1 /*
2  * Copyright (c) 2019-2022 Rockchip Eletronics Co., Ltd.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 #ifndef _RK_AIQ_BAYERTNR_HANDLE_INT_H_
17 #define _RK_AIQ_BAYERTNR_HANDLE_INT_H_
18 
19 #include "RkAiqHandle.h"
20 #include "abayertnr2/rk_aiq_uapi_abayertnr_int_v2.h"
21 #include "rk_aiq_api_private.h"
22 #include "rk_aiq_pool.h"
23 #include "xcam_mutex.h"
24 
25 namespace RkCam {
26 #if RKAIQ_HAVE_BAYERTNR_V2
27 
28 class RkAiqAbayertnrV2HandleInt : virtual public RkAiqHandle {
29 public:
RkAiqAbayertnrV2HandleInt(RkAiqAlgoDesComm * des,RkAiqCore * aiqCore)30     explicit RkAiqAbayertnrV2HandleInt(RkAiqAlgoDesComm* des, RkAiqCore* aiqCore)
31         : RkAiqHandle(des, aiqCore) {
32 #ifndef DISABLE_HANDLE_ATTRIB
33         updateStrength = false;
34         updateAtt      = false;
35         memset(&mCurStrength, 0x00, sizeof(mCurStrength));
36         memset(&mNewStrength, 0x00, sizeof(mNewStrength));
37         mCurStrength.percent = 1.0;
38         mNewStrength.percent = 1.0;
39         memset(&mCurAtt, 0x00, sizeof(mCurAtt));
40         memset(&mNewAtt, 0x00, sizeof(mNewAtt));
41 #endif
42     };
~RkAiqAbayertnrV2HandleInt()43     virtual ~RkAiqAbayertnrV2HandleInt() {
44         RkAiqHandle::deInit();
45     };
46     virtual XCamReturn updateConfig(bool needSync);
47     virtual XCamReturn prepare();
48     virtual XCamReturn preProcess();
49     virtual XCamReturn processing();
50     virtual XCamReturn postProcess();
51     virtual XCamReturn genIspResult(RkAiqFullParams* params, RkAiqFullParams* cur_params);
52     // TODO add algo specific methords, this is a sample
53     XCamReturn setAttrib(rk_aiq_bayertnr_attrib_v2_t* att);
54     XCamReturn getAttrib(rk_aiq_bayertnr_attrib_v2_t* att);
55     XCamReturn setStrength(rk_aiq_bayertnr_strength_v2_t* pStrength);
56     XCamReturn getStrength(rk_aiq_bayertnr_strength_v2_t* pStrength);
57     XCamReturn getInfo(rk_aiq_bayertnr_info_v2_t* pInfo);
58 
59 protected:
60     virtual void init();
deInit()61     virtual void deInit() {
62         RkAiqHandle::deInit();
63     };
64 
65 private:
66 #ifndef DISABLE_HANDLE_ATTRIB
67     // TODO
68     rk_aiq_bayertnr_attrib_v2_t mCurAtt;
69     rk_aiq_bayertnr_attrib_v2_t mNewAtt;
70     rk_aiq_bayertnr_strength_v2_t mCurStrength;
71     rk_aiq_bayertnr_strength_v2_t mNewStrength;
72     mutable std::atomic<bool> updateStrength;
73 #endif
74 
75 private:
76     DECLARE_HANDLE_REGISTER_TYPE(RkAiqAbayertnrV2HandleInt);
77 };
78 #endif
79 }  // namespace RkCam
80 
81 #endif
82