/*!
 *  @file Adafruit_SCD30.cpp
 *
 *  @mainpage Adafruit SCD30 CO2, Temperature, and Humidity sensor
 *
 *  @section intro_sec Introduction
 *
 * 	I2C Driver for the Library for the SCD30 CO2, Temperature, and Humidity
 * sensor
 *
 * 	This is a library for the Adafruit SCD30 breakout:
 * 	https://www.adafruit.com/product/48xx
 *
 * 	Adafruit invests time and resources providing this open source code,
 *  please support Adafruit and open-source hardware by purchasing products from
 * 	Adafruit!
 *
 *  @section dependencies Dependencies
 *  This library depends on the Adafruit BusIO library
 *
 *  This library depends on the Adafruit Unified Sensor library
 *
 *  @section author Author
 *
 *  Bryan Siepert for Adafruit Industries
 *
 * 	@section license License
 *
 * 	BSD (see license.txt)
 *
 * 	@section  HISTORY
 *
 *     v1.0 - First release
 */

#include "Arduino.h"
#include <Wire.h>

#include "Adafruit_SCD30.h"

static uint8_t crc8(const uint8_t *data, int len);

/**
 * @brief Construct a new Adafruit_SCD30::Adafruit_SCD30 object
 *
 */
Adafruit_SCD30::Adafruit_SCD30(void) {}

/**
 * @brief Destroy the Adafruit_SCD30::Adafruit_SCD30 object
 *
 */
Adafruit_SCD30::~Adafruit_SCD30(void) {
  if (temp_sensor)
    delete temp_sensor;
  if (humidity_sensor)
    delete humidity_sensor;
}

/*!
 *    @brief  Sets up the hardware and initializes I2C
 *    @param  i2c_address
 *            The I2C address to be used.
 *    @param  wire
 *            The Wire object to be used for I2C connections.
 *    @param  sensor_id
 *            The unique ID to differentiate the sensors from others
 *    @return True if initialization was successful, otherwise false.
 */
bool Adafruit_SCD30::begin(uint8_t i2c_address, TwoWire *wire,
                           int32_t sensor_id) {
  if (i2c_dev) {
    delete i2c_dev; // remove old interface
  }

  i2c_dev = new Adafruit_I2CDevice(i2c_address, wire);

  if (!i2c_dev->begin()) {
    return false;
  }

  return _init(sensor_id);
}
// bool Adafruit_SCD30::begin_UART(void){}

/*!  @brief Initializer for post i2c init
 *   @param sensor_id Optional unique ID for the sensor set
 *   @returns True if chip identified and initialized
 */
bool Adafruit_SCD30::_init(int32_t sensor_id) {

  _sensorid_humidity = sensor_id;
  _sensorid_temp = sensor_id + 1;

  reset();

  // first I2C xfer after reset can fail, double tapping seems to get by it
  if (!startContinuousMeasurement()) {
    if (!startContinuousMeasurement())
      return false;
  }
  if (!setMeasurementInterval(2)) {
    return false;
  }
  humidity_sensor = new Adafruit_SCD30_Humidity(this);
  temp_sensor = new Adafruit_SCD30_Temp(this);
  return true;
}

/**
 * @brief Performs a software reset initializing registers to their power on
 * state.
 *
 */
void Adafruit_SCD30::reset(void) {
  sendCommand(SCD30_CMD_SOFT_RESET);
  delay(30);
}

/**
 * @brief Ask the sensor if new data is ready to read
 *
 * @return true: data is available false: no new data available
 */
bool Adafruit_SCD30::dataReady(void) {
  return (readRegister(SCD30_CMD_GET_DATA_READY) == 1);
}

/**
 * @brief Set the amount of time between measurements
 *
 * @param interval The time between measurements in seconds. Must be from 2-1800
 * seconds. The default value set on sensor initialization is 2 seconds.
 *
 * @return true: success false: failure
 */
bool Adafruit_SCD30::setMeasurementInterval(uint16_t interval) {
  if ((interval < 2) || (interval > 1800)) {
    return false;
  }
  return sendCommand(SCD30_CMD_SET_MEASUREMENT_INTERVAL, interval);
}

/**
 * @brief Read the current amount of time between measurements
 *
 * @return uint16_t The current measurement interval in seconds.
 */
uint16_t Adafruit_SCD30::getMeasurementInterval(void) {
  return readRegister(SCD30_CMD_SET_MEASUREMENT_INTERVAL);
}

/**
 * @brief Gets the enable status of the SCD30's self calibration routine
 *
 * @return true: enabled false: disabled
 */
bool Adafruit_SCD30::selfCalibrationEnabled(void) {
  return (readRegister(SCD30_CMD_AUTOMATIC_SELF_CALIBRATION) == 1);
}

/**
 * @brief Enable or disable the SCD30's self calibration routine
 *
 * @param enabled true: enable false: disable
 * @return true: success false: failure
 */
bool Adafruit_SCD30::selfCalibrationEnabled(bool enabled) {
  return sendCommand(SCD30_CMD_AUTOMATIC_SELF_CALIBRATION, enabled);
}

/**
 * @brief Tell the SCD30 to start taking measurements continuously
 *
 * @param pressure an optional pressure offset to correct for in millibar (mBar)
 * @return true: succes false: failure
 */
bool Adafruit_SCD30::startContinuousMeasurement(uint16_t pressure) {
  return sendCommand(SCD30_CMD_CONTINUOUS_MEASUREMENT, pressure);
}

/**
 * @brief Read the current ambient pressure offset
 *
 * @return uint16_t  current ambient pressure offset in millibar (mBar)
 */
uint16_t Adafruit_SCD30::getAmbientPressureOffset(void) {
  return readRegister(SCD30_CMD_CONTINUOUS_MEASUREMENT);
}

/**
 * @brief Set the altitude offset that the SCD30 should correct for
 *
 * **Note:** This value is saved to the SCD30's internal storage and is reloaded
 * on sensor power up.
 * @param altitude The altitude offset in meters above sea level.
 * @return true: success false: failure
 */
bool Adafruit_SCD30::setAltitudeOffset(uint16_t altitude) {
  return sendCommand(SCD30_CMD_SET_ALTITUDE_COMPENSATION, altitude);
}
/**
 * @brief Get the current altitude offset
 *
 * @return uint16_t The current altitude offset value in meters above sea level.
 */
uint16_t Adafruit_SCD30::getAltitudeOffset(void) {
  return readRegister(SCD30_CMD_SET_ALTITUDE_COMPENSATION);
}

/**
 * @brief Set a temperature offset
 *
 * @param temp_offset The **positive** temperature offset to set in hundreths of
 * a degree C ie:
 *
 * 1015 => 10.15 degrees C
 * 31337 => 313.37 degrees C
 *
 *
 * **Note:** This value is saved to the SCD30's internal storage and is reloaded
 * on sensor power up.
 * @return true: success false: failure
 */
bool Adafruit_SCD30::setTemperatureOffset(uint16_t temp_offset) {
  return sendCommand(SCD30_CMD_SET_TEMPERATURE_OFFSET, temp_offset);
}
/**
 * @brief Get the current temperature offset in hundreths of a degree C
 *
 * @return uint16_t the current temperature offset
 */
uint16_t Adafruit_SCD30::getTemperatureOffset(void) {
  return readRegister(SCD30_CMD_SET_TEMPERATURE_OFFSET);
}

/**
 * @brief Force the SCD30 to recalibrate with a given reference value
 *
 * @param reference The calibration reference value in ppm from
 * 400-2000 ppm.
 *
 * **Note:** This value is saved to the SCD30's internal storage and is reloaded
 * on sensor power up.
 *
 * **Setting a reference value and forcing recalibration will override any
 * previous automatic self-calibration.**
 * @return true: success false: failure
 */
bool Adafruit_SCD30::forceRecalibrationWithReference(uint16_t reference) {
  if ((reference < 400) || (reference > 2000)) {
    return false;
  }
  return sendCommand(SCD30_CMD_SET_FORCED_RECALIBRATION_REF, reference);
}

/**
 * @brief Get the current forced recalibration reference value
 *
 * @return uint16_t The current reference value in ppm
 */
uint16_t Adafruit_SCD30::getForcedCalibrationReference(void) {
  return readRegister(SCD30_CMD_SET_FORCED_RECALIBRATION_REF);
}
/**
 * @brief  Updates the measurement data for all sensors simultaneously
 *
 * @return true: success false: failure
 */
bool Adafruit_SCD30::read(void) {
  uint8_t buffer[18];

  buffer[0] = (SCD30_CMD_READ_MEASUREMENT >> 8) & 0xFF;
  buffer[1] = SCD30_CMD_READ_MEASUREMENT & 0xFF;

  if (!i2c_dev->write(buffer, 2)) {
    return false;
  }

  delay(4); // delay between write and read specified by the datasheet

  if (!i2c_dev->read(buffer, 18)) {
    return false;
  }

  // loop through the bytes we read, 3 at a time for i=MSB, i+1=LSB, i+2=CRC
  for (uint8_t i = 0; i < 18; i += 3) {
    if (crc8(buffer + i, 2) != buffer[i + 2]) {
      // we got a bad CRC, fail out
      return false;
    }
  }
  // CRCs are good, unpack floats
  uint32_t co2 = 0, temp = 0, hum = 0;

  co2 |= buffer[0];
  co2 <<= 8;
  co2 |= buffer[1];
  co2 <<= 8;
  co2 |= buffer[3];
  co2 <<= 8;
  co2 |= buffer[4];

  temp |= buffer[6];
  temp <<= 8;
  temp |= buffer[7];
  temp <<= 8;
  temp |= buffer[9];
  temp <<= 8;
  temp |= buffer[10];

  hum |= buffer[12];
  hum <<= 8;
  hum |= buffer[13];
  hum <<= 8;
  hum |= buffer[15];
  hum <<= 8;
  hum |= buffer[16];

  memcpy(&CO2, &co2, sizeof(CO2));
  memcpy(&temperature, &temp, sizeof(temperature));
  memcpy(&relative_humidity, &hum, sizeof(relative_humidity));

  return true;
}

bool Adafruit_SCD30::sendCommand(uint16_t command) {
  uint8_t buffer[2];
  buffer[0] = (command >> 8) & 0xFF;
  buffer[1] = command & 0xFF;

  return i2c_dev->write(buffer, sizeof(buffer));
}
bool Adafruit_SCD30::sendCommand(uint16_t command, uint16_t argument) {

  uint8_t buffer[5];
  buffer[0] = (command >> 8) & 0xFF;
  buffer[1] = command & 0xFF;
  buffer[2] = argument >> 8;
  buffer[3] = argument & 0xFF;
  buffer[4] = crc8(buffer + 2, 2);
  return i2c_dev->write(buffer, sizeof(buffer));
}
uint16_t Adafruit_SCD30::readRegister(uint16_t reg_address) {
  uint8_t buffer[2];
  buffer[0] = (reg_address >> 8) & 0xFF;
  buffer[1] = reg_address & 0xFF;
  // the SCD30 really wants a stop before the read!
  i2c_dev->write(buffer, 2);
  delay(4); // delay between write and read specified by the datasheet
  i2c_dev->read(buffer, 2);
  return (uint16_t)(buffer[0] << 8 | (buffer[1] & 0xFF));
}

/*!
    @brief  Gets an Adafruit Unified Sensor object for the presure sensor
   component
    @return Adafruit_Sensor pointer to humidity sensor
 */
Adafruit_Sensor *Adafruit_SCD30::getHumiditySensor(void) {
  return humidity_sensor;
}

/*!
    @brief  Gets an Adafruit Unified Sensor object for the temp sensor component
    @return Adafruit_Sensor pointer to temperature sensor
 */
Adafruit_Sensor *Adafruit_SCD30::getTemperatureSensor(void) {
  return temp_sensor;
}

/**************************************************************************/
/*!
    @brief  Gets the humidity sensor and temperature values as sensor events
    @param  humidity Sensor event object that will be populated with humidity
   data
    @param  temp Sensor event object that will be populated with temp data
    @returns True
*/
/**************************************************************************/
bool Adafruit_SCD30::getEvent(sensors_event_t *humidity,
                              sensors_event_t *temp) {
  uint32_t t = millis();
  if (!read()) {
    return false;
  }

  // use helpers to fill in the events
  fillHumidityEvent(humidity, t);
  fillTempEvent(temp, t);
  return true;
}

void Adafruit_SCD30::fillHumidityEvent(sensors_event_t *humidity,
                                       uint32_t timestamp) {
  memset(humidity, 0, sizeof(sensors_event_t));
  humidity->version = sizeof(sensors_event_t);
  humidity->sensor_id = _sensorid_humidity;
  humidity->type = SENSOR_TYPE_RELATIVE_HUMIDITY;
  humidity->timestamp = timestamp;
  humidity->relative_humidity = relative_humidity;
}

void Adafruit_SCD30::fillTempEvent(sensors_event_t *temp, uint32_t timestamp) {
  memset(temp, 0, sizeof(sensors_event_t));
  temp->version = sizeof(sensors_event_t);
  temp->sensor_id = _sensorid_temp;
  temp->type = SENSOR_TYPE_AMBIENT_TEMPERATURE;
  temp->timestamp = timestamp;
  temp->temperature = temperature;
}

/**************************************************************************/
/*!
    @brief  Gets the sensor_t data for the SCD30's humidity
*/
/**************************************************************************/
void Adafruit_SCD30_Humidity::getSensor(sensor_t *sensor) {
  /* Clear the sensor_t object */
  memset(sensor, 0, sizeof(sensor_t));

  /* Insert the sensor name in the fixed length char array */
  strncpy(sensor->name, "SCD30_P", sizeof(sensor->name) - 1);
  sensor->name[sizeof(sensor->name) - 1] = 0;
  sensor->version = 1;
  sensor->sensor_id = _sensorID;
  sensor->type = SENSOR_TYPE_RELATIVE_HUMIDITY;
  sensor->min_delay = 0;
  sensor->min_value = 260;
  sensor->max_value = 1260;
  // 4096 LSB = 1 hPa >>  1 LSB = 1/4096 hPa >> 1 LSB =  2.441e-4 hPa
  sensor->resolution = 2.441e-4;
}

/**************************************************************************/
/*!
    @brief  Gets the humidity as a standard sensor event
    @param  event Sensor event object that will be populated
    @returns True
*/
/**************************************************************************/
bool Adafruit_SCD30_Humidity::getEvent(sensors_event_t *event) {
  _theSCD30->read();
  _theSCD30->fillHumidityEvent(event, millis());

  return true;
}

/**************************************************************************/
/*!
    @brief  Gets the sensor_t data for the SCD30's tenperature
*/
/**************************************************************************/
void Adafruit_SCD30_Temp::getSensor(sensor_t *sensor) {
  /* Clear the sensor_t object */
  memset(sensor, 0, sizeof(sensor_t));

  /* Insert the sensor name in the fixed length char array */
  strncpy(sensor->name, "SCD30_T", sizeof(sensor->name) - 1);
  sensor->name[sizeof(sensor->name) - 1] = 0;
  sensor->version = 1;
  sensor->sensor_id = _sensorID;
  sensor->type = SENSOR_TYPE_AMBIENT_TEMPERATURE;
  sensor->min_delay = 0;
  sensor->min_value = -30;
  sensor->max_value = 105;
  // 480 LSB = 1°C >> 1 LSB = 1/480°C >> 1 LSB =  0.00208 °C
  sensor->resolution = 0.00208;
}

/**************************************************************************/
/*!
    @brief  Gets the temperature as a standard sensor event
    @param  event Sensor event object that will be populated
    @returns True
*/
/**************************************************************************/
bool Adafruit_SCD30_Temp::getEvent(sensors_event_t *event) {
  _theSCD30->read();
  _theSCD30->fillTempEvent(event, millis());

  return true;
}

/**
 * Performs a CRC8 calculation on the supplied values.
 *
 * @param data  Pointer to the data to use when calculating the CRC8.
 * @param len   The number of bytes in 'data'.
 *
 * @return The computed CRC8 value.
 */
static uint8_t crc8(const uint8_t *data, int len) {
  /*
   *
   * CRC-8 formula from page 14 of SHT spec pdf
   *
   * Test data 0xBE, 0xEF should yield 0x92
   *
   * Initialization data 0xFF
   * Polynomial 0x31 (x8 + x5 +x4 +1)
   * Final XOR 0x00
   */

  const uint8_t POLYNOMIAL(0x31);
  uint8_t crc(0xFF);

  for (int j = len; j; --j) {
    crc ^= *data++;

    for (int i = 8; i; --i) {
      crc = (crc & 0x80) ? (crc << 1) ^ POLYNOMIAL : (crc << 1);
    }
  }
  return crc;
}