/**
 * @file ads1015_driver.h
 * @brief Header file for ADS1015 I2C ADC driver.
 *
 * This driver provides functions to configure and read from the ADS1015
 * 12-bit, 3300 SPS I2C ADC. The device supports up to four single-ended
 * inputs or two differential inputs.
 *
 * @note Based on Texas Instruments datasheet SBAS473F – MAY 2009 – REVISED JANUARY 2025
 */

#ifndef INC_ADS1015_DRIVER_H_
#define INC_ADS1015_DRIVER_H_

#include <stdint.h>
#include <math.h>
#include "main.h" // Assuming HAL is included via main.h
#include "stm32f1xx_hal.h"
#include "flash_manager.h"

#ifdef __cplusplus
extern "C" {
#endif

/*=========================================================================
    I2C ADDRESS/BITS - Connect the following pin to ADDR
    -----------------------------------------------------------------------*/
#define ADS1015_ADDR_GND (0x48 << 1) ///< 1001 000 (ADDR -> GND)
#define ADS1015_ADDR_VDD (0x49 << 1) ///< 1001 001 (ADDR -> VDD)
#define ADS1015_ADDR_SDA (0x4A << 1) ///< 1001 010 (ADDR -> SDA)
#define ADS1015_ADDR_SCL (0x4B << 1) ///< 1001 011 (ADDR -> SCL)
/*=========================================================================*/

/*=========================================================================
    CONVERSION DELAY (in mS)
    -----------------------------------------------------------------------*/
#define ADS1015_CONVERSIONDELAY (4) ///< Conversion delay
#define ADS1115_CONVERSIONDELAY (8) ///< Conversion delay
/*=========================================================================*/

/*=========================================================================
    POINTER REGISTER
    -----------------------------------------------------------------------*/
#define ADS1015_REG_POINTER_MASK (0x03)      ///< Point mask
#define ADS1015_REG_POINTER_CONVERT (0x00)   ///< Conversion
#define ADS1015_REG_POINTER_CONFIG (0x01)    ///< Configuration
#define ADS1015_REG_POINTER_LOWTHRESH (0x02) ///< Low threshold
#define ADS1015_REG_POINTER_HITHRESH (0x03)  ///< High threshold
/*=========================================================================*/

/*=========================================================================
    CONFIG REGISTER
    -----------------------------------------------------------------------*/
#define ADS1015_REG_CONFIG_OS_MASK (0x8000) ///< OS Mask
#define ADS1015_REG_CONFIG_OS_SINGLE        (0x8000) ///< Write: Set to start a single-conversion
#define ADS1015_REG_CONFIG_OS_BUSY          (0x0000) ///< Read: Bit = 0 when conversion is in progress
#define ADS1015_REG_CONFIG_OS_NOTBUSY       (0x8000) ///< Read: Bit = 1 when device is not performing a conversion

#define ADS1015_REG_CONFIG_MUX_MASK (0x7000) ///< Mux Mask
#define ADS1015_REG_CONFIG_MUX_DIFF_0_1    (0x0000) ///< Differential P = AIN0, N = AIN1 (default)
#define ADS1015_REG_CONFIG_MUX_DIFF_0_3    (0x1000) ///< Differential P = AIN0, N = AIN3
#define ADS1015_REG_CONFIG_MUX_DIFF_1_3   (0x2000) ///< Differential P = AIN1, N = AIN3
#define ADS1015_REG_CONFIG_MUX_DIFF_2_3   (0x3000) ///< Differential P = AIN2, N = AIN3
#define ADS1015_REG_CONFIG_MUX_SINGLE_0 (0x4000) ///< Single-ended AIN0
#define ADS1015_REG_CONFIG_MUX_SINGLE_1 (0x5000) ///< Single-ended AIN1
#define ADS1015_REG_CONFIG_MUX_SINGLE_2 (0x6000) ///< Single-ended AIN2
#define ADS1015_REG_CONFIG_MUX_SINGLE_3 (0x7000) ///< Single-ended AIN3

#define ADS1015_REG_CONFIG_PGA_MASK (0x0E00)   ///< PGA Mask
#define ADS1015_REG_CONFIG_PGA_6_144V (0x0000) ///< +/-6.144V range = Gain 2/3
#define ADS1015_REG_CONFIG_PGA_4_096V (0x0200) ///< +/-4.096V range = Gain 1
#define ADS1015_REG_CONFIG_PGA_2_048V  (0x0400) ///< +/-2.048V range = Gain 2 (default)
#define ADS1015_REG_CONFIG_PGA_1_024V (0x0600) ///< +/-1.024V range = Gain 4
#define ADS1015_REG_CONFIG_PGA_0_512V (0x0800) ///< +/-0.512V range = Gain 8
#define ADS1015_REG_CONFIG_PGA_0_256V (0x0A00) ///< +/-0.256V range = Gain 16

#define ADS1015_REG_CONFIG_MODE_MASK (0x0100)   ///< Mode Mask
#define ADS1015_REG_CONFIG_MODE_CONTIN (0x0000) ///< Continuous conversion mode
#define ADS1015_REG_CONFIG_MODE_SINGLE    (0x0100) ///< Power-down single-shot mode (default)

#define ADS1015_REG_CONFIG_DR_MASK (0x00E0)   ///< Data Rate Mask
#define ADS1015_REG_CONFIG_DR_128SPS (0x0000) ///< 128 samples per second
#define ADS1015_REG_CONFIG_DR_250SPS (0x0020) ///< 250 samples per second
#define ADS1015_REG_CONFIG_DR_490SPS (0x0040) ///< 490 samples per second
#define ADS1015_REG_CONFIG_DR_920SPS (0x0060) ///< 920 samples per second
#define ADS1015_REG_CONFIG_DR_1600SPS  (0x0080) ///< 1600 samples per second (default)
#define ADS1015_REG_CONFIG_DR_2400SPS (0x00A0) ///< 2400 samples per second
#define ADS1015_REG_CONFIG_DR_3300SPS (0x00C0) ///< 3300 samples per second

#define ADS1015_REG_CONFIG_CMODE_MASK (0x0010) ///< CMode Mask
#define ADS1015_REG_CONFIG_CMODE_TRAD (0x0000) ///< Traditional comparator with hysteresis (default)
#define ADS1015_REG_CONFIG_CMODE_WINDOW (0x0010) ///< Window comparator

#define ADS1015_REG_CONFIG_CPOL_MASK (0x0008) ///< CPol Mask
#define ADS1015_REG_CONFIG_CPOL_ACTVLOW     (0x0000) ///< ALERT/RDY pin is low when active (default)
#define ADS1015_REG_CONFIG_CPOL_ACTVHI      (0x0008) ///< ALERT/RDY pin is high when active

#define ADS1015_REG_CONFIG_CLAT_MASK       (0x0004) ///< Determines if ALERT/RDY pin latches once asserted
#define ADS1015_REG_CONFIG_CLAT_NONLAT     (0x0000) ///< Non-latching comparator (default)
#define ADS1015_REG_CONFIG_CLAT_LATCH (0x0004) ///< Latching comparator

#define ADS1015_REG_CONFIG_CQUE_MASK (0x0003) ///< CQue Mask
#define ADS1015_REG_CONFIG_CQUE_1CONV      (0x0000) ///< Assert ALERT/RDY after one conversions
#define ADS1015_REG_CONFIG_CQUE_2CONV      (0x0001) ///< Assert ALERT/RDY after two conversions
#define ADS1015_REG_CONFIG_CQUE_4CONV      (0x0002) ///< Assert ALERT/RDY after four conversions
#define ADS1015_REG_CONFIG_CQUE_NONE       (0x0003) ///< Disable the comparator and put ALERT/RDY in high state (default)

#define 		ADS1015_KELVIN_OFFSET	  	273.15f // Constante de conversão Kelvin
#define			ADS1015_ADC_MAX 			4095.0f
#define		 	ADS1015_ADC_VREF 			1024.0f //1.024V in mV
#define			PH_DELTA_CALIB				3.0f	// Difference between pH 7 and pH 4 used in calibration
#define			ADS1015_PH_AVERAGES			16

#define 		ADS1015_TEMPERATURE_ROW			0
#define 		ADS1015_PH_4_BUFFER_ROW			1
#define 		ADS1015_PH_7_BUFFER_ROW			2
#define			ADS1015_INTERPOLATION_ROWS		3
#define			ADS1015_PH_TEMP_INTER_POINTS	12
#define 		ADS1015_TIME_PER_SAMPLE			9 // 9ms
/*=========================================================================*/

/* Config Register para single-shot:
 * OS=1 (start), MUX[2:0]=000b (AIN0 single-ended, ajustar se necessário),
 * PGA[2:0]=010b (±2.048V default, ajustar ganho conforme sensor),
 * MODE=1b (single-shot/power-down), DR[2:0]=100b (1600SPS, ajustar se necessário),
 * COMP_QUE[1:0]=11b (comparador desabilitado, ALERT/RDY em alta impedância)
 */
#define 	ADS1015_CONFIG_START_SINGLE   (ADS1015_REG_CONFIG_OS_SINGLE | ADS1015_REG_CONFIG_MUX_DIFF_0_1 | ADS1015_REG_CONFIG_PGA_1_024V | ADS1015_REG_CONFIG_MODE_SINGLE | ADS1015_REG_CONFIG_DR_128SPS | ADS1015_REG_CONFIG_CMODE_TRAD | ADS1015_REG_CONFIG_CPOL_ACTVLOW | ADS1015_REG_CONFIG_CLAT_NONLAT | ADS1015_REG_CONFIG_CQUE_NONE)   // 0x8583  // valor de reset já tem OS=1, então só precisa garantir MODE=1b


#define ADS1015_BURST_SIZE      			1
#define ADS1015_TRIM_COUNT      			0
#define ADS1015_HISTORY_SIZE    			16
#define ADS1015_IIR_ALPHA       			0.1f

#define ADS1015_PH_FLOAT_SCALE_MIN			1000.0f
#define ADS1015_PH_FLOAT_SCALE_MAX			3000.0f
#define ADS1015_PH_OUT_SCALE_MIN			0
#define ADS1015_PH_OUT_SCALE_MAX			4096



/** Gain settings */
typedef enum {
  GAIN_TWOTHIRDS = ADS1015_REG_CONFIG_PGA_6_144V,
  GAIN_ONE = ADS1015_REG_CONFIG_PGA_4_096V,
  GAIN_TWO = ADS1015_REG_CONFIG_PGA_2_048V,
  GAIN_FOUR = ADS1015_REG_CONFIG_PGA_1_024V,
  GAIN_EIGHT = ADS1015_REG_CONFIG_PGA_0_512V,
  GAIN_SIXTEEN = ADS1015_REG_CONFIG_PGA_0_256V
} adsGain_t;

extern I2C_HandleTypeDef hi2c1;


typedef struct {
	uint16_t m_i2cAddress;      ///< the I2C address
	uint32_t m_conversionDelay; ///< conversion delay
	uint8_t m_bitShift;        ///< bit shift amount
	adsGain_t m_gain;          ///< ADC gain
	I2C_HandleTypeDef* hi2c;    // Handle for I2C
}ADS1015_I2C;

extern ADS1015_I2C i2c_ads1015;

typedef enum
{
    ADS1015_IDLE,
    ADS1015_START_CONVERSION,
    ADS1015_WAIT_CONVERSION,
    ADS1015_READ_DATA,
    ADS1015_WAIT_NEXT_SAMPLE
} ADS1015_State_t;

extern volatile ADS1015_State_t ads1015_state;

typedef struct {
    float    burst[ADS1015_BURST_SIZE];
    uint8_t  burst_index;
    float    history[ADS1015_HISTORY_SIZE];
    uint8_t  history_index;
    uint8_t  history_count;
    float    iir_state;
    uint8_t  iir_initialized;
    float    filtered_value;
    uint8_t  value_valid;
} ADS1015_filter_t;


extern ADS1015_filter_t g_ph_filter;
extern volatile uint32_t g_ms_counter; // Increment each 1ms Timer

void ADS1015_Init(void);
void ADS1015_Process_System(void);
void ADS1015_Sort_Array(float *v, uint8_t n);
float ADS1015_Trimmed_Mean(float *buffer, uint8_t n, uint8_t trim);
void ADS1015_Process_Sample(ADS1015_filter_t *ch, float raw);
void ADS1015_pH_NewSample(float raw);
uint16_t ADS1015_Compress_To_IntScale(float value);



#endif /* INC_ADS1015_DRIVER_H_ */
