Added support for casting within visit_SubroutineCall

TutuBinary · Sep 24, 2023 · 37cc6eb · 37cc6eb
1 parent 46df3e5
commit 37cc6eb
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/integration_tests/symbolics_09.py b/integration_tests/symbolics_09.py
@@ -13,7 +13,7 @@ def call_addInteger():
     e: S = cos(b)
     addInteger(c, d, e, 2)
     addInteger(c, sin(a), cos(b), 2)
-    addInteger(c, sin(Symbol("x")), cos(Symbol("y")), 2)
+    addInteger(pi, sin(Symbol("x")), cos(Symbol("y")), 2)
 
 def main0():
     call_addInteger()

diff --git a/src/libasr/pass/replace_symbolic.cpp b/src/libasr/pass/replace_symbolic.cpp
@@ -721,6 +721,17 @@ class ReplaceSymbolicVisitor : public PassUtils::PassVisitor<ReplaceSymbolicVisi
                 ASR::expr_t* target = ASRUtils::EXPR(ASR::make_Var_t(al, x.base.base.loc, arg));
                 process_intrinsic_function(al, x.base.base.loc, intrinsic_func, module_scope, target);
 
+                ASR::call_arg_t call_arg;
+                call_arg.loc = x.base.base.loc;
+                call_arg.m_value = target;
+                call_args.push_back(al, call_arg);
+            } else if (ASR::is_a<ASR::Cast_t>(*val)) {
+                ASR::Cast_t* cast_t = ASR::down_cast<ASR::Cast_t>(val);
+                if(cast_t->m_kind != ASR::cast_kindType::IntegerToSymbolicExpression) return;
+                this->visit_Cast(*cast_t);
+                ASR::symbol_t *var_sym = current_scope->get_symbol(symengine_stack.pop());
+                ASR::expr_t* target = ASRUtils::EXPR(ASR::make_Var_t(al, x.base.base.loc, var_sym));
+
                 ASR::call_arg_t call_arg;
                 call_arg.loc = x.base.base.loc;
                 call_arg.m_value = target;
@@ -793,9 +804,8 @@ class ReplaceSymbolicVisitor : public PassUtils::PassVisitor<ReplaceSymbolicVisi
             } else if (ASR::is_a<ASR::Cast_t>(*val)) {
                 ASR::Cast_t* cast_t = ASR::down_cast<ASR::Cast_t>(val);
                 if(cast_t->m_kind != ASR::cast_kindType::IntegerToSymbolicExpression) return;
-                ASR::symbol_t *var_sym = nullptr;
                 this->visit_Cast(*cast_t);
-                var_sym = current_scope->get_symbol(symengine_stack.pop());
+                ASR::symbol_t *var_sym = current_scope->get_symbol(symengine_stack.pop());
                 ASR::expr_t* target = ASRUtils::EXPR(ASR::make_Var_t(al, x.base.base.loc, var_sym));
 
                 // Now create the FunctionCall node for basic_str