/*------------------------------------------------------------------------------------------*\
\*------------------------------------------------------------------------------------------*/
#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/ar7.h>
#include <asm/mach_avm.h>
#include <asm/mach-ar7/hw_reset.h>

static struct _ar7_gpio_reset_bits {
    unsigned int Bit_Mask;
    unsigned int Polarity_Mask;
} ar7_gpio_reset_bits;

static spinlock_t ar7_reset_spinlock;

/*------------------------------------------------------------------------------------------*\
\*------------------------------------------------------------------------------------------*/
int ar7_reset_init(void) {
    spin_lock_init(&ar7_reset_spinlock);
    return 0;
}

/*------------------------------------------------------------------------------------------*\
\*------------------------------------------------------------------------------------------*/
int ar7_put_device_into_reset(unsigned int Bit) {
    struct _hw_reset *RESET = (struct _hw_reset *)AR7_RESET_BASE;
    if(Bit < AR7_RESET_START_GPIO) {
        unsigned int flags;
        spin_lock_irqsave(&ar7_reset_spinlock, flags);
        RESET->non_reset.Reg &= ~(1 << Bit);
        spin_unlock_irqrestore(&ar7_reset_spinlock, flags);
        return 0;
    }
    if(Bit < AR7_RESET_END_GPIO) {
        Bit -= AR7_RESET_START_GPIO;
        ar7_gpio_out_bit(Bit, (ar7_gpio_reset_bits.Polarity_Mask & (1 << Bit)) ? 1 : 0);
        return 0;
    }
#if defined(CONFIG_VLYNQ_SUPPORT)
    if(Bit < AR7_RESET_END_VIRTUAL) {
        return 0;
    }
#endif /*--- #if defined(CONFIG_VLYNQ_SUPPORT) ---*/
    return 1;
}

/*------------------------------------------------------------------------------------------*\
\*------------------------------------------------------------------------------------------*/
int ar7_take_device_out_of_reset(unsigned int Bit) {
    struct _hw_reset *RESET = (struct _hw_reset *)AR7_RESET_BASE;
    if(Bit < AR7_RESET_START_GPIO) {
        unsigned int flags;
        spin_lock_irqsave(&ar7_reset_spinlock, flags);
        RESET->non_reset.Reg |= (1 << Bit);
        spin_unlock_irqrestore(&ar7_reset_spinlock, flags);
        return 0;
    }
    if(Bit < AR7_RESET_END_GPIO) {
        Bit -= AR7_RESET_START_GPIO;
        ar7_gpio_out_bit(Bit, (ar7_gpio_reset_bits.Polarity_Mask & (1 << Bit)) ? 0 : 1);
        return 0;
    }
#if defined(CONFIG_VLYNQ_SUPPORT)
    if(Bit < AR7_RESET_END_VIRTUAL) {
        return 0;
    }
#endif /*--- #if defined(CONFIG_VLYNQ_SUPPORT) ---*/
    return 1;
}

/*------------------------------------------------------------------------------------------*\
\*------------------------------------------------------------------------------------------*/
int ar7_reset_device(unsigned int Bit, unsigned int msec_delay) {
    unsigned int ret;

    ret = ar7_put_device_into_reset(Bit);
    if(ret) return ret;
    if(msec_delay)
        mdelay(msec_delay);
    ret = ar7_take_device_out_of_reset(Bit);
    if(ret) return ret;
    return 0;
}

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

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

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

    ar7_gpio_reset_bits.Bit_Mask      |= (1 << gpio_pin);
    if(Polarity)
        ar7_gpio_reset_bits.Polarity_Mask |=  (1 << gpio_pin);
    else
        ar7_gpio_reset_bits.Polarity_Mask &= ~(1 << gpio_pin);
        
    ar7_gpio_ctrl(gpio_pin, GPIO_PIN, GPIO_OUTPUT_PIN);

    return gpio_pin + AR7_RESET_START_GPIO;
}

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

/*------------------------------------------------------------------------------------------*\
\*------------------------------------------------------------------------------------------*/
unsigned int ar7_reset_status(void) {
    struct _hw_reset *RESET = (struct _hw_reset *)AR7_RESET_BASE;
    
    return (unsigned int)RESET->reset_status.Bits.cause;
}

EXPORT_SYMBOL(ar7_put_device_into_reset);
EXPORT_SYMBOL(ar7_take_device_out_of_reset);
EXPORT_SYMBOL(ar7_reset_device);
EXPORT_SYMBOL(ar7_register_reset_gpio);
EXPORT_SYMBOL(ar7_release_reset_gpio);
EXPORT_SYMBOL(ar7_reset_status);