/*
   *  Routines for control of the TEA6330T circuit via i2c bus
   *  Sound fader control circuit for car radios by Philips Semiconductors

   *  Copyright (c) by Jaroslav Kysela <perex@perex.cz>

   *
   *
   *   This program is free software; you can redistribute it and/or modify
   *   it under the terms of the GNU General Public License as published by
   *   the Free Software Foundation; either version 2 of the License, or
   *   (at your option) any later version.
   *
   *   This program is distributed in the hope that it will be useful,
   *   but WITHOUT ANY WARRANTY; without even the implied warranty of
   *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   *   GNU General Public License for more details.
   *
   *   You should have received a copy of the GNU General Public License
   *   along with this program; if not, write to the Free Software
   *   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
   *
   */

  #include <linux/init.h>
  #include <linux/slab.h>

  #include <linux/module.h>

  #include <sound/core.h>

  #include <sound/control.h>

  #include <sound/tea6330t.h>

  MODULE_AUTHOR("Jaroslav Kysela <perex@perex.cz>");

  MODULE_DESCRIPTION("Routines for control of the TEA6330T circuit via i2c bus");
  MODULE_LICENSE("GPL");
  
  #define TEA6330T_ADDR			(0x80>>1) /* fixed address */
  
  #define TEA6330T_SADDR_VOLUME_LEFT	0x00	/* volume left */
  #define TEA6330T_SADDR_VOLUME_RIGHT	0x01	/* volume right */
  #define TEA6330T_SADDR_BASS		0x02	/* bass control */
  #define TEA6330T_SADDR_TREBLE		0x03	/* treble control */
  #define TEA6330T_SADDR_FADER		0x04	/* fader control */
  #define   TEA6330T_MFN			0x20	/* mute control for selected channels */
  #define   TEA6330T_FCH			0x10	/* select fader channels - front or rear */
  #define TEA6330T_SADDR_AUDIO_SWITCH	0x05	/* audio switch */
  #define   TEA6330T_GMU			0x80	/* mute control, general mute */
  #define   TEA6330T_EQN			0x40	/* equalizer switchover (0=equalizer-on) */

  
  struct tea6330t {
  	struct snd_i2c_device *device;
  	struct snd_i2c_bus *bus;

  	int equalizer;
  	int fader;
  	unsigned char regs[8];
  	unsigned char mleft, mright;
  	unsigned char bass, treble;
  	unsigned char max_bass, max_treble;

  };

  int snd_tea6330t_detect(struct snd_i2c_bus *bus, int equalizer)

  {
  	int res;
  
  	snd_i2c_lock(bus);
  	res = snd_i2c_probeaddr(bus, TEA6330T_ADDR);
  	snd_i2c_unlock(bus);
  	return res;
  }
  
  #if 0

  static void snd_tea6330t_set(struct tea6330t *tea,

  			     unsigned char addr, unsigned char value)
  {
  #if 0

  	printk(KERN_DEBUG "set - 0x%x/0x%x
  ", addr, value);

  #endif
  	snd_i2c_write(tea->bus, TEA6330T_ADDR, addr, value, 1);
  }
  #endif
  
  #define TEA6330T_MASTER_VOLUME(xname, xindex) \
  { .iface = SNDRV_CTL_ELEM_IFACE_MIXER, .name = xname, .index = xindex, \
    .info = snd_tea6330t_info_master_volume, \
    .get = snd_tea6330t_get_master_volume, .put = snd_tea6330t_put_master_volume }

  static int snd_tea6330t_info_master_volume(struct snd_kcontrol *kcontrol,
  					   struct snd_ctl_elem_info *uinfo)

  {
  	uinfo->type = SNDRV_CTL_ELEM_TYPE_INTEGER;
  	uinfo->count = 2;
  	uinfo->value.integer.min = 0;
  	uinfo->value.integer.max = 43;
  	return 0;
  }

  static int snd_tea6330t_get_master_volume(struct snd_kcontrol *kcontrol,
  					  struct snd_ctl_elem_value *ucontrol)

  {

  	struct tea6330t *tea = snd_kcontrol_chip(kcontrol);

  	
  	snd_i2c_lock(tea->bus);
  	ucontrol->value.integer.value[0] = tea->mleft - 0x14;
  	ucontrol->value.integer.value[1] = tea->mright - 0x14;
  	snd_i2c_unlock(tea->bus);
  	return 0;
  }

  static int snd_tea6330t_put_master_volume(struct snd_kcontrol *kcontrol,
  					  struct snd_ctl_elem_value *ucontrol)

  {

  	struct tea6330t *tea = snd_kcontrol_chip(kcontrol);

  	int change, count, err;
  	unsigned char bytes[3];
  	unsigned char val1, val2;
  	
  	val1 = (ucontrol->value.integer.value[0] % 44) + 0x14;
  	val2 = (ucontrol->value.integer.value[1] % 44) + 0x14;
  	snd_i2c_lock(tea->bus);
  	change = val1 != tea->mleft || val2 != tea->mright;
  	tea->mleft = val1;
  	tea->mright = val2;
  	count = 0;
  	if (tea->regs[TEA6330T_SADDR_VOLUME_LEFT] != 0) {
  		bytes[count++] = TEA6330T_SADDR_VOLUME_LEFT;
  		bytes[count++] = tea->regs[TEA6330T_SADDR_VOLUME_LEFT] = tea->mleft;
  	}
  	if (tea->regs[TEA6330T_SADDR_VOLUME_RIGHT] != 0) {
  		if (count == 0)
  			bytes[count++] = TEA6330T_SADDR_VOLUME_RIGHT;
  		bytes[count++] = tea->regs[TEA6330T_SADDR_VOLUME_RIGHT] = tea->mright;
  	}
  	if (count > 0) {
  		if ((err = snd_i2c_sendbytes(tea->device, bytes, count)) < 0)
  			change = err;
  	}
  	snd_i2c_unlock(tea->bus);
  	return change;
  }
  
  #define TEA6330T_MASTER_SWITCH(xname, xindex) \
  { .iface = SNDRV_CTL_ELEM_IFACE_MIXER, .name = xname, .index = xindex, \
    .info = snd_tea6330t_info_master_switch, \
    .get = snd_tea6330t_get_master_switch, .put = snd_tea6330t_put_master_switch }

  #define snd_tea6330t_info_master_switch		snd_ctl_boolean_stereo_info

  static int snd_tea6330t_get_master_switch(struct snd_kcontrol *kcontrol,
  					  struct snd_ctl_elem_value *ucontrol)

  {

  	struct tea6330t *tea = snd_kcontrol_chip(kcontrol);

  	
  	snd_i2c_lock(tea->bus);
  	ucontrol->value.integer.value[0] = tea->regs[TEA6330T_SADDR_VOLUME_LEFT] == 0 ? 0 : 1;
  	ucontrol->value.integer.value[1] = tea->regs[TEA6330T_SADDR_VOLUME_RIGHT] == 0 ? 0 : 1;
  	snd_i2c_unlock(tea->bus);
  	return 0;
  }

  static int snd_tea6330t_put_master_switch(struct snd_kcontrol *kcontrol,
  					  struct snd_ctl_elem_value *ucontrol)

  {

  	struct tea6330t *tea = snd_kcontrol_chip(kcontrol);

  	int change, err;
  	unsigned char bytes[3];
  	unsigned char oval1, oval2, val1, val2;
  	
  	val1 = ucontrol->value.integer.value[0] & 1;
  	val2 = ucontrol->value.integer.value[1] & 1;
  	snd_i2c_lock(tea->bus);
  	oval1 = tea->regs[TEA6330T_SADDR_VOLUME_LEFT] == 0 ? 0 : 1;
  	oval2 = tea->regs[TEA6330T_SADDR_VOLUME_RIGHT] == 0 ? 0 : 1;
  	change = val1 != oval1 || val2 != oval2;
  	tea->regs[TEA6330T_SADDR_VOLUME_LEFT] = val1 ? tea->mleft : 0;
  	tea->regs[TEA6330T_SADDR_VOLUME_RIGHT] = val2 ? tea->mright : 0;
  	bytes[0] = TEA6330T_SADDR_VOLUME_LEFT;
  	bytes[1] = tea->regs[TEA6330T_SADDR_VOLUME_LEFT];
  	bytes[2] = tea->regs[TEA6330T_SADDR_VOLUME_RIGHT];
  	if ((err = snd_i2c_sendbytes(tea->device, bytes, 3)) < 0)
  		change = err;
  	snd_i2c_unlock(tea->bus);
  	return change;
  }
  
  #define TEA6330T_BASS(xname, xindex) \
  { .iface = SNDRV_CTL_ELEM_IFACE_MIXER, .name = xname, .index = xindex, \
    .info = snd_tea6330t_info_bass, \
    .get = snd_tea6330t_get_bass, .put = snd_tea6330t_put_bass }

  static int snd_tea6330t_info_bass(struct snd_kcontrol *kcontrol,
  				  struct snd_ctl_elem_info *uinfo)

  {

  	struct tea6330t *tea = snd_kcontrol_chip(kcontrol);

  
  	uinfo->type = SNDRV_CTL_ELEM_TYPE_INTEGER;
  	uinfo->count = 1;
  	uinfo->value.integer.min = 0;
  	uinfo->value.integer.max = tea->max_bass;
  	return 0;
  }

  static int snd_tea6330t_get_bass(struct snd_kcontrol *kcontrol,
  				 struct snd_ctl_elem_value *ucontrol)

  {

  	struct tea6330t *tea = snd_kcontrol_chip(kcontrol);

  	
  	ucontrol->value.integer.value[0] = tea->bass;
  	return 0;
  }

  static int snd_tea6330t_put_bass(struct snd_kcontrol *kcontrol,
  				 struct snd_ctl_elem_value *ucontrol)

  {

  	struct tea6330t *tea = snd_kcontrol_chip(kcontrol);

  	int change, err;
  	unsigned char bytes[2];
  	unsigned char val1;
  	
  	val1 = ucontrol->value.integer.value[0] % (tea->max_bass + 1);
  	snd_i2c_lock(tea->bus);
  	tea->bass = val1;
  	val1 += tea->equalizer ? 7 : 3;
  	change = tea->regs[TEA6330T_SADDR_BASS] != val1;
  	bytes[0] = TEA6330T_SADDR_BASS;
  	bytes[1] = tea->regs[TEA6330T_SADDR_BASS] = val1;
  	if ((err = snd_i2c_sendbytes(tea->device, bytes, 2)) < 0)
  		change = err;
  	snd_i2c_unlock(tea->bus);
  	return change;
  }
  
  #define TEA6330T_TREBLE(xname, xindex) \
  { .iface = SNDRV_CTL_ELEM_IFACE_MIXER, .name = xname, .index = xindex, \
    .info = snd_tea6330t_info_treble, \
    .get = snd_tea6330t_get_treble, .put = snd_tea6330t_put_treble }

  static int snd_tea6330t_info_treble(struct snd_kcontrol *kcontrol,
  				    struct snd_ctl_elem_info *uinfo)

  {

  	struct tea6330t *tea = snd_kcontrol_chip(kcontrol);

  
  	uinfo->type = SNDRV_CTL_ELEM_TYPE_INTEGER;
  	uinfo->count = 1;
  	uinfo->value.integer.min = 0;
  	uinfo->value.integer.max = tea->max_treble;
  	return 0;
  }

  static int snd_tea6330t_get_treble(struct snd_kcontrol *kcontrol,
  				   struct snd_ctl_elem_value *ucontrol)

  {

  	struct tea6330t *tea = snd_kcontrol_chip(kcontrol);

  	
  	ucontrol->value.integer.value[0] = tea->treble;
  	return 0;
  }

  static int snd_tea6330t_put_treble(struct snd_kcontrol *kcontrol,
  				   struct snd_ctl_elem_value *ucontrol)

  {

  	struct tea6330t *tea = snd_kcontrol_chip(kcontrol);

  	int change, err;
  	unsigned char bytes[2];
  	unsigned char val1;
  	
  	val1 = ucontrol->value.integer.value[0] % (tea->max_treble + 1);
  	snd_i2c_lock(tea->bus);
  	tea->treble = val1;
  	val1 += 3;
  	change = tea->regs[TEA6330T_SADDR_TREBLE] != val1;
  	bytes[0] = TEA6330T_SADDR_TREBLE;
  	bytes[1] = tea->regs[TEA6330T_SADDR_TREBLE] = val1;
  	if ((err = snd_i2c_sendbytes(tea->device, bytes, 2)) < 0)
  		change = err;
  	snd_i2c_unlock(tea->bus);
  	return change;
  }

  static struct snd_kcontrol_new snd_tea6330t_controls[] = {

  TEA6330T_MASTER_SWITCH("Master Playback Switch", 0),
  TEA6330T_MASTER_VOLUME("Master Playback Volume", 0),
  TEA6330T_BASS("Tone Control - Bass", 0),
  TEA6330T_TREBLE("Tone Control - Treble", 0)
  };

  static void snd_tea6330_free(struct snd_i2c_device *device)

  {

  	kfree(device->private_data);

  }
                                          

  int snd_tea6330t_update_mixer(struct snd_card *card,
  			      struct snd_i2c_bus *bus,

  			      int equalizer, int fader)
  {

  	struct snd_i2c_device *device;
  	struct tea6330t *tea;
  	struct snd_kcontrol_new *knew;

  	unsigned int idx;
  	int err = -ENOMEM;
  	u8 default_treble, default_bass;
  	unsigned char bytes[7];

  	tea = kzalloc(sizeof(*tea), GFP_KERNEL);

  	if (tea == NULL)
  		return -ENOMEM;
  	if ((err = snd_i2c_device_create(bus, "TEA6330T", TEA6330T_ADDR, &device)) < 0) {
  		kfree(tea);
  		return err;
  	}
  	tea->device = device;
  	tea->bus = bus;
  	tea->equalizer = equalizer;
  	tea->fader = fader;
  	device->private_data = tea;
  	device->private_free = snd_tea6330_free;
  
  	snd_i2c_lock(bus);
  
  	/* turn fader off and handle equalizer */
  	tea->regs[TEA6330T_SADDR_FADER] = 0x3f;
  	tea->regs[TEA6330T_SADDR_AUDIO_SWITCH] = equalizer ? 0 : TEA6330T_EQN;
  	/* initialize mixer */
  	if (!tea->equalizer) {
  		tea->max_bass = 9;
  		tea->max_treble = 8;
  		default_bass = 3 + 4;
  		tea->bass = 4;
  		default_treble = 3 + 4;
  		tea->treble = 4;
  	} else {
  		tea->max_bass = 5;
  		tea->max_treble = 0;
  		default_bass = 7 + 4;
  		tea->bass = 4;
  		default_treble = 3;
  		tea->treble = 0;
  	}
  	tea->mleft = tea->mright = 0x14;
  	tea->regs[TEA6330T_SADDR_BASS] = default_bass;
  	tea->regs[TEA6330T_SADDR_TREBLE] = default_treble;
  
  	/* compose I2C message and put the hardware to initial state */
  	bytes[0] = TEA6330T_SADDR_VOLUME_LEFT;
  	for (idx = 0; idx < 6; idx++)
  		bytes[idx+1] = tea->regs[idx];
  	if ((err = snd_i2c_sendbytes(device, bytes, 7)) < 0)
  		goto __error;
  
  	strcat(card->mixername, ",TEA6330T");
  	if ((err = snd_component_add(card, "TEA6330T")) < 0)
  		goto __error;
  
  	for (idx = 0; idx < ARRAY_SIZE(snd_tea6330t_controls); idx++) {
  		knew = &snd_tea6330t_controls[idx];
  		if (tea->treble == 0 && !strcmp(knew->name, "Tone Control - Treble"))
  			continue;
  		if ((err = snd_ctl_add(card, snd_ctl_new1(knew, tea))) < 0)
  			goto __error;
  	}
  
  	snd_i2c_unlock(bus);
  	return 0;
  	
        __error:
        	snd_i2c_unlock(bus);
        	snd_i2c_device_free(device);
        	return err;
  }
  
  EXPORT_SYMBOL(snd_tea6330t_detect);
  EXPORT_SYMBOL(snd_tea6330t_update_mixer);
  
  /*
   *  INIT part
   */
  
  static int __init alsa_tea6330t_init(void)
  {
  	return 0;
  }
  
  static void __exit alsa_tea6330t_exit(void)
  {
  }
  
  module_init(alsa_tea6330t_init)
  module_exit(alsa_tea6330t_exit)