xref: /OK3568_Linux_fs/external/camera_engine_rkaiq/rkaiq/algos/asharp/rk_aiq_algo_asharp_itf.cpp (revision 4882a59341e53eb6f0b4789bf948001014eff981)
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 "asharp/rk_aiq_algo_asharp_itf.h"
21 #include "asharp/rk_aiq_asharp_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_ASHARP("%s: (enter)\n", __FUNCTION__ );
37 
38 #if 1
39     AsharpContext_t* pAsharpCtx = NULL;
40     #if(ASHARP_USE_JSON_PARA)
41     AsharpResult_t ret = AsharpInit_json(&pAsharpCtx, cfg->calibv2);
42     #else
43     AsharpResult_t ret = AsharpInit(&pAsharpCtx, cfg->calib);
44     #endif
45 
46     if(ret != ASHARP_RET_SUCCESS) {
47         result = XCAM_RETURN_ERROR_FAILED;
48         LOGE_ASHARP("%s: Initializaion Asharp failed (%d)\n", __FUNCTION__, ret);
49     } else {
50         *context = (RkAiqAlgoContext *)(pAsharpCtx);
51     }
52 #endif
53 
54     LOGI_ASHARP("%s: (exit)\n", __FUNCTION__ );
55     return result;
56 }
57 
58 static XCamReturn
destroy_context(RkAiqAlgoContext * context)59 destroy_context(RkAiqAlgoContext *context)
60 {
61     XCamReturn result = XCAM_RETURN_NO_ERROR;
62 
63     LOGI_ASHARP("%s: (enter)\n", __FUNCTION__ );
64 
65 #if 1
66     AsharpContext_t* pAsharpCtx = (AsharpContext_t*)context;
67     AsharpResult_t ret = AsharpRelease(pAsharpCtx);
68     if(ret != ASHARP_RET_SUCCESS) {
69         result = XCAM_RETURN_ERROR_FAILED;
70         LOGE_ASHARP("%s: release Asharp failed (%d)\n", __FUNCTION__, ret);
71     }
72 #endif
73 
74     LOGI_ASHARP("%s: (exit)\n", __FUNCTION__ );
75     return result;
76 }
77 
78 static XCamReturn
prepare(RkAiqAlgoCom * params)79 prepare(RkAiqAlgoCom* params)
80 {
81     XCamReturn result = XCAM_RETURN_NO_ERROR;
82 
83     LOGI_ASHARP("%s: (enter)\n", __FUNCTION__ );
84 
85 #if 1
86     AsharpContext_t* pAsharpCtx = (AsharpContext_t *)params->ctx;
87     RkAiqAlgoConfigAsharp* pCfgParam = (RkAiqAlgoConfigAsharp*)params;
88 	pAsharpCtx->prepare_type = params->u.prepare.conf_type;
89 
90 	if(!!(params->u.prepare.conf_type & RK_AIQ_ALGO_CONFTYPE_UPDATECALIB )){
91 		#if(ASHARP_USE_JSON_PARA)
92 		CalibDbV2_SharpV1_t *calibv2_sharp = (CalibDbV2_SharpV1_t *)(CALIBDBV2_GET_MODULE_PTR(pCfgParam->com.u.prepare.calibv2, sharp_v1));
93 		CalibDbV2_Edgefilter_t *calibv2_edgefilter = (CalibDbV2_Edgefilter_t*)(CALIBDBV2_GET_MODULE_PTR(pCfgParam->com.u.prepare.calibv2, edgefilter_v1));
94 		if(calibv2_sharp != NULL){
95 			sharp_calibdbV2_assign(&pAsharpCtx->sharp_v1, calibv2_sharp);
96 		}
97 		if(calibv2_edgefilter != NULL){
98 			edgefilter_calibdbV2_assign(&pAsharpCtx->edgefilter_v1, calibv2_edgefilter);
99 		}
100 		#else
101         pAsharpCtx->stSharpCalib = *(CalibDb_Sharp_2_t*)(CALIBDB_GET_MODULE_PTR(pCfgParam->com.u.prepare.calib, sharp));
102 		pAsharpCtx->stEdgeFltCalib = *(CalibDb_EdgeFilter_2_t*)(CALIBDB_GET_MODULE_PTR(pCfgParam->com.u.prepare.calib, edgeFilter));
103 		#endif
104 		pAsharpCtx->isIQParaUpdate = true;
105     }
106 
107     AsharpResult_t ret = AsharpPrepare(pAsharpCtx, &pCfgParam->stAsharpConfig);
108     if(ret != ASHARP_RET_SUCCESS) {
109         result = XCAM_RETURN_ERROR_FAILED;
110         LOGE_ASHARP("%s: config Asharp failed (%d)\n", __FUNCTION__, ret);
111     }
112 
113 #endif
114 
115     LOGI_ASHARP("%s: (exit)\n", __FUNCTION__ );
116     return result;
117 }
118 
119 static XCamReturn
pre_process(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)120 pre_process(const RkAiqAlgoCom* inparams, RkAiqAlgoResCom* outparams)
121 {
122     XCamReturn result = XCAM_RETURN_NO_ERROR;
123 
124     LOGI_ASHARP("%s: (enter)\n", __FUNCTION__ );
125 	AsharpContext_t* pAsharpCtx = (AsharpContext_t *)inparams->ctx;
126 
127 	RkAiqAlgoPreAsharp* pAsharpPreParams = (RkAiqAlgoPreAsharp*)inparams;
128 
129 	if (pAsharpPreParams->com.u.proc.gray_mode) {
130         pAsharpCtx->isGrayMode = true;
131     }else {
132         pAsharpCtx->isGrayMode = false;
133     }
134 
135     //nothing todo now
136     AsharpResult_t ret = AsharpPreProcess(pAsharpCtx);
137 	if(ret != ASHARP_RET_SUCCESS) {
138 		result = XCAM_RETURN_ERROR_FAILED;
139 		LOGE_ASHARP("%s: AsharpPreProcess failed (%d)\n", __FUNCTION__, ret);
140 	}
141 
142     LOGI_ASHARP("%s: (exit)\n", __FUNCTION__ );
143     return result;
144 }
145 
146 static XCamReturn
processing(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)147 processing(const RkAiqAlgoCom* inparams, RkAiqAlgoResCom* outparams)
148 {
149     XCamReturn result = XCAM_RETURN_NO_ERROR;
150 
151     LOGI_ASHARP("%s: (enter)\n", __FUNCTION__ );
152 
153 #if 1
154     RkAiqAlgoProcAsharp* pAsharpProcParams = (RkAiqAlgoProcAsharp*)inparams;
155     RkAiqAlgoProcResAsharp* pAsharpProcResParams = (RkAiqAlgoProcResAsharp*)outparams;
156     AsharpContext_t* pAsharpCtx = (AsharpContext_t *)inparams->ctx;
157     AsharpExpInfo_t stExpInfo;
158     memset(&stExpInfo, 0x00, sizeof(AsharpExpInfo_t));
159 
160     LOGD_ASHARP("%s:%d init:%d hdr mode:%d  \n",
161                 __FUNCTION__, __LINE__,
162                 inparams->u.proc.init,
163                 pAsharpProcParams->hdr_mode);
164 
165     stExpInfo.hdr_mode = 0; //pAnrProcParams->hdr_mode;
166     for(int i = 0; i < 3; i++) {
167         stExpInfo.arIso[i] = 50;
168         stExpInfo.arAGain[i] = 1.0;
169         stExpInfo.arDGain[i] = 1.0;
170         stExpInfo.arTime[i] = 0.01;
171     }
172 
173     if(pAsharpProcParams->hdr_mode == RK_AIQ_WORKING_MODE_NORMAL) {
174         stExpInfo.hdr_mode = 0;
175     } else if(pAsharpProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_2_FRAME_HDR
176               || pAsharpProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_2_LINE_HDR ) {
177         stExpInfo.hdr_mode = 1;
178     } else if(pAsharpProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_3_FRAME_HDR
179               || pAsharpProcParams->hdr_mode == RK_AIQ_ISP_HDR_MODE_3_LINE_HDR ) {
180         stExpInfo.hdr_mode = 2;
181     }
182 	stExpInfo.snr_mode = 0;
183 
184 #if 1
185     RKAiqAecExpInfo_t *preExp = pAsharpProcParams->com.u.proc.preExp;
186     RKAiqAecExpInfo_t *curExp = pAsharpProcParams->com.u.proc.curExp;
187     if(preExp != NULL && curExp != NULL) {
188         stExpInfo.cur_snr_mode = curExp->CISFeature.SNR;
189         stExpInfo.pre_snr_mode = preExp->CISFeature.SNR;
190         if(pAsharpProcParams->hdr_mode == RK_AIQ_WORKING_MODE_NORMAL) {
191             stExpInfo.hdr_mode = 0;
192             stExpInfo.arAGain[0] = curExp->LinearExp.exp_real_params.analog_gain;
193             stExpInfo.arDGain[0] = curExp->LinearExp.exp_real_params.digital_gain;
194             stExpInfo.arTime[0] = curExp->LinearExp.exp_real_params.integration_time;
195 			stExpInfo.arDcgMode[0] = curExp->LinearExp.exp_real_params.dcg_mode;
196             stExpInfo.arIso[0] = stExpInfo.arAGain[0] * stExpInfo.arDGain[0] * 50;
197 
198 			stExpInfo.preAGain[0] = preExp->LinearExp.exp_real_params.analog_gain;
199             stExpInfo.preDGain[0] = preExp->LinearExp.exp_real_params.digital_gain;
200             stExpInfo.preTime[0] = preExp->LinearExp.exp_real_params.integration_time;
201 			stExpInfo.preDcgMode[0] = preExp->LinearExp.exp_real_params.dcg_mode;
202             stExpInfo.preIso[0] = stExpInfo.preAGain[0] * stExpInfo.preDGain[0] * 50;
203             LOGD_ANR("asharp: %s-%d, preExp(%f, %f), curExp(%f, %f)\n",
204                     __FUNCTION__, __LINE__,
205                     preExp->LinearExp.exp_real_params.analog_gain,
206                     preExp->LinearExp.exp_real_params.integration_time,
207                     curExp->LinearExp.exp_real_params.analog_gain,
208                     curExp->LinearExp.exp_real_params.integration_time);
209         } else {
210             for(int i = 0; i < 3; i++) {
211                 stExpInfo.arAGain[i] = curExp->HdrExp[i].exp_real_params.analog_gain;
212                 stExpInfo.arDGain[i] = curExp->HdrExp[i].exp_real_params.digital_gain;
213                 stExpInfo.arTime[i] = curExp->HdrExp[i].exp_real_params.integration_time;
214 				stExpInfo.arDcgMode[i] = curExp->HdrExp[i].exp_real_params.dcg_mode;
215                 stExpInfo.arIso[i] = stExpInfo.arAGain[i] * stExpInfo.arDGain[i] * 50;
216 
217 				stExpInfo.preAGain[0] = preExp->HdrExp[i].exp_real_params.analog_gain;
218             	stExpInfo.preDGain[0] = preExp->HdrExp[i].exp_real_params.digital_gain;
219             	stExpInfo.preTime[0] = preExp->HdrExp[i].exp_real_params.integration_time;
220 				stExpInfo.preDcgMode[i] = preExp->HdrExp[i].exp_real_params.dcg_mode;
221             	stExpInfo.preIso[0] = stExpInfo.preAGain[0] * stExpInfo.preDGain[0] * 50;
222 
223                 LOGD_ASHARP("%s:%d index:%d again:%f dgain:%f time:%f iso:%d hdr_mode:%d\n",
224                             __FUNCTION__, __LINE__,
225                             i,
226                             stExpInfo.arAGain[i],
227                             stExpInfo.arDGain[i],
228                             stExpInfo.arTime[i],
229                             stExpInfo.arIso[i],
230                             stExpInfo.hdr_mode);
231             }
232         }
233     } else {
234         LOGE_ASHARP("%s:%d pAEPreRes is NULL, so use default instead \n", __FUNCTION__, __LINE__);
235     }
236 #endif
237 
238     AsharpResult_t ret = AsharpProcess(pAsharpCtx, &stExpInfo);
239     if(ret != ASHARP_RET_SUCCESS) {
240         result = XCAM_RETURN_ERROR_FAILED;
241         LOGE_ASHARP("%s: processing Asharp failed (%d)\n", __FUNCTION__, ret);
242     }
243 
244     AsharpGetProcResult(pAsharpCtx, &pAsharpProcResParams->stAsharpProcResult);
245 #endif
246 
247     LOGI_ASHARP("%s: (exit)\n", __FUNCTION__ );
248     return XCAM_RETURN_NO_ERROR;
249 }
250 
251 static XCamReturn
post_process(const RkAiqAlgoCom * inparams,RkAiqAlgoResCom * outparams)252 post_process(const RkAiqAlgoCom* inparams, RkAiqAlgoResCom* outparams)
253 {
254     LOGI_ASHARP("%s: (enter)\n", __FUNCTION__ );
255 
256     //nothing todo now
257 
258     LOGI_ASHARP("%s: (exit)\n", __FUNCTION__ );
259     return XCAM_RETURN_NO_ERROR;
260 }
261 
262 RkAiqAlgoDescription g_RkIspAlgoDescAsharp = {
263     .common = {
264         .version = RKISP_ALGO_ASHARP_VERSION,
265         .vendor  = RKISP_ALGO_ASHARP_VENDOR,
266         .description = RKISP_ALGO_ASHARP_DESCRIPTION,
267         .type    = RK_AIQ_ALGO_TYPE_ASHARP,
268         .id      = 0,
269         .create_context  = create_context,
270         .destroy_context = destroy_context,
271     },
272     .prepare = prepare,
273     .pre_process = pre_process,
274     .processing = processing,
275     .post_process = post_process,
276 };
277 
278 RKAIQ_END_DECLARE
279