/*
 * Copyright (c) 2018, ARM Limited and Contributors. All rights reserved.
 *
 * SPDX-License-Identifier: BSD-3-Clause
 */

#include <stdbool.h>

#include <arch.h>
#include <arch_helpers.h>
#include <common/debug.h>

#include <plat_imx8.h>
#include <sci/sci.h>

void __dead2 imx_system_off(void)
{
	sc_pm_set_sys_power_mode(ipc_handle, SC_PM_PW_MODE_OFF);
	wfi();
	ERROR("power off failed.\n");
	panic();
}

void __dead2 imx_system_reset(void)
{
	sc_pm_reboot(ipc_handle, SC_PM_RESET_TYPE_COLD);
	wfi();
	ERROR("system reset failed.\n");
	panic();
}

int imx_system_reset2(int is_vendor, int reset_type, u_register_t cookie)
{
	const char *reset_type_name = "";

	switch(reset_type) {
	case PSCI_RESET2_SYSTEM_WARM_RESET:
		sc_pm_reboot(ipc_handle, SC_PM_RESET_TYPE_WARM);
		reset_type_name = "warm";
		break;
	case PSCI_RESET2_SYSTEM_COLD_RESET:
		sc_pm_reboot(ipc_handle, SC_PM_RESET_TYPE_COLD);
		reset_type_name = "cold";
		break;
	case PSCI_RESET2_SYSTEM_BOARD_RESET:
		sc_pm_reset(ipc_handle, SC_PM_RESET_TYPE_BOARD);
		reset_type_name = "board";
		break;
	default:
		return PSCI_E_INVALID_PARAMS;
	}

	wfi();
	ERROR("system %s reset failed.\n", reset_type_name);
	panic();
}

int imx_validate_power_state(unsigned int power_state,
			 psci_power_state_t *req_state)
{
	int pwr_lvl = psci_get_pstate_pwrlvl(power_state);
	int pwr_type = psci_get_pstate_type(power_state);
	int state_id = psci_get_pstate_id(power_state);

	if (pwr_lvl > PLAT_MAX_PWR_LVL)
		return PSCI_E_INVALID_PARAMS;

	if (pwr_type == PSTATE_TYPE_POWERDOWN) {
		req_state->pwr_domain_state[MPIDR_AFFLVL0] = PLAT_MAX_OFF_STATE;
		if (!state_id)
			req_state->pwr_domain_state[MPIDR_AFFLVL1] = PLAT_MAX_RET_STATE;
		else
			req_state->pwr_domain_state[MPIDR_AFFLVL1] = PLAT_MAX_OFF_STATE;
	}

	return PSCI_E_SUCCESS;
}

void imx_get_sys_suspend_power_state(psci_power_state_t *req_state)
{
	unsigned int i;

	for (i = IMX_PWR_LVL0; i <= PLAT_MAX_PWR_LVL; i++)
		req_state->pwr_domain_state[i] = PLAT_MAX_RET_STATE;
}

void __dead2 imx_pwr_domain_pwr_down_wfi(const psci_power_state_t *target_state)
{
	while (1)
		wfi();
}