// Emisar D3AA helper functions
// Copyright (C) 2023 Selene ToyKeeper
// SPDX-License-Identifier: GPL-3.0-or-later
#pragma once

#include "fsm/chan-rgbaux.c"
#include "fsm/ramping.h"
#include "ui/anduril/misc.h"

void set_level_zero();

void set_level_main(uint8_t level);
bool gradual_tick_main(uint8_t gt);


Channel channels[] = {
    { // main LEDs
        .set_level    = set_level_main,
        .gradual_tick = gradual_tick_main
    },
    RGB_AUX_CHANNELS
};


void set_level_zero() {
    DAC_LVL  = 0;  // DAC off
    DAC_VREF = V10;  // low Vref
    HDR_ENABLE_PORT &= ~(1 << HDR_ENABLE_PIN);  // HDR off

    // prevent post-off flash
    IN_NFET_ENABLE_PORT |= (1 << IN_NFET_ENABLE_PIN);
    delay_4ms(IN_NFET_DELAY_TIME/4);
    IN_NFET_ENABLE_PORT &= ~(1 << IN_NFET_ENABLE_PIN);

    // turn off boost last
    BST_ENABLE_PORT &= ~(1 << BST_ENABLE_PIN);  // BST off
}

// single set of LEDs with 1 regulated power channel
// and low/high HDR plus low/high Vref as different "gears"
void set_level_main(uint8_t level) {
    uint8_t noflash = 0;

    // when turning on from off, use IN_NFET to prevent a flash
    if ((! actual_level) && (level < HDR_ENABLE_LEVEL_MIN)) {
        noflash = 1;
        IN_NFET_ENABLE_PORT |= (1 << IN_NFET_ENABLE_PIN);
    }

    // BST on first, to give it a few extra microseconds to spin up
    BST_ENABLE_PORT |= (1 << BST_ENABLE_PIN);

    // pre-load ramp data so it can be assigned faster later
    // DAC level register is left-aligned
    PWM1_DATATYPE dac_lvl  = PWM1_GET(level) << 6;
    PWM2_DATATYPE dac_vref = PWM2_GET(level);

    // enable HDR on top half of ramp
    if (level >= (HDR_ENABLE_LEVEL_MIN-1))
        HDR_ENABLE_PORT |= (1 << HDR_ENABLE_PIN);
    else
        HDR_ENABLE_PORT &= ~(1 << HDR_ENABLE_PIN);

    // set these in successive clock cycles to avoid getting out of sync
    // (minimizes ramp bumps when changing gears)
    DAC_LVL  = dac_lvl;
    DAC_VREF = dac_vref;

    if (noflash) {
        // wait for flash prevention to finish
        delay_4ms(IN_NFET_DELAY_TIME/4);
        IN_NFET_ENABLE_PORT &= ~(1 << IN_NFET_ENABLE_PIN);
    }
}

bool gradual_tick_main(uint8_t gt) {
    // if HDR and Vref "engine gear" is the same, do a small adjustment...
    // otherwise, simply jump to the next ramp level
    //   and let set_level() handle any gear changes

    // different gear = full adjustment
    PWM2_DATATYPE vref_next = PWM2_GET(gt);
    if (vref_next != DAC_VREF) return true;  // let parent set_level() for us

    // same gear = small adjustment
    PWM1_DATATYPE dac_now  = DAC_LVL >> 6;  // register is left-aligned
    PWM1_DATATYPE dac_next = PWM1_GET(gt);

    // only adjust 1 dac level, max is 1023
    // (but speed it up with "#define GRADUAL_ADJUST_SPEED  4" elsewhere)
    GRADUAL_ADJUST_SIMPLE(dac_next, dac_now);

    DAC_LVL = dac_now << 6;

    if (dac_next == dac_now) return true;  // done

    return false;  // not done yet
}


#ifdef USE_VOLTAGE_DIVIDER
uint8_t voltage_raw2cooked(uint16_t measurement) {
    // In : 65535 * BATTLVL / 1.024V
    // Out: uint8_t: Vbat * 40
    // BATTLVL = Vbat * (100.0/(330+100)) = Vbat / 4.3
    // So, Out = In * 4.3 / 1600
    // (plus a bit of fudging to fix the slope and offset,
    //  based on measuring actual hardware)
    uint8_t result = (uint32_t)(measurement + (65535 * 4 / 1024))
                     * 43 / 16000;
    return result;
}
#endif

#ifdef USE_WEAK_BATTERY_PROTECTION
uint8_t quick_volt_measurement() {
    // take the average of a few samples
    // (assumes the ADC is in voltage mode and running continuously)
    uint16_t total = 0;
    for (uint8_t i=0; i<8; i++) {
        uint16_t m = adc_raw[0];
        total += voltage_raw2cooked(m);
        delay_zero();
    }
    uint8_t v = total / 8;
    return v;
}

void detect_weak_battery() {
    // guess at the cell strength with a load test...
    // - measure voltage while LEDs off
    // - measure again with LEDs on
    // - determine how much to limit power
    // - blink to indicate weak battery mode, if active

    uint16_t resting, loaded;

    set_level(0);
    resting = quick_volt_measurement();

    set_level(WEAK_BATTERY_CHECK_LEVEL);
    loaded = quick_volt_measurement();
    set_level(0);

    int16_t diff = resting - loaded;
    uint8_t extra_blinks = 0;

    if (loaded <= DUAL_VOLTAGE_LOW_LOW) {
        // weak or empty AA battery has a low limit
        ramp_level_hard_limit = WEAK_BATTERY_LOWEST_LIMIT;
        extra_blinks = 2;
    } else if (loaded >= VOLTAGE_RED) {
        // reasonably strong li-ion battery
        ramp_level_hard_limit = WEAK_BATTERY_HIGHEST_LIMIT;
    } else if (diff <= (-5 * 4)) {
        // marginal battery, dropped a lot under mild load
        ramp_level_hard_limit = WEAK_BATTERY_MEDIUM_LIMIT;
        extra_blinks = 1;
    }

    for (uint8_t i=0; i<extra_blinks; i++) {
        delay_4ms(300/4);
        blink_once();
    }

    voltage = resting;
    battcheck();

    voltage = loaded;
    battcheck();

    //if (diff < 0)
    //    blink_num(-diff);
}
#endif