+ jIRt^RIHt^RIHtHtHtHtHtH t H t H t H t ^RI Ht^RIHt^RIHt^RIHt^RIHtHt^RIHt^R IHt^R IHtHt^R IH t ]!R 4t!]!R 4t"]!R4t#]!R4t$]!R4t%]!R4t&!RR] 4t'!RR] 4t(Rt)!RR] 4t*!RR] 4t+!RR]4t,!RR] 4t-!RR ] 4t.!R!R"] 4t/!R#R$] 4t0!R%R&]4t1]1!4t2R't3]3!R(4t4RRR*RSR+R)R,R)/R-llt5R.t6R/t7RTR0lt8RTR1lt9RRR2lt:RTR3lt;RTR4lt!R8R9] 4t?!R:R;]4t@!R<R=]@4tA!R>R?]@4tB!R@RA]@4tC!RBRC]@4tD!RDRE]@4tE!RFRG]A4tF!RHRI]4tG!RJRK]G4tH!RLRM]G4tI!RNRO] ]4tJ!RPRQ] ]4tKR)#)Uz AST nodes specific to Fortran. The functions defined in this module allows the user to express functions such as ``dsign`` as a SymPy function for symbolic manipulation. ) annotations) Attribute CodeBlock FunctionCallNodenoneStringToken _mk_TupleVariable)Basic)Tuple)Expr)Function)FloatInteger)Str)sympifytruefalse)iterablepure elemental intent_in intent_out intent_inout allocatablec8]tRt^"tRtR;tt]t] !R4t Rt R#)Programa9Represents a 'program' block in Fortran. Examples ======== >>> from sympy.codegen.ast import Print >>> from sympy.codegen.fnodes import Program >>> prog = Program('myprogram', [Print([42])]) >>> from sympy import fcode >>> print(fcode(prog, source_format='free')) program myprogram print *, 42 end program c \V!#Nrbodys&U/home/luca/screen-checking/.venv/lib/python3.14/site-packages/sympy/codegen/fnodes.pyProgram.4 40@N)namer$) __name__ __module__ __qualname____firstlineno____doc__ __slots___fieldsr_construct_name staticmethod_construct_body__static_attributes__r*r)r%rr"s#+*IO"#@AOr)rc*]tRt^7tRtR;tt]t]t Rt R#) use_renameaRepresents a renaming in a use statement in Fortran. Examples ======== >>> from sympy.codegen.fnodes import use_rename, use >>> from sympy import fcode >>> ren = use_rename("thingy", "convolution2d") >>> print(fcode(ren, source_format='free')) thingy => convolution2d >>> full = use('signallib', only=['snr', ren]) >>> print(fcode(full, source_format='free')) use signallib, only: snr, thingy => convolution2d r*N)localoriginal) r,r-r.r/r0r1r2r_construct_local_construct_originalr6r*r)r%r8r87s0/I r)r8cT\VR4'd VP#\V4#r+)hasattrr+rargs&r%_namerBKs"sFxxc{r)cb]tRt^QtRtR;ttR]R]/t] !] 4t ] !R4t ] !R4t RtR#) useaRepresents a use statement in Fortran. Examples ======== >>> from sympy.codegen.fnodes import use >>> from sympy import fcode >>> fcode(use('signallib'), source_format='free') 'use signallib' >>> fcode(use('signallib', [('metric', 'snr')]), source_format='free') 'use signallib, metric => snr' >>> fcode(use('signallib', only=['snr', 'convolution2d']), source_format='free') 'use signallib, only: snr, convolution2d' renameonlyc  |\VUu.uF$p\V\4'dTM\V!NK& up!#uupir!)r isinstancer8argsrAs& r%r& use.dsP%C:DCwzCQ[A\A\#blnqbr:rC:D3E:Ds*9c  \VUu.uF&p\V\4'dTM \V4NK( up!#uupir!)r rHr8rBrIs& r%r&rKes7vz7{vzorz#z?Z?Z`efi`j8jvz7{0|7{s,;r*N) namespacerErF)r,r-r.r/r0r1r2rdefaultsr4rB_construct_namespace_construct_rename_construct_onlyr6r*r)r%rDrDQsI:9I$-H'.$&EF"#|}Or)rDcZ]tRt^htRtR;ttR]!4/t] t ] R4t ] !R4tRtR#)Modulea'Represents a module in Fortran. Examples ======== >>> from sympy.codegen.fnodes import Module >>> from sympy import fcode >>> print(fcode(Module('signallib', ['implicit none'], []), source_format='free')) module signallib implicit none contains end module declarationsc VUu.uF&p\V\4'd \V4MTNK( pp\V!#uupir!)rHstrrr)clsrJrAs&& r%_construct_declarationsModule._construct_declarations~s<EIJTcJsC00Cc9TJ$Ks,<c \V!#r!r"r@s&r%r&Module.sior)r*N)r+rT definitions)r,r-r.r/r0r1r2r rNrr3 classmethodrXr4_construct_definitionsr6r*r)r%rSrShsE"BAI(HO  **EFr)rScj]tRt^tRtRt]]P,t]t ] !R4t ] R4t RtR#) SubroutineaRepresents a subroutine in Fortran. Examples ======== >>> from sympy import fcode, symbols >>> from sympy.codegen.ast import Print >>> from sympy.codegen.fnodes import Subroutine >>> x, y = symbols('x y', real=True) >>> sub = Subroutine('mysub', [x, y], [Print([x**2 + y**2, x*y])]) >>> print(fcode(sub, source_format='free', standard=2003)) subroutine mysub(x, y) real*8 :: x real*8 :: y print *, x**2 + y**2, x*y end subroutine c D\\\PV4!#r!)r mapr deduced)paramss&r%r&Subroutine.ss8CSCSU[?\8]r)c D\V\4'dV#\V!#r!)rHr)rWitrs&&r%r5Subroutine._construct_bodys c9 % %Jc? "r)r*N)r+ parametersr$)r,r-r.r/r0r1rr2rr3r4_construct_parametersr]r5r6r*r)r%r`r`s?$/I$,,&GO()]^##r)r`cB]tRt^tRtR;tt]!]4t ]!] 4t Rt R#)SubroutineCallzRepresents a call to a subroutine in Fortran. Examples ======== >>> from sympy.codegen.fnodes import SubroutineCall >>> from sympy import fcode >>> fcode(SubroutineCall('mysub', 'x y'.split())) ' call mysub(x, y)' r*N)r+subroutine_args) r,r-r.r/r0r1r2r4rBr3r _construct_subroutine_argsr6r*r)r%rlrls( 65I"5)O!-i!8r)rlc]tRt^tRtR;ttR]!^4R]/t ] !R4t ] !] 4t ] !] 4t] !] 4t] !] 4t] !R4tRtR#) Doa:Represents a Do loop in in Fortran. Examples ======== >>> from sympy import fcode, symbols >>> from sympy.codegen.ast import aug_assign, Print >>> from sympy.codegen.fnodes import Do >>> i, n = symbols('i n', integer=True) >>> r = symbols('r', real=True) >>> body = [aug_assign(r, '+', 1/i), Print([i, r])] >>> do1 = Do(body, i, 1, n) >>> print(fcode(do1, source_format='free')) do i = 1, n r = r + 1d0/i print *, i, r end do >>> do2 = Do(body, i, 1, n, 2) >>> print(fcode(do2, source_format='free')) do i = 1, n, 2 r = r + 1d0/i print *, i, r end do step concurrentc \V!#r!r"r#s&r%r& Do.r(r)c *V'd\#\#r!rr@s&r%r&rtsST5Ke5Kr)r*N)r$counterfirstlastrqrr)r,r-r.r/r0r1r2rrrNr4r5r_construct_counter_construct_first_construct_last_construct_step_construct_concurrentr6r*r)r%rprpsg4UTI L%8H"#@AO%g.#G,"7+O"7+O()KLr)rpc2]tRt^tRtR;tt]!]4t Rt R#)ArrayConstructora+Represents an array constructor. Examples ======== >>> from sympy import fcode >>> from sympy.codegen.fnodes import ArrayConstructor >>> ac = ArrayConstructor([1, 2, 3]) >>> fcode(ac, standard=95, source_format='free') '(/1, 2, 3/)' >>> fcode(ac, standard=2003, source_format='free') '[1, 2, 3]' r*N)elements) r,r-r.r/r0r1r2r4r _construct_elementsr6r*r)r%rrs ('I&y1r)rc]tRt^tRtR;ttR]!^4/t] !] 4t ] !] 4t ] !] 4t ] !] 4t] !] 4tRtR#) ImpliedDoLoopaRepresents an implied do loop in Fortran. Examples ======== >>> from sympy import Symbol, fcode >>> from sympy.codegen.fnodes import ImpliedDoLoop, ArrayConstructor >>> i = Symbol('i', integer=True) >>> idl = ImpliedDoLoop(i**3, i, -3, 3, 2) # -27, -1, 1, 27 >>> ac = ArrayConstructor([-28, idl, 28]) # -28, -27, -1, 1, 27, 28 >>> fcode(ac, standard=2003, source_format='free') '[-28, (i**3, i = -3, 3, 2), 28]' rqr*N)exprrvrwrxrq)r,r-r.r/r0r1r2rrNr4r_construct_exprryrzr{r|r6r*r)r%rrsV GFI #H"7+O%g.#G,"7+O"7+Or)rc&]tRtRtRtRtRtRtR#)Extentia Represents a dimension extent. Examples ======== >>> from sympy.codegen.fnodes import Extent >>> e = Extent(-3, 3) # -3, -2, -1, 0, 1, 2, 3 >>> from sympy import fcode >>> fcode(e, source_format='free') '-3:3' >>> from sympy.codegen.ast import Variable, real >>> from sympy.codegen.fnodes import dimension, intent_out >>> dim = dimension(e, e) >>> arr = Variable('x', real, attrs=[dim, intent_out]) >>> fcode(arr.as_Declaration(), source_format='free', standard=2003) 'real*8, dimension(-3:3, -3:3), intent(out) :: x' c \V4^8Xd/Vwr#\P!V\V4\V44#\V4^8Xg\V4^8Xd%V^,R9d\P!V4#\ R4h)z5Expected 0 or 2 args (or one argument == None or ':')):N)lenr __new__r ValueError)rWrJlowhighs&* r%rExtent.__new__sj t9>IC==gclGDMB B Y!^D Q47k3I==% %TU Ur)c ~\VP4^8XdR#RPRVP44#)rc38"TFp\V4xK R#5ir!)rV).0rAs& r% #Extent._sympystr..%s6ISCIs)rrJjoin)selfprinters&&r% _sympystrExtent._sympystr"s. tyy>Q xx6DII666r)r*N)r,r-r.r/r0rrr6r*r)r%rrs$V7r)rc \V4^8d \R4h.pVFp\V\4'dVP V4K,\V\ 4'd?VR8XdVP \44KdVP \ V44K\V4'dVP \V!4KVP \V44K \V4^8Xd \R4h\RV4#)aCreates a 'dimension' Attribute with (up to 7) extents. Examples ======== >>> from sympy import fcode >>> from sympy.codegen.fnodes import dimension, intent_in >>> dim = dimension('2', ':') # 2 rows, runtime determined number of columns >>> from sympy.codegen.ast import Variable, integer >>> arr = Variable('a', integer, attrs=[dim, intent_in]) >>> fcode(arr.as_Declaration(), source_format='free', standard=2003) 'integer*4, dimension(2, :), intent(in) :: a' z0Fortran only supports up to 7 dimensional arraysrzNeed at least one dimension dimension) rrrHrappendrVrrrr)rJrirAs* r%rr*s 4y1}KLLJ c6 " "   c " S ! !cz!!&(+!!&+. c]]   fcl +   gcl + 4yA~677 [* --r)*Nattrsvaluetypec\V\4'd7\VP4R8wd\ R\V4,4hM \ V!p\ V4V.,pVeCV\\\39dR\R\R\/V,pVPV4Vf\P!WVR7#\WWCR7#)aConvenience function for creating a Variable instance for a Fortran array. Parameters ========== symbol : symbol dim : Attribute or iterable If dim is an ``Attribute`` it need to have the name 'dimension'. If it is not an ``Attribute``, then it is passed to :func:`dimension` as ``*dim`` intent : str One of: 'in', 'out', 'inout' or None \*\*kwargs: Keyword arguments for ``Variable`` ('type' & 'value') Examples ======== >>> from sympy import fcode >>> from sympy.codegen.ast import integer, real >>> from sympy.codegen.fnodes import array >>> arr = array('a', '*', 'in', type=integer) >>> print(fcode(arr.as_Declaration(), source_format='free', standard=2003)) integer*4, dimension(*), intent(in) :: a >>> x = array('x', [3, ':', ':'], intent='out', type=real) >>> print(fcode(x.as_Declaration(value=1), source_format='free', standard=2003)) real*8, dimension(3, :, :), intent(out) :: x = 1 rz/Got an unexpected Attribute argument as dim: %sinoutinout)rr) rHrrVr+rrlistrrrrr rc)symboldimintentrrrs&&&$$$r%arrayrOs:#y!! sxx=K 'NQTUXQYYZ Z (o K3% E  )Z> >Iuj'<PQWXF V |5AAE??r)cZ\V\4'd \V4#\V4#r!)rHrVrrr@s&r% _printabler|s!$S#..6#;@GCL@r)c.\R\V4.4#)zCreates an AST node for a function call to Fortran's "allocated(...)" Examples ======== >>> from sympy import fcode >>> from sympy.codegen.fnodes import allocated >>> alloc = allocated('x') >>> fcode(alloc, source_format='free') 'allocated(x)' allocatedrr)rs&r%rrs  j&7%8 99r)c\R\V4.V'd \V4.M.,V'd\V4.,4#.,4#)a;Creates an AST node for a function call to Fortran's "lbound(...)" Parameters ========== array : Symbol or String dim : expr kind : expr Examples ======== >>> from sympy import fcode >>> from sympy.codegen.fnodes import lbound >>> lb = lbound('arr', dim=2) >>> fcode(lb, source_format='free') 'lbound(arr, 2)' lboundrrrkinds&&&r%rrsX(  E !*S/ r +#*T   - *, - r)c\R\V4.V'd \V4.M.,V'd\V4.,4#.,4#)uboundrrs&&&r%rrsV  E !*S/ r +#*T   - *, - r)cx\R\V4.V'd\V4.,4#.,4#)a!Creates an AST node for a function call to Fortran's "shape(...)" Parameters ========== source : Symbol or String kind : expr Examples ======== >>> from sympy import fcode >>> from sympy.codegen.fnodes import shape >>> shp = shape('x') >>> fcode(shp, source_format='free') 'shape(x)' shaper)sourcers&&r%rrsE&  F #*T   - *, - r)c\R\V4.V'd \V4.M.,V'd\V4.,4#.,4#)aHCreates an AST node for a function call to Fortran's "size(...)" Examples ======== >>> from sympy import fcode, Symbol >>> from sympy.codegen.ast import FunctionDefinition, real, Return >>> from sympy.codegen.fnodes import array, sum_, size >>> a = Symbol('a', real=True) >>> body = [Return((sum_(a**2)/size(a))**.5)] >>> arr = array(a, dim=[':'], intent='in') >>> fd = FunctionDefinition(real, 'rms', [arr], body) >>> print(fcode(fd, source_format='free', standard=2003)) real*8 function rms(a) real*8, dimension(:), intent(in) :: a rms = sqrt(sum(a**2)*1d0/size(a)) end function sizerrs&&&r%rrsX(  E !*S/ r +#*T   - *, - r)c\R\V4\V4.V'd \V4.M.,V'd\V4.,4#.,4#)zCreates an AST node for a function call to Fortran's "reshape(...)" Parameters ========== source : Symbol or String shape : ArrayExpr reshaper)rrpadorders&&&&r%rrs_  F Z./!*S/ r + #*U   - *, - r)cJ\RV'd\V4.4#.4#)aCreates an Attribute ``bind_C`` with a name. Parameters ========== name : str Examples ======== >>> from sympy import fcode, Symbol >>> from sympy.codegen.ast import FunctionDefinition, real, Return >>> from sympy.codegen.fnodes import array, sum_, bind_C >>> a = Symbol('a', real=True) >>> s = Symbol('s', integer=True) >>> arr = array(a, dim=[s], intent='in') >>> body = [Return((sum_(a**2)/s)**.5)] >>> fd = FunctionDefinition(real, 'rms', [arr, s], body, attrs=[bind_C('rms')]) >>> print(fcode(fd, source_format='free', standard=2003)) real*8 function rms(a, s) bind(C, name="rms") real*8, dimension(s), intent(in) :: a integer*4 :: s rms = sqrt(sum(a**2)/s) end function bind_C)rrr>s&r%rrs!6 Xt ~ >>2 >>r)cJ]tRtRtRtR;ttR]/t] !] 4t ] !] 4t RtR#)GoToizRepresents a goto statement in Fortran Examples ======== >>> from sympy.codegen.fnodes import GoTo >>> go = GoTo([10, 20, 30], 'i') >>> from sympy import fcode >>> fcode(go, source_format='free') 'go to (10, 20, 30), i' rr*N)labelsr)r,r-r.r/r0r1r2rrNr4r _construct_labelsrrr6r*r)r%rrs1 -,I~H$Y/"7+Or)rc:]tRtRtRtR;ttR]/t] !] 4t Rt R#) FortranReturni-aAST node explicitly mapped to a fortran "return". Explanation =========== Because a return statement in fortran is different from C, and in order to aid reuse of our codegen ASTs the ordinary ``.codegen.ast.Return`` is interpreted as assignment to the result variable of the function. If one for some reason needs to generate a fortran RETURN statement, this node should be used. Examples ======== >>> from sympy.codegen.fnodes import FortranReturn >>> from sympy import fcode >>> fcode(FortranReturn('x')) ' return x' return_valuer*N)r) r,r-r.r/r0r1r2rrNr4r_construct_return_valuer6r*r)r%rr-s((,+I%H*73r)rc ]tRtRt^MtRtRtR#) FFunctioniGc  *VPPpVPR,VP8d\ RW P3,4hRP VRP \VPVP444#)standardz%s requires Fortran %d or newerz{}({})z, ) __class__r, _settings_required_standardNotImplementedErrorformatrrb_printrJ)rrr+s&& r%_fcodeFFunction._fcodeJsw~~&&   Z (4+B+B B%&G'+-D-D&E'FG GtTYYs7>>499/M%NOOr)r*N)r,r-r.r/rrr6r*r)r%rrGsPr)rc]tRtRt^_tRtR#) F95FunctioniRr*N)r,r-r.r/rr6r*r)r%rrRsr)rc]tRtRtRt^tRtR#)isigniVz.Fortran sign intrinsic for integer arguments. r*Nr,r-r.r/r0nargsr6r*r)r%rrVs 9 Er)rc]tRtRtRt^tRtR#)dsigni[z7Fortran sign intrinsic for double precision arguments. r*Nrr*r)r%rr[s B Er)rc]tRtRtRt^tRtR#)cmplxi`z%Fortran complex conversion function. r*Nrr*r)r%rr`s 0 Er)rc]tRtRtRt^tRtR#)riezFortran kind function. r*Nrr*r)r%rre " Er)rc]tRtRtRt^tRtR#)mergeijzFortran merge function r*Nrr*r)r%rrjrr)rc2]tRtRt$R]R&R]R&RtRtR#) _literaliorV_tokenint _decimalsc hRPVP4V,PR4wrEVPR4P R4pV^,VR,P R4rvVR8XdRMTpT;'gRVP ,V,T;'gR,#)z%.{}ee0.:NN+)rrsplitstriprstriplstripr)rrrJkwargsmantissasgnd_exex_sgnex_nums&&*, r%r_literal._fcodess$^^DNN;dBII#N>>#&--c2 WR[%7%7%<}&C4;;.76==SIIr)r*N)r,r-r.r/__annotations__rr6r*r)r%rros KNJr)rc"]tRtRtRtRt^ tRtR#) literal_spi{z&Fortran single precision real literal rr*Nr,r-r.r/r0rrr6r*r)r%rr{s1 FIr)rc"]tRtRtRtRt^tRtR#) literal_dpiz&Fortran double precision real literal dr*Nrr*r)r%rrs1 FIr)rcJ]tRtRtR;ttR]R]/t]!] 4t ]!] 4t Rt R#)sum_irmaskr*Nrrr r,r-r.r/r1r2rrNr4r_construct_array_construct_dimr6r*r)r%rr/22ItVT*H#G,!'*Nr)rcJ]tRtRtR;ttR]R]/t]!] 4t ]!] 4t Rt R#)product_irrr*Nrrr*r)r%r r rr)r r!r*)NN)Lr0 __future__rsympy.codegen.astrrrrrrr r r sympy.core.basicr sympy.core.containersr sympy.core.exprrsympy.core.functionrsympy.core.numbersrrsympy.core.symbolrsympy.core.sympifyr sympy.logicrrsympy.utilities.iterablesrrrrrrrrr8rBrDrSr`rlrprrrassumed_extentr assumed_sizerrrrrrrrrrrrrrrrrrrrrrr r*r)r%rs##' (-!&#. k " k " | $ (  & BeB*!!( ~%~.GUG<##>9U9""M"MJ2u2&,E,07U7B .F~ +@R+@t+@$+@ZA : 848$?:,5,&4E44PP)I I I 9 K Ju J  +5$++ud+r)