3 dk@sdZddlmZmZmZddlmZmZmZm Z m Z m Z ddl m Z ddlmZGdddeZdd Ze e fd d ZGd d d eZGdddZddZddZedkredS)z Graphically display a Tree. )IntVarMenuTk) BoxWidget CanvasFrame CanvasWidget OvalWidget ParenWidget TextWidget)Tree)in_idlec@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*S)+TreeSegmentWidgeta A canvas widget that displays a single segment of a hierarchical tree. Each ``TreeSegmentWidget`` connects a single "node widget" to a sequence of zero or more "subtree widgets". By default, the bottom of the node is connected to the top of each subtree by a single line. However, if the ``roof`` attribute is set, then a single triangular "roof" will connect the node to all of its children. Attributes: - ``roof``: What sort of connection to draw between the node and its subtrees. If ``roof`` is true, draw a single triangular "roof" over the subtrees. If ``roof`` is false, draw a line between each subtree and the node. Default value is false. - ``xspace``: The amount of horizontal space to leave between subtrees when managing this widget. Default value is 10. - ``yspace``: The amount of space to place between the node and its children when managing this widget. Default value is 15. - ``color``: The color of the lines connecting the node to its subtrees; and of the outline of the triangular roof. Default value is ``'#006060'``. - ``fill``: The fill color for the triangular roof. Default value is ``''`` (no fill). - ``width``: The width of the lines connecting the node to its subtrees; and of the outline of the triangular roof. Default value is 1. - ``orientation``: Determines whether the tree branches downwards or rightwards. Possible values are ``'horizontal'`` and ``'vertical'``. The default value is ``'vertical'`` (i.e., branch downwards). - ``draggable``: whether the widget can be dragged by the user. c s||_||_d|_d|_d|_d|_d|_fdd|D|_jddddd d |_ |j |x|D]}|j |qdWd|_ t j |f|d S) zI :type node: :type subtrees: list(CanvasWidgetI) r Fc s g|]}jddddddqS)rz#006060)fill) create_line).0c)canvas./tmp/pip-build-v9q4h5k9/nltk/nltk/draw/tree.py Psz.TreeSegmentWidget.__init__..hiddenz#006060)rstateoutlineN)_label _subtrees _horizontal_roof_xspace_yspace_ordered_linesZcreate_polygon_polygon_add_child_widget _managingr__init__)selfrlabelsubtreesattribssubtreer)rrr'@s  zTreeSegmentWidget.__init__cCs|j}|dkr~||_|jrLx|jD]}|j|ddq$W|j|jddn.x|jD]}|j|ddqTW|j|jddn\|dkr|dkrd|_n|dkrd |_ntd n(|d krx|jD]}|j||d qW|j|j|d nt|to|d d kr$|jt |d}|j||d n|dkr@|j|j|d n|dkr|j|j||ix||jD]}|j|||iqdWnZ|dkr|dkr||_ n|dkr||_ |j |j n |dkr||_ntj|||dS)Nroofr)rZnormal orientation horizontalverticalrz*orientation must be horizontal or verticalcolor)r)rrwidthxspaceyspaceordered)r4r5)rrr#Z itemconfigr$r ValueError isinstancetupleintr r!updaterr"r __setitem__)r(attrvaluerlrrrr<_sL           zTreeSegmentWidget.__setitem__cCs|dkr|jS|dkr(|jj|j|S|dkrB|jj|jdSt|trz|ddkrz|jt|d}|jj|dS|dkr|jS|d kr|j S|d kr|j rd Sd Sn|d kr|j St j ||SdS)Nr-r3r2rrr0rr4r5r.r/r1r6)rrZitemcgetr$r8r9r#r:r r!rr"r __getitem__)r(r=r?rrrr@s(zTreeSegmentWidget.__getitem__cCs|jS)N)r)r(rrrr)szTreeSegmentWidget.labelcCs|jddS)N)r)r(rrrr*szTreeSegmentWidget.subtreescCs,|j|j|j|||_|j|jdS)z2 Set the node label to ``label``. N)_remove_child_widgetrr%r;)r(r)rrr set_labels  zTreeSegmentWidget.set_labelcCs8|jj|}||j|<|j||j||j|dS)zC Replace the child ``oldchild`` with ``newchild``. N)rindexrAr%r;)r(ZoldchildZnewchildrCrrr replace_childs     zTreeSegmentWidget.replace_childcCsB|jj|}|j|=|j||jj|jj|j|jdS)N) rrCrArdeleter#popr;r)r(childrCrrr remove_childs   zTreeSegmentWidget.remove_childcCsL|j}|jj|||j||jj|jdddddd|j|jdS)Nrz#006060)r) rrinsertr%r#appendrr;r)r(rCrGrrrr insert_childs  zTreeSegmentWidget.insert_childcCs|jr|jgS|jSdS)N)rr$r#)r(rrr_tagsszTreeSegmentWidget._tagscCsbt|tr|jj}n|j}|jrB|d|d|ddfS|d|dd|dfSdS)Nrr0g@)r8r r)bboxr)r(rGrOrrr _subtree_tops  zTreeSegmentWidget._subtree_topcCsL|jj}|jr,|d|d|ddfS|d|dd|dfSdS)NrNr0rMg@r)rrOr)r(rOrrr _node_bottoms zTreeSegmentWidget._node_bottomc Cslt|jdkrdS|jjdkr$dS||jkr6|j}n|g}|jrT|j rT|j|}|j\}}|jdj\}}}}xV|jddD]D} | j} t|| d}t|| d}t || d}t || d}qW|j r|j j |j |||||||| n |j j |j |||||||| xP|D]H} |j\}}|j|jj| } |j| \} } |j j | ||| | qWdS)Nrr0rNrM)lenrrrOr"r&_maintain_orderrQminmaxrrZcoordsr$r#rCrP)r(rGZ need_updatenodexnodeyZxminZyminZxmaxymaxr,rOlineZsubtreexZsubtreeyrrr_updates6     zTreeSegmentWidget._updatecCs|jr|j|S|j|SdS)N)r_maintain_order_horizontal_maintain_order_vertical)r(rGrrrrS s z!TreeSegmentWidget._maintain_ordercCs|j\}}}}||jkrdxB|jD]8}|j\}}} } ||j|kr"|jd||j|q"W|jS|g} |jj|} ||j} xrt| dt|jD]Z}|j|j\}}} } | |kr|j|j| |d| | ||j7} | j |j|qW||j} xrt| dddD]^}|j|j\}}} } | | kr|j|j| | d| | ||j8} | j |j|qW|jj\}}} } | ||jkr|jjd||j| |j} | S)Nrr0r]) rOrrr!moverCr rangerRrJ)r(rGlefttoprightbotr,x1y1x2y2movedrCxirrrr\s8      z*TreeSegmentWidget._maintain_order_verticalcCs|j\}}}}||jkrbx@|jD]6}|j\}}} } ||j|kr"|j||j|q"W|jS|g} |jj|} ||j} xrt| dt|jD]Z}|j|j\}}} } | |kr|j|jd| || | ||j7} | j |j|qW||j} xrt| dddD]^}|j|j\}}} } | | kr|j|jd| | | | ||j8} | j |j|qW|jj\}}} } | ||jkr|jj||j| d|j} | S)Nr0rr]r]) rOrrr r^rCr!r_rRrJ)r(rGr`rarbrcr,rdrerfrgrhrCyrjrrrr[<s8      z,TreeSegmentWidget._maintain_order_horizontalc Cs|j\}}d}xZ|jD]P}|j}||d|j}||d}|j||||d|d|j7}qWd}x |jD]}||j|d7}qxW|t|j}x|jD]}|jd||qWdS)Nrr0rMg)rQrrOr r^r!rPrR) r(rVrWrkr, subtree_bboxdxdycenterrrr_manage_horizontalfs      z$TreeSegmentWidget._manage_horizontalc Cs|j\}}d}xZ|jD]P}|j}||d|j}||d}|j||||d|d|j7}qWd}x*|jD] }||j|dt|j7}qxWx|jD]}|j||dqWdS)Nrr0rNg)rQrrOr!r^r rPrR) r(rVrWrir,rmrornrprrr_manage_vertical|s       z"TreeSegmentWidget._manage_verticalcCs`d|_|j\}}t|jdkr$dS|jr4|jn|jx|jD]}|j|qDWd|_dS)NTrF)r&rQrRrrrqrrrZ)r(rVrWr,rrr_manages   zTreeSegmentWidget._managecCsd|jd|jdS)Nz [TreeSeg z: ])rr)r(rrr__repr__szTreeSegmentWidget.__repr__N)__name__ __module__ __qualname____doc__r'r<r@r)r*rBrDrHrKrLrPrQrZrSr\r[rqrrrsrurrrrr s* ,    )**r c s\t|trJ|jf}fdd|D} t|| fS|fSdS)Nc s"g|]}t|qSr)_tree_to_treeseg)rrG)r leaf_attribs loc_attribs make_leaf make_node node_attribs tree_attribsrrrs z$_tree_to_treeseg..)r8r r)r ) rtr~r}rrr{r|r)r*r)rr{r|r}r~rrrrzs  rzc Ksi}i}i}i}xt|jD]\} } | dddkrH| || dd<q| dddkrj| || dd<q| dddkr| || dd<q| dddkr| || dd<qtd| qWt||||||||S) aS Convert a Tree into a ``TreeSegmentWidget``. :param make_node: A ``CanvasWidget`` constructor or a function that creates ``CanvasWidgets``. ``make_node`` is used to convert the Tree's nodes into ``CanvasWidgets``. If no constructor is specified, then ``TextWidget`` will be used. :param make_leaf: A ``CanvasWidget`` constructor or a function that creates ``CanvasWidgets``. ``make_leaf`` is used to convert the Tree's leafs into ``CanvasWidgets``. If no constructor is specified, then ``TextWidget`` will be used. :param attribs: Attributes for the canvas widgets that make up the returned ``TreeSegmentWidget``. Any attribute beginning with ``'tree_'`` will be passed to all ``TreeSegmentWidgets`` (with the ``'tree_'`` prefix removed. Any attribute beginning with ``'node_'`` will be passed to all nodes. Any attribute beginning with ``'leaf_'`` will be passed to all leaves. And any attribute beginning with ``'loc_'`` will be passed to all text locations (for Trees). NZtree_node_leaf_loc_zBad attribute: %s)listitemsr7rz) rrr~r}r+rrr{r|keyr>rrrtree_to_treesegments.rc@seZdZdZeefddZddZddZd$d d Zd%d d Z d&d dZ d'ddZ d(ddZ d)ddZ ddZddZddZddZddZdd Zd!d"Zd#S)* TreeWidgeta A canvas widget that displays a single Tree. ``TreeWidget`` manages a group of ``TreeSegmentWidgets`` that are used to display a Tree. Attributes: - ``node_attr``: Sets the attribute ``attr`` on all of the node widgets for this ``TreeWidget``. - ``node_attr``: Sets the attribute ``attr`` on all of the leaf widgets for this ``TreeWidget``. - ``loc_attr``: Sets the attribute ``attr`` on all of the location widgets for this ``TreeWidget`` (if it was built from a Tree). Note that a location widget is a ``TextWidget``. - ``xspace``: The amount of horizontal space to leave between subtrees when managing this widget. Default value is 10. - ``yspace``: The amount of space to place between the node and its children when managing this widget. Default value is 15. - ``line_color``: The color of the lines connecting each expanded node to its subtrees. - ``roof_color``: The color of the outline of the triangular roof for collapsed trees. - ``roof_fill``: The fill color for the triangular roof for collapsed trees. - ``width`` - ``orientation``: Determines whether the tree branches downwards or rightwards. Possible values are ``'horizontal'`` and ``'vertical'``. The default value is ``'vertical'`` (i.e., branch downwards). - ``shapeable``: whether the subtrees can be independently dragged by the user. THIS property simply sets the ``DRAGGABLE`` property on all of the ``TreeWidget``'s tree segments. - ``draggable``: whether the widget can be dragged by the user. cKs||_||_||_i|_i|_ddi|_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ d|_i|_i|_i|_g|_g|_|j||f|j||f|_|j|jtj||f|dS) Nr2z#008000z#008080r0z#c0c0c0Frr1) _make_node _make_leafZ_tree _nodeattribs _leafattribs _locattribs _line_color _line_width _roof_color _roof_fill _shapeabler r! _orientationr"_keys_expanded_trees_collapsed_trees_nodes_leaves_make_collapsed_trees_make_expanded_tree_treesegr%rr')r(rrr~r}r+rrrr'&s0  zTreeWidget.__init__cGs |j|S)aF Return the ``TreeSegmentWidget`` for the specified subtree. :param path_to_tree: A list of indices i1, i2, ..., in, where the desired widget is the widget corresponding to ``tree.children()[i1].children()[i2]....children()[in]``. For the root, the path is ``()``. )r)r( path_to_treerrr expanded_treeIs zTreeWidget.expanded_treecGs |j|S)aF Return the ``TreeSegmentWidget`` for the specified subtree. :param path_to_tree: A list of indices i1, i2, ..., in, where the desired widget is the widget corresponding to ``tree.children()[i1].children()[i2]....children()[in]``. For the root, the path is ``()``. )r)r(rrrrcollapsed_treeTs zTreeWidget.collapsed_treer0cCsLx"t|jjD]}|j||qWx"t|jjD]}|j||q4WdS)z5 Add a binding to all tree segments. N)rrvalues bind_clickr)r(callbackbuttontsegrrrbind_click_trees_szTreeWidget.bind_click_treescCsLx"t|jjD]}|j||qWx"t|jjD]}|j||q4WdS)z5 Add a binding to all tree segments. N)rrr bind_dragr)r(rrrrrrbind_drag_treeshszTreeWidget.bind_drag_treescCs<x|jD]}|j||qWx|jD]}|j||q$WdS)z. Add a binding to all leaves. N)rr)r(rrleafrrrbind_click_leavesqs  zTreeWidget.bind_click_leavescCs<x|jD]}|j||qWx|jD]}|j||q$WdS)z. Add a binding to all leaves. N)rr)r(rrrrrrbind_drag_leaveszs  zTreeWidget.bind_drag_leavescCs<x|jD]}|j||qWx|jD]}|j||q$WdS)z- Add a binding to all nodes. N)rr)r(rrnoderrrbind_click_nodess  zTreeWidget.bind_click_nodescCs<x|jD]}|j||qWx|jD]}|j||q$WdS)z- Add a binding to all nodes. N)rr)r(rrrrrrbind_drag_nodess  zTreeWidget.bind_drag_nodesc st|tsdSj}j||jfj}jj|fdd|jD}j |7_ t ||dj j j d}|j|<|j|<|jx0tt|D] }||} j| ||fqWdS)Ncsg|]}|fjqSr)r)rr?)rr}r(rrrsz4TreeWidget._make_collapsed_trees..r0)r-r2rr3)r8r rrr)rrrJleavesrr rrrrrhider_rRr) r(rrrr~rrtreesegrjrGr)rr}r(rrs,     z TreeWidget._make_collapsed_treesc sj}j}t|tr||jfj}jj||fddtt D}t ||j j d}|j <j|<|S||fj} jj| | SdS)Ncs$g|]}j||fqSr)r)rrj)rchildrenrr(rrrsz2TreeWidget._make_expanded_tree..)r2r3)rrr8r r)rrrJr_rRr rrrrrr) r(rrrr~r}rr*rrr)rrrr(rrs      zTreeWidget._make_expanded_treecCsN|dddkr4x|jD]}|||dd<qWn|dddkrhx|jD]}|||dd<qLWn|dkr||_xt|jjD] }||d<qWn|dkr||_xt|jjD] }||d<qWxt|jjD] }||d<qWn^|dkr"||_x t|jjD]}||d<q Wn(|d krX||_ x t|jjD]}||d <qBWn|d kr||_ x t|jjD]}||d <qxWx t|jjD]}||d <qWx|jD]}||d <qWn|d kr*||_ x t|jjD]}||d <qWx t|jjD]}||d <q W|j n |dkr||_ x t|jjD]}||d<qJWx t|jjD]}||d<qlW|j n|dkr||_x t|jjD]}||d<qWx t|jjD]}||d<qW|j nd|dkr<||_x t|jjD]}||d<qWx0t|jjD]}||d<q(Wntj|||dS)Nrrr line_colorr2 line_widthr3 roof_color roof_fillr shapeable draggabler4r5r.r6)rrrrrrrrrrrr manager!rr"rr<)r(r=r>rrrrrrr<sx              zTreeWidget.__setitem__cCs|dddkr&|jj|dddS|dddkrL|jj|dddS|dddkrr|jj|dddS|dkr|jS|dkr|jS|dkr|jS|d kr|jS|d kr|jS|d kr|j S|d kr|j S|d kr|j St j ||SdS)Nrrrrrrrrrrr4r5r.)rgetrrrrrrrr r!rrr@)r(r=rrrr@ s.zTreeWidget.__getitem__cCsgS)Nr)r(rrrrL#szTreeWidget._tagscCsNt|jjt|jj}x,|D]$}|jr"|j|j|jq"WdS)N)rrrrrshowrr)r(Zsegsrrrrrs&s zTreeWidget._managecCs|}|dr|j|j|}n|j|j|}|j|krV|j||j|||_n|jj|||j|j j dd\}}|j j dd\}}|j |||||j |jj |dS)z) Collapse/expand a tree. r-NrN)rrrparentrAr%rrDrr)rOr^rr;)r(rZ old_treesegZ new_treesegZnewxZnewyoldxoldyrrrtoggle_collapsed0s   zTreeWidget.toggle_collapsedN)r0)r0)r0)r0)r0)r0)rvrwrxryr r'rrrrrrrrrrr<r@rLrsrrrrrrs"'"  ? rc@s<eZdZddZddZddZddZd d Zd d Zd S)TreeViewc GsFddlm}m}||_t|_|jjd|jjd|j|jjd|jt |j}|_ |jjd|j j t |j|_ |j jdd|j j d f}d|j j f}t||t||_g|_x^tt|D]N}t|j|||d d d d d d|d }|j|j|jj||j|ddqW|j|j jddd|jdS)Nr)ceilsqrtZNLTKz z z helveticaboldz#008040z#004080z#004040whiter0) node_font leaf_color node_colorrrrr leaf_fontZboth)expandr)mathrrZ_treesr_toptitlebinddestroyr_cframe print_to_filer_sizesetrr:rR_width_widgetsr_rrrrrJ add_widget_layoutpack _init_menubar) r(treesrrcfrhelvrjwidgetrrrr'Vs@     zTreeView.__init__c Csd}}}}|j}xtt|jD]p}|j|}|jdd\}}||dkr\|}d}|j|||||jdd}t||jdd}q&WdS)NrrNrrM)rr_rRrrOr^rU) r(rjrirkrXr3rrrrrrrs  zTreeView._layoutcCst|j}t|dd}|jdd|jjdd|jdd|jdd|jd d|d t|dd}|jd |jdd |j d |jd|jdd|j d |jd|jdd|j d |jd|jdd|j d |jd|jdd|j d |jdd|d |jj |ddS)Nr)ZtearoffzPrint to PostscriptzCtrl-p)r) underlinecommandZ acceleratorZExitr0zCtrl-xZFile)r)rmenuZTinyr)r)variablerr>rZSmallrZMediumZLargeZHuge2ZZoom)r) rrZ add_commandrrrZ add_cascadeZadd_radiobuttonrresizeconfig)r(ZmenubarZfilemenuZzoommenurrrrsV        zTreeView._init_menubarcGsd|jj df}d|jj f}|jj}|jj}xf|jD]\}||d<||d<||d<||d<|jjdkrzd|d <q>|jjd krd |d <q>d |d <q>W|jdS) Nrrrrr4r5rlr0rrNrM)rrrr)r(errr4r5rrrrrs      zTreeView.resizecGs"|jdkrdS|jjd|_dS)N)rr)r(rrrrrs  zTreeView.destroycOstr dS|jj||dS)z Enter the Tkinter mainloop. This function must be called if this demo is created from a non-interactive program (e.g. from a secript); otherwise, the demo will close as soon as the script completes. N)r rmainloop)r(argskwargsrrrrszTreeView.mainloopN) rvrwrxr'rrrrrrrrrrUs + 7rcGst|jdS)zd Open a new window containing a graphical diagram of the given trees. :rtype: None N)rr)rrrr draw_treess rc s ddlfdd}tdddd}tjd}t|j|d d-d/ddddd }|j|dddd}dd}tjd}t|j|||d d}fdd}fdd} |j|j|j|j|j |d|j d j | d|j j | dt |j|} |j| |j dddtjd } t|j| dddd!} d | d"<|j| d|j ddd#d$} d%}t|j|j}t|j|dd d&}|j|| j dd|j ddtjd'}t|j|d d(d(d1d)d*d+}|j|j|| j dd|j dd|j | |j|jd|jdS)2Nrcsdjdd|d<dS)Nz#%06dri?Br)randint)Zcw)randomrrrszdemo..filli&irN)r3heightZ closeenoughzU (S (NP the very big cat) (VP (Adv sorta) (V saw) (NP (Det the) (N dog))))r0rrrritalicrZblackZgreen4Zblue2)rrrrrrrrcSsd}t|t|||dddS) Nrr)fontZgreen)r)rrr)rr )rtextbigrrrboxitszdemo..boxitcSst|t||ddS)NZcyan)r)rr )rrrrrovalitszdemo..ovalitz/(S (NP this tree) (VP (V is) (AdjP shapeable))))rcsdjdd|d<dS)Nz#%04d00ri'r2)r)r)rrrr2szdemo..colorcs,djdd|jd<d|jjd<dS)Nz#%06dri'rrr2)rr)rG)r)rrrcolor2!szdemo..color2rMzb (S (NP this tree) (AUX was) (VP (V built) (PP (P with) (NP (N tree_to_treesegment))))))Z tree_colorZ tree_xspaceZ tree_widthrcSs|ddkrx|jddjdjd|jddjdjd|jdjdjd|jjdjdd|d<nj|jddjdjd|jddjdjd|jdjdjd|jjdjdd|d<dS)Nr.r/r0rr1rM)rr*Zset_textr)Z treewidgetrrr orientswitch9s  zdemo..orientswitchaM Try clicking, right clicking, and dragging different elements of each of the trees. The top-left tree is a TreeWidget built from a Tree. The top-right is a TreeWidget built from a Tree, using non-default widget constructors for the nodes & leaves (BoxWidget and OvalWidget). The bottom-left tree is built from tree_to_treesegment.)rrz/(S (NP this tree) (VP (V is) (Adj horizontal)))Zbrown2Zbrown4r/)rrrrrr.)rrr)rrr)rrr)rrr fromstringrrrrrrrrr rOrr striprrr)rrrZtcrrZtreetokZtc2r2rZparenZtree3Ztc3rrZtwidgetZtextboxZtree4Ztc4r)rrdemosn       & & r__main__N)rytkinterrrrZnltk.draw.utilrrrrr r Z nltk.treer Z nltk.utilr r rzrrrrrrvrrrr s&    8Zo