#include <stdio.h>
#include <ogcsys.h>
#include <ogc/pad.h>

#include "sys.h"
#include "wpad.h"

/* Constants */
#define MAX_WIIMOTES	4

extern u8 shutdown;

void __Wpad_PowerCallback(s32 chan) {
    /* Poweroff console */
    shutdown = 1;
}

void WPad_SetIdleTime(u32 seconds) {
    /*Set idle time for wiimote*/
    WPAD_SetIdleTimeout(seconds);
}

s32 Wpad_Init(void) {
    s32 ret;

    /* Initialize Wiimote subsystem */
    ret = WPAD_Init();
    if (ret < 0)
        return ret;

    /* Set POWER button callback */
    WPAD_SetPowerButtonCallback(__Wpad_PowerCallback);

    return ret;
}

void Wpad_Disconnect(void) {
    u32 cnt;

    /* Disconnect Wiimotes */
    for (cnt = 0; cnt < MAX_WIIMOTES; cnt++)
        WPAD_Disconnect(cnt);

    /* Shutdown Wiimote subsystem */
    WPAD_Shutdown();
}

bool IsWpadConnected() {
    int i = 0;
    u32 test = 0;
    int notconnected = 0;
#ifdef HW_RVL
    for (i = 0; i < 4; i++) {
        if (WPAD_Probe(i, &test) == WPAD_ERR_NO_CONTROLLER) {
            notconnected++;
        }
    }
#endif
    if (notconnected < 4)
        return true;
    else
        return false;
}

u32 ButtonsHold(void) {

    int i;
    u32 buttons = 0;

#ifdef HW_RVL
    WPAD_ScanPads();
#endif
    PAD_ScanPads();

    for (i=3; i >= 0; i--) {
        buttons |= PAD_ButtonsHeld(i);
#ifdef HW_RVL
        buttons |= WPAD_ButtonsHeld(i);
#endif
    }
    return buttons;
}

u32 ButtonsPressed(void) {

    int i;
    u32 buttons = 0;

#ifdef HW_RVL
    WPAD_ScanPads();
#endif
    PAD_ScanPads();

    for (i=3; i >= 0; i--) {
        buttons |= PAD_ButtonsDown(i);
#ifdef HW_RVL
        buttons |= WPAD_ButtonsDown(i);
#endif
    }
    return buttons;

    /*  Don't remove this commented out code it might be useful for checking which buttons were pressed/hold

        if(buttons & (PAD_BUTTON_LEFT | PAD_BUTTON_RIGHT | PAD_BUTTON_DOWN | PAD_BUTTON_UP
                        | PAD_BUTTON_A | PAD_BUTTON_B | PAD_BUTTON_X | PAD_BUTTON_Y | PAD_BUTTON_MENU
                        | PAD_BUTTON_START | WPAD_BUTTON_2 | WPAD_BUTTON_1
                        | WPAD_BUTTON_B | WPAD_BUTTON_A | WPAD_BUTTON_MINUS
                        | WPAD_BUTTON_HOME | WPAD_BUTTON_LEFT | WPAD_BUTTON_RIGHT
                        | WPAD_BUTTON_DOWN | WPAD_BUTTON_UP | WPAD_BUTTON_PLUS
                        | WPAD_NUNCHUK_BUTTON_Z | WPAD_NUNCHUK_BUTTON_C
                        | WPAD_CLASSIC_BUTTON_UP | WPAD_CLASSIC_BUTTON_LEFT
                        | WPAD_CLASSIC_BUTTON_ZR | WPAD_CLASSIC_BUTTON_X
                        | WPAD_CLASSIC_BUTTON_A | WPAD_CLASSIC_BUTTON_Y
                        | WPAD_CLASSIC_BUTTON_B | WPAD_CLASSIC_BUTTON_ZL
                        | WPAD_CLASSIC_BUTTON_FULL_R | WPAD_CLASSIC_BUTTON_PLUS
                        | WPAD_CLASSIC_BUTTON_HOME | WPAD_CLASSIC_BUTTON_MINUS
                        | WPAD_CLASSIC_BUTTON_FULL_L | WPAD_CLASSIC_BUTTON_DOWN
                        | WPAD_CLASSIC_BUTTON_RIGHT | WPAD_GUITAR_HERO_3_BUTTON_STRUM_UP
                        | WPAD_GUITAR_HERO_3_BUTTON_YELLOW | WPAD_GUITAR_HERO_3_BUTTON_GREEN
                        | WPAD_GUITAR_HERO_3_BUTTON_BLUE | WPAD_GUITAR_HERO_3_BUTTON_RED
                        | WPAD_GUITAR_HERO_3_BUTTON_ORANGE | WPAD_GUITAR_HERO_3_BUTTON_PLUS
                        | WPAD_GUITAR_HERO_3_BUTTON_MINUS | WPAD_GUITAR_HERO_3_BUTTON_STRUM_DOWN)
          )
    */

}