#include "Keypad.h" // khai bao thư vien keypad
#include "Adafruit_GFX.h"       // khai bao thư vien Adafruit graphics library
#include "Adafruit_ILI9341.h"   // khai bao thư vien Adafruit ILI9341 TFT library
#include <map>
#include <iostream>

#define TFT_CS 8 // khai bao chan cs
#define TFT_RST 9 // khai bao chan reset
#define TFT_DC 53 // khai bao chan dc

// -------------------------LCD Setup-----------------------------------------
// initialize ILI9341 TFT library
Adafruit_ILI9341 tft = Adafruit_ILI9341(TFT_CS, TFT_DC, TFT_RST);

// -------------------------Keypad Setup---------------------------------------
const byte ROWS = 4; // Four Rows
const byte COLS = 4; // Four Columns
char keys[ROWS][COLS] = {
    {'7','8','9','A'},
    {'4','5','6','B'},
    {'1','2','3','C'},
    {'*','0','#','D'}
};

map <char, string> key_tittle_mapping = {
    {'7', "START"},
    {'8', "RESET"},
    {'9', "<"},
    {'A', ">"},
    {'4', "BACK RISE"},
    {'5', "BACK DOWN"},
    {'6', "^"},
    {'B', "v"},
    {'1', "SIT UP"},
    {'2', "LEG DOWN"},
    {'3', "LEFT TURN"},
    {'C', "RIGHT TURN"},
    {'*', "BEDPAN ON"},
    {'0', "BEDPAN OFF"},
    {'#', "AUTO TURN A"},
    {'D', "AUTO TURN B"},
}

byte rowPins[ROWS] = {2, 3, 4, 5}; // Arduino pins connected to the row pins of the keypad
byte colPins[COLS] = {6, 7, 8, 9}; // Arduino pins connected to the column pins of the keypad
Keypad keypad_key = Keypad(makeKeymap(keys), rowPins, colPins, ROWS, COLS); // Keypad Library definition

using namespace std;

pair <int, int> get_pos(char key) {  // get (x, y)
    for (int col=1; col < COLS; col++){
        for (int row=1; row <= ROWS; row++){
            if (keys[row - 1][col - 1] == key){
                return {col, row};
            }
        }
    }
    return {0, 0};
}

void draw_button(string tittle, int pos_x, int pos_y, int color = 0){
    pos_x = 15 + (pos_x - 1) * 45;
    pos_y = 40 + (pos_y - 1) * 45;

    uint16_t button_color;
    if (color == 0) { // green
        button_color = tft.color565(64, 237, 39);
    }
    else { // red
        button_color = tft.color565(238, 38, 91);
    }
    int pos_first_space = tittle.find(" ");
    if (pos_first_space != string::npos) {
        string word_up = tittle.substr(0, pos_first_space);
        string word_down = tittle.substr(pos_first_space + 1);

        tft.fillRect(pos_x, pos_y, 40, 40, button_color);
        tft.setCursor(pos_x + 10, pos_y + 10);
        tft.setTextColor(tft.color565(0, 0, 0));
        tft.setTextSize(1);
        tft.println(word_up);

        tft.setCursor(pos_x + 10, pos_y + 20);
        tft.setTextColor(tft.color565(0, 0, 0));
        tft.setTextSize(1);
        tft.println(word_down);
    }
    else {
        tft.fillRect(pos_x, pos_y, 40, 40, button_color);
        tft.setCursor(pos_x + 5, pos_y + 15);
        tft.setTextColor(tft.color565(0, 0, 0));
        tft.setTextSize(1);
        tft.println(tittle);
    }
}

void reset_screen(){
    tft.begin();
    tft.setRotation(3);
    tft.fillScreen(tft.color565(0, 0, 0));

    tft.fillRect(10, 10, 300, 220, tft.color565(116, 139, 248));
    tft.setCursor(20, 20);
    tft.setTextColor(tft.color565(159, 30, 6));
    tft.setTextSize(2);
    tft.println("MAIN CONTROL");

    for (int col=1; col < COLS; col++){
        for (int row=1; row <= ROWS; row++){
            string tittle = key_tittle_mapping[keys[row - 1][col - 1]];
            draw_button(tittle, col, row, 0);
        }
    }

    tft.drawLine(230, 130, 200, 150, ILI9341_BLACK);
    tft.drawLine(230, 130, 270, 130, ILI9341_BLACK);
    tft.drawLine(270, 130, 300, 100, ILI9341_BLACK);
}

void setup() {
    char key = keypad_key.getKey();

    if (key != NO_KEY){
        if (key == '7'){
            pair <int, int> pos = get_pos(key);
            string tittle = key_tittle_mapping[key];
            reset_screen()
            draw_button(tittle, pos.first, pos.second, 1);
        }
    }
}

void loop() {
    char key = keypad_key.getKey();

    if (key != NO_KEY){
        pair <int, int> pos = get_pos(key);
        string tittle = key_tittle_mapping[key];
        reset_screen()
        draw_button(tittle, pos.first, pos.second, 1);
    }
}