o o hJ@sbdZddlZddlmZmZmZmZmZmZm Z ddl m Z dZ GdddZ Gdd d e ZdS) aPrettyprinter by Jurjen Bos. (I hate spammers: mail me at pietjepuk314 at the reverse of ku.oc.oohay). All objects have a method that create a "stringPict", that can be used in the str method for pretty printing. Updates by Jason Gedge (email at cs mun ca) - terminal_string() method - minor fixes and changes (mostly to prettyForm) TODO: - Allow left/center/right alignment options for above/below and top/center/bottom alignment options for left/right N)hobjvobjxsymxobjpretty_use_unicode line_widthcenter)sympy_deprecation_warningcseZdZdZdZd2ddZeddZdd Zd d Z ed d Z ddZ ddZ eddZ ddZddZd3ddZddZd4d d!Zd"d#Zd$d%Zd&d'Zfd(d)Zd*d+Zd,d-Zd.d/Zd0d1ZZS)5 stringPictzZAn ASCII picture. The pictures are represented as a list of equal length strings. linercCs&||_t||_||_d|_dS)zHInitialize from string. Multiline strings are centered. N)sr equalLengths splitlinespicturebaselinebinding)selfr rrt/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/sympy/printing/pretty/stringpict.py__init__s zstringPict.__init__cs.|sdgStdd|Dfdd|DS)Ncss|]}t|VqdSN)r.0r rrr .sz*stringPict.equalLengths..csg|]}t|qSr)r rwidthrr /z+stringPict.equalLengths..)max)linesrrrr(szstringPict.equalLengthscC t|jS)z(The height of the picture in characters.)lenrrrrrheight1s zstringPict.heightcCst|jdS)z'The width of the picture in characters.r)rrr$rrrr5szstringPict.widthc Gsg}|D]}t|trt|}||qtdd|D}tdd|D}||}g}|D]$}d|g}||j} ||} ||| |j|| | q/ddt |D} d | |fS)zsPut a string of stringPicts next to each other. Returns string, baseline arguments for stringPict. css|]}|jVqdSrrrobjrrrrFsz"stringPict.next..css|] }||jVqdSr)r%rr'rrrrGs   cSg|]}d|qS)rjoin)rr!rrrrVrz#stringPict.next.. ) isinstancestrr appendr rrr%rzipr,) argsobjectsarg newBaselinenewHeightBelowBaseline newHeightpicturesr( oneEmptyLine basePadding totalPaddingresultrrrnext9s0     zstringPict.nextcGstj|g|RS)aPut pictures next to this one. Returns string, baseline arguments for stringPict. (Multiline) strings are allowed, and are given a baseline of 0. Examples ======== >>> from sympy.printing.pretty.stringpict import stringPict >>> print(stringPict("10").right(" + ",stringPict("1\r-\r2",1))[0]) 1 10 + - 2 r r=rr2rrrrightYszstringPict.rightcGstj||fS)ziPut pictures (left to right) at left. Returns string, baseline arguments for stringPict. r>r?rrrleftjszstringPict.leftcsg}|D]}|tjurt|trt|}||qtdd|Dttd}t|D] \}}|tjur;|||<q.fdd|D}|d|dj }d ||fS) aPut pictures on top of each other, from top to bottom. Returns string, baseline arguments for stringPict. The baseline is the baseline of the second picture. Everything is centered. Baseline is the baseline of the second picture. Strings are allowed. The special value stringPict.LINE is a row of '-' extended to the width. css"|] }|tjur|VqdSr)r LINErr'rrrrs z#stringPict.stack..-cs"g|] }|jD]}t|qqSr)rr )rr(r newWidthrrrs"z$stringPict.stack..rrr-) r rBr.r/r0r r enumerater%rr,)r2r3r4lineObjir( newPicturer5rrDrstackps    zstringPict.stackcGs tj|g|R\}}||jfS)aPut pictures under this picture. Returns string, baseline arguments for stringPict. Baseline is baseline of top picture Examples ======== >>> from sympy.printing.pretty.stringpict import stringPict >>> print(stringPict("x+3").below( ... stringPict.LINE, '3')[0]) #doctest: +NORMALIZE_WHITESPACE x+3 --- 3 )r rJr)rr2r rrrrbelows zstringPict.belowcGs6tj||f\}}t|||j}||fS)zPut pictures above this picture. Returns string, baseline arguments for stringPict. Baseline is baseline of bottom picture. )r rJr#rr%r)rr2stringrrrraboveszstringPict.above()Fc Cs~|}|j}|rtsd}d}|}|r$tt|||d}t||}|r6tt|||d}t||}d|j|jfS)zPut parentheses around self. Returns string, baseline arguments for stringPict. left or right can be None or empty string which means 'no paren from that side' rrr&r-)r%rrr rr@r,r) rrAr@ifascii_nouglyhbreslparenrparenrrrparenss zstringPict.parenscsRt|j|d|jdddfddtD}|t|dS)z6Precede object by a slash of the proper size. rr-c3s2|]}d|dtddd|VqdSr)r/N)rrrHr%rrrs  z'stringPict.leftslash..)r rr%r,rangerAr )rslashrr[r leftslashszstringPict.leftslashNcs|d|}|dfddtD}t|d}dkr,tdd}ntd}|d urN||krN|d ||}||}||}|j|||_||S) zUProduce a nice root symbol. Produces ugly results for big n inserts. _r-c3s,|]}d|ddd|VqdSrXrrZr[rrrs  z"stringPict.root..rrWz\ \\Nr)) rMrr%r,r\r rAr@r)rnr<r]downlinerootrr[rrcs    zstringPict.rootcstdurt|d<|ddurd|jS|ddur|dn|dkr)d|kr8t|jd|S d|d sBtrGd d nd d fdd}tt||j}dd|D}d|S)zReturn the string form of self. Unless the argument line_break is set to False, it will break the expression in a form that can be printed on the terminal without being broken up. N wrap_lineFr- num_columnsrPr use_unicodeu↪ u ↪z> z >c3s}t|d}}t|D](\}}t|}||kr4||||V}t||}}||7}q|||dVdS)z-Yields consecutive chunks of line_width ncolsrN)rrF)r prefixrstartrHxwxncols overflow_endoverflow_firstoverflow_startrrchunks-s  z!stringPict.render..chunkscSr*)r-r+)rrrrrr@rz%stringPict.render..z ) _GLOBAL_WRAP_LINEr,rterminal_widthrtyperr1map)rr2kwargsrqr8rrlrrenders,      zstringPict.rendercCstjdd}|jS)zCReturn the terminal width if possible, otherwise return 0. )rr)fallback)shutilget_terminal_sizecolumns)rsizerrrrsEs zstringPict.terminal_widthcCs4t|tr d|j|kSt|tr|j|jkSdS)Nr-F)r.r/r,rr )rorrr__eq__Ks   zstringPict.__eq__cs tSr)super__hash__r$ __class__rrrR zstringPict.__hash__cCs d|jS)Nr-)r,rr$rrr__str__Us zstringPict.__str__cCsdd|j|jfS)NzstringPict(%r,%d)r-)r,rrr$rrr__repr__XszstringPict.__repr__cCs |j|Sr)r)rindexrrr __getitem__[rzstringPict.__getitem__cCr"r)r#r r$rrr__len__^rzstringPict.__len__)r)rNrOFr)__name__ __module__ __qualname____doc__rBr staticmethodrr%rr=r@rArJrKrMrVr^rcrwrsr~rrrrr __classcell__rrrrr s6    #  "L r c@s|eZdZdZed\ZZZZZ Z Z Z dddZ eddZd d Zdd d ZddZddZddZgdZeddZdS) prettyForma Extension of the stringPict class that knows about basic math applications, optimizing double minus signs. "Binding" is interpreted as follows:: ATOM this is an atom: never needs to be parenthesized FUNC this is a function application: parenthesize if added (?) DIV this is a division: make wider division if divided POW this is a power: only parenthesize if exponent MUL this is a multiplication: parenthesize if powered ADD this is an addition: parenthesize if multiplied or powered NEG this is a negative number: optimize if added, parenthesize if multiplied or powered OPEN this is an open object: parenthesize if added, multiplied, or powered (example: Piecewise) rNcCs8t|||||_|durtdddd|p||_dS)z-Initialize from stringPict and binding power.Nz The unicode argument to prettyForm is deprecated. Only the s argument (the first positional argument) should be passed. 1.7$deprecated-pretty-printing-functionsdeprecated_since_versionactive_deprecations_target)r rrr _unicode)rr rrunicoderrrrvszprettyForm.__init__cCstdddd|jS)Nz} The prettyForm.unicode attribute is deprecated. Use the prettyForm.s attribute instead. rrr)r rr$rrrrs zprettyForm.unicodecGsz|}|jtjkrt|}|g}|D]}|jtjkr!t|}|jtjkr,|d||qttj|dtjiS)zTMake a pretty addition. Addition of negative numbers is simplified. z + r)rrNEGr rVr0r=ADD)rothersr4r<rrr__add__s       zprettyForm.__add__FcCsv|rtd|}|jtjkrt|}|jtjkr t|}|jtjkr-|dd}tt|tj |dtjiS)z4Make a pretty division; stacked or slashed. zCan't do slashed fraction yetr)rr) NotImplementedErrorrrDIVr rVrr@rJrB)rdenslashednumrrr __truediv__s     zprettyForm.__truediv__c Gsbddi}t|dkr |S|}|jtjkr |jtjkr t|}|g}|D])}|jd|vr7| t d|jtjkrI|jtjkrIt|}| |q%t|}t |D](}||dkr||dkr||dt dkr| || || |dqW|dddkrtj}|ddkr|d}|j|jddkrd|d<ntj}ttj|d |iS) zZMake a pretty multiplication. Parentheses are needed around +, - and neg. degree°r*rz-1rCz- r)r#rrMULrr rVrvaluesr0rr\popinsertrr=) rrquantityr4r<len_resrHbinr@rrr__mul__s<     ,    zprettyForm.__mul__cCsdd|j|j|jfS)NzprettyForm(%r,%d,%d)r-)r,rrrr$rrrrs  zprettyForm.__repr__cCs|}d}|jtjkrt|}|jtjkrt|}n|jtjkr2|dkr0t|}nd}|rM|jj||_t|j |}t| |j St| d| }t| d| }t| |dtjiS)zMake a pretty power. FrTr)r)rrPOWr rVFUNCr% prettyFuncrr@ prettyArgsrArrM)rrRause_inline_func_formfunctopbotrrr__pow__s"     zprettyForm.__pow__)sincostancGs|tjvr*t|dksJd||d}|jtjkr*t||ddtjiSg}|D]}|d||q.t t j |dd}t | }t||dtj iS)z,Functions of one or more variables. rz'Simple function %s must have 1 argumentrr)r,N) rsimpleFunctionsr# __pretty__rrrArr0r r=rVATOM)functionr2r4 argumentListrrrapplys$     zprettyForm.apply)rrN)F)rrrrr\rrrrrrrOPENrpropertyrrrrrrrrrrrrrrbs   +r)rrypretty_symbologyrrrrrrr sympy.utilities.exceptionsr rrr rrrrrs$ N