treat 0.1.1

data/lib/treat/resources/tags.rb

@@ -0,0 +1,393 @@
+# encoding: UTF-8
+
+module Treat
+  module Resources
+    class Tags
+
+      ClawsC5 = 0
+      Brown = 1
+      Penn = 2
+      Enju = 3
+
+      PTBWordTagToCategory = {
+        'CC' =>  :conjunction, # Coordinating conjunction
+        'CD' => :number, # Cardinal number
+        'DT' => :determiner, # Determiner
+        'DET' => :determiner, # Determiner
+        'EX' => :determiner, # Existential there
+        'FW' =>  :foreign, # Foreign word
+        'IN' =>  :preposition, # Preposition or subordinating conjunction
+        'JJ' =>  :adjective, # Adjective
+        'JJR' => :adjective, # Adjective, comparative
+        'JJS' => :adjective, # Adjective, superlative
+        'LS' =>  :list, # List item marker
+        'MD' =>  :modal, # Modal
+        'NN' =>  :noun, # Noun, singular or mass
+        'NNS' =>  :noun, # Noun, plural
+        'NNP' =>  :noun, # Proper noun, singular
+        'NNPS' =>  :noun, # Proper noun, plural
+        'PDT' =>  :determiner, # Predeterminer
+        'POS' =>  :determiner, # Possessive ending
+        'PRP' =>  :pronoun, # Personal pronoun
+        'PRP$' =>  :pronoun, # Possessive pronoun,
+        'PRPS' =>  :determiner, # Possessive determiner
+        'RB' =>  :adverb, # Adverb
+        'RBR' =>  :adverb, # Adverb, comparative
+        'RBS' =>  :adverb, # Adverb, superlative
+        'RP' =>  :particle, # Particle
+        'SYM' =>  :symbol, # Symbol
+        'TO' =>  :to, # to
+        'UH' =>  :interjection, # Interjection
+        'VB' =>  :verb, # Verb, base form
+        'VBD' =>  :verb, # Verb, past tense
+        'VBG' =>  :verb, # Verb, gerund or present participle
+        'VBN' =>  :verb, # Verb, past participle
+        'VBP' =>  :verb, # Verb, non-3rd person singular present
+        'VBZ' =>  :verb, # Verb, 3rd person singular present
+        'WDT' =>  :determiner, # Wh-determiner
+        'WP' =>  :pronoun, # Wh-pronoun
+        'WP$' =>  :pronoun, # Possessive wh-pronoun
+        'WRB' =>  :adverb, # Wh-adverb
+        ')' => :punctuation, # Right bracket
+        '(' => :punctuation,  # Left bracket
+        '.' => :punctuation, # Period
+        '\'\'' => :symbol, # Quote
+        ',' => :punctuation,
+        ';' => :punctuation
+      }
+
+      PTBClauseTagDescription = [
+        ['S', 'Simple declarative clause'],
+        ['SBAR', 'Clause introduced by a (possibly empty) subordinating conjunction'],
+        ['SBARQ', 'Direct question introduced by a wh-word or a wh-phrase'],
+        ['SINV', 'Inverted declarative sentence'],
+        ['SQ', 'Inverted yes/no question']
+      ]
+
+      PTBPhraseTagDescription = [
+        ['ADJP', 'Adjective phrase'],
+        ['ADVP', 'Adverb phrase'],
+        ['CONJP', 'Conjunction phrase'],
+        ['FRAG', 'Fragment'],
+        ['INTJ', 'Interjection'],
+        ['LST', 'List marker'],
+        ['NAC', 'Not a constituent'],
+        ['NP', 'Noun phrase'],
+        ['NX', 'Head of an NP'],
+        ['PP', 'Prepositional phrase'],
+        ['PRN', 'Parenthetical'],
+        ['PRT', 'Particle'],
+        ['QP', 'Quantifier phrase'],
+        ['RRC', 'Reduced relative clause'],
+        ['UCP', 'Unlike coordinated phrase'],
+        ['VP', 'Verb phrase'],
+        ['WHADJP', 'Wh-adjective phrase'],
+        ['WHAVP', 'Wh-adverb phrase'],
+        ['WHNP', 'Wh-noun phrase'],
+        ['WHPP', 'Wh-prepositional phrase'],
+        ['X', 'Unknown, uncertain, or unbracketable']
+      ]
+
+      PTBWordTagDescription = [
+=begin
+CC - Coordinating conjunction
+CD - Cardinal number
+DT - Determiner
+EX - Existential there
+FW - Foreign word
+IN - Preposition or subordinating conjunction
+JJ - Adjective
+JJR - Adjective, comparative
+JJS - Adjective, superlative
+LS - List item marker
+MD - Modal
+NN - Noun, singular or mass
+NNS - Noun, plural
+NNP - Proper noun, singular
+NNPS - Proper noun, plural
+PDT - Predeterminer
+POS - Possessive ending
+PRP - Personal pronoun
+PRP$ - Possessive pronoun (prolog version PRP-S)
+RB - Adverb
+RBR - Adverb, comparative
+RBS - Adverb, superlative
+RP - Particle
+SYM - Symbol
+TO - to
+UH - Interjection
+VB - Verb, base form
+VBD - Verb, past tense
+VBG - Verb, gerund or present participle
+VBN - Verb, past participle
+VBP - Verb, non-3rd person singular present
+VBZ - Verb, 3rd person singular present
+WDT - Wh-determiner
+WP - Wh-pronoun
+WP$ - Possessive wh-pronoun (prolog version WP-S)
+WRB - Wh-adverb
+
+=end  
+      ]
+      BrownWordTagDescription = [
+=begin
+
+Tag	Description	Examples
+
+.	sentence closer	. ; ? !
+(	left paren	 
+)	right paren	 
+*	not, n't	 
+--	dash	 
+,	comma	 
+:	colon	 
+ABL	pre-qualifier	quite, rather
+ABN	pre-quantifier	half, all
+ABX	pre-quantifier	both
+AP	post-determiner	many, several, next
+AT	article	a, the, no
+BE	be	 
+BED	were	 
+BEDZ	was	 
+BEG	being	 
+BEM	am	 
+BEN	been	 
+BER	are, art	 
+BEZ	is	 
+CC	coordinating conjunction	and, or
+CD	cardinal numeral	one, two, 2, etc.
+CS	subordinating conjunction	if, although
+DO	do	 
+DOD	did	 
+DOZ	does	 
+DT	singular determiner	this, that
+DTI	singular or plural determiner/quantifier	some, any
+DTS	plural determiner	these, those
+DTX	determiner/double conjunction	either
+EX	existentil there	 
+FW	foreign word (hyphenated before regular tag)	 
+HL	word occurring in headline (hyphenated after regular tag)	 
+HV	have	 
+HVD	had (past tense)	 
+HVG	having	 
+HVN	had (past participle)	 
+HVZ	has	 
+IN	preposition	 
+JJ	adjective	 
+JJR	comparative adjective	 
+JJS	semantically superlative adjective	 chief, top
+JJT	morphologically superlative adjective	biggest
+MD	modal auxiliary	can, should, will
+NC	cited word (hyphenated after regular tag)	 
+NN	singular or mass noun	 
+NN$	possessive singular noun	 
+NNS	plural noun	 
+NNS$	possessive plural noun	 
+NP	proper noun or part of name phrase	 
+NP$	possessive proper noun	 
+NPS	plural proper noun	 
+NPS$	possessive plural proper noun	 
+NR	adverbial noun	home, today, west
+NRS	plural adverbial noun
+OD	ordinal numeral	first, 2nd
+PN	nominal pronoun	everybody, nothing
+PN$	possessive nominal pronoun	 
+PP$	possessive personal pronoun	my, our
+PP$$	second (nominal) possessive pronoun	mine, ours
+PPL	singular reflexive/intensive personal pronoun	myself
+PPLS	plural reflexive/intensive personal pronoun	ourselves
+PPO	objective personal pronoun	me, him, it, them
+PPS	3rd. singular nominative pronoun	he, she, it, one
+PPSS	other nominative personal pronoun	I, we, they, you
+QL	qualifier	very, fairly
+QLP	post-qualifier	enough, indeed
+RB	adverb	 
+RBR	comparative adverb	 
+RBT	superlative adverb	 
+RN	nominal adverb	here then, indoors	 
+RP	adverb/particle	about, off, up
+TL	word occurring in title (hyphenated after	 
+ 	regular tag)	 
+TO	infinitive marker to	 
+UH	interjection, exclamation	 
+VB	verb, base form	 
+VBD	verb, past tense	 
+VBG	verb, present participle/gerund	 
+VBN	verb, past participle	 
+VBZ	verb, 3rd. singular present	 
+WDT	wh- determiner	what, which
+WP$	possessive wh- pronoun	whose
+WPO	objective wh- pronoun	whom, which, that
+WPS	nominative wh- pronoun	who, which, that
+WQL	wh- qualifier	how
+WRB	wh- adverb	how, where, when
+
+=end    
+      ]
+      EnjuCatDescription = [
+        ['ADJ',	'Adjective'],
+        ['ADV',	'Adverb'],
+        ['CONJ',	'Coordination conjunction'],
+        ['C',	'Complementizer'],
+        ['D',	'Determiner'],
+        ['N',	'Noun'],
+        ['P',	'Preposition'],
+        ['SC',	'Subordination conjunction'],
+        ['V',	'Verb'],
+        ['COOD',	'Part of coordination'],
+        ['PN',	'Punctuation'],
+        ['PRT',	'Particle'],
+        ['S',	'Sentence']
+      ]
+
+      # Description of the xcat in the Enju output specification.
+      EnjuXCatDescription = [
+        ['COOD',	'Coordinated phrase/clause'],
+        ['IMP',	'Imperative sentence'],
+        ['INV',	'Subject-verb inversion'],
+        ['Q',	'Interrogative sentence with subject-verb inversion'],
+        ['REL',	'A relativizer included'],
+        ['FREL', 'A free relative included'],
+        ['TRACE',	'A trace included'],
+        ['WH', 'A wh-question word included']
+      ]
+
+      EnjuCatXcatToPTB = [
+        ['ADJP', '', 'ADJP'],
+        ['ADJP', 'REL', 'WHADJP'],
+        ['ADJP', 'FREL', 'WHADJP'],
+        ['ADJP', 'WH', 'WHADJP'],
+        ['ADVP', '', 'ADVP'],
+        ['ADVP', 'REL', 'WHADVP'],
+        ['ADVP', 'FREL', 'WHADVP'],
+        ['ADVP', 'WH', 'WHADVP'],
+        ['CONJP', '', 'CONJP'],
+        ['CP', '', 'SBAR'],
+        ['DP', '', 'NP'],
+        ['NP', '', 'NP'],
+        ['NX', 'NX', 'NAC'],
+        ['NP'	'REL'	'WHNP'],
+        ['NP'	'FREL'	'WHNP'],
+        ['NP'	'WH'	'WHNP'],
+        ['PP', '', 'PP'],
+        ['PP', 'REL', 'WHPP'],
+        ['PP', 'WH', 'WHPP'],
+        ['PRT', '', 'PRT'],
+        ['S', '', 'S'],
+        ['S', 'INV', 'SINV'],
+        ['S', 'Q', 'SQ'],
+        ['S', 'REL', 'SBAR'],
+        ['S', 'FREL', 'SBAR'],
+        ['S', 'WH', 'SBARQ'],
+        ['SCP', '', 'SBAR'],
+        ['VP', '', 'VP'],
+        ['VP', '', 'VP'],
+        ['', '', 'UK']
+      ]
+
+      # Aligned tags for the Claws C5, Brown and Penn tag sets.
+      # Adapted from Manning, Christopher and Schütze, Hinrich,
+      # 1999. Foundations of Statistical Natural Language
+      # Processing. MIT Press, p. 141-142.
+      AlignedWordTags = [
+        'Adjective', ['AJ0', 'JJ', 'JJ'],
+        'Adjective, ordinal number', ['ORD', 'OD', 'JJ'],
+        'Adjective, comparative', ['AJC', 'JJR', 'JJR'],
+        'Adjective, superlative', ['AJS', 'JJT', 'JJS'],
+        'Adjective, superlative, semantically', ['AJ0', 'JJS', 'JJ'],
+        'Adjective, cardinal number', ['CRD', 'CD', 'CD'],
+        'Adjective, cardinal number, one', ['PNI', 'CD', 'CD'],
+        'Adverb', ['AV0', 'RB', 'RB'],
+        'Adverb, negative', ['XX0', '*', 'RB'],
+        'Adverb, comparative', ['AV0', 'RBR', 'RBR'],
+        'Adverb, superlative', ['AV0', 'RBT', 'RBS'],
+        'Adverb, particle', ['AVP', 'RP', 'RP'],
+        'Adverb, question', ['AVQ', 'WRB', 'WRB'],
+        'Adverb, degree & question', ['AVQ', 'WQL', 'WRB'],
+        'Adverb, degree', ['AV0', 'QL', 'RB'],
+        'Adverb, degree, postposed', ['AV0', 'QLP', 'RB'],
+        'Adverb, nominal', ['AV0', 'RN', 'RB'],
+        'Conjunction, coordination', ['CJC', 'CC', 'CC'],
+        'Conjunction, subordination', ['CJS', 'CS', 'IN'],
+        'Conjunction, complementizer, that', ['CJT', 'CS', 'IN'],
+        'Determiner', ['DT0', 'DT', 'DT'],
+        'Determiner, pronoun', ['DT0', 'DTI', 'DT'],
+        'Determiner, pronoun, plural', ['DT0', 'DTS', 'DT'],
+        'Determiner, prequalifier', ['DT0', 'ABL', 'DT'],
+        'Determiner, prequantifier', ['DT0', 'ABN', 'PDT'],
+        'Determiner, pronoun or double conjunction', ['DT0', 'ABX', 'PDT'],
+        'Determiner, pronoun or double conjunction', ['DT0', 'DTX', 'DT'],
+        'Determiner, article', ['AT0', 'AT', 'DT'],
+        'Determiner, postdeterminer', ['DT0', 'AP', 'JJ'],
+        'Determiner, possessive', ['DPS', 'PP$', 'PRP$'],
+        'Determiner, possessive, second', ['DPS', 'PP$$', 'PRP'],
+        'Determiner, question', ['DTQ', 'WDT', 'WDT'],
+        'Determiner, possessive & question', ['DTQ', 'WP$', 'WP$'],
+        'Noun', ['NN0', 'NN', 'NN'],
+        'Noun, singular', ['NN1', 'NN', 'NN'],
+        'Noun, plural', ['NN2', 'NNS', 'NNS'],
+        'Noun, proper, singular', ['NP0', 'NP', 'NNP'],
+        'Noun, proper, plural', ['NP0', 'NPS', 'NNPS'],
+        'Noun, adverbial', ['NN0', 'NR', 'NN'],
+        'Noun, adverbial, plural', ['NN2', 'NRS', 'NNS'],
+        'Pronoun, nominal (indefinite)', ['PNI', 'PN', 'NN'],
+        'Pronoun, personal, subject', ['PNP', 'PPSS', 'PRP'],
+        'Pronoun, personal, subject, 3SG', ['PNP', 'PPS', 'PRP'],
+        'Pronoun, personal, object', ['PNP', 'PPO', 'PRP'],
+        'Pronoun, reflexive', ['PNX', 'PPL', 'PRP'],
+        'Pronoun, reflexive, plural', ['PNX', 'PPLS', 'PRP'],
+        'Pronoun, question, subject', ['PNQ', 'WPS', 'WP'],
+        'Pronoun, question, object', ['PNQ', 'WPO', 'WP'],
+        'Pronoun, existential there', ['EX0', 'EX', 'EX'],
+        'Verb, base present form (not infinitive)', ['VVB', 'VB', 'VBP'],
+        'Verb, infinitive', ['VVI', 'VB', 'VB'],
+        'Verb, past tense', ['VVD', 'VBD', 'VBD'],
+        'Verb, present participle', ['VVG', 'VBG', 'VBG'],
+        'Verb, past/passive participle', ['VVN', 'VBN', 'VBN'],
+        'Verb, present, 3SG, -s form', ['VVZ', 'VBZ', 'VBZ'],
+        'Verb, auxiliary do, base', ['VDB', 'DO', 'VBP'],
+        'Verb, auxiliary do, infinitive', ['VDB', 'DO', 'VB'],
+        'Verb, auxiliary do, past', ['VDD', 'DOD', 'VBD'],
+        'Verb, auxiliary do, present participle', ['VDG', 'VBG', 'VBG'],
+        'Verb, auxiliary do, past participle', ['VDN', 'VBN', 'VBN'],
+        'Verb, auxiliary do, present 3SG', ['VDZ', 'DOZ', 'VBZ'],
+        'Verb, auxiliary have, base', ['VHB', 'HV', 'VBP'],
+        'Verb, auxiliary have, infinitive', ['VHI', 'HV', 'VB'],
+        'Verb, auxiliary have, past', ['VHD', 'HVD', 'VBD'],
+        'Verb, auxiliary have, present participle', ['VHG', 'HVG', 'VBG'],
+        'Verb, auxiliary have, past participle', ['VHN', 'HVN', 'VBN'],
+        'Verb, auxiliary have, present 3SG', ['VHZ', 'HVZ', 'VBZ'],
+        'Verb, auxiliary be, infinitive', ['VBI', 'BE', 'VB'],
+        'Verb, auxiliary be, past', ['VBD', 'BED', 'VBD'],
+        'Verb, auxiliary be, past, 3SG', ['VBD', 'BEDZ', 'VBD'],
+        'Verb, auxiliary be, present participle', ['VBG', 'BEG', 'VBG'],
+        'Verb, auxiliary be, past participle', ['VBN', 'BEN', 'VBN'],
+        'Verb, auxiliary be, present, 3SG', ['VBZ', 'BEZ', 'VBZ'],
+        'Verb, auxiliary be, present, 1SG', ['VBB', 'BEM', 'VBP'],
+        'Verb, auxiliary be, present', ['VBB', 'BER', 'VBP'],
+        'Verb, modal', ['VM0', 'MD', 'MD'],
+        'Preposition, to as infinitive marker', ['TO0', 'TO', 'TO'],
+        'Preposition, to', ['PRP', 'IN', 'TO'],
+        'Preposition', ['PRP', 'IN', 'IN'],
+        'Preposition, of', ['PRF', 'IN', 'IN'],
+        'Possessive', ['POS', '$', 'POS'],
+        'Interjection (or other isolate)', ['ITJ', 'UH', 'UH'],
+        'Punctuation, sentence ender', ['PUN', '.', '.'],
+        'Punctuation, semicolon', ['PUN', '.', '.'],
+        'Puncutation, colon or ellipsis', ['PUN', ':', ':'],
+        'Punctuationm, comma', ['PUN', ',', ','],
+        'Punctuation, dash', ['PUN', '-', '-'],
+        'Punctuation, dollar sign', ['PUN', '', '$'],
+        'Punctuation, left bracket', ['PUL', '(', '('],
+        'Punctuation, right bracket', ['PUR', ')', ')'],
+        'Punctuation, quotation mark, left', ['PUQ', '', '``'],
+        'Punctuation, quotation mark, right', ['PUQ', '', '"'],
+        'Unknown, foreign words (not in English lexicon)', ['UNZ', '(FW-)', 'FW'],
+        'Symbol', ['', '', 'SYM'],
+        'Symbol, alphabetical', ['ZZ0', '', ''],
+        'Symbol, list item', ['', '', 'LS']
+      ]
+
+    end
+  end
+end

data/lib/treat/sugar.rb

@@ -0,0 +1,43 @@
+module Treat
+  module Sugar
+    def edulcorate
+      return if @@edulcorated
+      @@edulcorated = true
+      each_entity_class do |type, klass|
+        unless type == :Symbol
+          Object.class_eval do
+            define_method(type) do |value='',id=nil|
+              klass.build(value, id)
+            end
+          end
+        end
+      end
+    end
+    def unedulcorate
+      return unless @@edulcorated
+      @@edulcorated = false
+      each_entity_class do |type, klass| 
+        unless type == :Symbol
+          Object.class_eval do
+             remove_method(type)
+          end
+        end
+      end
+    end
+    # Whtypeher syntactic sugar is
+    # enabled or not.
+    def edulcorated?; @@edulcorated; end
+    # Syntactic sugar is disabled by default.
+    @@edulcorated = false
+
+    private
+    
+    def each_entity_class
+      Treat::Entities.list.each do |entity_type|
+        type = :"#{cc(entity_type)}"
+        klass = Treat::Entities.const_get(type, klass)
+        yield type, klass
+      end
+    end
+  end
+end

data/lib/treat/tree.rb

@@ -0,0 +1,174 @@
+module Treat
+  # This module provides an abstract tree structure with
+  # nodes having an id, a value, children, features and edges.
+  module Tree
+    # This class models the nodes for an N-ary tree data structue
+    # with unique identifiers, text value, children, features
+    # (annotations) and edges. 
+    #
+    # This class was tightly based on the 'rubytree' gem.
+    # RubyTree is licensed under the BSD license and can
+    # be found at http://rubytree.rubyforge.org/rdoc/.
+    # I have made several modifications in order to better
+    # suit this library and to avoid monkey patching.
+    class Node
+      # Iterate over each children in the node.
+      def each
+        @children.each { |child| yield child }
+      end
+      # A string containing the node's value (or empty).
+      attr_accessor :value
+      # A unique identifier for the node.
+      attr_reader :id
+      # An array containing the children of this node.
+      attr_reader :children
+      # A hash containing the features of this node.
+      attr_accessor :features
+      # A hash containing the edges that link this 
+      # node to other nodes.
+      attr_accessor :edges
+      # The parent of the node.
+      attr_accessor :parent
+      # Initialize the node with its value and id.
+      # Setup containers for the children, features
+      # and edges of this node.
+      def initialize(value, id = nil)
+        @parent = nil
+        @value, @id = value, id
+        @children = []
+        @children_hash = {}
+        @features = {}
+        @edges = {}
+      end
+      # Boolean - does the node have edges?
+      def has_edges?; !@edges.empty?; end
+      # Boolean - does the node have children?
+      def has_children?; !@children.empty?; end
+      # Boolean - does the node have features?
+      def has_features?; !@features.empty?; end
+      # Boolean - does the node have a parent?
+      def has_parent?; !@parent.nil?; end
+      # Boolean - does the node not have a parent?
+      def is_root?; @parent.nil?; end
+      # Remove this node from its parent and set as root.
+      def set_as_root!; @parent = nil; self; end
+      # Boolean - is this node a leaf ?
+      # This is overriden in leaf classes.
+      def is_leaf?; !has_children?; end
+      # Add the nodes to the given child.
+      # This may be used with several nodes,
+      # for example: node << [child1, child2, child3]
+      def <<(nodes)
+        nodes = [nodes] unless nodes.is_a? Array
+        raise 'Trying to add a nil node.' if nodes.include? nil
+        nodes.each do |node|
+          node.parent = self
+          @children << node
+          @children_hash[node.id] = node
+        end
+        nodes[0]
+      end
+      def [](name_or_index)
+        if name_or_index == nil
+          raise Treat::Exception, 
+          "Non-nil name or index needs to be provided." 
+        end
+        if name_or_index.kind_of?(Integer) && 
+          name_or_index < 1000            # Fix
+          @children[name_or_index]
+        else
+          @children_hash[name_or_index]
+        end
+      end
+      def remove!(ion)
+        return nil unless ion
+        if ion.is_a? Treat::Tree::Node
+          @children.delete(ion)
+          @children_hash.delete(ion.id)
+          ion.set_as_root!
+        else
+          @children.delete(@children_hash[ion])
+          @children_hash.delete(ion)
+        end
+      end
+      def remove_all!
+        @children.each { |child| child.set_as_root! }
+        @children.clear
+        @children_hash.clear
+        self
+      end
+      # Previous sibling from the same parent.
+      def next_sibling
+        return nil if is_root?
+        id = @parent.children.index(self)
+        @parent.children.at(id + 1) if id
+      end
+      def left(n = 1); sibling(-1*n); end
+      def right(n = 1); sibling(1*n); end
+      def sibling(pos)
+        return nil if is_root?
+        id = @parent.children.index(self)
+        @parent.children.at(id + pos)
+      end
+      # There must be a cleaner way to do this.
+      def siblings
+        r = @parent.children.dup
+        r.delete(self)
+        r
+      end
+      # Total number of nodes in the subtree, including this one.
+      def size
+        @children.inject(1) { |sum, node| sum + node.size }
+      end
+      # Set the feature to the supplied value.
+      def set(feature, value)
+        @features ||= {}
+        @features[feature] = value
+      end
+      # Return the depth of this node in the tree.
+      def depth
+        return 0 if is_root?
+        1 + parent.depth
+      end
+      # Does the entity have a feature ?
+      def has_feature?(feature)
+        @features.has_key?(feature) ||
+        feature == :value
+      end
+      alias :has? :has_feature?
+      # Link this node to the target node with
+      # the supplied edge type.
+      def associate(id_or_node, edge_type = nil)
+        if id_or_node.is_a? Treat::Tree::Node
+          id = root.find(id_or_node).id
+        else
+          id = id_or_node
+        end
+        @edges[id] = edge_type if id
+      end
+      # Find the node in the tree with the given id.
+      def find(id_or_node)
+        if id_or_node.is_a? self.class
+          id = id_or_node.id
+        else
+          id = id_or_node
+        end
+        return @children_hash[id] if @children_hash[id]
+        self.each do |child|
+          r = child.find(id)
+          return r if r.is_a? Tree::Node
+        end
+      end
+      # Find the root of the tree within which
+      # this node is contained.
+      def root
+        return self if !has_parent?
+        ancestor = @parent
+        while ancestor.has_parent?
+          ancestor = ancestor.parent
+        end
+        ancestor
+      end
+    end
+  end
+end