add MathloggerDriver

AntaresSimulatorTeam · Nov 20, 2023 · d3527c4 · d3527c4
1 parent 4cfd573
commit d3527c4
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 1 deletion.
diff --git a/src/cpp/benders/logger/MathLogger.cpp b/src/cpp/benders/logger/MathLogger.cpp
@@ -1,7 +1,6 @@
 #include "logger/MathLogger.h"
 
 #include <iomanip>
-#include <iostream>
 #include <sstream>
 
 #include "LoggerUtils.h"
@@ -36,4 +35,16 @@ MathLoggerFile::MathLoggerFile(const std::filesystem::path &filename)
     std::cerr << PrefixMessage(LogUtils::LOGLEVEL::ERR, data.CONTEXT)
               << "Invalid file name passed as parameter" << std::endl;
   }
+}
+
+void MathLoggerDriver::write_header() {
+  for (auto logger : math_loggers_) {
+    logger->write_header();
+  }
+}
+
+void MathLoggerDriver::add_logger(MathLogger* logger) {
+  if (logger) {
+    math_loggers_.push_back(logger);
+  }
 }
diff --git a/src/cpp/benders/logger/include/logger/MathLogger.h b/src/cpp/benders/logger/include/logger/MathLogger.h
@@ -4,6 +4,8 @@
 #include <filesystem>
 #include <fstream>
 #include <iomanip>
+#include <iostream>
+#include <list>
 
 struct MathLoggerData {
   int iteration;
@@ -62,4 +64,19 @@ class MathLoggerFile : public MathLogger {
 
  private:
   std::ofstream file_stream_;
+};
+
+class MathLoggerOstream : public MathLogger {
+ public:
+  explicit MathLoggerOstream() : MathLogger(&std::cout) {}
+};
+
+class MathLoggerDriver {
+ public:
+  MathLoggerDriver() = default;
+  void write_header();
+  void MathLoggerDriver::add_logger(MathLogger* logger);
+
+ private:
+  std::list<MathLogger*> math_loggers_;
 };