1 /*
2  * rk_aiq_algo_camgroup_ablc_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_types.h"
21 #include "misc/rk_aiq_algo_camgroup_misc_itf.h"
22 #include "ablcV32/rk_aiq_ablc_algo_itf_v32.h"
23 #include "ablcV32/rk_aiq_ablc_algo_v32.h"
24 #include "ablcV32/rk_aiq_types_ablc_algo_prvt_v32.h"
25 
26 
27 RKAIQ_BEGIN_DECLARE
28 
29 
30 static XCamReturn
create_context(RkAiqAlgoContext ** context,const AlgoCtxInstanceCfg * cfg)31 create_context(RkAiqAlgoContext **context, const AlgoCtxInstanceCfg* cfg)
32 {
33     XCamReturn result = XCAM_RETURN_NO_ERROR;
34 
35     LOGI_ABLC("%s: (enter)\n", __FUNCTION__ );
36     AlgoCtxInstanceCfgCamGroup *cfgInt = (AlgoCtxInstanceCfgCamGroup*)cfg;
37 
38     AblcContext_V32_t* pAblcCtx = NULL;
39     AblcResult_V32_t ret = AblcV32Init(&pAblcCtx, (CamCalibDbV2Context_t*)(cfgInt->s_calibv2));//load json paras
40     if(ret != ABLC_V32_RET_SUCCESS) {
41         result = XCAM_RETURN_ERROR_FAILED;
42         LOGE_ABLC("%s: Initializaion Ablc failed (%d)\n", __FUNCTION__, ret);
43     } else {
44         *context = (RkAiqAlgoContext *)(pAblcCtx);
45     }
46 
47     LOGI_ABLC("%s: (exit)\n", __FUNCTION__ );
48     return result;
49 }
50 
51 static XCamReturn
destroy_context(RkAiqAlgoContext * context)52 destroy_context(RkAiqAlgoContext *context)
53 {
54     XCamReturn result = XCAM_RETURN_NO_ERROR;
55 
56     LOGI_ABLC("%s: (enter)\n", __FUNCTION__ );
57 
58 #if 1
59     AblcContext_V32_t* pAblcCtx = (AblcContext_V32_t*)context;
60     AblcResult_V32_t ret = AblcV32Release(pAblcCtx);
61     if(ret != ABLC_V32_RET_SUCCESS) {
62         result = XCAM_RETURN_ERROR_FAILED;
63         LOGE_ABLC("%s: release Ablc failed (%d)\n", __FUNCTION__, ret);
64     }
65 #endif
66 
67     LOGI_ABLC("%s: (exit)\n", __FUNCTION__ );
68     return result;
69 }
70 
71 static XCamReturn
prepare(RkAiqAlgoCom * params)72 prepare(RkAiqAlgoCom* params)
73 {
74     XCamReturn result = XCAM_RETURN_NO_ERROR;
75 
76     LOGD_ABLC("%s: (enter)\n", __FUNCTION__ );
77 
78     AblcContext_V32_t* pAblcCtx = (AblcContext_V32_t *)params->ctx;
79     RkAiqAlgoCamGroupPrepare* pCfgParam = (RkAiqAlgoCamGroupPrepare*)params;
80     pAblcCtx->prepare_type = params->u.prepare.conf_type;
81 
82     if(!!(params->u.prepare.conf_type & RK_AIQ_ALGO_CONFTYPE_UPDATECALIB )) {
83         CalibDbV2_Blc_V32_t* calibv2_ablc_calib =
84             (CalibDbV2_Blc_V32_t*)(CALIBDBV2_GET_MODULE_PTR((void*)(pCfgParam->s_calibv2), ablcV32_calib));
85         memcpy(&pAblcCtx->stBlcCalib, calibv2_ablc_calib, sizeof(CalibDbV2_Blc_V32_t));
86         pAblcCtx->isUpdateParam = true;
87         pAblcCtx->isReCalculate |= 1;
88     }
89 
90     LOGI_ABLC("%s: (exit)\n", __FUNCTION__ );
91     return result;
92 }
93 
94 static XCamReturn
processing(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)95 processing(const RkAiqAlgoCom* inparams, RkAiqAlgoResCom* outparams)
96 {
97     XCamReturn result = XCAM_RETURN_NO_ERROR;
98     int iso;
99     int delta_iso = 0;
100     LOGI_ABLC("%s: (enter)\n", __FUNCTION__ );
101 
102 #if 1
103     RkAiqAlgoCamGroupProcIn* procParaGroup = (RkAiqAlgoCamGroupProcIn*)inparams;
104     RkAiqAlgoCamGroupProcOut* procResParaGroup = (RkAiqAlgoCamGroupProcOut*)outparams;
105     AblcContext_V32_t* pAblcCtx = (AblcContext_V32_t *)inparams->ctx;
106     AblcExpInfo_V32_t stExpInfo;
107     AblcProc_V32_t   ablcV32_proc_res;
108 
109     memset(&stExpInfo, 0x00, sizeof(AblcExpInfo_t));
110 
111     LOGD_ABLC("%s:%d init:%d hdr mode:%d  \n",
112               __FUNCTION__, __LINE__,
113               inparams->u.proc.init,
114               procParaGroup->working_mode);
115 
116     stExpInfo.hdr_mode = 0; //pAnrProcParams->hdr_mode;
117     for(int i = 0; i < 3; i++) {
118         stExpInfo.arIso[i] = 50;
119         stExpInfo.arAGain[i] = 1.0;
120         stExpInfo.arDGain[i] = 1.0;
121         stExpInfo.arTime[i] = 0.01;
122         stExpInfo.isp_dgain[i] = 1.0;
123     }
124 
125     if(procParaGroup->working_mode == RK_AIQ_WORKING_MODE_NORMAL) {
126         stExpInfo.hdr_mode = 0;
127     } else if(procParaGroup->working_mode == RK_AIQ_ISP_HDR_MODE_2_FRAME_HDR
128               || procParaGroup->working_mode == RK_AIQ_ISP_HDR_MODE_2_LINE_HDR ) {
129         stExpInfo.hdr_mode = 1;
130     } else if(procParaGroup->working_mode == RK_AIQ_ISP_HDR_MODE_3_FRAME_HDR
131               || procParaGroup->working_mode == RK_AIQ_ISP_HDR_MODE_3_LINE_HDR ) {
132         stExpInfo.hdr_mode = 2;
133     }
134 
135     rk_aiq_singlecam_3a_result_t* scam_3a_res = procParaGroup->camgroupParmasArray[0];
136 
137     if(scam_3a_res->aec._bEffAecExpValid) {
138         RKAiqAecExpInfo_t* pAERes = &scam_3a_res->aec._effAecExpInfo;
139         if(procParaGroup->working_mode == RK_AIQ_WORKING_MODE_NORMAL) {
140             stExpInfo.arAGain[0] = pAERes->LinearExp.exp_real_params.analog_gain;
141             stExpInfo.arDGain[0] = pAERes->LinearExp.exp_real_params.digital_gain;
142             stExpInfo.arTime[0] = pAERes->LinearExp.exp_real_params.integration_time;
143             stExpInfo.isp_dgain[0] = pAERes->LinearExp.exp_real_params.isp_dgain;
144             stExpInfo.arIso[0] = stExpInfo.arAGain[0] * stExpInfo.arDGain[0] * 50;
145         } else {
146             for(int i = 0; i < 3; i++) {
147                 stExpInfo.arAGain[i] = pAERes->HdrExp[i].exp_real_params.analog_gain;
148                 stExpInfo.arDGain[i] = pAERes->HdrExp[i].exp_real_params.digital_gain;
149                 stExpInfo.arTime[i] = pAERes->HdrExp[i].exp_real_params.integration_time;
150                 stExpInfo.isp_dgain[i] = pAERes->HdrExp[i].exp_real_params.isp_dgain;
151                 stExpInfo.arIso[i] = stExpInfo.arAGain[i] * stExpInfo.arDGain[i] * 50;
152 
153                 LOGD_ABLC("%s:%d index:%d again:%f dgain:%f time:%f iso:%d hdr_mode:%d\n",
154                           __FUNCTION__, __LINE__,
155                           i,
156                           stExpInfo.arAGain[i],
157                           stExpInfo.arDGain[i],
158                           stExpInfo.arTime[i],
159                           stExpInfo.arIso[i],
160                           stExpInfo.hdr_mode);
161             }
162         }
163     } else {
164         LOGE_ABLC("%s:%d pAERes is NULL, so use default instead \n", __FUNCTION__, __LINE__);
165     }
166 
167     delta_iso = abs(stExpInfo.arIso[stExpInfo.hdr_mode] - pAblcCtx->stExpInfo.arIso[pAblcCtx->stExpInfo.hdr_mode]);
168     if(delta_iso > ABLC_V32_RECALCULATE_DELTE_ISO) {
169         pAblcCtx->isReCalculate |= 1;
170     }
171 
172 
173     if(pAblcCtx->isReCalculate ) {
174         AblcResult_V32_t ret = AblcV32Process(pAblcCtx, &stExpInfo);
175         if(ret != ABLC_V32_RET_SUCCESS) {
176             result = XCAM_RETURN_ERROR_FAILED;
177             LOGE_ABLC("%s: processing Ablc failed (%d)\n", __FUNCTION__, ret);
178         }
179         LOGD_ABLC("%s:%d processing ABLC recalculate delta_iso:%d \n", __FUNCTION__, __LINE__, delta_iso);
180     }
181 
182     Ablc_GetProcResult_V32(pAblcCtx, &ablcV32_proc_res);
183 
184     for (int i = 0; i < procResParaGroup->arraySize; i++) {
185         if (pAblcCtx->isReCalculate)
186             *(procResParaGroup->camgroupParmasArray[i]->ablc._blcConfig_v32) = ablcV32_proc_res;
187         //TODO
188         IS_UPDATE_MEM((procResParaGroup->camgroupParmasArray[i]->ablc._blcConfig_v32), procParaGroup->_offset_is_update) =
189             pAblcCtx->isReCalculate;
190     }
191 
192     pAblcCtx->isReCalculate = 0;
193 #if 0 //TODO: Is this necessary for group algo ?
194     //sensor blc setting
195     pAblcProcResParams->ablc_proc_res_com.SenDpccRes.enable = pAblcCtx->SenDpccRes.enable;
196     pAblcProcResParams->ablc_proc_res_com.SenDpccRes.total_blc = pAblcCtx->SenDpccRes.total_blc;
197     pAblcProcResParams->ablc_proc_res_com.SenDpccRes.cur_single_blc = pAblcCtx->SenDpccRes.cur_single_blc;
198     pAblcProcResParams->ablc_proc_res_com.SenDpccRes.cur_multiple_blc = pAblcCtx->SenDpccRes.cur_multiple_blc;
199 #endif
200 
201 #endif
202 
203     LOGI_ABLC("%s: (exit)\n", __FUNCTION__ );
204     return XCAM_RETURN_NO_ERROR;
205 }
206 
207 RkAiqAlgoDescription g_RkIspAlgoDescCamgroupAblcV32 = {
208     .common = {
209         .version = RKISP_ALGO_ABLC_V32_VERSION,
210         .vendor  = RKISP_ALGO_ABLC_V32_VENDOR,
211         .description = RKISP_ALGO_ABLC_V32_DESCRIPTION,
212         .type    = RK_AIQ_ALGO_TYPE_ABLC,
213         .id      = 0,
214         .create_context  = create_context,
215         .destroy_context = destroy_context,
216     },
217     .prepare = prepare,
218     .pre_process = NULL,
219     .processing = processing,
220     .post_process = NULL,
221 };
222 
223 RKAIQ_END_DECLARE
224