From f329a2686e13c16c09da778b3bf2e8d8f3532f68 Mon Sep 17 00:00:00 2001
From: Houtan Bastani <houtan@dynare.org>
Date: Thu, 9 Apr 2015 12:16:52 +0200
Subject: [PATCH] preprocessor: take care of extra exos in
 plot_conditional_forecast

---
 ComputingTasks.cc | 23 +++++++++++++++++++++++
 ComputingTasks.hh |  1 +
 2 files changed, 24 insertions(+)

diff --git a/ComputingTasks.cc b/ComputingTasks.cc
index ebfdb366..82327b03 100644
--- a/ComputingTasks.cc
+++ b/ComputingTasks.cc
@@ -2025,6 +2025,29 @@ PlotConditionalForecastStatement::PlotConditionalForecastStatement(int periods_a
 {
 }
 
+Statement *
+PlotConditionalForecastStatement::cloneAndReindexSymbIds(DataTree &dynamic_datatree, SymbolTable &orig_symbol_table)
+{
+  SymbolList new_symbol_list;
+  try
+    {
+      SymbolTable *new_symbol_table =  dynamic_datatree.getSymbolTable();
+      vector<string> symbols = symbol_list.get_symbols();
+      for (vector<string>::const_iterator it = symbols.begin(); it != symbols.end(); it++)
+        {
+          new_symbol_table->getID(*it);
+          new_symbol_list.addSymbol(*it);
+        }
+    }
+  catch (SymbolTable::UnknownSymbolNameException &e)
+    {
+      cerr << "ERROR: A variable in the plot_conditional_forecast statement was not found in the symbol table" << endl
+           << "       This likely means that you have declared a varexo that is not used in the model" << endl;
+      exit(EXIT_FAILURE);
+    }
+  return new PlotConditionalForecastStatement(periods, new_symbol_list);
+}
+
 void
 PlotConditionalForecastStatement::writeOutput(ostream &output, const string &basename) const
 {
diff --git a/ComputingTasks.hh b/ComputingTasks.hh
index 72c3649f..4c761a08 100644
--- a/ComputingTasks.hh
+++ b/ComputingTasks.hh
@@ -566,6 +566,7 @@ private:
 public:
   PlotConditionalForecastStatement(int periods_arg, const SymbolList &symbol_list_arg);
   virtual void writeOutput(ostream &output, const string &basename) const;
+  virtual Statement *cloneAndReindexSymbIds(DataTree &dynamic_datatree, SymbolTable &orig_symbol_table);
 };
 
 class CalibSmootherStatement : public Statement
-- 
GitLab