sh,tSSKJr \"S5r\(aSSKJrJrJrJrJrJ r J r SSK J r J r JrJr SSKJr \R$r\R$R$r\R(R*r\R,R.R.r"SS\R05rO "SS5rS rS rg ) ) import_modulelfortran)Variable IntBaseType FloatBaseTypeStringReturnFunctionDefinition Assignment)AddMulIntegerFloat)SymbolcN\rSrSrSrSrSrSrSrSr Sr S r S r S r S rg ) ASR2PyVisitor:z Visitor Class for LFortran ASR It is a Visitor class derived from asr.ASRVisitor which visits all the nodes of the LFortran ASR and creates corresponding AST node for each ASR node c/Ulg)zInitialize the ParserN_py_astselfs x/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/sympy/parsing/fortran/fortran_parser.py__init__ASR2PyVisitor.__init__Ds DLcURRH-nURRUnURU5 M/ URHnURU5 M g)zT Function to visit all the elements of the Translation Unit created by LFortran ASR N) global_scopesymbolsvisititems)rnodessymitems rvisit_TranslationUnit#ASR2PyVisitor.visit_TranslationUnitHsW &&..''//2 3/  4 #rc.[UR[R5(aURnURn[U[R5(a3[ [UR 5[UR 55nOg[U5[R:Xa4[U5nUH"n[ [UR 5U5nM$ O[S5e[S5eURRW5 g)aVisitor Function for Assignment Visits each Assignment is the LFortran ASR and creates corresponding assignment for SymPy. Notes ===== The function currently only supports variable assignment and binary operation assignments of varying multitudes. Any type of numberS or array is not supported. Raises ====== NotImplementedError() when called for Numeric assignments or Arrays z!Numeric assignments not supportedzArrays not supportedN) isinstancetargetasrrvaluer nametypeBinOp call_visitorNotImplementedErrorrappend)rr"r*r,new_nodeexp_astexprs rvisit_AssignmentASR2PyVisitor.visit_AssignmentSs($++s||44 eS\\22)$$*KK"%$)JJ"  H5kSYY.*51G '#- ( 5 $$!( ..QRR)*@AA LL   )rcURnURnURn[U5[R :Xa[ UR5nO?[U5[R:Xa[U5nUHnUnM O [S5e[U5[R :Xa[ UR5nO?[U5[R:Xa[U5n U HnUnM O [S5e[U[R5(a [WW5n O[U[R5(a[WW*5n OZ[U[R5(a[WSW- 5n O+[U[R5(a [WW5n UR R#W 5 g)auVisitor Function for Binary Operations Visits each binary operation present in the LFortran ASR like addition, subtraction, multiplication, division and creates the corresponding operation node in SymPy's AST In case of more than one binary operations, the function calls the call_visitor() function on the child nodes of the binary operations recursively until all the operations have been processed. Notes ===== The function currently only supports binary operations with Variables or other binary operations. Numerics are not supported as of yet. Raises ====== NotImplementedError() when called for Numeric assignments zNumbers Currently not supportedN)opleftrightr.r+rrr-r/r0r1r)r SubDivr rr2) rr"r:lhsrhs left_value l_exp_astexp right_value r_exp_astr3s r visit_BinOpASR2PyVisitor.visit_BinOpsU0B))C**CS S\\)#CHH- ccii'(- $C!$J%**KLLS S\\)$SXX. ccii'(- $C"%K%**KLL"cgg&&z;7B((zK<8B((z1[=9B((z;7 LL   )rc[UR[R5(a [ [ S55n[S5nOT[UR[R 5(a [[ S55n[S5nO [S5eURS:Xd?[UR5RUUS9nURRU5 gg) alVisitor Function for Variable Declaration Visits each variable declaration present in the ASR and creates a Symbol declaration for each variable Notes ===== The functions currently only support declaration of integer and real variables. Other data types are still under development. Raises ====== NotImplementedError() when called for unsupported data types integerrrealgData type not supportedin)r.r,N)r)r.r+rrrRealrrr1intentrr-as_Declarationrr2)rr"var_typer,r3s rvisit_VariableASR2PyVisitor.visit_Variables$$))S[[11&vi'89 DIIsxx00(8c )*CDDKK4'#II .#!!  ##H-(rcfUb.UH'nURR[U55 M) gg)zVisitor Function for code sequence Visits a code sequence/ block and calls the visitor function on all the children of the code block to create corresponding code in python N)rr2r0)rseqr"s rvisit_SequenceASR2PyVisitor.visit_Sequences/DLL'' T(:; rcg)zVisitor Function for Numbers in ASR This function is currently under development and will be updated with improvements in the LFortran ASR N)rr"s r visit_NumASR2PyVisitor.visit_Nums rcURVs/sHn[UR5PM nn/nURnURHn[ U5nM WnUR RH?n [ UR RU 5n U Hn URU 5 M MA UHn URU 5 M UR[[URR555 [URR[R5(a[[!S55n OS[URR[R"5(a[%[!S55n O ['S5e[)U UUUS9nUR*RU5 gs snf![ a /nGNf=f)aVisitor Function for function Definitions Visits each function definition present in the ASR and creates a function definition node in the Python AST with all the elements of the given function The functions declare all the variables required as SymPy symbols in the function before the function definition This function also the call_visior_function to parse the contents of the function body rIrJrK) return_typer- parametersbodyN)argsrr-r^r0UnboundLocalErrorsymtabrr2r return_varr)r.r+rrrrMrr1r r)rr"arg_iterfn_argsfn_bodyfn_nameifn_ast fn_body_exprr$declrelemret_typer3s rvisit_FunctionASR2PyVisitor.visit_Functions@DyyIy8x .yGIGiiGYY%a "% {{**#DKK$7$7$<=#GNN7+ $+%t$% NN,, $//.. <<&vi'89DOO00#((;;(8)*CDD)&.&%,& H LL   )EJ% "!  "sG G G! G!cUR$)zReturns the AST nodesrrs rret_astASR2PyVisitor.ret_ast&s << rrN)__name__ __module__ __qualname____firstlineno____doc__rr&r6rFrQrUrYrmrp__static_attributes__rXrrrr:s8   !+ *Z7 *r" .H < 1 *f rrc\rSrSrSrSrg)ri*c[S5e)Nzlfortran not available) ImportError)rr_kwargss rrr+s67 7rrXN)rrrsrtrurrwrXrrrr*s 8rc\[5nURU5 UR5nU$)auCalls the AST Visitor on the Module This function is used to call the AST visitor for a program or module It imports all the required modules and calls the visit() function on the given node Parameters ========== fort_node : LFortran ASR object Node for the operation for which the NodeVisitor is called Returns ======= res_ast : list list of SymPy AST Nodes )rr rp) fort_nodevres_asts rr0r0.s'( AGGIiikG NrcF[USS9n[U5n[U5nU$)aWrapper function to convert the given Fortran source code to SymPy Expressions Parameters ========== src : string A string with the Fortran source code Returns ======= py_src : string A string with the Python source code compatible with SymPy F)translation_unit) src_to_ast ast_to_asrr0)srca_astapy_srcs r src_to_sympyrHs( sU 3E5A !_F MrN)sympy.externalrrsympy.codegen.astrrrrr r r sympy.corer r rrsympy.core.symbolrr+asr_modastrsemanticr ASTVisitorrr0rrXrrrs(  $ KKK33(llG ,,  C((J""--88J'Tn n `884r