/*
	Copyright 2016 - 2020 Benjamin Vedder	benjamin@vedder.se

	This file is part of the VESC firmware.

	The VESC firmware 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 3 of the License, or
    (at your option) any later version.

    The VESC firmware 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/>.
    */

#include "ledpwm.h"
#include "ch.h"
#include "hal.h"
#include <string.h>
#include <math.h>

// Private variables
static volatile int led_values[LEDPWM_LED_NUM];
static uint8_t gamma_table[LEDPWM_CNT_TOP + 1];

void ledpwm_init(void) {
	memset((int*)led_values, 0, sizeof(led_values));

	// Generate gamma correction table
	for (int i = 0;i < (LEDPWM_CNT_TOP + 1);i++) {
		gamma_table[i] = (int)roundf(powf((float)i / (float)LEDPWM_CNT_TOP, 1.0 / 0.45) * (float)LEDPWM_CNT_TOP);
	}
}

/*
 * Set the intensity for one led. The intensity value is mapped to a PWM value
 * according to human luminance perception.
 *
 * Intensity range is 0.0 to 1.0
 */
void ledpwm_set_intensity(unsigned int led, float intensity) {
	if (led >= LEDPWM_LED_NUM) {
		return;
	}

	if (intensity < 0.0) {
		intensity = 0.0;
	}

	if (intensity > 1.0) {
		intensity = 1.0;
	}

	led_values[led] = gamma_table[(int)(intensity * LEDPWM_CNT_TOP)];
}

void ledpwm_led_on(int led) {
	if (led >= LEDPWM_LED_NUM) {
		return;
	}

	led_values[led] = LEDPWM_CNT_TOP;
}

void ledpwm_led_off(int led) {
	if (led >= LEDPWM_LED_NUM) {
		return;
	}

	led_values[led] = 0;
}

/*
 * Call this function as fast as possible, with a deterministic rate.
 */
void ledpwm_update_pwm(void) {
	static int cnt = 0;
	cnt++;
	if (cnt == LEDPWM_CNT_TOP) {
		cnt = 0;
	}

	if (cnt >= led_values[LED_GREEN]) {
		LED_GREEN_OFF();
	} else {
		LED_GREEN_ON();
	}

	if (cnt >= led_values[LED_RED]) {
		LED_RED_OFF();
	} else {
		LED_RED_ON();
	}

#ifdef LED_PWM1_ON
	if (cnt >= led_values[LED_HW1]) {
		LED_PWM1_OFF();
	} else {
		LED_PWM1_ON();
	}
#endif

#ifdef LED_PWM2_ON
	if (cnt >= led_values[LED_HW2]) {
		LED_PWM2_OFF();
	} else {
		LED_PWM2_ON();
	}
#endif

#ifdef LED_PWM3_ON
	if (cnt >= led_values[LED_HW3]) {
		LED_PWM3_OFF();
	} else {
		LED_PWM3_ON();
	}
#endif

#ifdef LED_PWM4_ON
	if (cnt >= led_values[LED_HW4]) {
		LED_PWM4_OFF();
	} else {
		LED_PWM4_ON();
	}
#endif

}