#include "SW_18010P.h"
#include <stm32f4xx_adc.h>
#include <stm32f4xx_gpio.h>
#include <stm32f4xx_rcc.h>
#include <stm32f4xx_dma.h>
#include <misc.h>
#include <Timer/TIM.h>
#include "../MotionAlarm_messageHandlers.h"

#define DEBUG

#ifdef DEBUG
#include <USART_dump/USART_dump.h>
#endif

#define VECTOR_TABLE 0x2001c000

static volatile uint16_t adcConvertedValue;
static uint8_t timer;
static uint16_t sensitivity;

static void initGpio() {
    RCC_AHB1PeriphClockCmd(RCC_AHB1Periph_GPIOC, ENABLE);

    GPIO_InitTypeDef gpioInitStructure;
    gpioInitStructure.GPIO_Pin = GPIO_Pin_1;
    gpioInitStructure.GPIO_Mode = GPIO_Mode_AN;
    gpioInitStructure.GPIO_PuPd = GPIO_PuPd_NOPULL;
    GPIO_Init(GPIOC, &gpioInitStructure);
}

static void initAdc() {
    ADC_CommonInitTypeDef adcCommonInitStructure;
    ADC_InitTypeDef adcInitStructure;
    adcCommonInitStructure.ADC_Mode = ADC_Mode_Independent;
    adcCommonInitStructure.ADC_Prescaler = ADC_Prescaler_Div2;
    adcCommonInitStructure.ADC_DMAAccessMode = ADC_DMAAccessMode_Disabled;
    adcCommonInitStructure.ADC_TwoSamplingDelay = ADC_TwoSamplingDelay_5Cycles;
    ADC_CommonInit(&adcCommonInitStructure);
    adcInitStructure.ADC_Resolution = ADC_Resolution_12b;
    adcInitStructure.ADC_ScanConvMode = DISABLE;
    adcInitStructure.ADC_ContinuousConvMode = DISABLE;
    adcInitStructure.ADC_ExternalTrigConvEdge = ADC_ExternalTrigConvEdge_Rising;
    adcInitStructure.ADC_ExternalTrigConv = ADC_ExternalTrigConv_T2_TRGO;
    adcInitStructure.ADC_DataAlign = ADC_DataAlign_Right;
    adcInitStructure.ADC_NbrOfConversion = 1;
    ADC_Init(ADC1, &adcInitStructure);
    ADC_RegularChannelConfig(ADC1, ADC_Channel_11, 1, ADC_SampleTime_480Cycles);
    ADC_DMARequestAfterLastTransferCmd(ADC1, ENABLE);
    ADC_DMACmd(ADC1, ENABLE);
    ADC_Cmd(ADC1, ENABLE);
}

static void initDma() {
    DMA_InitTypeDef dmaInitStructure;
    dmaInitStructure.DMA_Channel = DMA_Channel_0;
    dmaInitStructure.DMA_Memory0BaseAddr = (uint32_t) &adcConvertedValue;
    dmaInitStructure.DMA_PeripheralBaseAddr = (uint32_t)&ADC1->DR;
    dmaInitStructure.DMA_DIR = DMA_DIR_PeripheralToMemory;
    dmaInitStructure.DMA_BufferSize = 1;
    dmaInitStructure.DMA_PeripheralInc = DMA_PeripheralInc_Disable;
    dmaInitStructure.DMA_MemoryInc = DMA_MemoryInc_Enable;
    dmaInitStructure.DMA_PeripheralDataSize = DMA_PeripheralDataSize_HalfWord;
    dmaInitStructure.DMA_MemoryDataSize = DMA_MemoryDataSize_HalfWord;
    dmaInitStructure.DMA_Mode = DMA_Mode_Circular;
    dmaInitStructure.DMA_Priority = DMA_Priority_High;
    dmaInitStructure.DMA_FIFOMode = DMA_FIFOMode_Enable;
    dmaInitStructure.DMA_FIFOThreshold = DMA_FIFOThreshold_HalfFull;
    dmaInitStructure.DMA_MemoryBurst = DMA_MemoryBurst_Single;
    dmaInitStructure.DMA_PeripheralBurst = DMA_PeripheralBurst_Single;
    DMA_Init(DMA2_Stream0, &dmaInitStructure);
    DMA_ITConfig(DMA2_Stream0, DMA_IT_TC | DMA_IT_HT, ENABLE);
    DMA_Cmd(DMA2_Stream0, ENABLE);
}

static void dma2Stream0IrqHandler(void) {
    if (DMA_GetITStatus(DMA2_Stream0, DMA_IT_TCIF0)) {
        DMA_ClearITPendingBit(DMA2_Stream0, DMA_IT_TCIF0);
        if (adcConvertedValue < sensitivity) {
            MotionAlarm_triggerAlarm();
#ifdef DEBUG
            static char str[8];
            itoa(adcConvertedValue, str, 10);
            DUMP(str);
#endif
        }
    }
}

static void initNvic(void) {
    NVIC_InitTypeDef nvicInitStructure;
    nvicInitStructure.NVIC_IRQChannel = DMA2_Stream0_IRQn;
    nvicInitStructure.NVIC_IRQChannelPreemptionPriority = 0;
    nvicInitStructure.NVIC_IRQChannelSubPriority = 0;
    nvicInitStructure.NVIC_IRQChannelCmd = ENABLE;

    // Set irq vector
    *((void (**)(void)) (VECTOR_TABLE + 0x120)) = dma2Stream0IrqHandler;

    NVIC_Init(&nvicInitStructure);
}

void SW_18010P_init(uint16_t initialSensitivity) {
    sensitivity = initialSensitivity;
    initGpio();
    initNvic();
    timer = TIM_startAdcTimer(1);
    initDma();
    initAdc();
}

void SW_18010P_deinit() {
    if (timer) {
        TIM_cancelTimer(timer);
    }
}

uint16_t SW_18010P_getCurrentValue() {
    return adcConvertedValue;
}

uint16_t SW_18010P_getSensitivity() {
    return sensitivity;
}

void SW_18010P_setSensitivity(uint16_t newSensitivity) {
    sensitivity = newSensitivity;
}