1 /*
2 * rk_aiq_algo_anr_itf.c
3 *
4 * Copyright (c) 2019 Rockchip Corporation
5 *
6 * Licensed under the Apache License, Version 2.0 (the "License");
7 * you may not use this file except in compliance with the License.
8 * You may obtain a copy of the License at
9 *
10 * http://www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing, software
13 * distributed under the License is distributed on an "AS IS" BASIS,
14 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 * See the License for the specific language governing permissions and
16 * limitations under the License.
17 *
18 */
19
20 #include "aynr2/rk_aiq_aynr_algo_itf_v2.h"
21 #include "aynr2/rk_aiq_aynr_algo_v2.h"
22 #include "rk_aiq_algo_types.h"
23 #include "RkAiqCalibDbV2Helper.h"
24
25 RKAIQ_BEGIN_DECLARE
26
27 typedef struct _RkAiqAlgoContext {
28 Aynr_Context_V2_t AynrCtx;
29 } RkAiqAlgoContext;
30
31
32 static XCamReturn
create_context(RkAiqAlgoContext ** context,const AlgoCtxInstanceCfg * cfg)33 create_context(RkAiqAlgoContext **context, const AlgoCtxInstanceCfg* cfg)
34 {
35
36 XCamReturn result = XCAM_RETURN_NO_ERROR;
37 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
38
39 #if 1
40 Aynr_Context_V2_t* pAynrCtx = NULL;
41 #if (AYNR_USE_JSON_FILE_V2)
42 Aynr_result_t ret = Aynr_Init_V2(&pAynrCtx, cfg->calibv2);
43 #else
44 Aynr_result_t ret = Aynr_Init_V2(&pAynrCtx, cfg->calib);
45 #endif
46 if(ret != AYNR_RET_SUCCESS) {
47 result = XCAM_RETURN_ERROR_FAILED;
48 LOGE_ANR("%s: Initializaion ANR failed (%d)\n", __FUNCTION__, ret);
49 } else {
50 *context = (RkAiqAlgoContext *)(pAynrCtx);
51 }
52 #endif
53
54 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
55 return result;
56 }
57
58 static XCamReturn
destroy_context(RkAiqAlgoContext * context)59 destroy_context(RkAiqAlgoContext *context)
60 {
61 XCamReturn result = XCAM_RETURN_NO_ERROR;
62
63 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
64
65 #if 1
66 Aynr_Context_V2_t* pAynrCtx = (Aynr_Context_V2_t*)context;
67 Aynr_result_t ret = Aynr_Release_V2(pAynrCtx);
68 if(ret != AYNR_RET_SUCCESS) {
69 result = XCAM_RETURN_ERROR_FAILED;
70 LOGE_ANR("%s: release ANR failed (%d)\n", __FUNCTION__, ret);
71 }
72 #endif
73
74 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
75 return result;
76 }
77
78 static XCamReturn
prepare(RkAiqAlgoCom * params)79 prepare(RkAiqAlgoCom* params)
80 {
81 XCamReturn result = XCAM_RETURN_NO_ERROR;
82
83 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
84
85 Aynr_Context_V2_t* pAynrCtx = (Aynr_Context_V2_t *)params->ctx;
86 RkAiqAlgoConfigAynrV2* pCfgParam = (RkAiqAlgoConfigAynrV2*)params;
87 pAynrCtx->prepare_type = params->u.prepare.conf_type;
88
89 if(!!(params->u.prepare.conf_type & RK_AIQ_ALGO_CONFTYPE_UPDATECALIB )) {
90 #if AYNR_USE_JSON_FILE_V2
91 void *pCalibDbV2 = (void*)(pCfgParam->com.u.prepare.calibv2);
92 CalibDbV2_YnrV2_t *ynr_v2 = (CalibDbV2_YnrV2_t*)(CALIBDBV2_GET_MODULE_PTR((void*)pCalibDbV2, ynr_v2));
93 pAynrCtx->ynr_v2 = *ynr_v2;
94 #else
95 void *pCalibDb = (void*)(pCfgParam->com.u.prepare.calib);
96 pAynrCtx->list_ynr_v2 =
97 (struct list_head*)(CALIBDB_GET_MODULE_PTR((void*)pCalibDb, list_ynr_v2));
98 #endif
99 pAynrCtx->isIQParaUpdate = true;
100 pAynrCtx->isReCalculate |= 1;
101 }
102
103 Aynr_result_t ret = Aynr_Prepare_V2(pAynrCtx, &pCfgParam->stAynrConfig);
104 if(ret != AYNR_RET_SUCCESS) {
105 result = XCAM_RETURN_ERROR_FAILED;
106 LOGE_ANR("%s: config ANR failed (%d)\n", __FUNCTION__, ret);
107 }
108
109 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
110 return result;
111 }
112
113 static XCamReturn
pre_process(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)114 pre_process(const RkAiqAlgoCom* inparams, RkAiqAlgoResCom* outparams)
115 {
116 XCamReturn result = XCAM_RETURN_NO_ERROR;
117 bool oldGrayMode = false;
118
119 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
120 Aynr_Context_V2_t* pAynrCtx = (Aynr_Context_V2_t *)inparams->ctx;
121
122 RkAiqAlgoPreAynrV2* pAnrPreParams = (RkAiqAlgoPreAynrV2*)inparams;
123
124 oldGrayMode = pAynrCtx->isGrayMode;
125 if (pAnrPreParams->com.u.proc.gray_mode) {
126 pAynrCtx->isGrayMode = true;
127 } else {
128 pAynrCtx->isGrayMode = false;
129 }
130
131 if(oldGrayMode != pAynrCtx->isGrayMode) {
132 pAynrCtx->isReCalculate |= 1;
133 }
134
135 Aynr_result_t ret = Aynr_PreProcess_V2(pAynrCtx);
136 if(ret != AYNR_RET_SUCCESS) {
137 result = XCAM_RETURN_ERROR_FAILED;
138 LOGE_ANR("%s: ANRPreProcess failed (%d)\n", __FUNCTION__, ret);
139 }
140
141 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
142 return result;
143 }
144
145 static XCamReturn
processing(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)146 processing(const RkAiqAlgoCom* inparams, RkAiqAlgoResCom* outparams)
147 {
148 XCamReturn result = XCAM_RETURN_NO_ERROR;
149 int deltaIso = 0;
150 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
151
152 #if 1
153 RkAiqAlgoProcAynrV2* pAynrProcParams = (RkAiqAlgoProcAynrV2*)inparams;
154 RkAiqAlgoProcResAynrV2* pAynrProcResParams = (RkAiqAlgoProcResAynrV2*)outparams;
155 Aynr_Context_V2_t* pAynrCtx = (Aynr_Context_V2_t *)inparams->ctx;
156 Aynr_ExpInfo_t stExpInfo;
157 memset(&stExpInfo, 0x00, sizeof(Aynr_ExpInfo_t));
158
159 LOGD_ANR("%s:%d init:%d hdr mode:%d \n",
160 __FUNCTION__, __LINE__,
161 inparams->u.proc.init,
162 pAynrProcParams->hdr_mode);
163
164 stExpInfo.hdr_mode = 0; //pAnrProcParams->hdr_mode;
165 for(int i = 0; i < 3; i++) {
166 stExpInfo.arIso[i] = 50;
167 stExpInfo.arAGain[i] = 1.0;
168 stExpInfo.arDGain[i] = 1.0;
169 stExpInfo.arTime[i] = 0.01;
170 }
171
172 if(pAynrProcParams->hdr_mode == RK_AIQ_WORKING_MODE_NORMAL) {
173 stExpInfo.hdr_mode = 0;
174 } else if(pAynrProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_2_FRAME_HDR
175 || pAynrProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_2_LINE_HDR ) {
176 stExpInfo.hdr_mode = 1;
177 } else if(pAynrProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_3_FRAME_HDR
178 || pAynrProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_3_LINE_HDR ) {
179 stExpInfo.hdr_mode = 2;
180 }
181 stExpInfo.snr_mode = 0;
182
183 #if 0// TODO Merge:
184 XCamVideoBuffer* xCamAePreRes = pAynrProcParams->com.u.proc.res_comb->ae_pre_res;
185 RkAiqAlgoPreResAe* pAEPreRes = nullptr;
186 if (xCamAePreRes) {
187 // xCamAePreRes->ref(xCamAePreRes);
188 pAEPreRes = (RkAiqAlgoPreResAe*)xCamAePreRes->map(xCamAePreRes);
189 if (!pAEPreRes) {
190 LOGE_ANR("ae pre result is null");
191 } else {
192 LOGD_ANR("ae_pre_result: meanluma:%f time:%f gain:%f env_luma:%f \n",
193 pAEPreRes->ae_pre_res_rk.MeanLuma[0],
194 pAEPreRes->ae_pre_res_rk.LinearExp.exp_real_params.integration_time,
195 pAEPreRes->ae_pre_res_rk.LinearExp.exp_real_params.analog_gain,
196 pAEPreRes->ae_pre_res_rk.GlobalEnvLux);
197 }
198 // xCamAePreRes->unref(xCamAePreRes);
199 }
200 #endif
201
202 RKAiqAecExpInfo_t *curExp = pAynrProcParams->com.u.proc.curExp;
203 if(curExp != NULL) {
204 stExpInfo.snr_mode = curExp->CISFeature.SNR;
205 if(pAynrProcParams->hdr_mode == RK_AIQ_WORKING_MODE_NORMAL) {
206 stExpInfo.hdr_mode = 0;
207 stExpInfo.arAGain[0] = curExp->LinearExp.exp_real_params.analog_gain;
208 stExpInfo.arDGain[0] = curExp->LinearExp.exp_real_params.digital_gain;
209 stExpInfo.arTime[0] = curExp->LinearExp.exp_real_params.integration_time;
210 stExpInfo.arIso[0] = stExpInfo.arAGain[0] * stExpInfo.arDGain[0] * 50;
211 } else {
212 for(int i = 0; i < 3; i++) {
213 stExpInfo.arAGain[i] = curExp->HdrExp[i].exp_real_params.analog_gain;
214 stExpInfo.arDGain[i] = curExp->HdrExp[i].exp_real_params.digital_gain;
215 stExpInfo.arTime[i] = curExp->HdrExp[i].exp_real_params.integration_time;
216 stExpInfo.arIso[i] = stExpInfo.arAGain[i] * stExpInfo.arDGain[i] * 50;
217
218 LOGD_ANR("%s:%d index:%d again:%f dgain:%f time:%f iso:%d hdr_mode:%d\n",
219 __FUNCTION__, __LINE__,
220 i,
221 stExpInfo.arAGain[i],
222 stExpInfo.arDGain[i],
223 stExpInfo.arTime[i],
224 stExpInfo.arIso[i],
225 stExpInfo.hdr_mode);
226 }
227 }
228 } else {
229 LOGE_ANR("%s:%d curExp is NULL, so use default instead \n", __FUNCTION__, __LINE__);
230 }
231
232 #if 0
233 static int anr_cnt = 0;
234 anr_cnt++;
235
236 if(anr_cnt % 50 == 0) {
237 for(int i = 0; i < stExpInfo.hdr_mode + 1; i++) {
238 printf("%s:%d index:%d again:%f dgain:%f time:%f iso:%d hdr_mode:%d snr_mode:%d\n",
239 __FUNCTION__, __LINE__,
240 i,
241 stExpInfo.arAGain[i],
242 stExpInfo.arDGain[i],
243 stExpInfo.arTime[i],
244 stExpInfo.arIso[i],
245 stExpInfo.hdr_mode,
246 stExpInfo.snr_mode);
247 }
248 }
249 #endif
250
251 deltaIso = abs(stExpInfo.arIso[stExpInfo.hdr_mode] - pAynrCtx->stExpInfo.arIso[stExpInfo.hdr_mode]);
252 if(deltaIso > AYNRV2_RECALCULATE_DELTA_ISO) {
253 pAynrCtx->isReCalculate |= 1;
254 }
255
256 if(pAynrCtx->isReCalculate) {
257 Aynr_result_t ret = Aynr_Process_V2(pAynrCtx, &stExpInfo);
258 if(ret != AYNR_RET_SUCCESS) {
259 result = XCAM_RETURN_ERROR_FAILED;
260 LOGE_ANR("%s: processing ANR failed (%d)\n", __FUNCTION__, ret);
261 }
262
263 Aynr_GetProcResult_V2(pAynrCtx, &pAynrProcResParams->stAynrProcResult);
264 outparams->cfg_update = true;
265 LOGD_ANR("recalculate: %d delta_iso:%d \n ", pAynrCtx->isReCalculate, deltaIso);
266 } else {
267 outparams->cfg_update = false;
268 }
269 #endif
270
271 pAynrCtx->isReCalculate = 0;
272
273 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
274 return XCAM_RETURN_NO_ERROR;
275 }
276
277 static XCamReturn
post_process(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)278 post_process(const RkAiqAlgoCom* inparams, RkAiqAlgoResCom* outparams)
279 {
280 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
281
282 //nothing todo now
283
284 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
285 return XCAM_RETURN_NO_ERROR;
286 }
287
288 RkAiqAlgoDescription g_RkIspAlgoDescAynrV2 = {
289 .common = {
290 .version = RKISP_ALGO_AYNR_VERSION_V2,
291 .vendor = RKISP_ALGO_AYNR_VENDOR_V2,
292 .description = RKISP_ALGO_AYNR_DESCRIPTION_V2,
293 .type = RK_AIQ_ALGO_TYPE_AYNR,
294 .id = 0,
295 .create_context = create_context,
296 .destroy_context = destroy_context,
297 },
298 .prepare = prepare,
299 .pre_process = pre_process,
300 .processing = processing,
301 .post_process = post_process,
302 };
303
304 RKAIQ_END_DECLARE
305