diff --git a/src/DynamicModel.cc b/src/DynamicModel.cc
index 73d34bacfbdf85a8ab5013fb60c5a04460976261..5ee7bccfa0a8913c35e684cbb9f4a080db4119ae 100644
--- a/src/DynamicModel.cc
+++ b/src/DynamicModel.cc
@@ -3355,23 +3355,19 @@ DynamicModel::checkVarMinLag(vector<int> &eqnumber) const
int
DynamicModel::getVarMaxLag(StaticModel &static_model, vector<int> &eqnumber) const
{
- vector<expr_t> lhs;
+ set<expr_t> lhs;
for (vector<int>::const_iterator it = eqnumber.begin();
it != eqnumber.end(); it++)
+ equations[*it]->get_arg1()->collectVARLHSVariable(lhs);
+
+ if (eqnumber.size() != lhs.size())
{
- set<expr_t> lhs_set;
- equations[*it]->get_arg1()->collectVARLHSVariable(lhs_set);
- if (lhs_set.size() != 1)
- {
- cerr << "ERROR: in Equation "
- << ". A VAR may only have one endogenous variable on the LHS. " << endl;
- exit(EXIT_FAILURE);
- }
- lhs.push_back(*(lhs_set.begin()));
+ cerr << "The LHS variables of the VAR are not unique" << endl;
+ exit(EXIT_FAILURE);
}
set<expr_t> lhs_static;
- for(vector<expr_t>::const_iterator it = lhs.begin();
+ for(set<expr_t>::const_iterator it = lhs.begin();
it != lhs.end(); it++)
lhs_static.insert((*it)->toStatic(static_model));
@@ -3390,7 +3386,7 @@ DynamicModel::getVarLhsDiffAndInfo(vector<int> &eqnumber, vector<bool> &diff,
for (vector<int>::const_iterator it = eqnumber.begin();
it != eqnumber.end(); it++)
{
- diff.push_back(equations[*it]->get_arg1()->isDiffPresent());
+ equations[*it]->get_arg1()->countDiffs() > 0 ? diff.push_back(true) : diff.push_back(false);
if (diff.back())
{
set<pair<int, int> > diff_set;
diff --git a/src/ExprNode.cc b/src/ExprNode.cc
index c38154e00973d00f16ec6d01b7f70399ed6fc45f..a0a62ca41b88651fd5c4854f28d0caae244ba68f 100644
--- a/src/ExprNode.cc
+++ b/src/ExprNode.cc
@@ -294,12 +294,6 @@ ExprNode::isVariableNodeEqualTo(SymbolType type_arg, int variable_id, int lag_ar
return false;
}
-bool
-ExprNode::isDiffPresent() const
-{
- return false;
-}
-
void
ExprNode::getEndosAndMaxLags(map<string, int> &model_endos_and_lags) const
{
@@ -313,10 +307,10 @@ NumConstNode::NumConstNode(DataTree &datatree_arg, int id_arg) :
datatree.num_const_node_map[id] = this;
}
-bool
-NumConstNode::isDiffPresent() const
+int
+NumConstNode::countDiffs() const
{
- return false;
+ return 0;
}
void
@@ -389,6 +383,8 @@ NumConstNode::compile(ostream &CompileCode, unsigned int &instruction_number,
void
NumConstNode::collectVARLHSVariable(set<expr_t> &result) const
{
+ cerr << "ERROR: you can only have variables or unary ops on LHS of VAR" << endl;
+ exit(EXIT_FAILURE);
}
void
@@ -1144,7 +1140,7 @@ VariableNode::collectVARLHSVariable(set<expr_t> &result) const
result.insert(const_cast<VariableNode *>(this));
else
{
- cerr << "ERROR: A VAR must have one endogenous variable on the LHS." << endl;
+ cerr << "ERROR: you can only have endogenous variables or unary ops on LHS of VAR" << endl;
exit(EXIT_FAILURE);
}
}
@@ -1730,10 +1726,10 @@ VariableNode::detrend(int symb_id, bool log_trend, expr_t trend) const
}
}
-bool
-VariableNode::isDiffPresent() const
+int
+VariableNode::countDiffs() const
{
- return false;
+ return 0;
}
expr_t
@@ -2965,19 +2961,20 @@ UnaryOpNode::VarMaxLag(DataTree &static_datatree, set<expr_t> &static_lhs, int &
arg->VarMaxLag(static_datatree, static_lhs, max_lag);
else
{
- for (set<expr_t>::const_iterator it = static_lhs.begin();
- it != static_lhs.end(); it++)
- if (*it == this->toStatic(static_datatree))
- {
- int max_lag_tmp = arg->maxLag();
- if (max_lag_tmp > max_lag)
- max_lag = max_lag_tmp;
- return;
- }
- int max_lag_tmp = 0;
- arg->VarMaxLag(static_datatree, static_lhs, max_lag_tmp);
- if (max_lag_tmp + 1 > max_lag)
- max_lag = max_lag_tmp + 1;
+ set<expr_t>::const_iterator it = static_lhs.find(this->toStatic(static_datatree));
+ if (it != static_lhs.end())
+ {
+ int max_lag_tmp = arg->maxLag() - arg->countDiffs();
+ if (max_lag_tmp > max_lag)
+ max_lag = max_lag_tmp;
+ }
+ else
+ {
+ int max_lag_tmp = 0;
+ arg->VarMaxLag(static_datatree, static_lhs, max_lag_tmp);
+ if (max_lag_tmp + 1 > max_lag)
+ max_lag = max_lag_tmp + 1;
+ }
}
}
@@ -3027,12 +3024,12 @@ UnaryOpNode::substituteAdl() const
return retval;
}
-bool
-UnaryOpNode::isDiffPresent() const
+int
+UnaryOpNode::countDiffs() const
{
if (op_code == oDiff)
- return true;
- return arg->isDiffPresent();
+ return arg->countDiffs() + 1;
+ return arg->countDiffs();
}
bool
@@ -4414,8 +4411,8 @@ BinaryOpNode::VarMaxLag(DataTree &static_datatree, set<expr_t> &static_lhs, int
void
BinaryOpNode::collectVARLHSVariable(set<expr_t> &result) const
{
- arg1->collectVARLHSVariable(result);
- arg2->collectVARLHSVariable(result);
+ cerr << "ERROR: you can only have variables or unary ops on LHS of VAR" << endl;
+ exit(EXIT_FAILURE);
}
void
@@ -5011,10 +5008,10 @@ BinaryOpNode::substituteUnaryOpNodes(DataTree &static_datatree, diff_table_t &no
return buildSimilarBinaryOpNode(arg1subst, arg2subst, datatree);
}
-bool
-BinaryOpNode::isDiffPresent() const
+int
+BinaryOpNode::countDiffs() const
{
- return arg1->isDiffPresent() || arg2->isDiffPresent();
+ return arg1->countDiffs() + arg2->countDiffs();
}
expr_t
@@ -5657,9 +5654,8 @@ TrinaryOpNode::compileExternalFunctionOutput(ostream &CompileCode, unsigned int
void
TrinaryOpNode::collectVARLHSVariable(set<expr_t> &result) const
{
- arg1->collectVARLHSVariable(result);
- arg2->collectVARLHSVariable(result);
- arg3->collectVARLHSVariable(result);
+ cerr << "ERROR: you can only have variables or unary ops on LHS of VAR" << endl;
+ exit(EXIT_FAILURE);
}
void
@@ -5923,10 +5919,10 @@ TrinaryOpNode::substituteUnaryOpNodes(DataTree &static_datatree, diff_table_t &n
return buildSimilarTrinaryOpNode(arg1subst, arg2subst, arg3subst, datatree);
}
-bool
-TrinaryOpNode::isDiffPresent() const
+int
+TrinaryOpNode::countDiffs() const
{
- return arg1->isDiffPresent() || arg2->isDiffPresent() || arg3->isDiffPresent();
+ return arg1->countDiffs() + arg2->countDiffs() + arg3->countDiffs();
}
expr_t
@@ -6127,9 +6123,8 @@ AbstractExternalFunctionNode::compileExternalFunctionArguments(ostream &CompileC
void
AbstractExternalFunctionNode::collectVARLHSVariable(set<expr_t> &result) const
{
- for (vector<expr_t>::const_iterator it = arguments.begin();
- it != arguments.end(); it++)
- (*it)->collectVARLHSVariable(result);
+ cerr << "ERROR: you can only have variables or unary ops on LHS of VAR" << endl;
+ exit(EXIT_FAILURE);
}
void
@@ -6362,13 +6357,13 @@ AbstractExternalFunctionNode::substituteUnaryOpNodes(DataTree &static_datatree,
return buildSimilarExternalFunctionNode(arguments_subst, datatree);
}
-bool
-AbstractExternalFunctionNode::isDiffPresent() const
+int
+AbstractExternalFunctionNode::countDiffs() const
{
- bool result = false;
+ int ndiffs = 0;
for (vector<expr_t>::const_iterator it = arguments.begin(); it != arguments.end(); it++)
- result = result || (*it)->isDiffPresent();
- return result;
+ ndiffs += (*it)->countDiffs();
+ return ndiffs;
}
expr_t
@@ -7829,10 +7824,10 @@ VarExpectationNode::eval(const eval_context_t &eval_context) const throw (EvalEx
return it->second;
}
-bool
-VarExpectationNode::isDiffPresent() const
+int
+VarExpectationNode::countDiffs() const
{
- return false;
+ return 0;
}
void
@@ -7843,6 +7838,8 @@ VarExpectationNode::computeXrefs(EquationInfo &ei) const
void
VarExpectationNode::collectVARLHSVariable(set<expr_t> &result) const
{
+ cerr << "ERROR: you can only have variables or unary ops on LHS of VAR" << endl;
+ exit(EXIT_FAILURE);
}
void
@@ -8281,6 +8278,8 @@ PacExpectationNode::computeXrefs(EquationInfo &ei) const
void
PacExpectationNode::collectVARLHSVariable(set<expr_t> &result) const
{
+ cerr << "ERROR: you can only have variables or unary ops on LHS of VAR" << endl;
+ exit(EXIT_FAILURE);
}
void
@@ -8306,10 +8305,10 @@ PacExpectationNode::compile(ostream &CompileCode, unsigned int &instruction_numb
exit(EXIT_FAILURE);
}
-bool
-PacExpectationNode::isDiffPresent() const
+int
+PacExpectationNode::countDiffs() const
{
- return false;
+ return 0;
}
pair<int, expr_t >
diff --git a/src/ExprNode.hh b/src/ExprNode.hh
index 221b85aa055f5d9e44b1e5419af0d78b7d9f4f42..428ecc6cf64aee6b2330df6c65a2188107aa7eef 100644
--- a/src/ExprNode.hh
+++ b/src/ExprNode.hh
@@ -465,8 +465,8 @@ class ExprNode
//! Returns true if the expression contains one or several exogenous variable
virtual bool containsExogenous() const = 0;
- //! Returns true if the expression contains a diff operator
- virtual bool isDiffPresent(void) const = 0;
+ //! Returns the number of diffs present
+ virtual int countDiffs() const = 0;
//! Return true if the nodeID is a variable withe a type equal to type_arg, a specific variable id aqual to varfiable_id and a lag equal to lag_arg and false otherwise
/*!
@@ -595,7 +595,7 @@ public:
virtual bool isNumConstNodeEqualTo(double value) const;
virtual bool containsEndogenous(void) const;
virtual bool containsExogenous() const;
- virtual bool isDiffPresent(void) const;
+ virtual int countDiffs() const;
virtual bool isVariableNodeEqualTo(SymbolType type_arg, int variable_id, int lag_arg) const;
virtual expr_t replaceTrendVar() const;
virtual expr_t detrend(int symb_id, bool log_trend, expr_t trend) const;
@@ -685,7 +685,7 @@ public:
virtual bool isNumConstNodeEqualTo(double value) const;
virtual bool containsEndogenous(void) const;
virtual bool containsExogenous() const;
- virtual bool isDiffPresent(void) const;
+ virtual int countDiffs() const;
virtual bool isVariableNodeEqualTo(SymbolType type_arg, int variable_id, int lag_arg) const;
virtual expr_t replaceTrendVar() const;
virtual expr_t detrend(int symb_id, bool log_trend, expr_t trend) const;
@@ -799,7 +799,7 @@ public:
virtual bool isNumConstNodeEqualTo(double value) const;
virtual bool containsEndogenous(void) const;
virtual bool containsExogenous() const;
- virtual bool isDiffPresent(void) const;
+ virtual int countDiffs() const;
virtual bool isVariableNodeEqualTo(SymbolType type_arg, int variable_id, int lag_arg) const;
virtual expr_t replaceTrendVar() const;
virtual expr_t detrend(int symb_id, bool log_trend, expr_t trend) const;
@@ -928,7 +928,7 @@ public:
virtual bool isNumConstNodeEqualTo(double value) const;
virtual bool containsEndogenous(void) const;
virtual bool containsExogenous() const;
- virtual bool isDiffPresent(void) const;
+ virtual int countDiffs() const;
virtual bool isVariableNodeEqualTo(SymbolType type_arg, int variable_id, int lag_arg) const;
virtual expr_t replaceTrendVar() const;
virtual expr_t detrend(int symb_id, bool log_trend, expr_t trend) const;
@@ -1033,7 +1033,7 @@ public:
virtual bool isNumConstNodeEqualTo(double value) const;
virtual bool containsEndogenous(void) const;
virtual bool containsExogenous() const;
- virtual bool isDiffPresent(void) const;
+ virtual int countDiffs() const;
virtual bool isVariableNodeEqualTo(SymbolType type_arg, int variable_id, int lag_arg) const;
virtual expr_t replaceTrendVar() const;
virtual expr_t detrend(int symb_id, bool log_trend, expr_t trend) const;
@@ -1139,7 +1139,7 @@ public:
virtual bool isNumConstNodeEqualTo(double value) const;
virtual bool containsEndogenous(void) const;
virtual bool containsExogenous() const;
- virtual bool isDiffPresent(void) const;
+ virtual int countDiffs() const;
virtual bool isVariableNodeEqualTo(SymbolType type_arg, int variable_id, int lag_arg) const;
virtual void writePrhs(ostream &output, ExprNodeOutputType output_type, const temporary_terms_t &temporary_terms, deriv_node_temp_terms_t &tef_terms, const string &ending) const;
virtual expr_t replaceTrendVar() const;
@@ -1338,7 +1338,7 @@ public:
virtual void collectDynamicVariables(SymbolType type_arg, set<pair<int, int> > &result) const;
virtual bool containsEndogenous(void) const;
virtual bool containsExogenous() const;
- virtual bool isDiffPresent(void) const;
+ virtual int countDiffs() const;
virtual bool isNumConstNodeEqualTo(double value) const;
virtual expr_t differentiateForwardVars(const vector<string> &subset, subst_table_t &subst_table, vector<BinaryOpNode *> &neweqs) const;
virtual expr_t decreaseLeadsLagsPredeterminedVariables() const;
@@ -1423,7 +1423,7 @@ public:
virtual void collectDynamicVariables(SymbolType type_arg, set<pair<int, int> > &result) const;
virtual bool containsEndogenous(void) const;
virtual bool containsExogenous() const;
- virtual bool isDiffPresent(void) const;
+ virtual int countDiffs() const;
virtual bool isNumConstNodeEqualTo(double value) const;
virtual expr_t differentiateForwardVars(const vector<string> &subset, subst_table_t &subst_table, vector<BinaryOpNode *> &neweqs) const;
virtual expr_t decreaseLeadsLagsPredeterminedVariables() const;