sg,jddlmZedZerddlmZmZmZmZmZm Z m Z ddl m Z m Z mZmZddlmZej$Zej$j$Zej(j*Zej,j.j.Z Gddej0Zn GddZd Zd Zy ) ) import_modulelfortran)Variable IntBaseType FloatBaseTypeStringReturnFunctionDefinition Assignment)AddMulIntegerFloat)SymbolcFeZdZdZdZdZdZdZdZdZ dZ d Z d Z y ) ASR2PyVisitorz 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 cg|_y)zInitialize the ParserN_py_astselfs W/var/www/html/venv/lib/python3.12/site-packages/sympy/parsing/fortran/fortran_parser.py__init__zASR2PyVisitor.__init__Ds DLc|jjD],}|jj|}|j|.|jD]}|j|y)zx Function to visit all the elements of the Translation Unit created by LFortran ASR N) global_scopesymbolsvisititems)rnodessymitems rvisit_TranslationUnitz#ASR2PyVisitor.visit_TranslationUnitHsa &&.. ''//2 3   ! 4  !rct|jtjr|j}|j}t|tjr3t t|j t|j }ndt|tjk(r2t|}|D]!}t t|j |}#ntdtd|jjy)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_AssignmentzASR2PyVisitor.visit_AssignmentSs($++s||4 eS\\2)$$*KK"%$)JJ"  H5kSYY.*51G '#- ( 5 $$ ..QRR)*@AA LL   )rcx|j}|j}|j}t|tj k(rt |j}n|D]} |j| |jtt|jjt|jjtjrtt!d} nNt|jjtj"rt%t!d} n t'dt)| |||}|j*j|ycc}w#t $rg}YswxYw)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 rDrErF) return_typer* parametersbodyN)argsrr*rVr-UnboundLocalErrorsymtabrr/r return_varr&r+r(rrrrHrr.r r)rr arg_iterfn_argsfn_bodyfn_nameifn_ast fn_body_exprr"declrelemret_typer0s rvisit_FunctionzASR2PyVisitor.visit_Functions@DyyI8x .IGIGiiGYY )%a ) "% {{** ,#DKK$7$7$<=#,GNN7+, ,% %t$ % NN,, $//.. <&vi'89DOO00#((;(8)*CDD)&.&%,& H LL   )EJ% "!  "sF:F?? G Gc|jS)zReturns the AST nodesrrs rret_astzASR2PyVisitor.ret_ast&s << rN) __name__ __module__ __qualname____doc__rr$r3rBrLrOrRrergrQrrrr:s8   !+ *Z7 *r" .H < 1 *f rrceZdZdZy)rctd)Nzlfortran not available) ImportError)rrWkwargss rrzASR2PyVisitor.__init__+s67 7rN)rhrirjrrQrrrr*s 8rc\t}|j||j}|S)aCalls 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 )rrrg) fort_nodevres_asts rr-r-.s'( AGGIiikG NrcLt|d}t|}t|}|S)a$Wrapper 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_asrr-)srca_astapy_srcs r src_to_sympyr|Hs( sU 3E5A !_F MrN)sympy.externalrrsympy.codegen.astrrrrr r r sympy.corer r rrsympy.core.symbolrr(asr_modastrvsemanticrw ASTVisitorrr-r|rQrrrs(  $ KKK33(llG ,,  C((J""--88J'Tn n `884r