#include <linux/types.h>
#include <linux/pci.h>
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/delay.h>
#include <linux/mm.h>
#include <asm/danube/danube.h>
#include <asm/danube/irq.h>
#include <asm/addrspace.h>

#if defined(CONFIG_PCI) && defined(CONFIG_DANUBE_CHIP_A11)
#if defined(CONFIG_DANUBE_EBU_PCI_SOFTWARE_ARBITOR)
#define EBU_PCI_SOFTWARE_ARBITOR
#endif
#if defined(CONFIG_DANUBE_PPE_PCI_SOFTWARE_ARBITOR)
#define USE_FIX_FOR_PCI_PPE
#endif
#endif

#define DANUBE_PCI_REG32( addr )		  (*(volatile u32 *)(addr))
#undef DANUBE_PCI_TEST

#define DANUBE_PCI_MEM_BASE    0x18000000
#define DANUBE_PCI_MEM_SIZE    0x02000000
#define DANUBE_PCI_IO_BASE     0x1AE00000
#define DANUBE_PCI_IO_SIZE     0x00200000

#define DANUBE_PCI_CFG_BUSNUM_SHF 16
#define DANUBE_PCI_CFG_DEVNUM_SHF 11
#define DANUBE_PCI_CFG_FUNNUM_SHF 8

#define PCI_ACCESS_READ  0
#define PCI_ACCESS_WRITE 1

static struct resource pci_io_resource = {
	.name = "io pci IO space",
	.start = DANUBE_PCI_IO_BASE,
	.end = DANUBE_PCI_IO_BASE + DANUBE_PCI_IO_SIZE,
	.flags = IORESOURCE_IO
};

static struct resource pci_mem_resource = {
	.name = "ext pci memory space",
	.start = DANUBE_PCI_MEM_BASE,
	.end = DANUBE_PCI_MEM_BASE + DANUBE_PCI_MEM_SIZE,
	.flags = IORESOURCE_MEM
};

static int danube_pci_config_access(unsigned char access_type,
		struct pci_bus *bus, unsigned int devfn, unsigned int where, u32 *data){
	u32 pci_addr;
	u32 temp;

	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);

	/* Danube support slot from 0 to 15 */
	/* dev_fn 0&0x68 (AD29) is danube itself */
	if ((bus != 0) || ((devfn & 0xf8) > 0x78) || ((devfn & 0xf8) == 0)
		|| ((devfn & 0xf8) == 0x68)) {
		printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
		return 1;
	}

	printk("\nbus=%x dev_fn=%x %s", (u32)bus, devfn, (access_type == PCI_ACCESS_WRITE) ? "WRITE" : "READ");
	pci_addr = PCI_CFG_BASE | (bus->number << DANUBE_PCI_CFG_BUSNUM_SHF) | (devfn << DANUBE_PCI_CFG_FUNNUM_SHF) | (where & ~0x3);

	printk(" address=%x", pci_addr);
	/* Perform access */
	if (access_type == PCI_ACCESS_WRITE) {
#ifdef CONFIG_DANUBE_PCI_HW_SWAP
		*(volatile u32 *) pci_addr = swab32 (*data);
#else
		*(volatile u32 *) pci_addr = *data;
#endif
	} else {
		*data = *(volatile u32 *) pci_addr;
#ifdef CONFIG_DANUBE_PCI_HW_SWAP
		*data = swab32 (*data);
#endif
	}
	printk(" %x", *data);
		/* clean possible Master abort */
	/* use PCI access to clear it */
	pci_addr = PCI_CFG_BASE | (0x0 << DANUBE_PCI_CFG_FUNNUM_SHF) + 4;
	temp = *((volatile u32 *) pci_addr);
#ifdef CONFIG_DANUBE_PCI_HW_SWAP
	temp = swab32 (temp);
#endif /*CONFIG_DANUBE_PCI_HW_SWAP */
	pci_addr = PCI_CFG_BASE | (0x68 << DANUBE_PCI_CFG_FUNNUM_SHF) + 4;
	*(volatile u32 *) pci_addr = temp;

	//TODO: non-existed device
	if ((*data) == 0xffffffff) {
		return 1;
	}

	return 0;
}


static int danube_pci_read_config_dword(struct pci_bus *bus, unsigned int devfn, int where, int size, u32 *val){
	u32 data = 0;
printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);

	if(where & 3)
		return PCIBIOS_BAD_REGISTER_NUMBER;

	if (danube_pci_config_access(PCI_ACCESS_READ, bus, devfn, where, &data))
		return -1;

	*val = data;

	return PCIBIOS_SUCCESSFUL;
}

static int danube_pci_write_config_dword(struct pci_bus *bus, unsigned int devfn, int where, int size, u32 val){
	u32 data = 0;
printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);

	if (where & 3)
		return PCIBIOS_BAD_REGISTER_NUMBER;

	if (danube_pci_config_access(PCI_ACCESS_WRITE, bus, devfn, where, &data))
		return -1;

	return PCIBIOS_SUCCESSFUL;
}

struct pci_ops danube_pci_ops = {
	.read = danube_pci_read_config_dword,
	.write = danube_pci_write_config_dword
};


int pcibios_plat_dev_init(struct pci_dev *dev){
	u8 pin;
printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);

	dev->irq = 0;
	pci_read_config_byte(dev, PCI_INTERRUPT_PIN, &pin);
	switch(pin) {
		case 0:
			break;
		case 1:
			//PCI_INTA--shared with EBU
			//falling edge level triggered:0x4, low level:0xc, rising edge:0x2
			(*DANUBE_EBU_PCC_CON) |= 0xc;
			(*DANUBE_EBU_PCC_IEN) |= 0x10;
			dev->irq = INT_NUM_IM0_IRL22;
			pci_write_config_byte(dev, PCI_INTERRUPT_LINE, (u32)dev->irq);
			break;
		case 2:
		case 3:
		case 4:
			printk ("WARNING: interrupt pin %d not supported yet!\n", pin);
		default:
			printk ("WARNING: invalid interrupt pin %d\n", pin);
			return 1;
	}

	return 0;
}

static void __init danube_pci_startup (void){
	/*initialize the first PCI device--danube itself */
	u32 temp_buffer;
printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
*DANUBE_PMU_PWDCR &= ~(1 << 14); //FP0
	/* make sure PCI clock is slower than FPI clock */
	extern unsigned int danube_get_fpi_hz (void);
	//pliu20060613
	//if ( danube_get_fpi_hz() < CLOCK_60M){
	/*use 33.3M */
	/*TODO: trigger reset */
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	DANUBE_PCI_REG32 (DANUBE_CGU_IFCCR) &= (~(0xf00000));
	DANUBE_PCI_REG32 (DANUBE_CGU_IFCCR) |= ((0x800000));
	//pliu20060613
	//printk("PCI clock 33.3MHz\n");
	//}
	/* PCIS of IF_CLK of CGU   : 1 =>PCI Clock output
	   0 =>clock input
	   PADsel of PCI_CR of CGU : 1 =>From CGU
	   : 0 =>From pad
	 */
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	DANUBE_PCI_REG32 (DANUBE_CGU_IFCCR) |= ((1 << 16));
	//removed: pliu20060613
	//DANUBE_PCI_REG32(DANUBE_CGU_PCICR) = (1<<31);
	DANUBE_PCI_REG32 (DANUBE_CGU_PCICR) = ((1 << 31) | (1 << 30));

	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	/* prepare GPIO */
	/* PCI_RST: P1.5 ALT 01 */
	//pliu20060613: start
	*DANUBE_GPIO_P1_OUT |= ((1 << 5));
	*DANUBE_GPIO_P1_OD |= ((1 << 5));
	*DANUBE_GPIO_P1_DIR |= ((1 << 5));
	*DANUBE_GPIO_P1_ALTSEL1 &= (~(1 << 5));
	*DANUBE_GPIO_P1_ALTSEL0 &= (~(1 << 5));
	//pliu20060613: end
	/* PCI_REQ1: P1.13 ALT 01 */
	/* PCI_GNT1: P1.14 ALT 01 */
	*DANUBE_GPIO_P1_DIR &= (~(0x2000));
	*DANUBE_GPIO_P1_DIR |= ((0x4000));
	*DANUBE_GPIO_P1_ALTSEL1 &= (~(0x6000));
	*DANUBE_GPIO_P1_ALTSEL0 |= ((0x6000));
	/* PCI_REQ2: P1.15 ALT 10 */
	/* PCI_GNT2: P1.7 ALT 10 */

printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);

	/* enable auto-switching between PCI and EBU */
	DANUBE_PCI_REG32 (PCI_CR_CLK_CTRL_REG) = 0xA;
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	/* busy, i.e. configuration is not done, PCI access has to be retried */
	DANUBE_PCI_REG32 (PCI_CR_PCI_MOD_REG) &= (~(1 << 24));
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	wmb ();
	/* BUS Master/IO/MEM access */
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	DANUBE_PCI_REG32 (PCI_CS_STS_CMD_REG) |= (7);
printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);

	temp_buffer = DANUBE_PCI_REG32 (PCI_CR_PC_ARB_REG);
#ifdef EBU_PCI_SOFTWARE_ARBITOR
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	temp_buffer &= (~(0x1 << 16));	// enable 1st external pci master only
#else
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	/* enable external 2 PCI masters */
	temp_buffer &= (~(0xf << 16));
#endif
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	/* enable internal arbiter */
	temp_buffer |= (1 << INTERNAL_ARB_ENABLE_BIT);
	/* enable internal PCI master reqest */
	temp_buffer &= (~(3 << PCI_MASTER0_REQ_MASK_2BITS));

#ifdef EBU_PCI_SOFTWARE_ARBITOR
#if 1
	*DANUBE_GPIO_P0_OUT &= (~(1 << 7));
	*DANUBE_GPIO_P0_OD |= ((1 << 7));
	*DANUBE_GPIO_P0_DIR |= ((1 << 7));
	*DANUBE_GPIO_P0_ALTSEL1 &= (~(1 << 7));
	*DANUBE_GPIO_P0_ALTSEL0 &= (~(1 << 7));
#endif
	/* enable ebu */
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	temp_buffer &= (~(3 << PCI_MASTER1_REQ_MASK_2BITS));
	DANUBE_PCI_REG32 (PCI_CR_PC_ARB_REG) = temp_buffer;

	/* enable Config space access */
	DANUBE_PCI_REG32 (0xBE105430) = 0x00000103;

	DANUBE_PCI_REG32 (0xBE105400) &= ~0x2;
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	temp_buffer = DANUBE_PCI_REG32 (0xB700006C);
	temp_buffer &= (~0x00000001);
	DANUBE_PCI_REG32 (0xB700006C) = temp_buffer;
	temp_buffer = DANUBE_PCI_REG32 (0xB700006C);	// do dummy read
#if 1
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	printk ("\nwaiting ebu ownnership");
	while ((DANUBE_PCI_REG32 (PCI_CR_PC_ARB_REG) & 0x400000) != 0) {
		*DANUBE_GPIO_P0_OUT |= ((1 << 7));
		printk ("\n.\n");
	}
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	*DANUBE_GPIO_P0_OUT &= (~(1 << 7));
	printk ("done\n");
#endif
	/* disable ebu */
	DANUBE_PCI_REG32 (0xBE105400) |= 0x2;
	temp_buffer = DANUBE_PCI_REG32 (0xB700006C);
	temp_buffer |= (0x00000001);
	DANUBE_PCI_REG32 (0xB700006C) = temp_buffer;

	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	/* disable Config space access */
	DANUBE_PCI_REG32 (0xBE105430) = 0x01000103;

	temp_buffer = DANUBE_PCI_REG32 (PCI_CR_PC_ARB_REG);
	temp_buffer |= ((3 << PCI_MASTER1_REQ_MASK_2BITS));
#else
	/* enable EBU reqest */
	temp_buffer &= (~(3 << PCI_MASTER1_REQ_MASK_2BITS));
#endif
printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);

	/* enable all external masters request */
	temp_buffer &= (~(3 << PCI_MASTER2_REQ_MASK_2BITS));
	DANUBE_PCI_REG32 (PCI_CR_PC_ARB_REG) = temp_buffer;

	wmb ();

	/* FPI ==> PCI MEM address mapping */
	/* base: 0xb8000000 == > 0x18000000 */
	/* size: 8x4M = 32M */
	DANUBE_PCI_REG32 (PCI_CR_FCI_ADDR_MAP0_REG) = 0x18000000;
	DANUBE_PCI_REG32 (PCI_CR_FCI_ADDR_MAP1_REG) = 0x18400000;
	DANUBE_PCI_REG32 (PCI_CR_FCI_ADDR_MAP2_REG) = 0x18800000;
	DANUBE_PCI_REG32 (PCI_CR_FCI_ADDR_MAP3_REG) = 0x18c00000;
	DANUBE_PCI_REG32 (PCI_CR_FCI_ADDR_MAP4_REG) = 0x19000000;
	DANUBE_PCI_REG32 (PCI_CR_FCI_ADDR_MAP5_REG) = 0x19400000;
	DANUBE_PCI_REG32 (PCI_CR_FCI_ADDR_MAP6_REG) = 0x19800000;
	DANUBE_PCI_REG32 (PCI_CR_FCI_ADDR_MAP7_REG) = 0x19c00000;

	/* FPI ==> PCI IO address mapping */
	/* base: 0xbAE00000 == > 0xbAE00000 */
	/* size: 2M */
	DANUBE_PCI_REG32 (PCI_CR_FCI_ADDR_MAP11hg_REG) = 0xbae00000;

	/* PCI ==> FPI address mapping */
	/* base: 0x0 ==> 0x0 */
	/* size: 32M */
	/* BAR1 32M map to SDR address */
	DANUBE_PCI_REG32 (PCI_CR_BAR11MASK_REG) = 0x0e000008;
	DANUBE_PCI_REG32 (PCI_CR_PCI_ADDR_MAP11_REG) = 0x0;
	DANUBE_PCI_REG32 (PCI_CS_BASE_ADDR1_REG) = 0x0;
#ifdef CONFIG_DANUBE_PCI_HW_SWAP
	/* both TX and RX endian swap are enabled */
	DANUBE_PCI_REG32 (PCI_CR_PCI_EOI_REG) |= 3;
	wmb ();
#endif
	/*TODO: disable BAR2 & BAR3 */
	DANUBE_PCI_REG32 (PCI_CR_BAR12MASK_REG) |= 0x80000000;
	DANUBE_PCI_REG32 (PCI_CR_BAR13MASK_REG) |= 0x80000000;
printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	/*use 8 dw burse length */
	DANUBE_PCI_REG32 (PCI_CR_FCI_BURST_LENGTH_REG) = 0x303;

	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	DANUBE_PCI_REG32 (PCI_CR_PCI_MOD_REG) |= ((1 << 24));
	wmb ();
	//pliu20060613: start
	*DANUBE_GPIO_P1_OUT &= (~(1 << 5));
	wmb ();
	mdelay (1);
	*DANUBE_GPIO_P1_OUT |= (1 << 5);
}

int pcibios_map_irq(const struct pci_dev *dev, u8 slot, u8 pin)
{
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	switch (slot) {
		case 13:
			/* IDSEL = AD29 --> USB Host Controller */
			return INT_NUM_IM1_IRL17;
		case 14:
			/* IDSEL = AD30 --> mini PCI connector */
			return INT_NUM_IM2_IRL14;
		default:
			printk("Warning: no IRQ found for PCI device in slot %d, pin %d\n", slot, pin);
			return 0;
	}
}

#ifdef DANUBE_PCI_TEST
static void dump_cs (u32 base){
	int i;
	u32 data;
	for (i = 0; i < 64; i++) {
		data = DANUBE_PCI_REG32 (((base) | (i * 4)));
#ifdef CONFIG_DANUBE_PCI_HW_SWAP
		data = swab32 (data);
#endif
		printk ("%8x ", data);
		if (i % 8 == 7)
			printk ("\n");
	}
	printk ("\n");
}
#endif

static struct pci_controller danube_pci_controller = {
	.pci_ops = &danube_pci_ops,
	.mem_resource = &pci_mem_resource,
	.io_resource = &pci_io_resource
};

int pcibios_init(void){
#ifdef CONFIG_PCI_AUTO
	struct pci_channel *p;
#endif

	printk ("PCI: Probing PCI hardware on host bus 0.\n");
printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);

	danube_pci_startup ();
printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
#ifdef CONFIG_PCI_AUTO
	/* assign resources */
	busno = 0;
	for (p = mips_pci_channels; p->pci_ops != NULL; p++) {
		busno = pciauto_assign_resources (busno, p) + 1;
	}
#endif
printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
#ifdef DANUBE_PCI_TEST
	dump_cs (EXT_PCI14_CONFIG_SPACE_BASE_ADDR);
#endif
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	DANUBE_PCI_REG32 (PCI_CR_CLK_CTRL_REG) &= (~8);

	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	set_io_port_base(ioremap(DANUBE_PCI_IO_BASE, DANUBE_PCI_IO_SIZE));
	printk("%s:%s[%d]\n", __FILE__, __func__, __LINE__);
	register_pci_controller(&danube_pci_controller);

	return 0;
}

arch_initcall(pcibios_init);