/* Copyright 2017 Jason Williams
 * Copyright 2018 Jack Humbert
 * Copyright 2019 Clueboard
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifdef __AVR__
#    include <avr/interrupt.h>
#    include <avr/io.h>
#    include <util/delay.h>
#    define led_wait_us(us) wait_us(us)
#else
#    include "ch.h"
#    include "hal.h"
#    define led_wait_us(us) chSysPolledDelayX(US2RTC(STM32_SYSCLK, us))
#endif

#include <stdint.h>
#include <stdbool.h>
#include <string.h>
#include "led_matrix_pinmatrix.h"
#include "led_tables.h"
#include "progmem.h"
#include "quantum.h"
#include "backlight.h"


/*
 * g_pwm_buffer is an array that represents the duty cycle (0-255) for each LED.
 * FIXME: Map this from the wiring matrix to a physical location matrix at some point
 */
uint8_t g_pwm_buffer[LED_MATRIX_ROWS * LED_MATRIX_COLS];
const pin_t led_row_pins[LED_MATRIX_ROWS] = LED_MATRIX_ROW_PINS;
const pin_t led_col_pins[LED_MATRIX_COLS] = LED_MATRIX_COL_PINS;


void led_matrix_pinmatrix_init_pins(void) {
    /* Set all pins to output, we are not interested in reading any information.
     */
    for (uint8_t x = 0; x < LED_MATRIX_ROWS; x++) {
        setPinOutput(led_row_pins[x]);
        writePinLow(led_row_pins[x]);
    }

    for (uint8_t x = 0; x < LED_MATRIX_COLS; x++) {
        setPinOutput(led_col_pins[x]);
        writePinLow(led_col_pins[x]);
    }
}

void led_matrix_pinmatrix_set_value(int index, uint8_t value) {
    /* Set the brighness for a single LED.
     */
    if (index >= 0 && index < LED_DRIVER_LED_COUNT) {
        g_pwm_buffer[index] = value;
    }
}

void led_matrix_pinmatrix_set_value_all(uint8_t value) {
    /* Set the brighness for all LEDs.
     */
    for (int i = 0; i < LED_DRIVER_LED_COUNT; i++) {
        led_matrix_pinmatrix_set_value(i, value);
    }
}

void led_matrix_pinmatrix_flush(void) {
    /* This is a basic bit-banged pwm implementation.
     */
    uint8_t led_count = 0;
    for (uint8_t row = 0; row < LED_MATRIX_ROWS; row++) {
        writePinLow(led_row_pins[row]);
        for (uint8_t col = 0; col < LED_MATRIX_COLS; col++) {
            /* We spend ~128us on each LED, dividing that time between lit and unlit.
             */
            const uint8_t brightness = pgm_read_byte(&CIE1931_CURVE[g_pwm_buffer[led_count]]) / 2;
            if (brightness > 0) {
                writePinHigh(led_col_pins[col]);
                for (int i = 0; i < 128; i++) {
                    if (i == brightness) {
                        writePinLow(led_col_pins[col]);
                    }
                    led_wait_us(1);
                }
            } else {
                led_wait_us(128);
            }
            led_count++;
        }
    }
}