// SPDX-License-Identifier: MIT
/*
 * Copyright 2018 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/string.h>
#include <linux/acpi.h>

#include <drm/drm_probe_helper.h>
#include <drm/amdgpu_drm.h>
#include "dm_services.h"
#include "amdgpu.h"
#include "amdgpu_dm.h"
#include "amdgpu_dm_irq.h"
#include "amdgpu_pm.h"
#include "dm_pp_smu.h"
#include "dm_helpers.h"
#include "amdgpu_dm_pp_smu.h"

STATIC_IFN_KUNIT void build_pm_display_cfg(
		struct amd_pp_display_configuration *pm_display_cfg,
		const struct dm_pp_display_configuration *pp_display_cfg)
{
	int i;

	memset(pm_display_cfg, 0, sizeof(*pm_display_cfg));

	pm_display_cfg->cpu_cc6_disable = pp_display_cfg->cpu_cc6_disable;
	pm_display_cfg->cpu_pstate_disable = pp_display_cfg->cpu_pstate_disable;
	pm_display_cfg->cpu_pstate_separation_time = pp_display_cfg->cpu_pstate_separation_time;
	pm_display_cfg->nb_pstate_switch_disable = pp_display_cfg->nb_pstate_switch_disable;

	pm_display_cfg->num_display = pp_display_cfg->display_count;
	pm_display_cfg->num_path_including_non_display = pp_display_cfg->display_count;

	pm_display_cfg->min_core_set_clock = pp_display_cfg->min_engine_clock_khz/10;
	pm_display_cfg->min_core_set_clock_in_sr =
			pp_display_cfg->min_engine_clock_deep_sleep_khz/10;
	pm_display_cfg->min_mem_set_clock = pp_display_cfg->min_memory_clock_khz/10;

	pm_display_cfg->min_dcef_deep_sleep_set_clk =
			pp_display_cfg->min_engine_clock_deep_sleep_khz/10;
	pm_display_cfg->min_dcef_set_clk = pp_display_cfg->min_dcfclock_khz/10;

	pm_display_cfg->multi_monitor_in_sync = pp_display_cfg->all_displays_in_sync;
	pm_display_cfg->min_vblank_time = pp_display_cfg->avail_mclk_switch_time_us;

	pm_display_cfg->display_clk = pp_display_cfg->disp_clk_khz/10;

	pm_display_cfg->dce_tolerable_mclk_in_active_latency =
			pp_display_cfg->avail_mclk_switch_time_in_disp_active_us;

	pm_display_cfg->crtc_index = pp_display_cfg->crtc_index;
	pm_display_cfg->line_time_in_us = pp_display_cfg->line_time_in_us;

	pm_display_cfg->vrefresh = pp_display_cfg->disp_configs[0].v_refresh;
	pm_display_cfg->crossfire_display_index = -1;
	pm_display_cfg->min_bus_bandwidth = 0;

	for (i = 0; i < pp_display_cfg->display_count; i++) {
		const struct dm_pp_single_disp_config *dc_cfg =
					&pp_display_cfg->disp_configs[i];
		pm_display_cfg->displays[i].controller_id = dc_cfg->pipe_idx + 1;
		pm_display_cfg->displays[i].pixel_clock = dc_cfg->pixel_clock;
	}
}
EXPORT_IF_KUNIT(build_pm_display_cfg);

bool dm_pp_apply_display_requirements(
		const struct dc_context *ctx,
		const struct dm_pp_display_configuration *pp_display_cfg)
{
	struct amdgpu_device *adev = ctx->driver_context;

	if (adev->pm.dpm_enabled) {

		build_pm_display_cfg(&adev->pm.pm_display_cfg, pp_display_cfg);

		amdgpu_dpm_display_configuration_change(adev, &adev->pm.pm_display_cfg);

		amdgpu_dpm_compute_clocks(adev);
	}

	return true;
}
EXPORT_IF_KUNIT(dm_pp_apply_display_requirements);

STATIC_IFN_KUNIT void get_default_clock_levels(
		enum dm_pp_clock_type clk_type,
		struct dm_pp_clock_levels *clks)
{
	uint32_t disp_clks_in_khz[6] = {
			300000, 400000, 496560, 626090, 685720, 757900 };
	uint32_t sclks_in_khz[6] = {
			300000, 360000, 423530, 514290, 626090, 720000 };
	uint32_t mclks_in_khz[2] = { 333000, 800000 };

	switch (clk_type) {
	case DM_PP_CLOCK_TYPE_DISPLAY_CLK:
		clks->num_levels = 6;
		memmove(clks->clocks_in_khz, disp_clks_in_khz,
				sizeof(disp_clks_in_khz));
		break;
	case DM_PP_CLOCK_TYPE_ENGINE_CLK:
		clks->num_levels = 6;
		memmove(clks->clocks_in_khz, sclks_in_khz,
				sizeof(sclks_in_khz));
		break;
	case DM_PP_CLOCK_TYPE_MEMORY_CLK:
		clks->num_levels = 2;
		memmove(clks->clocks_in_khz, mclks_in_khz,
				sizeof(mclks_in_khz));
		break;
	default:
		clks->num_levels = 0;
		break;
	}
}
EXPORT_IF_KUNIT(get_default_clock_levels);

STATIC_IFN_KUNIT enum amd_pp_clock_type dc_to_pp_clock_type(
		enum dm_pp_clock_type dm_pp_clk_type)
{
	enum amd_pp_clock_type amd_pp_clk_type = 0;

	switch (dm_pp_clk_type) {
	case DM_PP_CLOCK_TYPE_DISPLAY_CLK:
		amd_pp_clk_type = amd_pp_disp_clock;
		break;
	case DM_PP_CLOCK_TYPE_ENGINE_CLK:
		amd_pp_clk_type = amd_pp_sys_clock;
		break;
	case DM_PP_CLOCK_TYPE_MEMORY_CLK:
		amd_pp_clk_type = amd_pp_mem_clock;
		break;
	case DM_PP_CLOCK_TYPE_DCEFCLK:
		amd_pp_clk_type  = amd_pp_dcef_clock;
		break;
	case DM_PP_CLOCK_TYPE_DCFCLK:
		amd_pp_clk_type = amd_pp_dcf_clock;
		break;
	case DM_PP_CLOCK_TYPE_PIXELCLK:
		amd_pp_clk_type = amd_pp_pixel_clock;
		break;
	case DM_PP_CLOCK_TYPE_FCLK:
		amd_pp_clk_type = amd_pp_f_clock;
		break;
	case DM_PP_CLOCK_TYPE_DISPLAYPHYCLK:
		amd_pp_clk_type = amd_pp_phy_clock;
		break;
	case DM_PP_CLOCK_TYPE_DPPCLK:
		amd_pp_clk_type = amd_pp_dpp_clock;
		break;
	default:
		DRM_ERROR("DM_PPLIB: invalid clock type: %d!\n",
				dm_pp_clk_type);
		break;
	}

	return amd_pp_clk_type;
}
EXPORT_IF_KUNIT(dc_to_pp_clock_type);

STATIC_IFN_KUNIT void pp_to_dc_clock_levels(
		const struct amd_pp_clocks *pp_clks,
		struct dm_pp_clock_levels *dc_clks,
		enum dm_pp_clock_type dc_clk_type)
{
	uint32_t i;

	if (pp_clks->count > DM_PP_MAX_CLOCK_LEVELS) {
		DRM_INFO("DM_PPLIB: Warning: %s clock: number of levels %d exceeds maximum of %d!\n",
				DC_DECODE_PP_CLOCK_TYPE(dc_clk_type),
				pp_clks->count,
				DM_PP_MAX_CLOCK_LEVELS);

		dc_clks->num_levels = DM_PP_MAX_CLOCK_LEVELS;
	} else
		dc_clks->num_levels = pp_clks->count;

	DRM_INFO("DM_PPLIB: values for %s clock\n",
			DC_DECODE_PP_CLOCK_TYPE(dc_clk_type));

	for (i = 0; i < dc_clks->num_levels; i++) {
		DRM_INFO("DM_PPLIB:\t %d\n", pp_clks->clock[i]);
		dc_clks->clocks_in_khz[i] = pp_clks->clock[i];
	}
}
EXPORT_IF_KUNIT(pp_to_dc_clock_levels);

STATIC_IFN_KUNIT void pp_to_dc_clock_levels_with_latency(
		const struct pp_clock_levels_with_latency *pp_clks,
		struct dm_pp_clock_levels_with_latency *clk_level_info,
		enum dm_pp_clock_type dc_clk_type)
{
	uint32_t i;

	if (pp_clks->num_levels > DM_PP_MAX_CLOCK_LEVELS) {
		DRM_INFO("DM_PPLIB: Warning: %s clock: number of levels %d exceeds maximum of %d!\n",
				DC_DECODE_PP_CLOCK_TYPE(dc_clk_type),
				pp_clks->num_levels,
				DM_PP_MAX_CLOCK_LEVELS);

		clk_level_info->num_levels = DM_PP_MAX_CLOCK_LEVELS;
	} else
		clk_level_info->num_levels = pp_clks->num_levels;

	DRM_DEBUG("DM_PPLIB: values for %s clock\n",
			DC_DECODE_PP_CLOCK_TYPE(dc_clk_type));

	for (i = 0; i < clk_level_info->num_levels; i++) {
		DRM_DEBUG("DM_PPLIB:\t %d in kHz\n", pp_clks->data[i].clocks_in_khz);
		clk_level_info->data[i].clocks_in_khz = pp_clks->data[i].clocks_in_khz;
		clk_level_info->data[i].latency_in_us = pp_clks->data[i].latency_in_us;
	}
}
EXPORT_IF_KUNIT(pp_to_dc_clock_levels_with_latency);

STATIC_IFN_KUNIT void pp_to_dc_clock_levels_with_voltage(
		const struct pp_clock_levels_with_voltage *pp_clks,
		struct dm_pp_clock_levels_with_voltage *clk_level_info,
		enum dm_pp_clock_type dc_clk_type)
{
	uint32_t i;

	if (pp_clks->num_levels > DM_PP_MAX_CLOCK_LEVELS) {
		DRM_INFO("DM_PPLIB: Warning: %s clock: number of levels %d exceeds maximum of %d!\n",
				DC_DECODE_PP_CLOCK_TYPE(dc_clk_type),
				pp_clks->num_levels,
				DM_PP_MAX_CLOCK_LEVELS);

		clk_level_info->num_levels = DM_PP_MAX_CLOCK_LEVELS;
	} else
		clk_level_info->num_levels = pp_clks->num_levels;

	DRM_INFO("DM_PPLIB: values for %s clock\n",
			DC_DECODE_PP_CLOCK_TYPE(dc_clk_type));

	for (i = 0; i < clk_level_info->num_levels; i++) {
		DRM_INFO("DM_PPLIB:\t %d in kHz, %d in mV\n", pp_clks->data[i].clocks_in_khz,
			 pp_clks->data[i].voltage_in_mv);
		clk_level_info->data[i].clocks_in_khz = pp_clks->data[i].clocks_in_khz;
		clk_level_info->data[i].voltage_in_mv = pp_clks->data[i].voltage_in_mv;
	}
}
EXPORT_IF_KUNIT(pp_to_dc_clock_levels_with_voltage);

STATIC_IFN_KUNIT void cap_clock_levels_to_validation(
		struct dm_pp_clock_levels *dc_clks,
		enum dm_pp_clock_type clk_type,
		const struct amd_pp_simple_clock_info *validation_clks)
{
	uint32_t i;

	/* Determine the highest non-boosted level from the Validation Clocks */
	if (clk_type == DM_PP_CLOCK_TYPE_ENGINE_CLK) {
		for (i = 0; i < dc_clks->num_levels; i++) {
			if (dc_clks->clocks_in_khz[i] > validation_clks->engine_max_clock) {
				/* This clock is higher the validation clock.
				 * Than means the previous one is the highest
				 * non-boosted one.
				 */
				DRM_INFO("DM_PPLIB: reducing engine clock level from %d to %d\n",
						dc_clks->num_levels, i);
				dc_clks->num_levels = i > 0 ? i : 1;
				break;
			}
		}
	} else if (clk_type == DM_PP_CLOCK_TYPE_MEMORY_CLK) {
		for (i = 0; i < dc_clks->num_levels; i++) {
			if (dc_clks->clocks_in_khz[i] > validation_clks->memory_max_clock) {
				DRM_INFO("DM_PPLIB: reducing memory clock level from %d to %d\n",
						dc_clks->num_levels, i);
				dc_clks->num_levels = i > 0 ? i : 1;
				break;
			}
		}
	}
}
EXPORT_IF_KUNIT(cap_clock_levels_to_validation);

bool dm_pp_get_clock_levels_by_type(
		const struct dc_context *ctx,
		enum dm_pp_clock_type clk_type,
		struct dm_pp_clock_levels *dc_clks)
{
	struct amdgpu_device *adev = ctx->driver_context;
	struct amd_pp_clocks pp_clks = { 0 };
	struct amd_pp_simple_clock_info validation_clks = { 0 };

	if (amdgpu_dpm_get_clock_by_type(adev,
		dc_to_pp_clock_type(clk_type), &pp_clks)) {
		/* Error in pplib. Provide default values. */
		get_default_clock_levels(clk_type, dc_clks);
		return true;
	}

	pp_to_dc_clock_levels(&pp_clks, dc_clks, clk_type);

	if (amdgpu_dpm_get_display_mode_validation_clks(adev, &validation_clks)) {
		/* Error in pplib. Provide default values. */
		DRM_INFO("DM_PPLIB: Warning: using default validation clocks!\n");
		validation_clks.engine_max_clock = 72000;
		validation_clks.memory_max_clock = 80000;
	}

	DRM_INFO("DM_PPLIB: Validation clocks:\n");
	DRM_INFO("DM_PPLIB:    engine_max_clock: %d\n",
			validation_clks.engine_max_clock);
	DRM_INFO("DM_PPLIB:    memory_max_clock: %d\n",
			validation_clks.memory_max_clock);

	/* Translate 10 kHz to kHz. */
	validation_clks.engine_max_clock *= 10;
	validation_clks.memory_max_clock *= 10;

	cap_clock_levels_to_validation(dc_clks, clk_type, &validation_clks);

	return true;
}
EXPORT_IF_KUNIT(dm_pp_get_clock_levels_by_type);

bool dm_pp_get_clock_levels_by_type_with_latency(
	const struct dc_context *ctx,
	enum dm_pp_clock_type clk_type,
	struct dm_pp_clock_levels_with_latency *clk_level_info)
{
	struct amdgpu_device *adev = ctx->driver_context;
	struct pp_clock_levels_with_latency pp_clks = { 0 };
	int ret;

	ret = amdgpu_dpm_get_clock_by_type_with_latency(adev,
					dc_to_pp_clock_type(clk_type),
					&pp_clks);
	if (ret)
		return false;

	pp_to_dc_clock_levels_with_latency(&pp_clks, clk_level_info, clk_type);

	return true;
}
EXPORT_IF_KUNIT(dm_pp_get_clock_levels_by_type_with_latency);

bool dm_pp_get_clock_levels_by_type_with_voltage(
	const struct dc_context *ctx,
	enum dm_pp_clock_type clk_type,
	struct dm_pp_clock_levels_with_voltage *clk_level_info)
{
	struct amdgpu_device *adev = ctx->driver_context;
	struct pp_clock_levels_with_voltage pp_clk_info = {0};
	int ret;

	ret = amdgpu_dpm_get_clock_by_type_with_voltage(adev,
					dc_to_pp_clock_type(clk_type),
					&pp_clk_info);
	if (ret)
		return false;

	pp_to_dc_clock_levels_with_voltage(&pp_clk_info, clk_level_info, clk_type);

	return true;
}
EXPORT_IF_KUNIT(dm_pp_get_clock_levels_by_type_with_voltage);

bool dm_pp_notify_wm_clock_changes(
	const struct dc_context *ctx,
	struct dm_pp_wm_sets_with_clock_ranges *wm_with_clock_ranges)
{
	struct amdgpu_device *adev = ctx->driver_context;

	/*
	 * Limit this watermark setting for Polaris for now
	 * TODO: expand this to other ASICs
	 */
	if ((adev->asic_type >= CHIP_POLARIS10) &&
	    (adev->asic_type <= CHIP_VEGAM) &&
	    !amdgpu_dpm_set_watermarks_for_clocks_ranges(adev,
							 (void *)wm_with_clock_ranges))
		return true;

	return false;
}
EXPORT_IF_KUNIT(dm_pp_notify_wm_clock_changes);

bool dm_pp_apply_clock_for_voltage_request(
	const struct dc_context *ctx,
	struct dm_pp_clock_for_voltage_req *clock_for_voltage_req)
{
	struct amdgpu_device *adev = ctx->driver_context;
	struct pp_display_clock_request pp_clock_request = {0};
	int ret = 0;

	pp_clock_request.clock_type = dc_to_pp_clock_type(clock_for_voltage_req->clk_type);
	pp_clock_request.clock_freq_in_khz = clock_for_voltage_req->clocks_in_khz;

	if (!pp_clock_request.clock_type)
		return false;

	ret = amdgpu_dpm_display_clock_voltage_request(adev, &pp_clock_request);
	if (ret && (ret != -EOPNOTSUPP))
		return false;

	return true;
}
EXPORT_IF_KUNIT(dm_pp_apply_clock_for_voltage_request);

STATIC_IFN_KUNIT void build_wm_clock_ranges_soc15(
		const struct pp_smu_wm_range_sets *ranges,
		struct dm_pp_wm_sets_with_clock_ranges_soc15 *wm_with_clock_ranges)
{
	struct dm_pp_clock_range_for_dmif_wm_set_soc15 *wm_dce_clocks =
			wm_with_clock_ranges->wm_dmif_clocks_ranges;
	struct dm_pp_clock_range_for_mcif_wm_set_soc15 *wm_soc_clocks =
			wm_with_clock_ranges->wm_mcif_clocks_ranges;
	int32_t i;

	wm_with_clock_ranges->num_wm_dmif_sets = ranges->num_reader_wm_sets;
	wm_with_clock_ranges->num_wm_mcif_sets = ranges->num_writer_wm_sets;

	for (i = 0; i < wm_with_clock_ranges->num_wm_dmif_sets; i++) {
		if (ranges->reader_wm_sets[i].wm_inst > 3)
			wm_dce_clocks[i].wm_set_id = WM_SET_A;
		else
			wm_dce_clocks[i].wm_set_id = ranges->reader_wm_sets[i].wm_inst;
		wm_dce_clocks[i].wm_max_dcfclk_clk_in_khz =
				ranges->reader_wm_sets[i].max_drain_clk_mhz * 1000;
		wm_dce_clocks[i].wm_min_dcfclk_clk_in_khz =
				ranges->reader_wm_sets[i].min_drain_clk_mhz * 1000;
		wm_dce_clocks[i].wm_max_mem_clk_in_khz =
				ranges->reader_wm_sets[i].max_fill_clk_mhz * 1000;
		wm_dce_clocks[i].wm_min_mem_clk_in_khz =
				ranges->reader_wm_sets[i].min_fill_clk_mhz * 1000;
	}

	for (i = 0; i < wm_with_clock_ranges->num_wm_mcif_sets; i++) {
		if (ranges->writer_wm_sets[i].wm_inst > 3)
			wm_soc_clocks[i].wm_set_id = WM_SET_A;
		else
			wm_soc_clocks[i].wm_set_id = ranges->writer_wm_sets[i].wm_inst;
		wm_soc_clocks[i].wm_max_socclk_clk_in_khz =
				ranges->writer_wm_sets[i].max_fill_clk_mhz * 1000;
		wm_soc_clocks[i].wm_min_socclk_clk_in_khz =
				ranges->writer_wm_sets[i].min_fill_clk_mhz * 1000;
		wm_soc_clocks[i].wm_max_mem_clk_in_khz =
				ranges->writer_wm_sets[i].max_drain_clk_mhz * 1000;
		wm_soc_clocks[i].wm_min_mem_clk_in_khz =
				ranges->writer_wm_sets[i].min_drain_clk_mhz * 1000;
	}
}
EXPORT_IF_KUNIT(build_wm_clock_ranges_soc15);

STATIC_IFN_KUNIT void pp_rv_set_wm_ranges(struct pp_smu *pp,
		struct pp_smu_wm_range_sets *ranges)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;
	struct dm_pp_wm_sets_with_clock_ranges_soc15 wm_with_clock_ranges;

	build_wm_clock_ranges_soc15(ranges, &wm_with_clock_ranges);

	amdgpu_dpm_set_watermarks_for_clocks_ranges(adev,
						    &wm_with_clock_ranges);
}
EXPORT_IF_KUNIT(pp_rv_set_wm_ranges);

STATIC_IFN_KUNIT void pp_rv_set_pme_wa_enable(struct pp_smu *pp)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;

	amdgpu_dpm_notify_smu_enable_pwe(adev);
}
EXPORT_IF_KUNIT(pp_rv_set_pme_wa_enable);

STATIC_IFN_KUNIT void pp_rv_set_active_display_count(struct pp_smu *pp, int count)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;

	amdgpu_dpm_set_active_display_count(adev, count);
}
EXPORT_IF_KUNIT(pp_rv_set_active_display_count);

STATIC_IFN_KUNIT void pp_rv_set_min_deep_sleep_dcfclk(struct pp_smu *pp, int clock)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;

	amdgpu_dpm_set_min_deep_sleep_dcefclk(adev, clock);
}
EXPORT_IF_KUNIT(pp_rv_set_min_deep_sleep_dcfclk);

STATIC_IFN_KUNIT void pp_rv_set_hard_min_dcefclk_by_freq(struct pp_smu *pp, int clock)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;

	amdgpu_dpm_set_hard_min_dcefclk_by_freq(adev, clock);
}
EXPORT_IF_KUNIT(pp_rv_set_hard_min_dcefclk_by_freq);

STATIC_IFN_KUNIT void pp_rv_set_hard_min_fclk_by_freq(struct pp_smu *pp, int mhz)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;

	amdgpu_dpm_set_hard_min_fclk_by_freq(adev, mhz);
}
EXPORT_IF_KUNIT(pp_rv_set_hard_min_fclk_by_freq);

STATIC_IFN_KUNIT enum pp_smu_status pp_nv_set_wm_ranges(struct pp_smu *pp,
		struct pp_smu_wm_range_sets *ranges)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;

	amdgpu_dpm_set_watermarks_for_clocks_ranges(adev, ranges);

	return PP_SMU_RESULT_OK;
}
EXPORT_IF_KUNIT(pp_nv_set_wm_ranges);

STATIC_IFN_KUNIT enum pp_smu_status pp_nv_set_display_count(struct pp_smu *pp, int count)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;
	int ret = 0;

	ret = amdgpu_dpm_set_active_display_count(adev, count);
	if (ret == -EOPNOTSUPP)
		return PP_SMU_RESULT_UNSUPPORTED;
	else if (ret)
		/* 0: successful or smu.ppt_funcs->set_display_count = NULL;  1: fail */
		return PP_SMU_RESULT_FAIL;

	return PP_SMU_RESULT_OK;
}
EXPORT_IF_KUNIT(pp_nv_set_display_count);

STATIC_IFN_KUNIT enum pp_smu_status
pp_nv_set_min_deep_sleep_dcfclk(struct pp_smu *pp, int mhz)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;
	int ret = 0;

	/* 0: successful or smu.ppt_funcs->set_deep_sleep_dcefclk = NULL;1: fail */
	ret = amdgpu_dpm_set_min_deep_sleep_dcefclk(adev, mhz);
	if (ret == -EOPNOTSUPP)
		return PP_SMU_RESULT_UNSUPPORTED;
	else if (ret)
		return PP_SMU_RESULT_FAIL;

	return PP_SMU_RESULT_OK;
}
EXPORT_IF_KUNIT(pp_nv_set_min_deep_sleep_dcfclk);

STATIC_IFN_KUNIT enum pp_smu_status pp_nv_set_hard_min_dcefclk_by_freq(
		struct pp_smu *pp, int mhz)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;
	struct pp_display_clock_request clock_req;
	int ret = 0;

	clock_req.clock_type = amd_pp_dcef_clock;
	clock_req.clock_freq_in_khz = mhz * 1000;

	/* 0: successful or smu.ppt_funcs->display_clock_voltage_request = NULL
	 * 1: fail
	 */
	ret = amdgpu_dpm_display_clock_voltage_request(adev, &clock_req);
	if (ret == -EOPNOTSUPP)
		return PP_SMU_RESULT_UNSUPPORTED;
	else if (ret)
		return PP_SMU_RESULT_FAIL;

	return PP_SMU_RESULT_OK;
}
EXPORT_IF_KUNIT(pp_nv_set_hard_min_dcefclk_by_freq);

STATIC_IFN_KUNIT enum pp_smu_status
pp_nv_set_hard_min_uclk_by_freq(struct pp_smu *pp, int mhz)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;
	struct pp_display_clock_request clock_req;
	int ret = 0;

	clock_req.clock_type = amd_pp_mem_clock;
	clock_req.clock_freq_in_khz = mhz * 1000;

	/* 0: successful or smu.ppt_funcs->display_clock_voltage_request = NULL
	 * 1: fail
	 */
	ret = amdgpu_dpm_display_clock_voltage_request(adev, &clock_req);
	if (ret == -EOPNOTSUPP)
		return PP_SMU_RESULT_UNSUPPORTED;
	else if (ret)
		return PP_SMU_RESULT_FAIL;

	return PP_SMU_RESULT_OK;
}
EXPORT_IF_KUNIT(pp_nv_set_hard_min_uclk_by_freq);

STATIC_IFN_KUNIT enum pp_smu_status pp_nv_set_pstate_handshake_support(
	struct pp_smu *pp, bool pstate_handshake_supported)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;

	if (amdgpu_dpm_display_disable_memory_clock_switch(adev,
							  !pstate_handshake_supported))
		return PP_SMU_RESULT_FAIL;

	return PP_SMU_RESULT_OK;
}
EXPORT_IF_KUNIT(pp_nv_set_pstate_handshake_support);

STATIC_IFN_KUNIT bool pp_smu_nv_clock_id_to_pp(enum pp_smu_nv_clock_id clock_id,
		enum amd_pp_clock_type *clock_type)
{
	switch (clock_id) {
	case PP_SMU_NV_DISPCLK:
		*clock_type = amd_pp_disp_clock;
		break;
	case PP_SMU_NV_PHYCLK:
		*clock_type = amd_pp_phy_clock;
		break;
	case PP_SMU_NV_PIXELCLK:
		*clock_type = amd_pp_pixel_clock;
		break;
	default:
		return false;
	}

	return true;
}
EXPORT_IF_KUNIT(pp_smu_nv_clock_id_to_pp);

STATIC_IFN_KUNIT enum pp_smu_status pp_nv_set_voltage_by_freq(struct pp_smu *pp,
		enum pp_smu_nv_clock_id clock_id, int mhz)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;
	struct pp_display_clock_request clock_req;
	int ret = 0;

	if (!pp_smu_nv_clock_id_to_pp(clock_id, &clock_req.clock_type))
		return PP_SMU_RESULT_FAIL;

	clock_req.clock_freq_in_khz = mhz * 1000;

	/* 0: successful or smu.ppt_funcs->display_clock_voltage_request = NULL
	 * 1: fail
	 */
	ret = amdgpu_dpm_display_clock_voltage_request(adev, &clock_req);
	if (ret == -EOPNOTSUPP)
		return PP_SMU_RESULT_UNSUPPORTED;
	else if (ret)
		return PP_SMU_RESULT_FAIL;

	return PP_SMU_RESULT_OK;
}
EXPORT_IF_KUNIT(pp_nv_set_voltage_by_freq);

STATIC_IFN_KUNIT enum pp_smu_status pp_nv_get_maximum_sustainable_clocks(
		struct pp_smu *pp, struct pp_smu_nv_clock_table *max_clocks)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;
	int ret = 0;

	ret = amdgpu_dpm_get_max_sustainable_clocks_by_dc(adev,
							  max_clocks);
	if (ret == -EOPNOTSUPP)
		return PP_SMU_RESULT_UNSUPPORTED;
	else if (ret)
		return PP_SMU_RESULT_FAIL;

	return PP_SMU_RESULT_OK;
}
EXPORT_IF_KUNIT(pp_nv_get_maximum_sustainable_clocks);

STATIC_IFN_KUNIT enum pp_smu_status pp_nv_get_uclk_dpm_states(struct pp_smu *pp,
		unsigned int *clock_values_in_khz, unsigned int *num_states)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;
	int ret = 0;

	ret = amdgpu_dpm_get_uclk_dpm_states(adev,
					     clock_values_in_khz,
					    num_states);
	if (ret == -EOPNOTSUPP)
		return PP_SMU_RESULT_UNSUPPORTED;
	else if (ret)
		return PP_SMU_RESULT_FAIL;

	return PP_SMU_RESULT_OK;
}
EXPORT_IF_KUNIT(pp_nv_get_uclk_dpm_states);

STATIC_IFN_KUNIT enum pp_smu_status pp_rn_get_dpm_clock_table(
		struct pp_smu *pp, struct dpm_clocks *clock_table)
{
	const struct dc_context *ctx = pp->dm;
	struct amdgpu_device *adev = ctx->driver_context;
	int ret = 0;

	ret = amdgpu_dpm_get_dpm_clock_table(adev, clock_table);
	if (ret == -EOPNOTSUPP)
		return PP_SMU_RESULT_UNSUPPORTED;
	else if (ret)
		return PP_SMU_RESULT_FAIL;

	return PP_SMU_RESULT_OK;
}
EXPORT_IF_KUNIT(pp_rn_get_dpm_clock_table);

void dm_pp_get_funcs(
		struct dc_context *ctx,
		struct pp_smu_funcs *funcs)
{
	switch (ctx->dce_version) {
	case DCN_VERSION_1_0:
	case DCN_VERSION_1_01:
		funcs->ctx.ver = PP_SMU_VER_RV;
		funcs->rv_funcs.pp_smu.dm = ctx;
		funcs->rv_funcs.set_wm_ranges = pp_rv_set_wm_ranges;
		funcs->rv_funcs.set_pme_wa_enable = pp_rv_set_pme_wa_enable;
		funcs->rv_funcs.set_display_count =
				pp_rv_set_active_display_count;
		funcs->rv_funcs.set_min_deep_sleep_dcfclk =
				pp_rv_set_min_deep_sleep_dcfclk;
		funcs->rv_funcs.set_hard_min_dcfclk_by_freq =
				pp_rv_set_hard_min_dcefclk_by_freq;
		funcs->rv_funcs.set_hard_min_fclk_by_freq =
				pp_rv_set_hard_min_fclk_by_freq;
		break;
	case DCN_VERSION_2_0:
		funcs->ctx.ver = PP_SMU_VER_NV;
		funcs->nv_funcs.pp_smu.dm = ctx;
		funcs->nv_funcs.set_display_count = pp_nv_set_display_count;
		funcs->nv_funcs.set_hard_min_dcfclk_by_freq =
				pp_nv_set_hard_min_dcefclk_by_freq;
		funcs->nv_funcs.set_min_deep_sleep_dcfclk =
				pp_nv_set_min_deep_sleep_dcfclk;
		funcs->nv_funcs.set_voltage_by_freq =
				pp_nv_set_voltage_by_freq;
		funcs->nv_funcs.set_wm_ranges = pp_nv_set_wm_ranges;

		/* todo set_pme_wa_enable cause 4k@6ohz display not light up */
		funcs->nv_funcs.set_pme_wa_enable = NULL;
		/* todo debug waring message */
		funcs->nv_funcs.set_hard_min_uclk_by_freq = pp_nv_set_hard_min_uclk_by_freq;
		/* todo  compare data with window driver*/
		funcs->nv_funcs.get_maximum_sustainable_clocks = pp_nv_get_maximum_sustainable_clocks;
		/*todo  compare data with window driver */
		funcs->nv_funcs.get_uclk_dpm_states = pp_nv_get_uclk_dpm_states;
		funcs->nv_funcs.set_pstate_handshake_support = pp_nv_set_pstate_handshake_support;
		break;

	case DCN_VERSION_2_1:
		funcs->ctx.ver = PP_SMU_VER_RN;
		funcs->rn_funcs.pp_smu.dm = ctx;
		funcs->rn_funcs.set_wm_ranges = pp_nv_set_wm_ranges;
		funcs->rn_funcs.get_dpm_clock_table = pp_rn_get_dpm_clock_table;
		break;
	default:
		DRM_ERROR("smu version is not supported !\n");
		break;
	}
}
EXPORT_IF_KUNIT(dm_pp_get_funcs);

int amdgpu_dm_smu_write_watermarks_table(struct amdgpu_device *adev)
{
	int ret = 0;

	/* This interface is for dGPU Navi1x.Linux dc-pplib interface depends
	 * on window driver dc implementation.
	 * For Navi1x, clock settings of dcn watermarks are fixed. the settings
	 * should be passed to smu during boot up and resume from s3.
	 * boot up: dc calculate dcn watermark clock settings within dc_create,
	 * dcn20_resource_construct
	 * then call pplib functions below to pass the settings to smu:
	 * smu_set_watermarks_for_clock_ranges
	 * smu_set_watermarks_table
	 * navi10_set_watermarks_table
	 * smu_write_watermarks_table
	 *
	 * For Renoir, clock settings of dcn watermark are also fixed values.
	 * dc has implemented different flow for window driver:
	 * dc_hardware_init / dc_set_power_state
	 * dcn10_init_hw
	 * notify_wm_ranges
	 * set_wm_ranges
	 * -- Linux
	 * smu_set_watermarks_for_clock_ranges
	 * renoir_set_watermarks_table
	 * smu_write_watermarks_table
	 *
	 * For Linux,
	 * dc_hardware_init -> amdgpu_dm_init
	 * dc_set_power_state --> dm_resume
	 *
	 * therefore, this function apply to navi10/12/14 but not Renoir
	 * *
	 */
	switch (amdgpu_ip_version(adev, DCE_HWIP, 0)) {
	case IP_VERSION(2, 0, 2):
	case IP_VERSION(2, 0, 0):
		break;
	default:
		return 0;
	}

	ret = amdgpu_dpm_write_watermarks_table(adev);
	if (ret) {
		drm_err(adev_to_drm(adev), "Failed to update WMTABLE!\n");
		return ret;
	}

	return 0;
}
EXPORT_IF_KUNIT(amdgpu_dm_smu_write_watermarks_table);