// 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");