// SPDX-License-Identifier: (GPL-2.0+ OR MIT)
/*
 * Copyright (C) 2014-2022 Amlogic, Inc. All rights reserved.
 *
 * All information contained herein is Amlogic confidential.
 *
 * This software is provided to you pursuant to Software License Agreement
 * (SLA) with Amlogic Inc ("Amlogic"). This software may be used
 * only in accordance with the terms of this agreement.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification is strictly prohibited without prior written permission from
 * Amlogic.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <command.h>
#include <linux/list.h>
#include <linux/kernel.h>
#include <amlogic/pm.h>
#include <amlogic/aml_ir.h>
#include <amlogic/aml_cec.h>
#include <amlogic/aml_hdmirx.h>
//#include <asm/arch/bl31_apis.h>
#include <stdio.h>
#include <malloc.h>
#include <stdarg.h>
//#include <spinlock.h>

#define PM_SUSPEND_ON		(0)
#define PM_SUSPEND_TO_IDLE	(1)
#define PM_RESUME	(2)
#define PM_SUSPEND_MEM		(3)
#define PM_SUSPEND_MIN		PM_SUSPEND_TO_IDLE
#define PM_SUSPEND_MAX		(4)
extern void aml_system_off(void);
#define PM_DEBUG

static struct list_head devices;
bool pm_resume_start;
//static spin_lock_t pm_suspend_lock;

#ifdef PM_DEBUG
static int test_pm_suspend(void *dev_info)
{
	printf("%s\n", __func__);
	return 0;
}

static int test_pm_resume(void *dev_info)
{
	printf("%s\n", __func__);
	return 0;
}

static int test_pm_poweroff(void *dev_info)
{
	printf("%s\n", __func__);
	return 0;
}
#endif

__weak int vpp_suspend(void *dev_info)
{
	printf("[pm]%s\n", __func__);
	return 0;
}

__weak int vpp_resume(void *dev_info)
{
	printf("[pm] %s\n", __func__);
	return 0;
}

__weak int vpp_poweroff(void *dev_info)
{
	printf("[pm] %s\n", __func__);
	return 0;
}

#ifdef PM_DEBUG
struct dev_pm_ops *dev_register_pm(char *name, FUNC_PTR suspend,
			FUNC_PTR resume, FUNC_PTR poweroff, int level, ...);
char name[] = "test_pm_name";
#endif

void pm_initialize(void)
{
	INIT_LIST_HEAD(&devices);
//	spin_lock_init(&pm_suspend_lock);
	dev_register_pm("vpp_pm", vpp_suspend, vpp_resume, vpp_poweroff, 2);
#ifdef PM_DEBUG
	dev_register_pm(name, test_pm_suspend, test_pm_resume, test_pm_poweroff, 0);
#endif
#if IS_ENABLED(CONFIG_CMD_IR)
	dev_register_pm("meson_ir_pm", meson_ir_suspend, meson_ir_resume,
			meson_ir_poweroff, 2);
#endif
	if (IS_ENABLED(CONFIG_CMD_CEC)) {
		dev_register_pm("meson_cec_pm", meson_cec_suspend, meson_cec_resume,
				meson_cec_poweroff, 2);
	}
	#if IS_ENABLED(CONFIG_CMD_HDMIRX)
	dev_register_pm("meson_hdmirx_pm", meson_hdmirx_suspend, meson_hdmirx_resume,
			meson_hdmirx_poweroff, 2);
	#endif
}

extern int cpu_suspend(unsigned long arg);
void pm_suspend(void)
{
	struct list_head *dentry;
	struct dev_pm_ops *pm_ops;

	debug("suspend entry\n");

	list_for_each(dentry, &devices) {
		pm_ops = list_entry(dentry, struct dev_pm_ops, link);
		if (pm_ops->suspend) {
			pm_ops->suspend((void *)pm_ops);
			debug("%s suspend done\n", pm_ops->name);
		} else {
			printf("suspend is not illeage!\n");
		}
	}

	printf("uboot str suspend done!\n");

	cpu_suspend(0);

	pm_resume_start = true;
	list_for_each_prev(dentry, &devices) {
		pm_ops = list_entry(dentry, struct dev_pm_ops, link);
		if (pm_ops->resume) {
			pm_ops->resume((void *)pm_ops);
			debug("%s resume done\n", pm_ops->name);
		} else {
			printf("suspend is not illeage!\n");
		}
	}
	printf("uboot str resume done!\n");
}

bool get_resume_state(void)
{
	return pm_resume_start;
}

void pm_poweroff(void)
{
	struct list_head *dentry;
	struct dev_pm_ops *pm_ops;

	debug("poweroff entry\n");

	list_for_each(dentry, &devices) {
		pm_ops = list_entry(dentry, struct dev_pm_ops, link);
		if (pm_ops->poweroff) {
			pm_ops->poweroff((void *)pm_ops);
			debug("%s poweroff done\n", pm_ops->name);
		} else {
			printf("poweroff is not illeage!\n");
		}
	}
	printf("uboot poweroff done!\n");
	aml_system_off();
}

/** level (0~99), The smaller the value, means earlier suspend, later resume.
* vpu vpp lcd , need resume in sequence,or else maybe flashing green screen.
* vpp mmc , vpp need resume later than mmc.
*
* eth/usb/...		0
* lcd_drv0_pm		1
* vpp_pm		2
* vpu_drv_pm		3
* meson_mmc_pm		99
**/
struct dev_pm_ops *dev_register_pm(char *name, FUNC_PTR suspend,
			FUNC_PTR resume, FUNC_PTR poweroff, int level, ...)
{
	struct dev_pm_ops *ops;
	struct list_head *pos;
	va_list pargs;

	printf("[%s]:%s.\n", __func__, name);
	va_start(pargs, level);
	if (!name || !suspend || !resume || !poweroff) {
		printf("Invalid pass parameters\n");
		va_end(pargs);
		return NULL;
	}

	ops = (struct dev_pm_ops *)malloc((size_t)sizeof(struct dev_pm_ops));

	if (!ops) {
		printf("Wakeup source malloc fail!!\n");
		va_end(pargs);
		return NULL;
	}

	ops->name = name;
	ops->suspend = suspend;
	ops->resume = resume;
	ops->poweroff = poweroff;
	ops->level = level;
	ops->private_data = va_arg(pargs, void *);

//	spin_lock(&pm_suspend_lock);
	list_for_each(pos, &devices) {
		struct dev_pm_ops *pm_ops;

		pm_ops = list_entry(pos, struct dev_pm_ops, link);
		if (pm_ops->level > ops->level)
			break;
	}
	list_add_tail(&ops->link, pos);
//	spin_unlock(&pm_suspend_lock);

	va_end(pargs);

	return ops;
}

int dev_unregister_pm(struct dev_pm_ops *arg)
{
	struct dev_pm_ops *ops = arg;

	if (!ops) {
		printf("Invalid parameters\n");
		return -1;
	}

	list_del(&ops->link);

	free(ops);
	ops = NULL;
	return 0;
}