def exitNode(self, node, data):
"""Replace a node if it's a unary function."""
ans = super().exitNode(node, data)
if type(ans) is EXPR.UnaryFunctionExpression:
return EXPR.UnaryFunctionExpression(
ans.args, ans.getname(), casadi_intrinsic[ans.getname()])
return ans
def visit(self, node, values):
"""Replace a node if it's a unary function."""
if type(node) is EXPR.UnaryFunctionExpression:
return EXPR.UnaryFunctionExpression(
values[0],
node._name,
casadi_intrinsic[node._name])
return node