Ed/ddlmZddlmZddlZddlmZmZm Z m Z ddl m Z ddl ZGddeZd ZGd d Zd ZeZd gZdZefdZGddZdZGddeZdZdZdS))Set)warnN)ordering ambiguitiessuper_signatureAmbiguityWarning) expand_tuplesceZdZdZdS)MDNotImplementedErrorz- A NotImplementedError for multiple dispatch N)__name__ __module__ __qualname____doc__Alib/python3.11/site-packages/sympy/multipledispatch/dispatcher.pyr r s7777rr cVtt|j|tdS)aC 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)r warning_textnamer ) dispatcherrs rambiguity_warnrs' jo{ 3 35EFFFFFrceZdZdZdZdZdS)RaiseNotImplementedErrorz*Raise ``NotImplementedError`` when called.c||_dSN)r)selfrs r__init__z!RaiseNotImplementedError.__init__$s $rctd|D}td|jjdt |d)Nc34K|]}t|VdSrtype).0as r z4RaiseNotImplementedError.__call__..(s(,,!d1gg,,,,,,rzAmbiguous signature for : <>)tupleNotImplementedErrorrr str_signature)rargskwargstypess r__call__z!RaiseNotImplementedError.__call__'sV,,t,,,,,!! O -"6"6"6"6    rN)r rrrrr.rrrrr!s844%%%     rrc|D]i}tt|}tt|dkr?||t |t jdS)a If super signature for ambiguous types is duplicate types, ignore it. Else, register instance of ``RaiseNotImplementedError`` for ambiguous types. 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 ambiguity_warn r on_ambiguityN)r(rlensetaddr#ambiguity_register_error_ignore_dup)rramb signatures rr5r5.s   /#..// s9~~  ! #   / ;;<       rTcdtd<dS)NFr)_resolverrr halt_orderingr:NsHQKKKrcdtd<tr8t}||t6dSdS)NTrr0)r9_unresolved_dispatcherspopreorder)r1rs rrestart_orderingr?RsYHQK !6,0022  555 "66666rceZdZdZdZddZdZedZedZ e fdZ e fd Z d Z d ZeZd Zd ZdZdZdZedZdZdZdZdZdS) Dispatcheraj Dispatch methods based on type signature Use ``dispatch`` to add implementations Examples -------- >>> from sympy.multipledispatch import dispatch >>> @dispatch(int) ... def f(x): ... return x + 1 >>> @dispatch(float) ... def f(x): # noqa: F811 ... return x - 1 >>> f(3) 4 >>> f(3.0) 2.0 )r rfuncsr_cachedocNcZ|x|_|_i|_i|_g|_||_dSr)rr rBrCrrD)rrrDs rrzDispatcher.__init__qs0$(( DM   rcfd}|S)a# 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] c&j|fi|Sr)r4)funcr,rr-s r_zDispatcher.register.._s$ DHUD + +F + + +Krr)rr-r,rIs``` rregisterzDispatcher.registerxs/6       rcttdr-tj|}|jSdS)Nr7)hasattrinspectr7 parametersvaluesclsrHsigs rget_func_paramszDispatcher.get_func_paramssB 7K ( ( +#D))C>((** * + +rc||}|rPtjfd|D}td|D}t fd|Ds|SdSdS)z; Get annotations of function positional parameters c3FK|]}|jjjfv|VdSr)kindPOSITIONAL_ONLYPOSITIONAL_OR_KEYWORD)r#param Parameters rr%z2Dispatcher.get_func_annotations..sS99 0 6889e999999rc3$K|] }|jV dSr) annotation)r#rYs rr%z2Dispatcher.get_func_annotations..s9 % %  % % % % % %rc3*K|] }|juVdSr)empty)r#annrZs rr%z2Dispatcher.get_func_annotations..s*EE#sio-EEEEEErN)rSrMrZr(any)rQrHparams annotationsrZs @rget_func_annotationszDispatcher.get_func_annotationss$$T**  #)I9999999F   % %# % % %%%KEEEEEEEEE #"" # # # #rc |s||}|r|}td|Dr+t|D]}||||dS|D]S}t |t s>> 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. c3@K|]}t|tVdSr) isinstancer()r#typs rr%z!Dispatcher.add..s,;;#z#u%%;;;;;;rN, c3lK|]/}t|tr|jnt|V0dSr)rfr"r str)r#cs rr%z!Dispatcher.add..s\$D$D453=Q2E2E%0AJJ),Q$D$D$D$D$D$DrzTried to dispatch on non-type: z In signature: In function: r0) rcr`r r4rfr"join TypeErrorrrBr>rCclear)rr7rHr1rbtypsrgstr_sigs rr4zDispatcher.addsH( (33D99K ('  ;;;;; ; ; %i00 3 3t\2222 F ; ;Cc4(( ;))$D$D9B$D$D$DDDi"%gggtyy!:;;; ;!% 9 , /// rctdr?t|j|_t|j}|r|||dSdSt|dS)Nr)r9rrBrr<r4)rr1r6s rr>zDispatcher.reorderst A; .$TZ00DMdj))C ( T3''''' ( ( $ ' ' - - - - -rc td|D} |j|}nN#t$rA|j|}|s(t d|jdt |d||j|<YnwxYw ||i|S#t$rc|j|}t||D]} ||i|ccYS#t$rYwxYwt d|jdt |dwxYw)Nc,g|]}t|Srr!r#args r z'Dispatcher.__call__..111StCyy111rCould not find signature for r&r'zMatching functions for z(> found, but none completed successfully) r(rCKeyErrordispatchr)rr*r dispatch_iternext)rr+r,r-rHrBs rr.zDispatcher.__call__s11D11122 &;u%DD & & & 4=%(D 7))YYY e 4 4 4 46777"&DK     & K4((( ($ K K K&D&.E KKK  400000000,D%%)-M%4H4H4H4H'JKK K Ks@ )AA43A48B'C-(B4/C-4 C>C-C,C-cd|jzS)Nz)rrs r__str__zDispatcher.__str__s 49,,rc||jvr |j|S t|j|S#t$rYdSwxYw)a` 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)rBr|r{ StopIterationrr-s rrzzDispatcher.dispatchs`, DJ  %:e$ $ **E233 3   44 s / ==c'Kt|}|jD]I}t||kr4ttt||r|j|}|VJdSr)r2rallmap issubclassrB)rr-nr7results rr{zDispatcher.dispatch_iter#sp JJ  I9~~" s3z5)+L+L'M'M I.   rc@tdt|j|S)z Deterimine appropriate implementation for this type signature .. deprecated:: 0.4.4 Use ``dispatch(*types)`` instead z-resolve() is deprecated, use dispatch(*types))rDeprecationWarningrzrs rresolvezDispatcher.resolve*s- <  ! ! !t}e$$rc |j|jdS)NrrBrr~s r __getstate__zDispatcher.__getstate__5s %% %rcz|d|_|d|_t|j|_i|_dS)NrrB)rrBrrC)rds r __setstate__zDispatcher.__setstate__9s4fI wZ  ,,  rc,d|jzg}|jr||jg}|jdddD]}|j|}|jr\dt |z}|dt|zdzz }||jz }||r|t ||r+|dd |zd |S) NzMultiply dispatched method: %sz Inputs: <%s> - zOther signatures: z  ) rrDappendrrBrr*r2striprl)rdocsotherrRrHss rrzDispatcher.__doc__?s049<= 8 " KK ! ! !=2& 1 1C:c?D| 1$}S'9'99S3q66\D((T\''))) A ]3//0000  J KK1HMM%4H4HH I I I{{4   rcF|jtt|jSr)rzrr"r)rr+s r_helpzDispatcher._helpVst}c$oo.66rc4t|j|dS)z: Print docstring for the function corresponding to inputs N)printrrr+r,s rhelpzDispatcher.helpYs jdj$     rc||jtt|}|stdt |S)NzNo function found)rzrr"rmsource)rr+rHs r_sourcezDispatcher._source]s;t}c$oo. 1/00 0d||rc4t|j|dS)z< Print source code for the function corresponding to inputs N)rrrs rrzDispatcher.sourcecs ldlD!"""""rr)r rrr __slots__rrJ classmethodrSrcrr4r>r.r__repr__rzr{rrrpropertyrrrrrrrrArAYs{*II@++[+ ##[#&1?****X$2....KKK4---H< % % %%%% !!X!,777!!! #####rrAcbdtj|z}|tj|z}|S)Nz File: %s )rM getsourcefile getsource)rHrs rrrhs1.t444A G d # ##A Hrc4eZdZdZedZdZdZdS)MethodDispatcherzP Dispatch methods based on type signature See Also: Dispatcher cttdrAtj|}tj|jddSdS)Nr7r)rLrMr7itlislicerNrOrPs rrSz MethodDispatcher.get_func_paramsusS 7K ( ( @#D))C:cn3355q$?? ? @ @rc"||_||_|Sr)objrQ)rinstanceowners r__get__zMethodDispatcher.__get__{s rctd|D}|j|}|s(td|jdt |d||jg|Ri|S)Nc,g|]}t|Srr!rts rrvz-MethodDispatcher.__call__..rwrrxr&r')r(rzr)rr*r)rr+r,r-rHs rr.zMethodDispatcher.__call__s11D11122t}e$ I%%'+yyy-2F2F2F2F'HII ItDH.t...v...rN)r rrrrrSrr.rrrrrnsZ @@[@  /////rrc@dd|DS)z String representation of type signature >>> from sympy.multipledispatch.dispatcher import str_signature >>> str_signature((int, float)) 'int, float' rhc3$K|] }|jV dSr)r )r#rQs rr%z str_signature..s$11cS\111111rrl)rRs rr*r*s% 9911S111 1 11rcdz}|dz }|D]*}|ddd|Dzdzz }+|dz }|dfd |Dz }|S) z! The text for ambiguity warnings z. Ambiguities exist in dispatched function %s z;The following signatures may result in ambiguous behavior:  rhc3@K|]}dt|zdzVdS)[]N)r*)r#rs rr%zwarning_text..s4AAqcM!,,,s2AAAAAArrz, Consider making the following additions: rcZg|]'}dtt|zdzz(S)z @dispatch(z ) def %s(...))r*r)r#rrs rrvz warning_text..sQAAA78& oa6H6H(I(II+d23AAArr)rr6textpairs` rrrs >$ GD JJDII  IIAADAAA A ABDHI I >>DFKKAAAArsNNNNNNNNNNNN 88888/888 GGG"            8#%% 6#16666L#L#L#L#L#L#L#L#^   /////z///6222     r