// SPDX-License-Identifier: GPL-2.0+
//
// max8997.c - mfd core driver for the Maxim 8966 and 8997
//
// Copyright (C) 2011 Samsung Electronics
// MyungJoo Ham <myungjoo.ham@samsung.com>
//
// This driver is based on max8998.c

#include <linux/err.h>
#include <linux/slab.h>
#include <linux/i2c.h>
#include <linux/of.h>
#include <linux/of_irq.h>
#include <linux/interrupt.h>
#include <linux/pm_runtime.h>
#include <linux/init.h>
#include <linux/mutex.h>
#include <linux/mfd/core.h>
#include <linux/mfd/max8997.h>
#include <linux/mfd/max8997-private.h>

#define I2C_ADDR_PMIC (0xCC >> 1)
#define I2C_ADDR_MUIC (0x4A >> 1)
#define I2C_ADDR_BATTERY (0x6C >> 1)
#define I2C_ADDR_RTC (0x0C >> 1)
#define I2C_ADDR_HAPTIC (0x90 >> 1)

static const struct mfd_cell max8997_devs[] = {
 { .name = "max8997-pmic", },
 { .name = "max8997-rtc", },
 { .name = "max8997-battery", },
 { .name = "max8997-haptic", },
 { .name = "max8997-muic", },
 { .name = "max8997-led", .id = 1 },
 { .name = "max8997-led", .id = 2 },
};

#ifdef CONFIG_OF
static const struct of_device_id max8997_pmic_dt_match[] = {
 { .compatible = "maxim,max8997-pmic", .data = (void *)TYPE_MAX8997 },
 {},
};
#endif

int max8997_read_reg(struct i2c_client *i2c, u8 reg, u8 *dest)
{
 struct max8997_dev *max8997 = i2c_get_clientdata(i2c);
 int ret;

 mutex_lock(&max8997->iolock);
 ret = i2c_smbus_read_byte_data(i2c, reg);
 mutex_unlock(&max8997->iolock);
 if (ret < 0)
  return ret;

 ret &= 0xff;
 *dest = ret;
 return 0;
}
EXPORT_SYMBOL_GPL(max8997_read_reg);

int max8997_bulk_read(struct i2c_client *i2c, u8 reg, int count, u8 *buf)
{
 struct max8997_dev *max8997 = i2c_get_clientdata(i2c);
 int ret;

 mutex_lock(&max8997->iolock);
 ret = i2c_smbus_read_i2c_block_data(i2c, reg, count, buf);
 mutex_unlock(&max8997->iolock);
 if (ret < 0)
  return ret;

 return 0;
}
EXPORT_SYMBOL_GPL(max8997_bulk_read);

int max8997_write_reg(struct i2c_client *i2c, u8 reg, u8 value)
{
 struct max8997_dev *max8997 = i2c_get_clientdata(i2c);
 int ret;

 mutex_lock(&max8997->iolock);
 ret = i2c_smbus_write_byte_data(i2c, reg, value);
 mutex_unlock(&max8997->iolock);
 return ret;
}
EXPORT_SYMBOL_GPL(max8997_write_reg);

int max8997_bulk_write(struct i2c_client *i2c, u8 reg, int count, u8 *buf)
{
 struct max8997_dev *max8997 = i2c_get_clientdata(i2c);
 int ret;

 mutex_lock(&max8997->iolock);
 ret = i2c_smbus_write_i2c_block_data(i2c, reg, count, buf);
 mutex_unlock(&max8997->iolock);
 if (ret < 0)
  return ret;

 return 0;
}
EXPORT_SYMBOL_GPL(max8997_bulk_write);

int max8997_update_reg(struct i2c_client *i2c, u8 reg, u8 val, u8 mask)
{
 struct max8997_dev *max8997 = i2c_get_clientdata(i2c);
 int ret;

 mutex_lock(&max8997->iolock);
 ret = i2c_smbus_read_byte_data(i2c, reg);
 if (ret >= 0) {
  u8 old_val = ret & 0xff;
  u8 new_val = (val & mask) | (old_val & (~mask));
  ret = i2c_smbus_write_byte_data(i2c, reg, new_val);
 }
 mutex_unlock(&max8997->iolock);
 return ret;
}
EXPORT_SYMBOL_GPL(max8997_update_reg);

/*
 * Only the common platform data elements for max8997 are parsed here from the
 * device tree. Other sub-modules of max8997 such as pmic, rtc and others have
 * to parse their own platform data elements from device tree.
 *
 * The max8997 platform data structure is instantiated here and the drivers for
 * the sub-modules need not instantiate another instance while parsing their
 * platform data.
 */
static struct max8997_platform_data *max8997_i2c_parse_dt_pdata(
     struct device *dev)
{
 struct max8997_platform_data *pd;

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

 pd->ono = irq_of_parse_and_map(dev->of_node, 1);

 return pd;
}

static int max8997_i2c_probe(struct i2c_client *i2c)
{
 struct max8997_dev *max8997;
 struct max8997_platform_data *pdata = dev_get_platdata(&i2c->dev);
 int ret = 0;

 max8997 = devm_kzalloc(&i2c->dev, sizeof(struct max8997_dev),
    GFP_KERNEL);
 if (max8997 == NULL)
  return -ENOMEM;

 i2c_set_clientdata(i2c, max8997);
 max8997->dev = &i2c->dev;
 max8997->i2c = i2c;
 max8997->type = (uintptr_t)i2c_get_match_data(i2c);
 max8997->irq = i2c->irq;

 if (IS_ENABLED(CONFIG_OF) && max8997->dev->of_node) {
  pdata = max8997_i2c_parse_dt_pdata(max8997->dev);
  if (IS_ERR(pdata))
   return PTR_ERR(pdata);
 }

 if (!pdata)
  return ret;

 max8997->pdata = pdata;
 max8997->ono = pdata->ono;

 mutex_init(&max8997->iolock);

 max8997->rtc = i2c_new_dummy_device(i2c->adapter, I2C_ADDR_RTC);
 if (IS_ERR(max8997->rtc)) {
  dev_err(max8997->dev, "Failed to allocate I2C device for RTC\n");
  return PTR_ERR(max8997->rtc);
 }
 i2c_set_clientdata(max8997->rtc, max8997);

 max8997->haptic = i2c_new_dummy_device(i2c->adapter, I2C_ADDR_HAPTIC);
 if (IS_ERR(max8997->haptic)) {
  dev_err(max8997->dev, "Failed to allocate I2C device for Haptic\n");
  ret = PTR_ERR(max8997->haptic);
  goto err_i2c_haptic;
 }
 i2c_set_clientdata(max8997->haptic, max8997);

 max8997->muic = i2c_new_dummy_device(i2c->adapter, I2C_ADDR_MUIC);
 if (IS_ERR(max8997->muic)) {
  dev_err(max8997->dev, "Failed to allocate I2C device for MUIC\n");
  ret = PTR_ERR(max8997->muic);
  goto err_i2c_muic;
 }
 i2c_set_clientdata(max8997->muic, max8997);

 pm_runtime_set_active(max8997->dev);

 max8997_irq_init(max8997);

 ret = mfd_add_devices(max8997->dev, -1, max8997_devs,
   ARRAY_SIZE(max8997_devs),
   NULL, 0, NULL);
 if (ret < 0) {
  dev_err(max8997->dev, "failed to add MFD devices %d\n", ret);
  goto err_mfd;
 }

 /*
  * TODO: enable others (flash, muic, rtc, battery, ...) and
  * check the return value
 */

 /* MAX8997 has a power button input. */
 device_init_wakeup(max8997->dev, true);

 return ret;

err_mfd:
 mfd_remove_devices(max8997->dev);
 i2c_unregister_device(max8997->muic);
err_i2c_muic:
 i2c_unregister_device(max8997->haptic);
err_i2c_haptic:
 i2c_unregister_device(max8997->rtc);
 return ret;
}

static const struct i2c_device_id max8997_i2c_id[] = {
 { "max8997", TYPE_MAX8997 },
 { "max8966", TYPE_MAX8966 },
 { }
};

static u8 max8997_dumpaddr_pmic[] = {
 MAX8997_REG_INT1MSK,
 MAX8997_REG_INT2MSK,
 MAX8997_REG_INT3MSK,
 MAX8997_REG_INT4MSK,
 MAX8997_REG_MAINCON1,
 MAX8997_REG_MAINCON2,
 MAX8997_REG_BUCKRAMP,
 MAX8997_REG_BUCK1CTRL,
 MAX8997_REG_BUCK1DVS1,
 MAX8997_REG_BUCK1DVS2,
 MAX8997_REG_BUCK1DVS3,
 MAX8997_REG_BUCK1DVS4,
 MAX8997_REG_BUCK1DVS5,
 MAX8997_REG_BUCK1DVS6,
 MAX8997_REG_BUCK1DVS7,
 MAX8997_REG_BUCK1DVS8,
 MAX8997_REG_BUCK2CTRL,
 MAX8997_REG_BUCK2DVS1,
 MAX8997_REG_BUCK2DVS2,
 MAX8997_REG_BUCK2DVS3,
 MAX8997_REG_BUCK2DVS4,
 MAX8997_REG_BUCK2DVS5,
 MAX8997_REG_BUCK2DVS6,
 MAX8997_REG_BUCK2DVS7,
 MAX8997_REG_BUCK2DVS8,
 MAX8997_REG_BUCK3CTRL,
 MAX8997_REG_BUCK3DVS,
 MAX8997_REG_BUCK4CTRL,
 MAX8997_REG_BUCK4DVS,
 MAX8997_REG_BUCK5CTRL,
 MAX8997_REG_BUCK5DVS1,
 MAX8997_REG_BUCK5DVS2,
 MAX8997_REG_BUCK5DVS3,
 MAX8997_REG_BUCK5DVS4,
 MAX8997_REG_BUCK5DVS5,
 MAX8997_REG_BUCK5DVS6,
 MAX8997_REG_BUCK5DVS7,
 MAX8997_REG_BUCK5DVS8,
 MAX8997_REG_BUCK6CTRL,
 MAX8997_REG_BUCK6BPSKIPCTRL,
 MAX8997_REG_BUCK7CTRL,
 MAX8997_REG_BUCK7DVS,
 MAX8997_REG_LDO1CTRL,
 MAX8997_REG_LDO2CTRL,
 MAX8997_REG_LDO3CTRL,
 MAX8997_REG_LDO4CTRL,
 MAX8997_REG_LDO5CTRL,
 MAX8997_REG_LDO6CTRL,
 MAX8997_REG_LDO7CTRL,
 MAX8997_REG_LDO8CTRL,
 MAX8997_REG_LDO9CTRL,
 MAX8997_REG_LDO10CTRL,
 MAX8997_REG_LDO11CTRL,
 MAX8997_REG_LDO12CTRL,
 MAX8997_REG_LDO13CTRL,
 MAX8997_REG_LDO14CTRL,
 MAX8997_REG_LDO15CTRL,
 MAX8997_REG_LDO16CTRL,
 MAX8997_REG_LDO17CTRL,
 MAX8997_REG_LDO18CTRL,
 MAX8997_REG_LDO21CTRL,
 MAX8997_REG_MBCCTRL1,
 MAX8997_REG_MBCCTRL2,
 MAX8997_REG_MBCCTRL3,
 MAX8997_REG_MBCCTRL4,
 MAX8997_REG_MBCCTRL5,
 MAX8997_REG_MBCCTRL6,
 MAX8997_REG_OTPCGHCVS,
 MAX8997_REG_SAFEOUTCTRL,
 MAX8997_REG_LBCNFG1,
 MAX8997_REG_LBCNFG2,
 MAX8997_REG_BBCCTRL,

 MAX8997_REG_FLASH1_CUR,
 MAX8997_REG_FLASH2_CUR,
 MAX8997_REG_MOVIE_CUR,
 MAX8997_REG_GSMB_CUR,
 MAX8997_REG_BOOST_CNTL,
 MAX8997_REG_LEN_CNTL,
 MAX8997_REG_FLASH_CNTL,
 MAX8997_REG_WDT_CNTL,
 MAX8997_REG_MAXFLASH1,
 MAX8997_REG_MAXFLASH2,
 MAX8997_REG_FLASHSTATUSMASK,

 MAX8997_REG_GPIOCNTL1,
 MAX8997_REG_GPIOCNTL2,
 MAX8997_REG_GPIOCNTL3,
 MAX8997_REG_GPIOCNTL4,
 MAX8997_REG_GPIOCNTL5,
 MAX8997_REG_GPIOCNTL6,
 MAX8997_REG_GPIOCNTL7,
 MAX8997_REG_GPIOCNTL8,
 MAX8997_REG_GPIOCNTL9,
 MAX8997_REG_GPIOCNTL10,
 MAX8997_REG_GPIOCNTL11,
 MAX8997_REG_GPIOCNTL12,

 MAX8997_REG_LDO1CONFIG,
 MAX8997_REG_LDO2CONFIG,
 MAX8997_REG_LDO3CONFIG,
 MAX8997_REG_LDO4CONFIG,
 MAX8997_REG_LDO5CONFIG,
 MAX8997_REG_LDO6CONFIG,
 MAX8997_REG_LDO7CONFIG,
 MAX8997_REG_LDO8CONFIG,
 MAX8997_REG_LDO9CONFIG,
 MAX8997_REG_LDO10CONFIG,
 MAX8997_REG_LDO11CONFIG,
 MAX8997_REG_LDO12CONFIG,
 MAX8997_REG_LDO13CONFIG,
 MAX8997_REG_LDO14CONFIG,
 MAX8997_REG_LDO15CONFIG,
 MAX8997_REG_LDO16CONFIG,
 MAX8997_REG_LDO17CONFIG,
 MAX8997_REG_LDO18CONFIG,
 MAX8997_REG_LDO21CONFIG,

 MAX8997_REG_DVSOKTIMER1,
 MAX8997_REG_DVSOKTIMER2,
 MAX8997_REG_DVSOKTIMER4,
 MAX8997_REG_DVSOKTIMER5,
};

static u8 max8997_dumpaddr_muic[] = {
 MAX8997_MUIC_REG_INTMASK1,
 MAX8997_MUIC_REG_INTMASK2,
 MAX8997_MUIC_REG_INTMASK3,
 MAX8997_MUIC_REG_CDETCTRL,
 MAX8997_MUIC_REG_CONTROL1,
 MAX8997_MUIC_REG_CONTROL2,
 MAX8997_MUIC_REG_CONTROL3,
};

static u8 max8997_dumpaddr_haptic[] = {
 MAX8997_HAPTIC_REG_CONF1,
 MAX8997_HAPTIC_REG_CONF2,
 MAX8997_HAPTIC_REG_DRVCONF,
 MAX8997_HAPTIC_REG_CYCLECONF1,
 MAX8997_HAPTIC_REG_CYCLECONF2,
 MAX8997_HAPTIC_REG_SIGCONF1,
 MAX8997_HAPTIC_REG_SIGCONF2,
 MAX8997_HAPTIC_REG_SIGCONF3,
 MAX8997_HAPTIC_REG_SIGCONF4,
 MAX8997_HAPTIC_REG_SIGDC1,
 MAX8997_HAPTIC_REG_SIGDC2,
 MAX8997_HAPTIC_REG_SIGPWMDC1,
 MAX8997_HAPTIC_REG_SIGPWMDC2,
 MAX8997_HAPTIC_REG_SIGPWMDC3,
 MAX8997_HAPTIC_REG_SIGPWMDC4,
};

static int max8997_freeze(struct device *dev)
{
 struct i2c_client *i2c = to_i2c_client(dev);
 struct max8997_dev *max8997 = i2c_get_clientdata(i2c);
 int i;

 for (i = 0; i < ARRAY_SIZE(max8997_dumpaddr_pmic); i++)
  max8997_read_reg(i2c, max8997_dumpaddr_pmic[i],
    &max8997->reg_dump[i]);

 for (i = 0; i < ARRAY_SIZE(max8997_dumpaddr_muic); i++)
  max8997_read_reg(i2c, max8997_dumpaddr_muic[i],
    &max8997->reg_dump[i + MAX8997_REG_PMIC_END]);

 for (i = 0; i < ARRAY_SIZE(max8997_dumpaddr_haptic); i++)
  max8997_read_reg(i2c, max8997_dumpaddr_haptic[i],
    &max8997->reg_dump[i + MAX8997_REG_PMIC_END +
    MAX8997_MUIC_REG_END]);

 return 0;
}

static int max8997_restore(struct device *dev)
{
 struct i2c_client *i2c = to_i2c_client(dev);
 struct max8997_dev *max8997 = i2c_get_clientdata(i2c);
 int i;

 for (i = 0; i < ARRAY_SIZE(max8997_dumpaddr_pmic); i++)
  max8997_write_reg(i2c, max8997_dumpaddr_pmic[i],
    max8997->reg_dump[i]);

 for (i = 0; i < ARRAY_SIZE(max8997_dumpaddr_muic); i++)
  max8997_write_reg(i2c, max8997_dumpaddr_muic[i],
    max8997->reg_dump[i + MAX8997_REG_PMIC_END]);

 for (i = 0; i < ARRAY_SIZE(max8997_dumpaddr_haptic); i++)
  max8997_write_reg(i2c, max8997_dumpaddr_haptic[i],
    max8997->reg_dump[i + MAX8997_REG_PMIC_END +
    MAX8997_MUIC_REG_END]);

 return 0;
}

static int max8997_suspend(struct device *dev)
{
 struct i2c_client *i2c = to_i2c_client(dev);
 struct max8997_dev *max8997 = i2c_get_clientdata(i2c);

 disable_irq(max8997->irq);
 if (device_may_wakeup(dev))
  irq_set_irq_wake(max8997->irq, 1);
 return 0;
}

static int max8997_resume(struct device *dev)
{
 struct i2c_client *i2c = to_i2c_client(dev);
 struct max8997_dev *max8997 = i2c_get_clientdata(i2c);

 if (device_may_wakeup(dev))
  irq_set_irq_wake(max8997->irq, 0);
 enable_irq(max8997->irq);
 return max8997_irq_resume(max8997);
}

static const struct dev_pm_ops max8997_pm = {
 .suspend = max8997_suspend,
 .resume = max8997_resume,
 .freeze = max8997_freeze,
 .restore = max8997_restore,
};

static struct i2c_driver max8997_i2c_driver = {
 .driver = {
     .name = "max8997",
     .pm = &max8997_pm,
     .suppress_bind_attrs = true,
     .of_match_table = of_match_ptr(max8997_pmic_dt_match),
 },
 .probe = max8997_i2c_probe,
 .id_table = max8997_i2c_id,
};

static int __init max8997_i2c_init(void)
{
 return i2c_add_driver(&max8997_i2c_driver);
}
/* init early so consumer devices can complete system boot */
subsys_initcall(max8997_i2c_init);