// SPDX-License-Identifier: GPL-2.0-only
/*
 * GPIO driver for RICOH583 power management chip.
 *
 * Copyright (c) 2012, NVIDIA CORPORATION.  All rights reserved.
 * Author: Laxman dewangan <ldewangan@nvidia.com>
 *
 * Based on code
 * Copyright (C) 2011 RICOH COMPANY,LTD
 */
#include <linux/init.h>
#include <linux/kernel.h>
#include <linux/slab.h>
#include <linux/platform_device.h>
#include <linux/device.h>
#include <linux/gpio/driver.h>
#include <linux/mfd/rc5t583.h>

struct rc5t583_gpio {
 struct gpio_chip gpio_chip;
 struct rc5t583 *rc5t583;
};

static int rc5t583_gpio_get(struct gpio_chip *gc, unsigned int offset)
{
 struct rc5t583_gpio *rc5t583_gpio = gpiochip_get_data(gc);
 struct device *parent = rc5t583_gpio->rc5t583->dev;
 uint8_t val = 0;
 int ret;

 ret = rc5t583_read(parent, RC5T583_GPIO_MON_IOIN, &val);
 if (ret < 0)
  return ret;

 return !!(val & BIT(offset));
}

static int rc5t583_gpio_set(struct gpio_chip *gc, unsigned int offset, int val)
{
 struct rc5t583_gpio *rc5t583_gpio = gpiochip_get_data(gc);
 struct device *parent = rc5t583_gpio->rc5t583->dev;
 int ret;

 if (val)
  ret = rc5t583_set_bits(parent, RC5T583_GPIO_IOOUT,
           BIT(offset));
 else
  ret = rc5t583_clear_bits(parent, RC5T583_GPIO_IOOUT,
      BIT(offset));

 return ret;
}

static int rc5t583_gpio_dir_input(struct gpio_chip *gc, unsigned int offset)
{
 struct rc5t583_gpio *rc5t583_gpio = gpiochip_get_data(gc);
 struct device *parent = rc5t583_gpio->rc5t583->dev;
 int ret;

 ret = rc5t583_clear_bits(parent, RC5T583_GPIO_IOSEL, BIT(offset));
 if (ret < 0)
  return ret;

 /* Set pin to gpio mode */
 return rc5t583_clear_bits(parent, RC5T583_GPIO_PGSEL, BIT(offset));
}

static int rc5t583_gpio_dir_output(struct gpio_chip *gc, unsigned offset,
   int value)
{
 struct rc5t583_gpio *rc5t583_gpio = gpiochip_get_data(gc);
 struct device *parent = rc5t583_gpio->rc5t583->dev;
 int ret;

 ret = rc5t583_gpio_set(gc, offset, value);
 if (ret)
  return ret;

 ret = rc5t583_set_bits(parent, RC5T583_GPIO_IOSEL, BIT(offset));
 if (ret < 0)
  return ret;

 /* Set pin to gpio mode */
 return rc5t583_clear_bits(parent, RC5T583_GPIO_PGSEL, BIT(offset));
}

static int rc5t583_gpio_to_irq(struct gpio_chip *gc, unsigned offset)
{
 struct rc5t583_gpio *rc5t583_gpio = gpiochip_get_data(gc);

 if (offset < RC5T583_MAX_GPIO)
  return rc5t583_gpio->rc5t583->irq_base +
    RC5T583_IRQ_GPIO0 + offset;
 return -EINVAL;
}

static void rc5t583_gpio_free(struct gpio_chip *gc, unsigned offset)
{
 struct rc5t583_gpio *rc5t583_gpio = gpiochip_get_data(gc);
 struct device *parent = rc5t583_gpio->rc5t583->dev;

 rc5t583_set_bits(parent, RC5T583_GPIO_PGSEL, BIT(offset));
}

static int rc5t583_gpio_probe(struct platform_device *pdev)
{
 struct rc5t583 *rc5t583 = dev_get_drvdata(pdev->dev.parent);
 struct rc5t583_platform_data *pdata = dev_get_platdata(rc5t583->dev);
 struct rc5t583_gpio *rc5t583_gpio;

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

 rc5t583_gpio->gpio_chip.label = "gpio-rc5t583",
 rc5t583_gpio->gpio_chip.owner = THIS_MODULE,
 rc5t583_gpio->gpio_chip.free = rc5t583_gpio_free,
 rc5t583_gpio->gpio_chip.direction_input = rc5t583_gpio_dir_input,
 rc5t583_gpio->gpio_chip.direction_output = rc5t583_gpio_dir_output,
 rc5t583_gpio->gpio_chip.set = rc5t583_gpio_set,
 rc5t583_gpio->gpio_chip.get = rc5t583_gpio_get,
 rc5t583_gpio->gpio_chip.to_irq = rc5t583_gpio_to_irq,
 rc5t583_gpio->gpio_chip.ngpio = RC5T583_MAX_GPIO,
 rc5t583_gpio->gpio_chip.can_sleep = true,
 rc5t583_gpio->gpio_chip.parent = &pdev->dev;
 rc5t583_gpio->gpio_chip.base = -1;
 rc5t583_gpio->rc5t583 = rc5t583;

 if (pdata && pdata->gpio_base)
  rc5t583_gpio->gpio_chip.base = pdata->gpio_base;

 return devm_gpiochip_add_data(&pdev->dev, &rc5t583_gpio->gpio_chip,
          rc5t583_gpio);
}

static struct platform_driver rc5t583_gpio_driver = {
 .driver = {
  .name    = "rc5t583-gpio",
 },
 .probe  = rc5t583_gpio_probe,
};

static int __init rc5t583_gpio_init(void)
{
 return platform_driver_register(&rc5t583_gpio_driver);
}
subsys_initcall(rc5t583_gpio_init);