Propagate global context in EvalFunction (#14176)

- Propagate the global context in EvalFunction so that evaluation of
  expressions inside of functions follow the normal context rules (such
  as skipping external functions in the instance API).

Author: perost

OMCompiler/Compiler/NFFrontEnd/NFEvalFunction.mo

@@ -91,13 +91,14 @@ algorithm
     // make sure we don't try to evaluate the non-differentiated function body.
     fail();
   else
-    result := evaluateNormal(fn, args);
+    result := evaluateNormal(fn, args, target.context);
   end if;
 end evaluate;
 
 function evaluateNormal
   input Function fn;
   input list<Expression> args;
+  input InstContext.Type context;
   output Expression result;
 protected
   list<Statement> fn_body;
@@ -106,6 +107,7 @@ protected
   Integer call_count, limit;
   Pointer<Integer> call_counter = fn.callCounter;
   FlowControl ctrl;
+  InstContext.Type body_context;
 algorithm
   // Functions contain a mutable call counter that's increased by one at the
   // start of each evaluation, and decreased by one when the evalution is
@@ -122,6 +124,8 @@ algorithm
 
   Pointer.update(call_counter, call_count);
 
+  body_context := InstContext.clearScopeFlags(context);
+
   try
     fn_body := Function.getBody(fn);
     arg_map := createArgumentMap(fn.inputs, fn.outputs, fn.locals, args, mutableParams = true);
@@ -130,7 +134,7 @@ algorithm
     //       sorted by dependencies first.
     fn_body := applyReplacements(arg_map, fn_body);
     fn_body := optimizeBody(fn_body);
-    ctrl := evaluateStatements(fn_body);
+    ctrl := evaluateStatements(fn_body, body_context);
 
     if ctrl <> FlowControl.ASSERTION then
       result := createResult(arg_map, fn.outputs);
@@ -712,10 +716,11 @@ end assertAssignedOutput;
 
 function evaluateStatements
   input list<Statement> stmts;
+  input InstContext.Type context;
   output FlowControl ctrl = FlowControl.NEXT;
 algorithm
   for s in stmts loop
-    ctrl := evaluateStatement(s);
+    ctrl := evaluateStatement(s, context);
 
     if ctrl <> FlowControl.NEXT then
       if ctrl == FlowControl.CONTINUE then
@@ -729,17 +734,18 @@ end evaluateStatements;
 
 function evaluateStatement
   input Statement stmt;
+  input InstContext.Type context;
   output FlowControl ctrl;
 algorithm
   // adrpo: we really need some error handling here to detect which statement cannot be evaluated
   // try
   ctrl := match stmt
-    case Statement.ASSIGNMENT() then evaluateAssignment(stmt.lhs, stmt.rhs, stmt.source);
-    case Statement.FOR()        then evaluateFor(stmt.iterator, stmt.range, stmt.body, stmt.source);
-    case Statement.IF()         then evaluateIf(stmt.branches, stmt.source);
-    case Statement.ASSERT()     then evaluateAssert(stmt.condition, stmt);
-    case Statement.NORETCALL()  then evaluateNoRetCall(stmt.exp, stmt.source);
-    case Statement.WHILE()      then evaluateWhile(stmt.condition, stmt.body, stmt.source);
+    case Statement.ASSIGNMENT() then evaluateAssignment(stmt.lhs, stmt.rhs, stmt.source, context);
+    case Statement.FOR()        then evaluateFor(stmt.iterator, stmt.range, stmt.body, stmt.source, context);
+    case Statement.IF()         then evaluateIf(stmt.branches, stmt.source, context);
+    case Statement.ASSERT()     then evaluateAssert(stmt.condition, stmt, stmt.source, context);
+    case Statement.NORETCALL()  then evaluateNoRetCall(stmt.exp, stmt.source, context);
+    case Statement.WHILE()      then evaluateWhile(stmt.condition, stmt.body, stmt.source, context);
     case Statement.RETURN()     then FlowControl.RETURN;
     case Statement.BREAK()      then FlowControl.BREAK;
     else
@@ -759,10 +765,11 @@ function evaluateAssignment
   input Expression lhsExp;
   input Expression rhsExp;
   input DAE.ElementSource source;
+  input InstContext.Type context;
   output FlowControl ctrl = FlowControl.NEXT;
 algorithm
   assignVariable(lhsExp,
-    Ceval.evalExp(rhsExp, EvalTarget.new(ElementSource.getInfo(source), STATEMENT_CONTEXT)));
+    Ceval.evalExp(rhsExp, evalTargetFromSource(source, STATEMENT_CONTEXT, context)));
 end evaluateAssignment;
 
 public
@@ -967,6 +974,7 @@ function evaluateFor
   input Option<Expression> range;
   input list<Statement> forBody;
   input DAE.ElementSource source;
+  input InstContext.Type context;
   output FlowControl ctrl = FlowControl.NEXT;
 protected
   RangeIterator range_iter;
@@ -976,7 +984,7 @@ protected
   Integer i = 0, limit = Flags.getConfigInt(Flags.EVAL_LOOP_LIMIT);
 algorithm
   range_exp := Ceval.evalExp(Util.getOption(range),
-    EvalTarget.new(ElementSource.getInfo(source), STATEMENT_CONTEXT));
+    evalTargetFromSource(source, STATEMENT_CONTEXT, context));
   range_iter := RangeIterator.fromExp(range_exp);
 
   if RangeIterator.hasNext(range_iter) then
@@ -987,7 +995,7 @@ algorithm
       (range_iter, value) := RangeIterator.next(range_iter);
       // Update the mutable expression with the iteration value and evaluate the statement.
       Mutable.update(iter_exp, value);
-      ctrl := evaluateStatements(body);
+      ctrl := evaluateStatements(body, context);
 
       if ctrl <> FlowControl.NEXT then
         if ctrl == FlowControl.BREAK then
@@ -1010,6 +1018,7 @@ end evaluateFor;
 function evaluateIf
   input list<tuple<Expression, list<Statement>>> branches;
   input DAE.ElementSource source;
+  input InstContext.Type context;
   output FlowControl ctrl;
 protected
   Expression cond;
@@ -1018,8 +1027,8 @@ algorithm
   for branch in branches loop
     (cond, body) := branch;
 
-    if Expression.isTrue(Ceval.evalExp(cond, EvalTarget.new(ElementSource.getInfo(source), IF_COND_CONTEXT))) then
-      ctrl := evaluateStatements(body);
+    if Expression.isTrue(Ceval.evalExp(cond, evalTargetFromSource(source, IF_COND_CONTEXT, context))) then
+      ctrl := evaluateStatements(body, context);
       return;
     end if;
   end for;
@@ -1030,11 +1039,12 @@ end evaluateIf;
 function evaluateAssert
   input Expression condition;
   input Statement assertStmt;
+  input DAE.ElementSource source;
+  input InstContext.Type context;
   output FlowControl ctrl = FlowControl.NEXT;
 protected
   Expression cond, msg, lvl;
-  SourceInfo info = ElementSource.getInfo(Statement.source(assertStmt));
-  EvalTarget target = EvalTarget.new(info, STATEMENT_CONTEXT);
+  EvalTarget target = evalTargetFromSource(source, STATEMENT_CONTEXT, context);
 algorithm
   if Expression.isFalse(Ceval.evalExp(condition, target)) then
     Statement.ASSERT(message = msg, level = lvl) := assertStmt;
@@ -1044,13 +1054,13 @@ algorithm
     () := match (msg, lvl)
       case (Expression.STRING(), Expression.ENUM_LITERAL(name = "warning"))
         algorithm
-          Error.addSourceMessage(Error.ASSERT_TRIGGERED_WARNING, {msg.value}, info);
+          Error.addSourceMessage(Error.ASSERT_TRIGGERED_WARNING, {msg.value}, EvalTarget.getInfo(target));
         then
           ();
 
       case (Expression.STRING(), Expression.ENUM_LITERAL(name = "error"))
         algorithm
-          Error.addSourceMessage(Error.ASSERT_TRIGGERED_ERROR, {msg.value}, info);
+          Error.addSourceMessage(Error.ASSERT_TRIGGERED_ERROR, {msg.value}, EvalTarget.getInfo(target));
           ctrl := FlowControl.ASSERTION;
         then
           ();
@@ -1068,22 +1078,24 @@ end evaluateAssert;
 function evaluateNoRetCall
   input Expression callExp;
   input DAE.ElementSource source;
+  input InstContext.Type context;
   output FlowControl ctrl = FlowControl.NEXT;
 algorithm
-  Ceval.evalExp(callExp, EvalTarget.new(ElementSource.getInfo(source), STATEMENT_CONTEXT));
+  Ceval.evalExp(callExp, evalTargetFromSource(source, STATEMENT_CONTEXT, context));
 end evaluateNoRetCall;
 
 function evaluateWhile
   input Expression condition;
   input list<Statement> body;
   input DAE.ElementSource source;
+  input InstContext.Type context;
   output FlowControl ctrl = FlowControl.NEXT;
 protected
   Integer i = 0, limit = Flags.getConfigInt(Flags.EVAL_LOOP_LIMIT);
-  EvalTarget target = EvalTarget.new(ElementSource.getInfo(source), STATEMENT_CONTEXT);
+  EvalTarget target = evalTargetFromSource(source, STATEMENT_CONTEXT, context);
 algorithm
   while Expression.isTrue(Ceval.evalExp(condition, target)) loop
-    ctrl := evaluateStatements(body);
+    ctrl := evaluateStatements(body, context);
 
     if ctrl <> FlowControl.NEXT then
       if ctrl == FlowControl.BREAK then
@@ -1102,6 +1114,13 @@ algorithm
   end while;
 end evaluateWhile;
 
+function evalTargetFromSource
+  input DAE.ElementSource source;
+  input InstContext.Type context;
+  input InstContext.Type currentContext;
+  output EvalTarget target = EvalTarget.new(ElementSource.getInfo(source), InstContext.set(context, currentContext));
+end evalTargetFromSource;
+
 function evaluateExternal2
   input String name;
   input Function fn;