//
// Created by MightyPork on 2018/02/03.
//

#include "platform.h"
#include "unit_base.h"

#define SIPO_INTERNAL
#include "_sipo_internal.h"

/** Allocate data structure and set defaults */
error_t USIPO_preInit(Unit *unit)
{
    struct priv *priv = unit->data = calloc_ck(1, sizeof(struct priv));
    if (priv == NULL) return E_OUT_OF_MEM;

    priv->store_pname = 'A';
    priv->store_pnum = 0;
    priv->store_pol = true;

    priv->shift_pname = 'A';
    priv->shift_pnum = 1;
    priv->shift_pol = true;

    priv->clear_pname = 'A';
    priv->clear_pnum = 2;
    priv->clear_pol = false;

    priv->data_pname = 'A';
    priv->data_pins = (1<<3);

    return E_SUCCESS;
}

/** Finalize unit set-up */
error_t USIPO_init(Unit *unit)
{
    bool suc = true;
    struct priv *priv = unit->data;

    // --- Parse config ---
    priv->store_ll = hw_pin2ll(priv->store_pnum, &suc);
    priv->store_port = hw_port2periph(priv->store_pname, &suc);
    Resource store_rsc = hw_pin2resource(priv->store_pname, priv->store_pnum, &suc);
    if (!suc) return E_BAD_CONFIG;
    TRY(rsc_claim(unit, store_rsc));

    priv->shift_ll = hw_pin2ll(priv->shift_pnum, &suc);
    priv->shift_port = hw_port2periph(priv->shift_pname, &suc);
    Resource shift_rsc = hw_pin2resource(priv->shift_pname, priv->shift_pnum, &suc);
    if (!suc) return E_BAD_CONFIG;
    TRY(rsc_claim(unit, shift_rsc));

    priv->clear_ll = hw_pin2ll(priv->clear_pnum, &suc);
    priv->clear_port = hw_port2periph(priv->clear_pname, &suc);
    Resource clear_rsc = hw_pin2resource(priv->clear_pname, priv->clear_pnum, &suc);
    if (!suc) return E_BAD_CONFIG;
    TRY(rsc_claim(unit, clear_rsc));

    // Claim all needed pins
    TRY(rsc_claim_gpios(unit, priv->data_pname, priv->data_pins));
    priv->data_port = hw_port2periph(priv->data_pname, &suc);

    // --- Init hardware ---

    priv->data_width = 0;
    for (int i = 0; i < 16; i++) {
        if (priv->data_pins & (1 << i)) {
            uint32_t ll_pin = hw_pin2ll((uint8_t) i, &suc);
            LL_GPIO_SetPinMode(priv->data_port, ll_pin, LL_GPIO_MODE_OUTPUT);
            LL_GPIO_SetPinOutputType(priv->data_port, ll_pin, LL_GPIO_OUTPUT_PUSHPULL);
            LL_GPIO_SetPinSpeed(priv->data_port, ll_pin, LL_GPIO_SPEED_FREQ_HIGH);
            priv->data_width++;
        }
    }

    // Set the initial state - zeros
    priv->data_port->ODR &= ~priv->data_pins;


    // STORE
    LL_GPIO_SetPinMode(priv->store_port, priv->store_ll, LL_GPIO_MODE_OUTPUT);
    LL_GPIO_SetPinOutputType(priv->store_port, priv->store_ll, LL_GPIO_OUTPUT_PUSHPULL);
    LL_GPIO_SetPinSpeed(priv->store_port, priv->store_ll, LL_GPIO_SPEED_FREQ_HIGH);

    if (priv->store_pol)
        LL_GPIO_ResetOutputPin(priv->store_port, priv->store_ll);
    else
        LL_GPIO_SetOutputPin(priv->store_port, priv->store_ll);

    // SHIFT
    LL_GPIO_SetPinMode(priv->shift_port, priv->shift_ll, LL_GPIO_MODE_OUTPUT);
    LL_GPIO_SetPinOutputType(priv->shift_port, priv->shift_ll, LL_GPIO_OUTPUT_PUSHPULL);
    LL_GPIO_SetPinSpeed(priv->shift_port, priv->shift_ll, LL_GPIO_SPEED_FREQ_HIGH);

    if (priv->shift_pol)
        LL_GPIO_ResetOutputPin(priv->shift_port, priv->shift_ll);
    else
        LL_GPIO_SetOutputPin(priv->shift_port, priv->shift_ll);

    // CLEAR
    LL_GPIO_SetPinMode(priv->clear_port, priv->clear_ll, LL_GPIO_MODE_OUTPUT);
    LL_GPIO_SetPinOutputType(priv->clear_port, priv->clear_ll, LL_GPIO_OUTPUT_PUSHPULL);
    LL_GPIO_SetPinSpeed(priv->clear_port, priv->clear_ll, LL_GPIO_SPEED_FREQ_HIGH);

    if (priv->clear_pol)
        LL_GPIO_ResetOutputPin(priv->clear_port, priv->clear_ll);
    else
        LL_GPIO_SetOutputPin(priv->clear_port, priv->clear_ll);

    // initial clear
    UU_SIPO_DirectClear(unit);

    return E_SUCCESS;
}


/** Tear down the unit */
void USIPO_deInit(Unit *unit)
{
    // Release all resources, deinit pins
    rsc_teardown(unit);

    // Free memory
    free_ck(unit->data);
}