/*
   * ITE 8213 IDE driver
   *
   * Copyright (C) 2006 Jack Lee
   * Copyright (C) 2006 Alan Cox
   * Copyright (C) 2007 Bartlomiej Zolnierkiewicz
   */

  #include <linux/kernel.h>
  #include <linux/types.h>
  #include <linux/module.h>
  #include <linux/pci.h>

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

  #define DRV_NAME "it8213"

  /**

   *	it8213_set_pio_mode	-	set host controller for PIO mode

   *	@hwif: port

   *	@drive: drive

   *

   *	Set the interface PIO mode.

   */

  static void it8213_set_pio_mode(ide_hwif_t *hwif, ide_drive_t *drive)

  {

  	struct pci_dev *dev	= to_pci_dev(hwif->dev);

  	int is_slave		= drive->dn & 1;

  	int master_port		= 0x40;
  	int slave_port		= 0x44;
  	unsigned long flags;
  	u16 master_data;
  	u8 slave_data;

  	static DEFINE_SPINLOCK(tune_lock);
  	int control = 0;

  	const u8 pio = drive->pio_mode - XFER_PIO_0;

  	static const u8 timings[][2] = {

  					{ 0, 0 },
  					{ 0, 0 },
  					{ 1, 0 },
  					{ 2, 1 },
  					{ 2, 3 }, };

  	spin_lock_irqsave(&tune_lock, flags);
  	pci_read_config_word(dev, master_port, &master_data);

  
  	if (pio > 1)
  		control |= 1;	/* Programmable timing on */
  	if (drive->media != ide_disk)
  		control |= 4;	/* ATAPI */

  	if (ide_pio_need_iordy(drive, pio))

  		control |= 2;	/* IORDY */

  	if (is_slave) {

  		master_data |=  0x4000;
  		master_data &= ~0x0070;

  		if (pio > 1)

  			master_data = master_data | (control << 4);

  		pci_read_config_byte(dev, slave_port, &slave_data);
  		slave_data = slave_data & 0xf0;

  		slave_data = slave_data | (timings[pio][0] << 2) | timings[pio][1];

  	} else {

  		master_data &= ~0x3307;

  		if (pio > 1)

  			master_data = master_data | control;

  		master_data = master_data | (timings[pio][0] << 12) | (timings[pio][1] << 8);
  	}
  	pci_write_config_word(dev, master_port, master_data);
  	if (is_slave)
  		pci_write_config_byte(dev, slave_port, slave_data);
  	spin_unlock_irqrestore(&tune_lock, flags);
  }

  /**

   *	it8213_set_dma_mode	-	set host controller for DMA mode

   *	@hwif: port

   *	@drive: drive

   *

   *	Tune the ITE chipset for the DMA mode.

   */

  static void it8213_set_dma_mode(ide_hwif_t *hwif, ide_drive_t *drive)

  {

  	struct pci_dev *dev	= to_pci_dev(hwif->dev);

  	u8 maslave		= 0x40;

  	int a_speed		= 3 << (drive->dn * 4);
  	int u_flag		= 1 << drive->dn;
  	int v_flag		= 0x01 << drive->dn;
  	int w_flag		= 0x10 << drive->dn;
  	int u_speed		= 0;
  	u16			reg4042, reg4a;

  	u8			reg48, reg54, reg55;

  	const u8 speed		= drive->dma_mode;

  
  	pci_read_config_word(dev, maslave, &reg4042);
  	pci_read_config_byte(dev, 0x48, &reg48);
  	pci_read_config_word(dev, 0x4a, &reg4a);
  	pci_read_config_byte(dev, 0x54, &reg54);
  	pci_read_config_byte(dev, 0x55, &reg55);

  	if (speed >= XFER_UDMA_0) {

  		u8 udma = speed - XFER_UDMA_0;
  
  		u_speed = min_t(u8, 2 - (udma & 1), udma) << (drive->dn * 4);

  		if (!(reg48 & u_flag))
  			pci_write_config_byte(dev, 0x48, reg48 | u_flag);

  		if (speed >= XFER_UDMA_5)

  			pci_write_config_byte(dev, 0x55, (u8) reg55|w_flag);

  		else

  			pci_write_config_byte(dev, 0x55, (u8) reg55 & ~w_flag);

  
  		if ((reg4a & a_speed) != u_speed)
  			pci_write_config_word(dev, 0x4a, (reg4a & ~a_speed) | u_speed);

  		if (speed > XFER_UDMA_2) {

  			if (!(reg54 & v_flag))
  				pci_write_config_byte(dev, 0x54, reg54 | v_flag);
  		} else
  			pci_write_config_byte(dev, 0x54, reg54 & ~v_flag);

  	} else {

  		const u8 mwdma_to_pio[] = { 0, 3, 4 };

  		if (reg48 & u_flag)
  			pci_write_config_byte(dev, 0x48, reg48 & ~u_flag);
  		if (reg4a & a_speed)
  			pci_write_config_word(dev, 0x4a, reg4a & ~a_speed);
  		if (reg54 & v_flag)
  			pci_write_config_byte(dev, 0x54, reg54 & ~v_flag);
  		if (reg55 & w_flag)
  			pci_write_config_byte(dev, 0x55, (u8) reg55 & ~w_flag);

  
  		if (speed >= XFER_MW_DMA_0)

  			drive->pio_mode =
  				mwdma_to_pio[speed - XFER_MW_DMA_0] + XFER_PIO_0;

  		else

  			drive->pio_mode = XFER_PIO_2; /* for SWDMA2 */

  		it8213_set_pio_mode(hwif, drive);

  	}

  }

  static u8 it8213_cable_detect(ide_hwif_t *hwif)

  {
  	struct pci_dev *dev = to_pci_dev(hwif->dev);
  	u8 reg42h = 0;
  
  	pci_read_config_byte(dev, 0x42, &reg42h);
  
  	return (reg42h & 0x02) ? ATA_CBL_PATA40 : ATA_CBL_PATA80;
  }

  static const struct ide_port_ops it8213_port_ops = {
  	.set_pio_mode		= it8213_set_pio_mode,
  	.set_dma_mode		= it8213_set_dma_mode,
  	.cable_detect		= it8213_cable_detect,
  };

  static const struct ide_port_info it8213_chipset __devinitdata = {

  	.name		= DRV_NAME,

  	.enablebits	= { {0x41, 0x80, 0x80} },
  	.port_ops	= &it8213_port_ops,
  	.host_flags	= IDE_HFLAG_SINGLE,
  	.pio_mask	= ATA_PIO4,
  	.swdma_mask	= ATA_SWDMA2_ONLY,
  	.mwdma_mask	= ATA_MWDMA12_ONLY,
  	.udma_mask	= ATA_UDMA6,

  };

  /**
   *	it8213_init_one	-	pci layer discovery entry
   *	@dev: PCI device
   *	@id: ident table entry
   *
   *	Called by the PCI code when it finds an ITE8213 controller. As
   *	this device follows the standard interfaces we can use the
   *	standard helper functions to do almost all the work for us.
   */
  
  static int __devinit it8213_init_one(struct pci_dev *dev, const struct pci_device_id *id)
  {

  	return ide_pci_init_one(dev, &it8213_chipset, NULL);

  }

  static const struct pci_device_id it8213_pci_tbl[] = {
  	{ PCI_VDEVICE(ITE, PCI_DEVICE_ID_ITE_8213), 0 },

  	{ 0, },
  };
  
  MODULE_DEVICE_TABLE(pci, it8213_pci_tbl);

  static struct pci_driver it8213_pci_driver = {

  	.name		= "ITE8213_IDE",
  	.id_table	= it8213_pci_tbl,
  	.probe		= it8213_init_one,

  	.remove		= ide_pci_remove,

  	.suspend	= ide_pci_suspend,
  	.resume		= ide_pci_resume,

  };
  
  static int __init it8213_ide_init(void)
  {

  	return ide_pci_register_driver(&it8213_pci_driver);

  }

  static void __exit it8213_ide_exit(void)
  {

  	pci_unregister_driver(&it8213_pci_driver);

  }

  module_init(it8213_ide_init);

  module_exit(it8213_ide_exit);

  MODULE_AUTHOR("Jack Lee, Alan Cox");

  MODULE_DESCRIPTION("PCI driver module for the ITE 8213");
  MODULE_LICENSE("GPL");