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 "anr/rk_aiq_algo_anr_itf.h"
21 #include "anr/rk_aiq_anr_algo.h"
22 #include "rk_aiq_algo_types.h"
23
24 RKAIQ_BEGIN_DECLARE
25
26 typedef struct _RkAiqAlgoContext {
27 void* place_holder[0];
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 ANRContext_t* pAnrCtx = NULL;
40 #if(ANR_USE_JSON_PARA)
41 ANRresult_t ret = ANRInit_json(&pAnrCtx, cfg->calibv2);
42 #else
43 ANRresult_t ret = ANRInit(&pAnrCtx, cfg->calib);
44 #endif
45 if(ret != ANR_RET_SUCCESS) {
46 result = XCAM_RETURN_ERROR_FAILED;
47 LOGE_ANR("%s: Initializaion ANR failed (%d)\n", __FUNCTION__, ret);
48 } else {
49 *context = (RkAiqAlgoContext *)(pAnrCtx);
50 }
51 #endif
52
53 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
54 return result;
55 }
56
57 static XCamReturn
destroy_context(RkAiqAlgoContext * context)58 destroy_context(RkAiqAlgoContext *context)
59 {
60 XCamReturn result = XCAM_RETURN_NO_ERROR;
61
62 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
63
64 #if 1
65 ANRContext_t* pAnrCtx = (ANRContext_t*)context;
66 ANRresult_t ret = ANRRelease(pAnrCtx);
67 if(ret != ANR_RET_SUCCESS) {
68 result = XCAM_RETURN_ERROR_FAILED;
69 LOGE_ANR("%s: release ANR failed (%d)\n", __FUNCTION__, ret);
70 }
71 #endif
72
73 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
74 return result;
75 }
76
77 static XCamReturn
prepare(RkAiqAlgoCom * params)78 prepare(RkAiqAlgoCom* params)
79 {
80 XCamReturn result = XCAM_RETURN_NO_ERROR;
81
82 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
83
84 ANRContext_t* pAnrCtx = (ANRContext_t *)params->ctx;
85 RkAiqAlgoConfigAnr* pCfgParam = (RkAiqAlgoConfigAnr*)params;
86 pAnrCtx->prepare_type = params->u.prepare.conf_type;
87
88 if(!!(params->u.prepare.conf_type & RK_AIQ_ALGO_CONFTYPE_UPDATECALIB )){
89 #if(ANR_USE_JSON_PARA)
90 void *pCalibDbV2 = (void*)(pCfgParam->com.u.prepare.calibv2);
91 CalibDbV2_BayerNrV1_t *bayernr_v1 = (CalibDbV2_BayerNrV1_t*)(CALIBDBV2_GET_MODULE_PTR((void*)pCalibDbV2, bayernr_v1));
92 bayernr_calibdbV2_assign(&pAnrCtx->bayernr_v1, bayernr_v1);
93 CalibDbV2_MFNR_t *mfnr_v1 = (CalibDbV2_MFNR_t*)(CALIBDBV2_GET_MODULE_PTR((void*)pCalibDbV2, mfnr_v1));
94 mfnr_calibdbV2_assign(&pAnrCtx->mfnr_v1, mfnr_v1);
95 CalibDbV2_YnrV1_t *ynr_v1 = (CalibDbV2_YnrV1_t*)(CALIBDBV2_GET_MODULE_PTR((void*)pCalibDbV2, ynr_v1));
96 ynr_calibdbV2_assign(&pAnrCtx->ynr_v1, ynr_v1);
97 CalibDbV2_UVNR_t *uvnr_v1 = (CalibDbV2_UVNR_t*)(CALIBDBV2_GET_MODULE_PTR((void*)pCalibDbV2, uvnr_v1));
98 uvnr_calibdbV2_assign(&pAnrCtx->uvnr_v1, uvnr_v1);
99 #else
100 void *pCalibDb = (void*)(pCfgParam->com.u.prepare.calib);
101 pAnrCtx->stBayernrCalib =
102 *(CalibDb_BayerNr_2_t*)(CALIBDB_GET_MODULE_PTR((void*)pCalibDb, bayerNr));
103 pAnrCtx->stUvnrCalib =
104 *(CalibDb_UVNR_2_t*)(CALIBDB_GET_MODULE_PTR((void*)pCalibDb, uvnr));
105 pAnrCtx->stYnrCalib =
106 *(CalibDb_YNR_2_t*)(CALIBDB_GET_MODULE_PTR((void*)pCalibDb, ynr));
107 pAnrCtx->stMfnrCalib =
108 *(CalibDb_MFNR_2_t*)(CALIBDB_GET_MODULE_PTR((void*)pCalibDb, mfnr));
109 #endif
110 pAnrCtx->isIQParaUpdate = true;
111 }
112
113 ANRresult_t ret = ANRPrepare(pAnrCtx, &pCfgParam->stANRConfig);
114 if(ret != ANR_RET_SUCCESS) {
115 result = XCAM_RETURN_ERROR_FAILED;
116 LOGE_ANR("%s: config ANR failed (%d)\n", __FUNCTION__, ret);
117 }
118
119 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
120 return result;
121 }
122
123 static XCamReturn
pre_process(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)124 pre_process(const RkAiqAlgoCom* inparams, RkAiqAlgoResCom* outparams)
125 {
126 XCamReturn result = XCAM_RETURN_NO_ERROR;
127
128 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
129 ANRContext_t* pAnrCtx = (ANRContext_t *)inparams->ctx;
130
131 RkAiqAlgoPreAnr* pAnrPreParams = (RkAiqAlgoPreAnr*)inparams;
132
133 if (pAnrPreParams->com.u.proc.gray_mode) {
134 pAnrCtx->isGrayMode = true;
135 }else {
136 pAnrCtx->isGrayMode = false;
137 }
138
139 ANRresult_t ret = ANRPreProcess(pAnrCtx);
140 if(ret != ANR_RET_SUCCESS) {
141 result = XCAM_RETURN_ERROR_FAILED;
142 LOGE_ANR("%s: ANRPreProcess failed (%d)\n", __FUNCTION__, ret);
143 }
144
145 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
146 return result;
147 }
148
149 static XCamReturn
processing(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)150 processing(const RkAiqAlgoCom* inparams, RkAiqAlgoResCom* outparams)
151 {
152 XCamReturn result = XCAM_RETURN_NO_ERROR;
153
154 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
155
156 #if 1
157 RkAiqAlgoProcAnr* pAnrProcParams = (RkAiqAlgoProcAnr*)inparams;
158 RkAiqAlgoProcResAnr* pAnrProcResParams = (RkAiqAlgoProcResAnr*)outparams;
159 ANRContext_t* pAnrCtx = (ANRContext_t *)inparams->ctx;
160 ANRExpInfo_t stExpInfo;
161 memset(&stExpInfo, 0x00, sizeof(ANRExpInfo_t));
162
163 LOGD_ANR("%s:%d init:%d hdr mode:%d \n",
164 __FUNCTION__, __LINE__,
165 inparams->u.proc.init,
166 pAnrProcParams->hdr_mode);
167
168 stExpInfo.hdr_mode = 0; //pAnrProcParams->hdr_mode;
169 for(int i = 0; i < 3; i++) {
170 stExpInfo.arIso[i] = 50;
171 stExpInfo.arAGain[i] = 1.0;
172 stExpInfo.arDGain[i] = 1.0;
173 stExpInfo.arTime[i] = 0.01;
174 }
175
176 if(pAnrProcParams->hdr_mode == RK_AIQ_WORKING_MODE_NORMAL) {
177 stExpInfo.hdr_mode = 0;
178 } else if(pAnrProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_2_FRAME_HDR
179 || pAnrProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_2_LINE_HDR ) {
180 stExpInfo.hdr_mode = 1;
181 } else if(pAnrProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_3_FRAME_HDR
182 || pAnrProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_3_LINE_HDR ) {
183 stExpInfo.hdr_mode = 2;
184 }
185 stExpInfo.snr_mode = 0;
186
187 #if 1
188 RKAiqAecExpInfo_t *preExp = pAnrProcParams->com.u.proc.preExp;
189 RKAiqAecExpInfo_t *curExp = pAnrProcParams->com.u.proc.curExp;
190
191 if(preExp != NULL && curExp != NULL) {
192 stExpInfo.cur_snr_mode = curExp->CISFeature.SNR;
193 stExpInfo.pre_snr_mode = preExp->CISFeature.SNR;
194 if(pAnrProcParams->hdr_mode == RK_AIQ_WORKING_MODE_NORMAL) {
195 stExpInfo.hdr_mode = 0;
196 stExpInfo.arAGain[0] = curExp->LinearExp.exp_real_params.analog_gain;
197 stExpInfo.arDGain[0] = curExp->LinearExp.exp_real_params.digital_gain;
198 stExpInfo.arTime[0] = curExp->LinearExp.exp_real_params.integration_time;
199 stExpInfo.arDcgMode[0] = curExp->LinearExp.exp_real_params.dcg_mode;
200 stExpInfo.arIso[0] = stExpInfo.arAGain[0] * stExpInfo.arDGain[0] * 50;
201
202 stExpInfo.preAGain[0] = preExp->LinearExp.exp_real_params.analog_gain;
203 stExpInfo.preDGain[0] = preExp->LinearExp.exp_real_params.digital_gain;
204 stExpInfo.preTime[0] = preExp->LinearExp.exp_real_params.integration_time;
205 stExpInfo.preDcgMode[0] = preExp->LinearExp.exp_real_params.dcg_mode;
206 stExpInfo.preIso[0] = stExpInfo.preAGain[0] * stExpInfo.preDGain[0] * 50;
207 LOGD_ANR("anr: %s-%d, preExp(%f, %f, %f, %d, %d), curExp(%f, %f, %f, %d, %d)\n",
208 __FUNCTION__, __LINE__,
209 preExp->LinearExp.exp_real_params.analog_gain,
210 preExp->LinearExp.exp_real_params.integration_time,
211 preExp->LinearExp.exp_real_params.digital_gain,
212 preExp->LinearExp.exp_real_params.dcg_mode,
213 preExp->CISFeature.SNR,
214 curExp->LinearExp.exp_real_params.analog_gain,
215 curExp->LinearExp.exp_real_params.integration_time,
216 curExp->LinearExp.exp_real_params.digital_gain,
217 curExp->LinearExp.exp_real_params.dcg_mode,
218 curExp->CISFeature.SNR);
219 } else {
220 for(int i = 0; i < 3; i++) {
221 stExpInfo.arAGain[i] = curExp->HdrExp[i].exp_real_params.analog_gain,
222 stExpInfo.arDGain[i] = curExp->HdrExp[i].exp_real_params.digital_gain;
223 stExpInfo.arTime[i] = curExp->HdrExp[i].exp_real_params.integration_time;
224 stExpInfo.arDcgMode[i] = curExp->HdrExp[i].exp_real_params.dcg_mode;
225 stExpInfo.arIso[i] = stExpInfo.arAGain[i] * stExpInfo.arDGain[i] * 50;
226
227 stExpInfo.preAGain[i] = preExp->HdrExp[i].exp_real_params.analog_gain,
228 stExpInfo.preDGain[i] = preExp->HdrExp[i].exp_real_params.digital_gain;
229 stExpInfo.preTime[i] = preExp->HdrExp[i].exp_real_params.integration_time;
230 stExpInfo.preDcgMode[i] = preExp->HdrExp[i].exp_real_params.dcg_mode;
231 stExpInfo.preIso[i] = stExpInfo.preAGain[i] * stExpInfo.preDGain[i] * 50;
232
233 LOGD_ANR("%s:%d index:%d again:%f %f dgain:%f %f time:%f %f iso:%d %d hdr_mode:%d \n",
234 __FUNCTION__, __LINE__,
235 i,
236 stExpInfo.preAGain[i], stExpInfo.arAGain[i],
237 stExpInfo.preDGain[i], stExpInfo.arDGain[i],
238 stExpInfo.preTime[i], stExpInfo.arTime[i],
239 stExpInfo.preIso[i], stExpInfo.arIso[i],
240 stExpInfo.hdr_mode);
241 }
242 }
243 } else {
244 LOGE_ANR("%s:%d preExp(%p) or curExp(%p) is NULL, so use default instead \n",
245 __FUNCTION__, __LINE__, preExp, curExp);
246 }
247
248 #if 0
249 static int anr_cnt = 0;
250 anr_cnt++;
251
252 if(anr_cnt % 50 == 0) {
253 for(int i = 0; i < stExpInfo.hdr_mode + 1; i++) {
254 printf("%s:%d index:%d again:%f dgain:%f time:%f iso:%d hdr_mode:%d\n",
255 __FUNCTION__, __LINE__,
256 i,
257 stExpInfo.arAGain[i],
258 stExpInfo.arDGain[i],
259 stExpInfo.arTime[i],
260 stExpInfo.arIso[i],
261 stExpInfo.hdr_mode);
262 }
263 }
264 #endif
265
266
267 #endif
268
269 ANRresult_t ret = ANRProcess(pAnrCtx, &stExpInfo);
270 if(ret != ANR_RET_SUCCESS) {
271 result = XCAM_RETURN_ERROR_FAILED;
272 LOGE_ANR("%s: processing ANR failed (%d)\n", __FUNCTION__, ret);
273 }
274
275 ANRGetProcResult(pAnrCtx, &pAnrProcResParams->stAnrProcResult);
276 #endif
277
278 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
279 return XCAM_RETURN_NO_ERROR;
280 }
281
282 static XCamReturn
post_process(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)283 post_process(const RkAiqAlgoCom* inparams, RkAiqAlgoResCom* outparams)
284 {
285 LOGI_ANR("%s: (enter)\n", __FUNCTION__ );
286
287 //nothing todo now
288
289 LOGI_ANR("%s: (exit)\n", __FUNCTION__ );
290 return XCAM_RETURN_NO_ERROR;
291 }
292
293 RkAiqAlgoDescription g_RkIspAlgoDescAnr = {
294 .common = {
295 .version = RKISP_ALGO_ANR_VERSION,
296 .vendor = RKISP_ALGO_ANR_VENDOR,
297 .description = RKISP_ALGO_ANR_DESCRIPTION,
298 .type = RK_AIQ_ALGO_TYPE_ANR,
299 .id = 0,
300 .create_context = create_context,
301 .destroy_context = destroy_context,
302 },
303 .prepare = prepare,
304 .pre_process = pre_process,
305 .processing = processing,
306 .post_process = post_process,
307 };
308
309 RKAIQ_END_DECLARE
310