ó ¡¼™\c@sëddlmZddlZddlmZmZmZmZddlm Z ddl Z de fd„ƒYZ d„ZeƒZegZd „Zed „Zd efd „ƒYZd „Zdefd„ƒYZd„Zd„ZdS(iÿÿÿÿ(twarnNi(torderingt ambiguitiestsuper_signaturetAmbiguityWarning(t expand_tuplestMDNotImplementedErrorcBseZdZRS(s- A NotImplementedError for multiple dispatch (t__name__t __module__t__doc__(((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyRscCstt|j|ƒtƒdS(sC Raise warning when ambiguity is detected Parameters ---------- dispatcher : Dispatcher The dispatcher on which the ambiguity was detected ambiguities : set Set of type signature pairs that are ambiguous within this dispatcher See Also: Dispatcher.add warning_text N(Rt warning_texttnameR(t dispatcherR((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pytambiguity_warn scCsttd>> from sympy.multipledispatch import dispatch >>> @dispatch(int) ... def f(x): ... return x + 1 >>> @dispatch(float) ... def f(x): ... return x - 1 >>> f(3) 4 >>> f(3.0) 2.0 RR tfuncsRt_cachetdoccCs>||_|_tƒ|_tƒ|_g|_||_dS(N(R RtdictRRRR(tselfR R((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyt__init__Ds    cs‡‡‡fd†}|S(s# Register dispatcher with new implementation >>> from sympy.multipledispatch.dispatcher import Dispatcher >>> f = Dispatcher('f') >>> @f.register(int) ... def inc(x): ... return x + 1 >>> @f.register(float) ... def dec(x): ... return x - 1 >>> @f.register(list) ... @f.register(tuple) ... def reverse(x): ... return x[::-1] >>> f(1) 2 >>> f(1.0) 0.0 >>> f([1, 2, 3]) [3, 2, 1] csˆjˆ|ˆ|S(N(tadd(tfunc(tkwargsRttypes(s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyt_fs((RR!R R"((R RR!s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pytregisterKscCs/ttdƒr+tj|ƒ}|jjƒSdS(Nt signature(thasattrtinspectR$t parameterstvalues(tclsRtsig((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pytget_func_paramskscsq|j|ƒ}|rmtj‰‡fd†|Dƒ}td„|Dƒƒ}t‡fd†|Dƒƒrm|SndS(s; Get annotations of function positional parameters c3s0|]&}|jˆjˆjfkr|VqdS(N(tkindtPOSITIONAL_ONLYtPOSITIONAL_OR_KEYWORD(t.0tparam(t Parameter(s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pys yscss|]}|jVqdS(N(t annotation(R/R0((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pys sc3s|]}|ˆjk VqdS(N(tempty(R/tann(R1(s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pys ‚sN(R+R&R1ttupletall(R)Rtparamst annotations((R1s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pytget_func_annotationsqs  cCsñ|s'|j|ƒ}|r'|}q'ntd„|Dƒƒrkx't|ƒD]}|j|||ƒqJWdSxU|D]M}t|tƒsrdjd„|Dƒƒ}td|||jfƒ‚qrqrW||j |<|j d|ƒ|j j ƒdS(s† Add new types/method pair to dispatcher >>> from sympy.multipledispatch import Dispatcher >>> D = Dispatcher('add') >>> D.add((int, int), lambda x, y: x + y) >>> D.add((float, float), lambda x, y: x + y) >>> D(1, 2) 3 >>> D(1, 2.0) Traceback (most recent call last): ... NotImplementedError: Could not find signature for add: When ``add`` detects a warning it calls the ``on_ambiguity`` callback with a dispatcher/itself, and a set of ambiguous type signature pairs as inputs. See ``ambiguity_warn`` for an example. css|]}t|tƒVqdS(N(t isinstanceR5(R/ttyp((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pys ŸsNs, css3|])}t|tƒr!|jn t|ƒVqdS(N(R:ttypeRtstr(R/tc((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pys ¦ssDTried to dispatch on non-type: %s In signature: <%s> In function: %sR( R9tanyRRR:R<tjoint TypeErrorR RRRtclear(RR$RRR8ttypsR;tstr_sig((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyR…s"     cCsUtdrDt|jƒ|_t|jƒ}|rQ|||ƒqQn tj|ƒdS(Ni(RRRRRR(RRtamb((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyR±s  cOs%tg|D]}t|ƒ^q ƒ}y|j|}WnUtk r|j|Œ}|s}td|jt|ƒfƒ‚n||j|sFMatching functions for %s: <%s> found, but none completed successfully( R5R<RtKeyErrortdispatchtNotImplementedErrorR t str_signatureRt dispatch_itertnext(RtargsR targR!RR((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyt__call__ºs,%     cCs d|jS(Ns(R (R((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyt__str__ÔscGsJ||jkr|j|Syt|j|ŒƒSWntk rEdSXdS(s` Deterimine appropriate implementation for this type signature This method is internal. Users should call this object as a function. Implementation resolution occurs within the ``__call__`` method. >>> from sympy.multipledispatch import dispatch >>> @dispatch(int) ... def inc(x): ... return x + 1 >>> implementation = inc.dispatch(int) >>> implementation(3) 4 >>> print(inc.dispatch(float)) None See Also: ``sympy.multipledispatch.conflict`` - module to determine resolution order N(RRKRJt StopIterationtNone(RR!((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyRGØs   cgsct|ƒ}xP|jD]E}t|ƒ|krttt||ƒƒr|j|}|VqqWdS(N(tlenRR6tmapt issubclassR(RR!tnR$tresult((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyRJös  * cCstdtƒ|j|ŒS(s“ Deterimine appropriate implementation for this type signature .. deprecated:: 0.4.4 Use ``dispatch(*types)`` instead s-resolve() is deprecated, use dispatch(*types)(RtDeprecationWarningRG(RR!((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pytresolveýscCsi|jd6|jd6S(NR R(R R(R((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyt __getstate__s cCs<|d|_|d|_t|jƒ|_tƒ|_dS(NR R(R RRRR(Rtd((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyt __setstate__ s  cCs÷d|jg}|jr,|j|jƒng}x’|jddd…D]z}|j|}|jr°dt|ƒ}|dt|ƒd7}||jjƒ7}|j|ƒqI|jt|ƒƒqIW|rê|jddj |ƒƒndj |ƒS( NsMultiply dispatched method: %siÿÿÿÿs Inputs: <%s> t-s sOther signatures: s s ( R RtappendRRR RIRRtstripR@(RtdocstotherR*Rts((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyR s   cGs|jtt|ƒŒjS(N(RGRSR<R (RRL((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyt_help)scOs|j|ŒGHdS(s: Print docstring for the function corresponding to inputs N(Rb(RRLR ((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pythelp,scGs7|jtt|ƒŒ}|s-tdƒ‚nt|ƒS(NsNo function found(RGRSR<RAtsource(RRLR((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyt_source0scOs|j|ŒGHdS(s< Print source code for the function corresponding to inputs N(Re(RRLR ((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyRd6s(RR RRRRN(RRR t __slots__RQRR#t classmethodR+R9R RRRNROt__repr__RGRJRXRYR[tpropertyRbRcReRd(((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyR,s*   ,         cCs*dtj|ƒ}|tj|ƒ}|S(Ns File: %s (R&t getsourcefilet getsource(RRa((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyRd;stMethodDispatchercBs/eZdZed„ƒZd„Zd„ZRS(sP Dispatch methods based on type signature See Also: Dispatcher cCs>ttdƒr:tj|ƒ}tj|jjƒddƒSdS(NR$i(R%R&R$titltisliceR'R(RQ(R)RR*((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyR+HscCs||_||_|S(N(tobjR)(Rtinstancetowner((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyt__get__Ns  cOsotg|D]}t|ƒ^q ƒ}|j|Œ}|s\td|jt|ƒfƒ‚n||j||ŽS(Ns%Could not find signature for %s: <%s>(R5R<RGRHR RIRo(RRLR RMR!R((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyRNSs %(RRR RgR+RrRN(((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyRlAs cCsdjd„|DƒƒS(s§ String representation of type signature >>> from sympy.multipledispatch.dispatcher import str_signature >>> str_signature((int, float)) 'int, float' s, css|]}|jVqdS(N(R(R/R)((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pys cs(R@(R*((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyRI\scCs–d|}|d7}x3|D]+}|ddjd„|Dƒƒd7}qW|d7}|djg|D]$}d tt|ƒƒd |^qdƒ7}|S( s! The text for ambiguity warnings s. Ambiguities exist in dispatched function %s s;The following signatures may result in ambiguous behavior: s s, css#|]}dt|ƒdVqdS(t[t]N(RI(R/Ra((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pys lss s, Consider making the following additions: s s @dispatch(s ) def %s(...)(R@RIR(R REttexttpairRa((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyR fs   &  2(twarningsRR&tconflictRRRRtutilsRt itertoolsRmRHRR tsetRRRRRtobjectRRdRlRIR (((s@lib/python2.7/site-packages/sympy/multipledispatch/dispatcher.pyts  "      ÿ