Quelle  pinctrl-loongson2.c

// SPDX-License-Identifier: GPL-2.0+
/*
 * Author: zhanghongchen <zhanghongchen@loongson.cn>
 *         Yinbo Zhu <zhuyinbo@loongson.cn>
 * Copyright (C) 2022-2023 Loongson Technology Corporation Limited
 */

#include <linux/init.h>
#include <linux/module.h>
#include <linux/platform_device.h>
#include <linux/mod_devicetable.h>
#include <linux/pinctrl/pinmux.h>
#include <linux/pinctrl/pinconf-generic.h>
#include <linux/pinctrl/pinctrl.h>
#include <linux/bitops.h>
#include <linux/io.h>
#include <linux/seq_file.h>

#include "core.h"
#include "pinctrl-utils.h"

#define PMX_GROUP(name, offset, bitv)     \
 {        \
  .grp = PINCTRL_PINGROUP((#name), (name ## _pins), \
    ARRAY_SIZE((name ## _pins))),  \
  .reg = offset,      \
  .bit = bitv,      \
 }

#define SPECIFIC_GROUP(group)      \
 static const char * const group##_groups[] = {   \
  #group       \
 }

#define FUNCTION(fn)       \
 {        \
  .name = #fn,      \
  .groups = fn ## _groups,    \
  .num_groups = ARRAY_SIZE(fn ## _groups),  \
 }

struct loongson2_pinctrl {
 struct device *dev;
 struct pinctrl_dev *pcdev;
 struct pinctrl_desc desc;
 struct device_node *of_node;
 spinlock_t lock;
 void __iomem *reg_base;
};

struct loongson2_pmx_group {
 struct pingroup grp;
 unsigned int reg;
 unsigned int bit;
};

struct loongson2_pmx_func {
 const char *name;
 const char * const *groups;
 unsigned int num_groups;
};

#define LOONGSON2_PIN(x) PINCTRL_PIN(x, "gpio"#x)
static const struct pinctrl_pin_desc loongson2_pctrl_pins[] = {
 LOONGSON2_PIN(0),  LOONGSON2_PIN(1),  LOONGSON2_PIN(2),  LOONGSON2_PIN(3),
 LOONGSON2_PIN(4),  LOONGSON2_PIN(5),  LOONGSON2_PIN(6),  LOONGSON2_PIN(7),
 LOONGSON2_PIN(8),  LOONGSON2_PIN(9),  LOONGSON2_PIN(10), LOONGSON2_PIN(11),
 LOONGSON2_PIN(12), LOONGSON2_PIN(13), LOONGSON2_PIN(14),
 LOONGSON2_PIN(16), LOONGSON2_PIN(17), LOONGSON2_PIN(18), LOONGSON2_PIN(19),
 LOONGSON2_PIN(20), LOONGSON2_PIN(21), LOONGSON2_PIN(22), LOONGSON2_PIN(23),
 LOONGSON2_PIN(24), LOONGSON2_PIN(25), LOONGSON2_PIN(26), LOONGSON2_PIN(27),
 LOONGSON2_PIN(28), LOONGSON2_PIN(29), LOONGSON2_PIN(30),
 LOONGSON2_PIN(32), LOONGSON2_PIN(33), LOONGSON2_PIN(34), LOONGSON2_PIN(35),
 LOONGSON2_PIN(36), LOONGSON2_PIN(37), LOONGSON2_PIN(38), LOONGSON2_PIN(39),
 LOONGSON2_PIN(40), LOONGSON2_PIN(41),
 LOONGSON2_PIN(44), LOONGSON2_PIN(45), LOONGSON2_PIN(46), LOONGSON2_PIN(47),
 LOONGSON2_PIN(48), LOONGSON2_PIN(49), LOONGSON2_PIN(50), LOONGSON2_PIN(51),
 LOONGSON2_PIN(52), LOONGSON2_PIN(53), LOONGSON2_PIN(54), LOONGSON2_PIN(55),
 LOONGSON2_PIN(56), LOONGSON2_PIN(57), LOONGSON2_PIN(58), LOONGSON2_PIN(59),
 LOONGSON2_PIN(60), LOONGSON2_PIN(61), LOONGSON2_PIN(62), LOONGSON2_PIN(63),
};

static const unsigned int gpio_pins[] = {0, 1, 2, 3, 4, 5, 6, 7,
      8, 9, 10, 11, 12, 13, 14,
      16, 17, 18, 19, 20, 21, 22, 23,
      24, 25, 26, 27, 28, 29, 30,
      32, 33, 34, 35, 36, 37, 38, 39,
      40,         43, 44, 45, 46, 47,
      48, 49, 50, 51, 52, 53, 46, 55,
      56, 57, 58, 59, 60, 61, 62, 63};
static const unsigned int sdio_pins[] = {36, 37, 38, 39, 40, 41};
static const unsigned int can1_pins[] = {34, 35};
static const unsigned int can0_pins[] = {32, 33};
static const unsigned int pwm3_pins[] = {23};
static const unsigned int pwm2_pins[] = {22};
static const unsigned int pwm1_pins[] = {21};
static const unsigned int pwm0_pins[] = {20};
static const unsigned int i2c1_pins[] = {18, 19};
static const unsigned int i2c0_pins[] = {16, 17};
static const unsigned int nand_pins[] = {44, 45, 46, 47, 48, 49, 50, 51,
      52, 53, 54, 55, 56, 57, 58, 59, 60,
      61, 62, 63};
static const unsigned int sata_led_pins[] = {14};
static const unsigned int i2s_pins[]    = {24, 25, 26, 27, 28};
static const unsigned int hda_pins[]    = {24, 25, 26, 27, 28, 29, 30};

static struct loongson2_pmx_group loongson2_pmx_groups[] = {
 PMX_GROUP(gpio, 0x0, 64),
 PMX_GROUP(sdio, 0x0, 20),
 PMX_GROUP(can1, 0x0, 17),
 PMX_GROUP(can0, 0x0, 16),
 PMX_GROUP(pwm3, 0x0, 15),
 PMX_GROUP(pwm2, 0x0, 14),
 PMX_GROUP(pwm1, 0x0, 13),
 PMX_GROUP(pwm0, 0x0, 12),
 PMX_GROUP(i2c1, 0x0, 11),
 PMX_GROUP(i2c0, 0x0, 10),
 PMX_GROUP(nand, 0x0, 9),
 PMX_GROUP(sata_led, 0x0, 8),
 PMX_GROUP(i2s, 0x0, 6),
 PMX_GROUP(hda, 0x0, 4),
};

SPECIFIC_GROUP(sdio);
SPECIFIC_GROUP(can1);
SPECIFIC_GROUP(can0);
SPECIFIC_GROUP(pwm3);
SPECIFIC_GROUP(pwm2);
SPECIFIC_GROUP(pwm1);
SPECIFIC_GROUP(pwm0);
SPECIFIC_GROUP(i2c1);
SPECIFIC_GROUP(i2c0);
SPECIFIC_GROUP(nand);
SPECIFIC_GROUP(sata_led);
SPECIFIC_GROUP(i2s);
SPECIFIC_GROUP(hda);

static const char * const gpio_groups[] = {
 "sdio",
 "can1", "can0",
 "pwm3", "pwm2", "pwm1", "pwm0",
 "i2c1", "i2c0",
 "nand",
 "sata_led",
 "i2s",
 "hda",
};

static const struct loongson2_pmx_func loongson2_pmx_functions[] = {
 FUNCTION(gpio),
 FUNCTION(sdio),
 FUNCTION(can1),
 FUNCTION(can0),
 FUNCTION(pwm3),
 FUNCTION(pwm2),
 FUNCTION(pwm1),
 FUNCTION(pwm0),
 FUNCTION(i2c1),
 FUNCTION(i2c0),
 FUNCTION(nand),
 FUNCTION(sata_led),
 FUNCTION(i2s),
 FUNCTION(hda),
};

static int loongson2_get_groups_count(struct pinctrl_dev *pcdev)
{
 return ARRAY_SIZE(loongson2_pmx_groups);
}

static const char *loongson2_get_group_name(struct pinctrl_dev *pcdev,
     unsigned int selector)
{
 return loongson2_pmx_groups[selector].grp.name;
}

static int loongson2_get_group_pins(struct pinctrl_dev *pcdev, unsigned int selector,
   const unsigned int **pins, unsigned int *num_pins)
{
 *pins = loongson2_pmx_groups[selector].grp.pins;
 *num_pins = loongson2_pmx_groups[selector].grp.npins;

 return 0;
}

static void loongson2_pin_dbg_show(struct pinctrl_dev *pcdev, struct seq_file *s,
          unsigned int offset)
{
 seq_printf(s, " %s", dev_name(pcdev->dev));
}

static const struct pinctrl_ops loongson2_pctrl_ops = {
 .get_groups_count = loongson2_get_groups_count,
 .get_group_name  = loongson2_get_group_name,
 .get_group_pins  = loongson2_get_group_pins,
 .dt_node_to_map  = pinconf_generic_dt_node_to_map_all,
 .dt_free_map  = pinctrl_utils_free_map,
 .pin_dbg_show  = loongson2_pin_dbg_show,
};

static int loongson2_pmx_set_mux(struct pinctrl_dev *pcdev, unsigned int func_num,
         unsigned int group_num)
{
 struct loongson2_pinctrl *pctrl = pinctrl_dev_get_drvdata(pcdev);
 void __iomem *reg = pctrl->reg_base +
    loongson2_pmx_groups[group_num].reg;
 unsigned int mux_bit = loongson2_pmx_groups[group_num].bit;
 unsigned int val;
 unsigned long flags;

 spin_lock_irqsave(&pctrl->lock, flags);
 val = readl(reg);
 if (func_num == 0)
  val &= ~BIT(mux_bit);
 else
  val |= BIT(mux_bit);
 writel(val, reg);
 spin_unlock_irqrestore(&pctrl->lock, flags);

 return 0;
}

static int loongson2_pmx_get_funcs_count(struct pinctrl_dev *pcdev)
{
 return ARRAY_SIZE(loongson2_pmx_functions);
}

static const char *loongson2_pmx_get_func_name(struct pinctrl_dev *pcdev,
        unsigned int selector)
{
 return loongson2_pmx_functions[selector].name;
}

static int loongson2_pmx_get_groups(struct pinctrl_dev *pcdev,
    unsigned int selector,
    const char * const **groups,
    unsigned int * const num_groups)
{
 *groups = loongson2_pmx_functions[selector].groups;
 *num_groups = loongson2_pmx_functions[selector].num_groups;

 return 0;
}

static const struct pinmux_ops loongson2_pmx_ops = {
 .set_mux = loongson2_pmx_set_mux,
 .get_functions_count = loongson2_pmx_get_funcs_count,
 .get_function_name = loongson2_pmx_get_func_name,
 .get_function_groups = loongson2_pmx_get_groups,
};

static int loongson2_pinctrl_probe(struct platform_device *pdev)
{
 struct device *dev = &pdev->dev;
 struct loongson2_pinctrl *pctrl;

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

 pctrl->reg_base = devm_platform_ioremap_resource(pdev, 0);
 if (IS_ERR(pctrl->reg_base))
  return PTR_ERR(pctrl->reg_base);

 spin_lock_init(&pctrl->lock);

 pctrl->dev = dev;
 pctrl->desc.name = "pinctrl-loongson2";
 pctrl->desc.owner = THIS_MODULE;
 pctrl->desc.pctlops = &loongson2_pctrl_ops;
 pctrl->desc.pmxops = &loongson2_pmx_ops;
 pctrl->desc.pins = loongson2_pctrl_pins;
 pctrl->desc.npins = ARRAY_SIZE(loongson2_pctrl_pins);

 pctrl->pcdev = devm_pinctrl_register(pctrl->dev, &pctrl->desc, pctrl);
 if (IS_ERR(pctrl->pcdev))
  return dev_err_probe(pctrl->dev, PTR_ERR(pctrl->pcdev),
         "can't register pinctrl device");

 return 0;
}

static const struct of_device_id loongson2_pinctrl_dt_match[] = {
 {
  .compatible = "loongson,ls2k-pinctrl",
 },
 { }
};
MODULE_DEVICE_TABLE(of, loongson2_pinctrl_dt_match);

static struct platform_driver loongson2_pinctrl_driver = {
 .probe  = loongson2_pinctrl_probe,
 .driver = {
  .name = "loongson2-pinctrl",
  .of_match_table = loongson2_pinctrl_dt_match,
 },
};

static int __init loongson2_pinctrl_init(void)
{
 return platform_driver_register(&loongson2_pinctrl_driver);
}
arch_initcall(loongson2_pinctrl_init);

static void __exit loongson2_pinctrl_exit(void)
{
 platform_driver_unregister(&loongson2_pinctrl_driver);
}
module_exit(loongson2_pinctrl_exit);

MODULE_DESCRIPTION("Loongson2 Pinctrl driver");
MODULE_LICENSE("GPL");