#include "main.h"

#include "stm32l100c_discovery.h"

#include "ssd1306.h"

// USB includes

#include "hw_config.h"

#include "usb_lib.h"

#include "usb_desc.h"

#include "usb_pwr.h"

#define LED_POWER GPIOB,GPIO_Pin_9

#define LED_STAT  GPIOA,GPIO_Pin_15

#define MAX_CS GPIOB,GPIO_Pin_12

// TODO: Grab buttonpresses with interrupts

#define SW_BTN  GPIOB, GPIO_Pin_3

#define SW_UP   GPIOB, GPIO_Pin_7

#define SW_DOWN GPIOB, GPIO_Pin_6

#define SW_LEFT GPIOB, GPIO_Pin_5

#define SW_RIGHT GPIOB, GPIO_Pin_4

// USB Supporting Vars

extern __IO uint8_t Receive_Buffer[64];

extern __IO  uint32_t Receive_length ;

extern __IO  uint32_t length ;

uint8_t Send_Buffer[64];

uint32_t packet_sent=1;

uint32_t packet_receive=1;

static __IO uint32_t TimingDelay;

// Move to header file

void init_gpio();

void init_spi();

void process();

void machine();

int main(void)

{

    // Init clocks

    SystemInit();

    init_gpio();

    // Init USB

    //Set_USBClock();

    //USB_Interrupts_Config();

    //USB_Init();

    GPIO_SetBits(LED_POWER);

    RCC_ClocksTypeDef RCC_Clocks;

    // SysTick end of count event each 1ms

    RCC_GetClocksFreq(&RCC_Clocks);

    SysTick_Config(RCC_Clocks.HCLK_Frequency / 1000);

    GPIO_ResetBits(LED_STAT);

    Delay(100);

    GPIO_SetBits(LED_POWER);

    Delay(500);

    GPIO_ResetBits(LED_POWER);

    init_spi();

    ssd1306_Init();

    ssd1306_DrawString("Hello World!", 0, 10);

  //  ssd1306_block_write();

    while(1)

   {  

        //ssd1306_block_write();

        // Process sensor inputs [TODO: 5hz?]

        process();

        // Run state machine [TODO: 50hz?]

        machine(); 

        // probably just passed the actual port

        // TODO: Grab buttonpresses with interrupts

        uint8_t sw_btn = GPIO_ReadInputDataBit(SW_BTN);

        uint8_t sw_up = GPIO_ReadInputDataBit(SW_UP);

        uint8_t sw_down = GPIO_ReadInputDataBit(SW_DOWN);

        uint8_t sw_left = GPIO_ReadInputDataBit(SW_LEFT);

        uint8_t sw_right = GPIO_ReadInputDataBit(SW_RIGHT);

        if(!sw_btn) {

            GPIO_ToggleBits(LED_STAT);

        }

        GPIO_SetBits(LED_POWER);

        Delay(50);

        GPIO_ResetBits(LED_POWER);

        Delay(50);

    }

}

int32_t temp = 0;

int32_t setpoint = 0;

int32_t p = 1;

int32_t i = 1;

int32_t d = 1;

// Process things

void process()

{

    // Read MAX temp sensor

    GPIO_ResetBits(MAX_CS);

    // Assert CS

    // This may not clock at all... might need to send 16 bits first

    uint8_t retval = 0;//SPI_I2S_ReceiveData(SPI2);

    // Deassert CS

    GPIO_SetBits(MAX_CS);

    if((!retval || (temp & 0x2) != 0))

        return; // Comms error - this is happening right now

    if((temp & 0x4)!= 0)

        return; // Open thermocouple

    temp = (temp & 0x7FF8) >> 5;

    // TODO: Add calibration offset (linear)

    // Perform PID calculations

    // Write output to SSR

}

enum state {

    STATE_IDLE = 0,

    STATE_SETP,

    STATE_SETI,

    STATE_SETD,

    STATE_PREHEAT_BREW,

    STATE_MAINTAIN_BREW,

    STATE_PREHEAT_STEAM,

    STATE_MAINTAIN_STEAM,

};

uint8_t state = STATE_IDLE;

// State machine

void machine()

{

    switch(state)

    {

        // Idle state

        case STATE_IDLE:

        {

            // Write text to OLED

            // [ therm :: idle ]

            // Button handler

            if(GPIO_ReadInputDataBit(SW_BTN)) {

                state = STATE_SETP;

            }

            // Event Handler

            // N/A

        } break;

        case STATE_SETP:

        {

            // Write text to OLED

            // [ therm :: set p ]

            // [ p = 12         ]

            // Button handler

            if(GPIO_ReadInputDataBit(SW_BTN)) {

                state = STATE_IDLE;

            }

            // Event Handler

            // N/A

        } break;

        case STATE_SETI:

        {

            // Write text to OLED

            // [ therm :: set i ]

            // [ i = 12         ]

            // Button handler

            if(GPIO_ReadInputDataBit(SW_BTN)) {

                state = STATE_IDLE;

            }

            // Event Handler

            // N/A

        } break;

        case STATE_SETD:

        {

            // Write text to OLED

            // [ therm :: set d ]

            // [ d = 12         ]

            // Button handler

            if(GPIO_ReadInputDataBit(SW_BTN)) {

                state = STATE_IDLE;

            }

            // Event Handler

            // N/A

        } break;

        case STATE_PREHEAT_BREW:

        {

            // Write text to OLED

            // [ therm : preheating brew ]

            // [ 30 => 120 C             ]

            // Button handler

            if(GPIO_ReadInputDataBit(SW_BTN)) {

                state = STATE_IDLE;

            }

            // Event Handler

            if(temp >= setpoint) {

                state = STATE_MAINTAIN_BREW;

            }

        } break;

        case STATE_MAINTAIN_BREW:

        {

            // Write text to OLED

            // [ therm : ready to brew ]

            // [ 30 => 120 C           ]

            // Button handler

            if(GPIO_ReadInputDataBit(SW_BTN)) {

                state = STATE_IDLE;

            }

            // Event Handler

            // N/A

        } break;

        case STATE_PREHEAT_STEAM:

        {

            // Write text to OLED

            // [ therm : preheating steam ]

            // [ 30 => 120 C           ]

            // Button handler

            if(GPIO_ReadInputDataBit(SW_BTN)) {

                state = STATE_IDLE;

            }

            // Event Handler

            if(temp >= setpoint) {

                state = STATE_MAINTAIN_STEAM;

            }

        } break;

        case STATE_MAINTAIN_STEAM:

        {

            // Write text to OLED

            // [ therm : ready to steam ]

            // [ 30 => 120 C            ]

            // Button handler

            if(GPIO_ReadInputDataBit(SW_BTN)) {

                state = STATE_IDLE;

            }

            // Event Handler

            // N/A

        } break;

        // Something is terribly wrong

        default:

        {

            state = STATE_IDLE;

        } break;

    }

}

/**

  * @brief  Inserts a delay time.

  * @param  nTime: specifies the delay time length, in 1 ms.

  * @retval None

  */

void Delay(__IO uint32_t nTime)

{

  TimingDelay = nTime;

  while(TimingDelay != 0);

}

/**

  * @brief  Decrements the TimingDelay variable.

  * @param  None

  * @retval None

  */

void TimingDelay_Decrement(void)

{

  if (TimingDelay != 0x00)

  { 

    TimingDelay--;

  }

}

void init_spi(void)

{

    SPI_InitTypeDef  SPI_InitStructure;

    // OLED IC

    SPI_InitStructure.SPI_Direction = SPI_Direction_1Line_Tx;

    SPI_InitStructure.SPI_Mode = SPI_Mode_Master;

    SPI_InitStructure.SPI_DataSize = SPI_DataSize_8b;

    SPI_InitStructure.SPI_CPOL = SPI_CPOL_High;

    SPI_InitStructure.SPI_CPHA = SPI_CPHA_2Edge;

    SPI_InitStructure.SPI_NSS = SPI_NSS_Soft;

    SPI_InitStructure.SPI_BaudRatePrescaler = SPI_BaudRatePrescaler_16;

    SPI_InitStructure.SPI_FirstBit = SPI_FirstBit_MSB;

    SPI_InitStructure.SPI_CRCPolynomial = 7;

    SPI_Init(SPI1, &SPI_InitStructure);

    SPI_Cmd(SPI1, ENABLE);           /* Enable the SPI  */   

    // MAX IC

    //SPI_InitStructure.SPI_Direction = SPI_Direction_1Line_Rx;

    SPI_InitStructure.SPI_Mode = SPI_Mode_Master;

    SPI_InitStructure.SPI_DataSize = SPI_DataSize_16b; // Andysworkshop

    SPI_InitStructure.SPI_CPOL = SPI_CPOL_Low; // From andysworkshop

    SPI_InitStructure.SPI_CPHA = SPI_CPHA_1Edge; // same

    // Can be used for CS... SPI_InitStructure.SPI_NSS = SPI_NSS_Soft;

    SPI_InitStructure.SPI_BaudRatePrescaler = SPI_BaudRatePrescaler_64;

    //SPI_InitStructure.SPI_FirstBit = SPI_FirstBit_MSB;

    //SPI_InitStructure.SPI_CRCPolynomial = 7;

    SPI_Init(SPI2, &SPI_InitStructure);

    SPI_Cmd(SPI2, ENABLE);           /* Enable the SPI */

}

void init_gpio(void) {

 GPIO_InitTypeDef GPIO_InitStruct;

  // Enable SPI clocks

  RCC_APB2PeriphClockCmd(RCC_APB2Periph_SPI1, ENABLE);

  RCC_APB1PeriphClockCmd(RCC_APB1Periph_SPI2, ENABLE);

  // Enable GPIO clocks

  RCC_AHBPeriphClockCmd(RCC_AHBPeriph_GPIOC|RCC_AHBPeriph_GPIOB|RCC_AHBPeriph_GPIOA, ENABLE);

  // Enable DMA clocks (Is AHB even the right thing???)

  RCC_AHBPeriphClockCmd(RCC_AHBPeriph_DMA1, ENABLE); // EMZ TODO get the right ones

  /*Configure GPIO pin : PC */

  GPIO_InitStruct.GPIO_Pin = GPIO_Pin_13;

  GPIO_InitStruct.GPIO_Mode = GPIO_Mode_OUT;

  GPIO_InitStruct.GPIO_OType = GPIO_OType_PP;

  GPIO_InitStruct.GPIO_PuPd = GPIO_PuPd_NOPULL;

  GPIO_InitStruct.GPIO_Speed = GPIO_Speed_400KHz;

  GPIO_Init(GPIOC, &GPIO_InitStruct);

  /*Configure GPIO pin : PB */

  GPIO_InitStruct.GPIO_Pin = GPIO_Pin_1|GPIO_Pin_2|GPIO_Pin_10|GPIO_Pin_12 

                          |GPIO_Pin_9;

  GPIO_InitStruct.GPIO_Mode = GPIO_Mode_OUT;

  GPIO_InitStruct.GPIO_OType = GPIO_OType_PP;

  GPIO_InitStruct.GPIO_PuPd = GPIO_PuPd_NOPULL;

  GPIO_InitStruct.GPIO_Speed = GPIO_Speed_400KHz;

  GPIO_Init(GPIOB, &GPIO_InitStruct);

  /*Configure GPIO pin : PA */

  GPIO_InitStruct.GPIO_Pin = GPIO_Pin_15;

  GPIO_InitStruct.GPIO_Mode = GPIO_Mode_OUT;

  GPIO_InitStruct.GPIO_OType = GPIO_OType_PP;

  GPIO_InitStruct.GPIO_PuPd = GPIO_PuPd_NOPULL;

  GPIO_InitStruct.GPIO_Speed = GPIO_Speed_400KHz;

  GPIO_Init(GPIOA, &GPIO_InitStruct);

  /*Configure GPIO pin : PB */

  GPIO_InitStruct.GPIO_Pin = GPIO_Pin_3|GPIO_Pin_4|GPIO_Pin_5|GPIO_Pin_6 

                          |GPIO_Pin_7;

  GPIO_InitStruct.GPIO_Mode = GPIO_Mode_IN;

  GPIO_InitStruct.GPIO_PuPd = GPIO_PuPd_UP;

  GPIO_Init(GPIOB, &GPIO_InitStruct);

  /** SPI1 GPIO Configuration  

  PA5   ------> SPI1_SCK

  PA7   ------> SPI1_MOSI

  */

  /*Enable or disable the AHB peripheral clock */

  RCC_AHBPeriphClockCmd(RCC_AHBPeriph_GPIOA, ENABLE);

  /*Configure GPIO pin : PA: MOSI,SCK */

  GPIO_InitStruct.GPIO_Pin = GPIO_Pin_5|GPIO_Pin_7;

  GPIO_InitStruct.GPIO_Mode = GPIO_Mode_AF;

  GPIO_InitStruct.GPIO_PuPd = GPIO_PuPd_NOPULL;

  GPIO_InitStruct.GPIO_Speed = GPIO_Speed_10MHz;

  GPIO_Init(GPIOA, &GPIO_InitStruct);

  /*Configure GPIO pin alternate function */

  GPIO_PinAFConfig(GPIOA, GPIO_PinSource5, GPIO_AF_SPI1);

  /*Configure GPIO pin alternate function */

  GPIO_PinAFConfig(GPIOA, GPIO_PinSource7, GPIO_AF_SPI1);

  /** SPI2 GPIO Configuration  

  PB13   ------> SPI2_SCK

  PB14   ------> SPI2_MISO

  PB15   ------> SPI2_MOSI

  */

  /*Enable or disable the AHB peripheral clock */

  RCC_AHBPeriphClockCmd(RCC_AHBPeriph_GPIOB, ENABLE);

// SPI PINSSS

  /*Configure GPIO pin : PB, MOSI, SCK */

  GPIO_InitStruct.GPIO_Pin = GPIO_Pin_13|GPIO_Pin_15;

  GPIO_InitStruct.GPIO_Mode = GPIO_Mode_AF;

  GPIO_InitStruct.GPIO_PuPd = GPIO_PuPd_NOPULL;

  GPIO_InitStruct.GPIO_Speed = GPIO_Speed_10MHz;

  GPIO_Init(GPIOB, &GPIO_InitStruct);

// MISO

  GPIO_InitStruct.GPIO_Pin = GPIO_Pin_14;

  GPIO_InitStruct.GPIO_Mode = GPIO_Mode_IN;

  GPIO_InitStruct.GPIO_PuPd = GPIO_PuPd_NOPULL;

  GPIO_InitStruct.GPIO_Speed = GPIO_Speed_10MHz;

  GPIO_Init(GPIOB, &GPIO_InitStruct);

  /*Configure GPIO pin alternate function */

  GPIO_PinAFConfig(GPIOB, GPIO_PinSource13, GPIO_AF_SPI2);

  /*Configure GPIO pin alternate function */

  GPIO_PinAFConfig(GPIOB, GPIO_PinSource14, GPIO_AF_SPI2);

  /*Configure GPIO pin alternate function */

  GPIO_PinAFConfig(GPIOB, GPIO_PinSource15, GPIO_AF_SPI2);

  /** USB GPIO Configuration  

  PA11   ------> USB_DM

  PA12   ------> USB_DP

  */

  /*Enable or disable the AHB peripheral clock */

  RCC_AHBPeriphClockCmd(RCC_AHBPeriph_GPIOA, ENABLE);

  /*Configure GPIO pin : PA */

  GPIO_InitStruct.GPIO_Pin = GPIO_Pin_11|GPIO_Pin_12;

/*******************************************************************************

* File Name          : ssd1306.c

* Author             : lxyppc

* Version            : V1.0

* Date               : 10-01-21

* Description        : ssd1306 operations

*                      the SSH1101A is compatible with ssd1306

*******************************************************************************/

/* Includes ------------------------------------------------------------------*/

#include "stm32l100c_discovery.h"

#include "bsp.h"

#include "ssd1306.h"

/* Private typedef -----------------------------------------------------------*/

/* Private define ------------------------------------------------------------*/

#define   SSD1306_PAGE_NUMBER           8

#define   SSD1306_COLUMN_NUMBER         128

#define   SSD1306_COLUMN_MARGIN_START   2

#define   SSD1306_COLUMN_MARGIN_END     2

#define   SSD1306_X_PIXEL   128

#define   SSD1306_Y_PIXEL   32

/* Private macro -------------------------------------------------------------*/

#define   ssd1306_Buffer    (_SSD1306_Buffer + SSD1306_COLUMN_MARGIN_START)

/* Private variables ---------------------------------------------------------*/

static  uint8_t  _SSD1306_Buffer[SSD1306_COLUMN_NUMBER*SSD1306_PAGE_NUMBER + SSD1306_COLUMN_MARGIN_START + SSD1306_COLUMN_MARGIN_END] = {0};

static  uint8_t  pageIndex = 0;

static  uint8_t  iS_SSD_On = 0;

static  uint8_t  pre_on = 0;

static  uint8_t  curContrast = 0xCC;

static  uint8_t  lastContrast = 0xCC;

/* Private function prototypes -----------------------------------------------*/

void  WriteCommand(unsigned char command);

void  WriteData(unsigned char data);

void  OnPageTransferDone(void);

unsigned long ssd1306_OFF(void);

unsigned long ssd1306_ON(void);

unsigned char* ssd1306_GetBuffer()

{

  return ssd1306_Buffer;

}

/*******************************************************************************

* Function Name  : WriteCommand

* Description    : Write command to the ssd1306

* Input          : None

* Output         : None

* Return         : None

*******************************************************************************/

void WriteCommand(unsigned char command)

{

  SSD_A0_Low();

  SPI_SendByte(command);

  SPI_Wait();

}

/*******************************************************************************

* Function Name  : WriteData

* Description    : Write data to the ssd1306

* Input          : None

* Output         : None

* Return         : None

*******************************************************************************/

void WriteData(unsigned char data)

{

  SSD_A0_High();

  SPI_SendByte(data);

  SPI_Wait();

}

/*******************************************************************************

* Function Name  : ssd1306_Init

* Description    : Initialize the ssd1306

* Input          : None

* Output         : None

* Return         : None

*******************************************************************************/

void ssd1306_Init(void)

{

  /* Generate a reset */

  SSD_Reset_Low();

  uint32_t i;

  for(i=5000; i>1; i--) 

  SSD_Reset_High();

  WriteCommand(0xAE);

  WriteCommand(0xD5);

  WriteCommand(0x80);

  WriteCommand(0xA8);

  WriteCommand(0x1F);

  WriteCommand(0xD3);

  WriteCommand(0x00);

  WriteCommand(0x40 | 0x00); // line #0

  WriteCommand(0x8D);

  WriteCommand(0x14); //10 or 14 if not externalvcc

  WriteCommand(0x20);

  WriteCommand(0x00);

  WriteCommand(0xA0 | 0x1); // segremap

  WriteCommand(0xC8); // comscandep

  WriteCommand(0xDA); // setcompins 

  WriteCommand(0x02);

  WriteCommand(0x81); // contrast

  WriteCommand(0x0F); // contrast value. 8f is a good one.

  WriteCommand(0xD9);

  WriteCommand(0xF1); //22 or F1 if not externalvcc

  WriteCommand(0xDB);

  WriteCommand(0x40);

  WriteCommand(0xA4); // dispalyallon_resume

  WriteCommand(0xA6); // normaldisplay

  WriteCommand(0xAF); // display on 

  // Clear buffer

//  for(i=0;i<900;i++)

//    ssd1306_Buffer[i] = 0;

  /*************************************************

  // ssd1306 Initialization Command

  *************************************************/

 /* 

  // Lower Column Address

  WriteCommand(0x00); // Set Lower Column Address 

  GPIO_SetBits(GPIOA,GPIO_Pin_15);//turn on status LED

  // High Column Address

  WriteCommand(0x10); // Set Higher Column Address

  // Display Start Line

  WriteCommand(0x40); // Set Display Start Line 

#ifdef    DEBUG_BOARD

  curContrast = lastContrast = 0x30;

#else

  curContrast = lastContrast = 0xCF;

#endif

  // Contrast Control Register

  WriteCommand(0x81); // Set Contrast Control 

  WriteCommand(lastContrast); // 0 ~ 255 0x1f

  // Re-map

  WriteCommand(0xA1); // [A0]:column address 0 is map to SEG0 , [A1]: columnaddress 131 is map to SEG0 

  // Entire Display ON/OFF

  WriteCommand(0xA4); // A4=ON 

  // Normal or Inverse Display

  WriteCommand(0XA6); // Normal Display

  // Multiplex Ratio

  WriteCommand(0xA8); // Set Multiplex Ratio 

  WriteCommand(0x3f); // Set to 36 Mux

  // Set DC-DC

  WriteCommand(0xAD); // Set DC-DC 

  WriteCommand(0x8B); // 8B=ON, 8A=Off 

  // Display ON/OFF

  WriteCommand(0xAE); // AF=ON , AE=OFF

  // Display Offset

  WriteCommand(0xD3); // Set Display Offset 

  WriteCommand(0x00); // No offset 

  // Display Clock Divide

  WriteCommand(0xD5); // Set Clock Divide 

  WriteCommand(0x20); // Set to 80Hz 

  // Area Color Mode

  WriteCommand(0xD8); // Set Area Color On or Off

  WriteCommand(0x00); // Mono Mode 

  // COM Pins Hardware Configuration

  WriteCommand(0xDA); // Set Pins HardwareConfiguration 

  WriteCommand(0x12);

  // VCOMH

  WriteCommand(0xDB); // Set VCOMH 

  WriteCommand(0x00);

  // VP

  WriteCommand(0xD9); // Set VP 

  WriteCommand(0x22); // P1=2 , P2=2 

  // Set Common output scan direction

  WriteCommand(0xc8);// Set COM scan direction 

  // For SSD1306 Set the address mode

  WriteCommand(0x20);// Set address mode 

  WriteCommand(0x00);// Set address mode horizontal 

  // Set the page start address

  WriteCommand(0xb0);

  WriteCommand(0x00);

  WriteCommand(0x10);

  // Turn on the controller 

  pre_on = ssd1306_ON();

//  // Set Charge pump

//  WriteCommand(0x8D); // Set Charge pump 

//  WriteCommand(0x14); // 0x14=ON, 0x10=Off 

//  

//  // Turn on the display

//  WriteCommand(0xaf);

*/

}

void ssd1306_block_write(void)

{

  // Set col start addr to 0

  WriteCommand(0x21); 

  WriteCommand(0x00); 

  // Set col end addr to width - 1

  WriteCommand(127);

  // Set page addr

  WriteCommand(0x22);

  WriteCommand(0x00); // start page addr

  WriteCommand(4); // end page addr (height / 8)

  uint32_t i = 0;

  for(i=0;i<512;i++)

    WriteData(ssd1306_Buffer[i]);

//  for(i=0;i<512;i++)

             // * * * * 

//    WriteData(0b00000011);

}

/*******************************************************************************

* Function Name  : ssd1306_TurnOff

* Description    : Turn off the ssd1306 controller

* Input          : None

* Output         : None

* Return         : None

*******************************************************************************/