From 4d01b85e998d8cac035f1eea279c044face7895d Mon Sep 17 00:00:00 2001
From: Joakim Hove <joakim.hove@gmail.com>
Date: Fri, 13 Mar 2020 12:17:41 +0100
Subject: [PATCH] Add operator<< support for TimeMap

---
 .../eclipse/EclipseState/Schedule/TimeMap.hpp  |  4 ++++
 .../eclipse/EclipseState/Schedule/TimeMap.cpp  | 18 ++++++++++++++++++
 2 files changed, 22 insertions(+)

diff --git a/opm/parser/eclipse/EclipseState/Schedule/TimeMap.hpp b/opm/parser/eclipse/EclipseState/Schedule/TimeMap.hpp
index 02d4a5c5f..7301cda50 100644
--- a/opm/parser/eclipse/EclipseState/Schedule/TimeMap.hpp
+++ b/opm/parser/eclipse/EclipseState/Schedule/TimeMap.hpp
@@ -27,6 +27,7 @@
 #include <ctime>
 #include <map>
 #include <utility>
+#include <iostream>
 
 #include <stddef.h>
 
@@ -99,6 +100,9 @@ namespace Opm {
         bool m_skiprest = false;
         std::size_t m_restart_offset = 0;
     };
+
+std::ostream& operator<<(std::ostream& stream, const TimeMap& tm);
+
 }
 
 
diff --git a/src/opm/parser/eclipse/EclipseState/Schedule/TimeMap.cpp b/src/opm/parser/eclipse/EclipseState/Schedule/TimeMap.cpp
index ccf5e375f..7e7b205a6 100644
--- a/src/opm/parser/eclipse/EclipseState/Schedule/TimeMap.cpp
+++ b/src/opm/parser/eclipse/EclipseState/Schedule/TimeMap.cpp
@@ -20,6 +20,7 @@
 #include <cassert>
 #include <ctime>
 #include <stddef.h>
+#include <iomanip>
 
 #include <opm/common/utility/TimeService.hpp>
 
@@ -436,6 +437,23 @@ namespace {
     bool TimeMap::skiprest() const {
         return this->m_skiprest;
     }
+
+std::ostream& operator<<(std::ostream& stream, const TimeMap& tm) {
+    std::stringstream ss;
+    ss << "{";
+    std::size_t index = 0;
+    for (const auto& tp : tm.timeList()) {
+        auto ts = TimeStampUTC(tp);
+        ss << ts.year() << "-" << std::setfill('0') << std::setw(2) << ts.month() << "-" << std::setfill('0') << std::setw(2) << ts.day();
+        index += 1;
+        if (index < tm.timeList().size())
+            ss << ", ";
+        if (index % 12 == 0)
+            ss << std::endl;
+    }
+    ss << "}";
+    return stream << ss.str();
+}
 }