/*
 *  Planex MZK-W04NU board support
 *
 *  Copyright (C) 2009 Gabor Juhos <juhosg@openwrt.org>
 *
 *  This program is free software; you can redistribute it and/or modify it
 *  under the terms of the GNU General Public License version 2 as published
 *  by the Free Software Foundation.
 */

#include <linux/mtd/mtd.h>
#include <linux/mtd/partitions.h>

#include <asm/mach-ar71xx/ar71xx.h>

#include "machtype.h"
#include "devices.h"
#include "dev-ar913x-wmac.h"
#include "dev-gpio-buttons.h"
#include "dev-leds-gpio.h"
#include "dev-m25p80.h"
#include "dev-usb.h"

#define MZK_W04NU_GPIO_LED_USB		0
#define MZK_W04NU_GPIO_LED_STATUS	1
#define MZK_W04NU_GPIO_LED_WPS		3
#define MZK_W04NU_GPIO_LED_WLAN		6
#define MZK_W04NU_GPIO_LED_AP		15
#define MZK_W04NU_GPIO_LED_ROUTER	16

#define MZK_W04NU_GPIO_BTN_APROUTER	5
#define MZK_W04NU_GPIO_BTN_WPS		12
#define MZK_W04NU_GPIO_BTN_RESET	21

#define MZK_W04NU_BUTTONS_POLL_INTERVAL	20

#ifdef CONFIG_MTD_PARTITIONS
static struct mtd_partition mzk_w04nu_partitions[] = {
	{
		.name		= "u-boot",
		.offset		= 0,
		.size		= 0x040000,
		.mask_flags	= MTD_WRITEABLE,
	} , {
		.name		= "u-boot-env",
		.offset		= 0x040000,
		.size		= 0x010000,
	} , {
		.name		= "kernel",
		.offset		= 0x050000,
		.size		= 0x160000,
	} , {
		.name		= "rootfs",
		.offset		= 0x1b0000,
		.size		= 0x630000,
	} , {
		.name		= "art",
		.offset		= 0x7e0000,
		.size		= 0x020000,
		.mask_flags	= MTD_WRITEABLE,
	} , {
		.name		= "firmware",
		.offset		= 0x050000,
		.size		= 0x790000,
	}
};
#endif /* CONFIG_MTD_PARTITIONS */

static struct flash_platform_data mzk_w04nu_flash_data = {
#ifdef CONFIG_MTD_PARTITIONS
        .parts          = mzk_w04nu_partitions,
        .nr_parts       = ARRAY_SIZE(mzk_w04nu_partitions),
#endif
};

static struct gpio_led mzk_w04nu_leds_gpio[] __initdata = {
	{
		.name		= "mzk-w04nu:green:status",
		.gpio		= MZK_W04NU_GPIO_LED_STATUS,
		.active_low	= 1,
	}, {
		.name		= "mzk-w04nu:blue:wps",
		.gpio		= MZK_W04NU_GPIO_LED_WPS,
		.active_low	= 1,
	}, {
		.name		= "mzk-w04nu:green:wlan",
		.gpio		= MZK_W04NU_GPIO_LED_WLAN,
		.active_low	= 1,
	}, {
		.name		= "mzk-w04nu:green:usb",
		.gpio		= MZK_W04NU_GPIO_LED_USB,
		.active_low	= 1,
	}, {
		.name		= "mzk-w04nu:green:ap",
		.gpio		= MZK_W04NU_GPIO_LED_AP,
		.active_low	= 1,
	}, {
		.name		= "mzk-w04nu:green:router",
		.gpio		= MZK_W04NU_GPIO_LED_ROUTER,
		.active_low	= 1,
	}
};

static struct gpio_button mzk_w04nu_gpio_buttons[] __initdata = {
	{
		.desc		= "reset",
		.type		= EV_KEY,
		.code		= BTN_0,
		.threshold	= 3,
		.gpio		= MZK_W04NU_GPIO_BTN_RESET,
		.active_low	= 1,
	}, {
		.desc		= "wps",
		.type		= EV_KEY,
		.code		= BTN_1,
		.threshold	= 3,
		.gpio		= MZK_W04NU_GPIO_BTN_WPS,
		.active_low	= 1,
	}, {
		.desc		= "aprouter",
		.type		= EV_KEY,
		.code		= BTN_2,
		.threshold	= 3,
		.gpio		= MZK_W04NU_GPIO_BTN_APROUTER,
		.active_low	= 0,
	}
};

#define MZK_W04NU_WAN_PHYMASK	BIT(4)
#define MZK_W04NU_MDIO_MASK	(~MZK_W04NU_WAN_PHYMASK)

static void __init mzk_w04nu_setup(void)
{
	u8 *eeprom = (u8 *) KSEG1ADDR(0x1fff1000);

	ar71xx_set_mac_base(eeprom);

	ar71xx_add_device_mdio(MZK_W04NU_MDIO_MASK);

	ar71xx_eth0_data.phy_if_mode = PHY_INTERFACE_MODE_RMII;
	ar71xx_eth0_data.speed = SPEED_100;
	ar71xx_eth0_data.duplex = DUPLEX_FULL;
	ar71xx_eth0_data.has_ar8216 = 1;

	ar71xx_eth1_data.phy_if_mode = PHY_INTERFACE_MODE_RMII;
	ar71xx_eth1_data.phy_mask = MZK_W04NU_WAN_PHYMASK;

	ar71xx_add_device_eth(0);
	ar71xx_add_device_eth(1);

	ar71xx_add_device_m25p80(&mzk_w04nu_flash_data);

	ar71xx_add_device_leds_gpio(-1, ARRAY_SIZE(mzk_w04nu_leds_gpio),
					mzk_w04nu_leds_gpio);

	ar71xx_add_device_gpio_buttons(-1, MZK_W04NU_BUTTONS_POLL_INTERVAL,
					ARRAY_SIZE(mzk_w04nu_gpio_buttons),
					mzk_w04nu_gpio_buttons);
	ar71xx_add_device_usb();

	ar913x_add_device_wmac(eeprom, NULL);
}

MIPS_MACHINE(AR71XX_MACH_MZK_W04NU, "MZK-W04NU", "Planex MZK-W04NU",
	     mzk_w04nu_setup);
