// SPDX-License-Identifier: GPL-2.0-only
/*
 * GPIO driver for the WinSystems WS16C48
 * Copyright (C) 2016 William Breathitt Gray
 */
#include <linux/bitfield.h>
#include <linux/bits.h>
#include <linux/device.h>
#include <linux/err.h>
#include <linux/gpio/regmap.h>
#include <linux/irq.h>
#include <linux/isa.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/spinlock.h>
#include <linux/regmap.h>
#include <linux/types.h>

#define WS16C48_EXTENT 11
#define MAX_NUM_WS16C48 max_num_isa_dev(WS16C48_EXTENT)

static unsigned int base[MAX_NUM_WS16C48];
static unsigned int num_ws16c48;
module_param_hw_array(base, uint, ioport, &num_ws16c48, 0);
MODULE_PARM_DESC(base, "WinSystems WS16C48 base addresses");

static unsigned int irq[MAX_NUM_WS16C48];
static unsigned int num_irq;
module_param_hw_array(irq, uint, irq, &num_irq, 0);
MODULE_PARM_DESC(irq, "WinSystems WS16C48 interrupt line numbers");

#define WS16C48_DAT_BASE 0x0
#define WS16C48_PAGE_LOCK 0x7
#define WS16C48_PAGE_BASE 0x8
#define WS16C48_POL WS16C48_PAGE_BASE
#define WS16C48_ENAB WS16C48_PAGE_BASE
#define WS16C48_INT_ID WS16C48_PAGE_BASE

#define PAGE_LOCK_PAGE_FIELD GENMASK(7, 6)
#define POL_PAGE u8_encode_bits(1, PAGE_LOCK_PAGE_FIELD)
#define ENAB_PAGE u8_encode_bits(2, PAGE_LOCK_PAGE_FIELD)
#define INT_ID_PAGE u8_encode_bits(3, PAGE_LOCK_PAGE_FIELD)

static const struct regmap_range ws16c48_wr_ranges[] = {
 regmap_reg_range(0x0, 0x5), regmap_reg_range(0x7, 0xA),
};
static const struct regmap_range ws16c48_rd_ranges[] = {
 regmap_reg_range(0x0, 0xA),
};
static const struct regmap_range ws16c48_volatile_ranges[] = {
 regmap_reg_range(0x0, 0x6), regmap_reg_range(0x8, 0xA),
};
static const struct regmap_access_table ws16c48_wr_table = {
 .yes_ranges = ws16c48_wr_ranges,
 .n_yes_ranges = ARRAY_SIZE(ws16c48_wr_ranges),
};
static const struct regmap_access_table ws16c48_rd_table = {
 .yes_ranges = ws16c48_rd_ranges,
 .n_yes_ranges = ARRAY_SIZE(ws16c48_rd_ranges),
};
static const struct regmap_access_table ws16c48_volatile_table = {
 .yes_ranges = ws16c48_volatile_ranges,
 .n_yes_ranges = ARRAY_SIZE(ws16c48_volatile_ranges),
};
static const struct regmap_config ws16c48_regmap_config = {
 .reg_bits = 8,
 .reg_stride = 1,
 .val_bits = 8,
 .io_port = true,
 .wr_table = &ws16c48_wr_table,
 .rd_table = &ws16c48_rd_table,
 .volatile_table = &ws16c48_volatile_table,
 .cache_type = REGCACHE_FLAT,
 .use_raw_spinlock = true,
};

#define WS16C48_NGPIO_PER_REG 8
#define WS16C48_REGMAP_IRQ(_id)       \
 [_id] = {        \
  .reg_offset = (_id) / WS16C48_NGPIO_PER_REG,   \
  .mask = BIT((_id) % WS16C48_NGPIO_PER_REG),   \
  .type = {       \
   .type_reg_offset = (_id) / WS16C48_NGPIO_PER_REG, \
   .types_supported = IRQ_TYPE_EDGE_BOTH,   \
  },        \
 }

/* Only the first 24 lines (Port 0-2) support interrupts */
#define WS16C48_NUM_IRQS 24
static const struct regmap_irq ws16c48_regmap_irqs[WS16C48_NUM_IRQS] = {
 WS16C48_REGMAP_IRQ(0), WS16C48_REGMAP_IRQ(1), WS16C48_REGMAP_IRQ(2), /* 0-2 */
 WS16C48_REGMAP_IRQ(3), WS16C48_REGMAP_IRQ(4), WS16C48_REGMAP_IRQ(5), /* 3-5 */
 WS16C48_REGMAP_IRQ(6), WS16C48_REGMAP_IRQ(7), WS16C48_REGMAP_IRQ(8), /* 6-8 */
 WS16C48_REGMAP_IRQ(9), WS16C48_REGMAP_IRQ(10), WS16C48_REGMAP_IRQ(11), /* 9-11 */
 WS16C48_REGMAP_IRQ(12), WS16C48_REGMAP_IRQ(13), WS16C48_REGMAP_IRQ(14), /* 12-14 */
 WS16C48_REGMAP_IRQ(15), WS16C48_REGMAP_IRQ(16), WS16C48_REGMAP_IRQ(17), /* 15-17 */
 WS16C48_REGMAP_IRQ(18), WS16C48_REGMAP_IRQ(19), WS16C48_REGMAP_IRQ(20), /* 18-20 */
 WS16C48_REGMAP_IRQ(21), WS16C48_REGMAP_IRQ(22), WS16C48_REGMAP_IRQ(23), /* 21-23 */
};

/**
 * struct ws16c48_gpio - GPIO device private data structure
 * @map: regmap for the device
 * @lock: synchronization lock to prevent I/O race conditions
 * @irq_mask: I/O bits affected by interrupts
 */
struct ws16c48_gpio {
 struct regmap *map;
 raw_spinlock_t lock;
 u8 irq_mask[WS16C48_NUM_IRQS / WS16C48_NGPIO_PER_REG];
};

static int ws16c48_handle_pre_irq(void *const irq_drv_data) __acquires(&ws16c48gpio->lock)
{
 struct ws16c48_gpio *const ws16c48gpio = irq_drv_data;

 /* Lock to prevent Page/Lock register change while we handle IRQ */
 raw_spin_lock(&ws16c48gpio->lock);

 return 0;
}

static int ws16c48_handle_post_irq(void *const irq_drv_data) __releases(&ws16c48gpio->lock)
{
 struct ws16c48_gpio *const ws16c48gpio = irq_drv_data;

 raw_spin_unlock(&ws16c48gpio->lock);

 return 0;
}

static int ws16c48_handle_mask_sync(const int index, const unsigned int mask_buf_def,
        const unsigned int mask_buf, void *const irq_drv_data)
{
 struct ws16c48_gpio *const ws16c48gpio = irq_drv_data;
 unsigned long flags;
 int ret = 0;

 raw_spin_lock_irqsave(&ws16c48gpio->lock, flags);

 /* exit early if no change since the last mask sync */
 if (mask_buf == ws16c48gpio->irq_mask[index])
  goto exit_unlock;
 ws16c48gpio->irq_mask[index] = mask_buf;

 ret = regmap_write(ws16c48gpio->map, WS16C48_PAGE_LOCK, ENAB_PAGE);
 if (ret)
  goto exit_unlock;

 /* Update ENAB register (inverted mask) */
 ret = regmap_write(ws16c48gpio->map, WS16C48_ENAB + index, ~mask_buf);
 if (ret)
  goto exit_unlock;

 ret = regmap_write(ws16c48gpio->map, WS16C48_PAGE_LOCK, INT_ID_PAGE);
 if (ret)
  goto exit_unlock;

exit_unlock:
 raw_spin_unlock_irqrestore(&ws16c48gpio->lock, flags);

 return ret;
}

static int ws16c48_set_type_config(unsigned int **const buf, const unsigned int type,
       const struct regmap_irq *const irq_data, const int idx,
       void *const irq_drv_data)
{
 struct ws16c48_gpio *const ws16c48gpio = irq_drv_data;
 unsigned int polarity;
 unsigned long flags;
 int ret;

 switch (type) {
 case IRQ_TYPE_EDGE_RISING:
  polarity = irq_data->mask;
  break;
 case IRQ_TYPE_EDGE_FALLING:
  polarity = 0;
  break;
 default:
  return -EINVAL;
 }

 raw_spin_lock_irqsave(&ws16c48gpio->lock, flags);

 ret = regmap_write(ws16c48gpio->map, WS16C48_PAGE_LOCK, POL_PAGE);
 if (ret)
  goto exit_unlock;

 /* Set interrupt polarity */
 ret = regmap_update_bits(ws16c48gpio->map, WS16C48_POL + idx, irq_data->mask, polarity);
 if (ret)
  goto exit_unlock;

 ret = regmap_write(ws16c48gpio->map, WS16C48_PAGE_LOCK, INT_ID_PAGE);
 if (ret)
  goto exit_unlock;

exit_unlock:
 raw_spin_unlock_irqrestore(&ws16c48gpio->lock, flags);

 return ret;
}

#define WS16C48_NGPIO 48
static const char *ws16c48_names[WS16C48_NGPIO] = {
 "Port 0 Bit 0", "Port 0 Bit 1", "Port 0 Bit 2", "Port 0 Bit 3",
 "Port 0 Bit 4", "Port 0 Bit 5", "Port 0 Bit 6", "Port 0 Bit 7",
 "Port 1 Bit 0", "Port 1 Bit 1", "Port 1 Bit 2", "Port 1 Bit 3",
 "Port 1 Bit 4", "Port 1 Bit 5", "Port 1 Bit 6", "Port 1 Bit 7",
 "Port 2 Bit 0", "Port 2 Bit 1", "Port 2 Bit 2", "Port 2 Bit 3",
 "Port 2 Bit 4", "Port 2 Bit 5", "Port 2 Bit 6", "Port 2 Bit 7",
 "Port 3 Bit 0", "Port 3 Bit 1", "Port 3 Bit 2", "Port 3 Bit 3",
 "Port 3 Bit 4", "Port 3 Bit 5", "Port 3 Bit 6", "Port 3 Bit 7",
 "Port 4 Bit 0", "Port 4 Bit 1", "Port 4 Bit 2", "Port 4 Bit 3",
 "Port 4 Bit 4", "Port 4 Bit 5", "Port 4 Bit 6", "Port 4 Bit 7",
 "Port 5 Bit 0", "Port 5 Bit 1", "Port 5 Bit 2", "Port 5 Bit 3",
 "Port 5 Bit 4", "Port 5 Bit 5", "Port 5 Bit 6", "Port 5 Bit 7"
};

static int ws16c48_irq_init_hw(struct regmap *const map)
{
 int err;

 err = regmap_write(map, WS16C48_PAGE_LOCK, ENAB_PAGE);
 if (err)
  return err;

 /* Disable interrupts for all lines */
 err = regmap_write(map, WS16C48_ENAB + 0, 0x00);
 if (err)
  return err;
 err = regmap_write(map, WS16C48_ENAB + 1, 0x00);
 if (err)
  return err;
 err = regmap_write(map, WS16C48_ENAB + 2, 0x00);
 if (err)
  return err;

 return regmap_write(map, WS16C48_PAGE_LOCK, INT_ID_PAGE);
}

static int ws16c48_probe(struct device *dev, unsigned int id)
{
 struct ws16c48_gpio *ws16c48gpio;
 const char *const name = dev_name(dev);
 int err;
 struct gpio_regmap_config gpio_config = {};
 void __iomem *regs;
 struct regmap_irq_chip *chip;
 struct regmap_irq_chip_data *chip_data;

 ws16c48gpio = devm_kzalloc(dev, sizeof(*ws16c48gpio), GFP_KERNEL);
 if (!ws16c48gpio)
  return -ENOMEM;

 if (!devm_request_region(dev, base[id], WS16C48_EXTENT, name)) {
  dev_err(dev, "Unable to lock port addresses (0x%X-0x%X)\n",
   base[id], base[id] + WS16C48_EXTENT);
  return -EBUSY;
 }

 regs = devm_ioport_map(dev, base[id], WS16C48_EXTENT);
 if (!regs)
  return -ENOMEM;

 ws16c48gpio->map = devm_regmap_init_mmio(dev, regs, &ws16c48_regmap_config);
 if (IS_ERR(ws16c48gpio->map))
  return dev_err_probe(dev, PTR_ERR(ws16c48gpio->map),
         "Unable to initialize register map\n");

 chip = devm_kzalloc(dev, sizeof(*chip), GFP_KERNEL);
 if (!chip)
  return -ENOMEM;

 chip->name = name;
 chip->status_base = WS16C48_INT_ID;
 chip->mask_base = WS16C48_ENAB;
 chip->ack_base = WS16C48_INT_ID;
 chip->num_regs = 3;
 chip->irqs = ws16c48_regmap_irqs;
 chip->num_irqs = ARRAY_SIZE(ws16c48_regmap_irqs);
 chip->handle_pre_irq = ws16c48_handle_pre_irq;
 chip->handle_post_irq = ws16c48_handle_post_irq;
 chip->handle_mask_sync = ws16c48_handle_mask_sync;
 chip->set_type_config = ws16c48_set_type_config;
 chip->irq_drv_data = ws16c48gpio;

 raw_spin_lock_init(&ws16c48gpio->lock);

 /* Initialize to prevent spurious interrupts before we're ready */
 err = ws16c48_irq_init_hw(ws16c48gpio->map);
 if (err)
  return err;

 err = devm_regmap_add_irq_chip(dev, ws16c48gpio->map, irq[id], 0, 0, chip, &chip_data);
 if (err)
  return dev_err_probe(dev, err, "IRQ registration failed\n");

 gpio_config.parent = dev;
 gpio_config.regmap = ws16c48gpio->map;
 gpio_config.ngpio = WS16C48_NGPIO;
 gpio_config.names = ws16c48_names;
 gpio_config.reg_dat_base = GPIO_REGMAP_ADDR(WS16C48_DAT_BASE);
 gpio_config.reg_set_base = GPIO_REGMAP_ADDR(WS16C48_DAT_BASE);
 /* Setting a GPIO to 0 allows it to be used as an input */
 gpio_config.reg_dir_out_base = GPIO_REGMAP_ADDR(WS16C48_DAT_BASE);
 gpio_config.ngpio_per_reg = WS16C48_NGPIO_PER_REG;
 gpio_config.irq_domain = regmap_irq_get_domain(chip_data);

 return PTR_ERR_OR_ZERO(devm_gpio_regmap_register(dev, &gpio_config));
}

static struct isa_driver ws16c48_driver = {
 .probe = ws16c48_probe,
 .driver = {
  .name = "ws16c48"
 },
};

module_isa_driver_with_irq(ws16c48_driver, num_ws16c48, num_irq);

MODULE_AUTHOR("William Breathitt Gray <vilhelm.gray@gmail.com>");
MODULE_DESCRIPTION("WinSystems WS16C48 GPIO driver");
MODULE_LICENSE("GPL v2");