#include "pico/stdlib.h"
#include "hardware/gpio.h"

// Definições dos pinos dos LEDs e do buzzer
#define LED_VERDE_PIN 11
#define LED_AZUL_PIN 12
#define LED_VERMELHO_PIN 13
#define BUZZER_PIN 21

// Pinos do teclado matricial
#define ROW1 2
#define ROW2 3
#define ROW3 4
#define ROW4 5
#define COL1 6
#define COL2 7
#define COL3 8
#define COL4 9

// Definições das teclas
#define TECLA_A 'A'
#define TECLA_B 'B'
#define TECLA_C 'C'
#define TECLA_D 'D'
#define TECLA_HASH '#'
#define TECLA_ESTRELA '*'

// Função de inicialização do teclado
void teclado_init() {
    gpio_init(ROW1); gpio_set_dir(ROW1, GPIO_IN);
    gpio_init(ROW2); gpio_set_dir(ROW2, GPIO_IN);
    gpio_init(ROW3); gpio_set_dir(ROW3, GPIO_IN);
    gpio_init(ROW4); gpio_set_dir(ROW4, GPIO_IN);
    gpio_init(COL1); gpio_set_dir(COL1, GPIO_OUT);
    gpio_init(COL2); gpio_set_dir(COL2, GPIO_OUT);
    gpio_init(COL3); gpio_set_dir(COL3, GPIO_OUT);
    gpio_init(COL4); gpio_set_dir(COL4, GPIO_OUT);
}

// Função para ler o teclado matricial
char ler_tecla() {
    const char matriz[4][4] = {
        {'1', '2', '3', TECLA_A},
        {'4', '5', '6', TECLA_B},
        {'7', '8', '9', TECLA_C},
        {TECLA_ESTRELA, '0', TECLA_HASH, TECLA_D}
    };

    for (int c = 0; c < 4; c++) {
        gpio_put(COL1, c == 0);
        gpio_put(COL2, c == 1);
        gpio_put(COL3, c == 2);
        gpio_put(COL4, c == 3);

        if (gpio_get(ROW1)) {
            sleep_ms(200);  
            return matriz[0][c];
        }
        if (gpio_get(ROW2)) {
            sleep_ms(200);  
            return matriz[1][c];
        }
        if (gpio_get(ROW3)) {
            sleep_ms(200);  
            return matriz[2][c];
        }
        if (gpio_get(ROW4)) {
            sleep_ms(200);  
            return matriz[3][c];
        }
    }
    return 0;
}

// Função para controlar os LEDs
void controle_leds(char tecla) {
    gpio_put(LED_VERDE_PIN, 0);
    gpio_put(LED_AZUL_PIN, 0);
    gpio_put(LED_VERMELHO_PIN, 0);

    if (tecla == TECLA_A) {
        gpio_put(LED_VERDE_PIN, 1);  // Liga o LED verde
    } else if (tecla == TECLA_B) {
        gpio_put(LED_AZUL_PIN, 1);   // Liga o LED azul
    } else if (tecla == TECLA_C) {
        gpio_put(LED_VERMELHO_PIN, 1);  // Liga o LED vermelho
    } else if (tecla == TECLA_D) {
        gpio_put(LED_VERDE_PIN, 1);  // Liga todos os LEDs
        gpio_put(LED_AZUL_PIN, 1);
        gpio_put(LED_VERMELHO_PIN, 1);
    }
}

// Função para controlar o buzzer
void controle_buzzer(char tecla) {
    if (tecla == TECLA_HASH) {
        // Gera um som no buzzer ligando e desligando o pino rapidamente
        for (int i = 0; i < 1000; i++) {  // Repete o som por 1 segundo (aproximadamente)
            gpio_put(BUZZER_PIN, 1);  // Liga o buzzer
            sleep_us(500);  // Aguarda 500 microssegundos
            gpio_put(BUZZER_PIN, 0);  // Desliga o buzzer
            sleep_us(500);  // Aguarda 500 microssegundos
        }
    } else if (tecla == TECLA_ESTRELA) {
        // Desliga tudo (LEDs e buzzer)
        gpio_put(LED_VERDE_PIN, 0);
        gpio_put(LED_AZUL_PIN, 0);
        gpio_put(LED_VERMELHO_PIN, 0);
        gpio_put(BUZZER_PIN, 0);
    }
}


int main() {
    // Configura os pinos dos LEDs e do buzzer como saída
    gpio_init(LED_VERDE_PIN); gpio_set_dir(LED_VERDE_PIN, GPIO_OUT);
    gpio_init(LED_AZUL_PIN); gpio_set_dir(LED_AZUL_PIN, GPIO_OUT);
    gpio_init(LED_VERMELHO_PIN); gpio_set_dir(LED_VERMELHO_PIN, GPIO_OUT);
    gpio_init(BUZZER_PIN); gpio_set_dir(BUZZER_PIN, GPIO_OUT);

    // Inicializa o teclado
    teclado_init();

    // Desliga tudo no início
    gpio_put(LED_VERDE_PIN, 0);
    gpio_put(LED_AZUL_PIN, 0);
    gpio_put(LED_VERMELHO_PIN, 0);
    gpio_put(BUZZER_PIN, 0);

    // Loop principal
    while (true) {
        char tecla = ler_tecla();  // Lê a tecla pressionada

        if (tecla) {  // Se uma tecla foi pressionada
            controle_leds(tecla);  // Controla os LEDs
            controle_buzzer(tecla);  // Controla o buzzer
        }

        sleep_ms(100);  // Aguarda um pouco antes de ler a próxima tecla
    }

    return 0;
}