/*
 * Copyright 2015 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 <linux/slab.h>
#include <linux/mm.h>

#include "dm_services.h"

#include "dc.h"

#include "core_status.h"
#include "core_types.h"
#include "hw_sequencer.h"
#include "dce/dce_hwseq.h"

#include "resource.h"

#include "clk_mgr.h"
#include "clock_source.h"
#include "dc_bios_types.h"

#include "bios_parser_interface.h"
#include "bios/bios_parser_helper.h"
#include "include/irq_service_interface.h"
#include "transform.h"
#include "dmcu.h"
#include "dpp.h"
#include "timing_generator.h"
#include "abm.h"
#include "virtual/virtual_link_encoder.h"
#include "hubp.h"

#include "link_hwss.h"
#include "link_encoder.h"
#include "link_enc_cfg.h"

#include "dc_link.h"
#include "dc_link_ddc.h"
#include "dm_helpers.h"
#include "mem_input.h"

#include "dc_link_dp.h"
#include "dc_dmub_srv.h"

#include "dsc.h"

#include "vm_helper.h"

#include "dce/dce_i2c.h"

#include "dmub/dmub_srv.h"

#include "i2caux_interface.h"
#include "dce/dmub_hw_lock_mgr.h"

#include "dc_trace.h"

#define CTX \
        dc->ctx

#define DC_LOGGER \
        dc->ctx->logger

static const char DC_BUILD_ID[] = "production-build";

/**
 * DOC: Overview
 *
 * DC is the OS-agnostic component of the amdgpu DC driver.
 *
 * DC maintains and validates a set of structs representing the state of the
 * driver and writes that state to AMD hardware
 *
 * Main DC HW structs:
 *
 * struct dc - The central struct.  One per driver.  Created on driver load,
 * destroyed on driver unload.
 *
 * struct dc_context - One per driver.
 * Used as a backpointer by most other structs in dc.
 *
 * struct dc_link - One per connector (the physical DP, HDMI, miniDP, or eDP
 * plugpoints).  Created on driver load, destroyed on driver unload.
 *
 * struct dc_sink - One per display.  Created on boot or hotplug.
 * Destroyed on shutdown or hotunplug.  A dc_link can have a local sink
 * (the display directly attached).  It may also have one or more remote
 * sinks (in the Multi-Stream Transport case)
 *
 * struct resource_pool - One per driver.  Represents the hw blocks not in the
 * main pipeline.  Not directly accessible by dm.
 *
 * Main dc state structs:
 *
 * These structs can be created and destroyed as needed.  There is a full set of
 * these structs in dc->current_state representing the currently programmed state.
 *
 * struct dc_state - The global DC state to track global state information,
 * such as bandwidth values.
 *
 * struct dc_stream_state - Represents the hw configuration for the pipeline from
 * a framebuffer to a display.  Maps one-to-one with dc_sink.
 *
 * struct dc_plane_state - Represents a framebuffer.  Each stream has at least one,
 * and may have more in the Multi-Plane Overlay case.
 *
 * struct resource_context - Represents the programmable state of everything in
 * the resource_pool.  Not directly accessible by dm.
 *
 * struct pipe_ctx - A member of struct resource_context.  Represents the
 * internal hardware pipeline components.  Each dc_plane_state has either
 * one or two (in the pipe-split case).
 */

/*******************************************************************************
 * Private functions
 ******************************************************************************/

static inline void elevate_update_type(enum surface_update_type *original, enum surface_update_type new)
{
        if (new > *original)
                *original = new;
}

static void destroy_links(struct dc *dc)
{
        uint32_t i;

        for (i = 0; i < dc->link_count; i++) {
                if (NULL != dc->links[i])
                        link_destroy(&dc->links[i]);
        }
}

static uint32_t get_num_of_internal_disp(struct dc_link **links, uint32_t num_links)
{
        int i;
        uint32_t count = 0;

        for (i = 0; i < num_links; i++) {
                if (links[i]->connector_signal == SIGNAL_TYPE_EDP ||
                                links[i]->is_internal_display)
                        count++;
        }

        return count;
}

static int get_seamless_boot_stream_count(struct dc_state *ctx)
{
        uint8_t i;
        uint8_t seamless_boot_stream_count = 0;

        for (i = 0; i < ctx->stream_count; i++)
                if (ctx->streams[i]->apply_seamless_boot_optimization)
                        seamless_boot_stream_count++;

        return seamless_boot_stream_count;
}

static bool create_links(
                struct dc *dc,
                uint32_t num_virtual_links)
{
        int i;
        int connectors_num;
        struct dc_bios *bios = dc->ctx->dc_bios;

        dc->link_count = 0;

        connectors_num = bios->funcs->get_connectors_number(bios);

        DC_LOG_DC("BIOS object table - number of connectors: %d", connectors_num);

        if (connectors_num > ENUM_ID_COUNT) {
                dm_error(
                        "DC: Number of connectors %d exceeds maximum of %d!\n",
                        connectors_num,
                        ENUM_ID_COUNT);
                return false;
        }

        dm_output_to_console(
                "DC: %s: connectors_num: physical:%d, virtual:%d\n",
                __func__,
                connectors_num,
                num_virtual_links);

        for (i = 0; i < connectors_num; i++) {
                struct link_init_data link_init_params = {0};
                struct dc_link *link;

                DC_LOG_DC("BIOS object table - printing link object info for connector number: %d, link_index: %d", i, dc->link_count);

                link_init_params.ctx = dc->ctx;
                /* next BIOS object table connector */
                link_init_params.connector_index = i;
                link_init_params.link_index = dc->link_count;
                link_init_params.dc = dc;
                link = link_create(&link_init_params);

                if (link) {
                                dc->links[dc->link_count] = link;
                                link->dc = dc;
                                ++dc->link_count;
                }
        }

        DC_LOG_DC("BIOS object table - end");

        for (i = 0; i < num_virtual_links; i++) {
                struct dc_link *link = kzalloc(sizeof(*link), GFP_KERNEL);
                struct encoder_init_data enc_init = {0};

                if (link == NULL) {
                        BREAK_TO_DEBUGGER();
                        goto failed_alloc;
                }

                link->link_index = dc->link_count;
                dc->links[dc->link_count] = link;
                dc->link_count++;

                link->ctx = dc->ctx;
                link->dc = dc;
                link->connector_signal = SIGNAL_TYPE_VIRTUAL;
                link->link_id.type = OBJECT_TYPE_CONNECTOR;
                link->link_id.id = CONNECTOR_ID_VIRTUAL;
                link->link_id.enum_id = ENUM_ID_1;
                link->link_enc = kzalloc(sizeof(*link->link_enc), GFP_KERNEL);

                if (!link->link_enc) {
                        BREAK_TO_DEBUGGER();
                        goto failed_alloc;
                }

                link->link_status.dpcd_caps = &link->dpcd_caps;

                enc_init.ctx = dc->ctx;
                enc_init.channel = CHANNEL_ID_UNKNOWN;
                enc_init.hpd_source = HPD_SOURCEID_UNKNOWN;
                enc_init.transmitter = TRANSMITTER_UNKNOWN;
                enc_init.connector = link->link_id;
                enc_init.encoder.type = OBJECT_TYPE_ENCODER;
                enc_init.encoder.id = ENCODER_ID_INTERNAL_VIRTUAL;
                enc_init.encoder.enum_id = ENUM_ID_1;
                virtual_link_encoder_construct(link->link_enc, &enc_init);
        }

        dc->caps.num_of_internal_disp = get_num_of_internal_disp(dc->links, dc->link_count);

        return true;

failed_alloc:
        return false;
}

static struct dc_perf_trace *dc_perf_trace_create(void)
{
        return kzalloc(sizeof(struct dc_perf_trace), GFP_KERNEL);
}

static void dc_perf_trace_destroy(struct dc_perf_trace **perf_trace)
{
        kfree(*perf_trace);
        *perf_trace = NULL;
}

/**
 *  dc_stream_adjust_vmin_vmax:
 *
 *  Looks up the pipe context of dc_stream_state and updates the
 *  vertical_total_min and vertical_total_max of the DRR, Dynamic Refresh
 *  Rate, which is a power-saving feature that targets reducing panel
 *  refresh rate while the screen is static
 *
 *  @dc:     dc reference
 *  @stream: Initial dc stream state
 *  @adjust: Updated parameters for vertical_total_min and vertical_total_max
 */
bool dc_stream_adjust_vmin_vmax(struct dc *dc,
                struct dc_stream_state *stream,
                struct dc_crtc_timing_adjust *adjust)
{
        int i;
        bool ret = false;

        stream->adjust.v_total_max = adjust->v_total_max;
        stream->adjust.v_total_mid = adjust->v_total_mid;
        stream->adjust.v_total_mid_frame_num = adjust->v_total_mid_frame_num;
        stream->adjust.v_total_min = adjust->v_total_min;

        for (i = 0; i < MAX_PIPES; i++) {
                struct pipe_ctx *pipe = &dc->current_state->res_ctx.pipe_ctx[i];

                if (pipe->stream == stream && pipe->stream_res.tg) {
                        dc->hwss.set_drr(&pipe,
                                        1,
                                        *adjust);

                        ret = true;
                }
        }
        return ret;
}

/**
 *****************************************************************************
 *  Function: dc_stream_get_last_vrr_vtotal
 *
 *  @brief
 *     Looks up the pipe context of dc_stream_state and gets the
 *     last VTOTAL used by DRR (Dynamic Refresh Rate)
 *
 *  @param [in] dc: dc reference
 *  @param [in] stream: Initial dc stream state
 *  @param [in] adjust: Updated parameters for vertical_total_min and
 *  vertical_total_max
 *****************************************************************************
 */
bool dc_stream_get_last_used_drr_vtotal(struct dc *dc,
                struct dc_stream_state *stream,
                uint32_t *refresh_rate)
{
        bool status = false;

        int i = 0;

        for (i = 0; i < MAX_PIPES; i++) {
                struct pipe_ctx *pipe = &dc->current_state->res_ctx.pipe_ctx[i];

                if (pipe->stream == stream && pipe->stream_res.tg) {
                        /* Only execute if a function pointer has been defined for
                         * the DC version in question
                         */
                        if (pipe->stream_res.tg->funcs->get_last_used_drr_vtotal) {
                                pipe->stream_res.tg->funcs->get_last_used_drr_vtotal(pipe->stream_res.tg, refresh_rate);

                                status = true;

                                break;
                        }
                }
        }

        return status;
}

bool dc_stream_get_crtc_position(struct dc *dc,
                struct dc_stream_state **streams, int num_streams,
                unsigned int *v_pos, unsigned int *nom_v_pos)
{
        /* TODO: Support multiple streams */
        const struct dc_stream_state *stream = streams[0];
        int i;
        bool ret = false;
        struct crtc_position position;

        for (i = 0; i < MAX_PIPES; i++) {
                struct pipe_ctx *pipe =
                                &dc->current_state->res_ctx.pipe_ctx[i];

                if (pipe->stream == stream && pipe->stream_res.stream_enc) {
                        dc->hwss.get_position(&pipe, 1, &position);

                        *v_pos = position.vertical_count;
                        *nom_v_pos = position.nominal_vcount;
                        ret = true;
                }
        }
        return ret;
}

#if defined(CONFIG_DRM_AMD_SECURE_DISPLAY)
bool dc_stream_forward_dmcu_crc_window(struct dc *dc, struct dc_stream_state *stream,
                             struct crc_params *crc_window)
{
        int i;
        struct dmcu *dmcu = dc->res_pool->dmcu;
        struct pipe_ctx *pipe;
        struct crc_region tmp_win, *crc_win;
        struct otg_phy_mux mapping_tmp, *mux_mapping;

        /*crc window can't be null*/
        if (!crc_window)
                return false;

        if ((dmcu != NULL && dmcu->funcs->is_dmcu_initialized(dmcu))) {
                crc_win = &tmp_win;
                mux_mapping = &mapping_tmp;
                /*set crc window*/
                tmp_win.x_start = crc_window->windowa_x_start;
                tmp_win.y_start = crc_window->windowa_y_start;
                tmp_win.x_end = crc_window->windowa_x_end;
                tmp_win.y_end = crc_window->windowa_y_end;

                for (i = 0; i < MAX_PIPES; i++) {
                        pipe = &dc->current_state->res_ctx.pipe_ctx[i];
                        if (pipe->stream == stream && !pipe->top_pipe && !pipe->prev_odm_pipe)
                                break;
                }

                /* Stream not found */
                if (i == MAX_PIPES)
                        return false;


                /*set mux routing info*/
                mapping_tmp.phy_output_num = stream->link->link_enc_hw_inst;
                mapping_tmp.otg_output_num = pipe->stream_res.tg->inst;

                dmcu->funcs->forward_crc_window(dmcu, crc_win, mux_mapping);
        } else {
                DC_LOG_DC("dmcu is not initialized");
                return false;
        }

        return true;
}

bool dc_stream_stop_dmcu_crc_win_update(struct dc *dc, struct dc_stream_state *stream)
{
        int i;
        struct dmcu *dmcu = dc->res_pool->dmcu;
        struct pipe_ctx *pipe;
        struct otg_phy_mux mapping_tmp, *mux_mapping;

        if ((dmcu != NULL && dmcu->funcs->is_dmcu_initialized(dmcu))) {
                mux_mapping = &mapping_tmp;

                for (i = 0; i < MAX_PIPES; i++) {
                        pipe = &dc->current_state->res_ctx.pipe_ctx[i];
                        if (pipe->stream == stream && !pipe->top_pipe && !pipe->prev_odm_pipe)
                                break;
                }

                /* Stream not found */
                if (i == MAX_PIPES)
                        return false;


                /*set mux routing info*/
                mapping_tmp.phy_output_num = stream->link->link_enc_hw_inst;
                mapping_tmp.otg_output_num = pipe->stream_res.tg->inst;

                dmcu->funcs->stop_crc_win_update(dmcu, mux_mapping);
        } else {
                DC_LOG_DC("dmcu is not initialized");
                return false;
        }

        return true;
}
#endif

/**
 * dc_stream_configure_crc() - Configure CRC capture for the given stream.
 * @dc: DC Object
 * @stream: The stream to configure CRC on.
 * @enable: Enable CRC if true, disable otherwise.
 * @crc_window: CRC window (x/y start/end) information
 * @continuous: Capture CRC on every frame if true. Otherwise, only capture
 *              once.
 *
 * By default, only CRC0 is configured, and the entire frame is used to
 * calculate the crc.
 */
bool dc_stream_configure_crc(struct dc *dc, struct dc_stream_state *stream,
                             struct crc_params *crc_window, bool enable, bool continuous)
{
        int i;
        struct pipe_ctx *pipe;
        struct crc_params param;
        struct timing_generator *tg;

        for (i = 0; i < MAX_PIPES; i++) {
                pipe = &dc->current_state->res_ctx.pipe_ctx[i];
                if (pipe->stream == stream && !pipe->top_pipe && !pipe->prev_odm_pipe)
                        break;
        }
        /* Stream not found */
        if (i == MAX_PIPES)
                return false;

        /* By default, capture the full frame */
        param.windowa_x_start = 0;
        param.windowa_y_start = 0;
        param.windowa_x_end = pipe->stream->timing.h_addressable;
        param.windowa_y_end = pipe->stream->timing.v_addressable;
        param.windowb_x_start = 0;
        param.windowb_y_start = 0;
        param.windowb_x_end = pipe->stream->timing.h_addressable;
        param.windowb_y_end = pipe->stream->timing.v_addressable;

        if (crc_window) {
                param.windowa_x_start = crc_window->windowa_x_start;
                param.windowa_y_start = crc_window->windowa_y_start;
                param.windowa_x_end = crc_window->windowa_x_end;
                param.windowa_y_end = crc_window->windowa_y_end;
                param.windowb_x_start = crc_window->windowb_x_start;
                param.windowb_y_start = crc_window->windowb_y_start;
                param.windowb_x_end = crc_window->windowb_x_end;
                param.windowb_y_end = crc_window->windowb_y_end;
        }

        param.dsc_mode = pipe->stream->timing.flags.DSC ? 1:0;
        param.odm_mode = pipe->next_odm_pipe ? 1:0;

        /* Default to the union of both windows */
        param.selection = UNION_WINDOW_A_B;
        param.continuous_mode = continuous;
        param.enable = enable;

        tg = pipe->stream_res.tg;

        /* Only call if supported */
        if (tg->funcs->configure_crc)
                return tg->funcs->configure_crc(tg, &param);
        DC_LOG_WARNING("CRC capture not supported.");
        return false;
}

/**
 * dc_stream_get_crc() - Get CRC values for the given stream.
 * @dc: DC object
 * @stream: The DC stream state of the stream to get CRCs from.
 * @r_cr: CRC value for the first of the 3 channels stored here.
 * @g_y:  CRC value for the second of the 3 channels stored here.
 * @b_cb: CRC value for the third of the 3 channels stored here.
 *
 * dc_stream_configure_crc needs to be called beforehand to enable CRCs.
 * Return false if stream is not found, or if CRCs are not enabled.
 */
bool dc_stream_get_crc(struct dc *dc, struct dc_stream_state *stream,
                       uint32_t *r_cr, uint32_t *g_y, uint32_t *b_cb)
{
        int i;
        struct pipe_ctx *pipe;
        struct timing_generator *tg;

        for (i = 0; i < MAX_PIPES; i++) {
                pipe = &dc->current_state->res_ctx.pipe_ctx[i];
                if (pipe->stream == stream)
                        break;
        }
        /* Stream not found */
        if (i == MAX_PIPES)
                return false;

        tg = pipe->stream_res.tg;

        if (tg->funcs->get_crc)
                return tg->funcs->get_crc(tg, r_cr, g_y, b_cb);
        DC_LOG_WARNING("CRC capture not supported.");
        return false;
}

void dc_stream_set_dyn_expansion(struct dc *dc, struct dc_stream_state *stream,
                enum dc_dynamic_expansion option)
{
        /* OPP FMT dyn expansion updates*/
        int i;
        struct pipe_ctx *pipe_ctx;

        for (i = 0; i < MAX_PIPES; i++) {
                if (dc->current_state->res_ctx.pipe_ctx[i].stream
                                == stream) {
                        pipe_ctx = &dc->current_state->res_ctx.pipe_ctx[i];
                        pipe_ctx->stream_res.opp->dyn_expansion = option;
                        pipe_ctx->stream_res.opp->funcs->opp_set_dyn_expansion(
                                        pipe_ctx->stream_res.opp,
                                        COLOR_SPACE_YCBCR601,
                                        stream->timing.display_color_depth,
                                        stream->signal);
                }
        }
}

void dc_stream_set_dither_option(struct dc_stream_state *stream,
                enum dc_dither_option option)
{
        struct bit_depth_reduction_params params;
        struct dc_link *link = stream->link;
        struct pipe_ctx *pipes = NULL;
        int i;

        for (i = 0; i < MAX_PIPES; i++) {
                if (link->dc->current_state->res_ctx.pipe_ctx[i].stream ==
                                stream) {
                        pipes = &link->dc->current_state->res_ctx.pipe_ctx[i];
                        break;
                }
        }

        if (!pipes)
                return;
        if (option > DITHER_OPTION_MAX)
                return;

        stream->dither_option = option;

        memset(&params, 0, sizeof(params));
        resource_build_bit_depth_reduction_params(stream, &params);
        stream->bit_depth_params = params;

        if (pipes->plane_res.xfm &&
            pipes->plane_res.xfm->funcs->transform_set_pixel_storage_depth) {
                pipes->plane_res.xfm->funcs->transform_set_pixel_storage_depth(
                        pipes->plane_res.xfm,
                        pipes->plane_res.scl_data.lb_params.depth,
                        &stream->bit_depth_params);
        }

        pipes->stream_res.opp->funcs->
                opp_program_bit_depth_reduction(pipes->stream_res.opp, &params);
}

bool dc_stream_set_gamut_remap(struct dc *dc, const struct dc_stream_state *stream)
{
        int i;
        bool ret = false;
        struct pipe_ctx *pipes;

        for (i = 0; i < MAX_PIPES; i++) {
                if (dc->current_state->res_ctx.pipe_ctx[i].stream == stream) {
                        pipes = &dc->current_state->res_ctx.pipe_ctx[i];
                        dc->hwss.program_gamut_remap(pipes);
                        ret = true;
                }
        }

        return ret;
}

bool dc_stream_program_csc_matrix(struct dc *dc, struct dc_stream_state *stream)
{
        int i;
        bool ret = false;
        struct pipe_ctx *pipes;

        for (i = 0; i < MAX_PIPES; i++) {
                if (dc->current_state->res_ctx.pipe_ctx[i].stream
                                == stream) {

                        pipes = &dc->current_state->res_ctx.pipe_ctx[i];
                        dc->hwss.program_output_csc(dc,
                                        pipes,
                                        stream->output_color_space,
                                        stream->csc_color_matrix.matrix,
                                        pipes->stream_res.opp->inst);
                        ret = true;
                }
        }

        return ret;
}

void dc_stream_set_static_screen_params(struct dc *dc,
                struct dc_stream_state **streams,
                int num_streams,
                const struct dc_static_screen_params *params)
{
        int i, j;
        struct pipe_ctx *pipes_affected[MAX_PIPES];
        int num_pipes_affected = 0;

        for (i = 0; i < num_streams; i++) {
                struct dc_stream_state *stream = streams[i];

                for (j = 0; j < MAX_PIPES; j++) {
                        if (dc->current_state->res_ctx.pipe_ctx[j].stream
                                        == stream) {
                                pipes_affected[num_pipes_affected++] =
                                                &dc->current_state->res_ctx.pipe_ctx[j];
                        }
                }
        }

        dc->hwss.set_static_screen_control(pipes_affected, num_pipes_affected, params);
}

static void dc_destruct(struct dc *dc)
{
        if (dc->current_state) {
                dc_release_state(dc->current_state);
                dc->current_state = NULL;
        }

        destroy_links(dc);

        if (dc->clk_mgr) {
                dc_destroy_clk_mgr(dc->clk_mgr);
                dc->clk_mgr = NULL;
        }

        dc_destroy_resource_pool(dc);

        if (dc->ctx->gpio_service)
                dal_gpio_service_destroy(&dc->ctx->gpio_service);

        if (dc->ctx->created_bios)
                dal_bios_parser_destroy(&dc->ctx->dc_bios);

        dc_perf_trace_destroy(&dc->ctx->perf_trace);

        kfree(dc->ctx);
        dc->ctx = NULL;

        kfree(dc->bw_vbios);
        dc->bw_vbios = NULL;

        kfree(dc->bw_dceip);
        dc->bw_dceip = NULL;

#ifdef CONFIG_DRM_AMD_DC_DCN
        kfree(dc->dcn_soc);
        dc->dcn_soc = NULL;

        kfree(dc->dcn_ip);
        dc->dcn_ip = NULL;

#endif
        kfree(dc->vm_helper);
        dc->vm_helper = NULL;

}

static bool dc_construct_ctx(struct dc *dc,
                const struct dc_init_data *init_params)
{
        struct dc_context *dc_ctx;
        enum dce_version dc_version = DCE_VERSION_UNKNOWN;

        dc_ctx = kzalloc(sizeof(*dc_ctx), GFP_KERNEL);
        if (!dc_ctx)
                return false;

        dc_ctx->cgs_device = init_params->cgs_device;
        dc_ctx->driver_context = init_params->driver;
        dc_ctx->dc = dc;
        dc_ctx->asic_id = init_params->asic_id;
        dc_ctx->dc_sink_id_count = 0;
        dc_ctx->dc_stream_id_count = 0;
        dc_ctx->dce_environment = init_params->dce_environment;

        /* Create logger */

        dc_version = resource_parse_asic_id(init_params->asic_id);
        dc_ctx->dce_version = dc_version;

        dc_ctx->perf_trace = dc_perf_trace_create();
        if (!dc_ctx->perf_trace) {
                ASSERT_CRITICAL(false);
                return false;
        }

        dc->ctx = dc_ctx;

        return true;
}

static bool dc_construct(struct dc *dc,
                const struct dc_init_data *init_params)
{
        struct dc_context *dc_ctx;
        struct bw_calcs_dceip *dc_dceip;
        struct bw_calcs_vbios *dc_vbios;
#ifdef CONFIG_DRM_AMD_DC_DCN
        struct dcn_soc_bounding_box *dcn_soc;
        struct dcn_ip_params *dcn_ip;
#endif

        dc->config = init_params->flags;

        // Allocate memory for the vm_helper
        dc->vm_helper = kzalloc(sizeof(struct vm_helper), GFP_KERNEL);
        if (!dc->vm_helper) {
                dm_error("%s: failed to create dc->vm_helper\n", __func__);
                goto fail;
        }

        memcpy(&dc->bb_overrides, &init_params->bb_overrides, sizeof(dc->bb_overrides));

        dc_dceip = kzalloc(sizeof(*dc_dceip), GFP_KERNEL);
        if (!dc_dceip) {
                dm_error("%s: failed to create dceip\n", __func__);
                goto fail;
        }

        dc->bw_dceip = dc_dceip;

        dc_vbios = kzalloc(sizeof(*dc_vbios), GFP_KERNEL);
        if (!dc_vbios) {
                dm_error("%s: failed to create vbios\n", __func__);
                goto fail;
        }

        dc->bw_vbios = dc_vbios;
#ifdef CONFIG_DRM_AMD_DC_DCN
        dcn_soc = kzalloc(sizeof(*dcn_soc), GFP_KERNEL);
        if (!dcn_soc) {
                dm_error("%s: failed to create dcn_soc\n", __func__);
                goto fail;
        }

        dc->dcn_soc = dcn_soc;

        dcn_ip = kzalloc(sizeof(*dcn_ip), GFP_KERNEL);
        if (!dcn_ip) {
                dm_error("%s: failed to create dcn_ip\n", __func__);
                goto fail;
        }

        dc->dcn_ip = dcn_ip;
#endif

        if (!dc_construct_ctx(dc, init_params)) {
                dm_error("%s: failed to create ctx\n", __func__);
                goto fail;
        }

        dc_ctx = dc->ctx;

        /* Resource should construct all asic specific resources.
         * This should be the only place where we need to parse the asic id
         */
        if (init_params->vbios_override)
                dc_ctx->dc_bios = init_params->vbios_override;
        else {
                /* Create BIOS parser */
                struct bp_init_data bp_init_data;

                bp_init_data.ctx = dc_ctx;
                bp_init_data.bios = init_params->asic_id.atombios_base_address;

                dc_ctx->dc_bios = dal_bios_parser_create(
                                &bp_init_data, dc_ctx->dce_version);

                if (!dc_ctx->dc_bios) {
                        ASSERT_CRITICAL(false);
                        goto fail;
                }

                dc_ctx->created_bios = true;
        }

        dc->vendor_signature = init_params->vendor_signature;

        /* Create GPIO service */
        dc_ctx->gpio_service = dal_gpio_service_create(
                        dc_ctx->dce_version,
                        dc_ctx->dce_environment,
                        dc_ctx);

        if (!dc_ctx->gpio_service) {
                ASSERT_CRITICAL(false);
                goto fail;
        }

        dc->res_pool = dc_create_resource_pool(dc, init_params, dc_ctx->dce_version);
        if (!dc->res_pool)
                goto fail;

        /* set i2c speed if not done by the respective dcnxxx__resource.c */
        if (dc->caps.i2c_speed_in_khz_hdcp == 0)
                dc->caps.i2c_speed_in_khz_hdcp = dc->caps.i2c_speed_in_khz;

        dc->clk_mgr = dc_clk_mgr_create(dc->ctx, dc->res_pool->pp_smu, dc->res_pool->dccg);
        if (!dc->clk_mgr)
                goto fail;
#ifdef CONFIG_DRM_AMD_DC_DCN
        dc->clk_mgr->force_smu_not_present = init_params->force_smu_not_present;
#endif

        if (dc->res_pool->funcs->update_bw_bounding_box)
                dc->res_pool->funcs->update_bw_bounding_box(dc, dc->clk_mgr->bw_params);

        /* Creation of current_state must occur after dc->dml
         * is initialized in dc_create_resource_pool because
         * on creation it copies the contents of dc->dml
         */

        dc->current_state = dc_create_state(dc);

        if (!dc->current_state) {
                dm_error("%s: failed to create validate ctx\n", __func__);
                goto fail;
        }

        dc_resource_state_construct(dc, dc->current_state);

        if (!create_links(dc, init_params->num_virtual_links))
                goto fail;

        /* Initialise DIG link encoder resource tracking variables. */
        link_enc_cfg_init(dc, dc->current_state);

        return true;

fail:
        return false;
}

static void disable_all_writeback_pipes_for_stream(
                const struct dc *dc,
                struct dc_stream_state *stream,
                struct dc_state *context)
{
        int i;

        for (i = 0; i < stream->num_wb_info; i++)
                stream->writeback_info[i].wb_enabled = false;
}

static void apply_ctx_interdependent_lock(struct dc *dc, struct dc_state *context,
                                          struct dc_stream_state *stream, bool lock)
{
        int i;

        /* Checks if interdependent update function pointer is NULL or not, takes care of DCE110 case */
        if (dc->hwss.interdependent_update_lock)
                dc->hwss.interdependent_update_lock(dc, context, lock);
        else {
                for (i = 0; i < dc->res_pool->pipe_count; i++) {
                        struct pipe_ctx *pipe_ctx = &context->res_ctx.pipe_ctx[i];
                        struct pipe_ctx *old_pipe_ctx = &dc->current_state->res_ctx.pipe_ctx[i];

                        // Copied conditions that were previously in dce110_apply_ctx_for_surface
                        if (stream == pipe_ctx->stream) {
                                if (!pipe_ctx->top_pipe &&
                                        (pipe_ctx->plane_state || old_pipe_ctx->plane_state))
                                        dc->hwss.pipe_control_lock(dc, pipe_ctx, lock);
                        }
                }
        }
}

static void disable_dangling_plane(struct dc *dc, struct dc_state *context)
{
        int i, j;
        struct dc_state *dangling_context = dc_create_state(dc);
        struct dc_state *current_ctx;

        if (dangling_context == NULL)
                return;

        dc_resource_state_copy_construct(dc->current_state, dangling_context);

        for (i = 0; i < dc->res_pool->pipe_count; i++) {
                struct dc_stream_state *old_stream =
                                dc->current_state->res_ctx.pipe_ctx[i].stream;
                bool should_disable = true;

                for (j = 0; j < context->stream_count; j++) {
                        if (old_stream == context->streams[j]) {
                                should_disable = false;
                                break;
                        }
                }
                if (should_disable && old_stream) {
                        dc_rem_all_planes_for_stream(dc, old_stream, dangling_context);
                        disable_all_writeback_pipes_for_stream(dc, old_stream, dangling_context);

                        if (dc->hwss.apply_ctx_for_surface) {
                                apply_ctx_interdependent_lock(dc, dc->current_state, old_stream, true);
                                dc->hwss.apply_ctx_for_surface(dc, old_stream, 0, dangling_context);
                                apply_ctx_interdependent_lock(dc, dc->current_state, old_stream, false);
                                dc->hwss.post_unlock_program_front_end(dc, dangling_context);
                        }
                        if (dc->hwss.program_front_end_for_ctx) {
                                dc->hwss.interdependent_update_lock(dc, dc->current_state, true);
                                dc->hwss.program_front_end_for_ctx(dc, dangling_context);
                                dc->hwss.interdependent_update_lock(dc, dc->current_state, false);
                                dc->hwss.post_unlock_program_front_end(dc, dangling_context);
                        }
                }
        }

        current_ctx = dc->current_state;

        dc->current_state = dangling_context;
        dc_release_state(current_ctx);
}

static void disable_vbios_mode_if_required(
                struct dc *dc,
                struct dc_state *context)
{
        unsigned int i, j;

        /* check if timing_changed, disable stream*/
        for (i = 0; i < dc->res_pool->pipe_count; i++) {
                struct dc_stream_state *stream = NULL;
                struct dc_link *link = NULL;
                struct pipe_ctx *pipe = NULL;

                pipe = &context->res_ctx.pipe_ctx[i];
                stream = pipe->stream;
                if (stream == NULL)
                        continue;

                // only looking for first odm pipe
                if (pipe->prev_odm_pipe)
                        continue;

                if (stream->link->local_sink &&
                        stream->link->local_sink->sink_signal == SIGNAL_TYPE_EDP) {
                        link = stream->link;
                }

                if (link != NULL && link->link_enc->funcs->is_dig_enabled(link->link_enc)) {
                        unsigned int enc_inst, tg_inst = 0;
                        unsigned int pix_clk_100hz;

                        enc_inst = link->link_enc->funcs->get_dig_frontend(link->link_enc);
                        if (enc_inst != ENGINE_ID_UNKNOWN) {
                                for (j = 0; j < dc->res_pool->stream_enc_count; j++) {
                                        if (dc->res_pool->stream_enc[j]->id == enc_inst) {
                                                tg_inst = dc->res_pool->stream_enc[j]->funcs->dig_source_otg(
                                                        dc->res_pool->stream_enc[j]);
                                                break;
                                        }
                                }

                                dc->res_pool->dp_clock_source->funcs->get_pixel_clk_frequency_100hz(
                                        dc->res_pool->dp_clock_source,
                                        tg_inst, &pix_clk_100hz);

                                if (link->link_status.link_active) {
                                        uint32_t requested_pix_clk_100hz =
                                                pipe->stream_res.pix_clk_params.requested_pix_clk_100hz;

                                        if (pix_clk_100hz != requested_pix_clk_100hz) {
                                                core_link_disable_stream(pipe);
                                                pipe->stream->dpms_off = false;
                                        }
                                }
                        }
                }
        }
}

static void wait_for_no_pipes_pending(struct dc *dc, struct dc_state *context)
{
        int i;
        PERF_TRACE();
        for (i = 0; i < MAX_PIPES; i++) {
                int count = 0;
                struct pipe_ctx *pipe = &context->res_ctx.pipe_ctx[i];

                if (!pipe->plane_state)
                        continue;

                /* Timeout 100 ms */
                while (count < 100000) {
                        /* Must set to false to start with, due to OR in update function */
                        pipe->plane_state->status.is_flip_pending = false;
                        dc->hwss.update_pending_status(pipe);
                        if (!pipe->plane_state->status.is_flip_pending)
                                break;
                        udelay(1);
                        count++;
                }
                ASSERT(!pipe->plane_state->status.is_flip_pending);
        }
        PERF_TRACE();
}

/*******************************************************************************
 * Public functions
 ******************************************************************************/

struct dc *dc_create(const struct dc_init_data *init_params)
{
        struct dc *dc = kzalloc(sizeof(*dc), GFP_KERNEL);
        unsigned int full_pipe_count;

        if (!dc)
                return NULL;

        if (init_params->dce_environment == DCE_ENV_VIRTUAL_HW) {
                if (!dc_construct_ctx(dc, init_params))
                        goto destruct_dc;
        } else {
                if (!dc_construct(dc, init_params))
                        goto destruct_dc;

                full_pipe_count = dc->res_pool->pipe_count;
                if (dc->res_pool->underlay_pipe_index != NO_UNDERLAY_PIPE)
                        full_pipe_count--;
                dc->caps.max_streams = min(
                                full_pipe_count,
                                dc->res_pool->stream_enc_count);

                dc->caps.max_links = dc->link_count;
                dc->caps.max_audios = dc->res_pool->audio_count;
                dc->caps.linear_pitch_alignment = 64;

                dc->caps.max_dp_protocol_version = DP_VERSION_1_4;

                if (dc->res_pool->dmcu != NULL)
                        dc->versions.dmcu_version = dc->res_pool->dmcu->dmcu_version;
        }

        /* Populate versioning information */
        dc->versions.dc_ver = DC_VER;

        dc->build_id = DC_BUILD_ID;

        DC_LOG_DC("Display Core initialized\n");



        return dc;

destruct_dc:
        dc_destruct(dc);
        kfree(dc);
        return NULL;
}

static void detect_edp_presence(struct dc *dc)
{
        struct dc_link *edp_links[MAX_NUM_EDP];
        struct dc_link *edp_link = NULL;
        enum dc_connection_type type;
        int i;
        int edp_num;

        get_edp_links(dc, edp_links, &edp_num);
        if (!edp_num)
                return;

        for (i = 0; i < edp_num; i++) {
                edp_link = edp_links[i];
                if (dc->config.edp_not_connected) {
                        edp_link->edp_sink_present = false;
                } else {
                        dc_link_detect_sink(edp_link, &type);
                        edp_link->edp_sink_present = (type != dc_connection_none);
                }
        }
}

void dc_hardware_init(struct dc *dc)
{

        detect_edp_presence(dc);
        if (dc->ctx->dce_environment != DCE_ENV_VIRTUAL_HW)
                dc->hwss.init_hw(dc);
}

void dc_init_callbacks(struct dc *dc,
                const struct dc_callback_init *init_params)
{
#ifdef CONFIG_DRM_AMD_DC_HDCP
        dc->ctx->cp_psp = init_params->cp_psp;
#endif
}

void dc_deinit_callbacks(struct dc *dc)
{
#ifdef CONFIG_DRM_AMD_DC_HDCP
        memset(&dc->ctx->cp_psp, 0, sizeof(dc->ctx->cp_psp));
#endif
}

void dc_destroy(struct dc **dc)
{
        dc_destruct(*dc);
        kfree(*dc);
        *dc = NULL;
}

static void enable_timing_multisync(
                struct dc *dc,
                struct dc_state *ctx)
{
        int i, multisync_count = 0;
        int pipe_count = dc->res_pool->pipe_count;
        struct pipe_ctx *multisync_pipes[MAX_PIPES] = { NULL };

        for (i = 0; i < pipe_count; i++) {
                if (!ctx->res_ctx.pipe_ctx[i].stream ||
                                !ctx->res_ctx.pipe_ctx[i].stream->triggered_crtc_reset.enabled)
                        continue;
                if (ctx->res_ctx.pipe_ctx[i].stream == ctx->res_ctx.pipe_ctx[i].stream->triggered_crtc_reset.event_source)
                        continue;
                multisync_pipes[multisync_count] = &ctx->res_ctx.pipe_ctx[i];
                multisync_count++;
        }

        if (multisync_count > 0) {
                dc->hwss.enable_per_frame_crtc_position_reset(
                        dc, multisync_count, multisync_pipes);
        }
}

static void program_timing_sync(
                struct dc *dc,
                struct dc_state *ctx)
{
        int i, j, k;
        int group_index = 0;
        int num_group = 0;
        int pipe_count = dc->res_pool->pipe_count;
        struct pipe_ctx *unsynced_pipes[MAX_PIPES] = { NULL };

        for (i = 0; i < pipe_count; i++) {
                if (!ctx->res_ctx.pipe_ctx[i].stream || ctx->res_ctx.pipe_ctx[i].top_pipe)
                        continue;

                unsynced_pipes[i] = &ctx->res_ctx.pipe_ctx[i];
        }

        for (i = 0; i < pipe_count; i++) {
                int group_size = 1;
                enum timing_synchronization_type sync_type = NOT_SYNCHRONIZABLE;
                struct pipe_ctx *pipe_set[MAX_PIPES];

                if (!unsynced_pipes[i])
                        continue;

                pipe_set[0] = unsynced_pipes[i];
                unsynced_pipes[i] = NULL;

                /* Add tg to the set, search rest of the tg's for ones with
                 * same timing, add all tgs with same timing to the group
                 */
                for (j = i + 1; j < pipe_count; j++) {
                        if (!unsynced_pipes[j])
                                continue;
                        if (sync_type != TIMING_SYNCHRONIZABLE &&
                                dc->hwss.enable_vblanks_synchronization &&
                                unsynced_pipes[j]->stream_res.tg->funcs->align_vblanks &&
                                resource_are_vblanks_synchronizable(
                                        unsynced_pipes[j]->stream,
                                        pipe_set[0]->stream)) {
                                sync_type = VBLANK_SYNCHRONIZABLE;
                                pipe_set[group_size] = unsynced_pipes[j];
                                unsynced_pipes[j] = NULL;
                                group_size++;
                        } else
                        if (sync_type != VBLANK_SYNCHRONIZABLE &&
                                resource_are_streams_timing_synchronizable(
                                        unsynced_pipes[j]->stream,
                                        pipe_set[0]->stream)) {
                                sync_type = TIMING_SYNCHRONIZABLE;
                                pipe_set[group_size] = unsynced_pipes[j];
                                unsynced_pipes[j] = NULL;
                                group_size++;
                        }
                }

                /* set first unblanked pipe as master */
                for (j = 0; j < group_size; j++) {
                        bool is_blanked;

                        if (pipe_set[j]->stream_res.opp->funcs->dpg_is_blanked)
                                is_blanked =
                                        pipe_set[j]->stream_res.opp->funcs->dpg_is_blanked(pipe_set[j]->stream_res.opp);
                        else
                                is_blanked =
                                        pipe_set[j]->stream_res.tg->funcs->is_blanked(pipe_set[j]->stream_res.tg);
                        if (!is_blanked) {
                                if (j == 0)
                                        break;

                                swap(pipe_set[0], pipe_set[j]);
                                break;
                        }
                }

                for (k = 0; k < group_size; k++) {
                        struct dc_stream_status *status = dc_stream_get_status_from_state(ctx, pipe_set[k]->stream);

                        status->timing_sync_info.group_id = num_group;
                        status->timing_sync_info.group_size = group_size;
                        if (k == 0)
                                status->timing_sync_info.master = true;
                        else
                                status->timing_sync_info.master = false;

                }
                /* remove any other unblanked pipes as they have already been synced */
                for (j = j + 1; j < group_size; j++) {
                        bool is_blanked;

                        if (pipe_set[j]->stream_res.opp->funcs->dpg_is_blanked)
                                is_blanked =
                                        pipe_set[j]->stream_res.opp->funcs->dpg_is_blanked(pipe_set[j]->stream_res.opp);
                        else
                                is_blanked =
                                        pipe_set[j]->stream_res.tg->funcs->is_blanked(pipe_set[j]->stream_res.tg);
                        if (!is_blanked) {
                                group_size--;
                                pipe_set[j] = pipe_set[group_size];
                                j--;
                        }
                }

                if (group_size > 1) {
                        if (sync_type == TIMING_SYNCHRONIZABLE) {
                                dc->hwss.enable_timing_synchronization(
                                        dc, group_index, group_size, pipe_set);
                        } else
                                if (sync_type == VBLANK_SYNCHRONIZABLE) {
                                dc->hwss.enable_vblanks_synchronization(
                                        dc, group_index, group_size, pipe_set);
                                }
                        group_index++;
                }
                num_group++;
        }
}

static bool context_changed(
                struct dc *dc,
                struct dc_state *context)
{
        uint8_t i;

        if (context->stream_count != dc->current_state->stream_count)
                return true;

        for (i = 0; i < dc->current_state->stream_count; i++) {
                if (dc->current_state->streams[i] != context->streams[i])
                        return true;
        }

        return false;
}

bool dc_validate_seamless_boot_timing(const struct dc *dc,
                                const struct dc_sink *sink,
                                struct dc_crtc_timing *crtc_timing)
{
        struct timing_generator *tg;
        struct stream_encoder *se = NULL;

        struct dc_crtc_timing hw_crtc_timing = {0};

        struct dc_link *link = sink->link;
        unsigned int i, enc_inst, tg_inst = 0;

        /* Support seamless boot on EDP displays only */
        if (sink->sink_signal != SIGNAL_TYPE_EDP) {
                return false;
        }

        /* Check for enabled DIG to identify enabled display */
        if (!link->link_enc->funcs->is_dig_enabled(link->link_enc))
                return false;

        enc_inst = link->link_enc->funcs->get_dig_frontend(link->link_enc);

        if (enc_inst == ENGINE_ID_UNKNOWN)
                return false;

        for (i = 0; i < dc->res_pool->stream_enc_count; i++) {
                if (dc->res_pool->stream_enc[i]->id == enc_inst) {

                        se = dc->res_pool->stream_enc[i];

                        tg_inst = dc->res_pool->stream_enc[i]->funcs->dig_source_otg(
                                dc->res_pool->stream_enc[i]);
                        break;
                }
        }

        // tg_inst not found
        if (i == dc->res_pool->stream_enc_count)
                return false;

        if (tg_inst >= dc->res_pool->timing_generator_count)
                return false;

        tg = dc->res_pool->timing_generators[tg_inst];

        if (!tg->funcs->get_hw_timing)
                return false;

        if (!tg->funcs->get_hw_timing(tg, &hw_crtc_timing))
                return false;

        if (crtc_timing->h_total != hw_crtc_timing.h_total)
                return false;

        if (crtc_timing->h_border_left != hw_crtc_timing.h_border_left)
                return false;

        if (crtc_timing->h_addressable != hw_crtc_timing.h_addressable)
                return false;

        if (crtc_timing->h_border_right != hw_crtc_timing.h_border_right)
                return false;

        if (crtc_timing->h_front_porch != hw_crtc_timing.h_front_porch)
                return false;

        if (crtc_timing->h_sync_width != hw_crtc_timing.h_sync_width)
                return false;

        if (crtc_timing->v_total != hw_crtc_timing.v_total)
                return false;

        if (crtc_timing->v_border_top != hw_crtc_timing.v_border_top)
                return false;

        if (crtc_timing->v_addressable != hw_crtc_timing.v_addressable)
                return false;

        if (crtc_timing->v_border_bottom != hw_crtc_timing.v_border_bottom)
                return false;

        if (crtc_timing->v_front_porch != hw_crtc_timing.v_front_porch)
                return false;

        if (crtc_timing->v_sync_width != hw_crtc_timing.v_sync_width)
                return false;

        /* block DSC for now, as VBIOS does not currently support DSC timings */
        if (crtc_timing->flags.DSC)
                return false;

        if (dc_is_dp_signal(link->connector_signal)) {
                unsigned int pix_clk_100hz;

                dc->res_pool->dp_clock_source->funcs->get_pixel_clk_frequency_100hz(
                        dc->res_pool->dp_clock_source,
                        tg_inst, &pix_clk_100hz);

                if (crtc_timing->pix_clk_100hz != pix_clk_100hz)
                        return false;

                if (!se->funcs->dp_get_pixel_format)
                        return false;

                if (!se->funcs->dp_get_pixel_format(
                        se,
                        &hw_crtc_timing.pixel_encoding,
                        &hw_crtc_timing.display_color_depth))
                        return false;

                if (hw_crtc_timing.display_color_depth != crtc_timing->display_color_depth)
                        return false;

                if (hw_crtc_timing.pixel_encoding != crtc_timing->pixel_encoding)
                        return false;
        }

        if (link->dpcd_caps.dprx_feature.bits.VSC_SDP_COLORIMETRY_SUPPORTED) {
                return false;
        }

        if (is_edp_ilr_optimization_required(link, crtc_timing)) {
                DC_LOG_EVENT_LINK_TRAINING("Seamless boot disabled to optimize eDP link rate\n");
                return false;
        }

        return true;
}

void dc_enable_stereo(
        struct dc *dc,
        struct dc_state *context,
        struct dc_stream_state *streams[],
        uint8_t stream_count)
{
        int i, j;
        struct pipe_ctx *pipe;

        for (i = 0; i < MAX_PIPES; i++) {
                if (context != NULL)
                        pipe = &context->res_ctx.pipe_ctx[i];
                else
                        pipe = &dc->current_state->res_ctx.pipe_ctx[i];
                for (j = 0 ; pipe && j < stream_count; j++)  {
                        if (streams[j] && streams[j] == pipe->stream &&
                                dc->hwss.setup_stereo)
                                dc->hwss.setup_stereo(pipe, dc);
                }
        }
}

void dc_trigger_sync(struct dc *dc, struct dc_state *context)
{
        if (context->stream_count > 1 && !dc->debug.disable_timing_sync) {
                enable_timing_multisync(dc, context);
                program_timing_sync(dc, context);
        }
}

static uint8_t get_stream_mask(struct dc *dc, struct dc_state *context)
{
        int i;
        unsigned int stream_mask = 0;

        for (i = 0; i < dc->res_pool->pipe_count; i++) {
                if (context->res_ctx.pipe_ctx[i].stream)
                        stream_mask |= 1 << i;
        }

        return stream_mask;
}

#if defined(CONFIG_DRM_AMD_DC_DCN)
void dc_z10_restore(struct dc *dc)
{
        if (dc->hwss.z10_restore)
                dc->hwss.z10_restore(dc);
}

void dc_z10_save_init(struct dc *dc)
{
        if (dc->hwss.z10_save_init)
                dc->hwss.z10_save_init(dc);
}
#endif
/*
 * Applies given context to HW and copy it into current context.
 * It's up to the user to release the src context afterwards.
 */
static enum dc_status dc_commit_state_no_check(struct dc *dc, struct dc_state *context)
{
        struct dc_bios *dcb = dc->ctx->dc_bios;
        enum dc_status result = DC_ERROR_UNEXPECTED;
        struct pipe_ctx *pipe;
        int i, k, l;
        struct dc_stream_state *dc_streams[MAX_STREAMS] = {0};

#if defined(CONFIG_DRM_AMD_DC_DCN)
        dc_z10_restore(dc);
        dc_allow_idle_optimizations(dc, false);
#endif

        for (i = 0; i < context->stream_count; i++)
                dc_streams[i] =  context->streams[i];

        if (!dcb->funcs->is_accelerated_mode(dcb)) {
                disable_vbios_mode_if_required(dc, context);
                dc->hwss.enable_accelerated_mode(dc, context);
        }

        if (context->stream_count > get_seamless_boot_stream_count(context) ||
                context->stream_count == 0)
                dc->hwss.prepare_bandwidth(dc, context);

        disable_dangling_plane(dc, context);
        /* re-program planes for existing stream, in case we need to
         * free up plane resource for later use
         */
        if (dc->hwss.apply_ctx_for_surface) {
                for (i = 0; i < context->stream_count; i++) {
                        if (context->streams[i]->mode_changed)
                                continue;
                        apply_ctx_interdependent_lock(dc, context, context->streams[i], true);
                        dc->hwss.apply_ctx_for_surface(
                                dc, context->streams[i],
                                context->stream_status[i].plane_count,
                                context); /* use new pipe config in new context */
                        apply_ctx_interdependent_lock(dc, context, context->streams[i], false);
                        dc->hwss.post_unlock_program_front_end(dc, context);
                }
        }

        /* Program hardware */
        for (i = 0; i < dc->res_pool->pipe_count; i++) {
                pipe = &context->res_ctx.pipe_ctx[i];
                dc->hwss.wait_for_mpcc_disconnect(dc, dc->res_pool, pipe);
        }

        result = dc->hwss.apply_ctx_to_hw(dc, context);

        if (result != DC_OK)
                return result;

        dc_trigger_sync(dc, context);

        /* Program all planes within new context*/
        if (dc->hwss.program_front_end_for_ctx) {
                dc->hwss.interdependent_update_lock(dc, context, true);
                dc->hwss.program_front_end_for_ctx(dc, context);
                dc->hwss.interdependent_update_lock(dc, context, false);
                dc->hwss.post_unlock_program_front_end(dc, context);
        }
        for (i = 0; i < context->stream_count; i++) {
                const struct dc_link *link = context->streams[i]->link;

                if (!context->streams[i]->mode_changed)
                        continue;

                if (dc->hwss.apply_ctx_for_surface) {
                        apply_ctx_interdependent_lock(dc, context, context->streams[i], true);
                        dc->hwss.apply_ctx_for_surface(
                                        dc, context->streams[i],
                                        context->stream_status[i].plane_count,
                                        context);
                        apply_ctx_interdependent_lock(dc, context, context->streams[i], false);
                        dc->hwss.post_unlock_program_front_end(dc, context);
                }

                /*
                 * enable stereo
                 * TODO rework dc_enable_stereo call to work with validation sets?
                 */
                for (k = 0; k < MAX_PIPES; k++) {
                        pipe = &context->res_ctx.pipe_ctx[k];

                        for (l = 0 ; pipe && l < context->stream_count; l++)  {
                                if (context->streams[l] &&
                                        context->streams[l] == pipe->stream &&
                                        dc->hwss.setup_stereo)
                                        dc->hwss.setup_stereo(pipe, dc);
                        }
                }

                CONN_MSG_MODE(link, "{%dx%d, %dx%d@%dKhz}",
                                context->streams[i]->timing.h_addressable,
                                context->streams[i]->timing.v_addressable,
                                context->streams[i]->timing.h_total,
                                context->streams[i]->timing.v_total,
                                context->streams[i]->timing.pix_clk_100hz / 10);
        }

        dc_enable_stereo(dc, context, dc_streams, context->stream_count);

        if (context->stream_count > get_seamless_boot_stream_count(context) ||
                context->stream_count == 0) {
                /* Must wait for no flips to be pending before doing optimize bw */
                wait_for_no_pipes_pending(dc, context);
                /* pplib is notified if disp_num changed */
                dc->hwss.optimize_bandwidth(dc, context);
        }

        if (dc->ctx->dce_version >= DCE_VERSION_MAX)
                TRACE_DCN_CLOCK_STATE(&context->bw_ctx.bw.dcn.clk);
        else
                TRACE_DCE_CLOCK_STATE(&context->bw_ctx.bw.dce);

        context->stream_mask = get_stream_mask(dc, context);

        if (context->stream_mask != dc->current_state->stream_mask)
                dc_dmub_srv_notify_stream_mask(dc->ctx->dmub_srv, context->stream_mask);

        for (i = 0; i < context->stream_count; i++)
                context->streams[i]->mode_changed = false;

        dc_release_state(dc->current_state);

        dc->current_state = context;

        dc_retain_state(dc->current_state);

        return result;
}

bool dc_commit_state(struct dc *dc, struct dc_state *context)
{
        enum dc_status result = DC_ERROR_UNEXPECTED;
        int i;

        if (!context_changed(dc, context))
                return DC_OK;

        DC_LOG_DC("%s: %d streams\n",
                                __func__, context->stream_count);

        for (i = 0; i < context->stream_count; i++) {
                struct dc_stream_state *stream = context->streams[i];

                dc_stream_log(dc, stream);
        }

        result = dc_commit_state_no_check(dc, context);

        return (result == DC_OK);
}

#if defined(CONFIG_DRM_AMD_DC_DCN)
bool dc_acquire_release_mpc_3dlut(
                struct dc *dc, bool acquire,
                struct dc_stream_state *stream,
                struct dc_3dlut **lut,
                struct dc_transfer_func **shaper)
{
        int pipe_idx;
        bool ret = false;
        bool found_pipe_idx = false;
        const struct resource_pool *pool = dc->res_pool;
        struct resource_context *res_ctx = &dc->current_state->res_ctx;
        int mpcc_id = 0;

        if (pool && res_ctx) {
                if (acquire) {
                        /*find pipe idx for the given stream*/
                        for (pipe_idx = 0; pipe_idx < pool->pipe_count; pipe_idx++) {
                                if (res_ctx->pipe_ctx[pipe_idx].stream == stream) {
                                        found_pipe_idx = true;
                                        mpcc_id = res_ctx->pipe_ctx[pipe_idx].plane_res.hubp->inst;
                                        break;
                                }
                        }
                } else
                        found_pipe_idx = true;/*for release pipe_idx is not required*/

                if (found_pipe_idx) {
                        if (acquire && pool->funcs->acquire_post_bldn_3dlut)
                                ret = pool->funcs->acquire_post_bldn_3dlut(res_ctx, pool, mpcc_id, lut, shaper);
                        else if (!acquire && pool->funcs->release_post_bldn_3dlut)
                                ret = pool->funcs->release_post_bldn_3dlut(res_ctx, pool, lut, shaper);
                }
        }
        return ret;
}
#endif
static bool is_flip_pending_in_pipes(struct dc *dc, struct dc_state *context)
{
        int i;
        struct pipe_ctx *pipe;

        for (i = 0; i < MAX_PIPES; i++) {
                pipe = &context->res_ctx.pipe_ctx[i];

                if (!pipe->plane_state)
                        continue;

                /* Must set to false to start with, due to OR in update function */
                pipe->plane_state->status.is_flip_pending = false;
                dc->hwss.update_pending_status(pipe);
                if (pipe->plane_state->status.is_flip_pending)
                        return true;
        }
        return false;
}

void dc_post_update_surfaces_to_stream(struct dc *dc)
{
        int i;
        struct dc_state *context = dc->current_state;

        if ((!dc->optimized_required) || get_seamless_boot_stream_count(context) > 0)
                return;

        post_surface_trace(dc);

        if (is_flip_pending_in_pipes(dc, context))
                return;

        for (i = 0; i < dc->res_pool->pipe_count; i++)
                if (context->res_ctx.pipe_ctx[i].stream == NULL ||
                    context->res_ctx.pipe_ctx[i].plane_state == NULL) {
                        context->res_ctx.pipe_ctx[i].pipe_idx = i;
                        dc->hwss.disable_plane(dc, &context->res_ctx.pipe_ctx[i]);
                }

        dc->hwss.optimize_bandwidth(dc, context);

        dc->optimized_required = false;
        dc->wm_optimized_required = false;
}

static void init_state(struct dc *dc, struct dc_state *context)
{
        /* Each context must have their own instance of VBA and in order to
         * initialize and obtain IP and SOC the base DML instance from DC is
         * initially copied into every context
         */
#ifdef CONFIG_DRM_AMD_DC_DCN
        memcpy(&context->bw_ctx.dml, &dc->dml, sizeof(struct display_mode_lib));
#endif
}

struct dc_state *dc_create_state(struct dc *dc)
{
        struct dc_state *context = kvzalloc(sizeof(struct dc_state),
                                            GFP_KERNEL);

        if (!context)
                return NULL;

        init_state(dc, context);

        kref_init(&context->refcount);

        return context;
}

struct dc_state *dc_copy_state(struct dc_state *src_ctx)
{
        int i, j;
        struct dc_state *new_ctx = kvmalloc(sizeof(struct dc_state), GFP_KERNEL);

        if (!new_ctx)
                return NULL;
        memcpy(new_ctx, src_ctx, sizeof(struct dc_state));

        for (i = 0; i < MAX_PIPES; i++) {
                        struct pipe_ctx *cur_pipe = &new_ctx->res_ctx.pipe_ctx[i];

                        if (cur_pipe->top_pipe)
                                cur_pipe->top_pipe =  &new_ctx->res_ctx.pipe_ctx[cur_pipe->top_pipe->pipe_idx];

                        if (cur_pipe->bottom_pipe)
                                cur_pipe->bottom_pipe = &new_ctx->res_ctx.pipe_ctx[cur_pipe->bottom_pipe->pipe_idx];

                        if (cur_pipe->prev_odm_pipe)
                                cur_pipe->prev_odm_pipe =  &new_ctx->res_ctx.pipe_ctx[cur_pipe->prev_odm_pipe->pipe_idx];

                        if (cur_pipe->next_odm_pipe)
                                cur_pipe->next_odm_pipe = &new_ctx->res_ctx.pipe_ctx[cur_pipe->next_odm_pipe->pipe_idx];

        }

        for (i = 0; i < new_ctx->stream_count; i++) {
                        dc_stream_retain(new_ctx->streams[i]);
                        for (j = 0; j < new_ctx->stream_status[i].plane_count; j++)
                                dc_plane_state_retain(
                                        new_ctx->stream_status[i].plane_states[j]);
        }

        kref_init(&new_ctx->refcount);

        return new_ctx;
}

void dc_retain_state(struct dc_state *context)
{
        kref_get(&context->refcount);
}

static void dc_state_free(struct kref *kref)
{
        struct dc_state *context = container_of(kref, struct dc_state, refcount);
        dc_resource_state_destruct(context);
        kvfree(context);
}

void dc_release_state(struct dc_state *context)
{
        kref_put(&context->refcount, dc_state_free);
}

bool dc_set_generic_gpio_for_stereo(bool enable,
                struct gpio_service *gpio_service)
{
        enum gpio_result gpio_result = GPIO_RESULT_NON_SPECIFIC_ERROR;
        struct gpio_pin_info pin_info;
        struct gpio *generic;
        struct gpio_generic_mux_config *config = kzalloc(sizeof(struct gpio_generic_mux_config),
                           GFP_KERNEL);

        if (!config)
                return false;
        pin_info = dal_gpio_get_generic_pin_info(gpio_service, GPIO_ID_GENERIC, 0);

        if (pin_info.mask == 0xFFFFFFFF || pin_info.offset == 0xFFFFFFFF) {
                kfree(config);
                return false;
        } else {
                generic = dal_gpio_service_create_generic_mux(
                        gpio_service,
                        pin_info.offset,
                        pin_info.mask);
        }

        if (!generic) {
                kfree(config);
                return false;
        }

        gpio_result = dal_gpio_open(generic, GPIO_MODE_OUTPUT);

        config->enable_output_from_mux = enable;
        config->mux_select = GPIO_SIGNAL_SOURCE_PASS_THROUGH_STEREO_SYNC;

        if (gpio_result == GPIO_RESULT_OK)
                gpio_result = dal_mux_setup_config(generic, config);

        if (gpio_result == GPIO_RESULT_OK) {
                dal_gpio_close(generic);
                dal_gpio_destroy_generic_mux(&generic);
                kfree(config);
                return true;
        } else {
                dal_gpio_close(generic);
                dal_gpio_destroy_generic_mux(&generic);
                kfree(config);
                return false;
        }
}

static bool is_surface_in_context(
                const struct dc_state *context,
                const struct dc_plane_state *plane_state)
{
        int j;

        for (j = 0; j < MAX_PIPES; j++) {
                const struct pipe_ctx *pipe_ctx = &context->res_ctx.pipe_ctx[j];

                if (plane_state == pipe_ctx->plane_state) {
                        return true;
                }
        }

        return false;
}

static enum surface_update_type get_plane_info_update_type(const struct dc_surface_update *u)
{
        union surface_update_flags *update_flags = &u->surface->update_flags;
        enum surface_update_type update_type = UPDATE_TYPE_FAST;

        if (!u->plane_info)
                return UPDATE_TYPE_FAST;

        if (u->plane_info->color_space != u->surface->color_space) {
                update_flags->bits.color_space_change = 1;
                elevate_update_type(&update_type, UPDATE_TYPE_MED);
        }

        if (u->plane_info->horizontal_mirror != u->surface->horizontal_mirror) {
                update_flags->bits.horizontal_mirror_change = 1;
                elevate_update_type(&update_type, UPDATE_TYPE_MED);
        }

        if (u->plane_info->rotation != u->surface->rotation) {
                update_flags->bits.rotation_change = 1;
                elevate_update_type(&update_type, UPDATE_TYPE_FULL);
        }

        if (u->plane_info->format != u->surface->format) {
                update_flags->bits.pixel_format_change = 1;
                elevate_update_type(&update_type, UPDATE_TYPE_FULL);
        }

        if (u->plane_info->stereo_format != u->surface->stereo_format) {
                update_flags->bits.stereo_format_change = 1;
                elevate_update_type(&update_type, UPDATE_TYPE_FULL);
        }

        if (u->plane_info->per_pixel_alpha != u->surface->per_pixel_alpha) {
                update_flags->bits.per_pixel_alpha_change = 1;
                elevate_update_type(&update_type, UPDATE_TYPE_MED);
        }

        if (u->plane_info->global_alpha_value != u->surface->global_alpha_value) {
                update_flags->bits.global_alpha_change = 1;
                elevate_update_type(&update_type, UPDATE_TYPE_MED);
        }

        if (u->plane_info->dcc.enable != u->surface->dcc.enable
                        || u->plane_info->dcc.independent_64b_blks != u->surface->dcc.independent_64b_blks
                        || u->plane_info->dcc.meta_pitch != u->surface->dcc.meta_pitch) {
                /* During DCC on/off, stutter period is calculated before
                 * DCC has fully transitioned. This results in incorrect
                 * stutter period calculation. Triggering a full update will
                 * recalculate stutter period.
                 */
                update_flags->bits.dcc_change = 1;
                elevate_update_type(&update_type, UPDATE_TYPE_FULL);
        }

        if (resource_pixel_format_to_bpp(u->plane_info->format) !=
                        resource_pixel_format_to_bpp(u->surface->format)) {
                /* different bytes per element will require full bandwidth
                 * and DML calculation
                 */
                update_flags->bits.bpp_change = 1;
                elevate_update_type(&update_type, UPDATE_TYPE_FULL);
        }

        if (u->plane_info->plane_size.surface_pitch != u->surface->plane_size.surface_pitch
                        || u->plane_info->plane_size.chroma_pitch != u->surface->plane_size.chroma_pitch) {
                update_flags->bits.plane_size_change = 1;
                elevate_update_type(&update_type, UPDATE_TYPE_MED);
        }