ó ¡¼™\c@sÝdZddlmZmZmZmZmZmZmZm Z m Z ddl m Z ddl mZddlmZddlmZddlmZddlmZmZdd lmZdd lmZmZdd lmZed ƒZed ƒZ edƒZ!edƒZ"edƒZ#edƒZ$defd„ƒYZ%defd„ƒYZ&d„Z'defd„ƒYZ(defd„ƒYZ)defd„ƒYZ*defd„ƒYZ+defd „ƒYZ,d!efd"„ƒYZ-d#efd$„ƒYZ.d%e fd&„ƒYZ/e/ƒZ0d'„Z1e1d(ƒZ2dNd)„Z4d*„Z5d+„Z6dNdNd,„Z7dNdNd-„Z8dNd.„Z9dNdNd/„Z:dNdNd0„Z;dNd1„Z<d2efd3„ƒYZ=d4efd5„ƒYZ>d6efd7„ƒYZ?d8e?fd9„ƒYZ@d:e?fd;„ƒYZAd<e?fd=„ƒYZBd>e?fd?„ƒYZCd@e?fdA„ƒYZDdBe@fdC„ƒYZEdDefdE„ƒYZFdFeFfdG„ƒYZGdHeFfdI„ƒYZHdJeefdK„ƒYZIdLeefdM„ƒYZJdNS(Os¬ 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. iÿÿÿÿ( t Attributet CodeBlockt FunctionCalltNodetnonetStringtTokent _mk_TupletVariable(tBasic(t string_types(tTuple(tExpr(tFunction(tFloattInteger(tsympify(ttruetfalse(titerabletpuret elementalt intent_int intent_outt intent_inoutt allocatabletProgramcBs/eZdZddgZeZed„ƒZRS(sm 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 tnametbodycCs t|ŒS(N(R(R((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyt2t(t__name__t __module__t__doc__t __slots__Rt_construct_namet staticmethodt_construct_body(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR!s t use_renamecBs&eZdZddgZeZeZRS(sÑ 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 tlocaltoriginal(RR R!R"Rt_construct_localt_construct_original(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR&5s cCs$t|dƒr|jSt|ƒSdS(NR(thasattrRR(targ((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyt_nameHstusecBs[eZdZdddgZied6ed6ZeeƒZed„ƒZ ed„ƒZ RS(sØ 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' t namespacetrenametonlycCs8tg|D]'}t|tƒr%|n t|Œ^q ŒS(N(R t isinstanceR&(targsR,((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR`RcCs8tg|D]'}t|tƒr%|n t|ƒ^q ŒS(N(R R2R&R-(R3R,((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRaR( RR R!R"RtdefaultsR$R-t_construct_namespacet_construct_renamet_construct_only(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR.Ns  tModulecBsQeZdZdddgZieƒd6ZeZed„ƒZ ed„ƒZ RS(sd 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 Rt declarationst definitionscCs t|ŒS(N(R(R,((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRyRcCs t|ŒS(N(R(R,((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRzR( RR R!R"R R4RR#R$t_construct_declarationst_construct_definitions(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR8ds t SubroutinecBsDeZdZddddgZeZed„ƒZed„ƒZ RS(s  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 Rt parametersRtattrscCstttj|ƒŒS(N(R tmapRtdeduced(tparams((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR’RcCs!t|tƒr|St|ŒSdS(N(R2R(tclstitr((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR%”s( RR R!R"RR#R$t_construct_parameterst classmethodR%(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR=}s tSubroutineCallcBs2eZdZddgZeeƒZeeƒZRS(s 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)' Rtsubroutine_args( RR R!R"R$R-R#Rt_construct_subroutine_args(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRG›s   tDocBsŽeZdZddddddgZiedƒd6ed6Zed„ƒZee ƒZ ee ƒZ ee ƒZ ee ƒZ ed „ƒZRS( s¯ 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 Rtcountertfirsttlasttstept concurrenticCs t|ŒS(N(R(R((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRÈRcCs|r tStS(N(RR(R,((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRÍR(RR R!R"RRR4R$R%Rt_construct_countert_construct_firstt_construct_lastt_construct_stept_construct_concurrent(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRJ«s    tArrayConstructorcBs#eZdZdgZeeƒZRS(s\ 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]' telements(RR R!R"R$Rt_construct_elements(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRUÐs t ImpliedDoLoopcBsreZdZdddddgZiedƒd6ZeeƒZeeƒZ eeƒZ eeƒZ eeƒZ RS(sÈ 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]' texprRKRLRMRNi( RR R!R"RR4R$Rt_construct_exprRPRQRRRS(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRXãs    tExtentcBs eZdZd„Zd„ZRS(sL 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' cGsŽt|ƒdkr=|\}}tj|t|ƒt|ƒƒSt|ƒdksqt|ƒdkr~|ddkr~tj|ƒStdƒ‚dS(Niiit:s5Expected 0 or 2 args (or one argument == None or ':')(R\N(tlenR t__new__RtNonet ValueError(RCR3tlowthigh((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR^s  4 cCs$t|jƒdkrdSd|jS(NiR\s%d:%d(R]R3(tselftprinter((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyt _sympystrs(RR R!R^Re(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR[ûs cGsþt|ƒdkr!tdƒ‚ng}x¦|D]ž}t|tƒrS|j|ƒq.t|tƒr—|dkr|jtƒƒqÌ|jt|ƒƒq.t|ƒr¹|jt|Œƒq.|jt|ƒƒq.Wt|ƒdkrñtdƒ‚nt d|ƒS(s 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' is0Fortran only supports up to 7 dimensional arraysR\isNeed at least one dimensiont dimension( R]R`R2R[tappendR RRRR(R3R>R,((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRfs    t*cKs$t|tƒr@t|jƒdkrLtdt|ƒƒ‚qLn t|Œ}t|jdgƒƒ|g}|d k r¾|t t t fkr®it d6t d6t d6|}n|j |ƒn|jdd ƒ}|jdd ƒ}|d krt j|d|d|ƒSt ||d|d|ƒSd S( só 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 Rfs/Got an unexpected Attribute argument as dim: %sR?tintouttinouttvaluettypeN(R2RtstrRR`RftlisttpopR_RRRRgRRA(tsymboltdimtintenttkwargsR?Rlttype_((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pytarrayDs  " cCs#t|tƒrt|ƒSt|ƒS(N(R2R RR(R,((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyt _printablerscCstdt|ƒgƒS(s  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)' t allocated(RRw(Rv((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRxvs cCsHtdt|ƒg|r't|ƒgng|r@t|ƒgngƒS(sx 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)' tlbound(RRw(RvRrtkind((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRy…s%cCsHtdt|ƒg|r't|ƒgng|r@t|ƒgngƒS(Ntubound(RRw(RvRrRz((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR{ s%cCs/tdt|ƒg|r't|ƒgngƒS(sZ 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)' tshape(RRw(tsourceRz((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR|©s cCsHtdt|ƒg|r't|ƒgng|r@t|ƒgngƒS(s± 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 tsize(RRw(RvRrRz((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR~Âs%cCsQtdt|ƒt|ƒg|r0t|ƒgng|rIt|ƒgngƒS(s Creates an AST node for a function call to Fortran's "reshape(...)" Parameters ========== source : Symbol or String shape : ArrayExpr treshape(RRw(R}R|tpadtorder((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRßs .cCs"td|rt|ƒgngƒS(s* 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 tbind_C(RR(R((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR‚ðstGoTocBs?eZdZddgZied6ZeeƒZee ƒZ RS(s 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' tlabelsRY( RR R!R"RR4R$Rt_construct_labelsRRZ(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRƒ s     t FortranReturncBs0eZdZdgZied6ZeeƒZRS(s2 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' t return_value( RR R!R"RR4R$Rt_construct_return_value(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR†s  t FFunctioncBseZdZd„ZRS(iMcCsf|jj}|jd|jkr>td||jfƒ‚ndj|djt|j|j ƒƒƒS(Ntstandards%s requires Fortran %d or newers{0}({1})s, ( t __class__Rt _settingst_required_standardtNotImplementedErrortformattjoinR@t_printR3(RcRdR((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyt_fcode8s  (RR RR’(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR‰5st F95FunctioncBseZdZRS(i_(RR R(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR“@stisigncBseZdZdZRS(s/ Fortran sign intrinsic for integer arguments. i(RR R!tnargs(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR”DstdsigncBseZdZdZRS(s8 Fortran sign intrinsic for double precision arguments. i(RR R!R•(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR–IstcmplxcBseZdZdZRS(s& Fortran complex conversion function. i(RR R!R•(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR—NsRzcBseZdZdZRS(s Fortran kind function. i(RR R!R•(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyRzSstmergecBseZdZdZRS(s Fortran merge function i(RR R!R•(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR˜Xst_literalcBseZdZdZd„ZRS(cOs’dj|jƒ|jdƒ\}}|jdƒjdƒ}|d|djdƒ}}|dkrmdn|}|p|d|j||pdS( Ns%.{0}etet0t.iit+R(Rt _decimalstsplittstriptrstriptlstript_token(RcRdR3Rttmantissatsgnd_extex_sgntex_num((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR’as %N(RR R_R£RžR’(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR™]st literal_spcBseZdZdZdZRS(s' Fortran single precision real literal Rši (RR R!R£Rž(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR¨ist literal_dpcBseZdZdZdZRS(s' Fortran double precision real literal tdi(RR R!R£Rž(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR©ostsum_cBsCeZdddgZied6ed6ZeeƒZeeƒZRS(RvRrtmask( RR R"RR4R$Rt_construct_arrayt_construct_dim(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR«us tproduct_cBsCeZdddgZied6ed6ZeeƒZeeƒZRS(RvRrR¬( RR R"RR4R$RR­R®(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pyR¯|s N(KR!tsympy.codegen.astRRRRRRRRRtsympy.core.basicR tsympy.core.compatibilityR tsympy.core.containersR tsympy.core.exprR tsympy.core.functionR tsympy.core.numbersRRtsympy.core.sympifyRt sympy.logicRRtsympy.utilities.iterablesRRRRRRRRR&R-R.R8R=RGRJRURXR[tassumed_extentRft assumed_sizeR_RvRwRxRyR{R|R~RR‚RƒR†R‰R“R”R–R—RzR˜R™R¨R©R«R¯(((s3lib/python2.7/site-packages/sympy/codegen/fnodes.pytsj@       %!  #  .