ESPHome  2022.11.3
aqi_calculator.h
Go to the documentation of this file.
1 #pragma once
2 
4 
5 namespace esphome {
6 namespace hm3301 {
7 
9  public:
10  uint16_t get_aqi(uint16_t pm2_5_value, uint16_t pm10_0_value) override {
11  int pm2_5_index = calculate_index_(pm2_5_value, pm2_5_calculation_grid_);
12  int pm10_0_index = calculate_index_(pm10_0_value, pm10_0_calculation_grid_);
13 
14  return (pm2_5_index < pm10_0_index) ? pm10_0_index : pm2_5_index;
15  }
16 
17  protected:
18  static const int AMOUNT_OF_LEVELS = 6;
19 
20  int index_grid_[AMOUNT_OF_LEVELS][2] = {{0, 51}, {51, 100}, {101, 150}, {151, 200}, {201, 300}, {301, 500}};
21 
22  int pm2_5_calculation_grid_[AMOUNT_OF_LEVELS][2] = {{0, 12}, {13, 35}, {36, 55}, {56, 150}, {151, 250}, {251, 500}};
23 
24  int pm10_0_calculation_grid_[AMOUNT_OF_LEVELS][2] = {{0, 54}, {55, 154}, {155, 254},
25  {255, 354}, {355, 424}, {425, 604}};
26 
27  int calculate_index_(uint16_t value, int array[AMOUNT_OF_LEVELS][2]) {
28  int grid_index = get_grid_index_(value, array);
29  int aqi_lo = index_grid_[grid_index][0];
30  int aqi_hi = index_grid_[grid_index][1];
31  int conc_lo = array[grid_index][0];
32  int conc_hi = array[grid_index][1];
33 
34  return (value - conc_lo) * (aqi_hi - aqi_lo) / (conc_hi - conc_lo) + aqi_lo;
35  }
36 
37  int get_grid_index_(uint16_t value, int array[AMOUNT_OF_LEVELS][2]) {
38  for (int i = 0; i < AMOUNT_OF_LEVELS; i++) {
39  if (value >= array[i][0] && value <= array[i][1]) {
40  return i;
41  }
42  }
43  return -1;
44  }
45 };
46 
47 } // namespace hm3301
48 } // namespace esphome
int pm10_0_calculation_grid_[AMOUNT_OF_LEVELS][2]
int pm2_5_calculation_grid_[AMOUNT_OF_LEVELS][2]
int get_grid_index_(uint16_t value, int array[AMOUNT_OF_LEVELS][2])
int calculate_index_(uint16_t value, int array[AMOUNT_OF_LEVELS][2])
Definition: a4988.cpp:4
int index_grid_[AMOUNT_OF_LEVELS][2]
static const int AMOUNT_OF_LEVELS
uint16_t get_aqi(uint16_t pm2_5_value, uint16_t pm10_0_value) override