James Kuszmaul | e485354 | 2023-05-15 20:35:48 -0700 | [diff] [blame] | 1 | #ifndef AOS_UTIL_SIMULATION_LOGGER_H_ |
| 2 | #define AOS_UTIL_SIMULATION_LOGGER_H_ |
Stephan Pleines | b117767 | 2024-05-27 17:48:32 -0700 | [diff] [blame] | 3 | #include <memory> |
| 4 | #include <string> |
James Kuszmaul | e485354 | 2023-05-15 20:35:48 -0700 | [diff] [blame] | 5 | #include <string_view> |
Stephan Pleines | b117767 | 2024-05-27 17:48:32 -0700 | [diff] [blame] | 6 | #include <vector> |
James Kuszmaul | e485354 | 2023-05-15 20:35:48 -0700 | [diff] [blame] | 7 | |
Stephan Pleines | b117767 | 2024-05-27 17:48:32 -0700 | [diff] [blame] | 8 | #include "aos/configuration_generated.h" |
| 9 | #include "aos/events/event_loop.h" |
| 10 | #include "aos/events/logging/log_namer.h" |
James Kuszmaul | e485354 | 2023-05-15 20:35:48 -0700 | [diff] [blame] | 11 | #include "aos/events/logging/log_writer.h" |
| 12 | #include "aos/events/simulated_event_loop.h" |
Stephan Pleines | b117767 | 2024-05-27 17:48:32 -0700 | [diff] [blame] | 13 | |
James Kuszmaul | e485354 | 2023-05-15 20:35:48 -0700 | [diff] [blame] | 14 | namespace aos::util { |
| 15 | |
| 16 | class LoggerState { |
| 17 | public: |
| 18 | LoggerState(aos::SimulatedEventLoopFactory *factory, const aos::Node *node, |
Pallavi Madhukar | 3076d5c | 2023-09-09 10:23:26 -0700 | [diff] [blame] | 19 | std::string_view output_folder, |
| 20 | bool do_skip_timing_report = true); |
James Kuszmaul | e485354 | 2023-05-15 20:35:48 -0700 | [diff] [blame] | 21 | |
| 22 | private: |
| 23 | std::unique_ptr<aos::EventLoop> event_loop_; |
| 24 | std::unique_ptr<aos::logger::LogNamer> namer_; |
| 25 | std::unique_ptr<aos::logger::Logger> logger_; |
| 26 | }; |
| 27 | |
| 28 | // Creates a logger for each of the specified nodes. This makes it so that you |
| 29 | // can easily setup some number of loggers in simulation or log replay without |
| 30 | // needing to redo all the boilerplate every time. |
| 31 | std::vector<std::unique_ptr<LoggerState>> MakeLoggersForNodes( |
| 32 | aos::SimulatedEventLoopFactory *factory, |
| 33 | const std::vector<std::string> &nodes_to_log, |
Pallavi Madhukar | 3076d5c | 2023-09-09 10:23:26 -0700 | [diff] [blame] | 34 | std::string_view output_folder, bool do_skip_timing_report = true); |
James Kuszmaul | e485354 | 2023-05-15 20:35:48 -0700 | [diff] [blame] | 35 | |
| 36 | // Creates loggers for all of the nodes. |
| 37 | std::vector<std::unique_ptr<LoggerState>> MakeLoggersForAllNodes( |
Pallavi Madhukar | 3076d5c | 2023-09-09 10:23:26 -0700 | [diff] [blame] | 38 | aos::SimulatedEventLoopFactory *factory, std::string_view output_folder, |
| 39 | bool do_skip_timing_report = true); |
James Kuszmaul | e485354 | 2023-05-15 20:35:48 -0700 | [diff] [blame] | 40 | |
| 41 | } // namespace aos::util |
| 42 | #endif // AOS_UTIL_SIMULATION_LOGGER_H_ |