diff --git a/preprocessor/DataTree.cc b/preprocessor/DataTree.cc index 712d4e90b663325a153c2dc9424a2bf20082c2a9..9612fd9035f2f223ddaa2bf871e219dc977730fc 100644 --- a/preprocessor/DataTree.cc +++ b/preprocessor/DataTree.cc @@ -484,11 +484,33 @@ DataTree::getDynJacobianCol(int deriv_id) const throw (UnknownDerivIDException) } bool -DataTree::containsSteadyStateOperator() const +DataTree::isUnaryOpUsed(UnaryOpcode opcode) const { for (unary_op_node_map_type::const_iterator it = unary_op_node_map.begin(); it != unary_op_node_map.end(); it++) - if (it->first.second == oSteadyState) + if (it->first.second == opcode) + return true; + + return false; +} + +bool +DataTree::isBinaryOpUsed(BinaryOpcode opcode) const +{ + for (binary_op_node_map_type::const_iterator it = binary_op_node_map.begin(); + it != binary_op_node_map.end(); it++) + if (it->first.second == opcode) + return true; + + return false; +} + +bool +DataTree::isTrinaryOpUsed(TrinaryOpcode opcode) const +{ + for (trinary_op_node_map_type::const_iterator it = trinary_op_node_map.begin(); + it != trinary_op_node_map.end(); it++) + if (it->first.second == opcode) return true; return false; diff --git a/preprocessor/DataTree.hh b/preprocessor/DataTree.hh index 039aae26da90a9e0185062fd1ee9d466b757de64..fade2b8bbf9a3b9326889e51d5fc15360628b2a1 100644 --- a/preprocessor/DataTree.hh +++ b/preprocessor/DataTree.hh @@ -67,9 +67,6 @@ protected: //! Internal implementation of AddVariable(), without the check on the lag VariableNode *AddVariableInternal(int symb_id, int lag); - //! Is there a steady state operator in the tree? - bool containsSteadyStateOperator() const; - private: typedef list<NodeID> node_list_type; //! The list of nodes @@ -181,6 +178,12 @@ public: NodeID AddUnknownFunction(const string &function_name, const vector<NodeID> &arguments); //! Checks if a given symbol is used somewhere in the data tree bool isSymbolUsed(int symb_id) const; + //! Checks if a given unary op is used somewhere in the data tree + bool isUnaryOpUsed(UnaryOpcode opcode) const; + //! Checks if a given binary op is used somewhere in the data tree + bool isBinaryOpUsed(BinaryOpcode opcode) const; + //! Checks if a given trinary op is used somewhere in the data tree + bool isTrinaryOpUsed(TrinaryOpcode opcode) const; //! Thrown when trying to access an unknown variable by deriv_id class UnknownDerivIDException { diff --git a/preprocessor/DynamicModel.cc b/preprocessor/DynamicModel.cc index 1bfb09d63a578b03c7ff4f90d2c6077e64c1ac2e..f510d0fe9f97d8cb88eee355d279eeb6a353d9ff 100644 --- a/preprocessor/DynamicModel.cc +++ b/preprocessor/DynamicModel.cc @@ -990,7 +990,7 @@ DynamicModel::writeDynamicMFile(const string &dynamic_basename) const << "% Warning : this file is generated automatically by Dynare" << endl << "% from model file (.mod)" << endl << endl; - if (containsSteadyStateOperator()) + if (isUnaryOpUsed(oSteadyState)) mDynamicModelFile << "global oo_;" << endl << endl; writeDynamicModel(mDynamicModelFile, false);