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

Julia specific interface for normcdf and normpdf

pac-components
MichelJuillard 2021-12-10 10:17:06 +01:00
parent be316b3dec
commit d047878333
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/ExprNode.cc
View File

@ -5939,6 +5939,17 @@ TrinaryOpNode::writeOutput(ostream &output, ExprNodeOutputType output_type,
arg3->writeOutput(output, output_type, temporary_terms, temporary_terms_idxs, tef_terms);
output << ")/M_SQRT2)))";
}
else if (isJuliaOutput(output_type))
{
// Julia API is normcdf(mu, sigma, x) !
output << "normcdf(";
arg2->writeOutput(output, output_type, temporary_terms, temporary_terms_idxs, tef_terms);
output << ",";
arg3->writeOutput(output, output_type, temporary_terms, temporary_terms_idxs, tef_terms);
output << ",";
arg1->writeOutput(output, output_type, temporary_terms, temporary_terms_idxs, tef_terms);
output << ")";
}
else
{
output << "normcdf(";
@ -5964,6 +5975,17 @@ TrinaryOpNode::writeOutput(ostream &output, ExprNodeOutputType output_type,
arg3->writeOutput(output, output_type, temporary_terms, temporary_terms_idxs, tef_terms);
output << ",2)/2)))";
}
else if (isJuliaOutput(output_type))
{
// Julia API is normpdf(mu, sigma, x) !
output << "normpdf(";
arg2->writeOutput(output, output_type, temporary_terms, temporary_terms_idxs, tef_terms);
output << ",";
arg3->writeOutput(output, output_type, temporary_terms, temporary_terms_idxs, tef_terms);
output << ",";
arg1->writeOutput(output, output_type, temporary_terms, temporary_terms_idxs, tef_terms);
output << ")";
}
else
{
output << "normpdf(";