128 lines
5.1 KiB
C++
128 lines
5.1 KiB
C++
/*
|
|
* Copyright © 2010-2015 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 <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <cassert>
|
|
#include <cmath>
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
|
|
#include "ExternalFunctionsTable.hh"
|
|
#include "SymbolTable.hh"
|
|
|
|
void
|
|
ExternalFunctionsTable::addExternalFunction(
|
|
int symb_id, const external_function_options& external_function_options_arg, bool track_nargs)
|
|
{
|
|
assert(symb_id >= 0);
|
|
assert(external_function_options_arg.nargs > 0);
|
|
|
|
// Change options to be saved so the table is consistent
|
|
external_function_options external_function_options_chng = external_function_options_arg;
|
|
if (external_function_options_arg.firstDerivSymbID == IDSetButNoNameProvided)
|
|
external_function_options_chng.firstDerivSymbID = symb_id;
|
|
|
|
if (external_function_options_arg.secondDerivSymbID == IDSetButNoNameProvided)
|
|
external_function_options_chng.secondDerivSymbID = symb_id;
|
|
|
|
if (!track_nargs)
|
|
external_function_options_chng.nargs = IDNotSet;
|
|
|
|
// Ensure 1st & 2nd deriv option consistency
|
|
if (external_function_options_chng.secondDerivSymbID == symb_id
|
|
&& external_function_options_chng.firstDerivSymbID != symb_id)
|
|
{
|
|
cerr << "ERROR: If the second derivative is provided by the top-level function "
|
|
<< "the first derivative must also be provided by the same function." << endl;
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
if ((external_function_options_chng.secondDerivSymbID != symb_id
|
|
&& external_function_options_chng.firstDerivSymbID == symb_id)
|
|
&& external_function_options_chng.secondDerivSymbID != IDNotSet)
|
|
{
|
|
cerr << "ERROR: If the first derivative is provided by the top-level function, the "
|
|
<< "second derivative cannot be provided by any other external function." << endl;
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
if (external_function_options_chng.secondDerivSymbID != IDNotSet
|
|
&& external_function_options_chng.firstDerivSymbID == IDNotSet)
|
|
{
|
|
cerr << "ERROR: If the second derivative is provided, the first derivative must also be "
|
|
"provided."
|
|
<< endl;
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
if (external_function_options_chng.secondDerivSymbID
|
|
== external_function_options_chng.firstDerivSymbID
|
|
&& external_function_options_chng.firstDerivSymbID != symb_id
|
|
&& external_function_options_chng.firstDerivSymbID != IDNotSet)
|
|
{
|
|
cerr << "ERROR: If the Jacobian and Hessian are provided by the same function, that "
|
|
<< "function must be the top-level function." << endl;
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
// Ensure that if we're overwriting something, we mean to do it
|
|
if (exists(symb_id))
|
|
{
|
|
bool ok_to_overwrite = false;
|
|
if (getNargs(symb_id)
|
|
== IDNotSet) // implies that the information stored about this function is not important
|
|
ok_to_overwrite = true;
|
|
|
|
if (!ok_to_overwrite) // prevents multiple non-compatible calls to
|
|
// external_function(name=funcname)
|
|
{ // e.g. e_f(name=a,nargs=1,fd,sd) and e_f(name=a,nargs=2,fd=b,sd=c) should cause an error
|
|
if (external_function_options_chng.nargs != getNargs(symb_id))
|
|
{
|
|
cerr << "ERROR: The number of arguments passed to the external_function() statement "
|
|
"do not "
|
|
<< "match the number of arguments passed to a previous call or declaration of "
|
|
"the top-level function."
|
|
<< endl;
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
if (external_function_options_chng.firstDerivSymbID != getFirstDerivSymbID(symb_id))
|
|
{
|
|
cerr << "ERROR: The first derivative function passed to the external_function() "
|
|
"statement does not "
|
|
<< "match the first derivative function passed to a previous call or "
|
|
"declaration of the top-level function."
|
|
<< endl;
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
if (external_function_options_chng.secondDerivSymbID != getSecondDerivSymbID(symb_id))
|
|
{
|
|
cerr << "ERROR: The second derivative function passed to the external_function() "
|
|
"statement does not "
|
|
<< "match the second derivative function passed to a previous call or "
|
|
"declaration of the top-level function."
|
|
<< endl;
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
}
|
|
|
|
externalFunctionTable[symb_id] = external_function_options_chng;
|
|
}
|