// SPDX-License-Identifier: GPL-2.0-only
/*
 * STMicroelectronics COMBOPHY STM32MP25 Controller driver.
 *
 * Copyright (C) 2024 STMicroelectronics
 * Author: Christian Bruel <christian.bruel@foss.st.com>
 */

#include <linux/bitfield.h>
#include <linux/clk.h>
#include <linux/mfd/syscon.h>
#include <linux/platform_device.h>
#include <linux/phy/phy.h>
#include <linux/pm_runtime.h>
#include <linux/regmap.h>
#include <linux/reset.h>
#include <dt-bindings/phy/phy.h>

#define SYSCFG_COMBOPHY_CR1 0x4c00
#define SYSCFG_COMBOPHY_CR2 0x4c04
#define SYSCFG_COMBOPHY_CR4 0x4c0c
#define SYSCFG_COMBOPHY_CR5 0x4c10
#define SYSCFG_COMBOPHY_SR  0x4c14
#define SYSCFG_PCIEPRGCR    0x6080

/* SYSCFG PCIEPRGCR */
#define STM32MP25_PCIEPRGCR_EN   BIT(0)
#define STM32MP25_PCIEPRG_IMPCTRL_OHM     GENMASK(3, 1)
#define STM32MP25_PCIEPRG_IMPCTRL_VSWING  GENMASK(5, 4)

/* SYSCFG SYSCFG_COMBOPHY_SR */
#define STM32MP25_PIPE0_PHYSTATUS BIT(1)

/* SYSCFG CR1 */
#define SYSCFG_COMBOPHY_CR1_REFUSEPAD BIT(0)
#define SYSCFG_COMBOPHY_CR1_MPLLMULT GENMASK(7, 1)
#define SYSCFG_COMBOPHY_CR1_REFCLKSEL GENMASK(16, 8)
#define SYSCFG_COMBOPHY_CR1_REFCLKDIV2 BIT(17)
#define SYSCFG_COMBOPHY_CR1_REFSSPEN BIT(18)
#define SYSCFG_COMBOPHY_CR1_SSCEN BIT(19)

/* SYSCFG CR4 */
#define SYSCFG_COMBOPHY_CR4_RX0_EQ GENMASK(2, 0)

#define MPLLMULT_19_2 (0x02u << 1)
#define MPLLMULT_20   (0x7du << 1)
#define MPLLMULT_24   (0x68u << 1)
#define MPLLMULT_25   (0x64u << 1)
#define MPLLMULT_26   (0x60u << 1)
#define MPLLMULT_38_4 (0x41u << 1)
#define MPLLMULT_48   (0x6cu << 1)
#define MPLLMULT_50   (0x32u << 1)
#define MPLLMULT_52   (0x30u << 1)
#define MPLLMULT_100  (0x19u << 1)

#define REFCLKSEL_0   0
#define REFCLKSEL_1   (0x108u << 8)

#define REFCLDIV_0    0

/* SYSCFG CR2 */
#define SYSCFG_COMBOPHY_CR2_MODESEL GENMASK(1, 0)
#define SYSCFG_COMBOPHY_CR2_ISO_DIS BIT(15)

#define COMBOPHY_MODESEL_PCIE 0
#define COMBOPHY_MODESEL_USB  3

/* SYSCFG CR5 */
#define SYSCFG_COMBOPHY_CR5_COMMON_CLOCKS BIT(12)

#define COMBOPHY_SUP_ANA_MPLL_LOOP_CTL 0xc0
#define COMBOPHY_PROP_CNTRL GENMASK(7, 4)

/* Required apb/ker clocks first, optional pad last. */
static const char * const combophy_clks[] = {"apb", "ker", "pad"};
#define APB_CLK 0
#define KER_CLK 1
#define PAD_CLK 2

struct stm32_combophy {
 struct phy *phy;
 struct regmap *regmap;
 struct device *dev;
 void __iomem *base;
 struct reset_control *phy_reset;
 struct clk_bulk_data clks[ARRAY_SIZE(combophy_clks)];
 int num_clks;
 bool have_pad_clk;
 unsigned int type;
 bool is_init;
 int irq_wakeup;
};

struct clk_impedance  {
 u32 microohm;
 u32 vswing[4];
};

/*
 * lookup table to hold the settings needed for a ref clock frequency
 * impedance, the offset is used to set the IMP_CTL and DE_EMP bit of the
 * PRG_IMP_CTRL register. Use ordered discrete values in the table
 */
static const struct clk_impedance imp_lookup[] = {
 { 6090000, { 442000, 564000, 684000, 802000 } },
 { 5662000, { 528000, 621000, 712000, 803000 } },
 { 5292000, { 491000, 596000, 700000, 802000 } },
 { 4968000, { 558000, 640000, 722000, 803000 } },
 { 4684000, { 468000, 581000, 692000, 802000 } },
 { 4429000, { 554000, 613000, 717000, 803000 } },
 { 4204000, { 511000, 609000, 706000, 802000 } },
 { 3999000, { 571000, 648000, 726000, 803000 } }
};
#define DEFAULT_IMP_INDEX 3 /* Default impedance is 50 Ohm */

static int stm32_impedance_tune(struct stm32_combophy *combophy)
{
 u8 imp_size = ARRAY_SIZE(imp_lookup);
 u8 vswing_size = ARRAY_SIZE(imp_lookup[0].vswing);
 u8 imp_of, vswing_of;
 u32 max_imp = imp_lookup[0].microohm;
 u32 min_imp = imp_lookup[imp_size - 1].microohm;
 u32 max_vswing;
 u32 min_vswing = imp_lookup[0].vswing[0];
 u32 val;

 if (!of_property_read_u32(combophy->dev->of_node, "st,output-micro-ohms", &val)) {
  if (val < min_imp || val > max_imp) {
   dev_err(combophy->dev, "Invalid value %u for output ohm\n", val);
   return -EINVAL;
  }

  for (imp_of = 0; imp_of < ARRAY_SIZE(imp_lookup); imp_of++)
   if (imp_lookup[imp_of].microohm <= val)
    break;

  if (WARN_ON(imp_of == ARRAY_SIZE(imp_lookup)))
   return -EINVAL;

  dev_dbg(combophy->dev, "Set %u micro-ohms output impedance\n",
   imp_lookup[imp_of].microohm);

  regmap_update_bits(combophy->regmap, SYSCFG_PCIEPRGCR,
       STM32MP25_PCIEPRG_IMPCTRL_OHM,
       FIELD_PREP(STM32MP25_PCIEPRG_IMPCTRL_OHM, imp_of));
 } else
  imp_of = DEFAULT_IMP_INDEX;

 if (!of_property_read_u32(combophy->dev->of_node, "st,output-vswing-microvolt", &val)) {
  max_vswing = imp_lookup[imp_of].vswing[vswing_size - 1];

  if (val < min_vswing || val > max_vswing) {
   dev_err(combophy->dev, "Invalid value %u for output vswing\n", val);
   return -EINVAL;
  }

  for (vswing_of = 0; vswing_of < ARRAY_SIZE(imp_lookup[imp_of].vswing); vswing_of++)
   if (imp_lookup[imp_of].vswing[vswing_of] >= val)
    break;

  if (WARN_ON(vswing_of == ARRAY_SIZE(imp_lookup[imp_of].vswing)))
   return -EINVAL;

  dev_dbg(combophy->dev, "Set %u microvolt swing\n",
    imp_lookup[imp_of].vswing[vswing_of]);

  regmap_update_bits(combophy->regmap, SYSCFG_PCIEPRGCR,
       STM32MP25_PCIEPRG_IMPCTRL_VSWING,
       FIELD_PREP(STM32MP25_PCIEPRG_IMPCTRL_VSWING, vswing_of));
 }

 return 0;
}

static int stm32_combophy_pll_init(struct stm32_combophy *combophy)
{
 int ret;
 u32 refclksel, pllmult, propcntrl, val;
 u32 clk_rate;
 struct clk *clk;
 u32 cr1_val = 0, cr1_mask = 0;

 if (combophy->have_pad_clk)
  clk = combophy->clks[PAD_CLK].clk;
 else
  clk = combophy->clks[KER_CLK].clk;

 clk_rate = clk_get_rate(clk);

 dev_dbg(combophy->dev, "%s pll init rate %d\n",
  combophy->have_pad_clk ? "External" : "Ker", clk_rate);

 if (combophy->type != PHY_TYPE_PCIE) {
  cr1_mask |= SYSCFG_COMBOPHY_CR1_REFSSPEN;
  cr1_val |= SYSCFG_COMBOPHY_CR1_REFSSPEN;
 }

 if (of_property_present(combophy->dev->of_node, "st,ssc-on")) {
  dev_dbg(combophy->dev, "Enabling clock with SSC\n");
  cr1_mask |= SYSCFG_COMBOPHY_CR1_SSCEN;
  cr1_val |= SYSCFG_COMBOPHY_CR1_SSCEN;
 }

 switch (clk_rate) {
 case 100000000:
  pllmult = MPLLMULT_100;
  refclksel = REFCLKSEL_0;
  propcntrl = 0x8u << 4;
  break;
 case 19200000:
  pllmult = MPLLMULT_19_2;
  refclksel = REFCLKSEL_1;
  propcntrl = 0x8u << 4;
  break;
 case 25000000:
  pllmult = MPLLMULT_25;
  refclksel = REFCLKSEL_0;
  propcntrl = 0xeu << 4;
  break;
 case 24000000:
  pllmult = MPLLMULT_24;
  refclksel = REFCLKSEL_1;
  propcntrl = 0xeu << 4;
  break;
 case 20000000:
  pllmult = MPLLMULT_20;
  refclksel = REFCLKSEL_0;
  propcntrl = 0xeu << 4;
  break;
 default:
  dev_err(combophy->dev, "Invalid rate 0x%x\n", clk_rate);
  return -EINVAL;
 }

 cr1_mask |= SYSCFG_COMBOPHY_CR1_REFCLKDIV2;
 cr1_val |= REFCLDIV_0;

 cr1_mask |= SYSCFG_COMBOPHY_CR1_REFCLKSEL;
 cr1_val |= refclksel;

 cr1_mask |= SYSCFG_COMBOPHY_CR1_MPLLMULT;
 cr1_val |= pllmult;

 /*
  * vddcombophy is interconnected with vddcore. Isolation bit should be unset
  * before using the ComboPHY.
 */
 regmap_update_bits(combophy->regmap, SYSCFG_COMBOPHY_CR2,
      SYSCFG_COMBOPHY_CR2_ISO_DIS, SYSCFG_COMBOPHY_CR2_ISO_DIS);

 reset_control_assert(combophy->phy_reset);

 if (combophy->type == PHY_TYPE_PCIE) {
  ret = stm32_impedance_tune(combophy);
  if (ret)
   goto out_iso;

  cr1_mask |= SYSCFG_COMBOPHY_CR1_REFUSEPAD;
  cr1_val |= combophy->have_pad_clk ? SYSCFG_COMBOPHY_CR1_REFUSEPAD : 0;
 }

 if (!of_property_read_u32(combophy->dev->of_node, "st,rx-equalizer", &val)) {
  dev_dbg(combophy->dev, "Set RX equalizer %u\n", val);
  if (val > SYSCFG_COMBOPHY_CR4_RX0_EQ) {
   dev_err(combophy->dev, "Invalid value %u for rx0 equalizer\n", val);
   ret = -EINVAL;
   goto out_iso;
  }

  regmap_update_bits(combophy->regmap, SYSCFG_COMBOPHY_CR4,
      SYSCFG_COMBOPHY_CR4_RX0_EQ, val);
 }

 regmap_update_bits(combophy->regmap, SYSCFG_COMBOPHY_CR1, cr1_mask, cr1_val);

 /*
  * Force elasticity buffer to be tuned for the reference clock as
  * the separated clock model is not supported
 */
 regmap_update_bits(combophy->regmap, SYSCFG_COMBOPHY_CR5,
      SYSCFG_COMBOPHY_CR5_COMMON_CLOCKS, SYSCFG_COMBOPHY_CR5_COMMON_CLOCKS);

 reset_control_deassert(combophy->phy_reset);

 ret = regmap_read_poll_timeout(combophy->regmap, SYSCFG_COMBOPHY_SR, val,
           !(val & STM32MP25_PIPE0_PHYSTATUS),
           10, 1000);
 if (ret) {
  dev_err(combophy->dev, "timeout, cannot lock PLL\n");
  if (combophy->type == PHY_TYPE_PCIE && !combophy->have_pad_clk)
   regmap_update_bits(combophy->regmap, SYSCFG_PCIEPRGCR,
        STM32MP25_PCIEPRGCR_EN, 0);

  if (combophy->type != PHY_TYPE_PCIE)
   regmap_update_bits(combophy->regmap, SYSCFG_COMBOPHY_CR1,
        SYSCFG_COMBOPHY_CR1_REFSSPEN, 0);

  goto out;
 }


 if (combophy->type == PHY_TYPE_PCIE) {
  if (!combophy->have_pad_clk)
   regmap_update_bits(combophy->regmap, SYSCFG_PCIEPRGCR,
        STM32MP25_PCIEPRGCR_EN, STM32MP25_PCIEPRGCR_EN);

  val = readl_relaxed(combophy->base + COMBOPHY_SUP_ANA_MPLL_LOOP_CTL);
  val &= ~COMBOPHY_PROP_CNTRL;
  val |= propcntrl;
  writel_relaxed(val, combophy->base + COMBOPHY_SUP_ANA_MPLL_LOOP_CTL);
 }

 return 0;

out_iso:
 reset_control_deassert(combophy->phy_reset);

out:
 regmap_update_bits(combophy->regmap, SYSCFG_COMBOPHY_CR2,
      SYSCFG_COMBOPHY_CR2_ISO_DIS, 0);

 return ret;
}

static struct phy *stm32_combophy_xlate(struct device *dev,
     const struct of_phandle_args *args)
{
 struct stm32_combophy *combophy = dev_get_drvdata(dev);
 unsigned int type;

 if (args->args_count != 1) {
  dev_err(dev, "invalid number of cells in 'phy' property\n");
  return ERR_PTR(-EINVAL);
 }

 type = args->args[0];
 if (type != PHY_TYPE_USB3 && type != PHY_TYPE_PCIE) {
  dev_err(dev, "unsupported device type: %d\n", type);
  return ERR_PTR(-EINVAL);
 }

 if (combophy->have_pad_clk && type != PHY_TYPE_PCIE) {
  dev_err(dev, "Invalid use of clk_pad for USB3 mode\n");
  return ERR_PTR(-EINVAL);
 }

 combophy->type = type;

 return combophy->phy;
}

static int stm32_combophy_set_mode(struct stm32_combophy *combophy)
{
 int type = combophy->type;
 u32 val;

 switch (type) {
 case PHY_TYPE_PCIE:
  dev_dbg(combophy->dev, "setting PCIe ComboPHY\n");
  val = COMBOPHY_MODESEL_PCIE;
  break;
 case PHY_TYPE_USB3:
  dev_dbg(combophy->dev, "setting USB3 ComboPHY\n");
  val = COMBOPHY_MODESEL_USB;
  break;
 default:
  dev_err(combophy->dev, "Invalid PHY mode %d\n", type);
  return -EINVAL;
 }

 return regmap_update_bits(combophy->regmap, SYSCFG_COMBOPHY_CR2,
      SYSCFG_COMBOPHY_CR2_MODESEL, val);
}

static int stm32_combophy_suspend_noirq(struct device *dev)
{
 struct stm32_combophy *combophy = dev_get_drvdata(dev);

 /*
  * Clocks should be turned off since it is not needed for
  * wakeup capability. In case usb-remote wakeup is not enabled,
  * combo-phy is already turned off by HCD driver using exit callback
 */
 if (combophy->is_init) {
  clk_bulk_disable_unprepare(combophy->num_clks, combophy->clks);

  /* since wakeup is enabled for ctrl */
  enable_irq_wake(combophy->irq_wakeup);
 }

 return 0;
}

static int stm32_combophy_resume_noirq(struct device *dev)
{
 struct stm32_combophy *combophy = dev_get_drvdata(dev);
 int ret;

 /*
  * If clocks was turned off by suspend call for wakeup then needs
  * to be turned back ON in resume. In case usb-remote wakeup is not
  * enabled, clocks already turned ON by HCD driver using init callback
 */
 if (combophy->is_init) {
  /* since wakeup was enabled for ctrl */
  disable_irq_wake(combophy->irq_wakeup);

  ret = clk_bulk_prepare_enable(combophy->num_clks, combophy->clks);
  if (ret) {
   dev_err(dev, "can't enable clocks (%d)\n", ret);
   return ret;
  }
 }

 return 0;
}

static int stm32_combophy_exit(struct phy *phy)
{
 struct stm32_combophy *combophy = phy_get_drvdata(phy);
 struct device *dev = combophy->dev;

 combophy->is_init = false;

 if (combophy->type == PHY_TYPE_PCIE && !combophy->have_pad_clk)
  regmap_update_bits(combophy->regmap, SYSCFG_PCIEPRGCR,
       STM32MP25_PCIEPRGCR_EN, 0);

 if (combophy->type != PHY_TYPE_PCIE)
  regmap_update_bits(combophy->regmap, SYSCFG_COMBOPHY_CR1,
       SYSCFG_COMBOPHY_CR1_REFSSPEN, 0);

 regmap_update_bits(combophy->regmap, SYSCFG_COMBOPHY_CR2,
      SYSCFG_COMBOPHY_CR2_ISO_DIS, 0);

 clk_bulk_disable_unprepare(combophy->num_clks, combophy->clks);

 pm_runtime_put_noidle(dev);

 return 0;
}

static int stm32_combophy_init(struct phy *phy)
{
 struct stm32_combophy *combophy = phy_get_drvdata(phy);
 struct device *dev = combophy->dev;
 int ret;

 pm_runtime_get_noresume(dev);

 ret = clk_bulk_prepare_enable(combophy->num_clks, combophy->clks);
 if (ret) {
  dev_err(dev, "can't enable clocks (%d)\n", ret);
  pm_runtime_put_noidle(dev);
  return ret;
 }

 ret = stm32_combophy_set_mode(combophy);
 if (ret) {
  dev_err(dev, "combophy mode not set\n");
  clk_bulk_disable_unprepare(combophy->num_clks, combophy->clks);
  pm_runtime_put_noidle(dev);
  return ret;
 }

 ret = stm32_combophy_pll_init(combophy);
 if (ret) {
  clk_bulk_disable_unprepare(combophy->num_clks, combophy->clks);
  pm_runtime_put_noidle(dev);
  return ret;
 }

 pm_runtime_disable(dev);
 pm_runtime_set_active(dev);
 pm_runtime_enable(dev);

 combophy->is_init = true;

 return ret;
}

static const struct phy_ops stm32_combophy_phy_data = {
 .init = stm32_combophy_init,
 .exit = stm32_combophy_exit,
 .owner = THIS_MODULE
};

static irqreturn_t stm32_combophy_irq_wakeup_handler(int irq, void *dev_id)
{
 return IRQ_HANDLED;
}

static int stm32_combophy_get_clocks(struct stm32_combophy *combophy)
{
 int i, ret;

 for (i = 0; i < ARRAY_SIZE(combophy_clks); i++)
  combophy->clks[i].id = combophy_clks[i];

 combophy->num_clks = ARRAY_SIZE(combophy_clks) - 1;

 ret = devm_clk_bulk_get(combophy->dev, combophy->num_clks, combophy->clks);
 if (ret)
  return ret;

 ret = devm_clk_bulk_get_optional(combophy->dev, 1, combophy->clks + combophy->num_clks);
 if (ret)
  return ret;

 if (combophy->clks[combophy->num_clks].clk != NULL) {
  combophy->have_pad_clk = true;
  combophy->num_clks++;
 }

 return 0;
}

static int stm32_combophy_probe(struct platform_device *pdev)
{
 struct stm32_combophy *combophy;
 struct device *dev = &pdev->dev;
 struct phy_provider *phy_provider;
 int ret, irq;

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

 combophy->dev = dev;

 dev_set_drvdata(dev, combophy);

 combophy->base = devm_platform_ioremap_resource(pdev, 0);
 if (IS_ERR(combophy->base))
  return PTR_ERR(combophy->base);

 ret = stm32_combophy_get_clocks(combophy);
 if (ret)
  return ret;

 combophy->phy_reset = devm_reset_control_get_exclusive(dev, "phy");
 if (IS_ERR(combophy->phy_reset))
  return dev_err_probe(dev, PTR_ERR(combophy->phy_reset),
         "Failed to get PHY reset\n");

 combophy->regmap = syscon_regmap_lookup_by_compatible("st,stm32mp25-syscfg");
 if (IS_ERR(combophy->regmap))
  return dev_err_probe(dev, PTR_ERR(combophy->regmap),
         "No syscfg specified\n");

 combophy->phy = devm_phy_create(dev, NULL, &stm32_combophy_phy_data);
 if (IS_ERR(combophy->phy))
  return dev_err_probe(dev, PTR_ERR(combophy->phy),
         "failed to create PCIe/USB3 ComboPHY\n");

 if (device_property_read_bool(dev, "wakeup-source")) {
  irq = platform_get_irq(pdev, 0);
  if (irq < 0)
   return dev_err_probe(dev, irq, "failed to get IRQ\n");
  combophy->irq_wakeup = irq;

  ret = devm_request_threaded_irq(dev, combophy->irq_wakeup, NULL,
      stm32_combophy_irq_wakeup_handler, IRQF_ONESHOT,
      NULL, NULL);
  if (ret)
   return dev_err_probe(dev, ret, "unable to request wake IRQ %d\n",
       combophy->irq_wakeup);
 }

 ret = devm_pm_runtime_enable(dev);
 if (ret)
  return dev_err_probe(dev, ret, "Failed to enable pm runtime\n");

 phy_set_drvdata(combophy->phy, combophy);

 phy_provider = devm_of_phy_provider_register(dev, stm32_combophy_xlate);

 return PTR_ERR_OR_ZERO(phy_provider);
}

static const struct dev_pm_ops stm32_combophy_pm_ops = {
 NOIRQ_SYSTEM_SLEEP_PM_OPS(stm32_combophy_suspend_noirq,
      stm32_combophy_resume_noirq)
};

static const struct of_device_id stm32_combophy_of_match[] = {
 { .compatible = "st,stm32mp25-combophy", },
 { },
};
MODULE_DEVICE_TABLE(of, stm32_combophy_of_match);

static struct platform_driver stm32_combophy_driver = {
 .probe = stm32_combophy_probe,
 .driver = {
     .name = "stm32-combophy",
     .of_match_table = stm32_combophy_of_match,
     .pm = pm_sleep_ptr(&stm32_combophy_pm_ops)
 }
};

module_platform_driver(stm32_combophy_driver);

MODULE_AUTHOR("Christian Bruel <christian.bruel@foss.st.com>");
MODULE_DESCRIPTION("STM32MP25 Combophy USB3/PCIe controller driver");
MODULE_LICENSE("GPL");