/*------------------------------------------------------------------------------------------*\
\*------------------------------------------------------------------------------------------*/
#include <linux/config.h>
#include <linux/module.h>
#include <linux/init.h>
#include <linux/irq.h>
#include <asm/irq.h>
#include <linux/interrupt.h>
#include <linux/spinlock.h>
#include <linux/kernel_stat.h>
#include <linux/delay.h>

#include <asm/mips-boards/ohio.h>
#include <asm/mach_avm.h>
#include <asm/mach-ohio/hw_clock.h>

static struct _ohio_gpio_power_bits {
    unsigned int Bit_Mask;
    unsigned int Polarity_Mask;
} ohio_gpio_power_bits;

static spinlock_t ohio_power_spinlock;

/*------------------------------------------------------------------------------------------*\
\*------------------------------------------------------------------------------------------*/
int ohio_power_down_init(void) {
    spin_lock_init(&ohio_power_spinlock);
    return 0;
}

/*------------------------------------------------------------------------------------------*\
\*------------------------------------------------------------------------------------------*/
int ohio_put_device_into_power_down(unsigned int Bit) {
    struct _hw_clock *clock = (struct _hw_clock *)OHIO_CLOCK_BASE;
    if(Bit < OHIO_POWER_START_GPIO) {
        unsigned int flags;
        spin_lock_irqsave(&ohio_power_spinlock, flags);
        clock->PDCR.Register |= (1 << Bit);
        spin_unlock_irqrestore(&ohio_power_spinlock, flags);
        return 0;
    }
    if(Bit < OHIO_POWER_END_GPIO) {
        Bit -= OHIO_POWER_START_GPIO;
        ohio_gpio_out_bit(Bit, (ohio_gpio_power_bits.Polarity_Mask & (1 << Bit)) ? 1 : 0);
        return 0;
    }
#if defined(CONFIG_VLYNQ_SUPPORT)
    if(Bit < OHIO_POWER_END_VIRTUAL) {
        return 0;
    }
#endif /*--- #if defined(CONFIG_VLYNQ_SUPPORT) ---*/
    return 1;
}

/*------------------------------------------------------------------------------------------*\
\*------------------------------------------------------------------------------------------*/
int ohio_take_device_out_of_power_down(unsigned int Bit) {
    struct _hw_clock *clock = (struct _hw_clock *)OHIO_CLOCK_BASE;
    if(Bit < OHIO_POWER_START_GPIO) {
        unsigned int flags;
        spin_lock_irqsave(&ohio_power_spinlock, flags);
        clock->PDCR.Register &= ~(1 << Bit);
        spin_unlock_irqrestore(&ohio_power_spinlock, flags);
        return 0;
    }
    if(Bit < OHIO_POWER_END_GPIO) {
        Bit -= OHIO_POWER_START_GPIO;
        ohio_gpio_out_bit(Bit, (ohio_gpio_power_bits.Polarity_Mask & (1 << Bit)) ? 0 : 1);
        return 0;
    }
#if defined(CONFIG_VLYNQ_SUPPORT)
    if(Bit < OHIO_POWER_END_VIRTUAL) {
        return 0;
    }
#endif /*--- #if defined(CONFIG_VLYNQ_SUPPORT) ---*/
    return 1;
}

/*------------------------------------------------------------------------------------------*\
\*------------------------------------------------------------------------------------------*/
int ohio_register_power_down_gpio(unsigned int gpio_pin, unsigned int Polarity) {

    if(gpio_pin >= 32)
        return (unsigned int)-1;

    if(ohio_gpio_power_bits.Bit_Mask & (1 << gpio_pin)) 
        return (unsigned int)-1;

    ohio_gpio_power_bits.Bit_Mask      |= (1 << gpio_pin);
    if(Polarity)
        ohio_gpio_power_bits.Polarity_Mask |=  (1 << Polarity);
    else
        ohio_gpio_power_bits.Polarity_Mask &= ~(1 << Polarity);

    ohio_gpio_ctrl(gpio_pin, GPIO_PIN, GPIO_OUTPUT_PIN);
    return gpio_pin + OHIO_POWER_START_GPIO;
}

/*------------------------------------------------------------------------------------------*\
\*------------------------------------------------------------------------------------------*/
int ohio_release_power_down_gpio(unsigned int Bit) {
    unsigned int gpio_pin = Bit - OHIO_POWER_START_GPIO;
    if(gpio_pin >= 32)
        return (unsigned int)-1;
    if(!(ohio_gpio_power_bits.Bit_Mask & (1 << gpio_pin)))
        return (unsigned int)-1;
    ohio_gpio_power_bits.Bit_Mask &= ~(1 << gpio_pin);
    return 0;
}

/*------------------------------------------------------------------------------------------*\
\*------------------------------------------------------------------------------------------*/
unsigned int ohio_power_down_status(void) {
    struct _hw_clock *clock = (struct _hw_clock *)OHIO_CLOCK_BASE;
    return (unsigned int)clock->PDCR.Register;
}

EXPORT_SYMBOL(ohio_put_device_into_power_down);
EXPORT_SYMBOL(ohio_take_device_out_of_power_down);
EXPORT_SYMBOL(ohio_register_power_down_gpio);
EXPORT_SYMBOL(ohio_release_power_down_gpio);
EXPORT_SYMBOL(ohio_power_down_status);