#include <SPI.h>

// pins
int slaveSelectPin = 10;
int rstPin = 6;
int dcPin = 5;

// constants
int divider = 84;
int maxBufferIndex = 504;

// buffer
char screenBuffer[504];

// ASCII data
unsigned char font[95][6] =
    {
        { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 }, // space
        { 0x00, 0x00, 0x5F, 0x00, 0x00, 0x00 }, // !
        { 0x00, 0x07, 0x00, 0x07, 0x00, 0x00 }, // "
        { 0x14, 0x7F, 0x14, 0x7F, 0x14, 0x00 }, // #
        { 0x24, 0x2A, 0x7F, 0x2A, 0x12, 0x00 }, // $
        { 0x23, 0x13, 0x08, 0x64, 0x62, 0x00 }, // %
        { 0x36, 0x49, 0x55, 0x22, 0x50, 0x00 }, // &
        { 0x00, 0x00, 0x07, 0x00, 0x00, 0x00 }, // '
        { 0x00, 0x1C, 0x22, 0x41, 0x00, 0x00 }, // (
        { 0x00, 0x41, 0x22, 0x1C, 0x00, 0x00 }, // )
        { 0x0A, 0x04, 0x1F, 0x04, 0x0A, 0x00 }, // *
        { 0x08, 0x08, 0x3E, 0x08, 0x08, 0x00 }, // +
        { 0x00, 0x50, 0x30, 0x00, 0x00, 0x00 }, // ,
        { 0x08, 0x08, 0x08, 0x08, 0x08, 0x00 }, // -
        { 0x00, 0x60, 0x60, 0x00, 0x00, 0x00 }, // .
        { 0x20, 0x10, 0x08, 0x04, 0x02, 0x00 }, // slash
        { 0x3E, 0x51, 0x49, 0x45, 0x3E, 0x00 }, // 0
        { 0x00, 0x42, 0x7F, 0x40, 0x00, 0x00 }, // 1
        { 0x42, 0x61, 0x51, 0x49, 0x46, 0x00 }, // 2
        { 0x21, 0x41, 0x45, 0x4B, 0x31, 0x00 }, // 3
        { 0x18, 0x14, 0x12, 0x7F, 0x10, 0x00 }, // 4
        { 0x27, 0x45, 0x45, 0x45, 0x39, 0x00 }, // 5
        { 0x3C, 0x4A, 0x49, 0x49, 0x30, 0x00 }, // 6
        { 0x03, 0x71, 0x09, 0x05, 0x03, 0x00 }, // 7
        { 0x36, 0x49, 0x49, 0x49, 0x36, 0x00 }, // 8
        { 0x06, 0x49, 0x49, 0x29, 0x1E, 0x00 }, // 9
        { 0x00, 0x36, 0x36, 0x00, 0x00, 0x00 }, // :
        { 0x00, 0x56, 0x36, 0x00, 0x00, 0x00 }, // ;
        { 0x08, 0x14, 0x22, 0x41, 0x00, 0x00 }, // <
        { 0x14, 0x14, 0x14, 0x14, 0x14, 0x00 }, // =
        { 0x00, 0x41, 0x22, 0x14, 0x08, 0x00 }, // >
        { 0x02, 0x01, 0x51, 0x09, 0x06, 0x00 }, // ?
        { 0x32, 0x49, 0x79, 0x41, 0x3E, 0x00 }, // @
        { 0x7E, 0x11, 0x11, 0x11, 0x7E, 0x00 }, // A
        { 0x7F, 0x49, 0x49, 0x49, 0x36, 0x00 }, // B
        { 0x3E, 0x41, 0x41, 0x41, 0x22, 0x00 }, // C
        { 0x7F, 0x41, 0x41, 0x41, 0x3E, 0x00 }, // D
        { 0x7F, 0x49, 0x49, 0x49, 0x41, 0x00 }, // E
        { 0x7F, 0x09, 0x09, 0x09, 0x01, 0x00 }, // F
        { 0x3E, 0x41, 0x41, 0x49, 0x7A, 0x00 }, // G
        { 0x7F, 0x08, 0x08, 0x08, 0x7F, 0x00 }, // H
        { 0x00, 0x41, 0x7F, 0x41, 0x00, 0x00 }, // I
        { 0x20, 0x40, 0x41, 0x3F, 0x01, 0x00 }, // J
        { 0x7F, 0x08, 0x14, 0x22, 0x41, 0x00 }, // K
        { 0x7F, 0x40, 0x40, 0x40, 0x40, 0x00 }, // L
        { 0x7F, 0x02, 0x0C, 0x02, 0x7F, 0x00 }, // M
        { 0x7F, 0x04, 0x08, 0x10, 0x7F, 0x00 }, // N
        { 0x3E, 0x41, 0x41, 0x41, 0x3E, 0x00 }, // O
        { 0x7F, 0x09, 0x09, 0x09, 0x06, 0x00 }, // P
        { 0x3E, 0x41, 0x51, 0x21, 0x5E, 0x00 }, // Q
        { 0x7F, 0x09, 0x19, 0x29, 0x46, 0x00 }, // R
        { 0x26, 0x49, 0x49, 0x49, 0x32, 0x00 }, // S
        { 0x01, 0x01, 0x7F, 0x01, 0x01, 0x00 }, // T
        { 0x3F, 0x40, 0x40, 0x40, 0x3F, 0x00 }, // U
        { 0x1F, 0x20, 0x40, 0x20, 0x1F, 0x00 }, // V
        { 0x3F, 0x40, 0x38, 0x40, 0x3F, 0x00 }, // W
        { 0x63, 0x14, 0x08, 0x14, 0x63, 0x00 }, // X
        { 0x07, 0x08, 0x70, 0x08, 0x07, 0x00 }, // Y
        { 0x61, 0x51, 0x49, 0x45, 0x43, 0x00 }, // Z
        { 0x00, 0x7F, 0x41, 0x41, 0x00, 0x00 }, // [
        { 0x02, 0x04, 0x08, 0x10, 0x20, 0x00 }, // backslash
        { 0x00, 0x41, 0x41, 0x7F, 0x00, 0x00 }, // ]
        { 0x04, 0x02, 0x01, 0x02, 0x04, 0x00 }, // ^
        { 0x40, 0x40, 0x40, 0x40, 0x40, 0x00 }, // _
        { 0x00, 0x01, 0x02, 0x04, 0x00, 0x00 }, // `
        { 0x20, 0x54, 0x54, 0x54, 0x78, 0x00 }, // a
        { 0x7F, 0x48, 0x44, 0x44, 0x38, 0x00 }, // b
        { 0x38, 0x44, 0x44, 0x44, 0x20, 0x00 }, // c
        { 0x38, 0x44, 0x44, 0x48, 0x7F, 0x00 }, // d
        { 0x38, 0x54, 0x54, 0x54, 0x18, 0x00 }, // e
        { 0x08, 0x7E, 0x09, 0x01, 0x02, 0x00 }, // f
        { 0x08, 0x54, 0x54, 0x54, 0x3C, 0x00 }, // g
        { 0x7F, 0x08, 0x04, 0x04, 0x78, 0x00 }, // h
        { 0x00, 0x48, 0x7D, 0x40, 0x00, 0x00 }, // i
        { 0x20, 0x40, 0x44, 0x3D, 0x00, 0x00 }, // j
        { 0x7F, 0x10, 0x28, 0x44, 0x00, 0x00 }, // k
        { 0x00, 0x41, 0x7F, 0x40, 0x00, 0x00 }, // l
        { 0x7C, 0x04, 0x78, 0x04, 0x78, 0x00 }, // m
        { 0x7C, 0x08, 0x04, 0x04, 0x78, 0x00 }, // n
        { 0x38, 0x44, 0x44, 0x44, 0x38, 0x00 }, // o
        { 0x7C, 0x14, 0x14, 0x14, 0x08, 0x00 }, // p
        { 0x08, 0x14, 0x14, 0x18, 0x7C, 0x00 }, // q
        { 0x7C, 0x08, 0x04, 0x04, 0x08, 0x00 }, // r
        { 0x48, 0x54, 0x54, 0x54, 0x20, 0x00 }, // s
        { 0x04, 0x3F, 0x44, 0x40, 0x20, 0x00 }, // t
        { 0x3C, 0x40, 0x40, 0x20, 0x7C, 0x00 }, // u
        { 0x1C, 0x20, 0x40, 0x20, 0x1C, 0x00 }, // v
        { 0x3C, 0x40, 0x30, 0x40, 0x3C, 0x00 }, // w
        { 0x44, 0x28, 0x10, 0x28, 0x44, 0x00 }, // x
        { 0x0C, 0x50, 0x50, 0x50, 0x3C, 0x00 }, // y
        { 0x44, 0x64, 0x54, 0x4C, 0x44, 0x00 }, // z
        { 0x00, 0x08, 0x36, 0x41, 0x00, 0x00 }, // {
        { 0x00, 0x00, 0x7F, 0x00, 0x00, 0x00 }, // |
        { 0x00, 0x41, 0x36, 0x08, 0x00, 0x00 }, // }
        { 0x10, 0x08, 0x08, 0x10, 0x08, 0x00 } // ~
    };

void setup()
{
    Serial.begin(9600);
    
    // intialize SPI
    SPI.begin(slaveSelectPin);
    SPI.setClockDivider(slaveSelectPin, divider);
    
    // initialize pins
    pinMode(dcPin, OUTPUT);
    pinMode(rstPin, OUTPUT);
    
    // initial reset of display
    resetDisplay();
    
    // initialization of display
    digitalWrite(dcPin, LOW);
    SPI.transfer(slaveSelectPin, 0x21);
    SPI.transfer(slaveSelectPin, 0x14);
    SPI.transfer(slaveSelectPin, 0xb0);
    SPI.transfer(slaveSelectPin, 0x20);
    SPI.transfer(slaveSelectPin, 0x0c);
    digitalWrite(dcPin, HIGH);
    
    // initialize screen buffer
    resetBuffer();
}

/**
 * Loop function for main code
 */
void loop()
{
  // write my data
  printChar(9, 33, 'J');
  printChar(9, 39, 'i');
  printChar(9, 45, 'm');
  printChar(19, 21, 'M');
  printChar(19, 27, 'a');
  printChar(19, 33, 'r');
  printChar(19, 39, 't');
  printChar(19, 45, 'e');
  printChar(19, 51, 'n');
  printChar(19, 57, 's');
  printChar(29, 21, '6');
  printChar(29, 27, '4');
  printChar(29, 33, '2');
  printChar(29, 39, '0');
  printChar(29, 45, '3');
  printChar(29, 51, '2');
  printChar(29, 57, '3');
  flushBuffer();
  resetBuffer();
  delay(5000);
  // write his data
  printChar(9, 21, 'J');
  printChar(9, 27, 'o');
  printChar(9, 33, 'a');
  printChar(9, 39, 'c');
  printChar(9, 45, 'h');
  printChar(9, 51, 'i');
  printChar(9, 57, 'm');
  printChar(19, 6, 'S');
  printChar(19, 12, 'c');
  printChar(19, 18, 'h');
  printChar(19, 24, 'm');
  printChar(19, 30, 'i');
  printChar(19, 36, 'd');
  printChar(19, 42, 'b');
  printChar(19, 48, 'e');
  printChar(19, 54, 'r');
  printChar(19, 60, 'g');
  printChar(19, 66, 'e');
  printChar(19, 72, 'r');
  printChar(29, 21, '6');
  printChar(29, 27, '5');
  printChar(29, 33, '3');
  printChar(29, 39, '6');
  printChar(29, 45, '4');
  printChar(29, 51, '9');
  printChar(29, 57, '6');
  flushBuffer();
  resetBuffer();
  delay(5000);
}

/**
 * Resets display.
 */
void resetDisplay()
{
  digitalWrite(rstPin, LOW);
  delay(500);
  digitalWrite(rstPin, HIGH);
}

/**
 * Writes the screen buffer into the display.
 *
 * It does NOT reset the buffer.
 */
void flushBuffer()
{
  for (int i = 0; i < maxBufferIndex; i++) {
    SPI.transfer(slaveSelectPin, screenBuffer[i]);
  }
}

/**
 * Resets the buffer.
 *
 * Writes 0 in all fields.
 */
void resetBuffer()
{
  for (int i = 0; i < maxBufferIndex; i++) {
    screenBuffer[i] = 0x0;
  }
}

/**
 * Sets the pixel at the given location.
 * @param x
 * @param y
 * @param value
 */
void setPixel(int x, int y, int value)
{
  int bank = x / 8;
  int relativeRow = x - bank * 8;
  int bankStartIndex = bank * 84;
  int index = bankStartIndex + y;
  char bitValue = 0x0;
  
  if (value == 1) {
    bitValue = 0xff;
  }
  
  char existingValue = screenBuffer[index];
  char pixelMask = 1 << relativeRow;
  pixelMask = 0x0 | pixelMask;
  char invMask = ~ pixelMask;
  char newValue = pixelMask & bitValue;
  
  char finalExistingValue = existingValue & invMask;
  char finalValue = finalExistingValue | newValue;
  screenBuffer[index] = finalValue;
}

/**
 * Prints given char at given location.
 *
 * @param x
 * @param y
 * @param value
 *
 * @return 0 on success, -1 otherwise
 */
int printChar(int x, int y, char value)
{
  // guardians
  if (x > 40 || y > 78) {
    // absolute guardian
    return -1;
  }
  // TODO relative guardian to existing buffer

  char space = ' ';
  int arrayIndex = value - space;

  for (int i = 0; i < 6; i++) {
    char data = font[arrayIndex][i];
    for (int _x = 0; _x < 8; _x++) {
      char bitmask = 1 << _x;
      bitmask = 0x0 | bitmask;
      char pixel = bitmask & data;
      int pixelInt = 0;
      if (pixel != 0x0) {
        pixelInt = 1;
      }
      setPixel(x + _x, y + i, pixelInt);
    }
  }

  return 0;
}

void printString(int x, int y, String text)
{
  
}