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 "acnr2/rk_aiq_acnr_algo_itf_v2.h"
21 #include "acnr2/rk_aiq_acnr_algo_v2.h"
22 #include "rk_aiq_algo_types.h"
23
24 RKAIQ_BEGIN_DECLARE
25
26 typedef struct _RkAiqAlgoContext {
27 Acnr_Context_V2_t AcnrCtx;
28 } RkAiqAlgoContext;
29
30
31 static XCamReturn
create_context(RkAiqAlgoContext ** context,const AlgoCtxInstanceCfg * cfg)32 create_context(RkAiqAlgoContext **context, const AlgoCtxInstanceCfg* cfg)
33 {
34
35 XCamReturn result = XCAM_RETURN_NO_ERROR;
36 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
37
38 #if 1
39 Acnr_Context_V2_t* pAcnrCtx = NULL;
40 #if ACNR_USE_JSON_FILE_V2
41 AcnrV2_result_t ret = Acnr_Init_V2(&pAcnrCtx, cfg->calibv2);
42 #endif
43
44 if(ret != ACNRV2_RET_SUCCESS) {
45 result = XCAM_RETURN_ERROR_FAILED;
46 LOGE_ANR("%s: Initializaion ANR failed (%d)\n", __FUNCTION__, ret);
47 } else {
48 *context = (RkAiqAlgoContext *)(pAcnrCtx);
49 }
50 #endif
51
52 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
53 return result;
54 }
55
56 static XCamReturn
destroy_context(RkAiqAlgoContext * context)57 destroy_context(RkAiqAlgoContext *context)
58 {
59 XCamReturn result = XCAM_RETURN_NO_ERROR;
60
61 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
62
63 #if 1
64 Acnr_Context_V2_t* pAcnrCtx = (Acnr_Context_V2_t*)context;
65 AcnrV2_result_t ret = Acnr_Release_V2(pAcnrCtx);
66 if(ret != ACNRV2_RET_SUCCESS) {
67 result = XCAM_RETURN_ERROR_FAILED;
68 LOGE_ANR("%s: release ANR failed (%d)\n", __FUNCTION__, ret);
69 }
70 #endif
71
72 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
73 return result;
74 }
75
76 static XCamReturn
prepare(RkAiqAlgoCom * params)77 prepare(RkAiqAlgoCom* params)
78 {
79 XCamReturn result = XCAM_RETURN_NO_ERROR;
80
81 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
82
83 Acnr_Context_V2_t* pAcnrCtx = (Acnr_Context_V2_t *)params->ctx;
84 RkAiqAlgoConfigAcnrV2* pCfgParam = (RkAiqAlgoConfigAcnrV2*)params;
85 pAcnrCtx->prepare_type = params->u.prepare.conf_type;
86
87 if(!!(params->u.prepare.conf_type & RK_AIQ_ALGO_CONFTYPE_UPDATECALIB )) {
88 #if ACNR_USE_JSON_FILE_V2
89 void *pCalibDbV2 = (void*)(pCfgParam->com.u.prepare.calibv2);
90 CalibDbV2_CNRV2_t *cnr_v2 =
91 (CalibDbV2_CNRV2_t*)(CALIBDBV2_GET_MODULE_PTR((void*)pCalibDbV2, cnr_v2));
92 pAcnrCtx->cnr_v2 = *cnr_v2;
93 #endif
94 pAcnrCtx->isIQParaUpdate = true;
95 pAcnrCtx->isReCalculate |= 1;
96
97 }
98 AcnrV2_result_t ret = Acnr_Prepare_V2(pAcnrCtx, &pCfgParam->stAcnrConfig);
99 if(ret != ACNRV2_RET_SUCCESS) {
100 result = XCAM_RETURN_ERROR_FAILED;
101 LOGE_ANR("%s: config ANR failed (%d)\n", __FUNCTION__, ret);
102 }
103
104 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
105 return result;
106 }
107
108 static XCamReturn
pre_process(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)109 pre_process(const RkAiqAlgoCom* inparams, RkAiqAlgoResCom* outparams)
110 {
111 XCamReturn result = XCAM_RETURN_NO_ERROR;
112 bool oldGrayMode = false;
113
114 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
115 Acnr_Context_V2_t* pAcnrCtx = (Acnr_Context_V2_t *)inparams->ctx;
116
117 RkAiqAlgoPreAcnrV2* pAnrPreParams = (RkAiqAlgoPreAcnrV2*)inparams;
118
119 oldGrayMode = pAcnrCtx->isGrayMode;
120 if (pAnrPreParams->com.u.proc.gray_mode) {
121 pAcnrCtx->isGrayMode = true;
122 } else {
123 pAcnrCtx->isGrayMode = false;
124 }
125
126 if(oldGrayMode != pAcnrCtx->isGrayMode) {
127 pAcnrCtx->isReCalculate |= 1;
128 }
129
130 AcnrV2_result_t ret = Acnr_PreProcess_V2(pAcnrCtx);
131 if(ret != ACNRV2_RET_SUCCESS) {
132 result = XCAM_RETURN_ERROR_FAILED;
133 LOGE_ANR("%s: ANRPreProcess failed (%d)\n", __FUNCTION__, ret);
134 }
135
136 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
137 return result;
138 }
139
140 static XCamReturn
processing(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)141 processing(const RkAiqAlgoCom* inparams, RkAiqAlgoResCom* outparams)
142 {
143 XCamReturn result = XCAM_RETURN_NO_ERROR;
144 int DeltaISO = 0;
145 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
146
147 #if 1
148 RkAiqAlgoProcAcnrV2* pAcnrProcParams = (RkAiqAlgoProcAcnrV2*)inparams;
149 RkAiqAlgoProcResAcnrV2* pAcnrProcResParams = (RkAiqAlgoProcResAcnrV2*)outparams;
150 Acnr_Context_V2_t* pAcnrCtx = (Acnr_Context_V2_t *)inparams->ctx;
151 AcnrV2_ExpInfo_t stExpInfo;
152 memset(&stExpInfo, 0x00, sizeof(AcnrV2_ExpInfo_t));
153
154 LOGD_ANR("%s:%d init:%d hdr mode:%d \n",
155 __FUNCTION__, __LINE__,
156 inparams->u.proc.init,
157 pAcnrProcParams->hdr_mode);
158
159 stExpInfo.hdr_mode = 0; //pAnrProcParams->hdr_mode;
160 for(int i = 0; i < 3; i++) {
161 stExpInfo.arIso[i] = 50;
162 stExpInfo.arAGain[i] = 1.0;
163 stExpInfo.arDGain[i] = 1.0;
164 stExpInfo.arTime[i] = 0.01;
165 }
166
167 if(pAcnrProcParams->hdr_mode == RK_AIQ_WORKING_MODE_NORMAL) {
168 stExpInfo.hdr_mode = 0;
169 } else if(pAcnrProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_2_FRAME_HDR
170 || pAcnrProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_2_LINE_HDR ) {
171 stExpInfo.hdr_mode = 1;
172 } else if(pAcnrProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_3_FRAME_HDR
173 || pAcnrProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_3_LINE_HDR ) {
174 stExpInfo.hdr_mode = 2;
175 }
176 stExpInfo.snr_mode = 0;
177
178 #if 0// TODO Merge:
179 XCamVideoBuffer* xCamAePreRes = pAcnrProcParams->com.u.proc.res_comb->ae_pre_res;
180 RkAiqAlgoPreResAe* pAEPreRes = nullptr;
181 if (xCamAePreRes) {
182 // xCamAePreRes->ref(xCamAePreRes);
183 pAEPreRes = (RkAiqAlgoPreResAe*)xCamAePreRes->map(xCamAePreRes);
184 if (!pAEPreRes) {
185 LOGE_ANR("ae pre result is null");
186 } else {
187 }
188 // xCamAePreRes->unref(xCamAePreRes);
189 }
190 #endif
191
192
193 RKAiqAecExpInfo_t *curExp = pAcnrProcParams->com.u.proc.curExp;
194 if(curExp != NULL) {
195 stExpInfo.snr_mode = curExp->CISFeature.SNR;
196 if(pAcnrProcParams->hdr_mode == RK_AIQ_WORKING_MODE_NORMAL) {
197 stExpInfo.hdr_mode = 0;
198 if(curExp->LinearExp.exp_real_params.analog_gain < 1.0) {
199 stExpInfo.arAGain[0] = 1.0;
200 LOGW_ANR("leanr mode again is wrong, use 1.0 instead\n");
201 } else {
202 stExpInfo.arAGain[0] = curExp->LinearExp.exp_real_params.analog_gain;
203 }
204 if(curExp->LinearExp.exp_real_params.digital_gain < 1.0) {
205 stExpInfo.arDGain[0] = 1.0;
206 LOGW_ANR("leanr mode dgain is wrong, use 1.0 instead\n");
207 } else {
208 stExpInfo.arDGain[0] = curExp->LinearExp.exp_real_params.digital_gain;
209 }
210 if(curExp->LinearExp.exp_real_params.isp_dgain < 1.0) {
211 stExpInfo.arDGain[0] *= 1.0;
212 LOGW_ANR("leanr mode dgain is wrong, use 1.0 instead\n");
213 } else {
214 stExpInfo.arDGain[0] *= curExp->LinearExp.exp_real_params.isp_dgain;
215 }
216 stExpInfo.arTime[0] = curExp->LinearExp.exp_real_params.integration_time;
217 stExpInfo.arIso[0] = stExpInfo.arAGain[0] * stExpInfo.arDGain[0] * 50;
218 } else {
219 for(int i = 0; i < 3; i++) {
220 if(curExp->HdrExp[i].exp_real_params.analog_gain < 1.0) {
221 stExpInfo.arAGain[i] = 1.0;
222 LOGW_ANR("hdr mode again is wrong, use 1.0 instead\n");
223 } else {
224 stExpInfo.arAGain[i] = curExp->HdrExp[i].exp_real_params.analog_gain;
225 }
226 if(curExp->HdrExp[i].exp_real_params.digital_gain < 1.0) {
227 stExpInfo.arDGain[i] = 1.0;
228 } else {
229 LOGW_ANR("hdr mode dgain is wrong, use 1.0 instead\n");
230 stExpInfo.arDGain[i] = curExp->HdrExp[i].exp_real_params.digital_gain;
231 }
232 stExpInfo.arTime[i] = curExp->HdrExp[i].exp_real_params.integration_time;
233 stExpInfo.arIso[i] = stExpInfo.arAGain[i] * stExpInfo.arDGain[i] * 50;
234
235 LOGD_ANR("%s:%d index:%d again:%f dgain:%f time:%f iso:%d hdr_mode:%d\n",
236 __FUNCTION__, __LINE__,
237 i,
238 stExpInfo.arAGain[i],
239 stExpInfo.arDGain[i],
240 stExpInfo.arTime[i],
241 stExpInfo.arIso[i],
242 stExpInfo.hdr_mode);
243 }
244 }
245 } else {
246 LOGE_ANR("%s:%d curExp is NULL, so use default instead \n", __FUNCTION__, __LINE__);
247 }
248
249 #if 0
250 static int anr_cnt = 0;
251 anr_cnt++;
252
253 if(anr_cnt % 1 == 0) {
254 for(int i = 0; i < stExpInfo.hdr_mode + 1; i++) {
255 printf("%s:%d index:%d again:%f dgain:%f time:%f iso:%d hdr_mode:%d snr_mode:%d\n",
256 __FUNCTION__, __LINE__,
257 i,
258 stExpInfo.arAGain[i],
259 stExpInfo.arDGain[i],
260 stExpInfo.arTime[i],
261 stExpInfo.arIso[i],
262 stExpInfo.hdr_mode,
263 stExpInfo.snr_mode);
264 }
265 }
266 #endif
267
268 DeltaISO = abs(stExpInfo.arIso[stExpInfo.hdr_mode] - pAcnrCtx->stExpInfo.arIso[pAcnrCtx->stExpInfo.hdr_mode]);
269 if(DeltaISO > ACNRV2_RECALCULATE_DELTA_ISO) {
270 pAcnrCtx->isReCalculate |= 1;
271 }
272
273 if(pAcnrCtx->isReCalculate) {
274 AcnrV2_result_t ret = Acnr_Process_V2(pAcnrCtx, &stExpInfo);
275 if(ret != ACNRV2_RET_SUCCESS) {
276 result = XCAM_RETURN_ERROR_FAILED;
277 LOGE_ANR("%s: processing ANR failed (%d)\n", __FUNCTION__, ret);
278 }
279
280 Acnr_GetProcResult_V2(pAcnrCtx, &pAcnrProcResParams->stAcnrProcResult);
281 outparams->cfg_update = true;
282 LOGD_ANR("recalculate: %d delta_iso:%d \n ", pAcnrCtx->isReCalculate, DeltaISO);
283 } else {
284 outparams->cfg_update = false;
285 }
286 #endif
287
288 pAcnrCtx->isReCalculate = 0;
289 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
290 return XCAM_RETURN_NO_ERROR;
291 }
292
293 static XCamReturn
post_process(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)294 post_process(const RkAiqAlgoCom* inparams, RkAiqAlgoResCom* outparams)
295 {
296 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
297
298 //nothing todo now
299
300 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
301 return XCAM_RETURN_NO_ERROR;
302 }
303
304 RkAiqAlgoDescription g_RkIspAlgoDescAcnrV2 = {
305 .common = {
306 .version = RKISP_ALGO_ACNR_VERSION_V2,
307 .vendor = RKISP_ALGO_ACNR_VENDOR_V2,
308 .description = RKISP_ALGO_ACNR_DESCRIPTION_V2,
309 .type = RK_AIQ_ALGO_TYPE_ACNR,
310 .id = 0,
311 .create_context = create_context,
312 .destroy_context = destroy_context,
313 },
314 .prepare = prepare,
315 .pre_process = pre_process,
316 .processing = processing,
317 .post_process = post_process,
318 };
319
320 RKAIQ_END_DECLARE
321