// SPDX-License-Identifier: GPL-2.0+
//
// soc-ac97.c  --  ALSA SoC Audio Layer AC97 support
//
// Copyright 2005 Wolfson Microelectronics PLC.
// Copyright 2005 Openedhand Ltd.
// Copyright (C) 2010 Slimlogic Ltd.
// Copyright (C) 2010 Texas Instruments Inc.
//
// Author: Liam Girdwood <lrg@slimlogic.co.uk>
//         with code, comments and ideas from :-
//         Richard Purdie <richard@openedhand.com>

#include <linux/ctype.h>
#include <linux/delay.h>
#include <linux/export.h>
#include <linux/gpio/consumer.h>
#include <linux/gpio/driver.h>
#include <linux/init.h>
#include <linux/of.h>
#include <linux/pinctrl/consumer.h>
#include <linux/slab.h>
#include <sound/ac97_codec.h>
#include <sound/soc.h>

struct snd_ac97_reset_cfg {
 struct pinctrl *pctl;
 struct pinctrl_state *pstate_reset;
 struct pinctrl_state *pstate_warm_reset;
 struct pinctrl_state *pstate_run;
 struct gpio_desc *reset_gpio;
 struct gpio_desc *sdata_gpio;
 struct gpio_desc *sync_gpio;
};

static struct snd_ac97_bus soc_ac97_bus = {
 .ops = NULL, /* Gets initialized in snd_soc_set_ac97_ops() */
};

static void soc_ac97_device_release(struct device *dev)
{
 kfree(to_ac97_t(dev));
}

#ifdef CONFIG_GPIOLIB
struct snd_ac97_gpio_priv {
 struct gpio_chip gpio_chip;
 unsigned int gpios_set;
 struct snd_soc_component *component;
};

static inline struct snd_soc_component *gpio_to_component(struct gpio_chip *chip)
{
 struct snd_ac97_gpio_priv *gpio_priv = gpiochip_get_data(chip);

 return gpio_priv->component;
}

static int snd_soc_ac97_gpio_request(struct gpio_chip *chip, unsigned int offset)
{
 if (offset >= AC97_NUM_GPIOS)
  return -EINVAL;

 return 0;
}

static int snd_soc_ac97_gpio_direction_in(struct gpio_chip *chip,
       unsigned int offset)
{
 struct snd_soc_component *component = gpio_to_component(chip);

 dev_dbg(component->dev, "set gpio %d to output\n", offset);
 return snd_soc_component_update_bits(component, AC97_GPIO_CFG,
       1 << offset, 1 << offset);
}

static int snd_soc_ac97_gpio_get(struct gpio_chip *chip, unsigned int offset)
{
 struct snd_soc_component *component = gpio_to_component(chip);
 int ret;

 ret = snd_soc_component_read(component, AC97_GPIO_STATUS);

 dev_dbg(component->dev, "get gpio %d : %d\n", offset,
  ret & (1 << offset));

 return !!(ret & (1 << offset));
}

static int snd_soc_ac97_gpio_set(struct gpio_chip *chip, unsigned int offset,
     int value)
{
 struct snd_ac97_gpio_priv *gpio_priv = gpiochip_get_data(chip);
 struct snd_soc_component *component = gpio_to_component(chip);

 gpio_priv->gpios_set &= ~(1 << offset);
 gpio_priv->gpios_set |= (!!value) << offset;
 snd_soc_component_write(component, AC97_GPIO_STATUS,
    gpio_priv->gpios_set);
 dev_dbg(component->dev, "set gpio %d to %d\n", offset, !!value);

 return 0;
}

static int snd_soc_ac97_gpio_direction_out(struct gpio_chip *chip,
         unsigned offset, int value)
{
 struct snd_soc_component *component = gpio_to_component(chip);
 int ret;

 dev_dbg(component->dev, "set gpio %d to output\n", offset);

 ret = snd_soc_ac97_gpio_set(chip, offset, value);
 if (ret)
  return ret;

 return snd_soc_component_update_bits(component, AC97_GPIO_CFG,
          1 << offset, 0);
}

static const struct gpio_chip snd_soc_ac97_gpio_chip = {
 .label   = "snd_soc_ac97",
 .owner   = THIS_MODULE,
 .request  = snd_soc_ac97_gpio_request,
 .direction_input = snd_soc_ac97_gpio_direction_in,
 .get   = snd_soc_ac97_gpio_get,
 .direction_output = snd_soc_ac97_gpio_direction_out,
 .set   = snd_soc_ac97_gpio_set,
 .can_sleep  = 1,
};

static int snd_soc_ac97_init_gpio(struct snd_ac97 *ac97,
      struct snd_soc_component *component)
{
 struct snd_ac97_gpio_priv *gpio_priv;
 int ret;

 gpio_priv = devm_kzalloc(component->dev, sizeof(*gpio_priv), GFP_KERNEL);
 if (!gpio_priv)
  return -ENOMEM;
 ac97->gpio_priv = gpio_priv;
 gpio_priv->component = component;
 gpio_priv->gpio_chip = snd_soc_ac97_gpio_chip;
 gpio_priv->gpio_chip.ngpio = AC97_NUM_GPIOS;
 gpio_priv->gpio_chip.parent = component->dev;
 gpio_priv->gpio_chip.base = -1;

 ret = gpiochip_add_data(&gpio_priv->gpio_chip, gpio_priv);
 if (ret != 0)
  dev_err(component->dev, "Failed to add GPIOs: %d\n", ret);
 return ret;
}

static void snd_soc_ac97_free_gpio(struct snd_ac97 *ac97)
{
 gpiochip_remove(&ac97->gpio_priv->gpio_chip);
}
#else
static int snd_soc_ac97_init_gpio(struct snd_ac97 *ac97,
      struct snd_soc_component *component)
{
 return 0;
}

static void snd_soc_ac97_free_gpio(struct snd_ac97 *ac97)
{
}
#endif

/**
 * snd_soc_alloc_ac97_component() - Allocate new a AC'97 device
 * @component: The COMPONENT for which to create the AC'97 device
 *
 * Allocated a new snd_ac97 device and intializes it, but does not yet register
 * it. The caller is responsible to either call device_add(&ac97->dev) to
 * register the device, or to call put_device(&ac97->dev) to free the device.
 *
 * Returns: A snd_ac97 device or an ERR_PTR in case of an error.
 */
struct snd_ac97 *snd_soc_alloc_ac97_component(struct snd_soc_component *component)
{
 struct snd_ac97 *ac97;

 ac97 = kzalloc(sizeof(struct snd_ac97), GFP_KERNEL);
 if (ac97 == NULL)
  return ERR_PTR(-ENOMEM);

 ac97->bus = &soc_ac97_bus;
 ac97->num = 0;

 ac97->dev.bus = &ac97_bus_type;
 ac97->dev.parent = component->card->dev;
 ac97->dev.release = soc_ac97_device_release;

 dev_set_name(&ac97->dev, "%d-%d:%s",
       component->card->snd_card->number, 0,
       component->name);

 device_initialize(&ac97->dev);

 return ac97;
}
EXPORT_SYMBOL(snd_soc_alloc_ac97_component);

/**
 * snd_soc_new_ac97_component - initailise AC97 device
 * @component: audio component
 * @id: The expected device ID
 * @id_mask: Mask that is applied to the device ID before comparing with @id
 *
 * Initialises AC97 component resources for use by ad-hoc devices only.
 *
 * If @id is not 0 this function will reset the device, then read the ID from
 * the device and check if it matches the expected ID. If it doesn't match an
 * error will be returned and device will not be registered.
 *
 * Returns: An ERR_PTR on failure or a valid snd_ac97 struct on success.
 */
struct snd_ac97 *snd_soc_new_ac97_component(struct snd_soc_component *component,
 unsigned int id, unsigned int id_mask)
{
 struct snd_ac97 *ac97;
 int ret;

 ac97 = snd_soc_alloc_ac97_component(component);
 if (IS_ERR(ac97))
  return ac97;

 if (id) {
  ret = snd_ac97_reset(ac97, false, id, id_mask);
  if (ret < 0) {
   dev_err(component->dev, "Failed to reset AC97 device: %d\n",
    ret);
   goto err_put_device;
  }
 }

 ret = device_add(&ac97->dev);
 if (ret)
  goto err_put_device;

 ret = snd_soc_ac97_init_gpio(ac97, component);
 if (ret)
  goto err_put_device;

 return ac97;

err_put_device:
 put_device(&ac97->dev);
 return ERR_PTR(ret);
}
EXPORT_SYMBOL_GPL(snd_soc_new_ac97_component);

/**
 * snd_soc_free_ac97_component - free AC97 component device
 * @ac97: snd_ac97 device to be freed
 *
 * Frees AC97 component device resources.
 */
void snd_soc_free_ac97_component(struct snd_ac97 *ac97)
{
 snd_soc_ac97_free_gpio(ac97);
 device_del(&ac97->dev);
 ac97->bus = NULL;
 put_device(&ac97->dev);
}
EXPORT_SYMBOL_GPL(snd_soc_free_ac97_component);

static struct snd_ac97_reset_cfg snd_ac97_rst_cfg;

static void snd_soc_ac97_warm_reset(struct snd_ac97 *ac97)
{
 struct pinctrl *pctl = snd_ac97_rst_cfg.pctl;

 pinctrl_select_state(pctl, snd_ac97_rst_cfg.pstate_warm_reset);

 gpiod_direction_output_raw(snd_ac97_rst_cfg.sync_gpio, 1);

 udelay(10);

 gpiod_direction_output_raw(snd_ac97_rst_cfg.sync_gpio, 0);

 pinctrl_select_state(pctl, snd_ac97_rst_cfg.pstate_run);
 msleep(2);
}

static void snd_soc_ac97_reset(struct snd_ac97 *ac97)
{
 struct pinctrl *pctl = snd_ac97_rst_cfg.pctl;

 pinctrl_select_state(pctl, snd_ac97_rst_cfg.pstate_reset);

 gpiod_direction_output_raw(snd_ac97_rst_cfg.sync_gpio, 0);
 gpiod_direction_output_raw(snd_ac97_rst_cfg.sdata_gpio, 0);
 gpiod_direction_output_raw(snd_ac97_rst_cfg.reset_gpio, 0);

 udelay(10);

 gpiod_direction_output_raw(snd_ac97_rst_cfg.reset_gpio, 1);

 pinctrl_select_state(pctl, snd_ac97_rst_cfg.pstate_run);
 msleep(2);
}

static int snd_soc_ac97_parse_pinctl(struct device *dev,
  struct snd_ac97_reset_cfg *cfg)
{
 struct pinctrl *p;
 struct pinctrl_state *state;

 p = devm_pinctrl_get(dev);
 if (IS_ERR(p)) {
  dev_err(dev, "Failed to get pinctrl\n");
  return PTR_ERR(p);
 }
 cfg->pctl = p;

 state = pinctrl_lookup_state(p, "ac97-reset");
 if (IS_ERR(state)) {
  dev_err(dev, "Can't find pinctrl state ac97-reset\n");
  return PTR_ERR(state);
 }
 cfg->pstate_reset = state;

 state = pinctrl_lookup_state(p, "ac97-warm-reset");
 if (IS_ERR(state)) {
  dev_err(dev, "Can't find pinctrl state ac97-warm-reset\n");
  return PTR_ERR(state);
 }
 cfg->pstate_warm_reset = state;

 state = pinctrl_lookup_state(p, "ac97-running");
 if (IS_ERR(state)) {
  dev_err(dev, "Can't find pinctrl state ac97-running\n");
  return PTR_ERR(state);
 }
 cfg->pstate_run = state;

 cfg->sync_gpio = devm_gpiod_get_index(dev, "ac97", 0, GPIOD_ASIS);
 if (IS_ERR(cfg->sync_gpio))
  return dev_err_probe(dev, PTR_ERR(cfg->sync_gpio), "Can't find ac97-sync gpio\n");
 gpiod_set_consumer_name(cfg->sync_gpio, "AC97 link sync");

 cfg->sdata_gpio = devm_gpiod_get_index(dev, "ac97", 1, GPIOD_ASIS);
 if (IS_ERR(cfg->sdata_gpio))
  return dev_err_probe(dev, PTR_ERR(cfg->sdata_gpio), "Can't find ac97-sdata gpio\n");
 gpiod_set_consumer_name(cfg->sdata_gpio, "AC97 link sdata");

 cfg->reset_gpio = devm_gpiod_get_index(dev, "ac97", 2, GPIOD_ASIS);
 if (IS_ERR(cfg->reset_gpio))
  return dev_err_probe(dev, PTR_ERR(cfg->reset_gpio), "Can't find ac97-reset gpio\n");
 gpiod_set_consumer_name(cfg->reset_gpio, "AC97 link reset");

 return 0;
}

struct snd_ac97_bus_ops *soc_ac97_ops;
EXPORT_SYMBOL_GPL(soc_ac97_ops);

int snd_soc_set_ac97_ops(struct snd_ac97_bus_ops *ops)
{
 if (ops == soc_ac97_ops)
  return 0;

 if (soc_ac97_ops && ops)
  return -EBUSY;

 soc_ac97_ops = ops;
 soc_ac97_bus.ops = ops;

 return 0;
}
EXPORT_SYMBOL_GPL(snd_soc_set_ac97_ops);

/**
 * snd_soc_set_ac97_ops_of_reset - Set ac97 ops with generic ac97 reset functions
 * @ops: bus ops
 * @pdev: platform device
 *
 * This function sets the reset and warm_reset properties of ops and parses
 * the device node of pdev to get pinctrl states and gpio numbers to use.
 */
int snd_soc_set_ac97_ops_of_reset(struct snd_ac97_bus_ops *ops,
  struct platform_device *pdev)
{
 struct device *dev = &pdev->dev;
 struct snd_ac97_reset_cfg cfg;
 int ret;

 ret = snd_soc_ac97_parse_pinctl(dev, &cfg);
 if (ret)
  return ret;

 ret = snd_soc_set_ac97_ops(ops);
 if (ret)
  return ret;

 ops->warm_reset = snd_soc_ac97_warm_reset;
 ops->reset = snd_soc_ac97_reset;

 snd_ac97_rst_cfg = cfg;
 return 0;
}
EXPORT_SYMBOL_GPL(snd_soc_set_ac97_ops_of_reset);