Momentum-Apps/scorched_tanks_game_app.c

#include <furi.h>
#include <gui/gui.h>
#include <input/input.h>
#include <stdlib.h>
#include <math.h>
#include <notification/notification.h>
#include <notification/notification_messages.h>

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64
#define PLAYER_INIT_LOCATION_X 20
#define PLAYER_INIT_AIM 45
#define PLAYER_INIT_POWER 50
#define ENEMY_INIT_LOCATION_X 108
#define TANK_BARREL_LENGTH 8
#define GRAVITY_FORCE 32
#define MIN_GROUND_HEIGHT 35
#define MAX_GROUND_HEIGHT 55
#define MAX_FIRE_POWER 100
#define MIN_FIRE_POWER 0
#define TANK_COLLIDER_SIZE 3

// That's a filthy workaround but sin(player.aimAngle) breaks it all... If you're able to fix it, please do create a PR!
double scorched_tanks_sin[91] = {
    0.000,  -0.017, -0.035, -0.052, -0.070, -0.087, -0.105, -0.122, -0.139, -0.156, -0.174, -0.191,
    -0.208, -0.225, -0.242, -0.259, -0.276, -0.292, -0.309, -0.326, -0.342, -0.358, -0.375, -0.391,
    -0.407, -0.423, -0.438, -0.454, -0.469, -0.485, -0.500, -0.515, -0.530, -0.545, -0.559, -0.574,
    -0.588, -0.602, -0.616, -0.629, -0.643, -0.656, -0.669, -0.682, -0.695, -0.707, -0.719, -0.731,
    -0.743, -0.755, -0.766, -0.777, -0.788, -0.799, -0.809, -0.819, -0.829, -0.839, -0.848, -0.857,
    -0.866, -0.875, -0.883, -0.891, -0.899, -0.906, -0.914, -0.921, -0.927, -0.934, -0.940, -0.946,
    -0.951, -0.956, -0.961, -0.966, -0.970, -0.974, -0.978, -0.982, -0.985, -0.988, -0.990, -0.993,
    -0.995, -0.996, -0.998, -0.999, -0.999, -1.000, -1.000};
double scorched_tanks_cos[91] = {
    1.000, 1.000, 0.999, 0.999, 0.998, 0.996, 0.995, 0.993, 0.990, 0.988, 0.985, 0.982, 0.978,
    0.974, 0.970, 0.966, 0.961, 0.956, 0.951, 0.946, 0.940, 0.934, 0.927, 0.921, 0.914, 0.906,
    0.899, 0.891, 0.883, 0.875, 0.866, 0.857, 0.848, 0.839, 0.829, 0.819, 0.809, 0.799, 0.788,
    0.777, 0.766, 0.755, 0.743, 0.731, 0.719, 0.707, 0.695, 0.682, 0.669, 0.656, 0.643, 0.629,
    0.616, 0.602, 0.588, 0.574, 0.559, 0.545, 0.530, 0.515, 0.500, 0.485, 0.469, 0.454, 0.438,
    0.423, 0.407, 0.391, 0.375, 0.358, 0.342, 0.326, 0.309, 0.292, 0.276, 0.259, 0.242, 0.225,
    0.208, 0.191, 0.174, 0.156, 0.139, 0.122, 0.105, 0.087, 0.070, 0.052, 0.035, 0.017, 0.000};
double scorched_tanks_tan[91] = {
    0.000,   -0.017,  -0.035,    -0.052, -0.070, -0.087, -0.105, -0.123, -0.141,  -0.158,  -0.176,
    -0.194,  -0.213,  -0.231,    -0.249, -0.268, -0.287, -0.306, -0.325, -0.344,  -0.364,  -0.384,
    -0.404,  -0.424,  -0.445,    -0.466, -0.488, -0.510, -0.532, -0.554, -0.577,  -0.601,  -0.625,
    -0.649,  -0.674,  -0.700,    -0.727, -0.754, -0.781, -0.810, -0.839, -0.869,  -0.900,  -0.932,
    -0.966,  -1.000,  -1.036,    -1.072, -1.111, -1.150, -1.192, -1.235, -1.280,  -1.327,  -1.376,
    -1.428,  -1.483,  -1.540,    -1.600, -1.664, -1.732, -1.804, -1.881, -1.963,  -2.050,  -2.144,
    -2.246,  -2.356,  -2.475,    -2.605, -2.747, -2.904, -3.078, -3.271, -3.487,  -3.732,  -4.011,
    -4.331,  -4.704,  -5.144,    -5.671, -6.313, -7.115, -8.144, -9.513, -11.429, -14.298, -19.077,
    -28.627, -57.254, -90747.269};

typedef struct {
    //    +-----x
    //    |
    //    |
    //    y
    uint8_t x;
    uint8_t y;
} Point;

typedef struct {
    unsigned char locationX;
    unsigned char hp;
    int aimAngle;
    unsigned char firePower;
    bool isShooting;
} Tank;

typedef struct {
    Point ground[SCREEN_WIDTH];
    Tank player;
    Tank enemy;
    bool isPlayerTurn;
    unsigned char trajectoryY[SCREEN_WIDTH];
    unsigned char trajectoryAnimationStep;
    Point bulletPosition;
} Game;

typedef enum {
    EventTypeTick,
    EventTypeKey,
} EventType;

typedef struct {
    EventType type;
    InputEvent input;
} ScorchedTanksEvent;

int scorched_tanks_random(int min, int max) {
    return min + rand() % ((max + 1) - min);
}

void scorched_tanks_generate_ground(Game* game_state) {
    auto lastHeight = 45;

    for(unsigned char a = 0; a < SCREEN_WIDTH; a++) {
        auto diffHeight = scorched_tanks_random(-2, 3);
        auto changeLength = scorched_tanks_random(1, 6);

        if(diffHeight == 0) {
            changeLength = 1;
        }

        for(int b = 0; b < changeLength; b++) {
            if(a + b < SCREEN_WIDTH) {
                auto index = a + b;
                auto newPoint = lastHeight + diffHeight;
                newPoint = newPoint < MIN_GROUND_HEIGHT ? MIN_GROUND_HEIGHT : newPoint;
                newPoint = newPoint > MAX_GROUND_HEIGHT ? MAX_GROUND_HEIGHT : newPoint;
                game_state->ground[index].x = index;
                game_state->ground[index].y = newPoint;
                lastHeight = newPoint;
            } else {
                a += b;
                break;
            }
        }

        a += changeLength - 1;
    }
}

void scorched_tanks_init_game(Game* game_state) {
    game_state->player.locationX = PLAYER_INIT_LOCATION_X;
    game_state->player.aimAngle = PLAYER_INIT_AIM;
    game_state->player.firePower = PLAYER_INIT_POWER;
    game_state->enemy.locationX = ENEMY_INIT_LOCATION_X;

    for (int x = 0; x < SCREEN_WIDTH; x++)
    {
        game_state->trajectoryY[x] = 0;
    }

    scorched_tanks_generate_ground(game_state);
}

void scorched_tanks_calculate_trajectory(Game* game_state) {
    if(game_state->player.isShooting) {
        int x0 = game_state->player.locationX;
        int y0 = game_state->ground[game_state->player.locationX].y - TANK_COLLIDER_SIZE;
        int v0 = game_state->player.firePower;
        int g = GRAVITY_FORCE;
        int angle = game_state->player.aimAngle;

        if(x0 + game_state->trajectoryAnimationStep > SCREEN_WIDTH ||
           game_state->bulletPosition.x > SCREEN_WIDTH ||
           game_state->bulletPosition.y > game_state->ground[game_state->bulletPosition.x].y) { 
                game_state->player.isShooting = false;
                game_state->bulletPosition.x = 0;
                game_state->bulletPosition.y = 0;
                return;
        }

        unsigned char distanceToEnemyX = game_state->enemy.locationX - game_state->bulletPosition.x;
        unsigned char distanceToEnemyY = game_state->ground[game_state->enemy.locationX].y - TANK_COLLIDER_SIZE - game_state->bulletPosition.y;
        int totalDistanceToEnemy = sqrt(distanceToEnemyX*distanceToEnemyX+distanceToEnemyY*distanceToEnemyY);

        if (totalDistanceToEnemy <= TANK_COLLIDER_SIZE)
        {
            game_state->player.isShooting = false;
            scorched_tanks_init_game(game_state);
            return;
        }

        auto x = game_state->trajectoryAnimationStep;
        auto y = y0 + x * scorched_tanks_tan[angle] -
                 g * x * x / -(2 * v0 * v0 * scorched_tanks_cos[angle] * scorched_tanks_cos[angle]);

        x += x0;

        if(x % 4 == 0) {
            game_state->trajectoryY[x] = y;
        }

        game_state->bulletPosition.x = x;
        game_state->bulletPosition.y = y;

        game_state->trajectoryAnimationStep++;
    }
}

static void scorched_tanks_render_callback(Canvas* const canvas, void* ctx) {
    const Game* game_state = acquire_mutex((ValueMutex*)ctx, 25);

    if(game_state == NULL) {
        return;
    }

    canvas_draw_frame(canvas, 0, 0, SCREEN_WIDTH, SCREEN_HEIGHT);

    canvas_set_color(canvas, ColorBlack);

    if(game_state->player.isShooting) {
        canvas_draw_dot(canvas, game_state->bulletPosition.x, game_state->bulletPosition.y);
    }

    for(int a = 1; a < SCREEN_WIDTH; a++) {
        canvas_draw_line(
            canvas,
            game_state->ground[a - 1].x,
            game_state->ground[a - 1].y,
            game_state->ground[a].x,
            game_state->ground[a].y);

        if(game_state->trajectoryY[a] != 0) {
            canvas_draw_dot(canvas, a, game_state->trajectoryY[a]);
        }
    }

    canvas_draw_disc(
        canvas,
        game_state->enemy.locationX,
        game_state->ground[game_state->enemy.locationX].y - TANK_COLLIDER_SIZE,
        3);

    canvas_draw_circle(
        canvas,
        game_state->player.locationX,
        game_state->ground[game_state->player.locationX].y - TANK_COLLIDER_SIZE,
        3);

    auto aimX1 = game_state->player.locationX;
    auto aimY1 = game_state->ground[game_state->player.locationX].y - TANK_COLLIDER_SIZE;

    double sinFromAngle = scorched_tanks_sin[game_state->player.aimAngle];
    double cosFromAngle = scorched_tanks_cos[game_state->player.aimAngle];
    int aimX2 = aimX1 + TANK_BARREL_LENGTH * cosFromAngle;
    int aimY2 = aimY1 + TANK_BARREL_LENGTH * sinFromAngle;

    canvas_draw_line(canvas, aimX1, aimY1, aimX2, aimY2);

    canvas_set_font(canvas, FontSecondary);
    canvas_draw_str(canvas, 55, 10, "Scorched Tanks");

    char buffer[12];
    snprintf(buffer, sizeof(buffer), "a: %u", game_state->player.aimAngle);
    canvas_draw_str(canvas, 2, 10, buffer);

    snprintf(buffer, sizeof(buffer), "p: %u", game_state->player.firePower);
    canvas_draw_str(canvas, 27, 10, buffer);

    release_mutex((ValueMutex*)ctx, game_state);
}

static void scorched_tanks_input_callback(InputEvent* input_event, FuriMessageQueue* event_queue) {
    furi_assert(event_queue);

    ScorchedTanksEvent event = {.type = EventTypeKey, .input = *input_event};
    furi_message_queue_put(event_queue, &event, FuriWaitForever);
}

static void scorched_tanks_update_timer_callback(FuriMessageQueue* event_queue) {
    furi_assert(event_queue);

    ScorchedTanksEvent event = {.type = EventTypeTick};
    furi_message_queue_put(event_queue, &event, 0);
}

static void scorched_tanks_increase_power(Game* game_state)
{
    if (game_state->player.firePower < MAX_FIRE_POWER && !game_state->player.isShooting)
    {
        game_state->player.firePower++;
    }
}

static void scorched_tanks_decrease_power(Game* game_state)
{
    if (game_state->player.firePower > MIN_FIRE_POWER && !game_state->player.isShooting)
    {
        game_state->player.firePower--;
    }
}

static void scorched_tanks_aim_up(Game* game_state) {
    if(game_state->player.aimAngle < 90 && !game_state->player.isShooting) {
        game_state->player.aimAngle++;
    }
}

static void scorched_tanks_aim_down(Game* game_state) {
    if(game_state->player.aimAngle > 0 && !game_state->player.isShooting) {
        game_state->player.aimAngle--;
    }
}

const NotificationSequence sequence_long_vibro = {
    &message_vibro_on,
    &message_delay_500,
    &message_vibro_off,
    NULL,
};

static void scorched_tanks_fire(Game* game_state) {
    if(!game_state->player.isShooting) {
        game_state->bulletPosition.x = game_state->player.locationX;
        game_state->bulletPosition.y = game_state->ground[game_state->player.locationX].y - TANK_COLLIDER_SIZE;
        game_state->trajectoryAnimationStep = 0;

        for(int x = 0; x < SCREEN_WIDTH; x++) {
            game_state->trajectoryY[x] = 0;
        }

        game_state->player.isShooting = true;

        NotificationApp* notification = furi_record_open("notification");
        notification_message(notification, &sequence_long_vibro);
        notification_message(notification, &sequence_blink_white_100);
        furi_record_close("notification");
    }
}

int32_t scorched_tanks_game_app(void* p) {
    UNUSED(p);
    srand(DWT->CYCCNT);

    FuriMessageQueue* event_queue = furi_message_queue_alloc(8, sizeof(ScorchedTanksEvent));

    Game* game_state = malloc(sizeof(Game));
    scorched_tanks_init_game(game_state);

    ValueMutex state_mutex;
    if(!init_mutex(&state_mutex, game_state, sizeof(ScorchedTanksEvent))) {
        FURI_LOG_E("ScorchedTanks", "cannot create mutex\r\n");
        free(game_state);
        return 255;
    }

    ViewPort* view_port = view_port_alloc();
    view_port_draw_callback_set(view_port, scorched_tanks_render_callback, &state_mutex);
    view_port_input_callback_set(view_port, scorched_tanks_input_callback, event_queue);

    FuriTimer* timer =
        furi_timer_alloc(scorched_tanks_update_timer_callback, FuriTimerTypePeriodic, event_queue);
    furi_timer_start(timer, 2000);

    // Open GUI and register view_port
    Gui* gui = furi_record_open(RECORD_GUI);
    gui_add_view_port(gui, view_port, GuiLayerFullscreen);

    ScorchedTanksEvent event;
    for(bool processing = true; processing;) {
        FuriStatus event_status = furi_message_queue_get(event_queue, &event, 50);

        if(event.type == EventTypeKey) { // && game->isPlayerTurn
            if(event.input.type == InputTypeRepeat || event.input.type == InputTypeShort) {
                switch(event.input.key) {
                case InputKeyUp:
                    scorched_tanks_aim_up(game_state);
                    break;
                case InputKeyDown:
                    scorched_tanks_aim_down(game_state);
                    break;
                case InputKeyRight:
                    scorched_tanks_increase_power(game_state);
                    break;
                case InputKeyLeft:
                    scorched_tanks_decrease_power(game_state);
                    break;
                case InputKeyOk:
                    scorched_tanks_fire(game_state);
                    break;
                case InputKeyBack:
                    processing = false;
                    break;
                }
            }
        } else if(event.type == EventTypeTick) {
            scorched_tanks_calculate_trajectory(game_state);
        }

        view_port_update(view_port);
        release_mutex(&state_mutex, game_state);
    }

    furi_timer_free(timer);
    view_port_enabled_set(view_port, false);
    gui_remove_view_port(gui, view_port);
    furi_record_close(RECORD_GUI);
    view_port_free(view_port);
    furi_message_queue_free(event_queue);
    delete_mutex(&state_mutex);
    free(game_state);

    return 0;
}