B [-J@sdZddlmZddlmZddlmZddlmZddl m Z ddl m Z m Z ddlmZdd lmZmZmZmZmZmZmZmZmZmZmZdd lmZmZdd lm Z ed Z!ed Z"edZ#edZ$edZ%edZ&GdddeZ'GdddeZ(ddZ)GdddeZ*GdddeZ+GdddeZ,GdddeZ-Gd d!d!eZ.Gd"d#d#eZ/Gd$d%d%eZ0Gd&d'd'eZ1e1Z2d(d)Z3e3d*Z4dZd,d-Z5d.d/Z6d0d1Z7d[d2d3Z8d\d4d5Z9d]d6d7Z:d^d8d9Z;d_d:d;Zd?d?eZ>Gd@dAdAeZ?GdBdCdCe Z@GdDdEdEe@ZAGdFdGdGe@ZBGdHdIdIe@ZCGdJdKdKe@ZDGdLdMdMe@ZEGdNdOdOeAZFGdPdQdQe ZGGdRdSdSeGZHGdTdUdUeGZIGdVdWdWeeZJGdXdYdYeeZKd+S)az 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. )Basic)Expr) string_types)Tuple)Function)FloatInteger)sympify) Attribute CodeBlock Declaration FunctionCallNodenoneStringTokenType _mk_TupleVariable)truefalse)iterablepure elemental intent_in intent_out intent_inout allocatablec@s(eZdZdZddgZeZeddZdS)Programam 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.printing import fcode >>> print(fcode(prog, source_format='free')) program myprogram print *, 42 end program namebodycCst|S)N)r )r r!3lib/python3.7/site-packages/sympy/codegen/fnodes.py2szProgram.N) __name__ __module__ __qualname____doc__ __slots__r_construct_name staticmethod_construct_bodyr!r!r!r"r!src@s eZdZdZddgZeZeZdS) use_renamea Represents a renaming in a use statement in Fortran Examples ======== >>> from sympy.codegen.fnodes import use_rename, use >>> from sympy.printing 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 ZlocalZoriginalN)r$r%r&r'r(rZ_construct_localZ_construct_originalr!r!r!r"r,5sr,cCst|dr|jSt|SdS)Nr)hasattrrr)argr!r!r"_nameHs r/c@sDeZdZdZdddgZeedZeeZ eddZ eddZ d S) usea Represents a use statement in Fortran Examples ======== >>> from sympy.codegen.fnodes import use >>> from sympy.printing 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)r2r3cCstdd|DS)NcSs"g|]}t|tr|nt|qSr!) isinstancer,).0r.r!r!r" `sz use...)r)argsr!r!r"r#`sz use.cCstdd|DS)NcSs"g|]}t|tr|nt|qSr!)r4r,r/)r5r.r!r!r"r6asz use...)r)r7r!r!r"r#asN) r$r%r&r'r(rdefaultsr*r/Z_construct_namespaceZ_construct_renameZ_construct_onlyr!r!r!r"r0Ns    r0c@s@eZdZdZdddgZdeiZeZe ddZ e ddZ dS) Modulead Represents a module in Fortran Examples ======== >>> from sympy.codegen.fnodes import Module >>> from sympy.printing import fcode >>> print(fcode(Module('signallib', ['implicit none'], []), source_format='free')) module signallib implicit none contains end module rZ declarationsZ definitionscCst|S)N)r )r.r!r!r"r#yszModule.cCst|S)N)r )r.r!r!r"r#zsN) r$r%r&r'r(rr8rr)r*Z_construct_declarationsZ_construct_definitionsr!r!r!r"r9ds    r9c@s8eZdZdZddddgZeZeddZe dd Z d S) Subroutinea  Represents a subroutine in Fortran Examples ======== >>> from sympy import symbols >>> from sympy.codegen.ast import Print >>> from sympy.codegen.fnodes import Subroutine >>> from sympy.printing import fcode >>> 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 attrscCstttj|S)N)rmaprdeduced)Zparamsr!r!r"r#szSubroutine.cCst|tr|St|SdS)N)r4r )clsZitrr!r!r"r+s zSubroutine._construct_bodyN) r$r%r&r'r(rr)r*Z_construct_parameters classmethodr+r!r!r!r"r:}s   r:c@s(eZdZdZddgZeeZeeZ dS)SubroutineCalla Represents a call to a subroutine in Fortran Examples ======== >>> from sympy.codegen.fnodes import SubroutineCall >>> from sympy.printing import fcode >>> fcode(SubroutineCall('mysub', 'x y'.split())) ' call mysub(x, y)' rZsubroutine_argsN) r$r%r&r'r(r*r/r)rZ_construct_subroutine_argsr!r!r!r"rAs rAc@sfeZdZdZddddddgZeded Zed d Z ee Z ee Z ee Z ee Zed d Zd S)Doa Represents a Do loop in in Fortran Examples ======== >>> from sympy import symbols >>> from sympy.codegen.ast import aug_assign, Print >>> from sympy.codegen.fnodes import Do >>> from sympy.printing import fcode >>> 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)rFrGcCst|S)N)r )r r!r!r"r#sz Do.cCs |rtStS)N)rr)r.r!r!r"r#sN)r$r%r&r'r(rrr8r*r+r _construct_counter_construct_first_construct_last_construct_stepZ_construct_concurrentr!r!r!r"rBs rBc@seZdZdZdgZeeZdS)ArrayConstructora\ Represents an array constructor Examples ======== >>> from sympy.printing 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*rZ_construct_elementsr!r!r!r"rMsrMc@sReZdZdZdddddgZdediZeeZ eeZ eeZ eeZ eeZ dS) 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]' exprrCrDrErFrHN)r$r%r&r'r(rr8r*r _construct_exprrIrJrKrLr!r!r!r"rOs rOc@s eZdZdZddZddZdS)ExtentaL Represents a dimension extent. Examples ======== >>> from sympy.codegen.fnodes import Extent >>> e = Extent(-3, 3) # -3, -2, -1, 0, 1, 2, 3 >>> from sympy.printing 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)NrrH):Nz5Expected 0 or 2 args (or one argument == None or ':'))lenr__new__r ValueError)r?r7ZlowZhighr!r!r"rVs  $ zExtent.__new__cCst|jdkrdSd|jS)NrrTz%d:%d)rUr7)selfprinterr!r!r" _sympystrszExtent._sympystrN)r$r%r&r'rVrZr!r!r!r"rRs rRcGst|dkrtdg}xx|D]p}t|tr8||qt|trh|dkrX|tq|t|qt|r|t|q|t|qWt|dkrtdt d|S)a Creates a 'dimension' Attribute with (up to 7) extents. Examples ======== >>> from sympy.printing 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 arraysrTrzNeed at least one dimension dimension) rUrWr4rRappendrrrr r )r7r;r.r!r!r"r\s       r\*NcKst|tr*t|jdkr2tdt|nt|}t|dg|g}|dk rx|tt t fkrntt t d|}| ||dd}|dd}|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 passsed to :func:`dimension` as ``*dim`` intent : str One of: 'in', 'out', 'inout' or None \*\*kwargs: Keyword arguments for ``Variable`` ('type' & 'value') Examples ======== >>> from sympy.printing 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 r\z/Got an unexpected Attribute argument as dim: %sr<N)inoutZinoutvaluetype)rar<)r4r strrrWr\listpoprrrr]rr>)ZsymboldimZintentkwargsr<raZtype_r!r!r"arrayDs    rhcCst|trt|St|S)N)r4rrr )r.r!r!r" _printablersricCstdt|gS)a  Creates an AST node for a function call to Fortran's "allocated(...)" Examples ======== >>> from sympy.printing import fcode >>> from sympy.codegen.fnodes import allocated >>> alloc = allocated('x') >>> fcode(alloc, source_format='free') 'allocated(x)' allocated)r ri)rhr!r!r"rjvs rjcCs4tdt|g|rt|gng|r,t|gngS)ax Creates an AST node for a function call to Fortran's "lbound(...)" Parameters ========== array : Symbol or String dim : expr kind : expr Examples ======== >>> from sympy.printing import fcode >>> from sympy.codegen.fnodes import lbound >>> lb = lbound('arr', dim=2) >>> fcode(lb, source_format='free') 'lbound(arr, 2)' lbound)r ri)rhrfkindr!r!r"rksrkcCs4tdt|g|rt|gng|r,t|gngS)Nubound)r ri)rhrfrlr!r!r"rmsrmcCs"tdt|g|rt|gngS)aZ Creates an AST node for a function call to Fortran's "shape(...)" Parameters ========== source : Symbol or String kind : expr Examples ======== >>> from sympy.printing import fcode >>> from sympy.codegen.fnodes import shape >>> shp = shape('x') >>> fcode(shp, source_format='free') 'shape(x)' shape)r ri)sourcerlr!r!r"rnsrncCs4tdt|g|rt|gng|r,t|gngS)a Creates an AST node for a function call to Fortran's "size(...)" Examples ======== >>> from sympy import Symbol >>> from sympy.printing import fcode >>> from sympy.codegen.ast import FunctionDefinition, real, Return, Variable >>> 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 size)r ri)rhrfrlr!r!r"rpsrpcCs: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 reshape)r ri)rornZpadorderr!r!r"rqs  rqcCstd|rt|gngS)a* Creates an Attribute ``bind_C`` with a name Parameters ========== name : str Examples ======== >>> from sympy import Symbol >>> from sympy.printing import fcode >>> from sympy.codegen.ast import FunctionDefinition, real, Return, Variable >>> from sympy.codegen.fnodes import array, sum_, size, 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)r r)rr!r!r"rssrsc@s0eZdZdZddgZdeiZeeZ 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.printing import fcode >>> fcode(go, source_format='free') 'go to (10, 20, 30), i' labelsrPN) r$r%r&r'r(rr8r*rZ_construct_labelsr rQr!r!r!r"rt s  rtc@s&eZdZdZdgZdeiZeeZ dS) FortranReturna2 AST node explicitly mapped to a fortran "return". 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.printing import fcode >>> fcode(FortranReturn('x')) ' return x' Z return_valueN) r$r%r&r'r(rr8r*r Z_construct_return_valuer!r!r!r"rvsrvc@seZdZdZddZdS) FFunctionMcCsF|jj}|jd|jkr*td||jfd|dt|j|j S)NZstandardz%s requires Fortran %d or newerz{0}({1})z, ) __class__r$Z _settings_required_standardNotImplementedErrorformatjoinr=Z_printr7)rXrYrr!r!r"_fcode8s zFFunction._fcodeN)r$r%r&rzr~r!r!r!r"rw5srwc@seZdZdZdS) F95Function_N)r$r%r&rzr!r!r!r"r@src@seZdZdZdZdS)isignz/ Fortran sign intrinsic for integer arguments. rSN)r$r%r&r'nargsr!r!r!r"rDsrc@seZdZdZdZdS)dsignz8 Fortran sign intrinsic for double precision arguments. rSN)r$r%r&r'rr!r!r!r"rIsrc@seZdZdZdZdS)cmplxz& Fortran complex conversion function. rSN)r$r%r&r'rr!r!r!r"rNsrc@seZdZdZdZdS)rlz Fortran kind function. rHN)r$r%r&r'rr!r!r!r"rlSsrlc@seZdZdZdZdS)mergez Fortran merge function N)r$r%r&r'rr!r!r!r"rXsrc@seZdZdZdZddZdS)_literalNcOspd|j|d\}}|dd}|d|ddd}}|dkrRdn|}|p\d|j||pldS) Nz%.{0}ee0.rrH+)r| _decimalssplitstriprstriplstrip_token)rXrYr7rgZmantissaZsgnd_exZex_sgnZex_numr!r!r"r~as z_literal._fcode)r$r%r&rrr~r!r!r!r"r]src@seZdZdZdZdZdS) literal_spz' Fortran single precision real literal r N)r$r%r&r'rrr!r!r!r"risrc@seZdZdZdZdZdS) literal_dpz' Fortran double precision real literal dN)r$r%r&r'rrr!r!r!r"rosrc@s0eZdZdddgZeedZeeZeeZ dS)sum_rhrfmask)rfrN) r$r%r&r(rr8r*r _construct_array_construct_dimr!r!r!r"rus  rc@s0eZdZdddgZeedZeeZeeZ dS)product_rhrfr)rfrN) r$r%r&r(rr8r*r rrr!r!r!r"r|s  r)N)NN)NN)N)NN)NN)N)Lr'Zsympy.core.basicrZsympy.core.exprrZsympy.core.compatibilityrZsympy.core.containersrZsympy.core.functionrZsympy.core.numbersrrZsympy.core.sympifyr Zsympy.codegen.astr r r r rrrrrrrZ sympy.logicrrZsympy.utilities.iterablesrrrrrrrrr,r/r0r9r:rArBrMrOrRZassumed_extentr\Z assumed_sizerhrirjrkrmrnrprqrsrtrvrwrrrrrlrrrrrrr!r!r!r"sj      4 %!# .