XhIbdZddlmZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZddlmZmZdd lmZdd lmZdd lmZmZdd lm Z ed Z!edZ"edZ#edZ$edZ%edZ&Gdde Z'Gdde Z(dZ)Gdde Z*Gdde Z+GddeZ,Gdde Z-Gd d!e Z.Gd"d#e Z/Gd$d%e Z0Gd&d'eZ1e1Z2d(Z3e3d)Z4dRd+d*d*d,d-Z5d.Z6d/Z7dSd0Z8dSd1Z9dRd2Z:dSd3Z;dSd4ZGd8d9e Z?Gd:d;eZ@Gd<d=e@ZAGd>d?e@ZBGd@dAe@ZCGdBdCe@ZDGdDdEe@ZEGdFdGeAZFGdHdIeZGGdJdKeGZHGdLdMeGZIGdNdOe eZJGdPdQe eZKd*S)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)BasicTuple)Expr)Function)FloatInteger)Str)sympifytruefalse)iterablepure elemental intent_in intent_out intent_inout allocatablec6eZdZdZdxZZeZedZ dS)Programaf Represents 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 )namebodyct|SNrr#s f/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/sympy/codegen/fnodes.pyzProgram.4 40@N) __name__ __module__ __qualname____doc__ __slots___fieldsr _construct_name staticmethod_construct_bodyr+r(r!r!"s<+*IO"l#@#@AAOOOr+r!c"eZdZdZdxZZeZeZdS) use_renamea Represents 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 )localoriginalN) r,r-r.r/r0r1r _construct_local_construct_originalr5r+r(r7r77s20/I r+r7cNt|dr|jSt|S)Nr")hasattrr"r args r(_namer@Ks'sFxc{{r+cjeZdZdZdxZZeedZee Z edZ edZ dS)usea Represents 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)rDrEc(td|DS)NcNg|]"}t|tr|nt|#Sr5) isinstancer7.0r?s r( z use...dsF:D:D:DwzCQ[A\A\:r##blnqbr:D:D:Dr+rargss r(r)z use.ds1%:D:DC:D:D:D3Er+c(td|DS)NcZg|](}t|tr|nt|)Sr5)rHr7r@rIs r(rKz use...es67{7{7{orz#z?Z?Z8j`efi`j`j7{7{7{r+rrLs r(r)z use.es7{7{vz7{7{7{0|r+N) r,r-r.r/r0r1rdefaultsr3r@_construct_namespace_construct_rename_construct_onlyr5r+r(rBrBQst:9I--H'<..$ &E&EFF"l#|#|}}OOOr+rBcdeZdZdZdxZZdeiZeZ e dZ e dZ dS)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 )r" declarations definitionsrVc,d|D}t|S)NcZg|](}t|trt|n|)Sr5)rHstrrrIs r(rKz2Module._construct_declarations..s1JJJcJsC009CcJJJr+r&)clsrMs r(_construct_declarationszModule._construct_declarations~s JJTJJJ$r+ct|Sr%r&r>s r(r)zModule.s ior+N)r,r-r.r/r0r1rrPr r2 classmethodr\r3_construct_definitionsr5r+r(rUrUhsk"BAI(HO  [ *\*E*EFFr+rUc\eZdZdZdZeejzZeZe dZ e dZ dS) Subroutinea Represents 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#cDtttj|Sr%)rmapr deduced)paramss r(r)zSubroutine.ss8CSU[?\?\8]r+cBt|tr|St|Sr%)rHr)r[itrs r(r4zSubroutine._construct_bodys# c9 % % #Jc? "r+N) r,r-r.r/r0rr1r r2r3_construct_parametersr^r4r5r+r(rarasc$/I$,&GO(L)])]^^##[###r+racFeZdZdZdxZZeeZee Z dS)SubroutineCallz Represents 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_argsN) r,r-r.r/r0r1r3r@r2r _construct_subroutine_argsr5r+r(rkrksE  65I"l5))O!-i!8!8r+rkceZdZdZdxZZededZe dZ e e Z e e Z e e Ze e Ze dZdS)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)rsrtct|Sr%r&r's r(r)z Do.r*r+c"|rtntSr%rr>s r(r)z Do.sS5KTTer+N)r,r-r.r/r0r1rrrPr3r4r_construct_counter_construct_first_construct_last_construct_step_construct_concurrentr5r+r(roros4UTI %88H"l#@#@AAO%g..#|G,,"l7++O"l7++O(L)K)KLLr+roc0eZdZdZdxZZeeZdS)ArrayConstructoraT 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]' )elementsN) r,r-r.r/r0r1r3r _construct_elementsr5r+r(r~r~s6  ('I&,y11r+r~ceZdZdZdxZZdediZee Z ee Z ee Z ee Z ee ZdS) ImpliedDoLoopa Represents 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]' )exprrprqrrrsrsruN)r,r-r.r/r0r1rrPr3r_construct_exprrxryrzr{r5r+r(rrs  GFI #H"l7++O%g..#|G,,"l7++O"l7++OOOr+rceZdZdZdZdZdS)ExtentaC 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' c8t|dkr5|\}}tj|t|t|St|dkst|dkr|ddvrtj|St d)Nrru):Nz5Expected 0 or 2 args (or one argument == None or ':'))lenr __new__r ValueError)r[rMlowhighs r(rzExtent.__new__s t99>>IC=gcllGDMMBB B YY!^^D Q47k3I3I=%% %TUU Ur+c~t|jdkrdSdd|jDS)Nrrc34K|]}t|VdSr%)rZrIs r( z#Extent._sympystr..%s(66SC666666r+)rrMjoin)selfprinters r( _sympystrzExtent._sympystr"s= ty>>Q  3xx66DI666666r+N)r,r-r.r/rrr5r+r(rrs?$VVV77777r+rcjt|dkrtdg}|D]}t|tr||-t|t rK|dkr"|tj|t |t|r|t||t|t|dkrtdtd|S)a Creates 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) rrrHrappendrZr rrr)rMrbr?s r(rr*s& 4yy1}}KLLLJ , , c6 " " ,   c " " " " S ! ! ,czz!!&((++++!!&++.... c]] ,   fcl + + + +   gcll + + + + 4yyA~~6777 [* - --r+*Nr5)attrsvaluetypect|tr8t|jdkrt dt|zn t |}t ||gz}|F|tttfvrtttd|}| ||tj |||St||||S)a Convenience 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: %sN)inoutinout)rr) rHrrZr"rrlistrrrrr re)symboldimintentrrrs r(arrayrOs:#y!! sx==K ' 'NQTUXQYQYYZZ Z (o KK3% E  )Z> > >%j<PPQWXF V |e5AAAAE????r+cht|trt|nt|Sr%)rHrZr rr>s r( _printabler|s'$S#.. @6#;;;GCLL@r+c>tdt|gS)a Creates 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&7%8 9 99r+ctdt|g|rt|gngz|rt|gngzS)ap 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(rrs_(  E  ! )*S//  r +# +*T     -  r+ctdt|g|rt|gngz|rt|gngzS)Nuboundrrs r(rrs]  E  ! )*S//  r +# +*T     -  r+chtdt|g|rt|gngzS)aR 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(rrsD&  F  # +*T     -  r+ctdt|g|rt|gngz|rt|gngzS)a Creates 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(rrs_(  E  ! )*S//  r +# +*T     -  r+ctdt|t|g|rt|gngz|rt|gngzS)z Creates an AST node for a function call to Fortran's "reshape(...)" Parameters ========== source : Symbol or String shape : ArrayExpr reshaper)rrpadorders r(rrsi  F  Z../! )*S//  r + # +*U     -  r+cFtd|rt|gngS)a Creates 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 X=t ~~2 > >>r+cNeZdZdZdxZZdeiZee Z ee Z dS)GoToa Represents 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' )labelsrrN) r,r-r.r/r0r1rrPr3r _construct_labelsrrr5r+r(rrsL  -,I~H$ Y//"l7++OOOr+rc8eZdZdZdxZZdeiZee Z dS) FortranReturnaK AST 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_valuerN) r,r-r.r/r0r1rrPr3r_construct_return_valuer5r+r(rr-s@(,+I%H*l733r+rceZdZdZdZdS) FFunctionMc |jj}|jd|jkrt d||jfzd|dt|j|j S)Nstandardz%s requires Fortran %d or newerz{}({})z, ) __class__r, _settings_required_standardNotImplementedErrorformatrrd_printrM)rrr"s r(_fcodezFFunction._fcodeJsy~&  Z (4+B B B%&G'+T-D&E'FGG GtTYYs7>49/M/M%N%NOOOr+N)r,r-r.rrr5r+r(rrGs.PPPPPr+rceZdZdZdS) F95Function_N)r,r-r.rr5r+r(rrRsr+rceZdZdZdZdS)isignz/ Fortran sign intrinsic for integer arguments. rNr,r-r.r/nargsr5r+r(rrVs99 EEEr+rceZdZdZdZdS)dsignz8 Fortran sign intrinsic for double precision arguments. rNrr5r+r(rr[sBB EEEr+rceZdZdZdZdS)cmplxz& Fortran complex conversion function. rNrr5r+r(rr`s00 EEEr+rceZdZdZdZdS)rz Fortran kind function. ruNrr5r+r(rre"" EEEr+rceZdZdZdZdS)mergez Fortran merge function Nrr5r+r(rrjrr+rc*eZdZUded<ded<dZdS)_literalrZ_tokenint _decimalsc>d|j|zd\}}|dd}|d|ddd}}|dkrdn|}|pd|jz|z|pdzS) Nz%.{}ee0.rru+)rrsplitstriprstriplstripr)rrrMkwargsmantissasgnd_exex_sgnex_nums r(rz_literal._fcodess$^^DN;;dBII#NN'>>#&&--c22 WQRR[%7%7%<%<}}&C4;.76=SIIr+N)r,r-r.__annotations__rr5r+r(rros<KKKNNNJJJJJr+rceZdZdZdZdZdS) literal_spz' Fortran single precision real literal r Nr,r-r.r/rrr5r+r(rr{s11 FIIIr+rceZdZdZdZdZdS) literal_dpz' Fortran double precision real literal dNrr5r+r(rrs11 FIIIr+rcLeZdZdxZZeedZeeZ eeZ dS)sum_rrmaskrrN r,r-r.r0r1rrPr3r_construct_array_construct_dimr5r+r(rrD22IT**H#|G,,!\'**NNNr+rcLeZdZdxZZeedZeeZ eeZ dS)product_rrNrr5r+r(rrrr+rr%)NN)Lr/ __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!r7r@rBrUrarkror~rrassumed_extentr assumed_sizerrrrrrrrrrrrrrrrrrrrrrrr5r+r(rs#"""""#"""""'''''' ((((((--------!!!!!!&&&&&&########......y Ik " " Ik " " Y| $ $ y(( i && BBBBBeBBB*!!!!!!!!( ~~~~~%~~~.GGGGGUGGG<########>99999U999""M"M"M"M"M"M"M"MJ22222u222&,,,,,E,,,077777U777B . . .Fy~~ +@Rt$+@+@+@+@+@ZAAA : : : 848$????:,,,,,5,,,&44444E4444PPPPPPPP)I I I 9 K J J J J Ju J J J  +++++5$++++++++ud+++++r+