U ZʹjI@sdZddlmZmZmZmZmZmZmZm Z m Z ddl m Z ddl mZddlmZddlmZddlmZmZddlmZdd lmZdd lmZmZdd lmZed Zed Z edZ!edZ"edZ#edZ$GdddeZ%GdddeZ&ddZ'GdddeZ(GdddeZ)GdddeZ*GdddeZ+Gd d!d!eZ,Gd"d#d#eZ-Gd$d%d%eZ.Gd&d'd'e Z/e/Z0d(d)Z1e1d*Z2d\d,d+d+d-d.d/Z3d0d1Z4d2d3Z5d]d4d5Z6d^d6d7Z7d_d8d9Z8d`d:d;Z9dadd?Z;Gd@dAdAeZGdFdGdGe>Z?GdHdIdIe>Z@GdJdKdKe>ZAGdLdMdMe>ZBGdNdOdOe>ZCGdPdQdQe?ZDGdRdSdSeZEGdTdUdUeEZFGdVdWdWeEZGGdXdYdYeeZHGdZd[d[eeZId+S)cz 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. ) Attribute CodeBlock FunctionCallNodenoneStringToken _mk_TupleVariable)BasicTuple)Expr)Function)FloatInteger)Str)sympifytruefalse)iterablepure elemental intent_in intent_out intent_inout allocatablec@s(eZdZdZdZZeZeddZ 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 )namebodycCst|SNrr r$S/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/codegen/fnodes.py3zProgram.N) __name__ __module__ __qualname____doc__ __slots___fieldsr_construct_name staticmethod_construct_bodyr$r$r$r%r!src@s eZdZdZdZZeZeZdS) 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+r,r-rZ_construct_localZ_construct_originalr$r$r$r%r16sr1cCst|dr|jSt|SdS)Nr)hasattrrrargr$r$r%_nameJs r7c@sBeZdZdZdZZeedZee Z eddZ eddZ 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)r:r;cCstdd|DS)NcSs"g|]}t|tr|nt|qSr$) isinstancer1.0r6r$r$r% cs use...r argsr$r$r%r&cr'z use.cCstdd|DS)NcSs"g|]}t|tr|nt|qSr$)r<r1r7r=r$r$r%r?dsr@r rAr$r$r%r&dr'N) r(r)r*r+r,r-rdefaultsr/r7Z_construct_namespaceZ_construct_renameZ_construct_onlyr$r$r$r%r8Ps   r8c@s>eZdZdZdZZdeiZeZ e ddZ e ddZ 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 declarationsZ definitionsrEcCsdd|D}t|S)NcSs"g|]}t|trt|n|qSr$)r<strrr=r$r$r%r?sz2Module._construct_declarations..r")clsrBr$r$r%_construct_declarations}szModule._construct_declarationscCst|Sr!r"r5r$r$r%r&r'zModule.N)r(r)r*r+r,r-r rCrr. classmethodrHr/Z_construct_definitionsr$r$r$r%rDgs  rDc@s:eZdZdZdZeejZeZe ddZ e ddZ 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 cCstttj|Sr!)r mapr deduced)paramsr$r$r%r&r'zSubroutine.cCst|tr|St|SdSr!)r<r)rGZitrr$r$r%r0s zSubroutine._construct_bodyN) r(r)r*r+r,rr-rr.r/Z_construct_parametersrIr0r$r$r$r%rJs  rJc@s(eZdZdZdZZeeZee 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)' )rZsubroutine_argsN) r(r)r*r+r,r-r/r7r.r Z_construct_subroutine_argsr$r$r$r%rOs rOc@s^eZdZdZdZZededZe ddZ e e Z e e Z e e Ze e Ze ddZdS) 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)rTrUcCst|Sr!r"r#r$r$r%r&r'z Do.cCs |rtStSr!rr5r$r$r%r&r'N)r(r)r*r+r,r-rrrCr/r0r_construct_counter_construct_first_construct_last_construct_stepZ_construct_concurrentr$r$r$r%rPs rPc@s eZdZdZdZZeeZdS)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+r,r-r/r Z_construct_elementsr$r$r$r%r[sr[c@sLeZdZdZdZZdediZee 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]' )exprrQrRrSrTrTrVN)r(r)r*r+r,r-rrCr/r_construct_exprrWrXrYrZr$r$r$r%r]s r]c@s eZdZdZddZddZdS)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' cGsdt|dkr*|\}}t|t|t|St|dksNt|dkrX|ddkrXt|StddS)NrrV):Nz5Expected 0 or 2 args (or one argument == None or ':'))lenr __new__r ValueError)rGrBlowhighr$r$r%rds  $ zExtent.__new__cCs(t|jdkrdSddd|jDS)Nrrbcss|]}t|VqdSr!)rFr=r$r$r% $sz#Extent._sympystr..)rcrBjoin)selfprinterr$r$r% _sympystr!szExtent._sympystrN)r(r)r*r+rdrlr$r$r$r%r`s r`cGst|dkrtdg}|D]p}t|tr6||qt|trf|dkrV|tq|t|qt|r~|t|q|t|qt|dkrtdt d|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 arraysrbrzNeed at least one dimension dimension) rcrer<r`appendrFrrrr)rBrKr6r$r$r%rn)s      rn*Nr$)attrsvaluetypecCst|tr*t|jdkr2tdt|nt|}t||g}|dk rp|ttt fkrfttt d|}| ||dkrt j |||dSt ||||dSdS)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 rnz/Got an unexpected Attribute argument as dim: %sN)inoutZinout)rrrq) r<rrFrrernlistrrrror rM)symboldimZintentrqrrrsr$r$r%arrayNs  rycCst|trt|St|Sr!)r<rFrrr5r$r$r% _printable{srzcCstdt|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)' allocatedrrz)ryr$r$r%r{s r{cCs4tdt|g|rt|gng|r,t|gngS)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)' lboundr|ryrxkindr$r$r%r}sr}cCs4tdt|g|rt|gng|r,t|gngS)Nuboundr|r~r$r$r%rsrcCs"tdt|g|rt|gngS)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|)sourcerr$r$r%rs rcCs4tdt|g|rt|gng|r,t|gngS)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 sizer|r~r$r$r%rsrcCs:tdt|t|g|r t|gng|r2t|gngS)z Creates an AST node for a function call to Fortran's "reshape(...)" Parameters ========== source : Symbol or String shape : ArrayExpr reshaper|)rrpadorderr$r$r%rs rcCstd|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)rr$r$r%rsrc@s0eZdZdZdZZdeiZee 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' )labelsr^r^N) r(r)r*r+r,r-rrCr/r Z_construct_labelsrr_r$r$r$r%rs  rc@s(eZdZdZdZZdeiZee 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+r,r-rrCr/rZ_construct_return_valuer$r$r$r%r,src@seZdZdZddZdS) FFunctionMcCsF|jj}|jd|jkr*td||jfd|dt|j|j S)Nstandardz%s requires Fortran %d or newerz{}({})z, ) __class__r(Z _settings_required_standardNotImplementedErrorformatrirLZ_printrB)rjrkrr$r$r%_fcodeIs zFFunction._fcodeN)r(r)r*rrr$r$r$r%rFsrc@seZdZdZdS) F95Function_N)r(r)r*rr$r$r$r%rQsrc@seZdZdZdZdS)isignz/ Fortran sign intrinsic for integer arguments. raNr(r)r*r+nargsr$r$r$r%rUsrc@seZdZdZdZdS)dsignz8 Fortran sign intrinsic for double precision arguments. raNrr$r$r$r%rZsrc@seZdZdZdZdS)cmplxz& Fortran complex conversion function. raNrr$r$r$r%r_src@seZdZdZdZdS)rz Fortran kind function. rVNrr$r$r$r%rdsrc@seZdZdZdZdS)mergez Fortran merge function Nrr$r$r$r%risrc@seZdZdZdZddZdS)_literalNcOspd|j|d\}}|dd}|d|ddd}}|dkrRdn|}|p\d|j||pldS) Nz%.{}ee0.rrV+)r _decimalssplitstriprstriplstrip_token)rjrkrBkwargsZmantissaZsgnd_exZex_sgnZex_numr$r$r%rrs z_literal._fcode)r(r)r*rrrr$r$r$r%rnsrc@seZdZdZdZdZdS) literal_spz' Fortran single precision real literal r Nr(r)r*r+rrr$r$r$r%rzsrc@seZdZdZdZdZdS) literal_dpz' Fortran double precision real literal dNrr$r$r$r%rsrc@s.eZdZdZZeedZeeZ eeZ dS)sum_ryrxmaskrxrN r(r)r*r,r-rrCr/rZ_construct_arrayZ_construct_dimr$r$r$r%rs rc@s.eZdZdZZeedZeeZ eeZ dS)product_rrNrr$r$r$r%rs r)N)NN)NN)N)NN)NN)N)Jr+Zsympy.codegen.astrrrrrrrr r Zsympy.core.basicr Zsympy.core.containersr Zsympy.core.exprrZsympy.core.functionrZsympy.core.numbersrrZsympy.core.symbolrZsympy.core.sympifyrZ sympy.logicrrZsympy.utilities.iterablesrrrrrrrrr1r7r8rDrJrOrPr[r]r`Zassumed_extentrnZ assumed_sizeryrzr{r}rrrrrrrrrrrrrrrrrrrr$r$r$r%sj,       %!#-