Wii Nunchuk module with nasty hacks from debugging on a FRDM-KL25Z

Fork of WiiChuk_compat by Greg Brush

WiiChuk_compat.cpp

Committer:
eisd
Date:
2017-01-04
Revision:
4:bf058247fb2f
Parent:
2:566662f892b1

File content as of revision 4:bf058247fb2f:

#include "WiiChuk_compat.hpp"



WiiChuck::WiiChuck(PinName data, PinName clk, Stream &_pc):_i2c(data, clk), pc(_pc) {
    _i2c.frequency(50000); // should work at 100000 too
    Error = true;

    // initialize Wii extension (nunchuk) based on "new way" initialization
    // http://wiibrew.org/wiki/Wiimote#The_New_Way
    // also disables need for decryption
    
    unsigned char cmd[] = {0xF0, 0x55};
    if (_i2c.write(NUNCHUCK_ADDR, (const char*)cmd, sizeof(cmd), true) == I2C_ACK) {
        Error = false;
    }
    
    cmd[0] = 0xfb; cmd[1]= 0x00;
    //pc.printf("%i\r\n", _i2c.write(NUNCHUCK_ADDR, (const char*)cmd, sizeof(cmd), true));
  
    _oldC = 0;
    _oldZ = 0;
}

bool WiiChuck::Read(uint8_t* joyX, uint8_t* joyY, uint16_t* accX, uint16_t* accY, uint16_t* accZ, uint8_t* buttonC, uint8_t* buttonZ) {

    //int i;
    char readBuf[NUNCHUCK_READLEN];
    
    if (Error) {
        return false;
    }
    
    const unsigned char cmd[] = {0x00};
    if (_i2c.write(NUNCHUCK_ADDR, (const char*)cmd, sizeof(cmd)) == I2C_ACK) {
        wait(I2C_READ_DELAY);
        if (_i2c.read(NUNCHUCK_ADDR, readBuf, sizeof(readBuf)) == I2C_ACK) {
            //init values
            //n->X = 0; n->Y = 0; n->aX = 0; n->aY = 0; n->aZ = 0; n->C = 0; n->Z = 0;
            *joyX = 0; *joyY = 0; *accX = 0; *accY = 0; *accZ = 0; *buttonC = 0; *buttonZ = 0;
   /*
            pc.printf("\r\n");
            //  Decrypting not required with new initialization
            for (int i = 0; i < NUNCHUCK_READLEN; ++i) {
//                readBuf[i] = (readBuf[i] ^ 0x17) + 0x17;
                pc.printf("%i ", readBuf[i]);
            }
            pc.printf("\r\n");
     */       
            
            *joyX = readBuf[Joy_X];
            *joyY = readBuf[Joy_Y];
            *accX = readBuf[Acc_X] << 2;
            *accY = readBuf[Acc_Y] << 2;
            *accZ = readBuf[Acc_Z] << 2;

            if (readBuf[Button] & 0x01) {
                *buttonZ = 0;
            } else {
                *buttonZ = 1;
            }
            if (readBuf[Button] & 0x02) {
                *buttonC = 0;
            } else {
                *buttonC = 1;
            }
            if (readBuf[Button] & 0x04) accX += 2;
            if (readBuf[Button] & 0x08) accX += 1;
            if (readBuf[Button] & 0x10) accY += 2;
            if (readBuf[Button] & 0x20) accY += 1;
            if (readBuf[Button] & 0x40) accZ += 2;
            if (readBuf[Button] & 0x80) accZ += 1;
            return true;
        }
        else
        {
        pc.printf("no read ack\r\n");
            return false;
        }
    } else {
        pc.printf("no write ack\r\n");
        return false;
    }

}

void WiiChuck::start()
{
    _getValues.attach(this, &WiiChuck::getValues,0.2);
}
void WiiChuck::stop()
{
     _getValues.detach();
}

void WiiChuck::getValues()
{
    uint8_t joyX = 0, joyY = 0;
    uint16_t accX = 0, accY = 0, accZ = 0;
    uint8_t buttonC = 0, buttonZ = 0;
     
    bool read = Read(&joyX,&joyY,&accX,&accY,&accZ,&buttonC,&buttonZ);
  
    if(read)
    {
        //analyse
        if(_oldC == 0 && buttonC == 1)
        {
            _oldC = 1;
            _callback_input(BUTTON_CANCEL_VALUE);
            return;
        }
        else
        {
            _oldC = buttonC;
        }
        
        //analyse
        if(_oldZ == 0 && buttonZ == 1)
        {
            _oldZ = 1;
            _callback_input(BUTTON_OK_VALUE);
            return;
        }
        else
        {
            _oldZ = buttonZ;
        }
        
        if(joyY>160)
        {
            _callback_input(BUTTON_VOLUME_PLUS);
            return;
        }
        if(joyY<80)
        {
        
            _callback_input(BUTTON_VOLUME_MINUS);
            return;
        }
        if(joyX>160)
        {
            _callback_input(BUTTON_NEXT_VALUE);
            return;
        }
        if(joyX<80)
        {
        
            _callback_input(BUTTON_PREV_VALUE);
            return;
        }
        
    }
}
void WiiChuck::attach(pt2Func function)
{
    _callback_input = function;
}