From e3e12f52b214121840b44c91de5b3e5af5d3eb84 Mon Sep 17 00:00:00 2001
From: hc <hc@nodka.com>
Date: Mon, 06 Nov 2023 03:04:41 +0000
Subject: [PATCH] rk3568 rt init

---
 kernel/sound/soc/rockchip/rockchip_multicodecs.c |  516 +++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 496 insertions(+), 20 deletions(-)

diff --git a/kernel/sound/soc/rockchip/rockchip_multicodecs.c b/kernel/sound/soc/rockchip/rockchip_multicodecs.c
index ee45076..8b1cb5f 100644
--- a/kernel/sound/soc/rockchip/rockchip_multicodecs.c
+++ b/kernel/sound/soc/rockchip/rockchip_multicodecs.c
@@ -19,9 +19,18 @@
  *
  */
 
+#include <linux/extcon-provider.h>
+#include <linux/gpio.h>
+#include <linux/iio/consumer.h>
+#include <linux/iio/iio.h>
+#include <linux/input.h>
+#include <linux/interrupt.h>
+#include <linux/irq.h>
 #include <linux/module.h>
+#include <linux/of_gpio.h>
 #include <linux/platform_device.h>
 #include <linux/slab.h>
+#include <linux/workqueue.h>
 #include <sound/core.h>
 #include <sound/jack.h>
 #include <sound/pcm.h>
@@ -29,39 +38,321 @@
 #include <sound/soc.h>
 #include <sound/soc-dapm.h>
 
-#include "../codecs/rk3308_codec_provider.h"
-
 #define DRV_NAME "rk-multicodecs"
-#define MAX_CODECS	2
 #define WAIT_CARDS	(SNDRV_CARDS - 1)
 #define DEFAULT_MCLK_FS	256
+
+struct adc_keys_button {
+	u32 voltage;
+	u32 keycode;
+};
+
+struct input_dev_poller {
+	void (*poll)(struct input_dev *dev);
+
+	unsigned int poll_interval_ms;
+	struct input_dev *input;
+	struct delayed_work work;
+};
 
 struct multicodecs_data {
 	struct snd_soc_card snd_card;
 	struct snd_soc_dai_link dai_link;
+	struct snd_soc_jack *jack_headset;
+	struct gpio_desc *hp_ctl_gpio;
+	struct gpio_desc *spk_ctl_gpio;
+	struct gpio_desc *hp_det_gpio;
+	struct iio_channel *adc;
+	struct extcon_dev *extcon;
+	struct delayed_work handler;
 	unsigned int mclk_fs;
 	bool codec_hp_det;
+	u32 num_keys;
+	u32 last_key;
+	u32 keyup_voltage;
+	const struct adc_keys_button *map;
+	struct input_dev *input;
+	struct input_dev_poller *poller;
 };
 
-static struct snd_soc_jack mc_hp_jack;
+static const unsigned int headset_extcon_cable[] = {
+	EXTCON_JACK_MICROPHONE,
+	EXTCON_JACK_HEADPHONE,
+	EXTCON_NONE,
+};
+
+static void mc_set_poll_interval(struct input_dev_poller *poller, unsigned int interval)
+{
+	if (poller)
+		poller->poll_interval_ms = interval;
+}
+
+static void mc_keys_poller_queue_work(struct input_dev_poller *poller)
+{
+	unsigned long delay;
+
+	delay = msecs_to_jiffies(poller->poll_interval_ms);
+	if (delay >= HZ)
+		delay = round_jiffies_relative(delay);
+
+	queue_delayed_work(system_freezable_wq, &poller->work, delay);
+}
+
+static void mc_keys_poller_work(struct work_struct *work)
+{
+	struct input_dev_poller *poller =
+		container_of(work, struct input_dev_poller, work.work);
+
+	poller->poll(poller->input);
+	mc_keys_poller_queue_work(poller);
+}
+
+static void mc_keys_poller_start(struct input_dev_poller *poller)
+{
+	if (poller->poll_interval_ms > 0) {
+		poller->poll(poller->input);
+		mc_keys_poller_queue_work(poller);
+	}
+}
+
+static void mc_keys_poller_stop(struct input_dev_poller *poller)
+{
+	cancel_delayed_work_sync(&poller->work);
+}
+
+static int mc_keys_setup_polling(struct multicodecs_data *mc_data,
+				 void (*poll_fn)(struct input_dev *dev))
+{
+	struct input_dev_poller *poller;
+
+	poller = devm_kzalloc(mc_data->snd_card.dev, sizeof(*poller), GFP_KERNEL);
+	if (!poller)
+		return -ENOMEM;
+
+	INIT_DELAYED_WORK(&poller->work, mc_keys_poller_work);
+	poller->input = mc_data->input;
+	poller->poll = poll_fn;
+	mc_data->poller = poller;
+
+	return 0;
+}
+
+static void mc_keys_poll(struct input_dev *input)
+{
+	struct multicodecs_data *mc_data = input_get_drvdata(input);
+	int i, value, ret;
+	u32 diff, closest = 0xffffffff;
+	int keycode = 0;
+
+	ret = iio_read_channel_processed(mc_data->adc, &value);
+	if (unlikely(ret < 0)) {
+		/* Forcibly release key if any was pressed */
+		value = mc_data->keyup_voltage;
+	} else {
+		for (i = 0; i < mc_data->num_keys; i++) {
+			diff = abs(mc_data->map[i].voltage - value);
+			if (diff < closest) {
+				closest = diff;
+				keycode = mc_data->map[i].keycode;
+			}
+		}
+	}
+
+	if (abs(mc_data->keyup_voltage - value) < closest)
+		keycode = 0;
+
+	if (mc_data->last_key && mc_data->last_key != keycode)
+		input_report_key(input, mc_data->last_key, 0);
+
+	if (keycode)
+		input_report_key(input, keycode, 1);
+
+	input_sync(input);
+	mc_data->last_key = keycode;
+}
+
+static int mc_keys_load_keymap(struct device *dev,
+			       struct multicodecs_data *mc_data)
+{
+	struct adc_keys_button *map;
+	struct fwnode_handle *child;
+	int i = 0;
+
+	mc_data->num_keys = device_get_child_node_count(dev);
+	if (mc_data->num_keys == 0) {
+		dev_err(dev, "keymap is missing\n");
+		return -EINVAL;
+	}
+
+	map = devm_kmalloc_array(dev, mc_data->num_keys, sizeof(*map), GFP_KERNEL);
+	if (!map)
+		return -ENOMEM;
+
+	device_for_each_child_node(dev, child) {
+		if (fwnode_property_read_u32(child, "press-threshold-microvolt",
+					     &map[i].voltage)) {
+			dev_err(dev, "Key with invalid or missing voltage\n");
+			fwnode_handle_put(child);
+			return -EINVAL;
+		}
+		map[i].voltage /= 1000;
+
+		if (fwnode_property_read_u32(child, "linux,code",
+					     &map[i].keycode)) {
+			dev_err(dev, "Key with invalid or missing linux,code\n");
+			fwnode_handle_put(child);
+			return -EINVAL;
+		}
+
+		i++;
+	}
+	mc_data->map = map;
+	return 0;
+}
+
+static void adc_jack_handler(struct work_struct *work)
+{
+	struct multicodecs_data *mc_data = container_of(to_delayed_work(work),
+						  struct multicodecs_data,
+						  handler);
+	struct snd_soc_jack *jack_headset = mc_data->jack_headset;
+	int adc, ret = 0;
+
+	if (!gpiod_get_value(mc_data->hp_det_gpio)) {
+		snd_soc_jack_report(jack_headset, 0, SND_JACK_HEADSET);
+		extcon_set_state_sync(mc_data->extcon,
+				EXTCON_JACK_HEADPHONE, false);
+		extcon_set_state_sync(mc_data->extcon,
+				EXTCON_JACK_MICROPHONE, false);
+		if (mc_data->poller)
+			mc_keys_poller_stop(mc_data->poller);
+
+		return;
+	}
+	if (!mc_data->adc) {
+		/* no ADC, so is headphone */
+		snd_soc_jack_report(jack_headset, SND_JACK_HEADPHONE, SND_JACK_HEADSET);
+		extcon_set_state_sync(mc_data->extcon, EXTCON_JACK_HEADPHONE, true);
+		extcon_set_state_sync(mc_data->extcon, EXTCON_JACK_MICROPHONE, false);
+		return;
+	}
+	ret = iio_read_channel_processed(mc_data->adc, &adc);
+	if (ret < 0) {
+		/* failed to read ADC, so assume headphone */
+		snd_soc_jack_report(jack_headset, SND_JACK_HEADPHONE, SND_JACK_HEADSET);
+		extcon_set_state_sync(mc_data->extcon, EXTCON_JACK_HEADPHONE, true);
+		extcon_set_state_sync(mc_data->extcon, EXTCON_JACK_MICROPHONE, false);
+
+	} else {
+		snd_soc_jack_report(jack_headset,
+				    snd_soc_jack_get_type(jack_headset, adc),
+				    SND_JACK_HEADSET);
+		extcon_set_state_sync(mc_data->extcon, EXTCON_JACK_HEADPHONE, true);
+
+		if (snd_soc_jack_get_type(jack_headset, adc) == SND_JACK_HEADSET) {
+			extcon_set_state_sync(mc_data->extcon, EXTCON_JACK_MICROPHONE, true);
+			if (mc_data->poller)
+				mc_keys_poller_start(mc_data->poller);
+		}
+	}
+};
+
+static irqreturn_t headset_det_irq_thread(int irq, void *data)
+{
+	struct multicodecs_data *mc_data = (struct multicodecs_data *)data;
+
+	queue_delayed_work(system_power_efficient_wq, &mc_data->handler, msecs_to_jiffies(200));
+
+	return IRQ_HANDLED;
+};
+
+static int mc_hp_event(struct snd_soc_dapm_widget *w,
+		       struct snd_kcontrol *kcontrol, int event)
+{
+	struct snd_soc_card *card = w->dapm->card;
+	struct multicodecs_data *mc_data = snd_soc_card_get_drvdata(card);
+
+	switch (event) {
+	case SND_SOC_DAPM_POST_PMU:
+		gpiod_set_value_cansleep(mc_data->hp_ctl_gpio, 1);
+		break;
+	case SND_SOC_DAPM_PRE_PMD:
+		gpiod_set_value_cansleep(mc_data->hp_ctl_gpio, 0);
+		break;
+	default:
+		return 0;
+
+	}
+
+	return 0;
+}
+
+static int mc_spk_event(struct snd_soc_dapm_widget *w,
+			struct snd_kcontrol *kcontrol, int event)
+{
+	struct snd_soc_card *card = w->dapm->card;
+	struct multicodecs_data *mc_data = snd_soc_card_get_drvdata(card);
+
+	switch (event) {
+	case SND_SOC_DAPM_POST_PMU:
+		gpiod_set_value_cansleep(mc_data->spk_ctl_gpio, 1);
+		break;
+	case SND_SOC_DAPM_PRE_PMD:
+		gpiod_set_value_cansleep(mc_data->spk_ctl_gpio, 0);
+		break;
+	default:
+		return 0;
+
+	}
+
+	return 0;
+}
+
+static const struct snd_soc_dapm_widget mc_dapm_widgets[] = {
+
+	SND_SOC_DAPM_HP("Headphone", NULL),
+	SND_SOC_DAPM_SPK("Speaker", NULL),
+	SND_SOC_DAPM_MIC("Main Mic", NULL),
+	SND_SOC_DAPM_MIC("Headset Mic", NULL),
+	SND_SOC_DAPM_SUPPLY("Speaker Power",
+			    SND_SOC_NOPM, 0, 0,
+			    mc_spk_event,
+			    SND_SOC_DAPM_POST_PMU |
+			    SND_SOC_DAPM_PRE_PMD),
+	SND_SOC_DAPM_SUPPLY("Headphone Power",
+			    SND_SOC_NOPM, 0, 0,
+			    mc_hp_event,
+			    SND_SOC_DAPM_POST_PMU |
+			    SND_SOC_DAPM_PRE_PMD),
+};
+
+static const struct snd_kcontrol_new mc_controls[] = {
+	SOC_DAPM_PIN_SWITCH("Headphone"),
+	SOC_DAPM_PIN_SWITCH("Speaker"),
+	SOC_DAPM_PIN_SWITCH("Main Mic"),
+	SOC_DAPM_PIN_SWITCH("Headset Mic"),
+};
 
 static int rk_multicodecs_hw_params(struct snd_pcm_substream *substream,
 				    struct snd_pcm_hw_params *params)
 {
 	struct snd_soc_pcm_runtime *rtd = substream->private_data;
-	struct snd_soc_dai *codec_dai = rtd->codec_dai;
 	struct snd_soc_dai *cpu_dai = rtd->cpu_dai;
+	struct snd_soc_dai *codec_dai;
 	struct multicodecs_data *mc_data = snd_soc_card_get_drvdata(rtd->card);
 	unsigned int mclk;
-	int ret;
+	int ret, i;
 
 	mclk = params_rate(params) * mc_data->mclk_fs;
 
-	ret = snd_soc_dai_set_sysclk(codec_dai, substream->stream, mclk,
-				     SND_SOC_CLOCK_IN);
-	if (ret && ret != -ENOTSUPP) {
-		pr_err("Set codec_dai sysclk failed: %d\n", ret);
-		goto out;
+	for (i = 0; i < rtd->num_codecs; i++) {
+		codec_dai = rtd->codec_dais[i];
+		ret = snd_soc_dai_set_sysclk(codec_dai, substream->stream, mclk,
+					     SND_SOC_CLOCK_IN);
+		if (ret && ret != -ENOTSUPP) {
+			pr_err("Set codec_dai sysclk failed: %d\n", ret);
+			goto out;
+		}
 	}
 
 	ret = snd_soc_dai_set_sysclk(cpu_dai, substream->stream, mclk,
@@ -80,16 +371,97 @@
 static int rk_dailink_init(struct snd_soc_pcm_runtime *rtd)
 {
 	struct multicodecs_data *mc_data = snd_soc_card_get_drvdata(rtd->card);
+	struct snd_soc_card *card = rtd->card;
+	struct snd_soc_jack *jack_headset;
+	int ret, irq;
+	struct snd_soc_jack_pin *pins;
+	struct snd_soc_jack_zone *zones;
+	struct snd_soc_jack_pin jack_pins[] = {
+		{
+			.pin = "Headphone",
+			.mask = SND_JACK_HEADPHONE,
+		}, {
+			.pin = "Headset Mic",
+			.mask = SND_JACK_MICROPHONE,
+		},
+	};
+	struct snd_soc_jack_zone headset_zones[] = {
+		{
+			.min_mv = 0,
+			.max_mv = 222,
+			.jack_type = SND_JACK_HEADPHONE,
+		}, {
+			.min_mv = 223,
+			.max_mv = 1500,
+			.jack_type = SND_JACK_HEADSET,
+		}, {
+			.min_mv = 1501,
+			.max_mv = UINT_MAX,
+			.jack_type = SND_JACK_HEADPHONE,
+		}
+	};
+
+	if ((!mc_data->codec_hp_det) && (gpiod_to_irq(mc_data->hp_det_gpio) < 0)) {
+		dev_info(card->dev, "Don't need to map headset detect gpio to irq\n");
+		return 0;
+	}
+
+	jack_headset = devm_kzalloc(card->dev, sizeof(*jack_headset), GFP_KERNEL);
+	if (!jack_headset)
+		return -ENOMEM;
+
+	pins = devm_kmemdup(card->dev, jack_pins,
+			    sizeof(*jack_pins) * ARRAY_SIZE(jack_pins), GFP_KERNEL);
+	if (!pins)
+		return -ENOMEM;
+
+	zones = devm_kmemdup(card->dev, headset_zones,
+			     sizeof(*headset_zones) * ARRAY_SIZE(headset_zones), GFP_KERNEL);
+	if (!zones)
+		return -ENOMEM;
+
+	ret = snd_soc_card_jack_new(card, "Headset",
+				    SND_JACK_HEADSET,
+				    jack_headset,
+				    pins, ARRAY_SIZE(jack_pins));
+	if (ret)
+		return ret;
+	ret = snd_soc_jack_add_zones(jack_headset, ARRAY_SIZE(headset_zones), zones);
+	if (ret)
+		return ret;
+
+	mc_data->jack_headset = jack_headset;
 
 	if (mc_data->codec_hp_det) {
-		snd_soc_card_jack_new(rtd->card, "Headphones",
-				      SND_JACK_HEADPHONE,
-				      &mc_hp_jack, NULL, 0);
+		struct snd_soc_dai *codec_dai;
+		int i;
 
-#ifdef CONFIG_SND_SOC_RK3308
-		if (rk3308_codec_set_jack_detect_cb)
-			rk3308_codec_set_jack_detect_cb(rtd->codec_dai->component, &mc_hp_jack);
-#endif
+		/* set jack for the first successful one */
+		for (i = 0; i < rtd->num_codecs; i++) {
+			codec_dai = rtd->codec_dais[i];
+			ret = snd_soc_component_set_jack(codec_dai->component,
+							 jack_headset, NULL);
+			if (ret >= 0)
+				break;
+		}
+	} else {
+		irq = gpiod_to_irq(mc_data->hp_det_gpio);
+		if (irq >= 0) {
+			ret = devm_request_threaded_irq(card->dev, irq, NULL,
+							headset_det_irq_thread,
+							IRQF_TRIGGER_RISING |
+							IRQF_TRIGGER_FALLING |
+							IRQF_ONESHOT,
+							"headset_detect",
+							mc_data);
+			if (ret) {
+				dev_err(card->dev, "Failed to request headset detect irq");
+				return ret;
+			}
+
+			queue_delayed_work(system_power_efficient_wq,
+					   &mc_data->handler, msecs_to_jiffies(50));
+		}
 	}
 
 	return 0;
@@ -216,8 +588,9 @@
 	struct multicodecs_data *mc_data;
 	struct of_phandle_args args;
 	struct device_node *node;
+	struct input_dev *input;
 	u32 val;
-	int count;
+	int count, value;
 	int ret = 0, i = 0, idx = 0;
 	const char *prefix = "rockchip,";
 
@@ -244,13 +617,18 @@
 	link->stream_name = link->name;
 	link->init = rk_dailink_init;
 	link->ops = &rk_ops;
+	link->ignore_pmdown_time = 1;
 
 	card->dai_link = link;
 	card->num_links = 1;
+	card->dapm_widgets = mc_dapm_widgets;
+	card->num_dapm_widgets = ARRAY_SIZE(mc_dapm_widgets);
+	card->controls = mc_controls;
+	card->num_controls = ARRAY_SIZE(mc_controls);
 	card->num_aux_devs = 0;
 
 	count = of_count_phandle_with_args(np, "rockchip,codec", NULL);
-	if (count < 0 || count > MAX_CODECS)
+	if (count < 0)
 		return -EINVAL;
 
 	/* refine codecs, remove unavailable node */
@@ -267,6 +645,9 @@
 
 	codecs = devm_kcalloc(&pdev->dev, idx,
 			      sizeof(*codecs), GFP_KERNEL);
+	if (!codecs)
+		return -ENOMEM;
+
 	link->codecs = codecs;
 	link->num_codecs = idx;
 	idx = 0;
@@ -305,6 +686,101 @@
 	mc_data->codec_hp_det =
 		of_property_read_bool(np, "rockchip,codec-hp-det");
 
+	mc_data->adc = devm_iio_channel_get(&pdev->dev, "adc-detect");
+
+	if (IS_ERR(mc_data->adc)) {
+		if (PTR_ERR(mc_data->adc) != -EPROBE_DEFER) {
+			mc_data->adc = NULL;
+			dev_warn(&pdev->dev, "Failed to get ADC channel");
+		}
+	} else {
+		if (mc_data->adc->channel->type != IIO_VOLTAGE)
+			return -EINVAL;
+
+		if (device_property_read_u32(&pdev->dev, "keyup-threshold-microvolt",
+					&mc_data->keyup_voltage)) {
+			dev_warn(&pdev->dev, "Invalid or missing keyup voltage\n");
+			return -EINVAL;
+		}
+		mc_data->keyup_voltage /= 1000;
+
+		ret = mc_keys_load_keymap(&pdev->dev, mc_data);
+		if (ret)
+			return ret;
+
+		input = devm_input_allocate_device(&pdev->dev);
+		if (IS_ERR(input)) {
+			dev_err(&pdev->dev, "failed to allocate input device\n");
+			return PTR_ERR(input);
+		}
+
+		input_set_drvdata(input, mc_data);
+
+		input->name = "headset-keys";
+		input->phys = "headset-keys/input0";
+		input->id.bustype = BUS_HOST;
+		input->id.vendor = 0x0001;
+		input->id.product = 0x0001;
+		input->id.version = 0x0100;
+
+		__set_bit(EV_KEY, input->evbit);
+		for (i = 0; i < mc_data->num_keys; i++)
+			__set_bit(mc_data->map[i].keycode, input->keybit);
+
+		if (device_property_read_bool(&pdev->dev, "autorepeat"))
+			__set_bit(EV_REP, input->evbit);
+
+		mc_data->input = input;
+		ret = mc_keys_setup_polling(mc_data, mc_keys_poll);
+		if (ret) {
+			dev_err(&pdev->dev, "Unable to set up polling: %d\n", ret);
+			return ret;
+		}
+
+		if (!device_property_read_u32(&pdev->dev, "poll-interval", &value))
+			mc_set_poll_interval(mc_data->poller, value);
+
+		ret = input_register_device(mc_data->input);
+		if (ret) {
+			dev_err(&pdev->dev, "Unable to register input device: %d\n", ret);
+			return ret;
+		}
+	}
+
+	INIT_DEFERRABLE_WORK(&mc_data->handler, adc_jack_handler);
+
+	mc_data->spk_ctl_gpio = devm_gpiod_get_optional(&pdev->dev,
+							"spk-con",
+							GPIOD_OUT_LOW);
+	if (IS_ERR(mc_data->spk_ctl_gpio))
+		return PTR_ERR(mc_data->spk_ctl_gpio);
+
+	mc_data->hp_ctl_gpio = devm_gpiod_get_optional(&pdev->dev,
+						       "hp-con",
+						       GPIOD_OUT_LOW);
+	if (IS_ERR(mc_data->hp_ctl_gpio))
+		return PTR_ERR(mc_data->hp_ctl_gpio);
+
+	mc_data->hp_det_gpio = devm_gpiod_get_optional(&pdev->dev, "hp-det", GPIOD_IN);
+	if (IS_ERR(mc_data->hp_det_gpio))
+		return PTR_ERR(mc_data->hp_det_gpio);
+
+	mc_data->extcon = devm_extcon_dev_allocate(&pdev->dev, headset_extcon_cable);
+	if (IS_ERR(mc_data->extcon)) {
+		dev_err(&pdev->dev, "allocate extcon failed\n");
+		return PTR_ERR(mc_data->extcon);
+	}
+
+	ret = devm_extcon_dev_register(&pdev->dev, mc_data->extcon);
+	if (ret) {
+		dev_err(&pdev->dev, "failed to register extcon: %d\n", ret);
+		return ret;
+	}
+
+	ret = snd_soc_of_parse_audio_routing(card, "rockchip,audio-routing");
+	if (ret < 0)
+		dev_warn(&pdev->dev, "Audio routing invalid/unspecified\n");
+
 	snd_soc_card_set_drvdata(card, mc_data);
 
 	ret = devm_snd_soc_register_card(&pdev->dev, card);

--
Gitblit v1.6.2