From c0f818a9e60c4f550ccd91b095260197f1e0cd45 Mon Sep 17 00:00:00 2001
From: Houtan Bastani <houtan@dynare.org>
Date: Thu, 4 May 2017 16:50:19 +0200
Subject: [PATCH] preprocessor: aesthetic fix: make writing to .m file nicer
---
ComputingTasks.cc | 15 +++++++++++++--
1 file changed, 13 insertions(+), 2 deletions(-)
diff --git a/ComputingTasks.cc b/ComputingTasks.cc
index c51a29ed..7a8fe14d 100644
--- a/ComputingTasks.cc
+++ b/ComputingTasks.cc
@@ -221,8 +221,19 @@ VarModelStatement::createVarModelMFunction(ostream &output, const map<string, se
stringstream ss;
set<int> horizons = var_expectation_functions_to_write.find(name)->second;
for (set<int>::const_iterator it = horizons.begin(); it != horizons.end(); it++)
- ss << *it << " ";
- output << "writeVarExpectationFunction('" << name << "', [" << ss.rdbuf() << "]);" << endl;
+ {
+ if (it != horizons.begin())
+ ss << " ";
+ ss << *it;
+ }
+
+ output << "writeVarExpectationFunction('" << name << "', ";
+ if (horizons.size() > 1)
+ output << "[";
+ output << ss.rdbuf();
+ if (horizons.size() > 1)
+ output << "]";
+ output << ");" << endl;
}
StochSimulStatement::StochSimulStatement(const SymbolList &symbol_list_arg,
--
GitLab