// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright 2025 Bootlin
 *
 * Author: Kamel BOUHARA <kamel.bouhara@bootlin.com>
 * Author: Mathieu Dubois-Briand <mathieu.dubois-briand@bootlin.com>
 *
 * PWM functionality of the MAX7360 multi-function device.
 * https://www.analog.com/media/en/technical-documentation/data-sheets/MAX7360.pdf
 *
 * Limitations:
 * - Only supports normal polarity.
 * - The period is fixed to 2 ms.
 * - Only the duty cycle can be changed, new values are applied at the beginning
 *   of the next cycle.
 * - When disabled, the output is put in Hi-Z immediately.
 */
#include <linux/bits.h>
#include <linux/dev_printk.h>
#include <linux/err.h>
#include <linux/math64.h>
#include <linux/mfd/max7360.h>
#include <linux/minmax.h>
#include <linux/mod_devicetable.h>
#include <linux/module.h>
#include <linux/platform_device.h>
#include <linux/pwm.h>
#include <linux/regmap.h>
#include <linux/time.h>
#include <linux/types.h>

#define MAX7360_NUM_PWMS                        8
#define MAX7360_PWM_MAX                         255
#define MAX7360_PWM_STEPS                       256
#define MAX7360_PWM_PERIOD_NS                   (2 * NSEC_PER_MSEC)

struct max7360_pwm_waveform {
        u8 duty_steps;
        bool enabled;
};

static int max7360_pwm_request(struct pwm_chip *chip, struct pwm_device *pwm)
{
        struct regmap *regmap = pwmchip_get_drvdata(chip);

        /*
         * Make sure we use the individual PWM configuration register and not
         * the global one.
         * We never need to use the global one, so there is no need to revert
         * that in the .free() callback.
         */
        return regmap_write_bits(regmap, MAX7360_REG_PWMCFG(pwm->hwpwm),
                                 MAX7360_PORT_CFG_COMMON_PWM, 0);
}

static int max7360_pwm_round_waveform_tohw(struct pwm_chip *chip,
                                           struct pwm_device *pwm,
                                           const struct pwm_waveform *wf,
                                           void *_wfhw)
{
        struct max7360_pwm_waveform *wfhw = _wfhw;
        u64 duty_steps;

        /*
         * Ignore user provided values for period_length_ns and duty_offset_ns:
         * we only support fixed period of MAX7360_PWM_PERIOD_NS and offset of 0.
         * Values from 0 to 254 as duty_steps will provide duty cycles of 0/256
         * to 254/256, while value 255 will provide a duty cycle of 100%.
         */
        if (wf->duty_length_ns >= MAX7360_PWM_PERIOD_NS) {
                duty_steps = MAX7360_PWM_MAX;
        } else {
                duty_steps = (u32)wf->duty_length_ns * MAX7360_PWM_STEPS / MAX7360_PWM_PERIOD_NS;
                if (duty_steps == MAX7360_PWM_MAX)
                        duty_steps = MAX7360_PWM_MAX - 1;
        }

        wfhw->duty_steps = duty_steps;
        wfhw->enabled = !!wf->period_length_ns;

        if (wf->period_length_ns && wf->period_length_ns < MAX7360_PWM_PERIOD_NS)
                return 1;
        else
                return 0;
}

static int max7360_pwm_round_waveform_fromhw(struct pwm_chip *chip, struct pwm_device *pwm,
                                             const void *_wfhw, struct pwm_waveform *wf)
{
        const struct max7360_pwm_waveform *wfhw = _wfhw;

        wf->period_length_ns = wfhw->enabled ? MAX7360_PWM_PERIOD_NS : 0;
        wf->duty_offset_ns = 0;

        if (wfhw->enabled) {
                if (wfhw->duty_steps == MAX7360_PWM_MAX)
                        wf->duty_length_ns = MAX7360_PWM_PERIOD_NS;
                else
                        wf->duty_length_ns = DIV_ROUND_UP(wfhw->duty_steps * MAX7360_PWM_PERIOD_NS,
                                                          MAX7360_PWM_STEPS);
        } else {
                wf->duty_length_ns = 0;
        }

        return 0;
}

static int max7360_pwm_write_waveform(struct pwm_chip *chip,
                                      struct pwm_device *pwm,
                                      const void *_wfhw)
{
        struct regmap *regmap = pwmchip_get_drvdata(chip);
        const struct max7360_pwm_waveform *wfhw = _wfhw;
        unsigned int val;
        int ret;

        if (wfhw->enabled) {
                ret = regmap_write(regmap, MAX7360_REG_PWM(pwm->hwpwm), wfhw->duty_steps);
                if (ret)
                        return ret;
        }

        val = wfhw->enabled ? BIT(pwm->hwpwm) : 0;
        return regmap_write_bits(regmap, MAX7360_REG_GPIOCTRL, BIT(pwm->hwpwm), val);
}

static int max7360_pwm_read_waveform(struct pwm_chip *chip,
                                     struct pwm_device *pwm,
                                     void *_wfhw)
{
        struct regmap *regmap = pwmchip_get_drvdata(chip);
        struct max7360_pwm_waveform *wfhw = _wfhw;
        unsigned int val;
        int ret;

        ret = regmap_read(regmap, MAX7360_REG_GPIOCTRL, &val);
        if (ret)
                return ret;

        if (val & BIT(pwm->hwpwm)) {
                wfhw->enabled = true;
                ret = regmap_read(regmap, MAX7360_REG_PWM(pwm->hwpwm), &val);
                if (ret)
                        return ret;

                wfhw->duty_steps = val;
        } else {
                wfhw->enabled = false;
                wfhw->duty_steps = 0;
        }

        return 0;
}

static const struct pwm_ops max7360_pwm_ops = {
        .sizeof_wfhw = sizeof(struct max7360_pwm_waveform),
        .request = max7360_pwm_request,
        .round_waveform_tohw = max7360_pwm_round_waveform_tohw,
        .round_waveform_fromhw = max7360_pwm_round_waveform_fromhw,
        .read_waveform = max7360_pwm_read_waveform,
        .write_waveform = max7360_pwm_write_waveform,
};

static int max7360_pwm_probe(struct platform_device *pdev)
{
        struct device *dev = &pdev->dev;
        struct pwm_chip *chip;
        struct regmap *regmap;
        int ret;

        regmap = dev_get_regmap(dev->parent, NULL);
        if (!regmap)
                return dev_err_probe(dev, -ENODEV, "Could not get parent regmap\n");

        /*
         * This MFD sub-device does not have any associated device tree node:
         * properties are stored in the device node of the parent (MFD) device
         * and this same node is used in phandles of client devices.
         * Reuse this device tree node here, as otherwise the PWM subsystem
         * would be confused by this topology.
         */
        device_set_of_node_from_dev(dev, dev->parent);

        chip = devm_pwmchip_alloc(dev, MAX7360_NUM_PWMS, 0);
        if (IS_ERR(chip))
                return PTR_ERR(chip);
        chip->ops = &max7360_pwm_ops;

        pwmchip_set_drvdata(chip, regmap);

        ret = devm_pwmchip_add(dev, chip);
        if (ret)
                return dev_err_probe(dev, ret, "Failed to add PWM chip\n");

        return 0;
}

static struct platform_driver max7360_pwm_driver = {
        .driver = {
                .name = "max7360-pwm",
                .probe_type = PROBE_PREFER_ASYNCHRONOUS,
        },
        .probe = max7360_pwm_probe,
};
module_platform_driver(max7360_pwm_driver);

MODULE_DESCRIPTION("MAX7360 PWM driver");
MODULE_AUTHOR("Kamel BOUHARA <kamel.bouhara@bootlin.com>");
MODULE_AUTHOR("Mathieu Dubois-Briand <mathieu.dubois-briand@bootlin.com>");
MODULE_LICENSE("GPL");