#pragma once
//    FILE: INA229.h
//  AUTHOR: Rob Tillaart
// VERSION: 0.6.1
//    DATE: 2025-01-22
// PURPOSE: Arduino library for the INA229, SPI, 20 bit, voltage, current and power sensor.
//     URL: https://github.com/RobTillaart/INA229
//
//  Read the datasheet for the details


#include "Arduino.h"
#include "SPI.h"


#define INA229_LIB_VERSION          (F("0.6.1"))


#ifndef __SPI_CLASS__
  //  MBED must be tested before RP2040
  #if defined(ARDUINO_ARCH_MBED)
  #define __SPI_CLASS__   SPIClass
  #elif defined(ARDUINO_ARCH_RP2040)
  #define __SPI_CLASS__   SPIClassRP2040
  #else
  #define __SPI_CLASS__   SPIClass
  #endif
#endif


//  for setMode() and getMode()
enum ina229_mode_enum {
  INA229_MODE_SHUTDOWN            = 0x00,
  INA229_MODE_TRIG_BUS            = 0x01,
  INA229_MODE_TRIG_SHUNT          = 0x02,
  INA229_MODE_TRIG_BUS_SHUNT      = 0x03,
  INA229_MODE_TRIG_TEMP           = 0x04,
  INA229_MODE_TRIG_TEMP_BUS       = 0x05,
  INA229_MODE_TRIG_TEMP_SHUNT     = 0x06,
  INA229_MODE_TRIG_TEMP_BUS_SHUNT = 0x07,

  INA229_MODE_SHUTDOWN2           = 0x08,
  INA229_MODE_CONT_BUS            = 0x09,
  INA229_MODE_CONT_SHUNT          = 0x0A,
  INA229_MODE_CONT_BUS_SHUNT      = 0x0B,
  INA229_MODE_CONT_TEMP           = 0x0C,
  INA229_MODE_CONT_TEMP_BUS       = 0x0D,
  INA229_MODE_CONT_TEMP_SHUNT     = 0x0E,
  INA229_MODE_CONT_TEMP_BUS_SHUNT = 0x0F
};


//  for setAverage() and getAverage()
enum ina229_average_enum {
    INA229_1_SAMPLE     = 0,
    INA229_4_SAMPLES    = 1,
    INA229_16_SAMPLES   = 2,
    INA229_64_SAMPLES   = 3,
    INA229_128_SAMPLES  = 4,
    INA229_256_SAMPLES  = 5,
    INA229_512_SAMPLES  = 6,
    INA229_1024_SAMPLES = 7
};


//  for Bus, shunt and temperature conversion timing.
enum ina229_timing_enum {
    INA229_50_us   = 0,
    INA229_84_us   = 1,
    INA229_150_us  = 2,
    INA229_280_us  = 3,
    INA229_540_us  = 4,
    INA229_1052_us = 5,
    INA229_2074_us = 6,
    INA229_4120_us = 7
};


//  for diagnose/alert() bit fields.
enum ina229_diag_enum {
  INA229_DIAG_MEMORY_STATUS      = 0,
  INA229_DIAG_CONVERT_COMPLETE   = 1,
  INA229_DIAG_POWER_OVER_LIMIT   = 2,
  INA229_DIAG_BUS_UNDER_LIMIT    = 3,
  INA229_DIAG_BUS_OVER_LIMIT     = 4,
  INA229_DIAG_SHUNT_UNDER_LIMIT  = 5,
  INA229_DIAG_SHUNT_OVER_LIMIT   = 6,
  INA229_DIAG_TEMP_OVER_LIMIT    = 7,
  INA229_DIAG_RESERVED           = 8,
  INA229_DIAG_MATH_OVERFLOW      = 9,
  INA229_DIAG_CHARGE_OVERFLOW    = 10,
  INA229_DIAG_ENERGY_OVERFLOW    = 11,
  INA229_DIAG_ALERT_POLARITY     = 12,
  INA229_DIAG_SLOW_ALERT         = 13,
  INA229_DIAG_CONVERT_READY      = 14,
  INA229_DIAG_ALERT_LATCH        = 15
};


class INA229
{
public:
  //  HARDWARE SPI
  INA229(uint8_t select, __SPI_CLASS__ * mySPI = &SPI);
  //  SOFTWARE SPI
  INA229(uint8_t select, uint8_t dataIn, uint8_t dataOut, uint8_t clock);

  bool     begin();

  //
  //  CORE FUNCTIONS + scale wrappers.
  //
  //       BUS VOLTAGE
  float    getBusVoltage();     //  Volt
  float    getBusVolt()         { return getBusVoltage(); };
  float    getBusMilliVolt()    { return getBusVoltage()   * 1e3; };
  float    getBusMicroVolt()    { return getBusVoltage()   * 1e6; };

  //       SHUNT VOLTAGE
  float    getShuntVoltage();   //  Volt
  float    getShuntVolt()       { return getShuntVoltage(); };
  float    getShuntMilliVolt()  { return getShuntVoltage() * 1e3; };
  float    getShuntMicroVolt()  { return getShuntVoltage() * 1e6; };

  //  raw integer interface.
  int32_t  getShuntVoltageRAW();

  //       SHUNT CURRENT
  float    getCurrent();        //  Ampere
  float    getAmpere()          { return getCurrent(); };
  float    getMilliAmpere()     { return getCurrent()      * 1e3; };
  float    getMicroAmpere()     { return getCurrent()      * 1e6; };

  //       POWER
  float    getPower();          //  Watt
  float    getWatt()            { return getPower(); };
  float    getMilliWatt()       { return getPower()        * 1e3; };
  float    getMicroWatt()       { return getPower()        * 1e6; };
  float    getKiloWatt()        { return getPower()        * 1e-3; };

  //       TEMPERATURE
  float    getTemperature();    //  Celsius

  //  the Energy and Charge functions are returning double as they have higher accuracy.
  //       ENERGY
  double   getEnergy();         //  Joule or watt second
  double   getJoule()           { return getEnergy(); };
  double   getMegaJoule()       { return getEnergy()       * 1e-6; };
  double   getKiloJoule()       { return getEnergy()       * 1e-3; };
  double   getMilliJoule()      { return getEnergy()       * 1e3; };
  double   getMicroJoule()      { return getEnergy()       * 1e6; };
  double   getWattHour()        { return getEnergy()       * (1.0  / 3600.0); };
  double   getKiloWattHour()    { return getEnergy()       * (1e-3 / 3600.0); };

  //  CHARGE
  double   getCharge();         //  Coulombs
  double   getCoulomb()         { return getCharge(); };
  double   getMilliCoulomb()    { return getCharge()       * 1e3; };
  double   getMicroCoulomb()    { return getCharge()       * 1e6; };


  //
  //  CONFIG REGISTER 0
  //  read datasheet for details, section 7.6.1.1, page 21
  //
  void     reset();
  //  value: 0 == normal operation,  1 = clear registers
  bool     setAccumulation(uint8_t value);
  bool     getAccumulation();
  //  Conversion delay in 0..255 steps of 2 ms
  void     setConversionDelay(uint8_t steps);
  uint8_t  getConversionDelay();
  void     setTemperatureCompensation(bool on);
  bool     getTemperatureCompensation();
  //  flag = false => 164 mV, true => 41 mV
  bool     setADCRange(bool flag);
  bool     getADCRange();

  //
  //  CONFIG ADC REGISTER 1
  //  read datasheet for details, section 7.6.1.2, page 21++
  //
  bool     setMode(uint8_t mode = INA229_MODE_CONT_TEMP_BUS_SHUNT);
  uint8_t  getMode();
  //  default value = ~1 milliseconds for all.
  bool     setBusVoltageConversionTime(uint8_t bvct = INA229_1052_us);
  uint8_t  getBusVoltageConversionTime();
  bool     setShuntVoltageConversionTime(uint8_t svct = INA229_1052_us);
  uint8_t  getShuntVoltageConversionTime();
  bool     setTemperatureConversionTime(uint8_t tct = INA229_1052_us);
  uint8_t  getTemperatureConversionTime();
  bool     setAverage(uint8_t avg = INA229_1_SAMPLE);
  uint8_t  getAverage();

  //
  //  SHUNT CALIBRATION REGISTER 2
  //  read datasheet for details. use with care.
  //  maxCurrent <= 204, (in fact no limit)
  //  shunt >= 0.0001.
  //  returns _current_LSB;
  int      setMaxCurrentShunt(float maxCurrent, float shunt);
  bool     isCalibrated()    { return _current_LSB != 0.0; };
  float    getMaxCurrent();
  float    getShunt();
  float    getCurrentLSB();
  float    getCurrentLSB_mA() { return _current_LSB * 1e3; };
  float    getCurrentLSB_uA() { return _current_LSB * 1e6; };

  //
  //  SHUNT TEMPERATURE COEFFICIENT REGISTER 3
  //  read datasheet for details, page 16.
  //  ppm = 0..16383.
  bool     setShuntTemperatureCoefficent(uint16_t ppm = 0);
  uint16_t getShuntTemperatureCoefficent();


  //
  //  DIAGNOSE ALERT REGISTER 11  (0x0B)
  //  read datasheet for details, section 7.6.1.12, page 26++.
  //
  void     setDiagnoseAlert(uint16_t flags);
  uint16_t getDiagnoseAlert();
  //  INA229.h has an enum for the bit fields.
  //  See INA229_diag_enum above
  void     setDiagnoseAlertBit(uint8_t bit);
  void     clearDiagnoseAlertBit(uint8_t bit);
  uint16_t getDiagnoseAlertBit(uint8_t bit);


  //
  //  THRESHOLD AND LIMIT REGISTERS 12-17
  //  read datasheet for details, section 7.3.7, page 16++
  //
  void     setShuntOvervoltageTH(uint16_t threshold);
  uint16_t getShuntOvervoltageTH();
  void     setShuntUndervoltageTH(uint16_t threshold);
  uint16_t getShuntUndervoltageTH();
  void     setBusOvervoltageTH(uint16_t threshold);
  uint16_t getBusOvervoltageTH();
  void     setBusUndervoltageTH(uint16_t threshold);
  uint16_t getBusUndervoltageTH();
  void     setTemperatureOverLimitTH(uint16_t threshold);
  uint16_t getTemperatureOverLimitTH();
  void     setPowerOverLimitTH(uint16_t threshold);
  uint16_t getPowerOverLimitTH();


  //
  //  MANUFACTURER and ID REGISTER 3E and 3F
  //
  //                               typical value
  uint16_t getManufacturer();  //  0x5449
  uint16_t getDieID();         //  0x0229
  uint16_t getRevision();      //  0x0001


  //       speed in Hz
  void     setSPIspeed(uint32_t speed);
  uint32_t getSPIspeed();

  //       Debugging
  bool     usesHWSPI();


private:
  //  max 4 bytes
  uint32_t _readRegister(uint8_t reg, uint8_t bytes);
  //  always 5 bytes, mode == U ==> unsigned, otherwise signed
  double   _readRegisterF(uint8_t reg, char mode);
  uint16_t _writeRegister(uint8_t reg, uint16_t value);

  float    _current_LSB;
  float    _shunt;
  float    _maxCurrent;
  bool     _ADCRange;

  uint8_t  _dataIn = 255;
  uint8_t  _dataOut= 255;
  uint8_t  _clock  = 255;
  uint8_t  _select = 255;
  bool     _hwSPI  = false;
  uint32_t _SPIspeed = 10000000;  //  max speed

  uint8_t  swSPI_transfer(uint8_t value);

  __SPI_CLASS__ * _mySPI;
  SPISettings   _spi_settings;
};


//  -- END OF FILE --


