瞤dp ^ddlmZddlmZddlZddlZddlZddlZddlmZddl Tddl m Z m Z ddl Z dZd Zd Zejd Zd Zd ZdZegdZgdZddegddegddegddegddegddegddegddegddegd degd!degg Zd"d#iZed$eDed%gzZGd&d'eZGd(d)eZGd*d+eZGd,d-e Z!Gd.d/eZ"Gd0d1eZ#Gd2d3e Z$d4Z%d)absolute_import)print_functionN)stderr)*) SVG_COLORS COLOR_SCHEMEScfd}|S)Nctj}|i|}tdjtj|z |S)Nz! -> TIME:)timeprint__name__)argskargst1rfs 2lib/python3.11/site-packages/ete3/treeview/main.pya_wrapper_accepting_argumentsz0tracktime..a_wrapper_accepting_arguments6sG Y[[ At u   11:ty{{R?OPPP)rrs` r tracktimer5s$ )(rc |dvS)N)rrrxs rr>s qG|rc,t|tSN) isinstanceintrs rrr?s*Q,,rz^#[A-Fa-f\d]{6}$cl|tvptjt|Sr)lowerrrematch _COLOR_MATCHrs rrrAs$17799 2Obh|Q6O6Orc |dvS)N)spherecirclesquarerrs rrrBs q$BBrc4t|tp|dvS)N)rr)r boolrs rrrCs:a..rH_s>>>aQqT>>>r_facesceZdZdZdZdS)_Borderc0d|_d|_d|_dS)Nr)widthtypecolorselfs r__init__z_Border.__init__bs   rc|j|}t|}|||jr"t t |j}nt tj}t||j | |j| tj |||SdSr)rM boundingRectQGraphicsRectItem setParentItemrOQPenQColorQtNoPen set_pen_stylerNsetWidth setCapStyleFlatCapsetPen)rQitemrborderpens rapplyz _Border.applygs : !!!##A&q))F   & & &z %6$*--..28nn #ty ) ) ) LL $ $ $ OOBJ ' ' ' MM#   M4rN)r __module__ __qualname__rRrcrrrrKrKas2 rrKceZdZdZdZdZdS) _Backgroundzc Set the background of the object :param color: RGB color code or :data:`SVG_COLORS` cd|_dSr)rOrPs rrRz_Background.__init__s  rc|jr|}t|}||t t |j}t t |j}||||| tj |SdSr) rOrTrUrVrWrXQBrushr_setBrushsetFlag QGraphicsItemItemStacksBehindParent)rQr`rbgrbbrushs rrcz_Background.applys : !!##A"1%%B   T " " "vdj))**C6$*--..E IIcNNN KK    JJ}; < < <I4rN)rrdre__doc__rRrcrrrrgrgxs<      rrgc<eZdZdZdZdZeeeZdZdS)_ActionDelegatorz4 Used to associate GUI Functions to nodes and faces c|jSr _delegaterPs r get_delegatez_ActionDelegator.get_delegates ~rc  t|dr||t|D]Q}|ds:|dkr4t ||}t ||t j||R||_dS)Ninit_) hasattrrydir startswithgetattrsetattrtypes MethodTyperv)rQdelegateattrfns r set_delegatez_ActionDelegator.set_delegates 8V $ $ MM$   MM @ @D??3'' @DFNNXt,,dE$4R$>$>???!rcd|_dSrrurPs rrRz_ActionDelegator.__init__s rN) rrdrerqrwrpropertyrrRrrrrsrssZ>>"""x l33Hrrsc2eZdZdZfdZdZfdZxZS)r0a .. versionadded:: 2.1 .. currentmodule:: ete3 A dictionary with all valid node graphical attributes. :argument #0030c1 fgcolor: RGB code or name in :data:`SVG_COLORS` :argument #FFFFFF bgcolor: RGB code or name in :data:`SVG_COLORS` :argument #FFFFFF node_bgcolor: RGB code or name in :data:`SVG_COLORS` :argument #FFFFFF partition_bgcolor: RGB code or name in :data:`SVG_COLORS` :argument #FFFFFF faces_bgcolor: RGB code or name in :data:`SVG_COLORS` :argument #000000 vt_line_color: RGB code or name in :data:`SVG_COLORS` :argument #000000 hz_line_color: RGB code or name in :data:`SVG_COLORS` :argument 0 hz_line_type: integer number :argument 0 vt_line_type: integer number :argument 3 size: integer number :argument "circle" shape: "circle", "square" or "sphere" :argument True draw_descendants: Mark an internal node as a leaf. :argument 0 hz_line_width: integer number representing the width of the line in pixels. A line width of zero indicates a cosmetic pen. This means that the pen width is always drawn one pixel wide, independent of the transformation set on the painter. :argument 0 vt_line_width: integer number representing the width of the line in pixels. A line width of zero indicates a cosmetic pen. This means that the pen width is always drawn one pixel wide, independent of the transformation set on the painter. cptt|j|i||dSr)superr0rRry)rQrr __class__s rrRzNodeStyle.__init__s6'i'7777 rctD]<\}}}||vr|||<|||std|d||=dS)N'z1' attribute in node style has not a valid value: )NODE_STYLE_DEFAULT ValueError)rQkeydvaluecheckers rryzNodeStyle.initsy$6 7 7 C$"S WT#Y'' 7 j&)cc499"6777 7 7 7rcD|dkr^td|dddttdttt|d|d }|d kr5d }td|dd dttdt|d kr5d}td|dddttdt|t vrt d|ztt|||dS)N line_typez WARNING: [z4] keyword is deprecated and it has been replaced by z[hz_line_type, vt_line_type].)filezGWARNING: Support for this keyword will be removed in next ETE versions.r9r:vlwidthr@z[vt_line_width]hlwidthr?z[hz_line_width]z4'%s' is not a valid keyword for a NodeStyle instance)r rrr0 __setitem__VALID_NODE_STYLE_KEYSr)rQrGvrs rrzNodeStyle.__setitem__sl   E2224:@ B B B B [bh i i i i )T " " . .~q A A AA >>A E%%%'-3 5 5 5 5 [bh i i i i >>A E%%%'-3 5 5 5 5 [bh i i i i ) ) )SUVVWW W i**1a00000r)rrdrerqrRryr __classcell__rs@rr0r0sk""H 777111111111rr0cBeZdZdZdZdZeeeZdZdZ dS)r1a.. versionadded:: 2.1 .. currentmodule:: ete3 Contains all the general image properties used to render a tree **-- About tree design --** :param None layout_fn: Layout function used to dynamically control the aspect of nodes. Valid values are: None or a pointer to a method, function, etc. **-- About tree shape --** :param "r" mode: Valid modes are 'c'(ircular) or 'r'(ectangular). :param 0 orientation: If 0, tree is drawn from left-to-right. If 1, tree is drawn from right-to-left. This property only makes sense when "r" mode is used. :param 0 rotation: Tree figure will be rotate X degrees (clock-wise rotation). :param 1 min_leaf_separation: Min separation, in pixels, between two adjacent branches :param 0 branch_vertical_margin: Leaf branch separation margin, in pixels. This will add a separation of X pixels between adjacent leaf branches. In practice, increasing this value work as increasing Y axis scale. :param 0 arc_start: When circular trees are drawn, this defines the starting angle (in degrees) from which leaves are distributed (clock-wise) around the total arc span (0 = 3 o'clock). :param 359 arc_span: Total arc used to draw circular trees (in degrees). :param 0 margin_left: Left tree image margin, in pixels. :param 0 margin_right: Right tree image margin, in pixels. :param 0 margin_top: Top tree image margin, in pixels. :param 0 margin_bottom: Bottom tree image margin, in pixels. **-- About Tree branches --** :param None scale: Scale used to draw branch lengths. If None, it will be automatically calculated. :param "mid" optimal_scale_level: Two levels of automatic branch scale detection are available: :attr:`"mid"` and :attr:`"full"`. In :attr:`full` mode, branch scale will me adjusted to fully avoid dotted lines in the tree image. In other words, scale will be increased until the extra space necessary to allocated all branch-top/bottom faces and branch-right faces (in circular mode) is covered by real branches. Note, however, that the optimal scale in trees with very unbalanced branch lengths might be huge. If :attr:`"mid"` mode is selected (as it is by default), optimal scale will only satisfy the space necessary to allocate branch-right faces in circular trees. Some dotted lines (artificial branch offsets) will still appear when branch-top/bottom faces are larger than branch length. Note that both options apply only when :attr:`scale` is set to None (automatic). :param 0.25 root_opening_factor: (from 0 to 1). It defines how much the center of a circular tree could be opened when adjusting optimal scale, referred to the total tree length. By default (0.25), a blank space up to 4 times smaller than the tree width could be used to calculate the optimal tree scale. A 0 value would mean that root node should always be tightly adjusted to the center of the tree. :param True complete_branch_lines_when_necessary: True or False. Draws an extra line (dotted by default) to complete branch lengths when the space to cover is larger than the branch itself. :param 2 extra_branch_line_type: 0=solid, 1=dashed, 2=dotted :param "gray" extra_branch_line_color: RGB code or name in :data:`SVG_COLORS` :param False force_topology: Convert tree branches to a fixed length, thus allowing to observe the topology of tight nodes :param False draw_guiding_lines: Draw guidelines from leaf nodes to aligned faces :param 2 guiding_lines_type: 0=solid, 1=dashed, 2=dotted. :param "gray" guiding_lines_color: RGB code or name in :data:`SVG_COLORS` **-- About node faces --** :param False allow_face_overlap: If True, node faces are not taken into account to scale circular tree images, just like many other visualization programs. Overlapping among branch elements (such as node labels) will be therefore ignored, and tree size will be a lot smaller. Note that in most cases, manual setting of tree scale will be also necessary. :param True draw_aligned_faces_as_table: Aligned faces will be drawn as a table, considering all columns in all node faces. :param True children_faces_on_top: When floating faces from different nodes overlap, children faces are drawn on top of parent faces. This can be reversed by setting this attribute to false. **-- Addons --** :param False show_border: Draw a border around the whole tree :param True show_scale: Include the scale legend in the tree image :param None scale_length: Scale length to be used as reference scale bar when visualizing tree. None = automatically adjusted. :param False show_leaf_name: Automatically adds a text Face to leaf nodes showing their names :param False show_branch_length: Automatically adds branch length information on top of branches :param False show_branch_support: Automatically adds branch support text in the bottom of tree branches **-- Tree surroundings --** The following options are actually Face containers, so graphical elements can be added just as it is done with nodes. In example, to add tree legend: :: TreeStyle.legend.add_face(CircleFace(10, "red"), column=0) TreeStyle.legend.add_face(TextFace("0.5 support"), column=1) :param aligned_header: a :class:`FaceContainer` aligned to the end of the tree and placed at the top part. :param aligned_foot: a :class:`FaceContainer` aligned to the end of the tree and placed at the bottom part. :param legend: a :class:`FaceContainer` with an arbitrary number of faces representing the legend of the figure. :param 4 legend_position=4: TopLeft corner if 1, TopRight if 2, BottomLeft if 3, BottomRight if 4 :param title: A Face container that can be used as tree title cg|_t|ttttt gvr|g}|D]}t |dus||j|0ddlm } |jt||`#t$r#}t|tdd}~wwxYwdS)NTr)layoutszBRequired layout is not a function pointer nor a valid layout name.)_layout_handlerrNrDlisttuple frozensetcallableappendrr~ Exceptionr r)rQlayoutlyres r set_layout_fnzTreeStyle.set_layout_fns! <.)sdrz[%s] wrong typez[%s] option is not supported)r{r~TREE_STYLE_CHECKERgetobject __setattr__r)rQrvals rrzTreeStyle.__setattr__'s 4   CgdL!&D&D C!%%dNN;;C@@ :""4s33333 !2D!8999;TABB BrN) rrdrerqrrr layout_fnrRrrrrr1r1s|UUnlll"$$$ 66I{{{|CCCCCrr1c.eZdZdZfdZfdZxZS) _FaceAreascTtD]}t||t dSr)FACE_POSITIONSrr2)rQas rrRz_FaceAreas.__init__1s4 . .A D!]__ - - - - . .rc|tvrtd|dttt|||SNz Face area [z ] not in )rAttributeErrorrrr)rQrrrs rrz_FaceAreas.__setattr__5sI ~ % % .dddNN!SUU UZ&&224===rc|tvrtd|dttt||Sr)rrrr __getattr__)rQrrs rrz_FaceAreas.__getattr__:sG ~ % % .dddNN!SUU UZ&&224888r)rrdrerRrrrrs@rrr0s`...>>>>> 999999999rrceZdZdZdZdS)r2zw .. versionadded:: 2.1 Use this object to create a grid of faces. You can add faces to different columns. cr|t|g|dS)zC add the face **face** to the specified **column** N) setdefaultr!r)rQfacecolumns radd_facezFaceContainer.add_faceEs2 F R((//55555rN)rrdrerqrrrrr2r2?s- 66666rr2cjt|do |jd }|p|S)N _img_styler>)r{ img_styleis_leaf)node collapseds rr3r3Ks4l++VDNCU4V0VI  & &rFr-c|dkrd}t|ddr+t|j|||dStd)a .. currentmodule:: ete3.treeview.faces Adds a Face to a given node. :argument face: A :class:`Face` instance .. currentmodule:: ete3 :argument node: a tree node instance (:class:`Tree`, :class:`PhyloTree`, etc.) :argument column: An integer number starting from 0 :argument "branch-right" position: Possible values are "branch-right", "branch-top", "branch-bottom", "float", "float-behind" and "aligned". Tr/ _temp_facesNzVThis function can only be called within a layout function. Use node.add_face() instead)r~rrr)rrrr/positions rr4r4Os_&$t]D))s (++44T6BBBBBqrr rrc|dkr!|tjdS|dkr!|tjdS|dkr!|tjdSdS)Nrrr)setStylerY SolidLineDashLineDotLine)rb line_styles rr[r[ksrQ R\""""" q R[!!!!! q RZ      rZpxcd}|dkrd}d}n|dkrd}d}n|drD |dd }n#t$rd}YnwxYwd}n-|dd }|} | | z } |s9|s7d }| }| }tj} n8|r|r tj } n'||| z}tj} n||| z }tj} |d kr|r|d z|z}|r|d z|z}n+|dkr|r||z}|r||z}n|d krntd|| z || z } } |dkrt}tdd||}| tt|t||||d|d|dkrRt'}t)|}|t,j||n||t5}||||||| ||dkrt=|}t?dn;t+|5}| }dddn #1swxYwY|!dd}tEj#dd|}|!dd}tEj#dd|}|rddl$m%}||S|dkr| | |fSt+|d5}|&|dddn #1swxYwYn|dks|dkrI|dkr tNj(}n tNj)}tOtNj*}|+||,||-tNj.|/ta||tNj1|2ddddtNj1|3d|4|t5|}tdd||}||||| nttdd||}tkt|t|tjj6}|7tqtj9:|;t|dz |<t|dz t5|}|=t4j>|=t4j?|=t4j@||||| ||rut'}t)|}|t,j|A|ddd l$mB}||CS|dkrit'}t)|}|t,j|A|d| | |DfS|A||| z || z fS)!NFz%%inlineTPNGz %%inlineSVGSVGz%%returnrrrmmgc6P(?inzwrong unit formatrz(Generated with ETE http://etetoolkit.orgz from memoryzxml:id=zid=zfont-size="(\d+)"zfont-size="\1pt"  z]+>\s*r)rwPDFPSg F%u?)Image)Er}splitupper IndexError sceneRectheightrMrYKeepAspectRatioIgnoreAspectRatior QSvgGeneratorQRectFsetSizeQSizer! setViewBoxsetTitlesetDescription QByteArrayQBufferopen QIODevice WriteOnlysetOutputDevice setFileNameQPainterbeginrenderendstrr readreplacer$subIPython.core.displayrwriteQPrinterPostScriptFormat PdfFormatHighResolution setResolutionsetOutputFormat setPageSizeA4 setPaperSizeQSizeF DevicePixelsetPageMargins setFullPagesetOutputFileNameQImage Format_ARGB32fillrXwhitergbsetDotsPerMeterXsetDotsPerMeterY setRenderHint AntialiasingTextAntialiasingSmoothPixmapTransformsaver datatoBase64)sceneimgNamer hdpi take_regionunitsipython_inlineext main_rect aspect_ratio ratio_modex_scaley_scalesvg targetRectbabufppcompatible_coderrformatprinteriir s rrFrFts:N* M ! !   J ' '- --$$Q'--//CC   CCC mmC  $**,,!!I##%% (9(99L (Q ( OO       ' (q()  '  '  }}  'L 3&A  'L 3&A $  CA  CA $ +,,,***Ai.>.>.@.@,@WG e||ooAq!Q''  E#a&&#a&&))*** z""" ?@@@ EFFF j B"++C HHY( ) ) )    $ $ $ $ OOG $ $ $ ZZ   RU__%6%6 CCC  j !"ggO - g +!"#&&(( + + + + + + + + + + + + + + + *11)UCC&!46I?[[)11$<<&!3RII  ) 0 0 0 0 0 03'' '  " "G_4 4gs## )q((( ) ) ) ) ) ) ) ) ) ) ) ) ) ) )  $;;.FF'F8233c"""'''HK(((VAq\\8+?@@@q!Q8+?@@@ D!!!!!'*** g  QAq))  RU__%6%6 CCCCAq!Q'' 3q663q666#7 8 8 rx  $$&&''' Cf --... Cf --... b\\ ./// 2333 7888 RU__%6%6 CCC   B"++C HHY( ) ) ) GGC    2 2 2 2 2 25## #  " "B"++C HHY( ) ) ) GGC   GR[[]]2 2 GGG    Y__   )"2"2"4"4 4 44s5-A A,+A,(N  N N P<<QQ)Fr-)NNrFr)) __future__rrcolorsysrandomr$rsysrqt svg_colorsrr r r_LINE_TYPE_CHECKER _SIZE_CHECKERcompiler&_COLOR_CHECKER_NODE_TYPE_CHECKER _BOOL_CHECKERrDr__all__rrrrrKrgrsdictr0r1rr2r3r4r[rFrrrrms`N'&&&&&%%%%%% 11111111 ))),+,, rz,-- OOBB<< hhhii d d d~Q~Q~Q~Q'9Q'9Q }Q '9Q}Qq}Qq}Q$ 2 2 >>+=>>>??##xj//Qf.&2v*P1P1P1P1P1P1P1P1jsCsCsCsCsCsCsCsCj 9 9 9 9 9 9 9 9 6 6 6 6 6D 6 6 6'''ssss8!!!.0&*Y5Y5Y5Y5Y5Y5r