// SPDX-License-Identifier: GPL-2.0-only
//
// rt-sdw-common.c
//
// Copyright(c) 2024 Realtek Semiconductor Corp.
//

/*
 * This file defines common functions used with Realtek soundwire codecs.
 */

#include <linux/module.h>
#include <linux/regmap.h>
#include <linux/bitops.h>
#include <linux/soundwire/sdw_registers.h>
#include <sound/jack.h>

#include "rt-sdw-common.h"

/**
 * rt_sdca_index_write - Write a value to Realtek defined register.
 *
 * @map: map for setting.
 * @nid: Realtek-defined ID.
 * @reg: register.
 * @value: value.
 *
 * A value of zero will be returned on success, a negative errno will
 * be returned in error cases.
 */
int rt_sdca_index_write(struct regmap *map, unsigned int nid,
	unsigned int reg, unsigned int value)
{
	unsigned int addr = (nid << 20) | reg;
	int ret;

	ret = regmap_write(map, addr, value);
	if (ret < 0)
		pr_err("Failed to set value: %06x <= %04x ret=%d\n",
			addr, value, ret);

	return ret;
}
EXPORT_SYMBOL_GPL(rt_sdca_index_write);

/**
 * rt_sdca_index_read - Read value from Realtek defined register.
 *
 * @map: map for setting.
 * @nid: Realtek-defined ID.
 * @reg: register.
 * @value: value.
 *
 * A value of zero will be returned on success, a negative errno will
 * be returned in error cases.
 */
int rt_sdca_index_read(struct regmap *map, unsigned int nid,
	unsigned int reg, unsigned int *value)
{
	unsigned int addr = (nid << 20) | reg;
	int ret;

	ret = regmap_read(map, addr, value);
	if (ret < 0)
		pr_err("Failed to get value: %06x => %04x ret=%d\n",
			addr, *value, ret);

	return ret;
}
EXPORT_SYMBOL_GPL(rt_sdca_index_read);

/**
 * rt_sdca_index_update_bits - Update value on Realtek defined register.
 *
 * @map: map for setting.
 * @nid: Realtek-defined ID.
 * @reg: register.
 * @mask: Bitmask to change
 * @val: New value for bitmask
 *
 * A value of zero will be returned on success, a negative errno will
 * be returned in error cases.
 */

int rt_sdca_index_update_bits(struct regmap *map,
	unsigned int nid, unsigned int reg, unsigned int mask, unsigned int val)
{
	unsigned int tmp;
	int ret;

	ret = rt_sdca_index_read(map, nid, reg, &tmp);
	if (ret < 0)
		return ret;

	set_mask_bits(&tmp, mask, val);
	return rt_sdca_index_write(map, nid, reg, tmp);
}
EXPORT_SYMBOL_GPL(rt_sdca_index_update_bits);

/**
 * rt_sdca_btn_type - Decision of button type.
 *
 * @buffer: UMP message buffer.
 *
 * A button type will be returned regarding to buffer,
 * it returns zero if buffer cannot be recognized.
 */
int rt_sdca_btn_type(unsigned char *buffer)
{
	u8 btn_type = 0;
	int ret = 0;

	btn_type |= buffer[0] & 0xf;
	btn_type |= (buffer[0] >> 4) & 0xf;
	btn_type |= buffer[1] & 0xf;
	btn_type |= (buffer[1] >> 4) & 0xf;

	if (btn_type & BIT(0))
		ret |= SND_JACK_BTN_2;
	if (btn_type & BIT(1))
		ret |= SND_JACK_BTN_3;
	if (btn_type & BIT(2))
		ret |= SND_JACK_BTN_0;
	if (btn_type & BIT(3))
		ret |= SND_JACK_BTN_1;

	return ret;
}
EXPORT_SYMBOL_GPL(rt_sdca_btn_type);

/**
 * rt_sdca_headset_detect - Headset jack type detection.
 *
 * @map: map for setting.
 * @entity_id: SDCA entity ID.
 *
 * A headset jack type will be returned, a negative errno will
 * be returned in error cases.
 */
int rt_sdca_headset_detect(struct regmap *map, unsigned int entity_id)
{
	unsigned int det_mode, jack_type;
	int ret;

	/* get detected_mode */
	ret = regmap_read(map, SDW_SDCA_CTL(SDCA_NUM_JACK_CODEC, entity_id,
			RT_SDCA_CTL_DETECTED_MODE, 0), &det_mode);

	if (ret < 0)
		goto io_error;

	switch (det_mode) {
	case 0x03:
		jack_type = SND_JACK_HEADPHONE;
		break;
	case 0x05:
		jack_type = SND_JACK_HEADSET;
		break;
	default:
		jack_type = 0;
		break;
	}

	/* write selected_mode */
	if (det_mode) {
		ret = regmap_write(map, SDW_SDCA_CTL(SDCA_NUM_JACK_CODEC, entity_id,
				RT_SDCA_CTL_SELECTED_MODE, 0), det_mode);
		if (ret < 0)
			goto io_error;
	}

	return jack_type;

io_error:
	pr_err_ratelimited("IO error in %s, ret %d\n", __func__, ret);
	return ret;
}
EXPORT_SYMBOL_GPL(rt_sdca_headset_detect);

/**
 * rt_sdca_button_detect - Read UMP message and decide button type.
 *
 * @map: map for setting.
 * @entity_id: SDCA entity ID.
 * @hid_buf_addr: HID buffer address.
 * @hid_id: Report ID for HID.
 *
 * A button type will be returned regarding to buffer,
 * it returns zero if buffer cannot be recognized.
 */
int rt_sdca_button_detect(struct regmap *map, unsigned int entity_id,
	unsigned int hid_buf_addr, unsigned int hid_id)
{
	unsigned int btn_type = 0, offset, idx, val, owner;
	unsigned char buf[3];
	int ret;

	/* get current UMP message owner */
	ret = regmap_read(map, SDW_SDCA_CTL(SDCA_NUM_HID, entity_id,
			RT_SDCA_CTL_HIDTX_CURRENT_OWNER, 0), &owner);
	if (ret < 0)
		return 0;

	/* if owner is device then there is no button event from device */
	if (owner == 1)
		return 0;

	/* read UMP message offset */
	ret = regmap_read(map, SDW_SDCA_CTL(SDCA_NUM_HID, entity_id,
			RT_SDCA_CTL_HIDTX_MESSAGE_OFFSET, 0), &offset);
	if (ret < 0)
		goto _end_btn_det_;

	for (idx = 0; idx < sizeof(buf); idx++) {
		ret = regmap_read(map, hid_buf_addr + offset + idx, &val);
		if (ret < 0)
			goto _end_btn_det_;
		buf[idx] = val & 0xff;
	}
	/* Report ID for HID */
	if (buf[0] == hid_id)
		btn_type = rt_sdca_btn_type(&buf[1]);

_end_btn_det_:
	/* Host is owner, so set back to device */
	if (owner == 0)
		/* set owner to device */
		regmap_write(map,
			SDW_SDCA_CTL(SDCA_NUM_HID, entity_id,
				RT_SDCA_CTL_HIDTX_CURRENT_OWNER, 0), 0x01);

	return btn_type;
}
EXPORT_SYMBOL_GPL(rt_sdca_button_detect);

MODULE_DESCRIPTION("Realtek soundwire common functions");
MODULE_AUTHOR("jack yu <jack.yu@realtek.com>");
MODULE_LICENSE("GPL");