LCOM1920-T5G03

#include <lcom/lcf.h>

#include "ent.h"

#include "graph.h"

#include "utils.h"

#include "rectangle.h"

#include "font.h"

#include "errors.h"

#include "queue.h"

#include <math.h>

#define GREEN_HEALTH_COLOR      0x009900

static double scale = 1.0;

static int16_t x_origin = 0;

static int16_t y_origin = 0;

void (ent_set_scale) (double n){ scale = n; }

void (ent_set_origin)(double x, double y){ x_origin = x; y_origin = y; }

double (ent_get_scale)  (void){ return scale; }

double (ent_get_XLength)(void){ return graph_get_XRes()/scale; }

double (ent_get_YLength)(void){ return graph_get_YRes()/scale; }

struct gunner{

    double x, y; //real position

    double spawn_x, spawn_y;

    sprite_t *dude;

    sprite_t *weapon;

    double health, current_health;

    text_t *txt;

    gunner_type type;

    int team;

};

gunner_t* (gunner_ctor)(basic_sprite_t *dude, basic_sprite_t *weapon, gunner_type type, int team){

    gunner_t *ret = malloc(sizeof(gunner_t));

    if(ret == NULL) return NULL;

    ret->spawn_x = 0.0;

    ret->spawn_y = 0.0;

    ret->x = 0.0;

    ret->y = 0.0;

    ret->health = 100;

    ret->current_health = ret->health;

    ret->txt = text_ctor(default_font, "");

    ret->type = type;

    ret->team = team;

    ret->dude   = sprite_ctor(dude  );

    ret->weapon = sprite_ctor(weapon);

    if(ret->dude == NULL || ret->weapon == NULL){

        gunner_dtor(ret);

        return NULL;

    }

    text_set_size(ret->txt, 15);

    text_set_valign(ret->txt, text_valign_center);

    text_set_halign(ret->txt, text_halign_center);

    text_set_color(ret->txt, GRAPH_WHITE);

    return ret;

}

void (gunner_dtor)(gunner_t *p){

    if(p == NULL) return;

    sprite_dtor(p->dude);

    sprite_dtor(p->weapon);

    text_dtor(p->txt);

    free(p);

}

void (gunner_set_pos)  (gunner_t *p, double x, double y){ p->x = x; p->y = y; }

void (gunner_set_spawn)  (gunner_t *p, double x, double y){ p->spawn_x = x; p->spawn_y = y; }

void (gunner_set_angle)(gunner_t *p, double angle      ){

    sprite_set_angle(p->dude  , angle);

    sprite_set_angle(p->weapon, angle);

}

void (gunner_set_health)        (gunner_t *p, double health) {

    if (health < 0) health = 0;

    p->health = health;

}

void (gunner_set_curr_health)   (gunner_t *p, double health) {

    if (health < 0) health = 0;

    p->current_health = health;

}

double  (gunner_get_x)              (const gunner_t *p){ return p->x; }

double  (gunner_get_y)              (const gunner_t *p){ return p->y; }

double  (gunner_get_spawn_x)        (const gunner_t *p){ return p->spawn_x; }

double  (gunner_get_spawn_y)        (const gunner_t *p){ return p->spawn_y; }

double  (gunner_get_angle)          (const gunner_t *p){ return sprite_get_angle(p->dude); }

double  (gunner_get_health)         (const gunner_t *p){ return p->health; }

double  (gunner_get_curr_health)    (const gunner_t *p){ return p->current_health; }

int16_t (gunner_get_x_screen)       (const gunner_t *p){ return (p->x-x_origin)*scale; }

int16_t (gunner_get_y_screen)       (const gunner_t *p){ return (p->y-y_origin)*scale; }

gunner_type (gunner_get_type)       (const gunner_t *p){ return p->type; }

int     (gunner_get_team)           (const gunner_t *p){ return p->team; }

void (gunner_draw)(gunner_t *p){

    const int16_t x_screen = gunner_get_x_screen(p);

    const int16_t y_screen = gunner_get_y_screen(p);

    sprite_set_pos  (p->dude  , x_screen, y_screen);

    sprite_set_pos  (p->weapon, x_screen, y_screen);

    sprite_set_scale(p->dude  , scale);

    sprite_set_scale(p->weapon, scale);

    sprite_draw     (p->weapon);

    sprite_draw     (p->dude  );

    gunner_draw_health(p);

}

void (gunner_draw_health)(const gunner_t *p) {

    int16_t w = sprite_get_w(p->dude);

    int16_t h = sprite_get_h(p->dude);

    double x = gunner_get_x_screen(p) - w/2;

    double y = gunner_get_y_screen(p) - h/2 - 10;

    double curr_health = gunner_get_curr_health(p);

    double health = gunner_get_health(p);

    double perc = curr_health/health;

    rectangle_t *green_bar = rectangle_ctor(x, y, (int16_t)(w*perc), 10);

    rectangle_set_fill_color(green_bar, GREEN_HEALTH_COLOR);

    rectangle_t *red_bar = rectangle_ctor(x+(int16_t)(w*perc), y, (int16_t)(w*(1-perc)), 10);

    rectangle_set_fill_color(red_bar, GRAPH_RED);

    char buf[20]; sprintf(buf, "%d/%d", (int)p->current_health, (int)p->health);

    text_set_text(p->txt, buf);

    text_set_pos(p->txt, x+w/2, y+10/2);

    rectangle_draw(green_bar);

    rectangle_draw(red_bar);

    text_draw(p->txt);

    rectangle_dtor(green_bar);

    rectangle_dtor(red_bar);

}

double (gunner_distance)(const gunner_t *p1, const gunner_t *p2){

    double dx = gunner_get_x(p1) - gunner_get_x(p2);

    double dy = gunner_get_y(p1) - gunner_get_y(p2);

    return sqrt(dx*dx+dy*dy);

}

struct bullet{

    const gunner_t *shooter;

    double x, y; //real position

    double vx, vy;

    sprite_t *b;

    double damage;

};

bullet_t* (bullet_ctor)(const gunner_t *shooter, const basic_sprite_t *b, double x, double y, double vx, double vy){

    bullet_t *ret = malloc(sizeof(bullet_t));

    if(ret == NULL) return NULL;

    ret->shooter = shooter;

    ret-> x =  x;

    ret-> y =  y;

    ret->vx = vx;

    ret->vy = vy;

    ret->damage = 10;

    ret->b = sprite_ctor(b);

    if(ret->b == NULL){

        bullet_dtor(ret);

        return NULL;

    }

    double angle = atan2(-ret->vy, ret->vx);

    sprite_set_angle(ret->b, angle-M_PI_2);

    return ret;

}

void (bullet_dtor)(bullet_t *p){

    if(p == NULL) return;

    sprite_dtor(p->b);

    free(p);

}

double  (bullet_get_x)       (const bullet_t *p){ return p->x; }

double  (bullet_get_y)       (const bullet_t *p){ return p->y; }

int16_t (bullet_get_x_screen)(const bullet_t *p){ return (p->x-x_origin)*scale; }

int16_t (bullet_get_y_screen)(const bullet_t *p){ return (p->y-y_origin)*scale; }

double  (bullet_get_damage)  (const bullet_t *p){ return p->damage; }

void    (bullet_set_damage)  (bullet_t *p, double damage) {

    if (damage < 0) damage = 0;

    p->damage = damage;

}

const gunner_t* (bullet_get_shooter)(const bullet_t *p){ return p->shooter; }

void (bullet_update_movement)(bullet_t *p){

    p->x += p->vx;

    p->y += p->vy;

}

void (bullet_update_movement_list)(list_t *bullet_list){

    if (list_size(bullet_list) == 0) return;

    list_node_t *it = list_begin(bullet_list);

    while (it != list_end(bullet_list)) {

        bullet_update_movement(*(bullet_t**)list_node_val(it));

        it = list_node_next(it);

    }

}

void (bullet_draw)(bullet_t *p){

    const int16_t x_screen = bullet_get_x_screen(p);

    const int16_t y_screen = bullet_get_y_screen(p);

    sprite_set_pos  (p->b, x_screen, y_screen);

    sprite_set_scale(p->b, scale);

    sprite_draw     (p->b);

}

void (gunner_draw_list)(list_t *shooter_list) {

    if (list_size(shooter_list) == 0) return;

    list_node_t *it = list_begin(shooter_list);

    while (it != list_end(shooter_list)) {

        gunner_draw(*(gunner_t**)list_node_val(it));

        it = list_node_next(it);

    }

}

void (bullet_draw_list)(list_t *bullet_list) {

    if (list_size(bullet_list) == 0) return;

    list_node_t *it = list_begin(bullet_list);

    while (it != list_end(bullet_list)) {

        bullet_draw(*(bullet_t**)list_node_val(it));

        it = list_node_next(it);

    }

}

struct map{

    basic_sprite_t *bsp_background;

    sprite_t *background;

    uint8_t *collide;

    uint8_t *collide_gunner;

    int32_t *prev;

    uint8_t *visited;

};

static int (map_collides_gunner_pos)(const map_t *p, double shooter_x, double shooter_y, double radius) {

    for (double x = -radius; x <= radius; x += 1) {

        double y1 = sqrt(radius*radius - x*x);

        double y2 = -y1;

        if (map_collides_point(p, shooter_x + x, shooter_y + y1) || map_collides_point(p, shooter_x + x, shooter_y + y2)) return 1;

    }

    return 0;

}

map_t* (map_ctor)(const char **background, const char **collide){

    map_t *ret = malloc(sizeof(map_t));

    if(ret == NULL) return NULL;

    ret->bsp_background = NULL;

    ret->background     = NULL;

    ret->collide        = NULL;

    ret->prev           = NULL;

    ret->visited        = NULL;

    ret->bsp_background = basic_sprite_ctor(background, 0, 0);

    ret->background     = sprite_ctor(ret->bsp_background);

    if(ret->bsp_background == NULL ||

       ret->background     == NULL){ map_dtor(ret); return NULL; }

    basic_sprite_t *bsp_collide = basic_sprite_ctor(collide, 0, 0);

    if(bsp_collide == NULL){ map_dtor(ret); return NULL; }

    const uint16_t W = basic_sprite_get_w(bsp_collide);

    const uint16_t H = basic_sprite_get_h(bsp_collide);

    ret->collide = malloc(W*H*sizeof(uint8_t));

    if(ret->collide == NULL){ map_dtor(ret); return NULL; }

    const uint8_t *m = basic_sprite_get_map(bsp_collide);

    for(unsigned i = 0; i < W*H; ++i){

        ret->collide[i] = (m[4*i+3] < ALPHA_THRESHOLD ? 1 : 0);

    }

    basic_sprite_dtor(bsp_collide);

    ret->collide_gunner = malloc(W*H*sizeof(uint8_t));

    if(ret->collide_gunner == NULL){ map_dtor(ret); return NULL; }

    for(size_t i = 0; i < W*H; ++i){

        int16_t x = i%W, y = i/W;

        ret->collide_gunner[i] = (map_collides_gunner_pos(ret, x, y, 36) ? 1 : 0);

    }

    ret->prev = malloc(W*H*sizeof(int32_t));

    ret->visited = malloc(W*H*sizeof(uint8_t));

    if(ret->prev == NULL || ret->visited == NULL){

        map_dtor(ret);

        return NULL;

    }

    return ret;

}

void (map_dtor)(map_t *p){

    if(p == NULL) return;

    sprite_dtor(p->background);

    basic_sprite_dtor(p->bsp_background);

    free(p->collide);

    free(p->prev);

    free(p->visited);

    free(p);

}

int16_t (map_get_x_screen)(const map_t *p){ return (-x_origin)*scale; }

int16_t (map_get_y_screen)(const map_t *p){ return (-y_origin)*scale; }

int16_t (map_get_width)   (const map_t *p){ return sprite_get_w(p->background); }

int16_t (map_get_height)  (const map_t *p){ return sprite_get_h(p->background); }

int (map_collides_point)(const map_t *p, double x, double y){

    const uint16_t w = sprite_get_w(p->background), h = sprite_get_h(p->background);

    int16_t x_ = x, y_ = y;

    if(x_ < 0 || w <= x_ || y_ < 0 || h <= y_) return 0;

    uint32_t pos = x_ + y_*w;

    return p->collide[pos];

}

int (map_collides_gunner)(const map_t *p, const gunner_t *shooter) {

    double radius = max(sprite_get_w(shooter->dude), sprite_get_h(shooter->dude))/2.0;

    return map_collides_gunner_pos(p, gunner_get_x(shooter), gunner_get_y(shooter), radius);

}

int (map_make_dijkstra)(map_t *p, int16_t x, int16_t y){

    const uint16_t W = basic_sprite_get_w(p->bsp_background),

                   H = basic_sprite_get_h(p->bsp_background);

    static uint8_t first_time = true;

    if(first_time){

        for(size_t i = 0; i < W*H; ++i) p->prev[i] = -1;

        first_time = false;

    }

    /// ACTUAL DIJKSTRA

    queue_t *q = queue_ctor();

    memset(p->visited, false, W*H*sizeof(uint8_t));

    int *ptr;

    int32_t c, pos;

    c = y*W+x; p->prev[c] = c; ptr = malloc(sizeof(int)); *ptr = c; queue_push(q, ptr);

    while(!queue_empty(q)){

        c = *(int*)queue_top(q); free(queue_top(q)); queue_pop(q);

        x = c%W, y = c/W;

        if(p->visited[c]) continue;

        p->visited[c] = true;

        if(p->collide_gunner[c]) continue;

        if(0   <= x-1){ pos = y*W+(x-1); if(!p->visited[pos] && p->prev[pos] != c){ p->prev[pos] = c; ptr = malloc(sizeof(int)); *ptr = pos; queue_push(q, ptr); }}

        if(x+1 <  W  ){ pos = y*W+(x+1); if(!p->visited[pos] && p->prev[pos] != c){ p->prev[pos] = c; ptr = malloc(sizeof(int)); *ptr = pos; queue_push(q, ptr); }}

        if(0   <= y-1){ pos = (y-1)*W+x; if(!p->visited[pos] && p->prev[pos] != c){ p->prev[pos] = c; ptr = malloc(sizeof(int)); *ptr = pos; queue_push(q, ptr); }}

        if(y+1 <  H  ){ pos = (y+1)*W+x; if(!p->visited[pos] && p->prev[pos] != c){ p->prev[pos] = c; ptr = malloc(sizeof(int)); *ptr = pos; queue_push(q, ptr); }}

    }

    queue_dtor(q);

    return SUCCESS;

}

int (map_where_to_follow)(const map_t *p, float x, float y, float *theta){

    const uint16_t W = basic_sprite_get_w(p->bsp_background);

    int x_ = x, y_ = y;

    int pos = y_*W+x_;

    //printf("Is in %d,%d\n", x_, y_);

    int newx = p->prev[pos]%W, newy = p->prev[pos]/W;

    //printf("from %d,%d to %d,%d\n", x_, y_, newx, newy);

    *theta = atan2(-(newy-y_), newx-x_);

    return SUCCESS;

}

int (map_collides_bullet)(const map_t *p, const bullet_t *bull){

    double radius = max(sprite_get_w(bull->b), sprite_get_h(bull->b))/2.0;

    double bullet_x = bullet_get_x(bull);

    double bullet_y = bullet_get_y(bull);

    for (double x = -radius; x < radius; x += 1){

        double y1 = sqrt(radius*radius - x*x);

        double y2 = -y1;

        if (map_collides_point(p, bullet_x + x, bullet_y + y1) || map_collides_point(p, bullet_x + x, bullet_y + y2)) return 1;

    }

    return 0;

}

int (gunner_collides_bullet)(const gunner_t *shooter, const bullet_t *bull){

    if(bull->shooter == shooter) return false;

    double shooter_radius = max(sprite_get_w(shooter->dude), sprite_get_h(shooter->dude))/2.0;

    double shooter_x = gunner_get_x(shooter);

    double shooter_y = gunner_get_y(shooter);

    double bullet_radius = max(sprite_get_w(bull->b), sprite_get_h(bull->b))/2.0;

    double bullet_x = bullet_get_x(bull);

    double bullet_y = bullet_get_y(bull);

    double dx = shooter_x - bullet_x;

    double dy = shooter_y - bullet_y;

    double distance = sqrt(dx*dx + dy*dy);

    return distance <= shooter_radius+bullet_radius;

}

double (distance_gunners)(const gunner_t *shooter1, const gunner_t *shooter2) {

    double shooter1_x = gunner_get_x(shooter1);

    double shooter1_y = gunner_get_y(shooter1);

    double shooter2_x = gunner_get_x(shooter2);

    double shooter2_y = gunner_get_y(shooter2);

    double dx = shooter1_x - shooter2_x;

    double dy = shooter1_y - shooter2_y;

    return sqrt(dx*dx + dy*dy);

}

int (gunner_collides_gunner)(const gunner_t *shooter1, const gunner_t *shooter2) {

    double shooter1_radius = max(sprite_get_w(shooter1->dude), sprite_get_h(shooter1->dude))/2.0;

    double shooter2_radius = max(sprite_get_w(shooter2->dude), sprite_get_h(shooter2->dude))/2.0;

    double distance = distance_gunners(shooter1, shooter2);

    return distance <= shooter1_radius+shooter2_radius;

}

void   (map_draw)(map_t *p){

    const int16_t x_screen = map_get_x_screen(p);

    const int16_t y_screen = map_get_y_screen(p);

    sprite_set_pos  (p->background, x_screen, y_screen);

    sprite_set_scale(p->background, scale);

    sprite_draw     (p->background);

}