/*
 *  state_machine.h
 *
 *  Created on: 07.07.2024
 *      Author: Hamza
 */

#ifndef INC_STATE_MACHINE_H
#define INC_STATE_MACHINE_H

#include <stdint.h>
#include <stdbool.h>
#include "ADBMS_LL_Driver.h"
#include <AMS_HighLevel.h>
#include <errors.h>
#include <PWM_control.h>
#include <status_LED.h>
#include <TMP1075.h>

typedef enum {              //  states -> 3 bit. valid transitions: (all could transition to error)
  STATE_INACTIVE,           // INACTIVE   ->  PRECHARGE, CHARGING, ERROR  
  STATE_PRECHARGE,          // PRECHARGE  ->  INACTIVE, READY, DISCHARGE, ERROR
  STATE_READY,              // READY      ->  ACTIVE, DISCHARGE, ERROR
  STATE_ACTIVE,             // ACTIVE     ->  READY, DISCHARGE, ERROR
  STATE_DISCHARGE,          // DISCHARGE  ->  INACTIVE, PRECHARGE, ERROR
  STATE_CHARGING_PRECHARGE,
  STATE_CHARGING,           // CHARGING   ->  INACTIVE, DISCHARGE, ERROR
  STATE_ERROR,              // ERROR      ->  INACTIVE, DISCHARGE, ERROR
} State;
  
typedef struct {  
  uint16_t bms_timeout : 1;
  uint16_t bms_fault : 1;
  uint16_t temperature_error : 1;
  uint16_t current_error : 1;
  uint16_t current_sensor_missing : 1;
  uint16_t voltage_error : 1;
  uint16_t voltage_missing : 1;
  uint16_t state_transition_fail : 1;
  uint16_t eeprom_error : 1;
  uint16_t : 7; // padding
} ErrorKind;

//typedef enum {} WarningKind;

typedef struct {
  State current_state;
  State target_state;
  uint16_t error_source; // TSErrorSource (bitmask)
  ErrorKind error_type; // TSErrorKind
} StateHandle;

typedef enum { RELAY_MAIN, RELAY_PRECHARGE } Relay;

extern StateHandle state;
extern int32_t RELAY_BAT_SIDE_VOLTAGE;
extern int32_t RELAY_ESC_SIDE_VOLTAGE;
extern int32_t CURRENT_MEASUREMENT;

void sm_init();
void sm_update();
void sm_handle_ams_in(const uint8 *data);
void sm_precharge_discharge_manager();
void sm_powerground_manager();
void sm_calibrate_powerground();
void sm_balancing();

void sm_check_errors();
void sm_set_error_source();

State sm_update_inactive();
State sm_update_precharge();
State sm_update_ready();
State sm_update_active();
State sm_update_discharge();
State sm_update_charging_precharge(); 
State sm_update_charging();
State sm_update_error();

void sm_set_relay_positions(State state);
void sm_set_relay(Relay relay, bool closed);
void sm_check_battery_temperature(int8_t* id, int16_t* temp);
int16_t sm_return_cell_temperature(int id);
int16_t sm_return_cell_voltage(int id);

void sm_test_cycle_states();

#endif /* "INC_STATE_MACHINE_H" */