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