// SPDX-License-Identifier: GPL-2.0-only
/*
 * RTC driver for Rockchip RK808
 *
 * Copyright (c) 2014, Fuzhou Rockchip Electronics Co., Ltd
 *
 * Author: Chris Zhong <zyw@rock-chips.com>
 * Author: Zhang Qing <zhangqing@rock-chips.com>
 */

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/rtc.h>
#include <linux/bcd.h>
#include <linux/mfd/rk808.h>
#include <linux/platform_device.h>

/* RTC_CTRL_REG bitfields */
#define BIT_RTC_CTRL_REG_STOP_RTC_M             BIT(0)

/* RK808 has a shadowed register for saving a "frozen" RTC time.
 * When user setting "GET_TIME" to 1, the time will save in this shadowed
 * register. If set "READSEL" to 1, user read rtc time register, actually
 * get the time of that moment. If we need the real time, clr this bit.
 */
#define BIT_RTC_CTRL_REG_RTC_GET_TIME           BIT(6)
#define BIT_RTC_CTRL_REG_RTC_READSEL_M          BIT(7)
#define BIT_RTC_INTERRUPTS_REG_IT_ALARM_M       BIT(3)
#define RTC_STATUS_MASK         0xFE

#define SECONDS_REG_MSK         0x7F
#define MINUTES_REG_MAK         0x7F
#define HOURS_REG_MSK           0x3F
#define DAYS_REG_MSK            0x3F
#define MONTHS_REG_MSK          0x1F
#define YEARS_REG_MSK           0xFF
#define WEEKS_REG_MSK           0x7

/* REG_SECONDS_REG through REG_YEARS_REG is how many registers? */

#define NUM_TIME_REGS   (RK808_WEEKS_REG - RK808_SECONDS_REG + 1)
#define NUM_ALARM_REGS  (RK808_ALARM_YEARS_REG - RK808_ALARM_SECONDS_REG + 1)

struct rk_rtc_compat_reg {
        unsigned int ctrl_reg;
        unsigned int status_reg;
        unsigned int alarm_seconds_reg;
        unsigned int int_reg;
        unsigned int seconds_reg;
};

struct rk808_rtc {
        struct regmap *regmap;
        struct rtc_device *rtc;
        struct rk_rtc_compat_reg *creg;
        int irq;
};

/*
 * The Rockchip calendar used by the RK808 counts November with 31 days. We use
 * these translation functions to convert its dates to/from the Gregorian
 * calendar used by the rest of the world. We arbitrarily define Jan 1st, 2016
 * as the day when both calendars were in sync, and treat all other dates
 * relative to that.
 * NOTE: Other system software (e.g. firmware) that reads the same hardware must
 * implement this exact same conversion algorithm, with the same anchor date.
 */
static time64_t nov2dec_transitions(struct rtc_time *tm)
{
        return (tm->tm_year + 1900) - 2016 + (tm->tm_mon + 1 > 11 ? 1 : 0);
}

static void rockchip_to_gregorian(struct rtc_time *tm)
{
        /* If it's Nov 31st, rtc_tm_to_time64() will count that like Dec 1st */
        time64_t time = rtc_tm_to_time64(tm);
        rtc_time64_to_tm(time + nov2dec_transitions(tm) * 86400, tm);
}

static void gregorian_to_rockchip(struct rtc_time *tm)
{
        time64_t extra_days = nov2dec_transitions(tm);
        time64_t time = rtc_tm_to_time64(tm);
        rtc_time64_to_tm(time - extra_days * 86400, tm);

        /* Compensate if we went back over Nov 31st (will work up to 2381) */
        if (nov2dec_transitions(tm) < extra_days) {
                if (tm->tm_mon + 1 == 11)
                        tm->tm_mday++;  /* This may result in 31! */
                else
                        rtc_time64_to_tm(time - (extra_days - 1) * 86400, tm);
        }
}

/* Read current time and date in RTC */
static int rk808_rtc_readtime(struct device *dev, struct rtc_time *tm)
{
        struct rk808_rtc *rk808_rtc = dev_get_drvdata(dev);
        u8 rtc_data[NUM_TIME_REGS];
        int ret;

        /* Force an update of the shadowed registers right now */
        ret = regmap_update_bits(rk808_rtc->regmap, rk808_rtc->creg->ctrl_reg,
                                 BIT_RTC_CTRL_REG_RTC_GET_TIME,
                                 BIT_RTC_CTRL_REG_RTC_GET_TIME);
        if (ret) {
                dev_err(dev, "Failed to update bits rtc_ctrl: %d\n", ret);
                return ret;
        }

        /*
         * After we set the GET_TIME bit, the rtc time can't be read
         * immediately. So we should wait up to 31.25 us, about one cycle of
         * 32khz. If we clear the GET_TIME bit here, the time of i2c transfer
         * certainly more than 31.25us: 16 * 2.5us at 400kHz bus frequency.
         */
        ret = regmap_update_bits(rk808_rtc->regmap, rk808_rtc->creg->ctrl_reg,
                                 BIT_RTC_CTRL_REG_RTC_GET_TIME,
                                 0);
        if (ret) {
                dev_err(dev, "Failed to update bits rtc_ctrl: %d\n", ret);
                return ret;
        }

        ret = regmap_bulk_read(rk808_rtc->regmap, rk808_rtc->creg->seconds_reg,
                               rtc_data, NUM_TIME_REGS);
        if (ret) {
                dev_err(dev, "Failed to bulk read rtc_data: %d\n", ret);
                return ret;
        }

        tm->tm_sec = bcd2bin(rtc_data[0] & SECONDS_REG_MSK);
        tm->tm_min = bcd2bin(rtc_data[1] & MINUTES_REG_MAK);
        tm->tm_hour = bcd2bin(rtc_data[2] & HOURS_REG_MSK);
        tm->tm_mday = bcd2bin(rtc_data[3] & DAYS_REG_MSK);
        tm->tm_mon = (bcd2bin(rtc_data[4] & MONTHS_REG_MSK)) - 1;
        tm->tm_year = (bcd2bin(rtc_data[5] & YEARS_REG_MSK)) + 100;
        tm->tm_wday = bcd2bin(rtc_data[6] & WEEKS_REG_MSK);
        rockchip_to_gregorian(tm);
        dev_dbg(dev, "RTC date/time %ptRd(%d) %ptRt\n", tm, tm->tm_wday, tm);

        return ret;
}

/* Set current time and date in RTC */
static int rk808_rtc_set_time(struct device *dev, struct rtc_time *tm)
{
        struct rk808_rtc *rk808_rtc = dev_get_drvdata(dev);
        u8 rtc_data[NUM_TIME_REGS];
        int ret;

        dev_dbg(dev, "set RTC date/time %ptRd(%d) %ptRt\n", tm, tm->tm_wday, tm);
        gregorian_to_rockchip(tm);
        rtc_data[0] = bin2bcd(tm->tm_sec);
        rtc_data[1] = bin2bcd(tm->tm_min);
        rtc_data[2] = bin2bcd(tm->tm_hour);
        rtc_data[3] = bin2bcd(tm->tm_mday);
        rtc_data[4] = bin2bcd(tm->tm_mon + 1);
        rtc_data[5] = bin2bcd(tm->tm_year - 100);
        rtc_data[6] = bin2bcd(tm->tm_wday);

        /* Stop RTC while updating the RTC registers */
        ret = regmap_update_bits(rk808_rtc->regmap, rk808_rtc->creg->ctrl_reg,
                                 BIT_RTC_CTRL_REG_STOP_RTC_M,
                                 BIT_RTC_CTRL_REG_STOP_RTC_M);
        if (ret) {
                dev_err(dev, "Failed to update RTC control: %d\n", ret);
                return ret;
        }

        ret = regmap_bulk_write(rk808_rtc->regmap, rk808_rtc->creg->seconds_reg,
                                rtc_data, NUM_TIME_REGS);
        if (ret) {
                dev_err(dev, "Failed to bull write rtc_data: %d\n", ret);
                return ret;
        }
        /* Start RTC again */
        ret = regmap_update_bits(rk808_rtc->regmap, rk808_rtc->creg->ctrl_reg,
                                 BIT_RTC_CTRL_REG_STOP_RTC_M, 0);
        if (ret) {
                dev_err(dev, "Failed to update RTC control: %d\n", ret);
                return ret;
        }
        return 0;
}

/* Read alarm time and date in RTC */
static int rk808_rtc_readalarm(struct device *dev, struct rtc_wkalrm *alrm)
{
        struct rk808_rtc *rk808_rtc = dev_get_drvdata(dev);
        u8 alrm_data[NUM_ALARM_REGS];
        uint32_t int_reg;
        int ret;

        ret = regmap_bulk_read(rk808_rtc->regmap,
                               rk808_rtc->creg->alarm_seconds_reg,
                               alrm_data, NUM_ALARM_REGS);
        if (ret) {
                dev_err(dev, "Failed to read RTC alarm date REG: %d\n", ret);
                return ret;
        }

        alrm->time.tm_sec = bcd2bin(alrm_data[0] & SECONDS_REG_MSK);
        alrm->time.tm_min = bcd2bin(alrm_data[1] & MINUTES_REG_MAK);
        alrm->time.tm_hour = bcd2bin(alrm_data[2] & HOURS_REG_MSK);
        alrm->time.tm_mday = bcd2bin(alrm_data[3] & DAYS_REG_MSK);
        alrm->time.tm_mon = (bcd2bin(alrm_data[4] & MONTHS_REG_MSK)) - 1;
        alrm->time.tm_year = (bcd2bin(alrm_data[5] & YEARS_REG_MSK)) + 100;
        rockchip_to_gregorian(&alrm->time);

        ret = regmap_read(rk808_rtc->regmap, rk808_rtc->creg->int_reg, &int_reg);
        if (ret) {
                dev_err(dev, "Failed to read RTC INT REG: %d\n", ret);
                return ret;
        }

        dev_dbg(dev, "alrm read RTC date/time %ptRd(%d) %ptRt\n",
                &alrm->time, alrm->time.tm_wday, &alrm->time);

        alrm->enabled = (int_reg & BIT_RTC_INTERRUPTS_REG_IT_ALARM_M) ? 1 : 0;

        return 0;
}

static int rk808_rtc_stop_alarm(struct rk808_rtc *rk808_rtc)
{
        int ret;

        ret = regmap_update_bits(rk808_rtc->regmap, rk808_rtc->creg->int_reg,
                                 BIT_RTC_INTERRUPTS_REG_IT_ALARM_M, 0);

        return ret;
}

static int rk808_rtc_start_alarm(struct rk808_rtc *rk808_rtc)
{
        int ret;

        ret = regmap_update_bits(rk808_rtc->regmap, rk808_rtc->creg->int_reg,
                                 BIT_RTC_INTERRUPTS_REG_IT_ALARM_M,
                                 BIT_RTC_INTERRUPTS_REG_IT_ALARM_M);

        return ret;
}

static int rk808_rtc_setalarm(struct device *dev, struct rtc_wkalrm *alrm)
{
        struct rk808_rtc *rk808_rtc = dev_get_drvdata(dev);
        u8 alrm_data[NUM_ALARM_REGS];
        int ret;

        ret = rk808_rtc_stop_alarm(rk808_rtc);
        if (ret) {
                dev_err(dev, "Failed to stop alarm: %d\n", ret);
                return ret;
        }
        dev_dbg(dev, "alrm set RTC date/time %ptRd(%d) %ptRt\n",
                &alrm->time, alrm->time.tm_wday, &alrm->time);

        gregorian_to_rockchip(&alrm->time);
        alrm_data[0] = bin2bcd(alrm->time.tm_sec);
        alrm_data[1] = bin2bcd(alrm->time.tm_min);
        alrm_data[2] = bin2bcd(alrm->time.tm_hour);
        alrm_data[3] = bin2bcd(alrm->time.tm_mday);
        alrm_data[4] = bin2bcd(alrm->time.tm_mon + 1);
        alrm_data[5] = bin2bcd(alrm->time.tm_year - 100);

        ret = regmap_bulk_write(rk808_rtc->regmap,
                                rk808_rtc->creg->alarm_seconds_reg,
                                alrm_data, NUM_ALARM_REGS);
        if (ret) {
                dev_err(dev, "Failed to bulk write: %d\n", ret);
                return ret;
        }
        if (alrm->enabled) {
                ret = rk808_rtc_start_alarm(rk808_rtc);
                if (ret) {
                        dev_err(dev, "Failed to start alarm: %d\n", ret);
                        return ret;
                }
        }
        return 0;
}

static int rk808_rtc_alarm_irq_enable(struct device *dev,
                                      unsigned int enabled)
{
        struct rk808_rtc *rk808_rtc = dev_get_drvdata(dev);

        if (enabled)
                return rk808_rtc_start_alarm(rk808_rtc);

        return rk808_rtc_stop_alarm(rk808_rtc);
}

/*
 * We will just handle setting the frequency and make use the framework for
 * reading the periodic interupts.
 *
 * @freq: Current periodic IRQ freq:
 * bit 0: every second
 * bit 1: every minute
 * bit 2: every hour
 * bit 3: every day
 */
static irqreturn_t rk808_alarm_irq(int irq, void *data)
{
        struct rk808_rtc *rk808_rtc = data;
        int ret;

        ret = regmap_write(rk808_rtc->regmap, rk808_rtc->creg->status_reg,
                           RTC_STATUS_MASK);
        if (ret) {
                dev_err(&rk808_rtc->rtc->dev,
                        "%s:Failed to update RTC status: %d\n", __func__, ret);
                return ret;
        }

        rtc_update_irq(rk808_rtc->rtc, 1, RTC_IRQF | RTC_AF);
        dev_dbg(&rk808_rtc->rtc->dev,
                 "%s:irq=%d\n", __func__, irq);
        return IRQ_HANDLED;
}

static const struct rtc_class_ops rk808_rtc_ops = {
        .read_time = rk808_rtc_readtime,
        .set_time = rk808_rtc_set_time,
        .read_alarm = rk808_rtc_readalarm,
        .set_alarm = rk808_rtc_setalarm,
        .alarm_irq_enable = rk808_rtc_alarm_irq_enable,
};

#ifdef CONFIG_PM_SLEEP
/* Turn off the alarm if it should not be a wake source. */
static int rk808_rtc_suspend(struct device *dev)
{
        struct rk808_rtc *rk808_rtc = dev_get_drvdata(dev);

        if (device_may_wakeup(dev))
                enable_irq_wake(rk808_rtc->irq);

        return 0;
}

/* Enable the alarm if it should be enabled (in case it was disabled to
 * prevent use as a wake source).
 */
static int rk808_rtc_resume(struct device *dev)
{
        struct rk808_rtc *rk808_rtc = dev_get_drvdata(dev);

        if (device_may_wakeup(dev))
                disable_irq_wake(rk808_rtc->irq);

        return 0;
}
#endif

static SIMPLE_DEV_PM_OPS(rk808_rtc_pm_ops,
        rk808_rtc_suspend, rk808_rtc_resume);

static struct rk_rtc_compat_reg rk808_creg = {
        .ctrl_reg = RK808_RTC_CTRL_REG,
        .status_reg = RK808_RTC_STATUS_REG,
        .alarm_seconds_reg = RK808_ALARM_SECONDS_REG,
        .int_reg = RK808_RTC_INT_REG,
        .seconds_reg = RK808_SECONDS_REG,
};

static struct rk_rtc_compat_reg rk817_creg = {
        .ctrl_reg = RK817_RTC_CTRL_REG,
        .status_reg = RK817_RTC_STATUS_REG,
        .alarm_seconds_reg = RK817_ALARM_SECONDS_REG,
        .int_reg = RK817_RTC_INT_REG,
        .seconds_reg = RK817_SECONDS_REG,
};

static int rk808_rtc_probe(struct platform_device *pdev)
{
        struct rk808 *rk808 = dev_get_drvdata(pdev->dev.parent);
        struct rk808_rtc *rk808_rtc;
        int ret;

        rk808_rtc = devm_kzalloc(&pdev->dev, sizeof(*rk808_rtc), GFP_KERNEL);
        if (rk808_rtc == NULL)
                return -ENOMEM;

        switch (rk808->variant) {
        case RK809_ID:
        case RK817_ID:
                rk808_rtc->creg = &rk817_creg;
                break;
        default:
                rk808_rtc->creg = &rk808_creg;
                break;
        }
        platform_set_drvdata(pdev, rk808_rtc);
        rk808_rtc->regmap = dev_get_regmap(pdev->dev.parent, NULL);
        if (!rk808_rtc->regmap)
                return -ENODEV;

        /* start rtc running by default, and use shadowed timer. */
        ret = regmap_update_bits(rk808_rtc->regmap, rk808_rtc->creg->ctrl_reg,
                                 BIT_RTC_CTRL_REG_STOP_RTC_M |
                                 BIT_RTC_CTRL_REG_RTC_READSEL_M,
                                 BIT_RTC_CTRL_REG_RTC_READSEL_M);
        if (ret) {
                dev_err(&pdev->dev,
                        "Failed to update RTC control: %d\n", ret);
                return ret;
        }

        ret = regmap_write(rk808_rtc->regmap, rk808_rtc->creg->status_reg,
                           RTC_STATUS_MASK);
        if (ret) {
                dev_err(&pdev->dev,
                        "Failed to write RTC status: %d\n", ret);
                return ret;
        }

        device_init_wakeup(&pdev->dev, true);

        rk808_rtc->rtc = devm_rtc_allocate_device(&pdev->dev);
        if (IS_ERR(rk808_rtc->rtc))
                return PTR_ERR(rk808_rtc->rtc);

        rk808_rtc->rtc->ops = &rk808_rtc_ops;

        rk808_rtc->irq = platform_get_irq(pdev, 0);
        if (rk808_rtc->irq < 0)
                return rk808_rtc->irq;

        /* request alarm irq of rk808 */
        ret = devm_request_threaded_irq(&pdev->dev, rk808_rtc->irq, NULL,
                                        rk808_alarm_irq, 0,
                                        "RTC alarm", rk808_rtc);
        if (ret) {
                dev_err(&pdev->dev, "Failed to request alarm IRQ %d: %d\n",
                        rk808_rtc->irq, ret);
                return ret;
        }

        return devm_rtc_register_device(rk808_rtc->rtc);
}

static struct platform_driver rk808_rtc_driver = {
        .probe = rk808_rtc_probe,
        .driver = {
                .name = "rk808-rtc",
                .pm = &rk808_rtc_pm_ops,
        },
};

module_platform_driver(rk808_rtc_driver);

MODULE_DESCRIPTION("RTC driver for the rk808 series PMICs");
MODULE_AUTHOR("Chris Zhong <zyw@rock-chips.com>");
MODULE_AUTHOR("Zhang Qing <zhangqing@rock-chips.com>");
MODULE_LICENSE("GPL");
MODULE_ALIAS("platform:rk808-rtc");