Subversion Repositories Code-Repo

#include "defines.h"
#include "CONTROLLERS.h"
#include "SNAKE.h"

static SNAKE_DATA *data_p;

void Snake_Init(SNAKE_DATA *data) {
    data_p = data;

    // Set starting point
    data_p->body[0] = (SNAKE_POINT){0,0,7};

    data_p->pos_head = 0;
    data_p->pos_tail = 0;
    data_p->length = 1;
    data_p->level = 1;
    data_p->delay = 800;

    // Generate a starting location for the candy
    data_p->candy_loc = Snake_Generate_Candy();

    // Draw the snake (head)
    Cube_Clear();
    uint32_t index = data_p->pos_head;
    Cube_Set_Pixel(data_p->body[index].z, data_p->body[index].x, data_p->body[index].y, SNAKE_HEAD_COLOR);
    while (index != data_p->pos_tail) {
        if (data_p->length > 1) {
            index = (index == 0) ? CUBE_PIXELS - 1 : index - 1;
            Cube_Set_Pixel(data_p->body[index].z, data_p->body[index].x, data_p->body[index].y, SNAKE_BODY_COLOR);
        }
    }
}

void Snake_Main(void) {
    // Main function, loops and delays while updating the frame every x milliseconds
    Delay_MS(2000);
    while (1) {
        Snake_Update_Frame();
        Delay_MS(data_p->delay);
    }
}

void Snake_Update_Direction(uint8_t p1, uint8_t p2) {
    // Determine the next direction for the snake based off the last button press
    SNAKE_DIRECTION dir;
    dir.value = p1 | p2;
    data_p->last_direction = dir.value;

    SNAKE_POINT point = data_p->body[data_p->pos_head];
    
    if (dir.up) {
        point.z = (point.z == CUBE_LAYER_COUNT - 1) ? 0 : point.z + 1;
    } else if (dir.down) {
        point.z = (point.z == 0) ? CUBE_LAYER_COUNT - 1 : point.z - 1;
    } else if (dir.forward) {
        point.x = (point.x == CUBE_ROW_COUNT - 1) ? 0 : point.x + 1;
    } else if (dir.right) {
        point.y = (point.y == CUBE_COLUMN_COUNT - 1) ? 0 : point.y + 1;
    } else if (dir.backward) {
        point.x = (point.x == 0) ? CUBE_ROW_COUNT - 1 : point.x - 1;
    } else if (dir.left) {
        point.y = (point.y== 0) ? CUBE_COLUMN_COUNT - 1 : point.y - 1;
    }

    data_p->direction = point;

    // Update the overlay with the candy location
    Cube_Overlay_Clear();
    Cube_Overlay_Set_Pixel(data_p->candy_loc.z, data_p->candy_loc.x, data_p->candy_loc.y, SNAKE_CANDY_COLOR);
}

void Snake_Update_Frame(void) {
    uint8_t om_nom_nom = 0;

    // Check if we are moving onto a candy, if so extend body
    if (data_p->direction.x == data_p->candy_loc.x &&
            data_p->direction.y == data_p->candy_loc.y &&
            data_p->direction.z == data_p->candy_loc.z) {
        data_p->pos_head = (data_p->pos_head == CUBE_PIXELS - 1) ? 0 : data_p->pos_head + 1;
        data_p->body[data_p->pos_head] = data_p->direction;
        data_p->length++;
        data_p->candy_loc = Snake_Generate_Candy();
        om_nom_nom = 1;
    }

    // Check if the location that we are moving to is overlapping the body
    uint32_t pos = data_p->pos_tail;
    while (pos != data_p->pos_head) {
        if (data_p->direction.x == data_p->body[pos].x &&
                data_p->direction.y == data_p->body[pos].y &&
                data_p->direction.z == data_p->body[pos].z) {
            // Indicate the overlapping pixel, delay, then return to idle state
            Cube_Overlay_Set_Pixel(data_p->direction.z, data_p->direction.x, data_p->direction.y, SNAKE_HEAD_COLOR);
            Delay_MS(3000);
            Reset_Board(BOARD_MODE_IDLE);
        }
        pos = (pos == CUBE_PIXELS - 1) ? 0 : pos + 1;
    }

    // If we didnt eat a candy, increment the frame to move the body along
    if (!om_nom_nom) {
        data_p->pos_head = (data_p->pos_head == CUBE_PIXELS - 1) ? 0 : data_p->pos_head + 1;
        data_p->pos_tail = (data_p->pos_tail == CUBE_PIXELS - 1) ? 0 : data_p->pos_tail + 1;
        data_p->body[data_p->pos_head] = data_p->direction;
    }

    // Draw updated snake location
    Cube_Clear();
    uint32_t index = data_p->pos_head;
    Cube_Set_Pixel(data_p->body[index].z, data_p->body[index].x, data_p->body[index].y, SNAKE_HEAD_COLOR);
    while (index != data_p->pos_tail) {
        if (data_p->length > 1) {
            index = (index == 0) ? CUBE_PIXELS - 1 : index - 1;
            Cube_Set_Pixel(data_p->body[index].z, data_p->body[index].x, data_p->body[index].y, SNAKE_BODY_COLOR);
        }
    }

    // Determine the next direction to take
    Snake_Update_Direction(data_p->last_direction, 0x0);

    // If we ate a candy, delay for a bit to rest
    if (om_nom_nom) {
        // Increase the level by one, show on controller if necessary
        data_p->level += 1;
        if (data_p->level % SNAKE_LEVEL_STEP == 0) {
            uint8_t tier = data_p->level / SNAKE_LEVEL_STEP;
            Controller_Set_Leds(tier, tier);
        }
        // Decrease the delay between frame updates by 5ms
        data_p->delay -= 5;
        // Clear the watchdog timer to prevent resets in a middle of a game
        ClearWDT();
    }
}

SNAKE_POINT Snake_Generate_Candy(void) {
    // Generates a random position within the cube that doesnt overlap anything
    SNAKE_POINT ret;
    uint32_t x, y, z, brk = 0;
    while(1) {
        x = rand() % 8;
        y = rand() % 8;
        z = rand() % 8;

        if (data_p->length != 1) {
            uint32_t pos = data_p->pos_tail;
            uint32_t overlap = 0;
            // Iterate through the frame till we finish or find an overlap
            while (pos != data_p->pos_head) {
                if (data_p->body[pos].x == x &&
                        data_p->body[pos].y == y &&
                        data_p->body[pos].z == z) {
                    overlap = 1;
                    break;
                } else {
                    pos = (pos == CUBE_PIXELS - 1) ? 0 : pos + 1;
                }
            }
            if (!overlap)
                brk = 1;
        } else {
            uint32_t pos = data_p->pos_tail;
            if (data_p->body[pos].x != x &&
                    data_p->body[pos].y != y &&
                    data_p->body[pos].z != z) {
                brk = 1;
            }
        }

        if (brk)
            break;
    }

    ret.x = x;
    ret.y = y;
    ret.z = z;
    return ret;
}