jNUdZddlmZddlmZmZmZmZmZddl m Z ddl Z ddl Z ddl mZddlmZmZmZmZmZmZdd lmZmZdd lmZdd lmZdd lmZdd l m Z erddl!Z" dd l#mZ$de_erddl%Z&n#e'$r dZ$de_YnwxYwd9d:dZ( d;dd2Z*iZ+d3e,d4<e)d?d6Z-Gd7d8Z.dS)@z Schemdraw Drawing class ) annotations)AnyMutableMappingUnionOptional TYPE_CHECKING)ChainMapN)default_canvas)BBoxBackends ImageFormat LinestyleXY ImageType)Element Container) SegmentType)Point)Figure) drawing_stack matplotlibsvgbackendr returnNonecV|dkrttd|t_dS)z6 Change default backend, either 'matplotlib' or 'svg' rNzCould not import Matplotlib.) mplFigure ValueErrorr )rs R/data/share/cad/hardware/.venv/lib/python3.11/site-packages/schemdraw/schemdraw.pyuser!s0,  ;<< <$+N!!!@?皙?,@ sans-serifblack@-unitfloatinches_per_unitlblofstfontsizefontstrcolorlwlsrfill Optional[str]bgcolormarginmathfontc |td<|td<|td<|td<|td<|td<|td<|td<|td <| td <| r | td <| r | td <d Sd S)a Set global schemdraw style configuration Args: unit: Full length of a 2-terminal element. Inner zig-zag portion of a resistor is 1.0 units. inches_per_unit: Inches per drawing unit for setting drawing scale lblofst: Default offset between element and its label fontsize: Default font size for text labels font: Default font family for text labels color: Default color name or RGB (0-1) tuple lw: Default line width for elements ls: Default line style fill: Deault fill color for closed elements margin: White space around the drawing in drawing units mathont: Font for math delimited by $..$ r+r-r.r/r0r2r3r4r5r8r7r9Nschemdrawstyle) r+r-r.r/r0r2r3r4r5r7r8r9s r configr=&s,"N6(7N$% 'N9!)N:!N6#N7N4N4!N6%N8,$+y!.%-z"""..r"Tdwgbboxboolelmbboxc.|td<|td<dS)z4 Debug - draw element and/or drawing bounding boxes r>r@Nr;)r>r@s r debugrBLs!(N9 'N9r"zdict[str, Any]r<defaultc|dkrtdd$S|dkrdtd<dtd<d$S|dkrd td<d td<d$S|d krd td<d td<d$S|dkrdtd<dtd<d$S|dkrdtd<dtd<d$S|dkrdtd<dtd<d$S|dkrdtd<dtd<d$S|dkrdtd<dtd<d$S|dkrdtd<dtd<d$S|d krd!td<d"td<d$Std#|)%a Set schemdraw theme (line color and background color). Themes match those in jupyter-themes package (https://github.com/dunovank/jupyter-themes). Available themes: * default (black on white) * dark (white on black) * solarizedd * solarizedl * onedork * oceans16 * monokai * gruvboxl * gruvboxd * grade3 * chesterish rCwhite)r7darkr2r(r7 solarizeddz#002b36z#657b83 solarizedlz#eee8d5z#073642onedorkz#373e4bz#899ab8oceans16z#384151z#CDD2E9monokaiz#232323z#BBBBBBgruvboxlz#ebdbb2z#3c3836gruvboxdz#1d2021z#d5c4a1grade3z#ffffffz#3f3d46 chesterishz#323A48z#92A2BDzUnknown theme N)r=r<r)themes r rPrPWs$ w &")w$+y!!! ,  $-y!"+w ,  $-y!"+w )  $-y!"+w *  $-y!"+w )  $-y!"+w *  $-y!"+w *  $-y!"+w (  $-y!"+w ,  $-y!"+w1%11222r"czeZdZdZ dXdYdZedZejdZedZejdZ dZd[dZ dZ dZ d\d Z d!Z d]d^d#Zd_d%Zd`d'Zd(Zd)Zdad,Zdbd-Zdcd0Zddd1Zdedfd5Zdgdhd:Zdid;Zddd<Zddd=Z djdkdJZdKZdldLZdldMZ dmdndPZ dodpdRZ!dqdrdWZ"dS)sDrawingaM A schematic drawing See `schemdraw.config` method for argument defaults Args: canvas: Canvas to draw on when using Drawing context manager. Can be string 'matplotlib' or 'svg' to create new canvas with these backends, or an instance of a matplotlib axis, or an instance of xml.etree.ElementTree containing SVG. Default is value set by schemdraw.use(). file: optional filename to save on exiting context manager or calling draw method. show: Show the drawing after exiting context manager transparent: Save to file with a transparent background dpi: Dots-per-inch when saving to a raster graphics file Attributes: here: (xy tuple) Current drawing position. The next element will be added at this position unless specified otherwise. theta: (float) Current drawing angle, in degrees. The next element will be added with this angle unless specified otherwise. NTFHcanvasPOptional[Union[Backends, xml.etree.ElementTree.Element, matplotlib.pyplot.Axes]]filer6showr? transparentdpir,c  ||_||_||_||d|_g|_i|_d|vr6|d|_tjdtdt |_ |j ||dtd|_t!d|_d|_g|_d |_d|_dS) NrXrYr-Use of `backend` is deprecated. Use `canvas`. stacklevelr+rrrF)outfilerTrWsaveoptselementsanchorspopwarningswarnDeprecationWarningr<copy dwgparamsupdategetr+r_here_theta_state _interactivefig)selfrTrVrWrXrYkwargss r __init__zDrawing.__init__s   (3C@@ ') OQ    **Y//DK MI, < < < <*8)<)<)>)> f%%%JJv~'9'9&'A'ABB v  13 !:>r"c8tjd|jSNr push_elementrmrrs r herez Drawing.heres"4(((zr"c<tjd||_dSrvrwrrvalues r rzz Drawing.heres"4((( r"c8tjd|jSrv)rrxrnrys r thetaz Drawing.thetas"4((({r"c||_dSrv)rnr|s r rz Drawing.thetas  r"333333?? cornerradiuspadxpadyc(t||||S)a Add a container to the drawing. Use as a context manager, such that elemnents inside the `with` are surrounded by the container. >>> with drawing.container(): >>> elm.Resistor() >>> ... Args: cornerradius: radius for box corners padx: space between contents and border in x direction pady: space between contents and border in y direction )rrr)r)rrrrrs r containerzDrawing.containersLt$OOOOr"c.tj||Srv)r push_drawingrys r __enter__zDrawing.__enter__s"4((( r"ctjdtj||j|j|jfi|j|js|d|jrVt|j dsC t|jdS#t$r|jYdSwxYwdSdS)z1 Exit context manager - save to file and display NFrWplot) rrx pop_drawingrasaverbrqdrawrWhasattrrTdisplay NameError)rrexc_typeexc_valexc_tbs r __exit__zDrawing.__exit__s"4(((!$''' < # DIdl 4 4dm 4 4 4x " II5I ! ! ! 9 WT[&99  !!!!!         sB#B?>B?namer1rrc|t|divr(t|d|Std|)z, Allow getting anchor position as attribute rdz'Drawing' has no attribute )varsrlAttributeErrorrrrs r __getattr__zDrawing.__getattr__sU 4::>>)R00 0 0::>>),,T2 2A4AABBBr"c||jvSrv)rcrrelements r __contains__zDrawing.__contains__s$-''r" interactivec||_dS)z Enable interactive mode (matplotlib backend only). Matplotlib must also be set to interactive with `plt.ion()`. N)rp)rrrs r rzDrawing.interactives(r"r crtj}tj }tj}tj }|jD]l}|d}t |j|}t |j|}t |j|}t |j |}mt||||S)z Get drawing bounding box T) transform) mathinfrcget_bboxminxminmaxxmaxyminymaxr )rrrrrrrbboxs r rzDrawing.get_bbox sxyxy} ( (G##d#33Dty$''Dty$''Dty$''Dty$''DDD$d+++r"list[SegmentType]cg}|jD]Itjjj|fdjDJ|S)z3 Get flattened list of all segments in the drawing c6g|]}|jjfiS)xformr).0srparamss r z(Drawing.get_segments..s@888!"%QWW%6AA&AA888r")rcr _userparams elmparamsdefaultsextendsegments)rrrrrs @@r get_segmentszDrawing.get_segmentss} 9 9Gg173DgFVWWF OO88888&-&6888 9 9 9 9r"cx|ddS)z SVG representation for Jupyter Frr)rgetimagedecoderys r _repr_svg_zDrawing._repr_svg_!s0yyey$$--e44;;===r"c|jdkst|jdr)|ddSdS)z PNG representation for Jupyter rrFrpngN)rTrrrrys r _repr_png_zDrawing._repr_png_%sG ;, & &'$+v*F*F &99%9((11%88 8tr"rrc0|||S)z( In-place add element, via += operator. addrs r __iadd__zDrawing.__iadd__+s  r"c0|j|j|jfi|j\|_|_|j||jr|j[t|j d|_d|jvr%|j |jd| |j|j | |jnd|_|S)zh Add an element to the drawing. Args: element: The element to add. Nr-)r-r7)_placermrnrjrcappendrprqrrlr7_drawset_bboxrrrs r rz Drawing.add0s #1'.T["["[DN"["[ DK W%%%   x$$(N$6$67H$I$IKKK..H$$T^I%>??? MM$( # # # H  dmmoo . . . H      DHr"rcrc:|D]}||dS)z& Add multiple elements to the drawing Nr)rrrcrs r add_elementszDrawing.add_elementsFs0  G HHW      r"c||jd|j|jD]}||j|jdj\|_|_|j| |j dS)z" Removes previously added element N) rcrerqclearrabsdroprmrnrrrrs r undoz Drawing.undoKs " } $ $G MM$( # # # #"&-"3"; DK $--//*** r"rdxdyctjdt|jd|z|jd|zf|_dS)z Move the current drawing position Args: dx: change in x position dy: change in y position Nrr )rrxrrm)rrrrs r movez Drawing.moveUsC "4(((DJqMB. 1 0BCDD r"refrrOptional[float]cvtjd|j|z|j|zf|_| ||_dSdS)zn Move drawing position relative to the reference point. Change drawing theta if provided. N)rrxxyrmrn)rrrrrrs r move_fromzDrawing.move_from_sF "4(((ebj#%"*-  DKKK  r"c$|j|j|<dS)z9 Define a Drawing anchor at the current drawing position N)rmrdrs r set_anchorzDrawing.set_anchorhs!Z Tr"ctjd|jt |j|jfdS)z\ Push/save the drawing state. Drawing.here and Drawing.theta are saved. N)rrxrorrrmrnrys r pushz Drawing.pushlsB "4((( E$*--t{;<<<<#%DN4 >#%DN4  %)DN6 "  (/DN9 %  '-DN8 $  )1DN: & & & r"cN|jD]}||jdS)z# Draw all the elements on self.fig N)rcrrqrs r _drawelementszDrawing._drawelementss4} $ $G MM$( # # # # $ $r"c |j|t||jd|jd|jdd|_d|jvr%|j|jd|j||dS)z Draw on Matplotlib Axis Nr-r8r>F)axr-r8showbboxr7)rqrrjrlr7rrr)rrrs r _drawmplzDrawing._drawmpls 8 r~ B151C1CDU1V1V(,x(@*..*<*F)rrr-r8rr7)rq svgFigurerrjrlr7r)rrrs r _drawsvgzDrawing._drawsvgs 8 s St}}151C1CDU1V1V(,(:(:8(D(D*..*<*[F >#2F \ ! ! MMOOOO VV $ $ " MMVM $ $ $ $ u__ MMOOOO MM& ! ! !   HMMOOO < # DIdl 4 4dm 4 4 4xr"fnamecz|j|d|j|||dS)a Save figure to a file Args: fname: Filename to save. In Matplotlib backend, the file type is automatically determined from extension (png, svg, jpg). SVG backend only supports saving SVG format. transparent: Save as transparent background, if available dpi: Dots-per-inch for raster formats NFrr[)rqrr)rrrrXrYs r rz Drawing.saves@ 8  II5I ! ! !  e# >>>>>r"rfmtImageFormat | ImageTypebytesc|jdkr'|dkrtd|j|d|j|S)z Get image data as bytes array Args: fmt: Format or file extension of the image type. SVG backend only supports 'svg' format. Returns: Image data as bytes rz$Format not available in SVG backend.NFr)ext)rTlowerrrqrr)rrrs r get_imagedatazDrawing.get_imagedatasg ;%  CIIKK5$8$8CDD D 8  II5I ! ! !x  S )))r")NNTFrS) rTrUrVr6rWr?rXr?rYr,)rrr)rr,rr,rr,)rr1rr)T)rr?)rr )rr)rr)rrrr)rcrrr)rrr`)rr,rr,rr)rrN) rrrr,rr,rrrr)rr1rr) NNNNNNNNNNN)r+rr-rr/rr0r6r2r6r3rr4rr5r6r7r6r8rr9r6rrrv)TNN)rWr?rr)TrS)rr1rXr?rYr,rr)r)rrrr)#__name__ __module__ __qualname____doc__rtpropertyrzsetterrrrrrrrrrrrrrrrrrrrrer=rrrrrrrr"r rRrRs2DH'+"%* ?????<X [[X \\/136PPPPP"    CCCC (((((((( , , , ,>>>  , EEEEE     ((((====8888W[GKbfDH)-)- +2+2+2+2+2Z$$$        !%8<*****X ? ? ? ? ?*******r"rR)r)rr rr) r#r$r%r&r'r(r)r*NNr%N)r+r,r-r,r.r,r/r,r0r1r2r1r3r,r4rr5r6r7r6r8r,r9r6rr)TT)r>r?r@r?)rC)/r __future__rtypingrrrrr collectionsr rfrr typesr r rrrrrcrrrrutilr backends.svgrrrxml.etree.ElementTree.Elementxml backends.mplrmatplotlib.pyplot.Axesr ImportErrorr!r=rBr<__annotations__rPrRrr"r rs]""""""FFFFFFFFFFFFFF  HHHHHHHHHHHHHHHH((((((((!!!!!!------)((((*111111$0N!&%%%%***I$)N!!!* ,,,,,8;3629+.@D:> #.#.#.#.#.L((((("$####33333333l**********s#A77BB