1 /*
2  * rk_aiq_algo_camgroup_awb_itf.c
3  *
4  *  Copyright (c) 2021 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 "rk_aiq_algo_camgroup_abayer2dnr_itf_v23.h"
21 #include "rk_aiq_algo_camgroup_types.h"
22 #include "rk_aiq_types_camgroup_abayer2dnr_prvt_v23.h"
23 #include "abayer2dnrV23/rk_aiq_abayer2dnr_algo_itf_v23.h"
24 #include "abayer2dnrV23/rk_aiq_abayer2dnr_algo_v23.h"
25 
26 
27 
28 RKAIQ_BEGIN_DECLARE
29 
30 #if RKAIQ_HAVE_BAYER2DNR_V23
groupAbayer2dnrV23CreateCtx(RkAiqAlgoContext ** context,const AlgoCtxInstanceCfg * cfg)31 static XCamReturn groupAbayer2dnrV23CreateCtx(RkAiqAlgoContext **context, const AlgoCtxInstanceCfg* cfg)
32 {
33     LOGI_ANR("%s enter \n", __FUNCTION__ );
34 
35     XCamReturn ret = XCAM_RETURN_NO_ERROR;
36     CamGroup_Abayer2dnrV23_Contex_t *abayernr_group_contex = NULL;
37     AlgoCtxInstanceCfgCamGroup *cfgInt = (AlgoCtxInstanceCfgCamGroup*)cfg;
38 
39 
40     if(CHECK_ISP_HW_V32() || CHECK_ISP_HW_V32_LITE()) {
41         abayernr_group_contex = (CamGroup_Abayer2dnrV23_Contex_t*)malloc(sizeof(CamGroup_Abayer2dnrV23_Contex_t));
42 #if ABAYER2DNR_USE_JSON_FILE_V23
43         Abayer2dnr_result_V23_t ret_v23 = ABAYER2DNR_V23_RET_SUCCESS;
44         ret_v23 = Abayer2dnr_Init_V23(&(abayernr_group_contex->abayer2dnr_contex_v23), (void *)cfgInt->s_calibv2);
45         if(ret_v23 != ABAYER2DNR_V23_RET_SUCCESS) {
46             ret = XCAM_RETURN_ERROR_FAILED;
47             LOGE_ANR("%s: Initializaion ANR failed (%d)\n", __FUNCTION__, ret);
48         }
49 #endif
50     }
51     else {
52         ret = XCAM_RETURN_ERROR_FAILED;
53         LOGE_ANR("module_hw_version of abayernr is invalid!!!!");
54     }
55 
56     if(ret != XCAM_RETURN_NO_ERROR) {
57         LOGE_ANR("%s: Initializaion group bayernr failed (%d)\n", __FUNCTION__, ret);
58     } else {
59         // to do got abayernrSurrViewClib and initinal paras for for surround view
60         abayernr_group_contex->group_CalibV23.groupMethod = CalibDbV2_CAMGROUP_ABAYER2DNRV23_METHOD_MEAN;// to do from json
61         abayernr_group_contex->camera_Num = cfgInt->camIdArrayLen;
62 
63         *context = (RkAiqAlgoContext *)(abayernr_group_contex);
64 
65         LOGI_ANR("%s:%d surrViewMethod(1-mean):%d, cameraNum %d \n",
66                  __FUNCTION__, __LINE__,
67                  abayernr_group_contex->group_CalibV23.groupMethod,
68                  abayernr_group_contex->camera_Num);
69     }
70 
71     LOGI_ANR("%s exit ret:%d\n", __FUNCTION__, ret);
72     return ret;
73 
74 }
75 
groupAbayer2dnrV23DestroyCtx(RkAiqAlgoContext * context)76 static XCamReturn groupAbayer2dnrV23DestroyCtx(RkAiqAlgoContext *context)
77 {
78     LOGI_ANR("%s enter \n", __FUNCTION__ );
79 
80     XCamReturn ret = XCAM_RETURN_NO_ERROR;
81 
82     CamGroup_Abayer2dnrV23_Contex_t *abayernr_group_contex = (CamGroup_Abayer2dnrV23_Contex_t*)context;
83 
84     if(CHECK_ISP_HW_V32() || CHECK_ISP_HW_V32_LITE()) {
85         Abayer2dnr_result_V23_t ret_v23 = ABAYER2DNR_V23_RET_SUCCESS;
86         ret_v23 = Abayer2dnr_Release_V23(abayernr_group_contex->abayer2dnr_contex_v23);
87         if(ret_v23 != ABAYER2DNR_V23_RET_SUCCESS) {
88             ret = XCAM_RETURN_ERROR_FAILED;
89             LOGE_ANR("%s: Initializaion ANR failed (%d)\n", __FUNCTION__, ret);
90         }
91     }
92     else {
93         ret = XCAM_RETURN_ERROR_FAILED;
94         LOGE_ANR("module_hw_version of abayer2dnr is isvalid!!!!");
95     }
96 
97     if(ret != XCAM_RETURN_NO_ERROR) {
98         LOGE_ANR("%s: release ANR failed (%d)\n", __FUNCTION__, ret);
99     } else {
100         free(abayernr_group_contex);
101     }
102 
103 
104     LOGI_ANR("%s exit ret:%d\n", __FUNCTION__, ret);
105     return ret;
106 }
107 
groupAbayer2dnrV23Prepare(RkAiqAlgoCom * params)108 static XCamReturn groupAbayer2dnrV23Prepare(RkAiqAlgoCom* params)
109 {
110     LOGI_ANR("%s enter \n", __FUNCTION__ );
111 
112     XCamReturn ret = XCAM_RETURN_NO_ERROR;
113 
114     CamGroup_Abayer2dnrV23_Contex_t * abayernr_group_contex = (CamGroup_Abayer2dnrV23_Contex_t *)params->ctx;
115     RkAiqAlgoCamGroupPrepare* para = (RkAiqAlgoCamGroupPrepare*)params;
116 
117     if(CHECK_ISP_HW_V32() || CHECK_ISP_HW_V32_LITE()) {
118         Abayer2dnr_Context_V23_t * abayer2dnr_contex_v23 = abayernr_group_contex->abayer2dnr_contex_v23;
119         if(!!(params->u.prepare.conf_type & RK_AIQ_ALGO_CONFTYPE_UPDATECALIB )) {
120             // todo  update calib pars for surround view
121 #if ABAYER2DNR_USE_JSON_FILE_V23
122             void *pCalibdbV2 = (void*)(para->s_calibv2);
123             CalibDbV2_Bayer2dnrV23_t *bayer2dnr_v23 = (CalibDbV2_Bayer2dnrV23_t*)(CALIBDBV2_GET_MODULE_PTR((void*)pCalibdbV2, bayer2dnr_v23));
124             abayer2dnr_contex_v23->bayernr_v23 = *bayer2dnr_v23;
125             abayer2dnr_contex_v23->isIQParaUpdate = true;
126             abayer2dnr_contex_v23->isReCalculate |= 1;
127 #endif
128         }
129         Abayer2dnr_Config_V23_t stAbayer2dnrConfigV23;
130         Abayer2dnr_result_V23_t ret_v23 = ABAYER2DNR_V23_RET_SUCCESS;
131         ret_v23 = Abayer2dnr_Prepare_V23(abayer2dnr_contex_v23, &stAbayer2dnrConfigV23);
132         if(ret_v23 != ABAYER2DNR_V23_RET_SUCCESS) {
133             ret = XCAM_RETURN_ERROR_FAILED;
134             LOGE_ANR("%s: config ANR failed (%d)\n", __FUNCTION__, ret);
135         }
136     }
137     else {
138         ret = XCAM_RETURN_ERROR_FAILED;
139         LOGE_ANR("module_hw_version of abayer2dnr is isvalid!!!!");
140     }
141 
142     LOGI_ANR("%s exit ret:%d\n", __FUNCTION__, ret);
143     return ret;
144 }
145 
groupAbayer2dnrV23Processing(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)146 static XCamReturn groupAbayer2dnrV23Processing(const RkAiqAlgoCom* inparams, RkAiqAlgoResCom* outparams)
147 {
148     LOGI_ANR("%s enter \n", __FUNCTION__ );
149     LOGI_ANR("----------------------------------------------frame_id (%d)----------------------------------------------\n", inparams->frame_id);
150 
151     XCamReturn ret = XCAM_RETURN_NO_ERROR;
152     RkAiqAlgoCamGroupProcIn* procParaGroup = (RkAiqAlgoCamGroupProcIn*)inparams;
153     RkAiqAlgoCamGroupProcOut* procResParaGroup = (RkAiqAlgoCamGroupProcOut*)outparams;
154     CamGroup_Abayer2dnrV23_Contex_t * abayernr_group_contex = (CamGroup_Abayer2dnrV23_Contex_t *)inparams->ctx;
155     int deltaIso = 0;
156 
157     //method error
158     if (abayernr_group_contex->group_CalibV23.groupMethod <= CalibDbV2_CAMGROUP_ABAYER2DNRV23_METHOD_MIN
159             ||  abayernr_group_contex->group_CalibV23.groupMethod >=  CalibDbV2_CAMGROUP_ABAYER2DNRV23_METHOD_MAX) {
160         return (ret);
161     }
162 
163     //group empty
164     if(procParaGroup->camgroupParmasArray == nullptr) {
165         LOGE_ANR("camgroupParmasArray is null");
166         return(XCAM_RETURN_ERROR_FAILED);
167     }
168 
169     //get cur ae exposure
170     Abayer2dnr_ExpInfo_V23_t stExpInfoV23;
171     memset(&stExpInfoV23, 0x00, sizeof(stExpInfoV23));
172     stExpInfoV23.hdr_mode = 0; //pAnrProcParams->hdr_mode;
173     stExpInfoV23.snr_mode = 0;
174     for(int i = 0; i < 3; i++) {
175         stExpInfoV23.arIso[i] = 50;
176         stExpInfoV23.arAGain[i] = 1.0;
177         stExpInfoV23.arDGain[i] = 1.0;
178         stExpInfoV23.arTime[i] = 0.01;
179     }
180 
181     stExpInfoV23.blc_ob_predgain = 1.0f;
182     if(procParaGroup != NULL) {
183         LOGD_ANR(" predgain:%f\n",
184                  procParaGroup->stAblcV32_proc_res.isp_ob_predgain);
185         stExpInfoV23.blc_ob_predgain = procParaGroup->stAblcV32_proc_res.isp_ob_predgain;
186 
187     }
188     //merge ae result, iso mean value
189     rk_aiq_singlecam_3a_result_t* scam_3a_res = procParaGroup->camgroupParmasArray[0];
190     if(scam_3a_res->aec._bEffAecExpValid) {
191         RKAiqAecExpInfo_t* pCurExp = &scam_3a_res->aec._effAecExpInfo;
192         stExpInfoV23.snr_mode = pCurExp->CISFeature.SNR;
193         if((rk_aiq_working_mode_t)procParaGroup->working_mode == RK_AIQ_WORKING_MODE_NORMAL) {
194             stExpInfoV23.hdr_mode = 0;
195             stExpInfoV23.arAGain[0] = pCurExp->LinearExp.exp_real_params.analog_gain;
196             stExpInfoV23.arDGain[0] = pCurExp->LinearExp.exp_real_params.digital_gain;
197             stExpInfoV23.arTime[0] = pCurExp->LinearExp.exp_real_params.integration_time;
198             if(stExpInfoV23.arAGain[0] < 1.0) {
199                 stExpInfoV23.arAGain[0] = 1.0;
200             }
201             if(stExpInfoV23.arDGain[0] < 1.0) {
202                 stExpInfoV23.arDGain[0] = 1.0;
203             }
204             float tmp_predgain = stExpInfoV23.blc_ob_predgain;
205             if(tmp_predgain < 1.0)
206                 tmp_predgain = 1.0;
207             stExpInfoV23.arIso[0] = stExpInfoV23.arAGain[0] * stExpInfoV23.arDGain[0] * tmp_predgain * 50;
208 
209         } else {
210             if(procParaGroup->working_mode == RK_AIQ_ISP_HDR_MODE_2_FRAME_HDR
211                     || procParaGroup->working_mode == RK_AIQ_ISP_HDR_MODE_2_LINE_HDR)
212                 stExpInfoV23.hdr_mode = 1;
213             else if (procParaGroup->working_mode == RK_AIQ_ISP_HDR_MODE_3_FRAME_HDR
214                      || procParaGroup->working_mode == RK_AIQ_ISP_HDR_MODE_3_LINE_HDR)
215                 stExpInfoV23.hdr_mode = 2;
216             else {
217                 stExpInfoV23.hdr_mode = 0;
218                 LOGE_ANR("mode error\n");
219             }
220 
221             for(int i = 0; i < 3; i++) {
222                 stExpInfoV23.arAGain[i] = pCurExp->HdrExp[i].exp_real_params.analog_gain;
223                 stExpInfoV23.arDGain[i] = pCurExp->HdrExp[i].exp_real_params.digital_gain;
224                 stExpInfoV23.arTime[i] = pCurExp->HdrExp[i].exp_real_params.integration_time;
225 
226                 if(stExpInfoV23.arAGain[i] < 1.0) {
227                     stExpInfoV23.arAGain[i] = 1.0;
228                 }
229                 if(stExpInfoV23.arDGain[i] < 1.0) {
230                     stExpInfoV23.arDGain[i] = 1.0;
231                 }
232                 stExpInfoV23.blc_ob_predgain = 1.0;
233 
234                 stExpInfoV23.arIso[i] = stExpInfoV23.arAGain[i] * stExpInfoV23.arDGain[i] * 50;
235             }
236 
237         }
238     } else {
239         LOGW("fail to get sensor gain form AE module,use default value ");
240     }
241 
242 
243 
244     if(CHECK_ISP_HW_V32() || CHECK_ISP_HW_V32_LITE()) {
245         Abayer2dnr_Context_V23_t * abayer2dnr_contex_v23 = abayernr_group_contex->abayer2dnr_contex_v23;
246         Abayer2dnr_ProcResult_V23_t stAbayer2dnrResultV23;
247         RK_Bayer2dnr_Fix_V23_t st2DFix;
248         stAbayer2dnrResultV23.st2DFix = &st2DFix;
249 
250         if(stExpInfoV23.blc_ob_predgain != abayer2dnr_contex_v23->stExpInfo.blc_ob_predgain) {
251             abayer2dnr_contex_v23->isReCalculate |= 1;
252         }
253         deltaIso = abs(stExpInfoV23.arIso[stExpInfoV23.hdr_mode] - abayer2dnr_contex_v23->stExpInfo.arIso[stExpInfoV23.hdr_mode]);
254         if(deltaIso > ABAYER2DNRV23_RECALCULATE_DELTA_ISO) {
255             abayer2dnr_contex_v23->isReCalculate |= 1;
256         }
257         if(abayer2dnr_contex_v23->isReCalculate) {
258             Abayer2dnr_result_V23_t ret_v23 = ABAYER2DNR_V23_RET_SUCCESS;
259             ret_v23 = Abayer2dnr_Process_V23(abayer2dnr_contex_v23, &stExpInfoV23);
260             if(ret_v23 != ABAYER2DNR_V23_RET_SUCCESS) {
261                 ret = XCAM_RETURN_ERROR_FAILED;
262                 LOGE_ANR("%s: processing ANR failed (%d)\n", __FUNCTION__, ret);
263             }
264             outparams->cfg_update = true;
265             LOGD_ANR("recalculate: %d delta_iso:%d \n ", abayer2dnr_contex_v23->isReCalculate, deltaIso);
266         } else {
267             outparams->cfg_update = false;
268         }
269         Abayer2dnr_GetProcResult_V23(abayer2dnr_contex_v23, &stAbayer2dnrResultV23);
270         for (int i = 0; i < procResParaGroup->arraySize; i++) {
271             *(procResParaGroup->camgroupParmasArray[i]->abayernr._abayer2dnr_procRes_v23) = *stAbayer2dnrResultV23.st2DFix;
272             IS_UPDATE_MEM((procResParaGroup->camgroupParmasArray[i]->abayernr._abayer2dnr_procRes_v23), procParaGroup->_offset_is_update) =
273                 outparams->cfg_update;
274         }
275         abayer2dnr_contex_v23->isReCalculate = 0;
276     }
277     else {
278         ret = XCAM_RETURN_ERROR_FAILED;
279         LOGE_ANR("module_hw_version of abayer2dnr is isvalid!!!!");
280     }
281 
282     LOGI_ANR("%s exit\n", __FUNCTION__);
283     return ret;
284 }
285 
286 RkAiqAlgoDescription g_RkIspAlgoDescCamgroupAbayer2dnrV23 = {
287     .common = {
288         .version = RKISP_ALGO_CAMGROUP_ABAYER2DNRV23_VERSION,
289         .vendor  = RKISP_ALGO_CAMGROUP_ABAYER2DNRV23_VENDOR,
290         .description = RKISP_ALGO_CAMGROUP_ABAYER2DNRV23_DESCRIPTION,
291         .type    = RK_AIQ_ALGO_TYPE_ARAWNR,
292         .id      = 0,
293         .create_context  = groupAbayer2dnrV23CreateCtx,
294         .destroy_context = groupAbayer2dnrV23DestroyCtx,
295     },
296     .prepare = groupAbayer2dnrV23Prepare,
297     .pre_process = NULL,
298     .processing = groupAbayer2dnrV23Processing,
299     .post_process = NULL,
300 };
301 #endif
302 
303 RKAIQ_END_DECLARE
304