/*
 * Copyright 2020 Advanced Micro Devices, Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 *
 * Authors: AMD
 *
 */


#include "reg_helper.h"
#include "resource.h"
#include "dwb.h"
#include "dcn30_dwb.h"


#define REG(reg)\
        dwbc30->dwbc_regs->reg

#define CTX \
        dwbc30->base.ctx

#undef FN
#define FN(reg_name, field_name) \
        dwbc30->dwbc_shift->field_name, dwbc30->dwbc_mask->field_name

#define DC_LOGGER \
        dwbc30->base.ctx->logger

static bool dwb3_get_caps(struct dwbc *dwbc, struct dwb_caps *caps)
{
        if (caps) {
                caps->adapter_id = 0;   /* we only support 1 adapter currently */
                caps->hw_version = DCN_VERSION_3_0;
                caps->num_pipes = 2;
                memset(&caps->reserved, 0, sizeof(caps->reserved));
                memset(&caps->reserved2, 0, sizeof(caps->reserved2));
                caps->sw_version = dwb_ver_2_0;
                caps->caps.support_dwb = true;
                caps->caps.support_ogam = true;
                caps->caps.support_wbscl = true;
                caps->caps.support_ocsc = false;
                caps->caps.support_stereo = true;
                return true;
        } else {
                return false;
        }
}

void dwb3_config_fc(struct dwbc *dwbc, struct dc_dwb_params *params)
{
        struct dcn30_dwbc *dwbc30 = TO_DCN30_DWBC(dwbc);

        /* Set DWB source size */
        REG_UPDATE_2(FC_SOURCE_SIZE, FC_SOURCE_WIDTH, params->cnv_params.src_width,
                        FC_SOURCE_HEIGHT, params->cnv_params.src_height);

        /* source size is not equal the source size, then enable cropping. */
        if (params->cnv_params.crop_en) {
                REG_UPDATE(FC_MODE_CTRL,    FC_WINDOW_CROP_EN, 1);
                REG_UPDATE(FC_WINDOW_START, FC_WINDOW_START_X, params->cnv_params.crop_x);
                REG_UPDATE(FC_WINDOW_START, FC_WINDOW_START_Y, params->cnv_params.crop_y);
                REG_UPDATE(FC_WINDOW_SIZE,  FC_WINDOW_WIDTH,   params->cnv_params.crop_width);
                REG_UPDATE(FC_WINDOW_SIZE,  FC_WINDOW_HEIGHT,  params->cnv_params.crop_height);
        } else {
                REG_UPDATE(FC_MODE_CTRL,    FC_WINDOW_CROP_EN, 0);
        }

        /* Set CAPTURE_RATE */
        REG_UPDATE(FC_MODE_CTRL, FC_FRAME_CAPTURE_RATE, params->capture_rate);

        dwb3_set_stereo(dwbc, &params->stereo_params);
}

bool dwb3_enable(struct dwbc *dwbc, struct dc_dwb_params *params)
{
        struct dcn30_dwbc *dwbc30 = TO_DCN30_DWBC(dwbc);
        DC_LOG_DWB("%s dwb3_enabled at inst = %d", __func__, dwbc->inst);

        /* Set WB_ENABLE (not double buffered; capture not enabled) */
        REG_UPDATE(DWB_ENABLE_CLK_CTRL, DWB_ENABLE, 1);

        /* Set FC parameters */
        dwb3_config_fc(dwbc, params);

        /* Program color processing unit */
        dwb3_program_hdr_mult(dwbc, params);
        dwb3_set_gamut_remap(dwbc, params);
        dwb3_ogam_set_input_transfer_func(dwbc, params->out_transfer_func);

        /* Program output denorm */
        dwb3_set_denorm(dwbc, params);

        /* Enable DWB capture enable (double buffered) */
        REG_UPDATE(FC_MODE_CTRL, FC_FRAME_CAPTURE_EN, DWB_FRAME_CAPTURE_ENABLE);

        /* First pixel count */
        REG_UPDATE(FC_FLOW_CTRL, FC_FIRST_PIXEL_DELAY_COUNT, 96);

        return true;
}

bool dwb3_disable(struct dwbc *dwbc)
{
        struct dcn30_dwbc *dwbc30 = TO_DCN30_DWBC(dwbc);

        /* disable FC */
        REG_UPDATE(FC_MODE_CTRL, FC_FRAME_CAPTURE_EN, DWB_FRAME_CAPTURE_DISABLE);

        /* disable WB */
        REG_UPDATE(DWB_ENABLE_CLK_CTRL, DWB_ENABLE, 0);

        DC_LOG_DWB("%s dwb3_disabled at inst = %d", __func__, dwbc->inst);
        return true;
}

void dwb3_set_fc_enable(struct dwbc *dwbc, enum dwb_frame_capture_enable enable)
{
        struct dcn30_dwbc *dwbc30 = TO_DCN30_DWBC(dwbc);
        unsigned int pre_locked;

        REG_GET(DWB_UPDATE_CTRL, DWB_UPDATE_LOCK, &pre_locked);

        /* Lock DWB registers */
        if (pre_locked == 0)
                REG_UPDATE(DWB_UPDATE_CTRL, DWB_UPDATE_LOCK, 1);

        /* Disable FC */
        REG_UPDATE(FC_MODE_CTRL, FC_FRAME_CAPTURE_EN, enable);

        /* Unlock DWB registers */
        if (pre_locked == 0)
                REG_UPDATE(DWB_UPDATE_CTRL, DWB_UPDATE_LOCK, 0);

        DC_LOG_DWB("%s dwb3_fc_disabled at inst = %d", __func__, dwbc->inst);
}


bool dwb3_update(struct dwbc *dwbc, struct dc_dwb_params *params)
{
        struct dcn30_dwbc *dwbc30 = TO_DCN30_DWBC(dwbc);
        unsigned int pre_locked;

        /*
         * Check if the caller has already locked DWB registers.
         * If so: assume the caller will unlock, so don't touch the lock.
         * If not: lock them for this update, then unlock after the
         * update is complete.
         */
        REG_GET(DWB_UPDATE_CTRL, DWB_UPDATE_LOCK, &pre_locked);
        DC_LOG_DWB("%s dwb update, inst = %d", __func__, dwbc->inst);

        if (pre_locked == 0) {
                /* Lock DWB registers */
                REG_UPDATE(DWB_UPDATE_CTRL, DWB_UPDATE_LOCK, 1);
        }

        /* Set FC parameters */
        dwb3_config_fc(dwbc, params);

        /* Program color processing unit */
        dwb3_program_hdr_mult(dwbc, params);
        dwb3_set_gamut_remap(dwbc, params);
        dwb3_ogam_set_input_transfer_func(dwbc, params->out_transfer_func);

        /* Program output denorm */
        dwb3_set_denorm(dwbc, params);

        if (pre_locked == 0) {
                /* Unlock DWB registers */
                REG_UPDATE(DWB_UPDATE_CTRL, DWB_UPDATE_LOCK, 0);
        }

        return true;
}

bool dwb3_is_enabled(struct dwbc *dwbc)
{
        struct dcn30_dwbc *dwbc30 = TO_DCN30_DWBC(dwbc);
        unsigned int dwb_enabled = 0;
        unsigned int fc_frame_capture_en = 0;

        REG_GET(DWB_ENABLE_CLK_CTRL, DWB_ENABLE, &dwb_enabled);
        REG_GET(FC_MODE_CTRL, FC_FRAME_CAPTURE_EN, &fc_frame_capture_en);

        return ((dwb_enabled != 0) && (fc_frame_capture_en != 0));
}

void dwb3_set_stereo(struct dwbc *dwbc,
                struct dwb_stereo_params *stereo_params)
{
        struct dcn30_dwbc *dwbc30 = TO_DCN30_DWBC(dwbc);

        if (stereo_params->stereo_enabled) {
                REG_UPDATE(FC_MODE_CTRL, FC_EYE_SELECTION,       stereo_params->stereo_eye_select);
                REG_UPDATE(FC_MODE_CTRL, FC_STEREO_EYE_POLARITY, stereo_params->stereo_polarity);
                DC_LOG_DWB("%s dwb stereo enabled", __func__);
        } else {
                REG_UPDATE(FC_MODE_CTRL, FC_EYE_SELECTION, 0);
                DC_LOG_DWB("%s dwb stereo disabled", __func__);
        }
}

void dwb3_set_new_content(struct dwbc *dwbc,
                                                bool is_new_content)
{
        struct dcn30_dwbc *dwbc30 = TO_DCN30_DWBC(dwbc);

        REG_UPDATE(FC_MODE_CTRL, FC_NEW_CONTENT, is_new_content);
}

void dwb3_set_denorm(struct dwbc *dwbc, struct dc_dwb_params *params)
{
        struct dcn30_dwbc *dwbc30 = TO_DCN30_DWBC(dwbc);

        /* Set output format*/
        REG_UPDATE(DWB_OUT_CTRL, OUT_FORMAT, params->cnv_params.fc_out_format);

        /* Set output denorm */
        if (params->cnv_params.fc_out_format == DWB_OUT_FORMAT_32BPP_ARGB ||
                        params->cnv_params.fc_out_format == DWB_OUT_FORMAT_32BPP_RGBA) {
                REG_UPDATE(DWB_OUT_CTRL, OUT_DENORM, params->cnv_params.out_denorm_mode);
                REG_UPDATE(DWB_OUT_CTRL, OUT_MAX,    params->cnv_params.out_max_pix_val);
                REG_UPDATE(DWB_OUT_CTRL, OUT_MIN,    params->cnv_params.out_min_pix_val);
        }
}


static const struct dwbc_funcs dcn30_dwbc_funcs = {
        .get_caps               = dwb3_get_caps,
        .enable                 = dwb3_enable,
        .disable                = dwb3_disable,
        .update                 = dwb3_update,
        .is_enabled             = dwb3_is_enabled,
        .set_fc_enable          = dwb3_set_fc_enable,
        .set_stereo             = dwb3_set_stereo,
        .set_new_content        = dwb3_set_new_content,
        .dwb_ogam_set_input_transfer_func       = dwb3_ogam_set_input_transfer_func, //TODO: rename
};

void dcn30_dwbc_construct(struct dcn30_dwbc *dwbc30,
                struct dc_context *ctx,
                const struct dcn30_dwbc_registers *dwbc_regs,
                const struct dcn30_dwbc_shift *dwbc_shift,
                const struct dcn30_dwbc_mask *dwbc_mask,
                int inst)
{
        dwbc30->base.ctx = ctx;

        dwbc30->base.inst = inst;
        dwbc30->base.funcs = &dcn30_dwbc_funcs;

        dwbc30->dwbc_regs = dwbc_regs;
        dwbc30->dwbc_shift = dwbc_shift;
        dwbc30->dwbc_mask = dwbc_mask;
}