�
�0�U�" � @ s d Z Gd d � d � Z d S)zt
GSC Software: Tree class
Author: Nick Becker
Department of Cognitive Science, Johns Hopkins University
Summer 2015
c @ sv e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d d � Z d S)�TreezN
The last tree class you will ever need. Currently under development.
c C sK | j | � | _ g | _ | j | j d � g | _ | j | j � d S)a�
Create an internal representation of a tree by passing in a text version of it.
Formatting rules apply: trees must be passed to this method in the form
A (B A (D E))
In other words, parentheses enclose a node's children, and sister nodes are
separated by whitespace. Parentheses were chosen over brackets to avoid any
confusion when entering HNF trees, which contain many bracketed symbols.
Any symbols can be used in node labels EXCEPT parentheses, commas, and
percent signs (%).
Calling this method automatically sets the recursive-style filler-role bindings
for each node, as well as the span-style filler-role bindings for each node.
�rN)�treeStringToList�treeList�recursiveFRbindings�setRecursiveRoles�spanFRbindings� setSpanRoles)�self�
treeString� r �;/projects/e0c271a6-81f7-42a7-9681-fa02642c85a5/code/tree.py�__init__ s
z
Tree.__init__c C s� | j d d � j � } d } x� | t | � k r| | } t | � d k r
| d d k so | d d k r� | d d � | | <| j | | d � | d 7} | | } n | d d k s� | d d k r
| d d
� | | <| j | d | d � | d 8} q
n | d 7} q! Wd } x� | t | � d k r�| | } | | d } | d k r| d k r| d k r| d k s�| d k r�| d k r�| d k r�| j | d d � | d 7} n | d 7} q!W| S) zs
Check formatting of the input string; if it is valid, prepare it for internal
operations.
�,� � � �(�)N�����r r r )�replace�split�len�insert)r � inputString�niceList�i�
currentSymbol�
nextSymbolr r r r ) s2
0$
zTree.treeStringToListc C s� t | � d k r� | d d k r5 t d � | } n~ | d d k rv t t | d � d � | d d � } | } n= | d d k r� | d d � } n | j j | d | f � | j | d d � | � n� d } x� | r}d } x� t t | j � d � D]w } t | j | d � t | j | d d � k r� | j | } | j | d | j | <| | j | d <d } q� q� Wq� Wd S) zE
Recursive function to populate recursiveFRbindings.
r r r r Nr TF)r �str�intr �appendr �range)r � treeNodes�level�newLevel� needSwappedr �tempr r r r P s& ( 2
zTree.setRecursiveRolesc C s� | j | � } d } i } xg t t | � � D]S } t | | d � d k r. t | � t | d � | | | d <| d 7} q. q. Wd } x� | s�x� t t | � � D]� } t | | d � d k r� d } d } xK | | d D]; } | | k r| | }
| |
d |
d } q� d } q� W| scd j t t | � d | j �� } | | | | d <qcq� q� Wt | � t | � k r� d } q� q� Wt | � d g } xE | j � D]7 } | j d � }
|
d | | f | t
|
d � <q�W| | _ d S)
a#
A dumb, iterative function to populate spanFRbindings. Once again, probably
bad programming practice to populate a variable like this instead of returning
a value.
A nice recursive algorithm to do this task is certainly possible and would
be far more elegant, but it turned out not to be as straighforward as I initially
thought... hence the inefficient iterative algorithm here. Luckily, this
shouldn't be a problem because most users will be looking at very small
trees.
r r Fr T�keyN�%r ) � findSubTreesr! r r �join�sorted�set�index�keysr r r )r r �subTrees�spanBase�spanDictr �done�span� invalidSpan�child� childSpan� tempBindingsr' �splitKeyr r r r n s8 &
$
&zTree.setSpanRolesc C s� d } xs t t | � d � D][ } | | d k r | | d k r | | d k r | | d t | � 7<| d 7} q q Wg } x:t t | � d � D]"} | | d k r� | | d k r� | | d k r� g } | d } | | d k } | r�d } | d 7} x� | t | � k r�| d k r�| | d k rC| d 7} nM | | d k r`| d 8} n0 | | d k r�| d k r�| j | | � n | d 7} qWn | j | | | f � q� q� W| S) zV
Given a tree in list form, find the immediate children of each node.
r r r r r r( r r )r! r r r ) r r �idNor r/ �currentChildren�j� hasChildren�embeddedr r r r) � s0 0 0
!
zTree.findSubTreesc C s | j S)N)r )r r r r �
getFRbindings� s zTree.getFRbindingsc C s� d } x� t t | j � � D]m } | d k r; | d 7} n | | j | d d | j | d 7} | t | j � d k r | d 7} q q W| d 7} | S)zT
Gets a pretty string for the recursive-style filler-role bindings.
�{r � �/r z;
z}
)r! r r )r � returnStringr r r r �recursiveFRtoString� s
(
zTree.recursiveFRtoStringc C s� d } x� t t | j � � D]m } | d k r; | d 7} n | | j | d d | j | d 7} | t | j � d k r | d 7} q q W| d 7} | S)zO
Gets a pretty string for the span-style filler-role bindings.
r? r r@ rA r z;
z}
)r! r r )r rB r r r r �spanFRtoString� s
(
zTree.spanFRtoStringN) �__name__�
__module__� __qualname__�__doc__r
r r r r) r> rC rD r r r r r s ' 2 $ r N)rH r r r r r �<module> s 