/*
 * Copyright 2022 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 "link_hwss_hpo_dp.h"
#include "dm_helpers.h"
#include "core_types.h"
#include "dccg.h"
#include "clk_mgr.h"

#define DC_LOGGER link->ctx->logger

void set_hpo_dp_throttled_vcp_size(struct pipe_ctx *pipe_ctx,
                struct fixed31_32 throttled_vcp_size)
{
        struct hpo_dp_stream_encoder *hpo_dp_stream_encoder =
                        pipe_ctx->stream_res.hpo_dp_stream_enc;
        struct hpo_dp_link_encoder *hpo_dp_link_encoder =
                        pipe_ctx->link_res.hpo_dp_link_enc;

        hpo_dp_link_encoder->funcs->set_throttled_vcp_size(hpo_dp_link_encoder,
                        hpo_dp_stream_encoder->inst,
                        throttled_vcp_size);
}

void set_hpo_dp_hblank_min_symbol_width(struct pipe_ctx *pipe_ctx,
                const struct dc_link_settings *link_settings,
                struct fixed31_32 throttled_vcp_size)
{
        struct hpo_dp_stream_encoder *hpo_dp_stream_encoder =
                        pipe_ctx->stream_res.hpo_dp_stream_enc;
        struct dc_crtc_timing *timing = &pipe_ctx->stream->timing;
        struct fixed31_32 h_blank_in_ms, time_slot_in_ms, mtp_cnt_per_h_blank;
        uint32_t link_bw_in_kbps =
                        hpo_dp_stream_encoder->ctx->dc->link_srv->dp_link_bandwidth_kbps(
                                        pipe_ctx->stream->link, link_settings);
        uint16_t hblank_min_symbol_width = 0;

        if (link_bw_in_kbps > 0) {
                h_blank_in_ms = dc_fixpt_div(dc_fixpt_from_int(
                                timing->h_total - timing->h_addressable),
                                dc_fixpt_from_fraction(timing->pix_clk_100hz, 10));
                time_slot_in_ms = dc_fixpt_from_fraction(32 * 4, link_bw_in_kbps);
                mtp_cnt_per_h_blank = dc_fixpt_div(h_blank_in_ms,
                                dc_fixpt_mul_int(time_slot_in_ms, 64));
                hblank_min_symbol_width = dc_fixpt_floor(
                                dc_fixpt_mul(mtp_cnt_per_h_blank, throttled_vcp_size));
        }

        hpo_dp_stream_encoder->funcs->set_hblank_min_symbol_width(hpo_dp_stream_encoder,
                        hblank_min_symbol_width);
}

void setup_hpo_dp_stream_encoder(struct pipe_ctx *pipe_ctx)
{
        struct hpo_dp_stream_encoder *stream_enc = pipe_ctx->stream_res.hpo_dp_stream_enc;
        struct hpo_dp_link_encoder *link_enc = pipe_ctx->link_res.hpo_dp_link_enc;

        stream_enc->funcs->enable_stream(stream_enc);
        stream_enc->funcs->map_stream_to_link(stream_enc, stream_enc->inst, link_enc->inst);
}

void reset_hpo_dp_stream_encoder(struct pipe_ctx *pipe_ctx)
{
        struct hpo_dp_stream_encoder *stream_enc = pipe_ctx->stream_res.hpo_dp_stream_enc;

        stream_enc->funcs->disable(stream_enc);
}

void setup_hpo_dp_stream_attribute(struct pipe_ctx *pipe_ctx)
{
        struct hpo_dp_stream_encoder *stream_enc = pipe_ctx->stream_res.hpo_dp_stream_enc;
        struct dc_stream_state *stream = pipe_ctx->stream;
        struct dc_link *link = stream->link;

        stream_enc->funcs->set_stream_attribute(
                        stream_enc,
                        &stream->timing,
                        stream->output_color_space,
                        stream->use_vsc_sdp_for_colorimetry,
                        stream->timing.flags.DSC,
                        false);
        link->dc->link_srv->dp_trace_source_sequence(link,
                        DPCD_SOURCE_SEQ_AFTER_DP_STREAM_ATTR);
}

void enable_hpo_dp_link_output(struct dc_link *link,
                const struct link_resource *link_res,
                enum signal_type signal,
                enum clock_source_id clock_source,
                const struct dc_link_settings *link_settings)
{
        if (!link_res->hpo_dp_link_enc) {
                DC_LOG_ERROR("%s: invalid hpo_dp_link_enc\n", __func__);
                return;
        }

        if (link->dc->res_pool->dccg->funcs->set_symclk32_le_root_clock_gating)
                link->dc->res_pool->dccg->funcs->set_symclk32_le_root_clock_gating(
                                link->dc->res_pool->dccg,
                                link_res->hpo_dp_link_enc->inst,
                                true);
        link_res->hpo_dp_link_enc->funcs->enable_link_phy(
                        link_res->hpo_dp_link_enc,
                        link_settings,
                        link->link_enc->transmitter,
                        link->link_enc->hpd_source);
}

void disable_hpo_dp_link_output(struct dc_link *link,
                const struct link_resource *link_res,
                enum signal_type signal)
{
        if (!link_res->hpo_dp_link_enc) {
                DC_LOG_ERROR("%s: invalid hpo_dp_link_enc\n", __func__);
                return;
        }

                link_res->hpo_dp_link_enc->funcs->link_disable(link_res->hpo_dp_link_enc);
                link_res->hpo_dp_link_enc->funcs->disable_link_phy(
                                link_res->hpo_dp_link_enc, signal);
                if (link->dc->res_pool->dccg->funcs->set_symclk32_le_root_clock_gating)
                        link->dc->res_pool->dccg->funcs->set_symclk32_le_root_clock_gating(
                                        link->dc->res_pool->dccg,
                                        link_res->hpo_dp_link_enc->inst,
                                        false);
}

static void set_hpo_dp_link_test_pattern(struct dc_link *link,
                const struct link_resource *link_res,
                struct encoder_set_dp_phy_pattern_param *tp_params)
{
        link_res->hpo_dp_link_enc->funcs->set_link_test_pattern(
                        link_res->hpo_dp_link_enc, tp_params);
        link->dc->link_srv->dp_trace_source_sequence(link, DPCD_SOURCE_SEQ_AFTER_SET_SOURCE_PATTERN);
}

static void set_hpo_dp_lane_settings(struct dc_link *link,
                const struct link_resource *link_res,
                const struct dc_link_settings *link_settings,
                const struct dc_lane_settings lane_settings[LANE_COUNT_DP_MAX])
{
        link_res->hpo_dp_link_enc->funcs->set_ffe(
                        link_res->hpo_dp_link_enc,
                        link_settings,
                        lane_settings[0].FFE_PRESET.raw);
}

void update_hpo_dp_stream_allocation_table(struct dc_link *link,
                const struct link_resource *link_res,
                const struct link_mst_stream_allocation_table *table)
{
        link_res->hpo_dp_link_enc->funcs->update_stream_allocation_table(
                        link_res->hpo_dp_link_enc,
                        table);
}

void setup_hpo_dp_audio_output(struct pipe_ctx *pipe_ctx,
                struct audio_output *audio_output, uint32_t audio_inst)
{
        pipe_ctx->stream_res.hpo_dp_stream_enc->funcs->dp_audio_setup(
                        pipe_ctx->stream_res.hpo_dp_stream_enc,
                        audio_inst,
                        &pipe_ctx->stream->audio_info);
}

void enable_hpo_dp_audio_packet(struct pipe_ctx *pipe_ctx)
{
        pipe_ctx->stream_res.hpo_dp_stream_enc->funcs->dp_audio_enable(
                        pipe_ctx->stream_res.hpo_dp_stream_enc);
}

void disable_hpo_dp_audio_packet(struct pipe_ctx *pipe_ctx)
{
        if (pipe_ctx->stream_res.audio)
                pipe_ctx->stream_res.hpo_dp_stream_enc->funcs->dp_audio_disable(
                                pipe_ctx->stream_res.hpo_dp_stream_enc);
}

static const struct link_hwss hpo_dp_link_hwss = {
        .setup_stream_encoder = setup_hpo_dp_stream_encoder,
        .reset_stream_encoder = reset_hpo_dp_stream_encoder,
        .setup_stream_attribute = setup_hpo_dp_stream_attribute,
        .disable_link_output = disable_hpo_dp_link_output,
        .setup_audio_output = setup_hpo_dp_audio_output,
        .enable_audio_packet = enable_hpo_dp_audio_packet,
        .disable_audio_packet = disable_hpo_dp_audio_packet,
        .ext = {
                .set_throttled_vcp_size = set_hpo_dp_throttled_vcp_size,
                .set_hblank_min_symbol_width = set_hpo_dp_hblank_min_symbol_width,
                .enable_dp_link_output = enable_hpo_dp_link_output,
                .set_dp_link_test_pattern  = set_hpo_dp_link_test_pattern,
                .set_dp_lane_settings = set_hpo_dp_lane_settings,
                .update_stream_allocation_table = update_hpo_dp_stream_allocation_table,
        },
};

bool can_use_hpo_dp_link_hwss(const struct dc_link *link,
                const struct link_resource *link_res)
{
        return link_res->hpo_dp_link_enc != NULL;
}

const struct link_hwss *get_hpo_dp_link_hwss(void)
{
        return &hpo_dp_link_hwss;
}