#include <linux/init.h>
#include <linux/device.h>
#include <linux/of.h>
#include <linux/of_device.h>
#include <linux/of_address.h>
#include <linux/clocksource.h>
#include <linux/err.h>
#include <linux/io.h>
#include <linux/module.h>
#include <linux/clk-provider.h>
#include <linux/kernel.h>
#include <linux/err.h>
#include <linux/platform_device.h>

#include <asm/mach/arch.h>
#include <clocksource/qcom-qtimer.h>

/* IO_ADDR */
#include <asm/mach/map.h>

const void * __iomem apcs_qtmr_base;
static unsigned int avm_prof_clk_rate_MHzX10;

unsigned int avm_get_profiling_clk_rate_MHzX10(void)
{
	return avm_prof_clk_rate_MHzX10;
}

static const struct of_device_id qtimer_match_table[] = {
	{ .compatible = "qcom,qtimer" },
	{ }
};
MODULE_DEVICE_TABLE(of, qtimer_match_table);

void qtimer_set_next_event(qtimer qtim, unsigned int cycles)
{
	__raw_writel(cycles, apcs_qtmr_base + (qtim * 0x1000) + APCS_QTMR_CNTP_TVAL);
}

void qtimer_enable(qtimer qtim, uint8_t enable, uint8_t mask_irq)
{
	uint32_t val = 0;

	if (enable)
		val |= 0x01;
	if (mask_irq)
		val |= 0x02;

	__raw_writel(val, apcs_qtmr_base + (qtim * 0x1000) + APCS_QTMR_CNTP_CTL);
}

static int qtimer_probe(struct platform_device *pdev)
{
	struct device *dev = &pdev->dev;
	const struct of_device_id *id;
	struct device_node *node = dev->of_node;
	uint32_t val;
	struct resource *mem;

	id = of_match_device(qtimer_match_table, dev);
	if (!id)
		return -ENODEV;

	if (of_property_read_u32(node, "avm,clk-rate", &val)) {
		pr_err("[qcom:%s] WARNING avm,clk-rate not set in dt, using built in default\n", __func__);
		avm_prof_clk_rate_MHzX10 = 0;
	} else {
		pr_info("[qcom:%s] avm,clk-rate set to (%d/10) MHz\n", __func__, val);
		avm_prof_clk_rate_MHzX10 = val;
	}

	mem = platform_get_resource(pdev, IORESOURCE_MEM, 0);
	apcs_qtmr_base = devm_ioremap_resource(dev, mem);
	if (IS_ERR(apcs_qtmr_base))
		return PTR_ERR(apcs_qtmr_base);
	return 0;
}

static int qtimer_remove(struct platform_device *pdev)
{
	return 0;
}

static struct platform_driver qtimer_driver = {
	.probe		= qtimer_probe,
	.remove		= qtimer_remove,
	.driver		= {
		.name	= "qcom,qtimer",
		.owner	= THIS_MODULE,
		.of_match_table = qtimer_match_table,
	},
};

static int __init qtimer_init(void)
{
	return platform_driver_register(&qtimer_driver);
}
core_initcall(qtimer_init);

static void __exit qtimer_exit(void)
{
	platform_driver_unregister(&qtimer_driver);
}
module_exit(qtimer_exit);