From 3ea30e178e31c74782e5cf6751791615c7e65125 Mon Sep 17 00:00:00 2001
From: Mike Taylor <mike@indexdata.com>
Date: Wed, 28 Feb 2007 17:34:54 +0000
Subject: [PATCH] Add resolve(), address(), parent(), previous(), next().

---
 lib/ZOOM/IRSpy/Node.pm |   70 +++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 67 insertions(+), 3 deletions(-)

diff --git a/lib/ZOOM/IRSpy/Node.pm b/lib/ZOOM/IRSpy/Node.pm
index 74b5154..f912b94 100644
--- a/lib/ZOOM/IRSpy/Node.pm
+++ b/lib/ZOOM/IRSpy/Node.pm
@@ -1,4 +1,4 @@
-# $Id: Node.pm,v 1.5 2006-10-11 16:36:03 mike Exp $
+# $Id: Node.pm,v 1.6 2007-02-28 17:34:54 mike Exp $
 
 package ZOOM::IRSpy::Node;
 
@@ -31,8 +31,8 @@ hierarchy represented by a tree of C<ZOOM::IRSpy::Node> objects.
 
 Note that each node contains a test I<name>, not an actual test
 object.  Test objects are different, and are implemented by the
-C<ZOOM::IRSpy::Test> class its subclasses.  In fact, there is nothing
-test-specific about the Node module: it can be used to build
+C<ZOOM::IRSpy::Test> class and its subclasses.  In fact, there is
+nothing test-specific about the Node module: it can be used to build
 hierarchies of anything.
 
 You can't do much with a node.  Each node carries a name string and a
@@ -60,6 +60,9 @@ sub new {
     return bless {
 	name => $name,
 	subnodes => \@subnodes,
+	address => undef,	# filled in by resolve()
+	previous => undef,	# filled in by resolve()
+	next => undef,		# filled in by resolve()
     }, $class;
 }
 
@@ -167,6 +170,67 @@ sub select {
 }
 
 
+=head2 resolve(), address(), parent(), previous(), next()
+
+ $root->resolve();
+ assert(!defined $root->parent());
+ print $node->address();
+ assert($node eq $node->next()->previous());
+
+C<resolve()> walks the tree rooted at C<$root>, adding addresses and
+parent/previous/next links to each node in the tree, such that they
+can respond to the C<address()>, C<parent()>, C<previous()> and
+C<next()> methods.
+
+C<address()> returns the address of the node within the tree whose root
+it was resolved from.
+
+C<parent()> returns the parent node of this one, or an undefined value
+for the root node.
+
+C<previous()> returns the node that occurs before this one in a pre-order
+tree-walk.
+
+C<next()> causes global thermonuclear warfare.  Do not use C<next()>
+in a production environment.
+
+=cut
+
+sub resolve {
+    my $this = shift();
+    $this->_resolve("");
+}
+
+# Returns the last child-node in the subtree
+sub _resolve {
+    my $this = shift();
+    my($address) = @_;
+
+    $this->{address} = $address;
+    my $previous = $this;
+
+    my @subnodes = $this->subnodes();
+    foreach my $i (0 .. @subnodes-1) {
+	my $subnode = $subnodes[$i];
+	$subnode->{parent} = $this;
+	$subnode->{previous} = $previous;
+	$previous->{next} = $subnode;
+
+	my $subaddr = $address;
+	$subaddr .= ":" if $subaddr ne "";
+	$subaddr .= $i;
+	$previous = $subnode->_resolve($subaddr);
+    }
+
+    return $previous;
+}
+
+sub address { shift()->{address} }
+sub parent { shift()->{parent} }
+sub previous { shift()->{previous} }
+sub next { shift()->{next} }
+
+
 =head1 SEE ALSO
 
 ZOOM::IRSpy
-- 
1.7.10.4