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_abayertnr_itf_v2.h"
21 #include "rk_aiq_algo_camgroup_types.h"
22 #include "rk_aiq_types_camgroup_abayertnr_prvt_v2.h"
23 #include "abayertnr2/rk_aiq_abayertnr_algo_itf_v2.h"
24 #include "abayertnr2/rk_aiq_abayertnr_algo_v2.h"
25 
26 
27 
28 RKAIQ_BEGIN_DECLARE
29 
30 
groupAbayertnrV2CreateCtx(RkAiqAlgoContext ** context,const AlgoCtxInstanceCfg * cfg)31 static XCamReturn groupAbayertnrV2CreateCtx(RkAiqAlgoContext **context, const AlgoCtxInstanceCfg* cfg)
32 {
33     LOGI_ANR("%s enter \n", __FUNCTION__ );
34 
35     XCamReturn ret = XCAM_RETURN_NO_ERROR;
36     CamGroup_AbayertnrV2_Contex_t *abayertnr_group_contex = NULL;
37     AlgoCtxInstanceCfgCamGroup *cfgInt = (AlgoCtxInstanceCfgCamGroup*)cfg;
38 
39 
40     if(CHECK_ISP_HW_V30()) {
41         abayertnr_group_contex = (CamGroup_AbayertnrV2_Contex_t*)malloc(sizeof(CamGroup_AbayertnrV2_Contex_t));
42 #if ABAYERTNR_USE_JSON_FILE_V2
43         Abayertnr_result_V2_t ret_v2 = ABAYERTNRV2_RET_SUCCESS;
44         ret_v2 = Abayertnr_Init_V2(&(abayertnr_group_contex->abayertnr_contex_v2), (void *)cfgInt->s_calibv2);
45         if(ret_v2 != ABAYERTNRV2_RET_SUCCESS) {
46             ret = XCAM_RETURN_ERROR_FAILED;
47             LOGE_ANR("%s: Initializaion ANR failed (%d)\n", __FUNCTION__, ret);
48         }
49 #endif
50     } else {
51         ret = XCAM_RETURN_ERROR_FAILED;
52         LOGE_ANR("module_hw_version of abayertnr  is invalid!!!!");
53     }
54 
55     if(ret != XCAM_RETURN_NO_ERROR) {
56         LOGE_ANR("%s: Initializaion group bayertnr failed (%d)\n", __FUNCTION__, ret);
57     } else {
58         // to do got abayertnrSurrViewClib and initinal paras for for surround view
59         abayertnr_group_contex->group_CalibV2.groupMethod = CalibDbV2_CAMGROUP_ABAYERTNRV2_METHOD_MEAN;// to do from json
60         abayertnr_group_contex->camera_Num = cfgInt->camIdArrayLen;
61 
62         *context = (RkAiqAlgoContext *)(abayertnr_group_contex);
63 
64         LOGI_ANR("%s:%d surrViewMethod(1-mean):%d, cameraNum %d \n",
65                  __FUNCTION__, __LINE__,
66                  abayertnr_group_contex->group_CalibV2.groupMethod,
67                  abayertnr_group_contex->camera_Num);
68     }
69 
70     LOGI_ANR("%s exit ret:%d\n", __FUNCTION__, ret);
71     return ret;
72 
73 }
74 
groupAbayertnrV2DestroyCtx(RkAiqAlgoContext * context)75 static XCamReturn groupAbayertnrV2DestroyCtx(RkAiqAlgoContext *context)
76 {
77     LOGI_ANR("%s enter \n", __FUNCTION__ );
78 
79     XCamReturn ret = XCAM_RETURN_NO_ERROR;
80 
81     CamGroup_AbayertnrV2_Contex_t *abayertnr_group_contex = (CamGroup_AbayertnrV2_Contex_t*)context;
82 
83     if(CHECK_ISP_HW_V30()) {
84         Abayertnr_result_V2_t ret_v2 = ABAYERTNRV2_RET_SUCCESS;
85         ret_v2 = Abayertnr_Release_V2(abayertnr_group_contex->abayertnr_contex_v2);
86         if(ret_v2 != ABAYERTNRV2_RET_SUCCESS) {
87             ret = XCAM_RETURN_ERROR_FAILED;
88             LOGE_ANR("%s: Initializaion ANR failed (%d)\n", __FUNCTION__, ret);
89         }
90     }
91     else {
92         ret = XCAM_RETURN_ERROR_FAILED;
93         LOGE_ANR("module_hw_version of abayertnr is isvalid!!!!");
94     }
95 
96     if(ret != XCAM_RETURN_NO_ERROR) {
97         LOGE_ANR("%s: release abayertnr failed (%d)\n", __FUNCTION__, ret);
98     } else {
99         free(abayertnr_group_contex);
100     }
101 
102 
103     LOGI_ANR("%s exit ret:%d\n", __FUNCTION__, ret);
104     return ret;
105 }
106 
groupAbayertnrV2Prepare(RkAiqAlgoCom * params)107 static XCamReturn groupAbayertnrV2Prepare(RkAiqAlgoCom* params)
108 {
109     LOGI_ANR("%s enter \n", __FUNCTION__ );
110 
111     XCamReturn ret = XCAM_RETURN_NO_ERROR;
112 
113     CamGroup_AbayertnrV2_Contex_t * abayertnr_group_contex = (CamGroup_AbayertnrV2_Contex_t *)params->ctx;
114     RkAiqAlgoCamGroupPrepare* para = (RkAiqAlgoCamGroupPrepare*)params;
115 
116     if(CHECK_ISP_HW_V30()) {
117         Abayertnr_Context_V2_t * abayertnr_contex_v2 = abayertnr_group_contex->abayertnr_contex_v2;
118         abayertnr_contex_v2->prepare_type = params->u.prepare.conf_type;
119         if(!!(params->u.prepare.conf_type & RK_AIQ_ALGO_CONFTYPE_UPDATECALIB )) {
120             // todo  update calib pars for surround view
121 #if ABAYERTNR_USE_JSON_FILE_V2
122             void *pCalibdbV2 = (void*)(para->s_calibv2);
123             CalibDbV2_BayerTnrV2_t *bayertnr_v2 = (CalibDbV2_BayerTnrV2_t*)(CALIBDBV2_GET_MODULE_PTR((void*)pCalibdbV2, bayertnr_v2));
124             abayertnr_contex_v2->bayertnr_v2 = *bayertnr_v2;
125             abayertnr_contex_v2->isIQParaUpdate = true;
126             abayertnr_contex_v2->isReCalculate |= 1;
127 #endif
128         }
129         Abayertnr_Config_V2_t stAbayertnrConfigV2;
130         Abayertnr_result_V2_t ret_v2 = ABAYERTNRV2_RET_SUCCESS;
131         ret_v2 = Abayertnr_Prepare_V2(abayertnr_contex_v2, &stAbayertnrConfigV2);
132         if(ret_v2 != ABAYERTNRV2_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 abayertnr is isvalid!!!!");
140     }
141 
142     LOGI_ANR("%s exit ret:%d\n", __FUNCTION__, ret);
143     return ret;
144 }
145 
groupAbayertnrV2Processing(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)146 static XCamReturn groupAbayertnrV2Processing(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_AbayertnrV2_Contex_t * abayertnr_group_contex = (CamGroup_AbayertnrV2_Contex_t *)inparams->ctx;
155     int deltaIso = 0;
156 
157     //method error
158     if (abayertnr_group_contex->group_CalibV2.groupMethod <= CalibDbV2_CAMGROUP_ABAYERTNRV2_METHOD_MIN
159             ||  abayertnr_group_contex->group_CalibV2.groupMethod >=  CalibDbV2_CAMGROUP_ABAYERTNRV2_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     Abayertnr_ExpInfo_V2_t stExpInfoV2;
171     memset(&stExpInfoV2, 0x00, sizeof(stExpInfoV2));
172     stExpInfoV2.hdr_mode = 0; //pAnrProcParams->hdr_mode;
173     stExpInfoV2.snr_mode = 0;
174     for(int i = 0; i < 3; i++) {
175         stExpInfoV2.arIso[i] = 50;
176         stExpInfoV2.arAGain[i] = 1.0;
177         stExpInfoV2.arDGain[i] = 1.0;
178         stExpInfoV2.arTime[i] = 0.01;
179     }
180 
181 
182     //merge ae result, iso mean value
183     rk_aiq_singlecam_3a_result_t* scam_3a_res = procParaGroup->camgroupParmasArray[0];
184     if(scam_3a_res->aec._bEffAecExpValid) {
185         RKAiqAecExpInfo_t* pCurExp = &scam_3a_res->aec._effAecExpInfo;
186         stExpInfoV2.snr_mode = pCurExp->CISFeature.SNR;
187         if((rk_aiq_working_mode_t)procParaGroup->working_mode == RK_AIQ_WORKING_MODE_NORMAL) {
188             stExpInfoV2.hdr_mode = 0;
189             stExpInfoV2.arAGain[0] = pCurExp->LinearExp.exp_real_params.analog_gain;
190             stExpInfoV2.arDGain[0] = pCurExp->LinearExp.exp_real_params.digital_gain * pCurExp->LinearExp.exp_real_params.isp_dgain;
191             stExpInfoV2.arTime[0] = pCurExp->LinearExp.exp_real_params.integration_time;
192             stExpInfoV2.arIso[0] = stExpInfoV2.arAGain[0] * stExpInfoV2.arDGain[0] * 50;
193 
194         } else {
195             if(procParaGroup->working_mode == RK_AIQ_ISP_HDR_MODE_2_FRAME_HDR
196                     || procParaGroup->working_mode == RK_AIQ_ISP_HDR_MODE_2_LINE_HDR)
197                 stExpInfoV2.hdr_mode = 1;
198             else if (procParaGroup->working_mode == RK_AIQ_ISP_HDR_MODE_3_FRAME_HDR
199                      || procParaGroup->working_mode == RK_AIQ_ISP_HDR_MODE_3_LINE_HDR)
200                 stExpInfoV2.hdr_mode = 2;
201             else {
202                 stExpInfoV2.hdr_mode = 0;
203                 LOGE_ANR("mode error\n");
204             }
205 
206             for(int i = 0; i < 3; i++) {
207                 stExpInfoV2.arAGain[i] = pCurExp->HdrExp[i].exp_real_params.analog_gain;
208                 stExpInfoV2.arDGain[i] = pCurExp->HdrExp[i].exp_real_params.digital_gain;
209                 stExpInfoV2.arTime[i] = pCurExp->HdrExp[i].exp_real_params.integration_time;
210                 stExpInfoV2.arIso[i] = stExpInfoV2.arAGain[i] * stExpInfoV2.arDGain[i] * 50;
211             }
212 
213         }
214     } else {
215         LOGW("fail to get sensor gain form AE module,use default value ");
216     }
217 
218 
219 
220     if(CHECK_ISP_HW_V30()) {
221         Abayertnr_Context_V2_t * abayertnr_contex_v2 = abayertnr_group_contex->abayertnr_contex_v2;
222         Abayertnr_ProcResult_V2_t stAbayertnrResultV2;
223         RK_Bayertnr_Fix_V2_t st3DFix;
224         stAbayertnrResultV2.st3DFix = &st3DFix;
225 
226         deltaIso = abs(stExpInfoV2.arIso[stExpInfoV2.hdr_mode] - abayertnr_contex_v2->stExpInfo.arIso[stExpInfoV2.hdr_mode]);
227         if(deltaIso > ABAYERTNRV2_DELTA_ISO) {
228             abayertnr_contex_v2->isReCalculate |= 1;
229         }
230         if(abayertnr_contex_v2->isReCalculate) {
231             Abayertnr_result_V2_t ret_v2 = ABAYERTNRV2_RET_SUCCESS;
232             ret_v2 = Abayertnr_Process_V2(abayertnr_contex_v2, &stExpInfoV2);
233             if(ret_v2 != ABAYERTNRV2_RET_SUCCESS) {
234                 ret = XCAM_RETURN_ERROR_FAILED;
235                 LOGE_ANR("%s: processing ANR failed (%d)\n", __FUNCTION__, ret);
236             }
237             outparams->cfg_update = true;
238             LOGD_ANR("recalculate: %d delta_iso:%d \n ", abayertnr_contex_v2->isReCalculate, deltaIso);
239         } else {
240             outparams->cfg_update = false;
241         }
242         Abayertnr_GetProcResult_V2(abayertnr_contex_v2, &stAbayertnrResultV2);
243         for (int i = 0; i < procResParaGroup->arraySize; i++) {
244             *(procResParaGroup->camgroupParmasArray[i]->abayertnr._abayertnr_procRes_v2) = *stAbayertnrResultV2.st3DFix;
245             IS_UPDATE_MEM((procResParaGroup->camgroupParmasArray[i]->abayertnr._abayertnr_procRes_v2), procParaGroup->_offset_is_update) =
246                 outparams->cfg_update;
247         }
248         abayertnr_contex_v2->isReCalculate = 0;
249     }
250 
251 
252     else {
253         ret = XCAM_RETURN_ERROR_FAILED;
254         LOGE_ANR("module_hw_version of abayertnr is isvalid!!!!");
255     }
256 
257     LOGI_ANR("%s exit\n", __FUNCTION__);
258     return ret;
259 }
260 
261 RkAiqAlgoDescription g_RkIspAlgoDescCamgroupAbayertnrV2 = {
262     .common = {
263         .version = RKISP_ALGO_CAMGROUP_ABAYERTNRV2_VERSION,
264         .vendor  = RKISP_ALGO_CAMGROUP_ABAYERTNRV2_VENDOR,
265         .description = RKISP_ALGO_CAMGROUP_ABAYERTNRV2_DESCRIPTION,
266         .type    = RK_AIQ_ALGO_TYPE_AMFNR,
267         .id      = 0,
268         .create_context  = groupAbayertnrV2CreateCtx,
269         .destroy_context = groupAbayertnrV2DestroyCtx,
270     },
271     .prepare = groupAbayertnrV2Prepare,
272     .pre_process = NULL,
273     .processing = groupAbayertnrV2Processing,
274     .post_process = NULL,
275 };
276 
277 RKAIQ_END_DECLARE
278