#include "01_automato.h"
#include <string.h>

const int (*get_figure_shape(Sprite_t sprite, int rotation))[4] {
    const int (*result)[4] = NULL;
    switch (sprite) {
        case I:
            switch (rotation % 4) {
                case 0: result =  i_fig_up(); break;
                case 1: result =  i_fig_right(); break;
                case 2: result =  i_fig_down(); break;
                case 3: result =  i_fig_left(); break;
            }
            break;
        case O:
            result =  o_fig();
            break;
        case T:
            switch (rotation % 4) {
                case 0: result =  t_fig_up(); break;
                case 1: result =  t_fig_right(); break;
                case 2: result =  t_fig_down(); break;
                case 3: result =  t_fig_left(); break;
            }
            break;
        case L:
            switch (rotation % 4) {
                case 0: result =  l_fig_up(); break;
                case 1: result =  l_fig_right(); break;
                case 2: result =  l_fig_down(); break;
                case 3: result =  l_fig_left(); break;
            }
            break;
        case J:
            switch (rotation % 4) {
                case 0: result =  j_fig_up(); break;
                case 1: result =  j_fig_right(); break;
                case 2: result =  j_fig_down(); break;
                case 3: result =  j_fig_left(); break;
            }
            break;
        case S:
            switch (rotation % 4) {
                case 0: result =  s_fig_up(); break;
                case 1: result =  s_fig_right(); break;
                case 2: result =  s_fig_down(); break;
                case 3: result =  s_fig_left(); break;
            }
            break;
        case Z:
            switch (rotation % 4) {
                case 0: result =  z_fig_up(); break;
                case 1: result =  z_fig_right(); break;
                case 2: result =  z_fig_down(); break;
                case 3: result =  z_fig_left(); break;
            }
            break;
        default: result = NULL;
    }
    return result;
}

// I
const int (*i_fig_up())[4] {
    static const int shape[4][4] = {
        {0, 0, 0, 0},
        {1, 1, 1, 1},
        {0, 0, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*i_fig_right())[4] {
    static const int shape[4][4] = {
        {0, 0, 1, 0},
        {0, 0, 1, 0},
        {0, 0, 1, 0},
        {0, 0, 1, 0}
    };
    return (const int (*)[4])shape;
}

const int (*i_fig_down())[4] {
    static const int shape[4][4] = {
        {0, 0, 0, 0},
        {0, 0, 0, 0},
        {1, 1, 1, 1},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*i_fig_left())[4] {
    static const int shape[4][4] = {
        {0, 1, 0, 0},
        {0, 1, 0, 0},
        {0, 1, 0, 0},
        {0, 1, 0, 0}
    };
    return (const int (*)[4])shape;
}

// O
const int (*o_fig())[4] {
    static const int shape[4][4] = {
        {0, 1, 1, 0},
        {0, 1, 1, 0},
        {0, 0, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

// T
const int (*t_fig_up())[4] {
    static const int shape[4][4] = {
        {0, 1, 0, 0},
        {1, 1, 1, 0},
        {0, 0, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*t_fig_right())[4] {
    static const int shape[4][4] = {
        {0, 1, 0, 0},
        {0, 1, 1, 0},
        {0, 1, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*t_fig_down())[4] {
    static const int shape[4][4] = {
        {0, 0, 0, 0},
        {1, 1, 1, 0},
        {0, 1, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*t_fig_left())[4] {
    static const int shape[4][4] = {
        {0, 1, 0, 0},
        {1, 1, 0, 0},
        {0, 1, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

// L
const int (*l_fig_up())[4] {
    static const int shape[4][4] = {
        {0, 0, 1, 0},
        {1, 1, 1, 0},
        {0, 0, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*l_fig_right())[4] {
    static const int shape[4][4] = {
        {1, 0, 0, 0},
        {1, 0, 0, 0},
        {1, 1, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*l_fig_down())[4] {
    static const int shape[4][4] = {
        {0, 0, 0, 0},
        {1, 1, 1, 0},
        {1, 0, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*l_fig_left())[4] {
    static const int shape[4][4] = {
        {1, 1, 0, 0},
        {0, 1, 0, 0},
        {0, 1, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

// J
const int (*j_fig_up())[4] {
    static const int shape[4][4] = {
        {1, 0, 0, 0},
        {1, 1, 1, 0},
        {0, 0, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*j_fig_right())[4] {
    static const int shape[4][4] = {
        {0, 1, 1, 0},
        {0, 1, 0, 0},
        {0, 1, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*j_fig_down())[4] {
    static const int shape[4][4] = {
        {0, 0, 0, 0},
        {1, 1, 1, 0},
        {0, 0, 1, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*j_fig_left())[4] {
    static const int shape[4][4] = {
        {0, 1, 0, 0},
        {0, 1, 0, 0},
        {1, 1, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

// S
const int (*s_fig_up())[4] {
    static const int shape[4][4] = {
        {0, 1, 1, 0},
        {1, 1, 0, 0},
        {0, 0, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*s_fig_right())[4] {
    static const int shape[4][4] = {
        {0, 1, 0, 0},
        {0, 1, 1, 0},
        {0, 0, 1, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*s_fig_down())[4] {
    static const int shape[4][4] = {
        {0, 1, 1, 0},
        {1, 1, 0, 0},
        {0, 0, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*s_fig_left())[4] {
    static const int shape[4][4] = {
        {0, 1, 0, 0},
        {0, 1, 1, 0},
        {0, 0, 1, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

// Z
const int (*z_fig_up())[4] {
    static const int shape[4][4] = {
        {1, 1, 0, 0},
        {0, 1, 1, 0},
        {0, 0, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*z_fig_right())[4] {
    static const int shape[4][4] = {
        {0, 0, 1, 0},
        {0, 1, 1, 0},
        {0, 1, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*z_fig_down())[4] {
    static const int shape[4][4] = {
        {1, 1, 0, 0},
        {0, 1, 1, 0},
        {0, 0, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*z_fig_left())[4] {
    static const int shape[4][4] = {
        {0, 0, 1, 0},
        {0, 1, 1, 0},
        {0, 1, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}

const int (*empty_fig())[4] {
    static const int shape[4][4] = {
        {0, 0, 0, 0},
        {0, 0, 0, 0},
        {0, 0, 0, 0},
        {0, 0, 0, 0}
    };
    return (const int (*)[4])shape;
}