2019-05-30 20:40:56 +02:00
|
|
|
/*
|
|
|
|
* Copyright © 2019 Dynare Team
|
|
|
|
*
|
|
|
|
* This file is part of Dynare.
|
|
|
|
*
|
|
|
|
* Dynare is free software: you can redistribute it and/or modify
|
|
|
|
* it under the terms of the GNU General Public License as published by
|
|
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* Dynare is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with Dynare. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "Environment.hh"
|
|
|
|
#include "Expressions.hh"
|
|
|
|
|
|
|
|
using namespace macro;
|
|
|
|
|
|
|
|
void
|
|
|
|
Environment::define(VariablePtr var, ExpressionPtr value)
|
|
|
|
{
|
|
|
|
string name = var->getName();
|
|
|
|
if (functions.find(name) != functions.end())
|
|
|
|
throw StackTrace("Variable " + name + " was previously defined as a function");
|
|
|
|
variables[move(name)] = value->eval();
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
|
|
|
Environment::define(FunctionPtr func, ExpressionPtr value)
|
|
|
|
{
|
|
|
|
string name = func->getName();
|
|
|
|
if (variables.find(name) != variables.end())
|
|
|
|
throw StackTrace("Variable " + name + " was previously defined as a variable");
|
|
|
|
functions[name] = {move(func), move(value)};
|
|
|
|
}
|
|
|
|
|
|
|
|
ExpressionPtr
|
|
|
|
Environment::getVariable(const string &name) const
|
|
|
|
{
|
|
|
|
auto it = variables.find(name);
|
|
|
|
if (it != variables.end())
|
|
|
|
return it->second;
|
|
|
|
|
|
|
|
if (!parent)
|
|
|
|
throw StackTrace("Unknown variable " + name);
|
|
|
|
|
|
|
|
return getGlobalEnv()->getVariable(name);
|
|
|
|
}
|
|
|
|
|
|
|
|
tuple<FunctionPtr, ExpressionPtr>
|
|
|
|
Environment::getFunction(const string &name) const
|
|
|
|
{
|
|
|
|
auto it = functions.find(name);
|
|
|
|
if (it != functions.end())
|
|
|
|
return it->second;
|
|
|
|
|
|
|
|
if (!parent)
|
|
|
|
throw StackTrace("Unknown function " + name);
|
|
|
|
|
|
|
|
return parent->getFunction(name);
|
|
|
|
}
|
|
|
|
|
|
|
|
codes::BaseType
|
|
|
|
Environment::getType(const string &name)
|
|
|
|
{
|
|
|
|
return getVariable(name)->eval()->getType();
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
Environment::isVariableDefined(const string &name) const noexcept
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
getVariable(name);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
catch (StackTrace &ex)
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
bool
|
|
|
|
Environment::isFunctionDefined(const string &name) const noexcept
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
getFunction(name);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
catch (StackTrace &ex)
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void
|
2019-06-24 11:00:02 +02:00
|
|
|
Environment::print(ostream &output, int line, bool save) const
|
2019-05-30 20:40:56 +02:00
|
|
|
{
|
2019-06-24 11:00:02 +02:00
|
|
|
if (!save && !variables.empty())
|
|
|
|
output << "Macro Variables:" << endl;
|
|
|
|
|
|
|
|
for (auto & it : variables)
|
2019-05-30 20:40:56 +02:00
|
|
|
{
|
2019-06-24 11:00:02 +02:00
|
|
|
output << (save ? "options_.macrovars_line_" + to_string(line) + "." : " " );
|
|
|
|
output << it.first << " = ";
|
2019-06-24 16:16:22 +02:00
|
|
|
getVariable(it.first)->eval()->print(output, save);
|
2019-06-24 11:00:02 +02:00
|
|
|
if (save)
|
|
|
|
output << ";";
|
|
|
|
output << endl;
|
2019-05-30 20:40:56 +02:00
|
|
|
}
|
|
|
|
|
2019-06-24 11:00:02 +02:00
|
|
|
if (!save && !functions.empty())
|
|
|
|
output << "Macro Functions:" << endl;
|
|
|
|
|
|
|
|
for (auto & it : functions)
|
2019-05-30 20:40:56 +02:00
|
|
|
{
|
2019-06-24 11:00:02 +02:00
|
|
|
output << (save ? "options_.macrovars_line_" + to_string(line) + ".function." : " " );
|
|
|
|
if (save)
|
2019-05-30 20:40:56 +02:00
|
|
|
{
|
2019-06-24 11:00:02 +02:00
|
|
|
get<0>(it.second)->printName(output);
|
|
|
|
output << " = '";
|
2019-05-30 20:40:56 +02:00
|
|
|
}
|
2019-06-24 11:00:02 +02:00
|
|
|
|
|
|
|
get<0>(it.second)->print(output);
|
|
|
|
output << " = ";
|
|
|
|
get<1>(it.second)->print(output);
|
|
|
|
|
|
|
|
if (save)
|
|
|
|
output << "';";
|
|
|
|
output << endl;
|
2019-05-30 20:40:56 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if (parent)
|
2019-06-24 11:00:02 +02:00
|
|
|
parent->print(output, line, save);
|
2019-05-30 20:40:56 +02:00
|
|
|
}
|