master

Dependencies:   mbed

Fork of ESE350-Whack-a-Mole by Eric Berdinis

WhackAMole.cpp

Committer:
shibulal
Date:
2015-10-22
Revision:
3:3c0c008bacb3
Parent:
2:4c79ab84c285

File content as of revision 3:3c0c008bacb3:

#include "mbed.h"
#include "MRF24J40.h"
#include <string>



// RF tranceiver to link with handheld.
MRF24J40 mrf(p11, p12, p13, p14, p21);

// LEDs you can treat these as variables (led2 = 1 will turn led2 on!)
DigitalOut led1(LED1);
DigitalOut led2(LED2);
DigitalOut led3(LED3);
DigitalOut led4(LED4);

// Timer
Timer timer;

// Serial port for showing RX data.
Serial pc(USBTX, USBRX);

// Used for sending and receiving
char txBuffer[128];
char rxBuffer[128];
int rxLen;

//***************** Do not change these methods (please) *****************//

/**
* Receive data from the MRF24J40.
*
* @param data A pointer to a char array to hold the data
* @param maxLength The max amount of data to read.
*/
int rf_receive(char *data, uint8_t maxLength)
{
    uint8_t len = mrf.Receive((uint8_t *)data, maxLength);
    uint8_t header[8]= {1, 8, 0, 0xA1, 0xB2, 0xC3, 0xD4, 0x00};

    if(len > 10) {
        //Remove the header and footer of the message
        for(uint8_t i = 0; i < len-2; i++) {
            if(i<8) {
                //Make sure our header is valid first
                if(data[i] != header[i])
                    return 0;
            } else {
                data[i-8] = data[i];
            }
        }

        //pc.printf("Received: %s length:%d\r\n", data, ((int)len)-10);
    }
    return ((int)len)-10;
}

/**
* Send data to another MRF24J40.
*
* @param data The string to send
* @param maxLength The length of the data to send.
*                  If you are sending a null-terminated string you can pass strlen(data)+1
*/
void rf_send(char *data, uint8_t len)
{
    //We need to prepend the message with a valid ZigBee header
    uint8_t header[8]= {1, 8, 0, 0xA1, 0xB2, 0xC3, 0xD4, 0x00};
    uint8_t *send_buf = (uint8_t *) malloc( sizeof(uint8_t) * (len+8) );

    for(uint8_t i = 0; i < len+8; i++) {
        //prepend the 8-byte header
        send_buf[i] = (i<8) ? header[i] : data[i-8];
    }
    //pc.printf("Sent: %s\r\n", send_buf+8);

    mrf.Send(send_buf, len+8);
    free(send_buf);
}


//***************** You can start coding here *****************//
int main (void)
{
    uint8_t channel = 5;
    
    //Set the Channel. 0 is default, 15 is max
    mrf.SetChannel(channel);
    int action;
    int round_counter;
    int score_counter;
    int best_score=0;
    int level;
    int number_of_nodes=2;
    char input;
    bool confirmed;
    //Start the timer
    timer.start();
    if (number_of_nodes>=3){
        confirmed=false;
        strcpy(txBuffer, "check_3");
        rf_send(txBuffer, strlen(txBuffer)+1);
        pc.printf("Sent: %s\r\n", txBuffer);
        timer.reset();
        while(timer.read_ms()<2000){
            rxLen = rf_receive(rxBuffer, 128); 
            if (rxLen>0){
                pc.printf("Received: %s\r\n", rxBuffer);   
                //strcpy(rxBuffer,"");
                if(!strcmp("confirm_3", rxBuffer)) {
                    confirmed=true;
                    pc.printf("Node 3 alive\r\n");
                    break;
                    }
                } 
            }
        if (!confirmed){
            pc.printf("Node 3 is dead. Repair your hardware!!!");
            return 0;
            }
        }
    if (number_of_nodes>=2){
        confirmed=false;
        strcpy(txBuffer, "check_2");
        rf_send(txBuffer, strlen(txBuffer)+1);
        pc.printf("Sent: %s\r\n", txBuffer);
        timer.reset();
        while(timer.read_ms()<2000){
            rxLen = rf_receive(rxBuffer, 128); 
            if (rxLen>0){
                pc.printf("Received: %s\r\n", rxBuffer);   
                //strcpy(rxBuffer,"");
                if(!strcmp("confirm_2", rxBuffer)) {
                    confirmed=true;
                    pc.printf("Node 2 alive\r\n");
                    break;
                    }
                } 
            }
        if (!confirmed){
            pc.printf("Node 2 is dead. Repair your hardware!!!");
            return 0;
            }
        }
    if (number_of_nodes>=1){
        confirmed=false;
        strcpy(txBuffer, "check_1");
        rf_send(txBuffer, strlen(txBuffer)+1);
        pc.printf("Sent: %s\r\n", txBuffer);
        timer.reset();
        while(timer.read_ms()<2000){
            rxLen = rf_receive(rxBuffer, 128); 
            if (rxLen>0){
                pc.printf("Received: %s\r\n", rxBuffer);   
                //strcpy(rxBuffer,"");
                if(!strcmp("confirm_1", rxBuffer)) {
                    confirmed=true;
                    pc.printf("Node 1 alive\r\n");
                    break;
                    }
                } 
            }
        if (!confirmed){
            pc.printf("Node 1 is dead. Repair your hardware!!!");
            return 0;
            }
        }
    while (true){
        pc.printf("Enter Level (1-3) \r\n");
        input=pc.getc();
        if (input=='1'){
            level=1;
            break;
            }
        else if (input=='2'){
            level=2;
            break;
            }
        else if (input=='3'){
            level=3;
            break;
            }
        }
    while (true){
        pc.printf("Press 's' to start \r\n");
        input=pc.getc();
        if (input=='s'){
            break;
            }
        }
    while(true) {

        pc.printf("start!");
        round_counter=0;
        score_counter=0;
        while(round_counter<10){
            action = rand()%(number_of_nodes+1);
            if(action==0){
                wait_ms(1000/level);
                }
            else if (action==1){
                timer.reset();
                strcpy(txBuffer, "activate_1");
                rf_send(txBuffer, strlen(txBuffer)+1);
                pc.printf("Sent: %s\r\n", txBuffer);
                while(true){
                    rxLen = rf_receive(rxBuffer, 128);
                    if (timer.read_ms()>((1000+(500*(10-round_counter)))/level)){
                        strcpy(txBuffer, "time_out_1");
                        rf_send(txBuffer, strlen(txBuffer)+1);
                        pc.printf("Sent: %s\r\n", txBuffer);
                        score_counter+=((500*(10-round_counter))/level)+2000;
                        break;
                        } 
                    else if (rxLen>0){  
                        pc.printf("Received: %s\r\n", rxBuffer); 
                        if(!strcmp("finished_1", rxBuffer)) { 
                            score_counter+=timer.read_ms();
                            break;
                            }
                        }
                    }
                    round_counter++;
                }
            else if (action==2){
                timer.reset();
                strcpy(txBuffer, "activate_2");
                rf_send(txBuffer, strlen(txBuffer)+1);
                pc.printf("Sent: %s\r\n", txBuffer);
                while(true){
                    rxLen = rf_receive(rxBuffer, 128);
                    if (timer.read_ms()>((1000+(500*(10-round_counter)))/level)){
                        strcpy(txBuffer, "time_out_2");
                        rf_send(txBuffer, strlen(txBuffer)+1);
                        pc.printf("Sent: %s\r\n", txBuffer);
                        score_counter+=((500*(10-round_counter))/level)+2000;
                        break;
                        } 
                    else if (rxLen>0){  
                        pc.printf("Received: %s\r\n", rxBuffer); 
                        if(!strcmp("finished_2", rxBuffer)) { 
                            score_counter+=timer.read_ms();
                            break;
                            }
                        }
                    }
                    round_counter++;
                }
            else if (action ==3){
                timer.reset();
                strcpy(txBuffer, "activate_3");
                rf_send(txBuffer, strlen(txBuffer)+1);
                pc.printf("Sent: %s\r\n", txBuffer);
                while(true){
                    rxLen = rf_receive(rxBuffer, 128);
                    if (timer.read_ms()>((1000+(500*(10-round_counter)))/level)){
                        strcpy(txBuffer, "time_out_3");
                        rf_send(txBuffer, strlen(txBuffer)+1);
                        pc.printf("Sent: %s\r\n", txBuffer);
                        score_counter+=((500*(10-round_counter))/level)+2000;
                        break;
                        } 
                    else if (rxLen>0){  
                        pc.printf("Received: %s\r\n", rxBuffer); 
                        if(!strcmp("finished_3", rxBuffer)) { 
                            score_counter+=timer.read_ms();
                            break;
                            }
                        }
                    }
                    round_counter++;
                }    
        }
        pc.printf("Score: %d \r\n", score_counter);
        if (score_counter<best_score){
            best_score=score_counter;
            }
        while (true){
            pc.printf("New game? (y/n)\r\n");
            input=pc.getc();
            if (input=='y'|| input=='n'){
                break;
                }
            }
        if (input=='n'){
            pc.printf("Best Score %d", best_score);
            break;
            }
        
    }
}