/*
 * Copyright (c) 2019-2022 Rockchip Eletronics Co., Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#include "RkAiqAtmoHandle.h"

#include "RkAiqCore.h"

namespace RkCam {

DEFINE_HANDLE_REGISTER_TYPE(RkAiqAtmoHandleInt);

void RkAiqAtmoHandleInt::init() {
    ENTER_ANALYZER_FUNCTION();

    RkAiqHandle::deInit();
    mConfig       = (RkAiqAlgoCom*)(new RkAiqAlgoConfigAtmo());
    mPreInParam   = (RkAiqAlgoCom*)(new RkAiqAlgoPreAtmo());
    mPreOutParam  = (RkAiqAlgoResCom*)(new RkAiqAlgoPreResAtmo());
    mProcInParam  = (RkAiqAlgoCom*)(new RkAiqAlgoProcAtmo());
    mProcOutParam = (RkAiqAlgoResCom*)(new RkAiqAlgoProcResAtmo());
    mPostInParam  = (RkAiqAlgoCom*)(new RkAiqAlgoPostAtmo());
    mPostOutParam = (RkAiqAlgoResCom*)(new RkAiqAlgoPostResAtmo());

    EXIT_ANALYZER_FUNCTION();
}
XCamReturn RkAiqAtmoHandleInt::updateConfig(bool needSync) {
    ENTER_ANALYZER_FUNCTION();

    XCamReturn ret = XCAM_RETURN_NO_ERROR;
    if (needSync) mCfgMutex.lock();
    // if something changed
    if (updateAtt) {
        mCurAtt   = mNewAtt;
        updateAtt = false;
        rk_aiq_uapi_atmo_SetAttrib(mAlgoCtx, mCurAtt, true);
        sendSignal();
    }
    if (needSync) mCfgMutex.unlock();

    EXIT_ANALYZER_FUNCTION();
    return ret;
}

XCamReturn RkAiqAtmoHandleInt::setAttrib(atmo_attrib_t att) {
    ENTER_ANALYZER_FUNCTION();

    XCamReturn ret = XCAM_RETURN_NO_ERROR;
    mCfgMutex.lock();
    // TODO
    // check if there is different between att & mCurAtt
    // if something changed, set att to mNewAtt, and
    // the new params will be effective later when updateConfig
    // called by RkAiqCore

    // if something changed
    if (0 != memcmp(&mCurAtt, &att, sizeof(atmo_attrib_t))) {
        mNewAtt   = att;
        updateAtt = true;
        waitSignal();
    }
    mCfgMutex.unlock();

    EXIT_ANALYZER_FUNCTION();
    return ret;
}
XCamReturn RkAiqAtmoHandleInt::getAttrib(atmo_attrib_t* att) {
    ENTER_ANALYZER_FUNCTION();

    XCamReturn ret = XCAM_RETURN_NO_ERROR;

    rk_aiq_uapi_atmo_GetAttrib(mAlgoCtx, att);

    EXIT_ANALYZER_FUNCTION();
    return ret;
}

XCamReturn RkAiqAtmoHandleInt::prepare() {
    ENTER_ANALYZER_FUNCTION();

    XCamReturn ret = XCAM_RETURN_NO_ERROR;

    ret = RkAiqHandle::prepare();
    RKAIQCORE_CHECK_RET(ret, "atmo handle prepare failed");

    RkAiqAlgoConfigAtmo* atmo_config_int = (RkAiqAlgoConfigAtmo*)mConfig;
    RkAiqCore::RkAiqAlgosGroupShared_t* shared =
        (RkAiqCore::RkAiqAlgosGroupShared_t*)(getGroupShared());
    RkAiqCore::RkAiqAlgosComShared_t* sharedCom = &mAiqCore->mAlogsComSharedParams;

    // TODO
    atmo_config_int->rawHeight    = sharedCom->snsDes.isp_acq_height;
    atmo_config_int->rawWidth     = sharedCom->snsDes.isp_acq_width;
    atmo_config_int->working_mode = sharedCom->working_mode;

    RkAiqAlgoDescription* des = (RkAiqAlgoDescription*)mDes;
    ret                       = des->prepare(mConfig);
    RKAIQCORE_CHECK_RET(ret, "atmo algo prepare failed");

    EXIT_ANALYZER_FUNCTION();
    return XCAM_RETURN_NO_ERROR;
}

XCamReturn RkAiqAtmoHandleInt::preProcess() {
    ENTER_ANALYZER_FUNCTION();

    XCamReturn ret = XCAM_RETURN_NO_ERROR;

    RkAiqAlgoPreAtmo* atmo_pre_int        = (RkAiqAlgoPreAtmo*)mPreInParam;
    RkAiqAlgoPreResAtmo* atmo_pre_res_int = (RkAiqAlgoPreResAtmo*)mPreOutParam;
    RkAiqCore::RkAiqAlgosGroupShared_t* shared =
        (RkAiqCore::RkAiqAlgosGroupShared_t*)(getGroupShared());
    RkAiqCore::RkAiqAlgosComShared_t* sharedCom = &mAiqCore->mAlogsComSharedParams;

    ret = RkAiqHandle::preProcess();
    if (ret) {
        RKAIQCORE_CHECK_RET(ret, "atmo handle preProcess failed");
    }

    RkAiqIspStats* xIspStats = nullptr;
    if (shared->ispStats) {
        xIspStats = (RkAiqIspStats*)shared->ispStats->map(shared->ispStats);
        if (!xIspStats) LOGE_AEC("isp stats is null");
    } else {
        LOGW_ADEHAZE("the xcamvideobuffer of isp stats is null");
    }

    if (!xIspStats || !xIspStats->atmo_stats_valid || !sharedCom->init) {
        LOGD("no atmo stats, ignore!");
        // TODO: keep last result ?
        //
        return XCAM_RETURN_BYPASS;
    }

    RkAiqAlgoDescription* des = (RkAiqAlgoDescription*)mDes;
    ret                       = des->pre_process(mPreInParam, mPreOutParam);
    RKAIQCORE_CHECK_RET(ret, "ahdr algo pre_process failed");

    EXIT_ANALYZER_FUNCTION();
    return XCAM_RETURN_NO_ERROR;
}

XCamReturn RkAiqAtmoHandleInt::processing() {
    ENTER_ANALYZER_FUNCTION();

    XCamReturn ret = XCAM_RETURN_NO_ERROR;

    RkAiqAlgoProcAtmo* atmo_proc_int        = (RkAiqAlgoProcAtmo*)mProcInParam;
    RkAiqAlgoProcResAtmo* atmo_proc_res_int = (RkAiqAlgoProcResAtmo*)mProcOutParam;
    RkAiqCore::RkAiqAlgosGroupShared_t* shared =
        (RkAiqCore::RkAiqAlgosGroupShared_t*)(getGroupShared());
    RkAiqCore::RkAiqAlgosComShared_t* sharedCom = &mAiqCore->mAlogsComSharedParams;

    ret = RkAiqHandle::processing();
    if (ret) {
        RKAIQCORE_CHECK_RET(ret, "atmo handle processing failed");
    }

    RkAiqIspStats* xIspStats = nullptr;
    if (shared->ispStats) {
        xIspStats = (RkAiqIspStats*)shared->ispStats->map(shared->ispStats);
        if (!xIspStats) LOGE_ATMO("isp stats is null");
    } else {
        LOGW_ATMO("the xcamvideobuffer of isp stats is null");
    }

    if (!xIspStats || !xIspStats->atmo_stats_valid || !sharedCom->init) {
        LOGD("no atmo stats, ignore!");
        // TODO: keep last result ?
        //
        return XCAM_RETURN_BYPASS;
    } else {
        memcpy(&atmo_proc_int->ispAtmoStats.tmo_stats,
               &xIspStats->AtmoStatsProxy->data()->atmo_stats.tmo_stats, sizeof(hdrtmo_stats_t));
        memcpy(atmo_proc_int->ispAtmoStats.other_stats.tmo_luma,
               xIspStats->AecStatsProxy->data()->aec_stats.ae_data.extra.rawae_big.channelg_xy,
               sizeof(atmo_proc_int->ispAtmoStats.other_stats.tmo_luma));

        if (sharedCom->working_mode == RK_AIQ_ISP_HDR_MODE_3_FRAME_HDR ||
            sharedCom->working_mode == RK_AIQ_ISP_HDR_MODE_3_LINE_HDR) {
            memcpy(atmo_proc_int->ispAtmoStats.other_stats.short_luma,
                   xIspStats->AecStatsProxy->data()->aec_stats.ae_data.chn[0].rawae_big.channelg_xy,
                   sizeof(atmo_proc_int->ispAtmoStats.other_stats.short_luma));
            memcpy(
                atmo_proc_int->ispAtmoStats.other_stats.middle_luma,
                xIspStats->AecStatsProxy->data()->aec_stats.ae_data.chn[1].rawae_lite.channelg_xy,
                sizeof(atmo_proc_int->ispAtmoStats.other_stats.middle_luma));
            memcpy(atmo_proc_int->ispAtmoStats.other_stats.long_luma,
                   xIspStats->AecStatsProxy->data()->aec_stats.ae_data.chn[2].rawae_big.channelg_xy,
                   sizeof(atmo_proc_int->ispAtmoStats.other_stats.long_luma));
        } else if (sharedCom->working_mode == RK_AIQ_ISP_HDR_MODE_2_FRAME_HDR ||
                   sharedCom->working_mode == RK_AIQ_ISP_HDR_MODE_2_LINE_HDR) {
            memcpy(atmo_proc_int->ispAtmoStats.other_stats.short_luma,
                   xIspStats->AecStatsProxy->data()->aec_stats.ae_data.chn[0].rawae_big.channelg_xy,
                   sizeof(atmo_proc_int->ispAtmoStats.other_stats.short_luma));
            memcpy(atmo_proc_int->ispAtmoStats.other_stats.long_luma,
                   xIspStats->AecStatsProxy->data()->aec_stats.ae_data.chn[1].rawae_big.channelg_xy,
                   sizeof(atmo_proc_int->ispAtmoStats.other_stats.long_luma));
        } else
            LOGD("Wrong working mode!!!");
    }

    RkAiqAlgoDescription* des = (RkAiqAlgoDescription*)mDes;
    ret                       = des->processing(mProcInParam, mProcOutParam);
    RKAIQCORE_CHECK_RET(ret, "atmo algo processing failed");

    EXIT_ANALYZER_FUNCTION();
    return ret;
}

XCamReturn RkAiqAtmoHandleInt::postProcess() {
    ENTER_ANALYZER_FUNCTION();

    XCamReturn ret = XCAM_RETURN_NO_ERROR;

    RkAiqAlgoPostAtmo* atmo_post_int        = (RkAiqAlgoPostAtmo*)mPostInParam;
    RkAiqAlgoPostResAtmo* atmo_post_res_int = (RkAiqAlgoPostResAtmo*)mPostOutParam;
    RkAiqCore::RkAiqAlgosGroupShared_t* shared =
        (RkAiqCore::RkAiqAlgosGroupShared_t*)(getGroupShared());
    RkAiqCore::RkAiqAlgosComShared_t* sharedCom = &mAiqCore->mAlogsComSharedParams;

    ret = RkAiqHandle::postProcess();
    if (ret) {
        RKAIQCORE_CHECK_RET(ret, "ahdr handle postProcess failed");
        return ret;
    }

    RkAiqIspStats* xIspStats = nullptr;
    if (shared->ispStats) {
        xIspStats = (RkAiqIspStats*)shared->ispStats->map(shared->ispStats);
        if (!xIspStats) LOGE_ATMO("isp stats is null");
    } else {
        LOGW_ATMO("the xcamvideobuffer of isp stats is null");
    }

    if (!xIspStats || !xIspStats->atmo_stats_valid || !sharedCom->init) {
        LOGD("no atmo stats, ignore!");
        // TODO: keep last result ?
        //
        return XCAM_RETURN_BYPASS;
    }

    RkAiqAlgoDescription* des = (RkAiqAlgoDescription*)mDes;
    ret                       = des->post_process(mPostInParam, mPostOutParam);
    RKAIQCORE_CHECK_RET(ret, "atmo algo post_process failed");

    EXIT_ANALYZER_FUNCTION();
    return ret;
}

XCamReturn RkAiqAtmoHandleInt::genIspResult(RkAiqFullParams* params, RkAiqFullParams* cur_params) {
    ENTER_ANALYZER_FUNCTION();

    XCamReturn ret = XCAM_RETURN_NO_ERROR;
    RkAiqCore::RkAiqAlgosGroupShared_t* shared =
        (RkAiqCore::RkAiqAlgosGroupShared_t*)(getGroupShared());
    RkAiqCore::RkAiqAlgosComShared_t* sharedCom = &mAiqCore->mAlogsComSharedParams;
    RkAiqAlgoProcResAtmo* atmo_com              = (RkAiqAlgoProcResAtmo*)mProcOutParam;

    rk_aiq_isp_tmo_params_v20_t* tmo_param = params->mTmoParams->data().ptr();

    if (!atmo_com) {
        LOGD_ANALYZER("no atmo result");
        return XCAM_RETURN_NO_ERROR;
    }

    if (!this->getAlgoId()) {
        RkAiqAlgoProcResAtmo* atmo_rk = (RkAiqAlgoProcResAtmo*)atmo_com;

        if (sharedCom->init) {
            tmo_param->frame_id = 0;
        } else {
            tmo_param->frame_id = shared->frameId;
        }

        tmo_param->result.Res.sw_hdrtmo_lgmax      = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_lgmax;
        tmo_param->result.Res.sw_hdrtmo_lgscl      = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_lgscl;
        tmo_param->result.Res.sw_hdrtmo_lgscl_inv  = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_lgscl_inv;
        tmo_param->result.Res.sw_hdrtmo_clipratio0 = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_clipratio0;
        tmo_param->result.Res.sw_hdrtmo_clipratio1 = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_clipratio1;
        tmo_param->result.Res.sw_hdrtmo_clipgap0   = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_clipgap0;
        tmo_param->result.Res.sw_hdrtmo_clipgap1   = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_clipgap1;
        tmo_param->result.Res.sw_hdrtmo_ratiol     = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_ratiol;
        tmo_param->result.Res.sw_hdrtmo_hist_min   = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_hist_min;
        tmo_param->result.Res.sw_hdrtmo_hist_low   = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_hist_low;
        tmo_param->result.Res.sw_hdrtmo_hist_high  = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_hist_high;
        tmo_param->result.Res.sw_hdrtmo_hist_0p3   = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_hist_0p3;
        tmo_param->result.Res.sw_hdrtmo_hist_shift = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_hist_shift;
        tmo_param->result.Res.sw_hdrtmo_palpha_0p18 =
            atmo_rk->AtmoProcRes.Res.sw_hdrtmo_palpha_0p18;
        tmo_param->result.Res.sw_hdrtmo_palpha_lw0p5 =
            atmo_rk->AtmoProcRes.Res.sw_hdrtmo_palpha_lw0p5;
        tmo_param->result.Res.sw_hdrtmo_palpha_lwscl =
            atmo_rk->AtmoProcRes.Res.sw_hdrtmo_palpha_lwscl;
        tmo_param->result.Res.sw_hdrtmo_maxpalpha = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_maxpalpha;
        tmo_param->result.Res.sw_hdrtmo_maxgain   = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_maxgain;
        tmo_param->result.Res.sw_hdrtmo_cfg_alpha = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_cfg_alpha;
        tmo_param->result.Res.sw_hdrtmo_set_gainoff =
            atmo_rk->AtmoProcRes.Res.sw_hdrtmo_set_gainoff;
        tmo_param->result.Res.sw_hdrtmo_set_lgmin  = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_set_lgmin;
        tmo_param->result.Res.sw_hdrtmo_set_lgmax  = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_set_lgmax;
        tmo_param->result.Res.sw_hdrtmo_set_lgmean = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_set_lgmean;
        tmo_param->result.Res.sw_hdrtmo_set_weightkey =
            atmo_rk->AtmoProcRes.Res.sw_hdrtmo_set_weightkey;
        tmo_param->result.Res.sw_hdrtmo_set_lgrange0 =
            atmo_rk->AtmoProcRes.Res.sw_hdrtmo_set_lgrange0;
        tmo_param->result.Res.sw_hdrtmo_set_lgrange1 =
            atmo_rk->AtmoProcRes.Res.sw_hdrtmo_set_lgrange1;
        tmo_param->result.Res.sw_hdrtmo_set_lgavgmax =
            atmo_rk->AtmoProcRes.Res.sw_hdrtmo_set_lgavgmax;
        tmo_param->result.Res.sw_hdrtmo_set_palpha = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_set_palpha;
        tmo_param->result.Res.sw_hdrtmo_gain_ld_off1 =
            atmo_rk->AtmoProcRes.Res.sw_hdrtmo_gain_ld_off1;
        tmo_param->result.Res.sw_hdrtmo_gain_ld_off2 =
            atmo_rk->AtmoProcRes.Res.sw_hdrtmo_gain_ld_off2;
        tmo_param->result.Res.sw_hdrtmo_cnt_vsize  = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_cnt_vsize;
        tmo_param->result.Res.sw_hdrtmo_big_en     = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_big_en;
        tmo_param->result.Res.sw_hdrtmo_nobig_en   = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_nobig_en;
        tmo_param->result.Res.sw_hdrtmo_newhist_en = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_newhist_en;
        tmo_param->result.Res.sw_hdrtmo_cnt_mode   = atmo_rk->AtmoProcRes.Res.sw_hdrtmo_cnt_mode;
        tmo_param->result.Res.sw_hdrtmo_expl_lgratio =
            atmo_rk->AtmoProcRes.Res.sw_hdrtmo_expl_lgratio;
        tmo_param->result.Res.sw_hdrtmo_lgscl_ratio =
            atmo_rk->AtmoProcRes.Res.sw_hdrtmo_lgscl_ratio;

        tmo_param->result.LongFrameMode = atmo_rk->AtmoProcRes.LongFrameMode;

        tmo_param->result.isHdrGlobalTmo = atmo_rk->AtmoProcRes.isHdrGlobalTmo;

        tmo_param->result.bTmoEn = atmo_rk->AtmoProcRes.bTmoEn;

        tmo_param->result.isLinearTmo = atmo_rk->AtmoProcRes.isLinearTmo;

        tmo_param->result.TmoFlicker.GlobalTmoStrengthDown =
            atmo_rk->AtmoProcRes.TmoFlicker.GlobalTmoStrengthDown;
        tmo_param->result.TmoFlicker.GlobalTmoStrength =
            atmo_rk->AtmoProcRes.TmoFlicker.GlobalTmoStrength;
        tmo_param->result.TmoFlicker.iir    = atmo_rk->AtmoProcRes.TmoFlicker.iir;
        tmo_param->result.TmoFlicker.iirmax = atmo_rk->AtmoProcRes.TmoFlicker.iirmax;
        tmo_param->result.TmoFlicker.height = atmo_rk->AtmoProcRes.TmoFlicker.height;
        tmo_param->result.TmoFlicker.width  = atmo_rk->AtmoProcRes.TmoFlicker.width;

        tmo_param->result.TmoFlicker.PredictK.correction_factor =
            atmo_rk->AtmoProcRes.TmoFlicker.PredictK.correction_factor;
        tmo_param->result.TmoFlicker.PredictK.correction_offset =
            atmo_rk->AtmoProcRes.TmoFlicker.PredictK.correction_offset;
        tmo_param->result.TmoFlicker.PredictK.Hdr3xLongPercent =
            atmo_rk->AtmoProcRes.TmoFlicker.PredictK.Hdr3xLongPercent;
        tmo_param->result.TmoFlicker.PredictK.UseLongUpTh =
            atmo_rk->AtmoProcRes.TmoFlicker.PredictK.UseLongUpTh;
        tmo_param->result.TmoFlicker.PredictK.UseLongLowTh =
            atmo_rk->AtmoProcRes.TmoFlicker.PredictK.UseLongLowTh;
        for (int i = 0; i < 3; i++)
            tmo_param->result.TmoFlicker.LumaDeviation[i] =
                atmo_rk->AtmoProcRes.TmoFlicker.LumaDeviation[i];
        tmo_param->result.TmoFlicker.StableThr = atmo_rk->AtmoProcRes.TmoFlicker.StableThr;
    }

    cur_params->mTmoParams = params->mTmoParams;

    EXIT_ANALYZER_FUNCTION();

    return ret;
}

};  // namespace RkCam