\hISrSSKJr SSKJrJrJrJrJrJ r J r J r J r SSK Jr SSKJr SSKJr SSKJr SSKJrJr SS KJr SS KJr SS KJrJr SS KJ r \"S 5r!\"S5r"\"S5r#\"S5r$\"S5r%\"S5r&"SS\ 5r'"SS\ 5r(Sr)"SS\ 5r*"SS\ 5r+"SS\5r,"SS\ 5r-"S S!\ 5r."S"S#\ 5r/"S$S%\ 5r0"S&S'\5r1\1"5r2S(r3\3"S)5r4SRS+S*S*S,.S-jjr5S.r6S/r7SSS0jr8SSS1jr9SRS2jr:SSS3jr;SSS4jr"S8S9\ 5r?"S:S;\5r@"S<S=\@5rA"S>S?\@5rB"S@SA\@5rC"SBSC\@5rD"SDSE\@5rE"SFSG\A5rF"SHSI\5rG"SJSK\G5rH"SLSM\G5rI"SNSO\ \5rJ"SPSQ\ \5rKg*)Tz 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 allocatablec6\rSrSrSrS=rr\r\ "S5r Sr g)Program"a9Represents 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 )namebodyc[U6$Nrr#s L/var/www/auris/envauris/lib/python3.13/site-packages/sympy/codegen/fnodes.pyProgram.4 40@N) __name__ __module__ __qualname____firstlineno____doc__ __slots___fieldsr _construct_name staticmethod_construct_body__static_attributes__r-r,r(r r "s#+*IO"#@AOr,r c(\rSrSrSrS=rr\r\r Sr g) use_rename7aRepresents 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 )localoriginalr-N) r.r/r0r1r2r3r4r _construct_local_construct_originalr8r-r,r(r:r:7s0/I r,r:cR[US5(a UR$[U5$)Nr")hasattrr"r args r(_namerDKs"sFxxc{r,c^\rSrSrSrS=rr\\S.r\ "\ 5r \ "S5r \ "S5r Srg) useQaRepresents 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' ) namespacerenameonly)rIrJc x[UVs/sH"n[U[5(aUO[U6PM$ sn6$s snfr%)r isinstancer:argsrCs r(r) use.dsP%C:DCwzCQ[A\A\#blnqbr:rC:D3E:Ds)7c ~[UVs/sH%n[U[5(aUO [U5PM' sn6$s snfr%)rrLr:rDrMs r(r)rOes7vz7{vzorz#z?Z?Z`efi`j8jvz7{0|7{s,:r-N)r.r/r0r1r2r3r4rdefaultsr6rD_construct_namespace_construct_rename_construct_onlyr8r-r,r(rFrFQsE:9I-H'.$&EF"#|}Or,rFcX\rSrSrSrS=rrS\"50r\ r \ S5r \ "S5rSrg) Moduleha'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 )r" declarations definitionsrXcUVs/sH%n[U[5(a [U5OUPM' nn[U6$s snfr%)rLstrrr)clsrNrCs r(_construct_declarationsModule._construct_declarations~s<EIJTcJsC00Cc9TJ$Ks,;c[U6$r%r&rBs r(r)Module.sior,r-N)r.r/r0r1r2r3r4rrQr r5 classmethodr]r6_construct_definitionsr8r-r,r(rVrVhsE"BAI(HO  **EFr,rVc`\rSrSrSrSr\\R-r\r \ "S5r \ S5r Srg) 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 )r" parametersr#cB[[[RU56$r%)rmapr deduced)paramss r(r)Subroutine.ss8CSCSU[?\8]r,c@[U[5(aU$[U6$r%)rLr)r\itrs r(r7Subroutine._construct_bodys c9 % %Jc? "r,r-N)r.r/r0r1r2r3rr4r r5r6_construct_parametersrar7r8r-r,r(rdrds?$/I$,,&GO()]^##r,rdc@\rSrSrSrS=rr\"\5r \"\ 5r Sr g)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"subroutine_argsr-N) r.r/r0r1r2r3r4r6rDr5r _construct_subroutine_argsr8r-r,r(rqrqs( 65I"5)O!-i!8r,rqc\rSrSrSrS=rr\"S5\S.r \ "S5r \ "\ 5r \ "\ 5r\ "\ 5r\ "\ 5r\ "S5rSrg ) 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 )r#counterfirstlaststep concurrent)r{r|c[U6$r%r&r's r(r) Do.r+r,c(U(a[$[$r%rrBs r(r)rsST5Ke5Kr,r-N)r.r/r0r1r2r3r4rrrQr6r7r_construct_counter_construct_first_construct_last_construct_step_construct_concurrentr8r-r,r(rvrvsc4UTI %8H"#@AO%g.#G,"7+O"7+O()KLr,rvc0\rSrSrSrS=rr\"\5r Sr g)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]' )elementsr-N) r.r/r0r1r2r3r4r6r _construct_elementsr8r-r,r(rrs ('I&y1r,rc\rSrSrSrS=rrS\"S50r\ "\ 5r \ "\ 5r \ "\ 5r \ "\ 5r\ "\ 5rSrg) 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]' )exprrxryrzr{r{r}r-N)r.r/r0r1r2r3r4rrQr6r_construct_exprrrrrr8r-r,r(rrsV GFI #H"7+O%g.#G,"7+O"7+Or,rc$\rSrSrSrSrSrSrg)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 [U5S:Xa.Uup#[R"U[U5[U55$[U5S:Xd[U5S:XaUSS;a[R"U5$[ S5e)Nrr}):Nz5Expected 0 or 2 args (or one argument == None or ':'))lenr __new__r ValueError)r\rNlowhighs r(rExtent.__new__sj t9>IC==gclGDMB B Y!^D Q47k3I==% %TU Ur,cz[UR5S:XagSRSUR55$)Nrrc38# UHn[U5v M g7fr%)r[).0rCs r( #Extent._sympystr..%s6ISCIs)rrNjoin)selfprinters r( _sympystrExtent._sympystr"s. tyy>Q xx6DII666r,r-N)r.r/r0r1r2rrr8r-r,r(rrs$V7r,rc[U5S:a [S5e/nUHn[U[5(aUR U5 M+[U[ 5(a=US:XaUR [55 MaUR [ U55 M}[U5(aUR [U65 MUR [U55 M [U5S:Xa [S5e[SU5$)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 arraysrrzNeed at least one dimension dimension) rrrLrappendr[r rrr)rNrfrCs r(rr*s 4y1}KLLJ c6 " "   c " S ! !cz!!&(+!!&+. c]]   fcl +   gcl + 4yA~677 [* --r,*Nr-)attrsvaluetypec|[U[5(a1[UR5S:wa[ S[U5-5eO[ U6n[ U5U/-nUb;U[[[4;a[[[S.UnURU5 Uc[R"XUS9$[XXCS9$)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: %s)inoutinout)rr) rLrr[r"rrlistrrrrr ri)symboldimintentrrrs r(arrayrOs:#y!! sxx=K 'NQTUXQYYZ Z (o K3% E  )Z> >%j<PQWXF V |5AAE??r,cX[U[5(a [U5$[U5$r%)rLr[r rrBs r( _printabler|s!$S#..6#;@GCL@r,c.[S[U5/5$)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[S[U5/U(a [U5/O/-U(a[U5/-5$/-5$)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[S[U5/U(a [U5/O/-U(a[U5/-5$/-5$)Nuboundrrs r(rrsV  E !*S/ r +#*T   - *, - r,cf[S[U5/U(a[U5/-5$/-5$)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[S[U5/U(a [U5/O/-U(a[U5/-5$/-5$)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[S[U5[U5/U(a [U5/O/-U(a[U5/-5$/-5$)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,cH[SU(a[U5/5$/5$)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)rr )r"s r(rrs!6 Xt ~ >>2 >>r,cH\rSrSrSrS=rrS\0r\ "\ 5r \ "\ 5r Srg)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' )labelsrrr-N)r.r/r0r1r2r3r4rrQr6r _construct_labelsrrr8r-r,r(rrs1 -,I~H$Y/"7+Or,rc8\rSrSrSrS=rrS\0r\ "\ 5r Sr g) 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_valuerr-N) r.r/r0r1r2r3r4rrQr6r_construct_return_valuer8r-r,r(rr-s((,+I%H*73r,rc\rSrSrSrSrSrg) FFunctioniGMc URRnURSUR:a[ SX R4-5eSR USR [URUR555$)Nstandardz%s requires Fortran %d or newerz{}({})z, ) __class__r. _settings_required_standardNotImplementedErrorformatrrh_printrN)rrr"s r(_fcodeFFunction._fcodeJsw~~&&   Z (4+B+B B%&G'+-D-D&E'FG GtTYYs7>>499/M%NOOr,r-N)r.r/r0r1rrr8r-r,r(rrGsPr,rc\rSrSrSrSrg) F95FunctioniR_r-N)r.r/r0r1rr8r-r,r(rrRsr,rc\rSrSrSrSrSrg)isigniVz.Fortran sign intrinsic for integer arguments. rr-Nr.r/r0r1r2nargsr8r-r,r(rrVs 9 Er,rc\rSrSrSrSrSrg)dsigni[z7Fortran sign intrinsic for double precision arguments. rr-Nrr-r,r(rr[s B Er,rc\rSrSrSrSrSrg)cmplxi`z%Fortran complex conversion function. rr-Nrr-r,r(rr`s 0 Er,rc\rSrSrSrSrSrg)riezFortran kind function. r}r-Nrr-r,r(rre " Er,rc\rSrSrSrSrSrg)mergeijzFortran merge function r-Nrr-r,r(rrjrr,rc0\rSrSr%S\S'S\S'SrSrg) _literalior[_tokenint _decimalsc2SRUR5U-RS5upEURS5R S5nUSUSSR S5pvUS:XaSOUnU=(d SUR -U-U=(d S-$) Nz%.{}ee0.rr}+)rrsplitstriprstriplstripr)rrrNkwargsmantissasgnd_exex_sgnex_nums r(r_literal._fcodess$^^DNN;dBII#N>>#&--c2 WQR[%7%7%<}&C4;;.76=SIIr,r-N)r.r/r0r1__annotations__rr8r-r,r(rros KNJr,rc \rSrSrSrSrSrSrg) literal_spi{z&Fortran single precision real literal r r-Nr.r/r0r1r2rrr8r-r,r(rr{s1 FIr,rc \rSrSrSrSrSrSrg) literal_dpiz&Fortran double precision real literal dr-Nr r-r,r(r r s1 FIr,r cF\rSrSrS=rr\\S.r\"\ 5r \"\ 5r Sr g)sum_irrmaskrrr-N r.r/r0r1r3r4rrQr6r_construct_array_construct_dimr8r-r,r(rr+22IT*H#G,!'*Nr,rcF\rSrSrS=rr\\S.r\"\ 5r \"\ 5r Sr g)product_irrr-Nrr-r,r(rrrr,rr%)NN)Lr2 __future__rsympy.codegen.astrrrrrr r r r sympy.core.basicr sympy.core.containersrsympy.core.exprrsympy.core.functionrsympy.core.numbersrrsympy.core.symbolrsympy.core.sympifyr sympy.logicrrsympy.utilities.iterablesrrrrrrrr r:rDrFrVrdrqrvrrrassumed_extentr assumed_sizerrrrrrrrrrrrrrrrrrrrr rrr-r,r(r's##' (-!&#. k " k " | $ (  & BeB*!!( ~%~.GUG<##>9U9""M"MJ2u2&,E,07U7B .F~ +@Rt$+@ZA : 848$?:,5,&4E44PP)I I I 9 K Ju J  +5$++ud+r,