// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2013 Capella Microsystems Inc.
 * Author: Kevin Tsai <ktsai@capellamicro.com>
 */

#include <linux/acpi.h>
#include <linux/delay.h>
#include <linux/err.h>
#include <linux/i2c.h>
#include <linux/mutex.h>
#include <linux/module.h>
#include <linux/mod_devicetable.h>
#include <linux/interrupt.h>
#include <linux/regulator/consumer.h>
#include <linux/iio/iio.h>
#include <linux/iio/sysfs.h>
#include <linux/iio/events.h>
#include <linux/init.h>

/* Registers Address */
#define CM32181_REG_ADDR_CMD            0x00
#define CM32181_REG_ADDR_WH             0x01
#define CM32181_REG_ADDR_WL             0x02
#define CM32181_REG_ADDR_TEST           0x03
#define CM32181_REG_ADDR_ALS            0x04
#define CM32181_REG_ADDR_STATUS         0x06
#define CM32181_REG_ADDR_ID             0x07

/* Number of Configurable Registers */
#define CM32181_CONF_REG_NUM            4

/* CMD register */
#define CM32181_CMD_ALS_DISABLE         BIT(0)
#define CM32181_CMD_ALS_INT_EN          BIT(1)
#define CM32181_CMD_ALS_THRES_WINDOW    BIT(2)

#define CM32181_CMD_ALS_PERS_SHIFT      4
#define CM32181_CMD_ALS_PERS_MASK       (0x03 << CM32181_CMD_ALS_PERS_SHIFT)
#define CM32181_CMD_ALS_PERS_DEFAULT    (0x01 << CM32181_CMD_ALS_PERS_SHIFT)

#define CM32181_CMD_ALS_IT_SHIFT        6
#define CM32181_CMD_ALS_IT_MASK         (0x0F << CM32181_CMD_ALS_IT_SHIFT)
#define CM32181_CMD_ALS_IT_DEFAULT      (0x00 << CM32181_CMD_ALS_IT_SHIFT)

#define CM32181_CMD_ALS_SM_SHIFT        11
#define CM32181_CMD_ALS_SM_MASK         (0x03 << CM32181_CMD_ALS_SM_SHIFT)
#define CM32181_CMD_ALS_SM_DEFAULT      (0x01 << CM32181_CMD_ALS_SM_SHIFT)

#define CM32181_LUX_PER_BIT             500     /* ALS_SM=01 IT=800ms */
#define CM32181_LUX_PER_BIT_RESOLUTION  100000
#define CM32181_LUX_PER_BIT_BASE_IT     800000  /* Based on IT=800ms */
#define CM32181_CALIBSCALE_DEFAULT      100000
#define CM32181_CALIBSCALE_RESOLUTION   100000

#define SMBUS_ALERT_RESPONSE_ADDRESS    0x0c

/* CPM0 Index 0: device-id (3218 or 32181), 1: Unknown, 2: init_regs_bitmap */
#define CPM0_REGS_BITMAP                2
#define CPM0_HEADER_SIZE                3

/* CPM1 Index 0: lux_per_bit, 1: calibscale, 2: resolution (100000) */
#define CPM1_LUX_PER_BIT                0
#define CPM1_CALIBSCALE                 1
#define CPM1_SIZE                       3

/* CM3218 Family */
static const int cm3218_als_it_bits[] = { 0, 1, 2, 3 };
static const int cm3218_als_it_values[] = { 100000, 200000, 400000, 800000 };

/* CM32181 Family */
static const int cm32181_als_it_bits[] = { 12, 8, 0, 1, 2, 3 };
static const int cm32181_als_it_values[] = {
        25000, 50000, 100000, 200000, 400000, 800000
};

struct cm32181_chip {
        struct i2c_client *client;
        struct device *dev;
        struct mutex lock;
        u16 conf_regs[CM32181_CONF_REG_NUM];
        unsigned long init_regs_bitmap;
        int calibscale;
        int lux_per_bit;
        int lux_per_bit_base_it;
        int num_als_it;
        const int *als_it_bits;
        const int *als_it_values;
};

static int cm32181_read_als_it(struct cm32181_chip *cm32181, int *val2);

#ifdef CONFIG_ACPI
/**
 * cm32181_acpi_get_cpm() - Get CPM object from ACPI
 * @dev:        pointer of struct device.
 * @obj_name:   pointer of ACPI object name.
 * @values:     pointer of array for return elements.
 * @count:      maximum size of return array.
 *
 * Convert ACPI CPM table to array.
 *
 * Return: -ENODEV for fail.  Otherwise is number of elements.
 */
static int cm32181_acpi_get_cpm(struct device *dev, char *obj_name,
                                u64 *values, int count)
{
        struct acpi_buffer buffer = { ACPI_ALLOCATE_BUFFER, NULL };
        union acpi_object *cpm, *elem;
        acpi_handle handle;
        acpi_status status;
        int i;

        handle = ACPI_HANDLE(dev);
        if (!handle)
                return -ENODEV;

        status = acpi_evaluate_object(handle, obj_name, NULL, &buffer);
        if (ACPI_FAILURE(status)) {
                dev_err(dev, "object %s not found\n", obj_name);
                return -ENODEV;
        }

        cpm = buffer.pointer;
        if (cpm->package.count > count)
                dev_warn(dev, "%s table contains %u values, only using first %d values\n",
                         obj_name, cpm->package.count, count);

        count = min_t(int, cpm->package.count, count);
        for (i = 0; i < count; i++) {
                elem = &(cpm->package.elements[i]);
                values[i] = elem->integer.value;
        }

        kfree(buffer.pointer);

        return count;
}

static void cm32181_acpi_parse_cpm_tables(struct cm32181_chip *cm32181)
{
        u64 vals[CPM0_HEADER_SIZE + CM32181_CONF_REG_NUM];
        struct device *dev = cm32181->dev;
        int i, count;

        count = cm32181_acpi_get_cpm(dev, "CPM0", vals, ARRAY_SIZE(vals));
        if (count <= CPM0_HEADER_SIZE)
                return;

        count -= CPM0_HEADER_SIZE;

        cm32181->init_regs_bitmap = vals[CPM0_REGS_BITMAP];
        cm32181->init_regs_bitmap &= GENMASK(count - 1, 0);
        for_each_set_bit(i, &cm32181->init_regs_bitmap, count)
                cm32181->conf_regs[i] = vals[CPM0_HEADER_SIZE + i];

        count = cm32181_acpi_get_cpm(dev, "CPM1", vals, ARRAY_SIZE(vals));
        if (count != CPM1_SIZE)
                return;

        cm32181->lux_per_bit = vals[CPM1_LUX_PER_BIT];

        /* Check for uncalibrated devices */
        if (vals[CPM1_CALIBSCALE] == CM32181_CALIBSCALE_DEFAULT)
                return;

        cm32181->calibscale = vals[CPM1_CALIBSCALE];
        /* CPM1 lux_per_bit is for the current it value */
        cm32181_read_als_it(cm32181, &cm32181->lux_per_bit_base_it);
}
#else
static void cm32181_acpi_parse_cpm_tables(struct cm32181_chip *cm32181)
{
}
#endif /* CONFIG_ACPI */

/**
 * cm32181_reg_init() - Initialize CM32181 registers
 * @cm32181:    pointer of struct cm32181.
 *
 * Initialize CM32181 ambient light sensor register to default values.
 *
 * Return: 0 for success; otherwise for error code.
 */
static int cm32181_reg_init(struct cm32181_chip *cm32181)
{
        struct i2c_client *client = cm32181->client;
        int i;
        s32 ret;

        ret = i2c_smbus_read_word_data(client, CM32181_REG_ADDR_ID);
        if (ret < 0)
                return ret;

        /* check device ID */
        switch (ret & 0xFF) {
        case 0x18: /* CM3218 */
                cm32181->num_als_it = ARRAY_SIZE(cm3218_als_it_bits);
                cm32181->als_it_bits = cm3218_als_it_bits;
                cm32181->als_it_values = cm3218_als_it_values;
                break;
        case 0x81: /* CM32181 */
        case 0x82: /* CM32182, fully compat. with CM32181 */
                cm32181->num_als_it = ARRAY_SIZE(cm32181_als_it_bits);
                cm32181->als_it_bits = cm32181_als_it_bits;
                cm32181->als_it_values = cm32181_als_it_values;
                break;
        default:
                return -ENODEV;
        }

        /* Default Values */
        cm32181->conf_regs[CM32181_REG_ADDR_CMD] =
                        CM32181_CMD_ALS_IT_DEFAULT | CM32181_CMD_ALS_SM_DEFAULT;
        cm32181->init_regs_bitmap = BIT(CM32181_REG_ADDR_CMD);
        cm32181->calibscale = CM32181_CALIBSCALE_DEFAULT;
        cm32181->lux_per_bit = CM32181_LUX_PER_BIT;
        cm32181->lux_per_bit_base_it = CM32181_LUX_PER_BIT_BASE_IT;

        cm32181_acpi_parse_cpm_tables(cm32181);

        /* Initialize registers*/
        for_each_set_bit(i, &cm32181->init_regs_bitmap, CM32181_CONF_REG_NUM) {
                ret = i2c_smbus_write_word_data(client, i,
                                                cm32181->conf_regs[i]);
                if (ret < 0)
                        return ret;
        }

        return 0;
}

/**
 *  cm32181_read_als_it() - Get sensor integration time (ms)
 *  @cm32181:   pointer of struct cm32181
 *  @val2:      pointer of int to load the als_it value.
 *
 *  Report the current integration time in milliseconds.
 *
 *  Return: IIO_VAL_INT_PLUS_MICRO for success, otherwise -EINVAL.
 */
static int cm32181_read_als_it(struct cm32181_chip *cm32181, int *val2)
{
        u16 als_it;
        int i;

        als_it = cm32181->conf_regs[CM32181_REG_ADDR_CMD];
        als_it &= CM32181_CMD_ALS_IT_MASK;
        als_it >>= CM32181_CMD_ALS_IT_SHIFT;
        for (i = 0; i < cm32181->num_als_it; i++) {
                if (als_it == cm32181->als_it_bits[i]) {
                        *val2 = cm32181->als_it_values[i];
                        return IIO_VAL_INT_PLUS_MICRO;
                }
        }

        return -EINVAL;
}

/**
 * cm32181_write_als_it() - Write sensor integration time
 * @cm32181:    pointer of struct cm32181.
 * @val:        integration time by millisecond.
 *
 * Convert integration time (ms) to sensor value.
 *
 * Return: i2c_smbus_write_word_data command return value.
 */
static int cm32181_write_als_it(struct cm32181_chip *cm32181, int val)
{
        struct i2c_client *client = cm32181->client;
        u16 als_it;
        int ret, i, n;

        n = cm32181->num_als_it;
        for (i = 0; i < n; i++)
                if (val <= cm32181->als_it_values[i])
                        break;
        if (i >= n)
                i = n - 1;

        als_it = cm32181->als_it_bits[i];
        als_it <<= CM32181_CMD_ALS_IT_SHIFT;

        mutex_lock(&cm32181->lock);
        cm32181->conf_regs[CM32181_REG_ADDR_CMD] &=
                ~CM32181_CMD_ALS_IT_MASK;
        cm32181->conf_regs[CM32181_REG_ADDR_CMD] |=
                als_it;
        ret = i2c_smbus_write_word_data(client, CM32181_REG_ADDR_CMD,
                        cm32181->conf_regs[CM32181_REG_ADDR_CMD]);
        mutex_unlock(&cm32181->lock);

        return ret;
}

/**
 * cm32181_get_lux() - report current lux value
 * @cm32181:    pointer of struct cm32181.
 *
 * Convert sensor raw data to lux.  It depends on integration
 * time and calibscale variable.
 *
 * Return: Positive value is lux, otherwise is error code.
 */
static int cm32181_get_lux(struct cm32181_chip *cm32181)
{
        struct i2c_client *client = cm32181->client;
        int ret;
        int als_it;
        u64 lux;

        ret = cm32181_read_als_it(cm32181, &als_it);
        if (ret < 0)
                return -EINVAL;

        lux = cm32181->lux_per_bit;
        lux *= cm32181->lux_per_bit_base_it;
        lux = div_u64(lux, als_it);

        ret = i2c_smbus_read_word_data(client, CM32181_REG_ADDR_ALS);
        if (ret < 0)
                return ret;

        lux *= ret;
        lux *= cm32181->calibscale;
        lux = div_u64(lux, CM32181_CALIBSCALE_RESOLUTION);
        lux = div_u64(lux, CM32181_LUX_PER_BIT_RESOLUTION);

        if (lux > 0xFFFF)
                lux = 0xFFFF;

        return lux;
}

static int cm32181_read_raw(struct iio_dev *indio_dev,
                            struct iio_chan_spec const *chan,
                            int *val, int *val2, long mask)
{
        struct cm32181_chip *cm32181 = iio_priv(indio_dev);
        int ret;

        switch (mask) {
        case IIO_CHAN_INFO_PROCESSED:
                ret = cm32181_get_lux(cm32181);
                if (ret < 0)
                        return ret;
                *val = ret;
                return IIO_VAL_INT;
        case IIO_CHAN_INFO_CALIBSCALE:
                *val = cm32181->calibscale;
                return IIO_VAL_INT;
        case IIO_CHAN_INFO_INT_TIME:
                *val = 0;
                ret = cm32181_read_als_it(cm32181, val2);
                return ret;
        }

        return -EINVAL;
}

static int cm32181_write_raw(struct iio_dev *indio_dev,
                             struct iio_chan_spec const *chan,
                             int val, int val2, long mask)
{
        struct cm32181_chip *cm32181 = iio_priv(indio_dev);
        int ret;

        switch (mask) {
        case IIO_CHAN_INFO_CALIBSCALE:
                cm32181->calibscale = val;
                return val;
        case IIO_CHAN_INFO_INT_TIME:
                ret = cm32181_write_als_it(cm32181, val2);
                return ret;
        }

        return -EINVAL;
}

/**
 * cm32181_get_it_available() - Get available ALS IT value
 * @dev:        pointer of struct device.
 * @attr:       pointer of struct device_attribute.
 * @buf:        pointer of return string buffer.
 *
 * Display the available integration time values by millisecond.
 *
 * Return: string length.
 */
static ssize_t cm32181_get_it_available(struct device *dev,
                        struct device_attribute *attr, char *buf)
{
        struct cm32181_chip *cm32181 = iio_priv(dev_to_iio_dev(dev));
        int i, n, len;

        n = cm32181->num_als_it;
        for (i = 0, len = 0; i < n; i++)
                len += sprintf(buf + len, "0.%06u ", cm32181->als_it_values[i]);
        return len + sprintf(buf + len, "\n");
}

static const struct iio_chan_spec cm32181_channels[] = {
        {
                .type = IIO_LIGHT,
                .info_mask_separate =
                        BIT(IIO_CHAN_INFO_PROCESSED) |
                        BIT(IIO_CHAN_INFO_CALIBSCALE) |
                        BIT(IIO_CHAN_INFO_INT_TIME),
        }
};

static IIO_DEVICE_ATTR(in_illuminance_integration_time_available,
                        S_IRUGO, cm32181_get_it_available, NULL, 0);

static struct attribute *cm32181_attributes[] = {
        &iio_dev_attr_in_illuminance_integration_time_available.dev_attr.attr,
        NULL,
};

static const struct attribute_group cm32181_attribute_group = {
        .attrs = cm32181_attributes
};

static const struct iio_info cm32181_info = {
        .read_raw               = &cm32181_read_raw,
        .write_raw              = &cm32181_write_raw,
        .attrs                  = &cm32181_attribute_group,
};

static void cm32181_unregister_dummy_client(void *data)
{
        struct i2c_client *client = data;

        /* Unregister the dummy client */
        i2c_unregister_device(client);
}

static int cm32181_probe(struct i2c_client *client)
{
        struct device *dev = &client->dev;
        struct cm32181_chip *cm32181;
        struct iio_dev *indio_dev;
        int ret;

        indio_dev = devm_iio_device_alloc(dev, sizeof(*cm32181));
        if (!indio_dev)
                return -ENOMEM;

        i2c_set_clientdata(client, indio_dev);

        /*
         * Some ACPI systems list 2 I2C resources for the CM3218 sensor, the
         * SMBus Alert Response Address (ARA, 0x0c) and the actual I2C address.
         * Detect this and take the following step to deal with it:
         * 1. When a SMBus Alert capable sensor has an Alert asserted, it will
         *    not respond on its actual I2C address. Read a byte from the ARA
         *    to clear any pending Alerts.
         * 2. Create a "dummy" client for the actual I2C address and
         *    use that client to communicate with the sensor.
         */
        if (ACPI_HANDLE(dev) && client->addr == SMBUS_ALERT_RESPONSE_ADDRESS) {
                struct i2c_board_info board_info = { .type = "dummy" };

                i2c_smbus_read_byte(client);

                client = i2c_acpi_new_device(dev, 1, &board_info);
                if (IS_ERR(client))
                        return PTR_ERR(client);

                ret = devm_add_action_or_reset(dev, cm32181_unregister_dummy_client, client);
                if (ret)
                        return ret;
        }

        cm32181 = iio_priv(indio_dev);
        cm32181->client = client;
        cm32181->dev = dev;

        mutex_init(&cm32181->lock);
        indio_dev->channels = cm32181_channels;
        indio_dev->num_channels = ARRAY_SIZE(cm32181_channels);
        indio_dev->info = &cm32181_info;
        indio_dev->name = dev_name(dev);
        indio_dev->modes = INDIO_DIRECT_MODE;

        ret = cm32181_reg_init(cm32181);
        if (ret) {
                dev_err(dev, "%s: register init failed\n", __func__);
                return ret;
        }

        ret = devm_iio_device_register(dev, indio_dev);
        if (ret) {
                dev_err(dev, "%s: register device failed\n", __func__);
                return ret;
        }

        return 0;
}

static int cm32181_suspend(struct device *dev)
{
        struct cm32181_chip *cm32181 = iio_priv(dev_get_drvdata(dev));
        struct i2c_client *client = cm32181->client;

        return i2c_smbus_write_word_data(client, CM32181_REG_ADDR_CMD,
                                         CM32181_CMD_ALS_DISABLE);
}

static int cm32181_resume(struct device *dev)
{
        struct cm32181_chip *cm32181 = iio_priv(dev_get_drvdata(dev));
        struct i2c_client *client = cm32181->client;

        return i2c_smbus_write_word_data(client, CM32181_REG_ADDR_CMD,
                                         cm32181->conf_regs[CM32181_REG_ADDR_CMD]);
}

static DEFINE_SIMPLE_DEV_PM_OPS(cm32181_pm_ops, cm32181_suspend, cm32181_resume);

static const struct of_device_id cm32181_of_match[] = {
        { .compatible = "capella,cm3218" },
        { .compatible = "capella,cm32181" },
        { }
};
MODULE_DEVICE_TABLE(of, cm32181_of_match);

#ifdef CONFIG_ACPI
static const struct acpi_device_id cm32181_acpi_match[] = {
        { "CPLM3218", 0 },
        { }
};
MODULE_DEVICE_TABLE(acpi, cm32181_acpi_match);
#endif

static struct i2c_driver cm32181_driver = {
        .driver = {
                .name   = "cm32181",
                .acpi_match_table = ACPI_PTR(cm32181_acpi_match),
                .of_match_table = cm32181_of_match,
                .pm = pm_sleep_ptr(&cm32181_pm_ops),
        },
        .probe          = cm32181_probe,
};

module_i2c_driver(cm32181_driver);

MODULE_AUTHOR("Kevin Tsai <ktsai@capellamicro.com>");
MODULE_DESCRIPTION("CM32181 ambient light sensor driver");
MODULE_LICENSE("GPL");