stepan
/
dynare
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

preprocessor: move adl to unaryopnode

time-shift
Houtan Bastani 2017-07-03 16:38:44 +02:00
parent 253f92630d
commit 85eebf231e
6 changed files with 82 additions and 93 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
preprocessor/CodeInterpreter.hh
View File

@ -201,7 +201,8 @@ enum UnaryOpcode
oSteadyStateParam2ndDeriv, // for the 2nd derivative of the STEADY_STATE operator w.r.t. to a parameter
oExpectation,
oErf,
oDiff
oDiff,
oAdl
};
enum BinaryOpcode
@ -220,8 +221,7 @@ enum BinaryOpcode
oLessEqual,
oGreaterEqual,
oEqualEqual,
oDifferent,
oAdl
oDifferent
};
enum TrinaryOpcode

5
preprocessor/DataTree.cc
View File

@ -264,10 +264,9 @@ DataTree::AddDiff(expr_t iArg1)
}
expr_t
DataTree::AddAdl(expr_t iArg1, const string &name, expr_t iArg2)
DataTree::AddAdl(expr_t iArg1, const string &name, int lag)
{
expr_t adlnode = AddBinaryOp(iArg1, oAdl, iArg2, 0, string(name));
return adlnode;
return AddUnaryOp(oAdl, iArg1, 0, 0, 0, string(name), lag);
}
expr_t

25
preprocessor/DataTree.hh
View File

@ -63,10 +63,11 @@ protected:
typedef map<pair<int, int>, VariableNode *> variable_node_map_t;
variable_node_map_t variable_node_map;
//! Pair( Pair(arg1, UnaryOpCode), Pair( Expectation Info Set, Pair(param1_symb_id, param2_symb_id)) ))
typedef map<pair<pair<expr_t, UnaryOpcode>, pair<int, pair<int, int> > >, UnaryOpNode *> unary_op_node_map_t;
typedef map<pair<pair<expr_t, UnaryOpcode>, pair<pair<int, pair<int, int> >, pair<string, int> > >, UnaryOpNode *> unary_op_node_map_t;
unary_op_node_map_t unary_op_node_map;
//! Pair( Pair( Pair(arg1, arg2), order of Power Derivative), opCode)
typedef map<pair<pair<pair<expr_t, expr_t>, pair<int, string> >, BinaryOpcode>, BinaryOpNode *> binary_op_node_map_t;
typedef map<pair<pair<pair<expr_t, expr_t>, int>, BinaryOpcode>, BinaryOpNode *> binary_op_node_map_t;
binary_op_node_map_t binary_op_node_map;
typedef map<pair<pair<pair<expr_t, expr_t>, expr_t>, TrinaryOpcode>, TrinaryOpNode *> trinary_op_node_map_t;
trinary_op_node_map_t trinary_op_node_map;
@ -103,8 +104,8 @@ private:
int node_counter;
inline expr_t AddPossiblyNegativeConstant(double val);
inline expr_t AddUnaryOp(UnaryOpcode op_code, expr_t arg, int arg_exp_info_set = 0, int param1_symb_id = 0, int param2_symb_id = 0);
inline expr_t AddBinaryOp(expr_t arg1, BinaryOpcode op_code, expr_t arg2, int powerDerivOrder = 0, const string &adlparam = "");
inline expr_t AddUnaryOp(UnaryOpcode op_code, expr_t arg, int arg_exp_info_set = 0, int param1_symb_id = 0, int param2_symb_id = 0, const string &adl_param_name = "", int adl_param_lag = -1);
inline expr_t AddBinaryOp(expr_t arg1, BinaryOpcode op_code, expr_t arg2, int powerDerivOrder = 0);
inline expr_t AddTrinaryOp(expr_t arg1, TrinaryOpcode op_code, expr_t arg2, expr_t arg3);
public:
@ -165,7 +166,7 @@ public:
//! Adds "diff(arg)" to model tree
expr_t AddDiff(expr_t iArg1);
//! Adds "adl(arg1, arg2)" to model tree
expr_t AddAdl(expr_t iArg1, const string &name, expr_t iArg2);
expr_t AddAdl(expr_t iArg1, const string &name, int lag);
//! Adds "exp(arg)" to model tree
expr_t AddExp(expr_t iArg1);
//! Adds "log(arg)" to model tree
@ -316,10 +317,10 @@ DataTree::AddPossiblyNegativeConstant(double v)
}
inline expr_t
DataTree::AddUnaryOp(UnaryOpcode op_code, expr_t arg, int arg_exp_info_set, int param1_symb_id, int param2_symb_id)
DataTree::AddUnaryOp(UnaryOpcode op_code, expr_t arg, int arg_exp_info_set, int param1_symb_id, int param2_symb_id, const string &adl_param_name, int adl_param_lag)
{
// If the node already exists in tree, share it
unary_op_node_map_t::iterator it = unary_op_node_map.find(make_pair(make_pair(arg, op_code), make_pair(arg_exp_info_set, make_pair(param1_symb_id, param2_symb_id))));
unary_op_node_map_t::iterator it = unary_op_node_map.find(make_pair(make_pair(arg, op_code), make_pair(make_pair(arg_exp_info_set, make_pair(param1_symb_id, param2_symb_id)), make_pair(adl_param_name, adl_param_lag))));
if (it != unary_op_node_map.end())
return it->second;
@ -338,15 +339,13 @@ DataTree::AddUnaryOp(UnaryOpcode op_code, expr_t arg, int arg_exp_info_set, int
{
}
}
return new UnaryOpNode(*this, op_code, arg, arg_exp_info_set, param1_symb_id, param2_symb_id);
return new UnaryOpNode(*this, op_code, arg, arg_exp_info_set, param1_symb_id, param2_symb_id, adl_param_name, adl_param_lag);
}
inline expr_t
DataTree::AddBinaryOp(expr_t arg1, BinaryOpcode op_code, expr_t arg2, int powerDerivOrder, const string &adlparam)
DataTree::AddBinaryOp(expr_t arg1, BinaryOpcode op_code, expr_t arg2, int powerDerivOrder)
{
binary_op_node_map_t::iterator it = binary_op_node_map.find(make_pair(make_pair(make_pair(arg1, arg2),
make_pair(powerDerivOrder, adlparam)),
op_code));
binary_op_node_map_t::iterator it = binary_op_node_map.find(make_pair(make_pair(make_pair(arg1, arg2), powerDerivOrder), op_code));
if (it != binary_op_node_map.end())
return it->second;
@ -361,7 +360,7 @@ DataTree::AddBinaryOp(expr_t arg1, BinaryOpcode op_code, expr_t arg2, int powerD
catch (ExprNode::EvalException &e)
{
}
return new BinaryOpNode(*this, arg1, op_code, arg2, powerDerivOrder, adlparam);
return new BinaryOpNode(*this, arg1, op_code, arg2, powerDerivOrder);
}
inline expr_t

130
preprocessor/ExprNode.cc
View File

@ -1601,18 +1601,21 @@ VariableNode::getEndosAndMaxLags(map<string, int> &model_endos_and_lags) const
model_endos_and_lags[varname] = lag;
}
UnaryOpNode::UnaryOpNode(DataTree &datatree_arg, UnaryOpcode op_code_arg, const expr_t arg_arg, int expectation_information_set_arg, int param1_symb_id_arg, int param2_symb_id_arg) :
UnaryOpNode::UnaryOpNode(DataTree &datatree_arg, UnaryOpcode op_code_arg, const expr_t arg_arg, int expectation_information_set_arg, int param1_symb_id_arg, int param2_symb_id_arg, const string &adl_param_name_arg, int adl_param_lag_arg) :
ExprNode(datatree_arg),
arg(arg_arg),
expectation_information_set(expectation_information_set_arg),
param1_symb_id(param1_symb_id_arg),
param2_symb_id(param2_symb_id_arg),
op_code(op_code_arg)
op_code(op_code_arg),
adl_param_name(adl_param_name_arg),
adl_param_lag(adl_param_lag_arg)
{
// Add myself to the unary op map
datatree.unary_op_node_map[make_pair(make_pair(arg, op_code),
make_pair(expectation_information_set,
make_pair(param1_symb_id, param2_symb_id)))] = this;
make_pair(make_pair(expectation_information_set,
make_pair(param1_symb_id, param2_symb_id)),
make_pair(adl_param_name, adl_param_lag)))] = this;
}
void
@ -1761,6 +1764,9 @@ UnaryOpNode::composeDerivatives(expr_t darg, int deriv_id)
case oDiff:
cerr << "UnaryOpNode::composeDerivatives: not implemented on oDiff" << endl;
exit(EXIT_FAILURE);
case oAdl:
cerr << "UnaryOpNode::composeDerivatives: not implemented on oAdl" << endl;
exit(EXIT_FAILURE);
}
// Suppress GCC warning
exit(EXIT_FAILURE);
@ -1845,6 +1851,9 @@ UnaryOpNode::cost(int cost, bool is_matlab) const
case oDiff:
cerr << "UnaryOpNode::cost: not implemented on oDiff" << endl;
exit(EXIT_FAILURE);
case oAdl:
cerr << "UnaryOpNode::cost: not implemented on oAdl" << endl;
exit(EXIT_FAILURE);
}
else
// Cost for C files
@ -1890,6 +1899,9 @@ UnaryOpNode::cost(int cost, bool is_matlab) const
case oDiff:
cerr << "UnaryOpNode::cost: not implemented on oDiff" << endl;
exit(EXIT_FAILURE);
case oAdl:
cerr << "UnaryOpNode::cost: not implemented on oAdl" << endl;
exit(EXIT_FAILURE);
}
exit(EXIT_FAILURE);
}
@ -2036,6 +2048,12 @@ UnaryOpNode::writeJsonOutput(ostream &output,
case oDiff:
output << "diff";
break;
case oAdl:
output << "adl";
output << "(";
arg->writeJsonOutput(output, temporary_terms, tef_terms);
output << "," << adl_param_name << "," << adl_param_lag << ")";
return;
case oSteadyState:
output << "(";
arg->writeJsonOutput(output, temporary_terms, tef_terms);
@ -2259,6 +2277,9 @@ UnaryOpNode::writeOutput(ostream &output, ExprNodeOutputType output_type,
case oDiff:
output << "diff";
break;
case oAdl:
output << "adl";
break;
}
bool close_parenthesis = false;
@ -2367,6 +2388,9 @@ UnaryOpNode::eval_opcode(UnaryOpcode op_code, double v) throw (EvalException, Ev
case oDiff:
cerr << "UnaryOpNode::eval_opcode: not implemented on oDiff" << endl;
exit(EXIT_FAILURE);
case oAdl:
cerr << "UnaryOpNode::eval_opcode: not implemented on oAdl" << endl;
exit(EXIT_FAILURE);
}
// Suppress GCC warning
exit(EXIT_FAILURE);
@ -2618,6 +2642,8 @@ UnaryOpNode::buildSimilarUnaryOpNode(expr_t alt_arg, DataTree &alt_datatree) con
return alt_datatree.AddErf(alt_arg);
case oDiff:
return alt_datatree.AddDiff(alt_arg);
case oAdl:
return alt_datatree.AddAdl(alt_arg, adl_param_name, adl_param_lag);
}
// Suppress GCC warning
exit(EXIT_FAILURE);
@ -2676,15 +2702,36 @@ UnaryOpNode::maxLead() const
expr_t
UnaryOpNode::substituteAdlAndDiff() const
{
if (op_code != oDiff)
if (op_code != oDiff && op_code != oAdl)
{
expr_t argsubst = arg->substituteAdlAndDiff();
return buildSimilarUnaryOpNode(argsubst, datatree);
}
expr_t argsubst = arg->substituteAdlAndDiff();
return datatree.AddMinus(argsubst,
argsubst->decreaseLeadsLags(1));
if (op_code == oDiff)
{
expr_t argsubst = arg->substituteAdlAndDiff();
return datatree.AddMinus(argsubst,
argsubst->decreaseLeadsLags(1));
}
expr_t arg1subst = arg->substituteAdlAndDiff();
int i = 1;
ostringstream inttostr;
inttostr << i;
expr_t retval = datatree.AddTimes(datatree.AddVariable(datatree.symbol_table.getID(adl_param_name + "_lag_" + inttostr.str()), 0),
arg1subst->decreaseLeadsLags(i));
i++;
for (; i <= adl_param_lag; i++)
{
inttostr.clear();
inttostr.str("");
inttostr << i;
retval = datatree.AddPlus(retval,
datatree.AddTimes(datatree.AddVariable(datatree.symbol_table.getID(adl_param_name + "_lag_" + inttostr.str()), 0),
arg1subst->decreaseLeadsLags(i)));
}
return retval;
}
expr_t
@ -2887,10 +2934,9 @@ BinaryOpNode::BinaryOpNode(DataTree &datatree_arg, const expr_t arg1_arg,
arg1(arg1_arg),
arg2(arg2_arg),
op_code(op_code_arg),
powerDerivOrder(0),
adlparam("")
powerDerivOrder(0)
{
datatree.binary_op_node_map[make_pair(make_pair(make_pair(arg1, arg2), make_pair(powerDerivOrder, adlparam)), op_code)] = this;
datatree.binary_op_node_map[make_pair(make_pair(make_pair(arg1, arg2), powerDerivOrder), op_code)] = this;
}
BinaryOpNode::BinaryOpNode(DataTree &datatree_arg, const expr_t arg1_arg,
@ -2899,25 +2945,10 @@ BinaryOpNode::BinaryOpNode(DataTree &datatree_arg, const expr_t arg1_arg,
arg1(arg1_arg),
arg2(arg2_arg),
op_code(op_code_arg),
powerDerivOrder(powerDerivOrder_arg),
adlparam("")
powerDerivOrder(powerDerivOrder_arg)
{
assert(powerDerivOrder >= 0);
datatree.binary_op_node_map[make_pair(make_pair(make_pair(arg1, arg2), make_pair(powerDerivOrder, adlparam)), op_code)] = this;
}
BinaryOpNode::BinaryOpNode(DataTree &datatree_arg, const expr_t arg1_arg,
BinaryOpcode op_code_arg, const expr_t arg2_arg,
int powerDerivOrder_arg, string adlparam_arg) :
ExprNode(datatree_arg),
arg1(arg1_arg),
arg2(arg2_arg),
op_code(op_code_arg),
powerDerivOrder(powerDerivOrder_arg),
adlparam(adlparam_arg)
{
assert(powerDerivOrder >= 0);
datatree.binary_op_node_map[make_pair(make_pair(make_pair(arg1, arg2), make_pair(powerDerivOrder, adlparam)), op_code)] = this;
datatree.binary_op_node_map[make_pair(make_pair(make_pair(arg1, arg2), powerDerivOrder), op_code)] = this;
}
void
@ -3051,9 +3082,6 @@ BinaryOpNode::composeDerivatives(expr_t darg1, expr_t darg2)
return datatree.AddPlus(t14, t12);
case oEqual:
return datatree.AddMinus(darg1, darg2);
case oAdl:
cerr << "BinaryOpNode::composeDerivatives not implemented for oAdl";
exit(EXIT_FAILURE);
}
// Suppress GCC warning
exit(EXIT_FAILURE);
@ -3120,9 +3148,6 @@ BinaryOpNode::precedence(ExprNodeOutputType output_type, const temporary_terms_t
case oMin:
case oMax:
return 100;
case oAdl:
cerr << "BinaryOpNode::precedence not implemented for oAdl";
exit(EXIT_FAILURE);
}
// Suppress GCC warning
exit(EXIT_FAILURE);
@ -3159,7 +3184,6 @@ BinaryOpNode::precedenceJson(const temporary_terms_t &temporary_terms) const
return 5;
case oMin:
case oMax:
case oAdl:
return 100;
}
// Suppress GCC warning
@ -3220,9 +3244,6 @@ BinaryOpNode::cost(int cost, bool is_matlab) const
return cost + (MIN_COST_MATLAB/2+1);
case oEqual:
return cost;
case oAdl:
cerr << "BinaryOpNode::cost not implemented for oAdl";
exit(EXIT_FAILURE);
}
else
// Cost for C files
@ -3250,9 +3271,6 @@ BinaryOpNode::cost(int cost, bool is_matlab) const
return cost + (MIN_COST_C/2+1);;
case oEqual:
return cost;
case oAdl:
cerr << "BinaryOpNode::cost not implemented for oAdl";
exit(EXIT_FAILURE);
}
// Suppress GCC warning
exit(EXIT_FAILURE);
@ -3365,9 +3383,6 @@ BinaryOpNode::eval_opcode(double v1, BinaryOpcode op_code, double v2, int derivO
return (v1 != v2);
case oEqual:
throw EvalException();
case oAdl:
cerr << "BinaryOpNode::eval_opcode not implemented for oAdl";
exit(EXIT_FAILURE);
}
// Suppress GCC warning
exit(EXIT_FAILURE);
@ -4116,8 +4131,6 @@ BinaryOpNode::buildSimilarBinaryOpNode(expr_t alt_arg1, expr_t alt_arg2, DataTre
return alt_datatree.AddDifferent(alt_arg1, alt_arg2);
case oPowerDeriv:
return alt_datatree.AddPowerDeriv(alt_arg1, alt_arg2, powerDerivOrder);
case oAdl:
return alt_datatree.AddAdl(alt_arg1, adlparam, alt_arg2);
}
// Suppress GCC warning
exit(EXIT_FAILURE);
@ -4307,30 +4320,9 @@ BinaryOpNode::substituteExpectation(subst_table_t &subst_table, vector<BinaryOpN
expr_t
BinaryOpNode::substituteAdlAndDiff() const
{
if (op_code != oAdl)
{
expr_t arg1subst = arg1->substituteAdlAndDiff();
expr_t arg2subst = arg2->substituteAdlAndDiff();
return buildSimilarBinaryOpNode(arg1subst, arg2subst, datatree);
}
expr_t arg1subst = arg1->substituteAdlAndDiff();
int i = 1;
ostringstream inttostr;
inttostr << i;
int param_symb_id = datatree.symbol_table.getID(adlparam + "_lag_" + inttostr.str());
expr_t retval = datatree.AddTimes(datatree.AddVariable(param_symb_id, 0), arg1subst->decreaseLeadsLags(i));
i++;
for (; i <= (int) arg2->eval(eval_context_t());)
{
inttostr.clear();
inttostr.str("");
inttostr << i++;
retval = datatree.AddPlus(retval,
datatree.AddTimes(datatree.AddVariable(datatree.symbol_table.getID(adlparam + "_lag_" + inttostr.str()), 0),
arg1subst->decreaseLeadsLags(i)));
}
return retval;
expr_t arg2subst = arg2->substituteAdlAndDiff();
return buildSimilarBinaryOpNode(arg1subst, arg2subst, datatree);
}
expr_t

6
preprocessor/ExprNode.hh
View File

@ -633,6 +633,8 @@ private:
//! Only used for oSteadyStateParamDeriv and oSteadyStateParam2ndDeriv
const int param1_symb_id, param2_symb_id;
const UnaryOpcode op_code;
const string adl_param_name;
const int adl_param_lag;
virtual expr_t computeDerivative(int deriv_id);
virtual int cost(int cost, bool is_matlab) const;
virtual int cost(const temporary_terms_t &temporary_terms, bool is_matlab) const;
@ -640,7 +642,7 @@ private:
//! Returns the derivative of this node if darg is the derivative of the argument
expr_t composeDerivatives(expr_t darg, int deriv_id);
public:
UnaryOpNode(DataTree &datatree_arg, UnaryOpcode op_code_arg, const expr_t arg_arg, int expectation_information_set_arg, int param1_symb_id_arg, int param2_symb_id_arg);
UnaryOpNode(DataTree &datatree_arg, UnaryOpcode op_code_arg, const expr_t arg_arg, int expectation_information_set_arg, int param1_symb_id_arg, int param2_symb_id_arg, const string &adl_param_name_arg, int adl_param_lag_arg);
virtual void prepareForDerivation();
virtual void computeTemporaryTerms(map<expr_t, pair<int, NodeTreeReference> > &reference_count,
map<NodeTreeReference, temporary_terms_t> &temp_terms_map,
@ -736,8 +738,6 @@ public:
BinaryOpcode op_code_arg, const expr_t arg2_arg);
BinaryOpNode(DataTree &datatree_arg, const expr_t arg1_arg,
BinaryOpcode op_code_arg, const expr_t arg2_arg, int powerDerivOrder);
BinaryOpNode(DataTree &datatree_arg, const expr_t arg1_arg,
BinaryOpcode op_code_arg, const expr_t arg2_arg, int powerDerivOrder_arg, string adlparam_arg);
virtual void prepareForDerivation();
virtual int precedenceJson(const temporary_terms_t &temporary_terms) const;
virtual int precedence(ExprNodeOutputType output_type, const temporary_terms_t &temporary_terms) const;

3
preprocessor/ParsingDriver.cc
View File

@ -2594,8 +2594,7 @@ ParsingDriver::add_diff(expr_t arg1)
expr_t
ParsingDriver::add_adl(expr_t arg1, string *name, string *lag)
{
expr_t id = data_tree->AddAdl(arg1, *name,
data_tree->AddNonNegativeConstant(*lag));
expr_t id = data_tree->AddAdl(arg1, *name, atoi(lag->c_str()));
// Declare parameters here so that parameters can be initialized after the model block
int i = 0;