a Gd"d#d#e6Z?dS)$aq Fortran code printer The FCodePrinter converts single SymPy expressions into single Fortran expressions, using the functions defined in the Fortran 77 standard where possible. Some useful pointers to Fortran can be found on wikipedia: https://en.wikipedia.org/wiki/Fortran Most of the code below is based on the "Professional Programmer's Guide to Fortran77" by Clive G. Page: http://www.star.le.ac.uk/~cgp/prof77.html Fortran is a case-insensitive language. This might cause trouble because SymPy is case sensitive. So, fcode adds underscores to variable names when it is necessary to make them different for Fortran. )DictAny) defaultdict)chainN) Assignment DeclarationPointer value_constfloat32float64float80 complex64 complex128int8int16int32int64intcrealintegerbool_complex_) allocatableisigndsigncmplxmerge literal_dp elementalpure intent_in intent_out intent_inout)SAddNFloatSymbol)Function)Eq)Range) CodePrinter) precedence PRECEDENCE)printer_context)fcode print_fcodesincostanasinacosatanatan2sinhcoshtanhlogexperfabsZconjgmaxmin)r1r2r3r4r5r6r7r8r9r:r;r<r=Abs conjugateZMaxZMincseZdZdZdZdZeeee e e iZ e dede dedede d ed ed ed ed edi Ze ddiiZdddiddddddd ZddddddZdd iZdfd!d" Zed#d$Zfd%d&Zd'd(Zd)d*Zd+d,Z d-d.Z!d/d0Z"d1d2Z#d3d4Z$d5d6Z%d7d8Z&d9d:Z'd;d<Z(d=d>Z)d?d@Z*dAdBZ+dCdDZ,dEdFZ-dGdHZ.dIdJZ/dKdLZ0dMdNZ1dOdPZ2dQdRZ3dSdTZ4dUdVZ5dWdXZ6dYdZZ7d[d\Z8d]d^Z9d_d`Z:dadbZ;dcddZdidjZ?dkdlZ@dmdnZAdodpZBdqdrZCdsdtZDdudvZEdwdxZFdydzZGd{d|ZHd}d~ZIddZJddZKddZLddZMddZNddZOddZPddZQddZRddZSddZTddZUddZVZWS) FCodePrinterzAA printer to convert SymPy expressions to strings of Fortran codeZ_fcodeZFortranzinteger(c_int)zreal*4zreal*8zreal*10z complex*8z complex*16z integer*1z integer*2z integer*4z integer*8logical iso_c_bindingc_intNautoTFfixedM) orderZ full_prec precisionuser_functionsZhumanZallow_unknown_functions source_formatZcontractstandard name_manglingz.and.z.or.z.neqv.z.eqv.z.not. )andorxorZ equivalentnotz!=z/=cs|si}i|_g|_tt|j|di|_tt|j|di|_t |tt |_ | di}|j |hd}|j d|vrtd|j dtt|_dS)N type_aliases type_mappingsrM>BrJiZ_rOzUnknown Fortran standard: %s)mangled_symbols used_namedictrrUitemspoprVsuper__init__known_functionsgetupdate _settings ValueErrorrset module_uses)selfsettingsZ userfuncsZ standards __class__f/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/sympy/printing/fortran.pyrazs(      zFCodePrinter.__init__cCsJ|jddkrddddS|jddkr4ddd dStd |jddS) NrNrIz z @ zC )codecontcommentfreez! zUnknown source format: %s)rerfrirmrmrn_leads   zFCodePrinter._leadcs|jddkrv||jvrj|j}||jvr6|d7}q|j|||jkr\||j|<nt||j|<||j}t |}|S)NrPT_) rer[namelowerr\appendr'Zxreplacer` _print_Symbol)riexprrwrkrmrnrzs      zFCodePrinter._print_SymbolcCs | dS)Nrm)riprmrmrn_rate_index_positionsz!FCodePrinter._rate_index_positioncCs|SNrm)riZ codestringrmrmrn_get_statementszFCodePrinter._get_statementcCs d|S)Nz! {})format)ritextrmrmrn _get_commentszFCodePrinter._get_commentcCsd|||S)Nzparameter ({} = {}))r_print)rirwvaluermrmrn_declare_number_constsz"FCodePrinter._declare_number_constcCs(|j|t||jdft|S)NrL)Z_number_symbolsaddr&Zevalfrestrrir{rmrmrn_print_NumberSymbols z FCodePrinter._print_NumberSymbolcCs|||Sr) _wrap_fortran indent_code)rilinesrmrmrn _format_codeszFCodePrinter._format_codecs |j\}fddt|DS)Nc3s$|]}tD]}||fVqqdSr)range).0jirowsrmrn z8FCodePrinter._traverse_matrix_indices..)shaper)rimatcolsrmrrn_traverse_matrix_indicess z%FCodePrinter._traverse_matrix_indicescCs^g}g}|D]H}t|j|j|jd|jdg\}}}|d|||f|dq ||fS)Nzdo %s = %s, %send do)maprlabelrxupperry)riindicesZ open_linesZ close_linesrvarstartstoprmrmrn_get_loop_opening_endings  z%FCodePrinter._get_loop_opening_endingcCsddlm}|j\}|jr4tdtd|t|d}nb|js@|jrrtt t dt d|||t||t d}n$tt dt t d|t|t d}| |S)Nr)rAr) Z$sympy.functions.elementary.complexesrAargs is_integerrrr)Z is_complex is_infiniterrrr)rir{rAargZnew_exprrmrmrn _print_signs  2$zFCodePrinter._print_signc Cs:|jdjdkrtdg}|trt|jD]t\}\}}|dkrZ|d||n:|t|jdkr|dkr|dn|d|||||q0|d d |S|j d d kr.d }||jdj }t |jdd}|r*| \}}||}||} |j||| d}q|StddS)NTzAll Piecewise expressions must contain an (expr, True) statement to be used as a default condition. Without one, the generated expression may not evaluate to anything under some condition.rz if (%s) thenrelsezelse if (%s) thenend if rOrZzmerge({T}, {F}, {COND}))TFZCONDzmUsing Piecewise as an expression using inline operators is not supported in standards earlier than Fortran95.)rcondrfZhasr enumerateryrlenjoinrer{listr_rNotImplementedError) rir{rrecpatternroZtermsrrmrmrn_print_Piecewises0       zFCodePrinter._print_PiecewisecCs,d|j|jtddd|jd|jdS)Nz {}({}, {})ZAtomT)strictr)r parenthesizeparentr-rrrrmrmrn_print_MatrixElements  z!FCodePrinter._print_MatrixElementc Csg}g}g}|jD]>}|jr.|jr.||q|jrF|jrF||q||q|r|rt|}t|}||}|drd} |dd}nd} t||krd|}d|t||t j t|| |fSd|t||t j t|fSn t ||SdS)N-r+z(%s)zcmplx(%s,%s) %s %sz cmplx(%s,%s)) r is_numberZis_realry is_imaginaryr,r$r startswithr# ImaginaryUnitr+ _print_Add) rir{Z pure_realZpure_imaginarymixedrPRECZtermtsignrmrmrnrs<           zFCodePrinter._print_AddcsR|jdfdd|jD}|j|}t|ts<||St||j|SdS)NrLcsg|]}t|qSrm)r%)raprecrmrn 0rz0FCodePrinter._print_Function..)rerfunc isinstancer(rr+_print_Function)rir{rZ eval_exprrmrrnr-s     zFCodePrinter._print_FunctioncCsB|jddvrd}t|n"|j\}}d||||SdS)NrO)rWrJz]Python % operator and SymPy's Mod() function are not supported by Fortran 66 or 77 standards.z modulo({}, {}))rerrrr)rir{msgxyrmrmrn _print_Mod7s   zFCodePrinter._print_ModcCsdS)Nz cmplx(0,1)rmrrmrmrn_print_ImaginaryUnitDsz!FCodePrinter._print_ImaginaryUnitcCst|Srrrrmrmrn _print_intHszFCodePrinter._print_intcCs2|jr"|jr"d|tj |St||SdS)Nz cmplx(0,%s))rrrr#rr+ _print_MulrrmrmrnrKs  zFCodePrinter._print_MulcCst|}|jdkr2d|td||j|fS|jdkr|jjrn|jjr\d||jSd||jSqd||jSn t ||SdS)Nrz%s/%srg?z sqrt(%s.0d0)zsqrt(dble(%s))zsqrt(%s)) r,r<rrrbaserZ is_Numberr+ _print_Pow)rir{rrmrmrnrTs    zFCodePrinter._print_PowcCs"t|jt|j}}d||fS)Nz %d.0d0/%d.0d0)intr}q)rir{r}rrmrmrn_print_RationalgszFCodePrinter._print_RationalcCsFt||}|d}|dkr>d|d|||ddfSd|S)Nrrz%sd%srz%sd0)r+ _print_Floatfind)rir{Zprintedrrmrmrnrks    zFCodePrinter._print_FloatcCsD||j}||j}|j}||jvr,|n|j|}d|||S)Nz{} {} {})rlhsrhsZrel_op _relationalsr)rir{lhs_coderhs_codeoprmrmrn_print_Relationalrs   zFCodePrinter._print_Relationalcs0fdd|jD}d|jjd|fS)Ncsg|]}|qSrmr)rrrtrmrnrzrz/FCodePrinter._print_Indexed..%s(%s), )rrrrr)rir{Zindsrmrtrn_print_IndexedyszFCodePrinter._print_IndexedcCs ||jSr)rrrrmrmrn _print_Idx}szFCodePrinter._print_Idxcs>|j}|j}djtfdd||j|gS)Nz{0} = {0} {1} {2}cs |Srrrrtrmrnrz9FCodePrinter._print_AugmentedAssignment..)rrrrrrbinop)rir{rrrmrtrn_print_AugmentedAssignments    z'FCodePrinter._print_AugmentedAssignmentcCs^||j}|jdkr*|d||j7}|jdkrH|d||j7}d|jjd|fS)Nrz, mask=rrv)rarraydimmaskrl__name__rstrip)rismparamsrmrmrn _print_sum_s    zFCodePrinter._print_sum_cCs ||Sr)r)riprodrmrmrn_print_product_szFCodePrinter._print_product_cs\dg}|jdkr |dd}nd}d|djfd|jr>dndi|jfd d |d S) N concurrentrsteprs, {step}z*do {concurrent}{counter} = {first}, {last}z {body} end do z concurrent cs |Srrrrtrmrnrrz(FCodePrinter._print_Do..)applyexclude)rryrrkwargs)ridoZexclrrmrtrn _print_Dos    zFCodePrinter._print_Docs:|jdkrdnd}d|djfi|jfdddS) Nrrsrz$({expr}, {counter} = {first}, {last})cs |Srrrrtrmrnrrz3FCodePrinter._print_ImpliedDoLoop..r)rrr)riZidlrrmrtrn_print_ImpliedDoLoopsz!FCodePrinter._print_ImpliedDoLoopcCsP||j}t|jtr(|jj\}}}ntd||j}dj|||||dS)Nz*Only iterable currently supported is Rangez3do {target} = {start}, {stop}, {step} {body} end do)targetrrrbody) rrriterabler*rrrr)rir{rrrrrrmrmrn _print_Fors   zFCodePrinter._print_ForcCsP|j||}|j||j}|j|}|rL|D]\}}|j||q2|Sr)rUrcrVrw type_modulesrhr)ritype_Ztype_strrhkvrmrmrn _print_Types  zFCodePrinter._print_Typecs,dj|jdtfdd|jdS)Nz{symbol}({idxs})rcs |Srrrrtrmrnrrz-FCodePrinter._print_Element..)symbolZidxs)rrr rrr)rielemrmrtrn_print_Elements zFCodePrinter._print_ElementcCst|Srr)riextrmrmrn _print_ExtentszFCodePrinter._print_Extentc sT|jj}d}fddtttfD}|ddkrBd}n2|ddkrhdgd |d}n td t t rt d j d d krdj jtjvrdnd|rddtfdd|nd|tjvrdndjd}|dkrP|d|7}n:tjvs(|r0t ddtfddjjg}|S)NZ dimensioncsg|]}|jvqSrm)attrs)rintent)rrmrnrrz3FCodePrinter._print_Declaration..Trrsrz , intent(%s))inoutZinoutz!Multiple intents specified for %sz1Pointers are not available by default in Fortran.rOrYz"{t}{vc}{dim}{intent}{alloc} :: {s}z , parameterz, dimension(%s)rcs |Srrrrtrmrnrrz1FCodePrinter._print_Declaration..z , allocatable)rZvcrrZallocsz = %sz0F77 init./parameter statem. req. multiple lines. cs |Srrrrtrmrnrr)variabler attr_paramsr r!r"countindexrfrrrrerrtyper rrrrr )rir{valrZintentsrresultrm)rirrn_print_Declarations4    "   zFCodePrinter._print_DeclarationcCsd|tdS)Nz(huge(%s) + 1)r)rrrrmrmrn_print_InfinityszFCodePrinter._print_Infinitycs djfi|jfdddS)Nz$do while ({condition}) {body} end docs |Srrrrtrmrnrrz+FCodePrinter._print_While..rrrrrmrtrn _print_Whiles  zFCodePrinter._print_WhilecCsdS)Nz.true.rmrrmrmrn_print_BooleanTrueszFCodePrinter._print_BooleanTruecCsdS)Nz.false.rmrrmrmrn_print_BooleanFalsesz FCodePrinter._print_BooleanFalsecCsRg}|D]D}|dr8||jd|ddq||jd|q|S)N!rqrro)rryrulstrip)rirrlinermrmrn_pad_leading_columnss  "z!FCodePrinter._pad_leading_columnscstdtjtjtdfdd}g}|jddkrBd}nd}|D]}||jd r t|d kr|d d d }|d krd }|d|}||d }| ||r|d dd}|d kst|dkrd}|d|}||d }| d|jd |fqn | |qJ||jdr||d }|d| }||d }|rj||7}| ||r||d}|d| }||d }|r||7}| d|jd|fqtqJ| |qJ|S)zWrap long Fortran lines Argument: lines -- a list of lines (without \n character) A comment line is split at white space. Code lines are split with a more complex rule to give nice results. z_+-.z ()csJt|krtS|}fdd}||sF|d8}|dkr(|Sq(|S)Ncsp|vr|dvpn|vr8|dvpn|vrT|dvpn|von|dvS)Nrrm)pos)r'my_alnummy_whitermrnrs zDFCodePrinter._wrap_fortran..split_pos_code..rr)r)r'endposr)splitr*r+)r'rnsplit_pos_code s z2FCodePrinter._wrap_fortran..split_pos_coderNrrz &rsrqHrrNrrW%s%sroArp) rgstringdigits ascii_lettersrerrurrfindr&ryr)rirr/rZtrailingr'r)Zhunkrmr.rnrsP          zFCodePrinter._wrap_fortrancs@t|tr$||d}d|S|jddk}dd|D}ddfd d|D}fd d|D}d d|D}d }d }d } g} t|D]\} } | dvr| | q||| 8}|rd|| |} n d|| } d| | f} |s|| gd } | | || rd| }nd }||| 7}q|s<| | S| S)z0Accepts a string of code or a list of code linesTrsrNrrcSsg|]}|dqS)z )r&rr'rmrmrnrLrz,FCodePrinter.indent_code..)zdo zif(zif zdo rprogramZ interface)rZenddorendifrz end programz end interfacecs g|]}ttt|jqSrmranyrrr8) inc_keywordrmrnrQscs g|]}ttt|jqSrmr;r8) dec_keywordrmrnrSscSs$g|]}ttt|jddgqS)&z& )rr<rendswithr8rmrmrnrUsr)rsrrr2) rrr splitlinesrrerryr(r)riroZ code_linesrrZincreaseZdecreaseZ continuationlevelZ cont_paddingZtabwidthZnew_coderr'paddingrm)r>r=rnrEsN            zFCodePrinter.indent_codecsL|jr2djdtfdd|j|jdS|j\}d|SdS)Nzgo to ({labels}), {expr}rcs |Srrrrtrmrnrzrz*FCodePrinter._print_GoTo..)labelsr{zgo to %s)r{rrrrFr)rigotoZlblrmrtrn _print_GoTows zFCodePrinter._print_GoTocs djfi|jfdddS)Nz"program {name} {body} end program cs |Srrrrtrmrnrrz-FCodePrinter._print_Program..rr!)riprogrmrtrn_print_ProgramszFCodePrinter._print_Programcs djfi|jfdddS)NzAmodule {name} {declarations} contains {definitions} end module cs |Srrrrtrmrnrrz,FCodePrinter._print_Module..rr!)rimodrmrtrn _print_ModuleszFCodePrinter._print_ModulecCsp|jdkr,|jddkr,|jdddS|jdkrX|jddkrX|jddd S|jdkrfd S|jSdS) NstdoutrOrXrEzstdint=>input_unitZ input_unitstderrzstdint=>error_unitZ error_unit*)rwrerhr)ristrmrmrmrn _print_Streams zFCodePrinter._print_Streamcs@|jdkr|j}nd}dj|dtfdd|jdS)NrOzprint {fmt}, {iolist}rcs |Srrrrtrmrnrrz+FCodePrinter._print_Print..)fmtZiolist) format_stringrrrrZ print_args)riZpsrRrmrtrn _print_Prints   zFCodePrinter._print_PrintcCs&|j\}dj|jdd||dS)Nz{result_name} = {arg} result_nameZ sympy_result)rUr)rr_contextrcr)rirsrrmrmrn _print_Returns  zFCodePrinter._print_ReturncCs"|j\}|rd||SdSdS)Nz return %sreturn)rr)riZfrsrrmrmrn_print_FortranReturnsz!FCodePrinter._print_FortranReturnc s|d}|durd}n|r(d|dnd}jdd}dj||jdfd d |jD|rnd |nd|d tfd d|jdS)NZbind_Crsz bind(C, name="%s")rz bind(C)rUz<{entity}{name}({arg_names}){result}{bind} {arg_declarations}rcsg|]}|jqSrm)rr )rrrtrmrnrrz&FCodePrinter._head..z result(%s)rcst|Sr)rrrrtrmrnrrz$FCodePrinter._head..)entityrw arg_namesrbindZarg_declarations) rrercrrrwr parametersr)rir[fprZ bind_C_paramsr]rUrmrtrn_heads  zFCodePrinter._headcCs&d||j}dj|||dS)N {} function z4interface {function_head} end function end interface) function_head)rr return_typer`)rir_r[rmrmrn_print_FunctionPrototypes  z%FCodePrinter._print_FunctionPrototypecCst|jvrd}nt|jvr d}nd}d||j}t||jd0dj||||||j dWdS1sz0YdS)Nz elemental zpure rsra)rUz,{prefix}{function_head} {body} end function )prefixrbr) rrrrrrcr.rwr`r)rifdrer[rmrmrn_print_FunctionDefinitions    z&FCodePrinter._print_FunctionDefinitioncCsdj|d|||jdS)Nz({subroutine_head} {body} end subroutine z subroutine )Zsubroutine_headr)rr`rr)risubrmrmrn_print_Subroutines   zFCodePrinter._print_Subroutinecs,dj|jdtfdd|jdS)Nzcall {name}({args})rcs |Srrrrtrmrnrrz4FCodePrinter._print_SubroutineCall..)rwr)rrrwrrZsubroutine_args)riZscallrmrtrn_print_SubroutineCalls z"FCodePrinter._print_SubroutineCallcsdttfdd|jS)Nz%s => %scs |Srrrrtrmrnrrz0FCodePrinter._print_use_rename..)tuplerr)rirnmrmrtrn_print_use_renameszFCodePrinter._print_use_renamecsld|j}|jdkr<|ddfdd|jD7}|jdkrh|ddfdd|jD7}|S)Nzuse %srcsg|]}|qSrmr)rrlrtrmrnrrz+FCodePrinter._print_use..z, only: csg|]}|qSrmr)rZnlyrtrmrnrr)r namespacerenameronly)riZuserrmrtrn _print_uses  " "zFCodePrinter._print_usecCsdS)Nexitrmrirvrmrmrn_print_BreakTokenszFCodePrinter._print_BreakTokencCsdS)Ncyclermrsrmrmrn_print_ContinueTokensz!FCodePrinter._print_ContinueTokencs4jddkrdnd}|dtfdd|jS)NrOrXz[%s]z(/%s/)rcs |Srrrrtrmrnrrz6FCodePrinter._print_ArrayConstructor..)rerrelements)riacZfmtstrrmrtrn_print_ArrayConstructorsz$FCodePrinter._print_ArrayConstructor)N)Xr __module__ __qualname____doc__Z printmethodlanguagerrrr rrrUrr r r rrrrrVrZ_default_settings _operatorsrrapropertyrurzr~rrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrrr r"r#r$r(rrrHrJrLrQrTrXrZr`rdrgrirjrmrqrtrvry __classcell__rmrmrkrnrCDs     *'     ! G2     rC)@r|typingrZtDictr collectionsr itertoolsrr4Zsympy.codegen.astrrrr r r r r rrrrrrrrrrZsympy.codegen.fnodesrrrrrrrrr r!r"Z sympy.corer#r$r%r&r'Zsympy.core.functionr(Zsympy.core.relationalr)Z sympy.setsr*Zsympy.printing.codeprinterr+Zsympy.printing.precedencer,r-Zsympy.printing.printerr.r/r0rbrCrmrmrmrnsB  P4