#include <stdio.h>
#include "pico/stdlib.h"
#include "hardware/adc.h"
#include "hardware/dma.h"
#include "hardware/irq.h"

#define SAMPLES 20     // Número de amostras por transferência DMA
#define TEMP_CHANNEL 4 // Canal 4 para o sensor de temperatura interno

// Variáveis globais
static int dma_channel;
static volatile uint16_t adc_buffer[SAMPLES]; // 'volatile' para evitar otimizações indesejadas
static volatile bool dma_complete = false;    // Flag para sinalizar conclusão

int count = 0;

// Handler de interrupção do DMA
void dma_handler()
{
    // Limpa o flag de interrupção do DMA
    dma_hw->ints0 = 1u << dma_channel;

    // Marca a transferência como concluída
    dma_complete = true;

    // Reinicia a transferência DMA para o próximo ciclo
    dma_channel_set_write_addr(dma_channel, (void *)adc_buffer, true);
}

// Inicializa o ADC
void initialize_adc()
{
    adc_init();
    adc_set_temp_sensor_enabled(true);
    adc_select_input(TEMP_CHANNEL);

    // Configura o FIFO do ADC
    adc_fifo_setup(
        true,  // Habilita FIFO
        true,  // Habilita requisição DMA
        1,     // Threshold de 1 amostra (DMA é chamado a cada amostra)
        false, // Sem erro
        false  // Sem shift (12 bits)
    );

    // Inicia o ADC em modo contínuo
    adc_run(true);
}

// Inicializa o DMA
void initialize_dma()
{
    dma_channel = dma_claim_unused_channel(true);
    dma_channel_config config = dma_channel_get_default_config(dma_channel);
    channel_config_set_transfer_data_size(&config, DMA_SIZE_16);
    channel_config_set_read_increment(&config, false);
    channel_config_set_write_increment(&config, true);
    channel_config_set_dreq(&config, DREQ_ADC);

    // Configura o canal DMA
    dma_channel_configure(
        dma_channel,
        &config,
        (void *)adc_buffer, // Destino (cast para evitar warning)
        &adc_hw->fifo,      // Origem (FIFO do ADC)
        SAMPLES,            // Número de transferências
        false               // Não inicia ainda
    );

    // Habilita interrupção DMA
    dma_channel_set_irq0_enabled(dma_channel, true);
    irq_set_exclusive_handler(DMA_IRQ_0, dma_handler);
    irq_set_enabled(DMA_IRQ_0, true);

    // Inicia a primeira transferência
    dma_channel_start(dma_channel);
}

// Função para exibir as amostras
void print_samples()
{
    printf("DMA concluído! Amostras:\n");
    printf("Número de médias de amostras: %d\n", count++);
    for (int i = 0; i < SAMPLES; i++)
    {
        float voltage = adc_buffer[i] * 3.3f / (1 << 12);
        float temp = 27 - (voltage - 0.706) / 0.001721;
        printf("Amostra %d: %.2f°C\n", i, temp);
    }
    printf("\n");
}

// Função principal
int main()
{
    stdio_init_all();
    sleep_ms(2000); // Aguarda o terminal
    printf("Sistema iniciando...\n");

    initialize_adc();
    initialize_dma();

    // Loop principal
    while (true)
    {
        if (dma_complete)
        {
            print_samples();      // Exibe as amostras
            dma_complete = false; // Reseta a flag
            sleep_ms(1000);       // Pausa para facilitar a leitura
        }
        tight_loop_contents(); // Mantém a CPU livre para interrupções
    }

    return 0;
}