From 3947903f3352a6787deeb5e47e7ebe17a9615109 Mon Sep 17 00:00:00 2001
From: Houtan Bastani <houtan@dynare.org>
Date: Fri, 12 Apr 2019 14:16:42 +0200
Subject: [PATCH] add JSON output for shock_groups

---
 src/Shocks.cc | 35 +++++++++++++++++++++++++++++++++++
 src/Shocks.hh |  1 +
 2 files changed, 36 insertions(+)

diff --git a/src/Shocks.cc b/src/Shocks.cc
index a07b3960..3225c687 100644
--- a/src/Shocks.cc
+++ b/src/Shocks.cc
@@ -635,3 +635,38 @@ ShockGroupsStatement::writeOutput(ostream &output, const string &basename, bool
         }
     }
 }
+
+void
+ShockGroupsStatement::writeJsonOutput(ostream &output) const
+{
+  output << R"({"statementName": "shock_groups", "name": ")" << name << R"(", "groups": {)";
+  bool unique_label = true;
+  bool printed_group = false;
+  for (auto it = shock_groups.begin(); it != shock_groups.end(); it++, unique_label = true)
+    {
+      for (auto it1 = it+1; it1 != shock_groups.end(); it1++)
+        if (it->name == it1->name)
+          {
+            unique_label = false;
+            break;
+          }
+
+      if (unique_label)
+        {
+          if (printed_group)
+            output << ", ";
+          else
+            printed_group = true;
+          output << R"("group_name": ")" << it->name << R"(",)"
+                 << R"("shocks": [)";
+          for (auto it1 = it->list.begin(); it1 != it->list.end(); it1++)
+            {
+              if (it1 != it->list.begin())
+                output << ", ";
+              output << R"(")" << *it1 << R"(")";
+            }
+          output << "]";
+        }
+    }
+  output << "}}";
+}
diff --git a/src/Shocks.hh b/src/Shocks.hh
index c1879ad1..ad420e30 100644
--- a/src/Shocks.hh
+++ b/src/Shocks.hh
@@ -163,6 +163,7 @@ private:
 public:
   ShockGroupsStatement(group_t shock_groups_arg, string name_arg);
   void writeOutput(ostream &output, const string &basename, bool minimal_workspace) const override;
+  void writeJsonOutput(ostream &output) const override;
 };
 
 #endif