#include "oled.h"
#include "sh1106.h"
#include "connection.h"

void SH1106_Init(OLED_HandleTypeDef *hOled)
{
    uint8_t display = SH1106_DISPLAYOFF;
    oled_SendCommand(hOled, &display, 1);

    uint8_t config[] = {
        SH1106_LOWCOLADDR | 0x00,                      // 0x00h (0000 XXXX)
        SH1106_HIGHCOLADDR | 0x00,                     // 0x10h (0001 XXXX)
        SH1106_PUMPVOLTAGE | 0x02,                     // 0x32h (0011 00XX)
        SH1106_LINEADDRESS | 0x00,                     // 0x40h (01XX XXXX)
        SH1106_CONTRASTMODE, SH1106_CONTRASTVALUE,     //       (1000 0001)  // 0x80h (0x00h - 0xFFh) (XXXX XXXX)
        SH1106_SEGMENTREMAP | 0x00,                    // 0xA0h (1010 000X) right(0), left(1)
        SH1106_ENTIREDISPALY | 0x00,                   // 0xA4h (1010 010X) normal(0), entire(1)
        SH1106_NORMALORREVERS | 0x00,                  // 0xA6h (1010 011X) normal(0), reverse(1)
        SH1106_MULTIPLEXMODE, SH1106_MILTIPLEXVALUE,   //       (1010 1000)  // 0x3Fh (00XX XXXX) (0x00h - 0x3Fh)
        SH1106_DCCONTROLMODE, SH1106_DCCONTROLVALON,   //       (1010 1101)  // 0x8Bh (1000 101X) ON(1), OFF(0)
        SH1106_PAGEADDRESS | 0x00,                     // 0xB0h (1011 XXXX)
        SH1106_OUTPUTSCANDIR | 0x00,                   // 0xC0h (1100 X000) Scan from COM0 to COM [N- 1] (0x00) or Scan from COM [N-1] to COM0 (0x08)
        SH1106_OFFSETMODE, SH1106_OFFSETVALUE,         //       (1101 0011)  // 0x00h (00XX XXXX) COM0-63
        SH1106_DIVRATIOFREQM, SH1106_DIVRATIOFREQV,    //       (1101 0101)  // 0x50h (FFFF DDDD) Frequency, Divide
        SH1106_CHARGEMODE, SH1106_CHARGEVALUE,         //       (1101 1001)  // 0x22h (DDDD PPPP) Dis-charge Period, Pre-charge Period
        SH1106_COMPADSCONFMODE, SH1106_COMPADSCONFVAL, //       (1101 1010)  // 0x12h (000X 0010)
        SH1106_VCOMDESMODE, SH1106_VCOMDESVALUE        //       (1101 1101)  // 0x35h (XXXX XXXX) VCOM (Beta x Vref)
    };
    oled_SendCommand(hOled, config, sizeof(config));

    display = SH1106_DISPLAYON;
    oled_SendCommand(hOled, &display, 1);
}

void SH1106_display_all(OLED_HandleTypeDef *hOled)
{
    while (SH1106_display_page(hOled) != SENDALL);
}

OLED_SendStatus_t SH1106_display_page(OLED_HandleTypeDef *hOled)
{
    static uint8_t page = 0;

    uint8_t config[] = {
        SH1106_PAGEADDRESS | page,
        SH1106_LOWCOLADDR | 0x00,
        SH1106_HIGHCOLADDR | 0x00,
    };
    oled_SendCommand(hOled, config, sizeof(config));

    oled_SendData(hOled, hOled->Buffer + (page * hOled->Width), hOled->Width);

    if (++page > 7)
    {
        page = 0;
        return SENDALL;
    }
    else
    {
        return SENDPAGE;
    }
}