File content as of revision 2:663b9aadf00c:

/* Joystick
 
Example code of how to read a joystick
 
https://www.sparkfun.com/products/9032
 
Craig A. Evans
7 March 2015
*/
 
#include "mbed.h"
#include "N5110.h"
 
// change this to alter tolerance of joystick direction
#define DIRECTION_TOLERANCE 0.05

//         VCC,    SCE,   RST,   D/C,   MOSI,  SCLK,   LED
N5110 lcd (PTD3, PTA0 , PTC4 , PTD0 , PTD2 , PTD1 , PTC3);
// Can also power (VCC) directly from VOUT (3.3 V) -
// Can give better performance due to current limitation from GPIO pin 
 
// connections for joystick
DigitalIn button(PTB18);
AnalogIn xPot(PTC11);
AnalogIn yPot(PTC10);

// timer to regularly read the joystick
Ticker pollJoystick;
// Serial for debug
Serial serial(USBTX,USBRX);


 
// create enumerated type (0,1,2,3 etc. for direction)
// could be extended for diagonals etc.
enum DirectionName {
    UP,
    DOWN,
    LEFT,
    RIGHT,
    CENTRE,
    UNKNOWN
};

/// create enumerated type (0,1,2,3 etc. for current direction snake is travelling (not joystick reading))
enum CurrentDirection {
    up,
    down,
    left,
    right,
    centre,
};
CurrentDirection currentDirection = centre; /// intialise direction at beginning
 
// struct for Joystick
typedef struct JoyStick Joystick;
struct JoyStick {
    float x;    // current x value
    float x0;   // 'centred' x value
    float y;    // current y value
    float y0;   // 'centred' y value
    int button; // button state (assume pull-down used, so 1 = pressed, 0 = unpressed)
    DirectionName direction;  // current direction
};
// create struct variable
Joystick joystick;

int printFlag = 0;
 
int pixels[84][48];

int randomX =  rand() % 83 + 1;     // randomX in the range 1 to 81
int randomY =  rand() % 47 + 1;     // randomY in the range 1 to 47
int randomXoddEven = randomX%2; // find out whether odd or even
int randomYoddEven = randomY%2;

// function prototypes
void calibrateJoystick();
void updateJoystick();
void generateFood();
 
int main()
{
    
    calibrateJoystick();  // get centred values of joystick
    pollJoystick.attach(&updateJoystick,1.0/10.0);  // read joystick 10 times per second

    
    int i = 41; // snake head origin x
    int j = 23; // snake head origin y
    
    srand(time(NULL));
      

    lcd.init();
    lcd.drawRect(i,j,1,1,1); // default snake position
    
    generateFood();
 
    while(1) {
        
        if (printFlag) {  // if flag set, clear flag and print joystick values to serial port
            printFlag = 0;
            serial.printf("x = %f y = %f button = %d ",joystick.x,joystick.y,joystick.button);
 
            // check joystick direction
            if (joystick.direction == UP)
                serial.printf(" UP\n");
            if (joystick.direction == DOWN)
                serial.printf(" DOWN\n");
            if (joystick.direction == LEFT)
                serial.printf(" LEFT\n");
            if (joystick.direction == RIGHT)
                serial.printf(" RIGHT\n");
            if (joystick.direction == CENTRE)
                serial.printf(" CENTRE\n");
            if (joystick.direction == UNKNOWN)
                serial.printf(" Unsupported direction\n");
                
            if (joystick.direction == LEFT) {
                if (currentDirection != right) { // user can't turn opposite way
                currentDirection = left; // direct enum updated
                }
                while (currentDirection == left) { // enters left movement loop
                lcd.clear();
                i -= 2; // shifts two pixels (snake thickness)
                lcd.drawRect(i,j,1,1,1); // snake head
                lcd.drawRect(randomX,randomY,1,1,1); // food
                if (i == randomX && j == randomY) {
                    randomX =  rand() % 83 + 1;     // randomX in the range 1 to 81
                    randomY =  rand() % 47 + 1;     // randomY in the range 1 to 47
                    randomXoddEven = randomX%2; // find out whether odd or even
                    randomYoddEven = randomY%2;
                    generateFood();  
                    }
                wait(1.0);
                if (joystick.direction == UP) {
                    currentDirection = up; // direction enum updated, this while loop ends, up while loop starts
                    }
                if (joystick.direction == DOWN) {
                    currentDirection = down; // direction enum updated, this while loop ends, down while loop starts
                    }
                }
            }
            if (joystick.direction == RIGHT) {
                if (currentDirection != left) {
                currentDirection = right;
                }
                while (currentDirection == right) {
                lcd.clear();
                i += 2;
                lcd.drawRect(i,j,1,1,1); // snake head
                lcd.drawRect(randomX,randomY,1,1,1); // food
                if (i == randomX && j == randomY) {
                    randomX =  rand() % 83 + 1;     // randomX in the range 1 to 81
                    randomY =  rand() % 47 + 1;     // randomY in the range 1 to 47
                    randomXoddEven = randomX%2; // find out whether odd or even
                    randomYoddEven = randomY%2;
                    generateFood();  
                    }
                wait(1.0);
                if (joystick.direction == UP) {
                    currentDirection = up; // direction enum updated, this while loop ends, up while loop starts
                    }
                if (joystick.direction == DOWN) {
                    currentDirection = down; // direction enum updated, this while loop ends, down while loop starts
                    }
                }
            }
            if (joystick.direction == UP) {
                if (currentDirection != down) {
                currentDirection = up; 
                }
                while (currentDirection == up) {
                lcd.clear();
                j -= 2;
                lcd.drawRect(i,j,1,1,1); // snake head
                lcd.drawRect(randomX,randomY,1,1,1); // food
                if (i == randomX && j == randomY) {
                    randomX =  rand() % 83 + 1;     // randomX in the range 1 to 81
                    randomY =  rand() % 47 + 1;     // randomY in the range 1 to 47
                    randomXoddEven = randomX%2; // find out whether odd or even
                    randomYoddEven = randomY%2;
                    generateFood();  
                    }
                wait(1.0);
                if (joystick.direction == LEFT) {
                    currentDirection = left; // direction enum updated, this while loop ends, left while loop starts
                    }
                if (joystick.direction == RIGHT) {
                    currentDirection = right; // direction enum updated, this while loop ends, right while loop starts
                    }
                }
            }
            if (joystick.direction == DOWN) {
                if (currentDirection != up) {
                currentDirection = down;
                }
                while (currentDirection == down) { 
                lcd.clear();
                j += 2;
                lcd.drawRect(i,j,1,1,1); // snake head
                lcd.drawRect(randomX,randomY,1,1,1); // food
                if (i == randomX && j == randomY) {
                    randomX =  rand() % 83 + 1;     // randomX in the range 1 to 81
                    randomY =  rand() % 47 + 1;     // randomY in the range 1 to 47
                    randomXoddEven = randomX%2; // find out whether odd or even
                    randomYoddEven = randomY%2;
                    generateFood();  
                    }
                wait(1.0);
                if (joystick.direction == LEFT) {
                    currentDirection = left; // direction enum updated, this while loop ends, left while loop starts
                    }
                if (joystick.direction == RIGHT) {
                    currentDirection = right; // direction enum updated, this while loop ends, right while loop starts
                    } 
                }
            }
            
                
 
        }
                // put the MCU to sleep until an interrupt wakes it up
                sleep();
    }
}
 
// read default positions of the joystick to calibrate later readings
void calibrateJoystick()
{
    button.mode(PullDown);
    // must not move during calibration
    joystick.x0 = xPot;  // initial positions in the range 0.0 to 1.0 (0.5 if centred exactly)
    joystick.y0 = yPot;
}
void updateJoystick()
{
    // read current joystick values relative to calibrated values (in range -0.5 to 0.5, 0.0 is centred)
    joystick.x = xPot - joystick.x0;
    joystick.y = yPot - joystick.y0;
    // read button state
    joystick.button = button;
 
    // calculate direction depending on x,y values
    // tolerance allows a little lee-way in case joystick not exactly in the stated direction
    if ( fabs(joystick.y) < DIRECTION_TOLERANCE && fabs(joystick.x) < DIRECTION_TOLERANCE) {
        joystick.direction = CENTRE;
    } else if ( joystick.y > DIRECTION_TOLERANCE && fabs(joystick.x) < DIRECTION_TOLERANCE) {
        joystick.direction = UP;
    } else if ( joystick.y < DIRECTION_TOLERANCE && fabs(joystick.x) < DIRECTION_TOLERANCE) {
        joystick.direction = DOWN;
    } else if ( joystick.x > DIRECTION_TOLERANCE && fabs(joystick.y) < DIRECTION_TOLERANCE) {
        joystick.direction = RIGHT;
    } else if ( joystick.x < DIRECTION_TOLERANCE && fabs(joystick.y) < DIRECTION_TOLERANCE) {
        joystick.direction = LEFT;
    } else {
        joystick.direction = UNKNOWN;
    }
 
    // set flag for printing
    printFlag = 1;
}

void generateFood()
{
    while (randomXoddEven ==0 || randomYoddEven ==0 || lcd.getPixel(randomX,randomY) >= 1) { // do while x or y is even or pixel is on
        
        randomX =  rand() % 83 + 1;     // randomX in the range 1 to 81
        randomY =  rand() % 47 + 1;     // randomY in the range 1 to 47
   
        serial.printf("X = %i\n",randomX); // debug
        serial.printf("Y = %i\n\n",randomY);
        
        randomXoddEven = randomX%2; // find out whether odd or even
        randomYoddEven = randomY%2;  
        }
        
        lcd.drawRect(randomX,randomY,1,1,1);

}