1 #pragma once
2 
3 #include "interfaces/sensor.hpp"
4 #include "interfaces/sensor_listener.hpp"
5 #include "interfaces/threshold.hpp"
6 #include "interfaces/trigger_action.hpp"
7 #include "types/duration_types.hpp"
8 #include "types/trigger_types.hpp"
9 #include "utils/threshold_operations.hpp"
10 
11 #include <boost/asio/steady_timer.hpp>
12 
13 #include <chrono>
14 #include <map>
15 #include <memory>
16 #include <vector>
17 
18 class NumericThreshold :
19     public interfaces::Threshold,
20     public interfaces::SensorListener,
21     public std::enable_shared_from_this<NumericThreshold>
22 {
23   public:
24     NumericThreshold(
25         boost::asio::io_context& ioc, Sensors sensors,
26         std::vector<std::unique_ptr<interfaces::TriggerAction>> actions,
27         Milliseconds dwellTime, numeric::Direction direction,
28         double thresholdValue, numeric::Type type);
29     ~NumericThreshold()
30     {}
31 
32     void initialize() override;
33     void sensorUpdated(interfaces::Sensor&, Milliseconds) override;
34     void sensorUpdated(interfaces::Sensor&, Milliseconds, double) override;
35     LabeledThresholdParam getThresholdParam() const override;
36     void updateSensors(Sensors newSensors) override;
37 
38   private:
39     boost::asio::io_context& ioc;
40     const std::vector<std::unique_ptr<interfaces::TriggerAction>> actions;
41     const Milliseconds dwellTime;
42     const numeric::Direction direction;
43     const double thresholdValue;
44     const numeric::Type type;
45     bool initialized = false;
46 
47     struct ThresholdDetail
48     {
49         std::string sensorName;
50         double prevValue;
51         bool dwell;
52         boost::asio::steady_timer timer;
53 
54         ThresholdDetail(const std::string& name, double prevValue, bool dwell,
55                         boost::asio::io_context& ioc) :
56             sensorName(name),
57             prevValue(prevValue), dwell(dwell), timer(ioc)
58         {}
59     };
60     using SensorDetails =
61         std::unordered_map<std::shared_ptr<interfaces::Sensor>,
62                            std::shared_ptr<ThresholdDetail>>;
63     SensorDetails sensorDetails;
64 
65     friend ThresholdOperations;
66 
67     void startTimer(ThresholdDetail&, Milliseconds, double);
68     void commit(const std::string&, Milliseconds, double);
69     ThresholdDetail& getDetails(const interfaces::Sensor& sensor);
70     std::shared_ptr<ThresholdDetail> makeDetails(const std::string& sensorName);
71 };
72