#include <stdio.h>
#include <conio.h>
#include <c64.h>
#include <cbm_petscii_charmap.h>
#include <string.h>
#include <stdlib.h>
#include <stdint.h>
#include <joystick.h>

typedef uint8_t byte;
typedef uint16_t word;
typedef int8_t sbyte;

#define COLS 40
#define ROWS 25

void raster_wait(unsigned char line) {
  while (VIC.rasterline < line) ;
}

void wait_vblank() {
  raster_wait(255); // TODO
}

sbyte scroll_fine_x;
sbyte scroll_fine_y;
byte origin_x;
byte origin_y;
byte curbuf;
byte* scrnbuf[2];	// screen buffer(s)
byte tempbuf[COLS*ROWS];

void draw_cell(byte x, byte y) {
  byte xx = x + origin_x;
  byte yy = y + origin_y;
  byte ch = xx ^ yy;
  word ofs = x+y*COLS;
  scrnbuf[curbuf][ofs] = ch; // character
  tempbuf[ofs] = ch; // color
}

void scroll_draw_column(byte col) {
  byte y;
  for (y=0; y<ROWS; y++) {
    draw_cell(col, y);
  }
}

void scroll_draw_row(byte row) {
  byte x;
  for (x=0; x<COLS; x++) {
    draw_cell(x, row);
  }
}

void scroll_update_regs() {
  VIC.ctrl1 = (VIC.ctrl1 & 0xf8) | scroll_fine_y;
  VIC.ctrl2 = (VIC.ctrl2 & 0xf8) | scroll_fine_x;
}

void scroll_swap() {
  // swap hidden and visible buffers
  curbuf ^= 1;
  // wait for vblank and update registers
  wait_vblank();
  scroll_update_regs();
  VIC.addr = (VIC.addr & 0xf) | (curbuf ? 0x00 : 0x10);
}

void scroll_copy() {
  // copy temp buf to color ram
  memcpy(COLOR_RAM, tempbuf, COLS*ROWS);
  // copy visible buffer to hidden buffer
  memcpy(scrnbuf[curbuf], scrnbuf[curbuf^1], COLS*ROWS);
}

// TODO: left and up can be faster, b/c we can copy color ram downward

void scroll_left() {
  memcpy(scrnbuf[curbuf], scrnbuf[curbuf^1]+1, COLS*ROWS-1);
  ++origin_x;
  memcpy(tempbuf, COLOR_RAM+1, COLS*ROWS-1);
  scroll_draw_column(COLS-1);
  scroll_swap();
  scroll_copy();
}

void scroll_up() {
  memcpy(scrnbuf[curbuf], scrnbuf[curbuf^1]+COLS, COLS*(ROWS-1));
  ++origin_y;
  memcpy(tempbuf, COLOR_RAM+COLS, COLS*(ROWS-1));
  scroll_draw_row(ROWS-1);
  scroll_swap();
  scroll_copy();
}

void scroll_right() {
  memcpy(scrnbuf[curbuf]+1, scrnbuf[curbuf^1], COLS*ROWS-1);
  --origin_x;
  memcpy(tempbuf+1, COLOR_RAM, COLS*ROWS-1);
  scroll_draw_column(0);
  scroll_swap();
  scroll_copy();
}

void scroll_down() {
  memcpy(scrnbuf[curbuf]+COLS, scrnbuf[curbuf^1], COLS*(ROWS-1));
  --origin_y;
  memcpy(tempbuf+COLS, COLOR_RAM, COLS*(ROWS-1));
  scroll_draw_row(0);
  scroll_swap();
  scroll_copy();
}

void scroll_horiz(sbyte delta_x) {
  scroll_fine_x += delta_x;
  while (scroll_fine_x < 0) {
    scroll_fine_x += 8;
    scroll_left();
  }
  while (scroll_fine_x >= 8) {
    scroll_fine_x -= 8;
    scroll_right();
  }
}

void scroll_vert(sbyte delta_y) {
  scroll_fine_y += delta_y;
  while (scroll_fine_y < 0) {
    scroll_fine_y += 8;
    scroll_up();
  }
  while (scroll_fine_y >= 8) {
    scroll_fine_y -= 8;
    scroll_down();
  }
}

void scroll_setup() {
  scroll_fine_x = 0;
  scroll_fine_y = 0;
  origin_x = 0x80;
  origin_y = 0x80;
  curbuf = 0;
  // get screen buffer addresses
  scrnbuf[0] = (byte*) 0x8000;
  scrnbuf[1] = (byte*) 0x8400;
  // copy existing text to screen 0
  memcpy(scrnbuf[0], (byte*)0x400, COLS*ROWS);
  // copy screen 1 to screen 0
  memcpy(scrnbuf[1], scrnbuf[0], COLS*ROWS);
  
  // set VIC bank ($4000-$7FFF)
  // https://www.c64-wiki.com/wiki/VIC_bank
  CIA2.pra = 0x01;
  
  VIC.ctrl1 = 0x10; // 24 lines
  VIC.ctrl2 = 0x00; // 38 columns
}

void main(void) {
  sbyte n =0;

  clrscr();
  printf("\r\n\r\n\r\n                           Hello World!");
  printf("\r\n\r\n\r\n                  This is how we scroll");
  printf("\r\n\r\n\r\n               But color RAM can't move");
  printf("\r\n\r\n\r\n        So we have to use a temp buffer");
  printf("\r\n\r\n\r\n               And copy it just in time");
  
  scroll_setup();

  // install the joystick driver
  joy_install (joy_static_stddrv);
  
  // infinite loop
  while (1) {
    // get joystick bits
    char joy = joy_read(0);
    // move sprite based on arrow keys
    if (JOY_LEFT(joy)) scroll_horiz(-1);
    if (JOY_UP(joy)) scroll_vert(-1);
    if (JOY_RIGHT(joy)) scroll_horiz(1);
    if (JOY_DOWN(joy)) scroll_vert(1);
    // update regs
    wait_vblank();
    scroll_update_regs();
  }
}