#pragma once

#include "esphome/components/i2c/i2c.h"  // Ensure this header defines i2c::I2CDevice
#include "esphome/core/component.h"     // Ensure this header defines Component
#include "esphome/components/sensor/sensor.h"

#include <vector>
#include <esphome/core/gpio.h>

namespace esphome {
namespace ads1115_int {

enum ADS1115Multiplexer {
  ADS1115_MULTIPLEXER_P0_N1 = 0b000,
  ADS1115_MULTIPLEXER_P0_N3 = 0b001,
  ADS1115_MULTIPLEXER_P1_N3 = 0b010,
  ADS1115_MULTIPLEXER_P2_N3 = 0b011,
  ADS1115_MULTIPLEXER_P0_NG = 0b100,
  ADS1115_MULTIPLEXER_P1_NG = 0b101,
  ADS1115_MULTIPLEXER_P2_NG = 0b110,
  ADS1115_MULTIPLEXER_P3_NG = 0b111,
  ADS1115_MULTIPLEXER_INVALID = 0xFF
};

enum ADS1115Gain {
  ADS1115_GAIN_6P144 = 0b000,
  ADS1115_GAIN_4P096 = 0b001,
  ADS1115_GAIN_2P048 = 0b010,
  ADS1115_GAIN_1P024 = 0b011,
  ADS1115_GAIN_0P512 = 0b100,
  ADS1115_GAIN_0P256 = 0b101,
};

enum ADS1115Resolution {
  ADS1115_16_BITS = 16,
  ADS1015_12_BITS = 12,
};

enum ADS1115Samplerate {
  ADS1115_8SPS = 0b000,
  ADS1115_16SPS = 0b001,
  ADS1115_32SPS = 0b010,
  ADS1115_64SPS = 0b011,
  ADS1115_128SPS = 0b100,
  ADS1115_250SPS = 0b101,
  ADS1115_475SPS = 0b110,
  ADS1115_860SPS = 0b111
};

class ADS1115Component : public Component, public i2c::I2CDevice {
  private:
    struct mux_data_item {
      ADS1115Multiplexer multiplexer;
      ADS1115Gain gain;
      ADS1115Resolution resolution;
      ADS1115Samplerate samplerate;
      sensor::Sensor *sensor_ptr; // ADS1115Sensor *sensor_ptr
      mux_data_item() { multiplexer = ADS1115_MULTIPLEXER_INVALID; }
    }; 
    void request_data();
    int next_mux_data_index();
    void update_sensor();
    int mux_data_index_{0};
    mux_data_item muxdata_[4];
    uint64_t count_{0};
    
  public:
    void setup() override;
    void dump_config() override;
    /// HARDWARE_LATE setup priority
    // we will force single-shot mode for interrupt operation; kept continuous_mode_ for ease of reversal if needed
    //void set_continuous_mode(bool continuous_mode) { continuous_mode_ = continuous_mode; }

    // Method to read converted data                            
    double read_data(int16_t& raw_value, int mux_index);
    bool start_single_shot_conversion();
    void set_alert_pin(InternalGPIOPin *pin) { alert_pin_ = pin; }
    bool set_data_ready_mode();
    void set_mux_data(ADS1115Multiplexer multiplexer, ADS1115Gain gain, ADS1115Resolution resolution, ADS1115Samplerate samplerate, sensor::Sensor *sensor_ptr);
  
 protected:
    uint16_t prev_config_{0};
    bool continuous_mode_{false};
    InternalGPIOPin *alert_pin_;
    static void isr(ADS1115Component *arg);

};

}  // namespace ads1115_int
}  // namespace esphome