#include <avr/io.h>
#include <avr/pgmspace.h>
#include "matrix.h"
#include "led.h"
#include "wait.h"
#include "debug.h"

#define CLK_HI() (PORTD |=  (1<<0))
#define CLK_LO() (PORTD &= ~(1<<0))
#define STATE()  (PIND & (1<<1))
#define RST_HI() (PORTD |=  (1<<3))
#define RST_LO() (PORTD &= ~(1<<3))
#define SENSE()  (PIND & (1<<2))

static matrix_row_t matrix[8] = {};


void matrix_init(void)
{
    debug_enable = true;
    debug_keyboard = true;
    debug_matrix = true;

    // PD0:Clock PD1:State PD2:Sense_All PD3:Reset(Scan_All)
    DDRD  = (1<<3) | (1<<0);
    PORTD = (1<<2) | (1<<1);

    dprintf("init\n");
}

uint8_t matrix_scan(void)
{

    // Scan_all resets counter
    RST_HI();
    wait_us(10);
    // TODO: cannot get reliable value from SENSE()
    //uint8_t s = SENSE() | STATE();
    //if (!SENSE()) return 0; // no activated key
    RST_LO();
    wait_us(10);
    //if (!s) return 0;

    // 8x8 matrix:  row:sense, col:drive, key_on:hi
    for (uint8_t col = 0; col < 8; col++) {
        for (uint8_t row = 0; row < 8; row++) {
            CLK_HI();
            wait_us(10);

            if (STATE()) {
                matrix[row] |=  (1<<col);
            } else {
                matrix[row] &= ~(1<<col);
            }
            
            // clock lo - next row
            CLK_LO();
            wait_us(10);
        }
    }
    return 1;
}

matrix_row_t matrix_get_row(uint8_t row)
{
    return matrix[row];
}

void led_set(uint8_t usb_led)
{
}