/* Copyright 2023 Dual Tachyon
 * https://github.com/DualTachyon
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 *     Unless required by applicable law or agreed to in writing, software
 *     distributed under the License is distributed on an "AS IS" BASIS,
 *     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *     See the License for the specific language governing permissions and
 *     limitations under the License.
 */

#include <stdint.h>
#include "bsp/dp32g030/gpio.h"
#include "bsp/dp32g030/spi.h"
#include "driver/gpio.h"
#include "driver/spi.h"
#include "driver/st7565.h"
#include "driver/system.h"

uint8_t gStatusLine[128];
uint8_t gFrameBuffer[7][128];

void ST7565_DrawLine(uint8_t Column, uint8_t Line, uint16_t Size, const uint8_t *pBitmap, bool bIsClearMode)
{
	uint16_t i;

	SPI_ToggleMasterMode(&SPI0->CR, false);
	ST7565_SelectColumnAndLine(Column + 4U, Line);
	GPIO_SetBit(&GPIOB->DATA, 9);

	if (bIsClearMode == false) {
		for (i = 0; i < Size; i++) {
			while ((SPI0->FIFOST & SPI_FIFOST_TFF_MASK) != SPI_FIFOST_TFF_NOT_FULL) {
			}
			SPI0->WDR = pBitmap[i];
		}
	} else {
		for (i = 0; i < Size; i++) {
			while ((SPI0->FIFOST & SPI_FIFOST_TFF_MASK) != SPI_FIFOST_TFF_NOT_FULL) {
			}
			SPI0->WDR = 0;
		}
	}

	SPI_WaitForUndocumentedTxFifoStatusBit();
	SPI_ToggleMasterMode(&SPI0->CR, true);
}

void ST7565_BlitFullScreen(void)
{
	uint8_t Line;
	uint8_t Column;

	SPI_ToggleMasterMode(&SPI0->CR,false);
	ST7565_WriteByte(0x40);

	for (Line = 0; Line < 7; Line++) {
		ST7565_SelectColumnAndLine(4U, Line + 1U);
		GPIO_SetBit(&GPIOB->DATA, 9U);
		for (Column = 0; Column < 128; Column++) {
			while ((SPI0->FIFOST & SPI_FIFOST_TFF_MASK) != SPI_FIFOST_TFF_NOT_FULL) {
			}
			SPI0->WDR = gFrameBuffer[Line][Column];
		}
		SPI_WaitForUndocumentedTxFifoStatusBit();
	}

	SYSTEM_DelayUs(20);
	SPI_ToggleMasterMode(&SPI0->CR, true);
}

void ST7565_BlitStatusLine(void)
{
	uint8_t i;

	SPI_ToggleMasterMode(&SPI0->CR, false);
	ST7565_WriteByte(0x40);
	ST7565_SelectColumnAndLine(4, 0);
	GPIO_SetBit(&GPIOB->DATA, 9);

	for (i = 0; i < 0x80; i++) {
		while ((SPI0->FIFOST & SPI_FIFOST_TFF_MASK) != SPI_FIFOST_TFF_NOT_FULL) {
		}
		SPI0->WDR = gStatusLine[i];
	}
	SPI_WaitForUndocumentedTxFifoStatusBit();
	SPI_ToggleMasterMode(&SPI0->CR, true);
}

void ST7565_FillScreen(uint8_t Value)
{
	uint8_t i, j;

	SPI_ToggleMasterMode(&SPI0->CR, false);
	for (i = 0; i < 8; i++) {
		ST7565_SelectColumnAndLine(0, i);
		GPIO_SetBit(&GPIOB->DATA, 9);
		for (j = 0; j < 132; j++) {
			while ((SPI0->FIFOST & SPI_FIFOST_TFF_MASK) != SPI_FIFOST_TFF_NOT_FULL) {
			}
			SPI0->WDR = Value;
		}
		SPI_WaitForUndocumentedTxFifoStatusBit();
	}
	SPI_ToggleMasterMode(&SPI0->CR, true);
}

void ST7565_Init(void)
{
	SPI0_Init();
	ST7565_Configure_GPIO_B11();
	SPI_ToggleMasterMode(&SPI0->CR, false);
	ST7565_WriteByte(0xE2);
	SYSTEM_DelayUs(0x78);
	ST7565_WriteByte(0xA2);
	ST7565_WriteByte(0xC0);
	ST7565_WriteByte(0xA1);
	ST7565_WriteByte(0xA6);
	ST7565_WriteByte(0xA4);
	ST7565_WriteByte(0x24);
	ST7565_WriteByte(0x81);
	ST7565_WriteByte(0x1F);
	ST7565_WriteByte(0x2B);
	SYSTEM_DelayUs(1);
	ST7565_WriteByte(0x2E);
	SYSTEM_DelayUs(1);
	ST7565_WriteByte(0x2F);
	ST7565_WriteByte(0x2F);
	ST7565_WriteByte(0x2F);
	ST7565_WriteByte(0x2F);
	SYSTEM_DelayUs(0x28);
	ST7565_WriteByte(0x40);
	ST7565_WriteByte(0xAF);
	SPI_WaitForUndocumentedTxFifoStatusBit();
	SPI_ToggleMasterMode(&SPI0->CR, true);
	ST7565_FillScreen(0x00);
}

void ST7565_Configure_GPIO_B11(void)
{
	GPIO_SetBit(&GPIOB->DATA, 11);
	SYSTEM_DelayUs(1);
	GPIO_ClearBit(&GPIOB->DATA, 11);
	SYSTEM_DelayUs(20);
	GPIO_SetBit(&GPIOB->DATA, 11);
	SYSTEM_DelayUs(120);
}

void ST7565_SelectColumnAndLine(uint8_t Column, uint8_t Line)
{
	GPIO_ClearBit(&GPIOB->DATA, 9);
	while ((SPI0->FIFOST & SPI_FIFOST_TFF_MASK) != SPI_FIFOST_TFF_NOT_FULL) {
	}
	SPI0->WDR = Line + 0xB0;
	while ((SPI0->FIFOST & SPI_FIFOST_TFF_MASK) != SPI_FIFOST_TFF_NOT_FULL) {
	}
	SPI0->WDR = ((Column >> 4) & 0x0F) | 0x10;
	while ((SPI0->FIFOST & SPI_FIFOST_TFF_MASK) != SPI_FIFOST_TFF_NOT_FULL) {
	}
	SPI0->WDR = ((Column >> 0) & 0x0F);
	SPI_WaitForUndocumentedTxFifoStatusBit();
}

void ST7565_WriteByte(uint8_t Value)
{
	GPIO_ClearBit(&GPIOB->DATA, 9);
	while ((SPI0->FIFOST & SPI_FIFOST_TFF_MASK) != SPI_FIFOST_TFF_NOT_FULL) {
	}
	SPI0->WDR = Value;
}