zh_htu/zh_htu.c
2024-07-24 14:12:16 +03:00

229 lines
9.0 KiB
C
Executable File

#include "zh_htu.h"
#define I2C_DATA_SIZE 3 // Sensor data size (in bytes).
#define TEMPERATURE_MEASUREMENT_TIME 50 // Sensor temperature measurement time (in milliseconds).
#define HUMIDITY_MEASUREMENT_TIME 20 // Sensor humidity measurement time (in milliseconds).
#define RESET_TIME 20 // Sensor reset time (in milliseconds).
#define I2C_ADDRESS 0x40 // Sensor I2C address.
#define I2C_TEMPERATURE_DATA_READ_COMMAND 0xF3 // Command for read sensor temperature data.
#define I2C_HUMIDITY_DATA_READ_COMMAND 0xF5 // Command for read sensor humidity data.
#define I2C_RESET_COMMAND 0xFE // Command for reset sensor.
#define I2C_REGISTER_READ_COMMAND 0xE7 // Command for read sensor register.
#define I2C_REGISTER_WRITE_COMMAND 0xE6 // Command for write sensor register.
static zh_htu_init_config_t _init_config = {0};
static bool _is_initialized = false;
#ifndef CONFIG_IDF_TARGET_ESP8266
static i2c_master_dev_handle_t _htu_handle = {0};
#endif
static const char *TAG = "zh_htu";
static uint8_t _calc_crc(const uint8_t *buf, size_t len);
esp_err_t zh_htu_init(const zh_htu_init_config_t *config)
{
ESP_LOGI(TAG, "HTU initialization begin.");
if (config == NULL)
{
ESP_LOGE(TAG, "HTU initialization fail. Invalid argument.");
return ESP_ERR_INVALID_ARG;
}
_init_config = *config;
#ifndef CONFIG_IDF_TARGET_ESP8266
i2c_device_config_t htu_config = {
.dev_addr_length = I2C_ADDR_BIT_LEN_7,
.device_address = I2C_ADDRESS,
.scl_speed_hz = 100000,
};
i2c_master_bus_add_device(_init_config.i2c_handle, &htu_config, &_htu_handle);
if (i2c_master_probe(_init_config.i2c_handle, I2C_ADDRESS, 1000 / portTICK_PERIOD_MS) != ESP_OK)
{
ESP_LOGE(TAG, "HTU initialization fail. Sensor not connected or not responded.");
return ESP_ERR_NOT_FOUND;
}
#endif
esp_err_t esp_err = ESP_OK;
uint8_t command = I2C_STATUS_READ_COMMAND;
uint8_t sensor_data = 0;
#ifdef CONFIG_IDF_TARGET_ESP8266
i2c_cmd_handle_t i2c_cmd_handle = i2c_cmd_link_create();
i2c_master_start(i2c_cmd_handle);
i2c_master_write_byte(i2c_cmd_handle, I2C_ADDRESS << 1 | I2C_MASTER_WRITE, true);
i2c_master_write_byte(i2c_cmd_handle, command, true);
i2c_master_stop(i2c_cmd_handle);
esp_err = i2c_master_cmd_begin(_init_config.i2c_port, i2c_cmd_handle, 1000 / portTICK_PERIOD_MS);
i2c_cmd_link_delete(i2c_cmd_handle);
#else
esp_err = i2c_master_transmit(_htu_handle, &command, sizeof(command), 1000 / portTICK_PERIOD_MS);
#endif
if (esp_err != ESP_OK)
{
ESP_LOGE(TAG, "HTU initialization fail. I2C driver error at line %d.", __LINE__);
return ESP_ERR_INVALID_RESPONSE;
}
#ifdef CONFIG_IDF_TARGET_ESP8266
i2c_cmd_handle = i2c_cmd_link_create();
i2c_master_start(i2c_cmd_handle);
i2c_master_write_byte(i2c_cmd_handle, I2C_ADDRESS << 1 | I2C_MASTER_READ, true);
i2c_master_read_byte(i2c_cmd_handle, &sensor_data, I2C_MASTER_NACK);
i2c_master_stop(i2c_cmd_handle);
esp_err = i2c_master_cmd_begin(_init_config.i2c_port, i2c_cmd_handle, 1000 / portTICK_PERIOD_MS);
i2c_cmd_link_delete(i2c_cmd_handle);
#else
esp_err = i2c_master_receive(_htu_handle, &sensor_data, sizeof(sensor_data), 1000 / portTICK_PERIOD_MS);
#endif
if (esp_err != ESP_OK)
{
ESP_LOGE(TAG, "HTU initialization fail. I2C driver error at line %d.", __LINE__);
return ESP_ERR_INVALID_RESPONSE;
}
if ((sensor_data & 0b00001000) == 0)
{
uint8_t command[] = {I2C_INIT_COMMAND};
#ifdef CONFIG_IDF_TARGET_ESP8266
i2c_cmd_handle_t i2c_cmd_handle = i2c_cmd_link_create();
i2c_master_start(i2c_cmd_handle);
i2c_master_write_byte(i2c_cmd_handle, I2C_ADDRESS << 1 | I2C_MASTER_WRITE, true);
i2c_master_write_byte(i2c_cmd_handle, command[0], true);
i2c_master_write_byte(i2c_cmd_handle, command[1], true);
i2c_master_write_byte(i2c_cmd_handle, command[2], true);
i2c_master_stop(i2c_cmd_handle);
esp_err = i2c_master_cmd_begin(_init_config.i2c_port, i2c_cmd_handle, 1000 / portTICK_PERIOD_MS);
i2c_cmd_link_delete(i2c_cmd_handle);
#else
esp_err = i2c_master_transmit(_htu_handle, command, sizeof(command), 1000 / portTICK_PERIOD_MS);
#endif
if (esp_err != ESP_OK)
{
ESP_LOGE(TAG, "HTU initialization fail. I2C driver error at line %d.", __LINE__);
return ESP_ERR_INVALID_RESPONSE;
}
}
ESP_LOGI(TAG, "HTU initialization success.");
_is_initialized = true;
return ESP_OK;
}
esp_err_t zh_htu_read(float *humidity, float *temperature)
{
ESP_LOGI(TAG, "HTU read begin.");
if (humidity == NULL || temperature == NULL)
{
ESP_LOGE(TAG, "HTU read fail. Invalid argument.");
return ESP_ERR_INVALID_ARG;
}
if (_is_initialized == false)
{
ESP_LOGE(TAG, "HTU read fail. HTU not initialized.");
return ESP_ERR_NOT_FOUND;
}
esp_err_t esp_err = ESP_OK;
uint8_t sensor_data[I2C_DATA_SIZE] = {0};
uint8_t command[] = {I2C_DATA_READ_COMMAND};
#ifdef CONFIG_IDF_TARGET_ESP8266
i2c_cmd_handle_t i2c_cmd_handle = i2c_cmd_link_create();
i2c_master_start(i2c_cmd_handle);
i2c_master_write_byte(i2c_cmd_handle, I2C_ADDRESS << 1 | I2C_MASTER_WRITE, true);
i2c_master_write_byte(i2c_cmd_handle, command[0], true);
i2c_master_write_byte(i2c_cmd_handle, command[1], true);
i2c_master_write_byte(i2c_cmd_handle, command[2], true);
i2c_master_stop(i2c_cmd_handle);
esp_err = i2c_master_cmd_begin(_init_config.i2c_port, i2c_cmd_handle, 1000 / portTICK_PERIOD_MS);
i2c_cmd_link_delete(i2c_cmd_handle);
#else
esp_err = i2c_master_transmit(_htu_handle, command, sizeof(command), 1000 / portTICK_PERIOD_MS);
#endif
if (esp_err != ESP_OK)
{
ESP_LOGE(TAG, "HTU read fail. I2C driver error at line %d.", __LINE__);
return ESP_ERR_INVALID_RESPONSE;
}
// vTaskDelay(MEASUREMENT_TIME / portTICK_PERIOD_MS);
#ifdef CONFIG_IDF_TARGET_ESP8266
i2c_cmd_handle = i2c_cmd_link_create();
i2c_master_start(i2c_cmd_handle);
i2c_master_write_byte(i2c_cmd_handle, I2C_ADDRESS << 1 | I2C_MASTER_READ, true);
for (uint8_t i = 0; i < sizeof(sensor_data); ++i)
{
i2c_master_read_byte(i2c_cmd_handle, &sensor_data[i], i == (sizeof(sensor_data) - 1) ? I2C_MASTER_NACK : I2C_MASTER_ACK);
}
i2c_master_stop(i2c_cmd_handle);
esp_err = i2c_master_cmd_begin(_init_config.i2c_port, i2c_cmd_handle, 1000 / portTICK_PERIOD_MS);
i2c_cmd_link_delete(i2c_cmd_handle);
#else
esp_err = i2c_master_receive(_htu_handle, sensor_data, sizeof(sensor_data), 1000 / portTICK_PERIOD_MS);
#endif
if (esp_err != ESP_OK)
{
ESP_LOGE(TAG, "HTU read fail. I2C driver error at line %d.", __LINE__);
return ESP_ERR_INVALID_RESPONSE;
}
if ((sensor_data[0] & 0b01000000) != 0)
{
ESP_LOGE(TAG, "HTU read fail. Timeout exceeded.");
return ESP_ERR_TIMEOUT;
}
if (_calc_crc(sensor_data, I2C_DATA_SIZE - 1) != sensor_data[6])
{
ESP_LOGE(TAG, "HTU read fail. Invalid CRC.");
return ESP_ERR_INVALID_CRC;
}
*humidity = (((((uint32_t)sensor_data[1]) << 16) | (((uint32_t)sensor_data[2]) << 8) | (((uint32_t)sensor_data[3]) << 0)) >> 4) / 1048576.0 * 100.0;
*temperature = (((((uint32_t)sensor_data[3]) << 16) | (((uint32_t)sensor_data[4]) << 8) | (((uint32_t)sensor_data[5]) << 0)) & 0xFFFFF) / 1048576.0 * 200.0 - 50.0;
ESP_LOGI(TAG, "HTU read success.");
return ESP_OK;
}
esp_err_t zh_htu_reset(void)
{
ESP_LOGI(TAG, "HTU reset begin.");
if (_is_initialized == false)
{
ESP_LOGE(TAG, "HTU reset fail. HTU not initialized.");
return ESP_ERR_NOT_FOUND;
}
esp_err_t esp_err = ESP_OK;
uint8_t command = I2C_RESET_COMMAND;
#ifdef CONFIG_IDF_TARGET_ESP8266
i2c_cmd_handle_t i2c_cmd_handle = i2c_cmd_link_create();
i2c_master_start(i2c_cmd_handle);
i2c_master_write_byte(i2c_cmd_handle, I2C_ADDRESS << 1 | I2C_MASTER_WRITE, true);
i2c_master_write_byte(i2c_cmd_handle, command, true);
i2c_master_stop(i2c_cmd_handle);
esp_err = i2c_master_cmd_begin(_init_config.i2c_port, i2c_cmd_handle, 1000 / portTICK_PERIOD_MS);
i2c_cmd_link_delete(i2c_cmd_handle);
#else
esp_err = i2c_master_transmit(_htu_handle, &command, sizeof(command), 1000 / portTICK_PERIOD_MS);
#endif
if (esp_err != ESP_OK)
{
ESP_LOGE(TAG, "HTU reset fail. I2C driver error at line %d.", __LINE__);
return ESP_ERR_INVALID_RESPONSE;
}
vTaskDelay(RESET_TIME / portTICK_PERIOD_MS);
// Write user register.
ESP_LOGI(TAG, "HTU reset success.");
return ESP_OK;
}
static uint8_t _calc_crc(const uint8_t *buf, size_t len)
{
uint8_t crc = 0xFF;
for (uint8_t byte = 0; byte < len; byte++)
{
crc ^= buf[byte];
for (uint8_t i = 8; i > 0; --i)
{
if ((crc & 0x80) != 0)
{
crc = (crc << 1) ^ 0x31;
}
else
{
crc = crc << 1;
}
}
}
return crc;
}