// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright (C) 2023 Loongson Technology Corporation Limited
 */

#include <drm/drm_managed.h>
#include <drm/drm_print.h>

#include "lsdc_drv.h"
#include "lsdc_output.h"

/*
 * __lsdc_gpio_i2c_set - set the state of a gpio pin indicated by mask
 * @mask: gpio pin mask
 * @state: "0" for low, "1" for high
 */
static void __lsdc_gpio_i2c_set(struct lsdc_i2c * const li2c, int mask, int state)
{
        struct lsdc_device *ldev = to_lsdc(li2c->ddev);
        unsigned long flags;
        u8 val;

        spin_lock_irqsave(&ldev->reglock, flags);

        if (state) {
                /*
                 * Setting this pin as input directly, write 1 for input.
                 * The external pull-up resistor will pull the level up
                 */
                val = readb(li2c->dir_reg);
                val |= mask;
                writeb(val, li2c->dir_reg);
        } else {
                /* First set this pin as output, write 0 for output */
                val = readb(li2c->dir_reg);
                val &= ~mask;
                writeb(val, li2c->dir_reg);

                /* Then, make this pin output 0 */
                val = readb(li2c->dat_reg);
                val &= ~mask;
                writeb(val, li2c->dat_reg);
        }

        spin_unlock_irqrestore(&ldev->reglock, flags);
}

/*
 * __lsdc_gpio_i2c_get - read value back from the gpio pin indicated by mask
 * @mask: gpio pin mask
 * return "0" for low, "1" for high
 */
static int __lsdc_gpio_i2c_get(struct lsdc_i2c * const li2c, int mask)
{
        struct lsdc_device *ldev = to_lsdc(li2c->ddev);
        unsigned long flags;
        u8 val;

        spin_lock_irqsave(&ldev->reglock, flags);

        /* First set this pin as input */
        val = readb(li2c->dir_reg);
        val |= mask;
        writeb(val, li2c->dir_reg);

        /* Then get level state from this pin */
        val = readb(li2c->dat_reg);

        spin_unlock_irqrestore(&ldev->reglock, flags);

        return (val & mask) ? 1 : 0;
}

static void lsdc_gpio_i2c_set_sda(void *i2c, int state)
{
        struct lsdc_i2c * const li2c = (struct lsdc_i2c *)i2c;
        /* set state on the li2c->sda pin */
        return __lsdc_gpio_i2c_set(li2c, li2c->sda, state);
}

static void lsdc_gpio_i2c_set_scl(void *i2c, int state)
{
        struct lsdc_i2c * const li2c = (struct lsdc_i2c *)i2c;
        /* set state on the li2c->scl pin */
        return __lsdc_gpio_i2c_set(li2c, li2c->scl, state);
}

static int lsdc_gpio_i2c_get_sda(void *i2c)
{
        struct lsdc_i2c * const li2c = (struct lsdc_i2c *)i2c;
        /* read value from the li2c->sda pin */
        return __lsdc_gpio_i2c_get(li2c, li2c->sda);
}

static int lsdc_gpio_i2c_get_scl(void *i2c)
{
        struct lsdc_i2c * const li2c = (struct lsdc_i2c *)i2c;
        /* read the value from the li2c->scl pin */
        return __lsdc_gpio_i2c_get(li2c, li2c->scl);
}

static void lsdc_destroy_i2c(struct drm_device *ddev, void *data)
{
        struct lsdc_i2c *li2c = (struct lsdc_i2c *)data;

        if (li2c) {
                i2c_del_adapter(&li2c->adapter);
                kfree(li2c);
        }
}

/*
 * The DC in ls7a1000/ls7a2000/ls2k2000 has builtin gpio hardware
 *
 * @reg_base: gpio reg base
 * @index: output channel index, 0 for PIPE0, 1 for PIPE1
 */
int lsdc_create_i2c_chan(struct drm_device *ddev,
                         struct lsdc_display_pipe *dispipe,
                         unsigned int index)
{
        struct lsdc_device *ldev = to_lsdc(ddev);
        struct i2c_adapter *adapter;
        struct lsdc_i2c *li2c;
        int ret;

        li2c = kzalloc_obj(*li2c);
        if (!li2c)
                return -ENOMEM;

        dispipe->li2c = li2c;

        if (index == 0) {
                li2c->sda = 0x01;  /* pin 0 */
                li2c->scl = 0x02;  /* pin 1 */
        } else if (index == 1) {
                li2c->sda = 0x04;  /* pin 2 */
                li2c->scl = 0x08;  /* pin 3 */
        } else {
                return -ENOENT;
        }

        li2c->ddev = ddev;
        li2c->dir_reg = ldev->reg_base + LS7A_DC_GPIO_DIR_REG;
        li2c->dat_reg = ldev->reg_base + LS7A_DC_GPIO_DAT_REG;

        li2c->bit.setsda = lsdc_gpio_i2c_set_sda;
        li2c->bit.setscl = lsdc_gpio_i2c_set_scl;
        li2c->bit.getsda = lsdc_gpio_i2c_get_sda;
        li2c->bit.getscl = lsdc_gpio_i2c_get_scl;
        li2c->bit.udelay = 5;
        li2c->bit.timeout = usecs_to_jiffies(2200);
        li2c->bit.data = li2c;

        adapter = &li2c->adapter;
        adapter->algo_data = &li2c->bit;
        adapter->owner = THIS_MODULE;
        adapter->dev.parent = ddev->dev;
        adapter->nr = -1;

        snprintf(adapter->name, sizeof(adapter->name), "lsdc-i2c%u", index);

        i2c_set_adapdata(adapter, li2c);

        ret = i2c_bit_add_bus(adapter);
        if (ret) {
                kfree(li2c);
                return ret;
        }

        ret = drmm_add_action_or_reset(ddev, lsdc_destroy_i2c, li2c);
        if (ret)
                return ret;

        drm_info(ddev, "%s(sda pin mask=%u, scl pin mask=%u) created\n",
                 adapter->name, li2c->sda, li2c->scl);

        return 0;
}