/*
 * Copyright (c) 2006-2022, RT-Thread Development Team
 *
 * SPDX-License-Identifier: Apache-2.0
 *
 * Change Logs:
 * Date           Author       Notes
 * 2022-12-06     GuEe-GUI     first version
 */

#define DBG_TAG "drv.rk_timer"
#define DBG_LVL DBG_INFO
#include <rtdbg.h>

#include <rthw.h>
#include <rtthread.h>
#include <rtdevice.h>

#ifdef RT_USING_KTIME
#include <ktime.h>
#endif

#define HZ      100
#define KHZ     1000
#define MHZ     1000000
#define OSC_HZ  (24 * MHZ)

#define TIMER_LOAD_COUNT0       0x00
#define TIMER_LOAD_COUNT1       0x04
#define TIMER_CURRENT_VALUE0    0x08
#define TIMER_CURRENT_VALUE1    0x0c
#define TIMER_CONTROL_REG3288   0x10
#define TIMER_CONTROL_REG3399   0x1c
#define TIMER_INT_STATUS        0x18

#define TIMER_DISABLE                   0x0
#define TIMER_ENABLE                    0x1
#define TIMER_MODE_FREE_RUNNING         (0 << 1)
#define TIMER_MODE_USER_DEFINED_COUNT   (1 << 1)
#define TIMER_INT_UNMASK                (1 << 2)

struct rk_timer
{
    struct rt_hwtimer_device parent;

    void *base;
    void *ctrl;
    struct rt_clk *clk;
    struct rt_clk *pclk;

    int irq;
    rt_uint32_t freq;
    rt_uint32_t cycle;
    rt_bool_t status;

    struct rt_hwtimer_info info;
};
#ifdef RT_USING_KTIME
struct hrt_timer
{
    struct rk_timer *timer;
    uint64_t cnt;
    void (*outcb)(void *param);
    void *param;
};
static struct hrt_timer _timer0 = {0};
static struct rt_spinlock  _spinlock;
#endif
#define raw_to_rk_timer(raw) rt_container_of(raw, struct rk_timer, parent)

struct rk_timer_data
{
    rt_uint32_t ctrl_reg;
};

rt_inline void rk_timer_disable(struct rk_timer *timer)
{
    HWREG32(timer->ctrl) = TIMER_DISABLE;
}

rt_inline void rk_timer_enable(struct rk_timer *timer, rt_uint32_t flags)
{
    HWREG32(timer->ctrl) = TIMER_ENABLE | flags;
}

rt_inline rt_uint32_t rk_timer_current_value(struct rk_timer *timer)
{
    return HWREG32(timer->base + TIMER_CURRENT_VALUE0);
}

static void rk_timer_update_counter(unsigned long cycles, struct rk_timer *timer)
{
    HWREG32(timer->base + TIMER_LOAD_COUNT0) = cycles;
    HWREG32(timer->base + TIMER_LOAD_COUNT1) = 0;
}

static void rk_timer_interrupt_clear(struct rk_timer *timer)
{
    HWREG32(timer->base + TIMER_INT_STATUS) = 1;
}

static void rk_timer_init(struct rt_hwtimer_device *timer, rt_uint32_t state)
{
}

static rt_err_t rk_timer_start(struct rt_hwtimer_device *timer, rt_uint32_t cnt, rt_hwtimer_mode_t mode)
{
    rt_err_t err = RT_EOK;
    struct rk_timer *rk_timer = raw_to_rk_timer(timer);

    switch (mode)
    {
    case HWTIMER_MODE_ONESHOT:
        rk_timer_disable(rk_timer);
        rk_timer_update_counter(cnt, rk_timer);
        rk_timer_enable(rk_timer, TIMER_MODE_USER_DEFINED_COUNT | TIMER_INT_UNMASK);
        break;

    case HWTIMER_MODE_PERIOD:
        rk_timer_disable(rk_timer);
        rk_timer_update_counter(rk_timer->freq / HZ - 1, rk_timer);
        rk_timer_enable(rk_timer, TIMER_MODE_FREE_RUNNING | TIMER_INT_UNMASK);
        break;

    default:
        err = -RT_EINVAL;
        break;
    }

    if (!err)
    {
        rk_timer->cycle = cnt;
        rk_timer->status = RT_TRUE;
    }

    return err;
}

static void rk_timer_stop(struct rt_hwtimer_device *timer)
{
    struct rk_timer *rk_timer = raw_to_rk_timer(timer);

    rk_timer->status = RT_FALSE;
    rk_timer_disable(rk_timer);
}

static rt_uint32_t rk_timer_count_get(struct rt_hwtimer_device *timer)
{
    struct rk_timer *rk_timer = raw_to_rk_timer(timer);

    return rk_timer_current_value(rk_timer);
}

static rt_err_t rk_timer_ctrl(struct rt_hwtimer_device *timer, rt_uint32_t cmd, void *args)
{
    rt_err_t err = RT_EOK;
    struct rk_timer *rk_timer = raw_to_rk_timer(timer);

    switch (cmd)
    {
    case HWTIMER_CTRL_FREQ_SET:
        err = -RT_ENOSYS;
        break;

    case HWTIMER_CTRL_STOP:
        rk_timer_stop(timer);
        break;

    case HWTIMER_CTRL_INFO_GET:
        if (args)
        {
            rt_memcpy(args, &rk_timer->info, sizeof(rk_timer->info));
        }
        else
        {
            err = -RT_ERROR;
        }
        break;

    case HWTIMER_CTRL_MODE_SET:
        err = rk_timer_start(timer, rk_timer->cycle, (rt_hwtimer_mode_t)args);
        break;

    default:
        err = -RT_EINVAL;
        break;
    }

    return err;
}

const static struct rt_hwtimer_ops rk_timer_ops =
{
    .init = rk_timer_init,
    .start = rk_timer_start,
    .stop = rk_timer_stop,
    .count_get = rk_timer_count_get,
    .control = rk_timer_ctrl,
};

static void rk_timer_isr(int irqno, void *param)
{
    struct hrt_timer *timer = &_timer0;
    struct rk_timer *time = timer->timer;

    rk_timer_interrupt_clear(time);

    rt_ktime_hrtimer_process();
}

void rt_ktime_hrtimer_bind(rt_bitmap_t *affinity)
{
    struct rk_timer *timer = _timer0.timer;

    if (rt_pic_irq_set_affinity(timer->irq, affinity) == -RT_ENOSYS)
    {
        LOG_E("timer irq affinity init fail\n");
    }
    else
    {
        LOG_D("timer irq(%d) binding done\n", timer->irq);
    }
}

static rt_err_t rk_timer_probe(struct rt_platform_device *pdev)
{
    rt_err_t err = RT_EOK;
    const char *dev_name;
    struct rt_device *dev = &pdev->parent;
    struct rk_timer *timer = rt_calloc(1, sizeof(*timer));
    const struct rk_timer_data *timer_data = pdev->id->data;
    if (!timer)
    {
        return -RT_ENOMEM;
    }
#ifdef RT_USING_KTIME
    _timer0.timer = timer;
    rt_spin_lock_init(&_spinlock);
#endif
    if (!(timer->pclk = rt_clk_get_by_name(dev, "pclk")))
    {
        err = -RT_EIO;

        goto _fail;
    }

    if (!(timer->clk = rt_clk_get_by_name(dev, "timer")))
    {
        err = -RT_EIO;

        goto _fail;
    }

    timer->base = rt_dm_dev_iomap(dev, 0);

    if (!timer->base)
    {
        err = -RT_EIO;

        goto _fail;
    }

    timer->ctrl = timer->base + timer_data->ctrl_reg;

    rt_clk_enable(timer->pclk);
    rt_clk_enable(timer->clk);
    timer->freq = rt_clk_get_rate(timer->clk);
    timer->irq = rt_dm_dev_get_irq(dev, 0);

    rk_timer_interrupt_clear(timer);
    rk_timer_disable(timer);

    timer->parent.ops = &rk_timer_ops;
    timer->parent.info = &timer->info;

    timer->info.maxfreq = timer->freq;
    timer->info.minfreq = timer->freq;
    timer->info.maxcnt = 0xffffffff;
    timer->info.cntmode = HWTIMER_CNTMODE_DW;

    rt_dm_dev_set_name_auto(&timer->parent.parent, "timer");
    dev_name = rt_dm_dev_get_name(&timer->parent.parent);

    rt_device_hwtimer_register(&timer->parent, dev_name, RT_NULL);

    RT_BITMAP_DECLARE(affinity, RT_CPUS_NR) = { 0 };
    rt_bitmap_set_bit(affinity, RT_CPUS_NR - 1);
    rt_ktime_hrtimer_bind(affinity);

    rt_pic_attach_irq(timer->irq, rk_timer_isr, timer, dev_name, RT_IRQ_F_NONE);
    rt_pic_irq_unmask(timer->irq);

#if KTIMER_BIND_CPU
    RT_BITMAP_DECLARE(affinity, RT_CPUS_NR) = {0};
    rt_bitmap_set_bit(affinity, 1);
    rt_pic_irq_set_affinity(timer->irq, affinity);
#endif

    return err;

_fail:
    if (timer->base)
    {
        rt_iounmap(timer->base);
    }
    if (timer->pclk)
    {
        rt_clk_put(timer->pclk);
    }
    if (timer->clk)
    {
        rt_clk_put(timer->clk);
    }
    rt_free(timer);

    return err;
}

static const struct rk_timer_data rk3288_timer_data =
{
    .ctrl_reg = TIMER_CONTROL_REG3288,
};

static const struct rk_timer_data rk3399_timer_data =
{
    .ctrl_reg = TIMER_CONTROL_REG3399,
};

#ifdef RT_USING_KTIME

uint64_t rt_ktime_hrtimer_getfrq(void)
{
    return (24 * 1000 * 1000UL);
}

uint64_t rt_ktime_hrtimer_getres(void)
{
    return ((1000UL * 1000 * 1000) * RT_KTIME_RESMUL) / (24 * 1000 * 1000UL);
}

uint64_t rt_ktime_hrtimer_getcnt(void)
{
    return rk_timer_current_value(_timer0.timer);
}

/**
 * @brief set the timeout function for hrtimer framework
 *
 * @warning application should not call this API directly
 *
 * @param cnt the count of timer dealy
 * @return rt_err_t 0 forever
 */
rt_err_t rt_ktime_hrtimer_settimeout(unsigned long cnt)
{
    struct hrt_timer *timer = &_timer0;
    struct rk_timer *time = timer->timer;

    timer->cnt = cnt;

    if (cnt)
    {
        rk_timer_disable(time);
        rk_timer_update_counter(cnt, time);
        rk_timer_enable(time, TIMER_MODE_USER_DEFINED_COUNT | TIMER_INT_UNMASK);
    }

    return 0;
}
#endif

static const struct rt_ofw_node_id rk_timer_ofw_ids[] =
{
    { .compatible = "rockchip,rk3288-timer", .data = &rk3288_timer_data },
    { .compatible = "rockchip,rk3399-timer", .data = &rk3399_timer_data },
    { /* sentinel */ }
};

static struct rt_platform_driver rk_timer_driver =
{
    .name = "hwtimer-rockchip",
    .ids = rk_timer_ofw_ids,

    .probe = rk_timer_probe,
};

static int rk_timer_drv_register(void)
{
    rt_platform_driver_register(&rk_timer_driver);

    return 0;
}
INIT_DRIVER_EARLY_EXPORT(rk_timer_drv_register);