| // Copyright 2021 The Chromium OS Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef ML_SMART_BATTERY_SMART_BATTERY_SERVICE_H_ |
| #define ML_SMART_BATTERY_SMART_BATTERY_SERVICE_H_ |
| |
| #include <memory> |
| #include <string> |
| #include <vector> |
| |
| #include "dbus_adaptors/org.chromium.MachineLearningSmartBattery.h" |
| #include "ml/smart_battery/tf_model_graph_executor.h" |
| |
| namespace ml { |
| |
| // Implementation of the smart battery dbus interface. |
| class SmartBatteryService |
| : public org::chromium::MachineLearningSmartBatteryAdaptor, |
| public org::chromium::MachineLearningSmartBatteryInterface { |
| public: |
| explicit SmartBatteryService( |
| std::unique_ptr<brillo::dbus_utils::DBusObject> dbus_object); |
| SmartBatteryService(const SmartBatteryService&) = delete; |
| SmartBatteryService& operator=(const SmartBatteryService&) = delete; |
| ~SmartBatteryService(); |
| |
| // Register DBus object and interfaces. |
| void RegisterAsync( |
| const brillo::dbus_utils::AsyncEventSequencer::CompletionAction& |
| completion_callback); |
| |
| // org::chromium::MachineLearningSmartBattery: (see |
| // dbus_bindings/org.chromium.MachineLearningSmartBattery.xml). |
| void RequestSmartBatteryDecision( |
| std::unique_ptr< |
| brillo::dbus_utils::DBusMethodResponse<bool, std::vector<double>>> |
| response, |
| const std::string& serialized_example_proto) override; |
| |
| private: |
| const std::unique_ptr<brillo::dbus_utils::DBusObject> dbus_object_; |
| std::unique_ptr<TfModelGraphExecutor> tf_model_graph_executor_; |
| }; |
| |
| } // namespace ml |
| |
| #endif // ML_SMART_BATTERY_SMART_BATTERY_SERVICE_H_ |