EY\ D d Z dgZddlZddlZddlmZ ddlmZ G d de Z G d d e Z G d
d e Z
G d d
e Z G d de Z
G d de ZefZeefZd Z G d de Zd Z G d d Zd Zd Zd Zd;dZd;d Z d;d Z dz{Fortran/C symbolic expressions
References:
- J3/21-007: Draft Fortran 202x. https://j3-fortran.org/doc/year/21/21-007.pdf
Expr N)Enum)gcdc e Zd ZdZdZdZdZdS )Languagez2
Used as Expr.tostring language argument.
r N)__name__
__module__ __qualname____doc__PythonFortranC g/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/f2py/symbolic.pyr r s) FG AAAr r c N e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d Zd
ZdZdZdZdS )Opz$
Used as Expr op attribute.
( d i, i i i i N)r
r r r
INTEGERREALCOMPLEXSTRINGARRAYSYMBOLTERNARYAPPLYINDEXINGCONCAT
RELATIONALTERMSFACTORSREFDEREFr r r r r & se G
DG
F E
FG EH
FJ EG
C EEEr r c b e Zd ZdZdZdZdZdZdZdZ e
e j fd Z
e j fd Zd
S ) RelOpzH
Used in Op.RELATIONAL expression to specify the function part.
r r c n |t j u r]t j t j t j t j t j t j d| S t j t j t j t j t j t j d| S )N).eq..ne..lt..le..gt..ge.)==!=<<=>>=)
r r r0 EQNELTLEGTGElower)clsslanguages r
fromstringzRelOp.fromstringF s x' ' ' !Heh !Heh !Heh8 889 C
C hehUXhUXUX? ??@B Br c ^ |t j u rPt j dt j dt j dt j dt j dt j di| S t j dt j dt j d t j d
t j d t j d i| S )
Nr6 r7 r8 r9 r: r; r< r= r> r? r@ rA ) r r r0 rB rC rD rE rF rG selfrK s r tostringzRelOp.tostringO s x' ' 'HfehHfehHfeh88<>
> $$#ux#ux//35 5r N)r
r r r
rB rC rD rE rF rG classmethodr r rL rP r r r r0 r0 ; s
B
B
B
B
B
B$,J B B B [B !)
5 5 5 5 5 5r r0 c . e Zd ZdZdZdZdZdZdZdZ dZ
d S )
ArithOpzC
Used in Op.APPLY expression to specify the function part.
r r r1 r2 r3 r4 N) r
r r r
POSNEGADDSUBMULDIVPOWr r r rS rS Y s= C
C
C
C
C
C
CCCr rS c e Zd ZdS )OpErrorNr
r r r r r r] r] f Dr r] c F e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d Zd
ZdZdS )
Precedencez4
Used as Expr.tostring precedence argument.
r r r r1 r2 r4 rT r
r r N)r
r r r
ATOMPOWERUNARYPRODUCTSUMrD rB LANDLORr& ASSIGNTUPLENONEr r r ra ra j s[
D
E
EG
C
B
B
D
CG
F E DDDr ra c d | | }||| |< d S ||z }|r|| |< d S | |= d S N)get)dkvcs r
_pairs_addrv sL aAy!
E AaDDD!r c e Zd ZdS ) ExprWarningNr^ r r r rx rx r_ r rx c > t j | t d d S )Nr )
stacklevel)warningswarnrx )messages r ewarnr~ s M';1555555r c
e Zd ZdZeej fd Zd Zd Z d Z
d Z d Z d Z
d Zd
Zd Zej ej fd Zd
Zd Zd Zd Zd Zd Zd Zd Z d Z d Z d Z d Z d Z!d Z"d Z#d Z$d Z%d Z&d Z'd Z(d! Z)d"S )#r zfRepresents a Fortran expression as a op-data pair.
Expr instances are hashable and sortable.
c $ t | | S )z.Parse a Fortran expression to a Expr.
rK )rL )rJ rK s r parsez
Expr.parse s !h////r c t |t sJ |t j u rut |t rt | dk sJ t |d t
sJ t |d t
t f s
J | n|t j u rut |t rt | dk sJ t |d t sJ t |d t
t f s
J | n!|t j u r,t |t rt | dk sJ n|t j
u rt |t rt | dk sJ t |d t r)|d d d t |d dz
dv sJ t |d t
t f s
J | n;|t j u rt | J n|t j
t j fv rz Expr.__init__.. s, ??$z$--??????r r1 z$unknown op or missing sanity check: ) r r r tuplelenintstrr! floatr" r# r% hashr$ r) allr+ r, dictr' r( r& r- r. r r* NotImplementedErroropdata)rO r r s r __init__z
Expr.__init__ s "b!!!!! dE** =s4yyA~~~= d1gs++ + ++ d1gSz22 8 8D 8 82 8
27]] dE** =s4yyA~~~= d1gu-- - -- d1gSz22 8 8D 8 82 8
2:
dE** =s4yyA~~~=~
29__ dE** =s4yyA~~~= tAw,,
I Q 0 0#d1g,,q. 015GGGGH d1gSz22 8 8D 8 82 8
29__::))))
BHbi(
(
( dE** * **??$????? E E E E? E
BHbj)
)
) dD)) ) )) )
28^^ dE** =s4yyA~~~=Q==,,, d1gu-- - -- d1gt,, , ,, ,
2;
dE** =s4yyA~~~=Q==,,,,
2:
dE** =s4yyA~~~=~
BFBH%
%
% dD)) ) )) )
2=
dE** =s4yyA~~~=~%;r;;= =
= r c h t |t o | j |j u o| j |j k S rp )r r r r rO others r __eq__z Expr.__eq__ s6 5$'' ,Gux',I+ -r c | j t j t j fv r4t t | j }nd| j t j u rJ| j d d t t | j d z }n| j }t | j |f S Nr )
r r r+ r, r sortedr itemsr' r )rO r s r __hash__z
Expr.__hash__ s 7rx, , , 1 12233DD
W
9RaR=5 ! 0B0B0D0D)E)E#F#FFDD9DTWdO$$$r c H t |t r| j |j ur| j j |j j k S | j t j t j fv rht t | j
t t |j
k S | j t j u r| j d d |j d d k r | j d d |j d d k S t t | j d
t t |j d
k S | j |j k S t S r )
r r r valuer r+ r, r r r r r' NotImplementedr s r __lt__z Expr.__lt__ sP eT " " *weh&&w}ux~55w28RZ000 fTY__%6%6 7 788 uz'7'7'9'9 : :;;< =w"(""9RaR=EJrrN22 9RaR=5:bqb>99 VDIaL$6$6$8$8 9 9::U5:a=..0011>3 >3 3 39uz) ) r c | |k p| |k S rp r r s r __le__z Expr.__le__ s DEM$ATE\ Ar c | |k S rp r r s r __gt__z Expr.__gt__ s $7 7r c | |k S rp r r s r __ge__z Expr.__ge__ s $6 6r c P t | j d| j d| j dS N(, ))typer
r r rO s r __repr__z
Expr.__repr__ s. t**%AAAA49AAAAr c * | S rp )rP r s r __str__z Expr.__str__
s }} r c l | j t j t j fv ri| j d dk r t
j n t
j }t | j d | j d dk rd| j d ndz }n| j t j u r. J 1 1 " --
(88-LL 1 1 1 1 1 1r r r c 3 Z K | ]%}| t j V &dS r r r s r r z Expr.tostring..$ r r []z - + r * -0r r izpow(z ** 1/c R g | ]#}| t j $S r rP ra rh r argrK s r
z!Expr.tostring..d sF 1 1 1$' !$ Z-?6> !- !@ !@ 1 1 1r z / c R g | ]#}| t j $S r r r s r r z!Expr.tostring..j s< ) ) ) Z%5 II ) ) )r c ^ g | ]*\ }}|d z | t j z +S )=)rP ra rn r rs rt s r r z!Expr.tostring..l sC 5 5 5 !Q S1::jo#>#>> 5 5 5r c R g | ]#}| t j $S r r r s r r z!Expr.tostring..r s< . . . LL!1HLEE . . .r Nc R g | ]#}| t j $S r r r s r r z!Expr.tostring..w s< * * * LL!3hLGG * * *r z // c R g | ]#}| t j $S r r )r arK s r r z!Expr.tostring..| sF "6 "6 "6&' #$**Z-=6> #- #@ #@ "6 "6 "6r ?:z if z else zmerge(z
tostring for z and &* ztostring for op )-r r r r! r ra ri re r r" joinr% r# r$ r+ r r rP as_numberrh appendr, r r rangerm rf r' rS rZ r( r) r& r r r r- rg r. r* r0 rB rC rD r )rO parent_precedencerK
precedencertermstermcoeffr factorstailbaseexpfactornameargskwargsnumerdenomcondexpr1expr2ropleftrights ` r rP z
Expr.tostring
s 7rz27+ + +,0IaL1,<,<*.. )
DIaL!!(, ! (9(9 &81 %7%7%7?ACAA
W
"
" 1 1 1 1&*i1 1 1 1 1Aa#
A# JJ
W
!
!# JDIAA
W
!
! ! Ay|q IaL3&*# JJ
W
1 1 1 1&*i1 1 1 1 1Aa#
A# JJ
W
E %dioo&7&7 8 8
#
# e199 B "FEE BA:: == (=KKDDy||++ "5zz"'}}}t}} & . 0= 0C 0C C &LL$$$$5[[LL%%% T""""%#A+0E joJJ
W
"
"GD #DIOO$5$5 6 6
'
' c!88 !]]:+=4< + > >FF ++eArll**!% z/A8@ "/ "B "B!&VHsN!;!; c1
--!% z/A8@ "/ "B "B C4/ !% z/?8@ "/ "B "B!8!8!8#!8!8!8 !]]:+;4< + > >@Ls MF 'NN2&&&v&&&&
= %u$GCejj&6&6<< 'CA/6K++JOJJ
W
!% D$w{""x8:'='= 1 1 1 1+/ 1 1 1 u (((( ' /
) ) ) )#') ) ) 5 5%+\\^^5 5 5 500diioo000 '_
W
#
#9Q > ># JJ
W
di(()9H(MMMA#)JJ
W
di(()9H(MMMA#)JJ
W
%
% #y Cu+.58UX2F+F+F*-- )}
==h=??DNN:NAAE,,,11C''#''''AA%&B&B&BCC C "Z%5 5 5 7S= r c | S rp r r s r __pos__z Expr.__pos__ s r c | dz S )Nr r s r __neg__z Expr.__neg__ s byr c ( t | }t |t rg| j |j u r'| j t j t j fv rMt | j d |j d z t | j d |j d S | j t j
u r*| j \ }}|j \ }}t ||z ||z S | j t j u rkt | j t | j }|j D ]\ }}t |j || t! | S | j t j
u r1|j t j t j fv r| t | z S | j t j t j fv r%|j t j
u rt | |z S | j t j u r2|j t j u r | t# || j d z S | j t j u r2|j t j u r t# | |j d |z S t% | t% | z S t& S Nr r )kind)as_exprr r r r r r! r r maxr"
as_complexr+ r r rv normalizeas_realas_termsr ) rO r r1i1r2i2r rs rt s r __add__z Expr.__add__ s/ eT " " 4w%(""7rz27 333$ ! uz!}4DIaL%*Q-88 : : : 7bj(( !YFB "ZFB%b2grBw7777bh&& TWd49oo66A %
0 0 2 2 1 11"161a0000$Q<<'w"*$$bj"'5J)J)Jj / ///RZ 111eh"*6L6L!$''%//BG##BJ(>(>ge$)A, ? ? ???BJ&&58rw+>+> t%*Q-8885@@D>>HUOO3 3 r c ^ t |t rt | | z S t S rp r number_typesr r r s r __radd__z
Expr.__radd__ - e\ * * + U##d* * r c | | z S rp r r s r __sub__z Expr.__sub__ s v r c ^ t |t rt | | z
S t S rp r r s r __rsub__z
Expr.__rsub__ r r c t | }t |t rN| j |j u r| j t j t j fv rMt | j d |j d z t | j d |j d S | j t j
u r6| j \ }}|j \ }}t ||z ||z z
||z ||z z S | j t j u rkt | j t | j }|j D ]\ }}t |j || t! | S | j t j u r~t | j i }| j D ]@\ } }
|j D ]!\ } } t |j | | z |
| z "At! | S | j t j
u r1|j t j t j fv r| t | z S |j t j
u r1| j t j t j fv rt | |z S | j t j u r2|j t j u r | t% || j d z S | j t j u r2|j t j u r t% | |j d |z S | j t j u r| t' | z S |j t j u rt' | |z S t) | t) | z S t* S r )r r r r r r r! r r r r" r r, r r rv r r+ r r
as_factorsr )
rO r r r r r r rs rt t1c1t2c2s
r __mul__z Expr.__mul__ s eT " " $ 8w%(""7rz27 333$TYq\EJqM%A%(1 uz!}%E%E G G GW
** !YFB "ZFB%b2gR&7b279JKKK7bj(( TWd49oo66A %
0 0 2 2 1 11"161a0000$Q<<'W(( TWb))A"&)//"3"3 A AB&+j&6&6&8&8 A AFB &qvrBwR @ @ @ @A$Q<<'w"*$$bj"'5J)J)Jj / ///RZ''DG
BG7L,L,L!$''%//BG##BJ(>(>ge$)A, ? ? ???BJ&&58rw+>+> t%*Q-8885@@w"(""huoo--RX%% ~~-- d##j&7&77 7 r c ^ t |t rt | | z S t S rp r r s r __rmul__z
Expr.__rmul__ r r c F t | }t |t r|j t j u r|j d }|dk rt d S |dk r| S |dk rn| j t j u rPt | j i }| j D ]\ }}||z |j |< t | S | | |dz
z z S |dk r | | z dz S t t j | |i S t t j
| | S t S )Nr r r )r r r r r r r r r, r r as_applyrS r[ r )rO r exponentr rs rt s r __pow__z Expr.__pow__ s! eT " " 6x2:%% :a=q==$Q<<'q== Ka<<w"*,, " - -$(IOO$5$5 5 5DAq()H AF1II (||+ 4HqL#9:: ^^ hY /B66BJx(8999GKu55 5 r c t | }t |t r(t t t
j | | S t S rp )r r r r r rS rZ r r s r __truediv__zExpr.__truediv__ sD eT " " A Xgk4 ? ?@@ @ r c b t | }t |t r|| z S t S rp r r r r r s r __rtruediv__zExpr.__rtruediv__ s/ eT " " 4< r c t | }t |t r)t t t j | |f S t S rp )r r r r r r) r r s r __floordiv__zExpr.__floordiv__! sE eT " " = T")dE] ; ;<< < r c b t | }t |t r|| z S t S rp r r s r
__rfloordiv__zExpr.__rfloordiv__) s/ eT " " !D= r c t | gt t | R i t d | D S )Nc 3 > K | ]\ }}|t | fV d S rp r r s r r z Expr.__call__..6 s0 J J41a71:: J J J J J Jr )r mapr r r )rO r r s r __call__z
Expr.__call__/ sa Ls7D 1 1 L L L J J6<<>> J J JJJL L Lr c t | }t |t s|f}t | dk rt d| d t t j | f|z S )Nr z/C-index should be a single expression but got ``)r r r r r~ r r r( )rO indexs r __getitem__zExpr.__getitem__8 si %'' FE u::>> LELLL M M MBK$5 111r c | j t j u rs | }|| S t j d| j }|r<| \ }}|dv rt | S |dk s J ||f |S | j t j t j
t j fv r| S | j t j t j
fv r3t | j t fd| j D S | j t j u r@t# t | j t fd| j D S | j t j u ryd}| j D ];\ }}|| |z } || |z z
}<| t+ d t- d S |S | j t j u ryd}| j D ];\ } }
|| |
z } || |
z z }<| t+ d t- d
S |S | j t j u r| j \ } } }
t3 | t r| } t fd | D } t5 fd |
D }
t# t | j | | |
f S | j t j u r| j d }t3 |t r| }t fd
| j d
d D } t# t | j |f| z S | j t j u rBt fd| j D }t# t | j | S | j t j t j fv r:t# t | j | j S | j t j u rZ| j \ }}}| }| }t# t | j |||f S tA d| j d| )zRecursively substitute symbols with values in symbols map.
Symbols map is a dictionary of symbol-expression pairs.
Nz$\A(@__f2py_PARENTHESIS_(\w+)_\d+@)\ZROUNDDIVSQUAREROUNDc 3 B K | ]}| V d S rp
substituter r symbols_maps r r z"Expr.substitute..V sC '> '>+/ (,{'C'C '> '> '> '> '> '>r c 3 B K | ]}| V d S rp r1 r3 s r r z"Expr.substitute..Y sK 1H 1H59 26 1M1M 1H 1H 1H 1H 1H 1Hr z?substitute: empty TERMS expression interpreted as int-literal 0r zAsubstitute: empty FACTORS expression interpreted as int-literal 1r c 3 B K | ]}| V d S rp r1 r r r4 s r r z"Expr.substitute..w s/ AAq k22AAAAAAr c 3 L K | ] \ }}|| fV d S rp r1 )r rs rt r4 s r r z"Expr.substitute..x sL 6 6 "a all; 7 78 6 6 6 6 6 6r c 3 B K | ]}| V d S rp r1 r7 s r r z"Expr.substitute.. s/ JJq k22JJJJJJr c 3 B K | ]}| V d S rp r1 r7 s r r z"Expr.substitute.. s/ J J1Q\\+ 6 6 J J J J J Jr zsubstitute method for z: )!r r r% rq rematchr groupsas_arrayr r! r# r$ r" r r r) r r+ r r2 r~ r r, r' r r r( r& r- r. r* r )rO r4 r mr parenr r r r r targetr r funcoperandsr r r s ` r r2 zExpr.substituteB s
7bi OOD))E} @$)LLA
8 xxzz u 222#E??* '''% '''L 7rz27BI6 6 6K 7rx, , , '> '> '> '>379'> '> '> "> "> ? ?
? 7bi T$'5 1H 1H 1H 1H=AY1H 1H 1H ,H ,H I I J J
J 7bh A #y 0 0
>
> e9 44u : :;; ; 7bj J J J J J J JJJH T$'8 4 455 5 7rvrx( ( ( T$'49+?+? +L+L M MNN N 7bm # # #y Cu??;//D$$[11E T$'Cu+= > >?? ?!"N47"N"Nd"N"NOOOr c
R | gR i }||S | j t j t j t j t j fv r| S | j t j t j t j t j fv rBt t | j t fd| j
D S | j t j t j fv ri }| j
D ]R\ }} |j gR i }t% |t r |j gR i n|}||v r || |z }|||< St t | j | S | j t j u r| j
d }t% |t r |j gR i n|} t fd| j
d D }
t) fd| j
d D } t t | j | |
| f S | j t j u r| j
d }t% |t r |j gR i n|}t fd| j
dd D } t t | j |f| z S | j t j t j fv r6t t | j | j
j gR i S | j t j u rR| j
\ }
}} |j gR i } |j gR i }t t | j |
||f S t3 d | j )
aB Traverse expression tree with visit function.
The visit function is applied to an expression with given args
and kwargs.
Traverse call returns an expression returned by visit when not
None, otherwise return a new normalized expression with
traverse-visit sub-expressions.
Nc 3 : K | ]} |j gR i V d S rp traverse)r r r r visits r r z Expr.traverse.. sS 1' 1'
e5d555f551' 1' 1' 1' 1' 1'r r c 3 : K | ]} |j gR i V d S rp rF )r operandr r rH s r r z Expr.traverse.. sT : :!( .W -e Ed E E Ef E E : : : : : :r r c 3 D K | ]\ }}| |j gR i fV d S rp rF )r rs rt r r rH s r r z Expr.traverse.. sd @ @"&!Q !*!*U"DT"D"D"DV"D"D E @ @ @ @ @ @r r c 3 : K | ]} |j gR i V d S rp rF )r r) r r rH s r r z Expr.traverse.. sS 8 8 % +EN5 B4 B B B6 B B 8 8 8 8 8 8r ztraverse method for )r r r r! r# r% r" r$ r) r& r r r r r+ r, r rG r r' r r( r- r. r* r )rO rH r r resultr rs rt objrB rC
kwoperandsindicesr r r s ``` r rG z
Expr.traverse ss t-d---f-- M 7rz27BIryA A AK
WRXry"*E
E
E T$'5 1' 1' 1' 1' 1' 1' I1' 1' 1' ,' ,' ( ( ) )
) W2:.
.
.D ))
1 AJu6t666v66"1d++3ZQZ777777712 99 Q! AQ T$'4 0 011 1
W
)A,C!#t,,6LCL888888825
: : : : : :,0IaL : : : : :H @ @ @ @ @ @*.)A,*<*<*>*> @ @ @ @ @J T$'D(J+G H HII I
W
#
#)A,C d++5<3<777777714
8 8 8 8 8 8)-122 8 8 8 8 8G T$'C6G+; < <== =
W*
*
* T$'"4$)"4U"LT"L"L"LV"L"L N N O O
O
W
%
% #y Cu 4=888888D"EN5:4:::6::E T$'Cu+= > >?? ?!"B"B"BCCCr c f g }|ffd }| | t | dk S )z&Check if self contains other.
c J |r| S | k r| d | S d S )Nr )r )exprfoundr s r rH z Expr.contains..visit s4
Q r r )rG r )rO r rT rH s ` r containsz
Expr.contains sO #
e 5zzQ r c V t }|fd}| | |S )z3Return a set of symbols contained in self.
c Z | j t j u r| | d S d S rp )r r r% add)rS rT s r rH zExpr.symbols..visit s, w")## $ $#r setrG rO rT rH s r symbolsz Expr.symbols s= #
e r c \ t }|ffd | |S )zFReturn a set of expressions used as atoms in polynomial self.
c L | j t j u r!| j D ]}| | S | j t j t j fv rd S | j t j u rct | j d t rC| j d t j
u r(| j d d | S d S | j t j t j fv r| S |
| | j t j t j fv r| S d S )Nr r )r r r, r rG r+ r" r' r rS r[ r r! rX r( )rS rT brH s r rH z$Expr.polynomial_atoms..visit s w"*$$ & &AJJu%%%% w28RZ000w"(""z$)A,'H'H"9Q<7;..IaLO,,U333 Kw2:rw/// IIdOOOw2;111 21r rY r[ s @r polynomial_atomszExpr.polynomial_atoms sH # (
e r c 6 | |t d i }| |z
}| |t d i }t ||z |z
\ }}|t d k r t d| d| d| d| d|
||fS )zReturn a, b such that a * symbol + b == self.
If self is not linear with respect to symbol, raise RuntimeError.
r r znot a z-linear equation: r r z == )r2 r as_numer_denom RuntimeError)rO symbolr_ axr zeror s r linear_solvezExpr.linear_solve s
OOVYq\\ 2 3 3
AX MM69Q<<0 1 1 Vb 11a 9Q<< B B B#$ B B)/ B B45 B B;? B B C C
C!t r N)*r
r r r
staticmethodr r r r r r r r r r r r ra rn r rP r r r r r r r r r r r r r! r& r* r2 rG rU r\ r` rg r r r r r s2
"* 0 0 0 \0
@ @ @D- - -
% % %
BAA777666B B B *4 "*P P P Pd 8
' ' ' R
. L L L2 2 2IP IP IPV6D 6D 6Dp : r c z t | t s| S | j t j u ri }| j D ]\ }}|dk r |j t j u r
|dk r||z }d}|j t j u r4|j D ]\ }}t ||||z st ||| t | dk rt d S t | dk r!| \ \ }}|dk r|S t t j | S | j t j u rVd}i }| j D ]1\ }}|dk r
|j t j u r(t |t r|dk r
|||dz
z z }d}|j t j
t j fv rC|dk r||j d z }~|dk r||j d |z z }t ||| |j t j u rc|dk rJt |t r5|j D ]\ } }
t || |
|z
t ||| t ||| 3t | dk s|dk r&t |t sJ t | S t | dk r`| \ \ }}|dk r|}nt t j | }|dk r|S t t j ||i S |dk rt t j | S t t j t t j | |i S | j t j u r| j d t" j u r| j d \ } } t' | \ }}t' | \ }
}t |t r0t |t rt) || }||z ||z }}n||z d}}|j t j u rc|j d t" j u rJ|j d d |z }|j d d |
z |z }t+ t" j || S |
j t j u rc|
j d t" j u rJ|
j d d |z |z }|
j d d |z }t+ t" j || S t- t/ | j }t/ |
j D ]\ }}t ||| i i }}| D ]\ }}|dk r|||< | ||< t1 t t j | |z }t1 t t j | |z }|j t j
t j fv r|j d dk r|S t+ t" j || S | j t j u rE| j d g}| j dd D ]}|d }|j t j u r|j t j u r|j d d dv r|j d d |j d d k rct7 |j d dd |j d dd z t9 |j d |j d }||d< | | t | dk r|d S t t j t= | S | j t j u r_tA t0 | j \ }}}|j t j
u r|j d r|n|S t t j |||f S | S )z7Normalize Expr and apply basic evaluation methods.
r r Nr z"')!r r r r r+ r r r" rv r r r,
integer_typesr r! r r' rS rZ
as_term_coeffr r r r
r r) r# as_stringr r r r& r% )rN rr tru r r r r_ eb1e1dividenddivisorr r gr r lstrJ lastnew_lastr r r s r r r s c4
v HNN$$
$
$DAqAvvtrz!!a1ffEtrx fllnn . .FB q"b1f----. 1a#### q66Q;; Q<<
VVq[[ggiiGFQAvvBHa
v HNN$$ $ $DAqAvvtrx Jq-$@$@ QUUq1u &t
BG,,,66QVAY&EEUUQVAY!^+EE q!Q''''##q55Z= 9 95"#&,,.. 2 2B"1b"q&11112 q!Q''''1a#### q66Q;;%1** e\22 2 22 U## #
VVq[[ggiiGFQAvvQ''zz1e*-- -
aZZ
A&& &4
A#6#6">?? ?
vchqkW[ 8 8 HQK' x((B w''B b- ( ( ZM-J-J B AUBEBBUAB
5BH w{!:!:GAJqMB&EGAJqMB&+EGK66 6
5BH w{!:!:GAJqMB&+EGAJqMB&EGK66 6 B $ % % rNN'--// ! !DAq q!aR 2uGGII DAq1uu a 2a$rz51122R7$rz51122R7 8
BG, , ,A !1C1CL UE222
vx{m!"" Ar7DGry(( )) ! Q500q ! ! R(888$TYq\#2#%6122 %F%(1 qvay%A%A C C"B
1
s88q==q6MBIuSzz***
v CH 5 5eU 7bj IaL355e 3BJue 4555 Jr c | t | t rt | j | j S t | t
rt
| S t | t r t t | S t | t r"t t t | S | S )z.Convert non-Expr objects to Expr objects.
)
r complexr realimagr r r rl reprr r% r rN s r r r s #w .#(CH---#|$$ ~~ #s $ c####u (S#&&''' Jr c 6 t t j | S )zJReturn object as SYMBOL expression (variable or unparsed expression).
)r r r% r| s r as_symbolr~ s 3 r r2 c X t | t r t t j | |f S t | t
r t t j | |f S t | t r!| j t j t j fv r| S t d| d )z/Return object as INTEGER or REAL constant.
cannot convert z to INTEGER or REAL constant) r r r r r r r! r r] rN r s r r r s #s -BJd ,,,#u *BGc4[)))#t 6bj"'* * *J
ECEEE
F
FFr c t | t r t t j | |f S t | t r| j t j u r| S t
d| d )z'Return object as INTEGER constant.
r z to INTEGER constant)r r r r r r r] r s r
as_integerr sl #s -BJd ,,,#t 6RZ J
=C===
>
>>r c t | t r)t t j t | |f S t | t
r t t j | |f S t | t r\| j t j u r| S | j t j u r4t t j t | j d |f S t d| d )z$Return object as REAL constant.
r r z to REAL constant)
r r r r r! r r r r r] r s r r r s #s 1BGeCjj$ /000#u *BGc4[)))#t = 6RW J
Vrz
!
!% "4"4d!;<< <
:C:::
;
;;r r c : t t j | |f S )zBReturn object as STRING expression (string literal constant).
)r r r# r s r rl rl s C; ' ''r c f t | t r| f} t t j | S )z8Return object as ARRAY expression (array constant).
)r r r r$ r| s r r> r> s/ #t d # r c n t t j t | t | f S )zDReturn object as COMPLEX expression (complex literal constant).
)r r r" r )ry rz s r r r s'
WT]]GDMM : ; ;;r c
t t j | t t t
| t
d | D f S )zIReturn object as APPLY expression (function call, constructor, etc.)
c 3 > K | ]\ }}|t | fV d S rp r$ r s r r zas_apply.. s0 AA$!Qq'!**oAAAAAAr )r r r' r r% r r r )rB r r s r r r s\ uS$ / /00AA&,,..AAAAAC D D Dr c < t t j | ||f S )z
>>r c " t | t rZt | } | j t j u r| S | j t j u rt | j dk rv| j \ \ }}|dk r t t j |di S t t j |dt
| di S | j t j u rf| j d t j
u rM| j d s@t t j | j d d d| j d d di S t t j | di S t dt | d )z-Return expression as FACTORS expression.
r r r r r r )r r r r r r, r+ r r r numberr' rS rZ r] r )rN r r s r r
r
+ sI #t *nn 6RZ J 6RX 38}}!!!$ !1!1
uA::
T1I666BJq$++e2D2Da(HIII
Vrx
! ++Xa[ ,
SXa[^Q A$KLL LBJa)))
=DII===
>
>>r c t | t rSt | } | j t j u r(t
d| j d | j d fS | j t j u r(t d| j d | j d fS | j t j
u r:t | j dk r"| j \ \ }}||fS | j t j
u rj| j d t j u rQt! | j d d \ }}t# t j || j d d |fS | dfS t% dt' | d )z0Return expression as term-coefficient pair.
r r r z to term and coeff)r r r r r r r r r! r r+ r r r' rS rZ rk r r] r )rN r r rm ru s r rk rk @ sL #t nn 6RZ a!--sx{: : 6RW 1chqk**CHQK7 7 6RX 38}}!!!$ !1!1
uU{" 6RX #(1+"<"< !Q 00DAqGKCHQKN;;Q> >Av
ADIIAAA
B
BBr c t | t rt | } | j t j t j t j t j t j t j
fv r| t d fS | j t j u r|| j
d t j u rR| j
d sEt! t" | j
d \ }}|d |d z |d |d z fS | t d fS | j t j u r#g g }}| j
D ]F\ }}t# | \ }}||z }| | | | Gt d t d }}t+ t- | D ]J} || }t+ t- | D ]}
| |
k r |||
z }||z
}||| z }K|j t j t j fv r|j
d dk r| | }}||fS | j t j u rt d t d }}| j
D ]F\ } } t# | \ }
}| dk r||
| z z }||| z z }.| dk r||| z z }||
| z z }G||fS t1 dt3 | d )z+Return expression as numer-denom pair.
r r r r z to numer and denom)r r r r r r r! r" r% r( r& r r' r rS rZ r% rb r+ r r r r r, r] r )rN numersdenomsr r nrr r r ijr_ rn bnumerbdenoms r rb rb U s #t & nn 6bj"'2:ryk2:/ / / ! $ $
Vrx
x{gk))#(1+)!$^SXa[!A!A ay6!9,fQi&).CCC ! $ $
Vrx
FF "x~~ / /
!
! e%d++1I
a
a $Q<<1 5E3v;;''
#
#1I s6{{++ ' 'AAvvVAY
"xBJ000UZ]Q5F5F %vvu%<
Vrz
!
!$Q<<1 5E((
,
,1!/!2!2q55Vq[(EVq[(EEUUV^+EV^+E%<
BDIIBBB
C
CCr c # K d} | dz
} | V
)Nr Tr r )counters r _counterr s' G1
r c i fd}t j d ddd || }d|vsJ d|vsJ |fS ) zgReplace quoted substrings of input string.
Return a new string and a mapping of replacements.
c | d d \ }}|r
|d d }ddd|d }| d| dt d }||< |S )
Nr r SINGLEDOUBLE)'"r z@__f2py_QUOTES_r @)r= COUNTER__next__)r? r r prs rr s r replz eliminate_quotes..repl s hhjj!n e 9D * *58 4 = =A = =(8(8(:(: = = =!r z+({kind}_|)({single_quoted}|{double_quoted})z
\w[\w\d_]*z('([^'\\]|(\\.))*')z("([^"\\]|(\\.))*"))r
single_quoted
double_quotedr r )r; subformat)rJ r new_srr s @r eliminate_quotesr s
A
FAHH
,, I . .
a
E e e !8Or c | D ]B\ }}|d| d }|r|dz
}| |||z } C| S )z!Inverse of eliminate_quotes.
Nr r )r findreplace)rJ rr rs rt r s r
insert_quotesr se # #1 !&&++ CKD
IIa " " Hr c d\ }}t | }dD ]-\ }}| | }|dk r!||k r|}||}}.|| i fS |}| || }| ||dz | | ||dz | k rm| ||dz }|dk rt d||z d| | ||dz | | ||dz | k mdd d
d d | }d
| dt d} | |t | z | }
t
| |t | z d \ } } |
| | < | d| | z | z | fS )zzReplace substrings of input that are enclosed in parenthesis.
Return a new string and a mapping of replacements.
)NN))(//)z()z{}z[]r Nr z Mismatch of z parenthesis in r/ r. CURLYr- )r r {r @__f2py_PARENTHESIS_r r )r r count
ValueErrorr r replace_parenthesis)
rJ r r mn_ileft_right_r r r rs rt r rr s
r r r s KD% q66D ( (
v
FF5MM 77 t88D %D |"u A uaA
''$Aq
!
!QWWUAE1%=%=
=
=
FF5!a% 77 MDJ M M M MNN N ''$Aq
!
!QWWUAE1%=%=
=
=
H7*EEdKA8q887#3#3#5#5888A !CII+a-A q3u:: / 0 0DAq AaD RaR519q=! r c v | d s
J | | d d S )Nr r r2 )
startswithsplit)rJ s r _get_parenthesis_kindr s7 <<. / /2222 / 773<<?r c | D ]b\ }}t | }t dddd | }t dddd | }| |||z |z } c| S )
z$Inverse of replace_parenthesis.
r r r r )r/ r. r r- r r }r )r r r r )rJ rr rs rt r r r s r unreplace_parenthesisr s + +1 !! $ $#ctDDDQG3s#EEEaH
IIaE) * * Hr c t | | }t |t r|S t d| d| d )zCreate an expression from a string.
This is a "lazy" parser, that is, only arithmetic operations are
resolved, non-arithmetic operations are treated as symbols.
r zfailed to parse `z` to Expr instance: got `r( )_FromStringWorkerr r r r )rJ rK r s r rL rL s[ 8,,,22155A!T
IIIQIII
J
JJr c e Zd Zd Zd Zd ZdS )_Pairc " || _ || _ d S rp r )rO r r s r r z_Pair.__init__ s
r c | j | j }}t |t r| | }t |t r| | }t || S rp )r r r r r2 r )rO r4 r r s r r2 z_Pair.substitute sg ie dD ! ! 0??;//D eT " " 2$$[11ET5!!!r c P t | j d| j d| j dS r )r r
r r r s r r z_Pair.__repr__ s. t**%BB BBTZBBBBr N)r
r r r r2 r r r r r r sF " " "C C C C Cr r c 6 e Zd Zej fdZd Zd ZddZdS )r c 0 d | _ d | _ || _ d S rp )original
quotes_maprK rN s r r z_FromStringWorker.__init__ s
r c , t || j S rp )r r )rO rJ s r finalize_stringz!_FromStringWorker.finalize_string s Q000r c h || _ t | \ }| _ | | S rp )r r r process)rO inpunquoteds r r z_FromStringWorker.parse s/
$4S$9$9!$/||H%%%r rS c t |t t f r% t | fd|D S t |t sJ t | |f t | \ }| }fdd|v r | d }dk r"t | S dk r/t | dk r t | S t d d| t j
d | }|rdk s
J | \ }}} | } | } | }t ||| S j t" j u r!t j
d
|t j }nt j
d | }|r| \ } }
} j t" j u rd |
z d z }
| | f \ } } t( |
j
}
t- t. j |
| | f S t j
d| }|rE| \ } }
|
}
t3 | |
S t j d| }t | dk r |d pd }t5 |ddd |ddd D ]P\ }} | }| }|dk r||z
}C|dk sJ ||z }Q|S j t" j u r\d|v rX | d }t- t. j t | S t j d j t" j u r|n| dd }t | dk r | } j t" j ur d |D } |d }t5 |ddd |ddd D ]G\ }} | }| }|dk r||z }:|d k sJ ||z }H|S | d s| d r[t. j t. j d |d } |dd }t- || S j t" j urd|v r|t tC | d } |d }|dd D ] } | }||z } |S t j
d " d d!" | }|r\| \ }
}}|r#|# rtI | }tK tI |
|pd# S t j
d$" d%d&d!' | }|r| \ }
}}}|r#|# rtI | }|
& }
d(|
v r3tO tQ |
d(d) |pd* S tO tQ |
|pd# S | j) v r:|d|* d+ }tW j) | |pd S |v rtY | } | |d,k rdnd }|d,k rt |t, r|S |d-v r't |t, r|f}t[ | S t j
d.| }|r| \ }}} | } | dd/ d }t |t s|f}|d,k rBt] d0 |D }t d1 |D }t_ |g|R i |S |d2k sJ || S t j
d3| }|rta | S 1 | }te d4|d5 j3 d6 ta | S )7a Parse string within the given context.
The context may define the result in case of ambiguous
expressions. For instance, consider expressions `f(x, y)` and
`(x, y) + (a, b)` where `f` is a function and pair `(x, y)`
denotes complex number. Specifying context as "args" or
"expr", the subexpression `(x, y)` will be parse to an
argument list or to a complex number, respectively.
c 3 D K | ]} | V d S rp )r )r s_contextrO s r r z,_FromStringWorker.process..) s1 AA4<<G44AAAAAAr c t | t t f r& t | t | S t | S rp )r listr r r% r )r raw_symbols_maprestores r r z*_FromStringWorker.process..restore1 sG !dE]++
0 tAwws7A///(O<< ][=]|[>])\s*(.+)\Z.r z\A(\w[\w\d_]*)\s*[=](.*)\Zz((?.~ s7 5 5 5 ' $OO,BD I I 5 5 5r r r r )r r z \A({digit_string})({kind}|)\Zz\d+z_(\d+|\w[\w\d_]*)) digit_stringr r2 z7\A({significant}({exponent}|)|\d+{exponent})({kind}|)\Zz[.]\d+|\d+[.]\d*z[edED][+-]?\d+) significantr r rr rn r r/ r, z4\A(.+)\s*(@__f2py_PARENTHESIS_(ROUND|SQUARE)_\d+@)\Zr c 3 \ K | ]'}t |t |j |j fV (d S rp )r r r r r r s r r z,_FromStringWorker.process.. sM 7 7A!+Au!5!5 7qvqw / 7 7 7 7 7 7r c 3 D K | ]}t |t |V d S rp )r r r s r r z,_FromStringWorker.process.. s1 I I1Jq%4H4H IQ I I I I I Ir r. z\A\w[\w\d_]*\Zzfromstring: treating z as symbol (original=r )4r r r r r r stripr r r r r r; r< r= r rK r r Ir0 rL r r r* r zipr) r r r r. r- reversedr isdigitr r rH r r r r rl r r> r r r~ r r~ r )rO rJ r r rC r? operr r r r r keynamer rM r rJ r r r@ r rA r r r r s ` ` @@r r z_FromStringWorker.process sX
a$ ' ' B477AAAAAqAAAAA A!S!!//DGGQ<//! 1 3 3?
GGII = = = = = = !88 wqwws||,,H& T\\( 3 3444& x==A%%%t||H'='=>>%HHHQHHJ J
J
H4a 8 8 2f$$$g$$$!(!4!4 D%<<%%DLL''ELL''E dE511 1 =H , , ,>24I IAA JAO OA ; xxzz D#u} 000Ci#o ,,wwe}'='=>>KD%""3 "??C
T5'9:: :
H2A 6 6 7 XXZZNGUGENNE$,,u"5"566 6 8 2A66 x==1 \\''(1+*<"="=>>F "8ADqD>8ADqD> B B
&
& G ,,www'7'788XXZZ99g%FF 9999g%FFM =H , , , wqwwt}}--H dll8 4 4557 7
7 8 6"&-8:"="=QQ"#))D2H"I"IL L x==1 wx((H}HJ.. 5 5+3 5 5 5 \\(1+..F "8ADqD>8ADqD> B B
&
& G ,,w//XXZZ99g%FF 9999g%FFM
<< % S 1 1 %xbf--ad3Bll771QRR5>>22GG$$ $ =
* *tqyyHWWQWWT]]%;%; < <==H\\(1+..F#ABB<
+
+ ,,w// F*M
H5<< % = ' '()
+
+
5 XXZZNE1d
!
!4yy c%jj$)!44 4
HO f%8"3 2 4 4 67
8
8
4 !
E1a
!
!4yyKKMMEe|| uU]]3%<%< = =tyqIII5<<33 3
\affSkk\?D T_Q /;; ;
)!,,ELL );! r r r r r r r r r r r r r
rk rb r r r r r r r r rL r r r r r r sN & (
t
*5 5 5 5 5D 5 5 5<
d
i &
U|
+ 6 6 6w w w w w w w w tF F FR
G
G
G
G? ? ? ? < < < < ( ( ( ( < < < < D D D2 2 2 8 8 88 8 88 8 88 8 88 8 88 8 8 ? ? ? ? ? ?*C C C**D *D *DZ (** :
( ( ( V
$: K K K KC C C C C C C C&W W W W W W W W W W r