Added NaryTree root node getter support

nathanklick · May 7, 2014 · 35974dc · 35974dc
1 parent 321078c
commit 35974dc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/tjsd/collections/NaryTree.php b/tjsd/collections/NaryTree.php
@@ -12,8 +12,10 @@ class NaryTree implements Tree {
     public function insert(types\Comparable $element) {
 		if($this->isEmpty()) {
 			$this->rootNode = new types\NaryTreeNode($element);
+
+			return $this->rootNode;
 		} else {
-			$this->rootNode->insert($element);
+			return $this->rootNode->insert($element);
 		}
     }
 
@@ -24,6 +26,10 @@ public function __toString() {
     public function clear() {
 		$this->rootNode = null;
     }
+
+	public function root() {
+		return $this->rootNode;
+	}
 
     public function contains($element) {
 		if($element instanceof types\Comparable) {

diff --git a/tjsd/collections/types/NaryTreeNode.php b/tjsd/collections/types/NaryTreeNode.php
@@ -91,6 +91,8 @@ public function insert(Comparable $element) {
 			$item = new self($element);
 			$item->setParent($this);
 			$this->children[] = $item;
+
+			return $item;
 		} else {
 		   throw new DuplicateEntryException('Value is already present - cannot insert two elements with same value.'); 
 		}