tooled 0.0.2 → 0.1.0

data/lib/tooled.rb

@@ -2,6 +2,8 @@ require "tooled/version"
 require "tooled/vector2d"
 require "tooled/vector3d"
 require "tooled/stack"
+require "tooled/tree_node"
+require "tooled/tree_node_collection"
 require "tooled/queue"
 require "tooled/publisher"
 

data/lib/tooled/tree_node.rb

@@ -0,0 +1,48 @@
+require "tooled/tree_node_collection"
+
+module Tooled
+  class TreeNode
+    def initialize
+      @children = TreeNodeCollection.new
+      parent = nil
+    end
+
+    def children
+      @children
+    end
+
+    def each_child
+      return to_enum(:each_child) unless block_given?
+      children.each do |child|
+        yield child
+        child.each_child { |c| yield c }
+      end
+    end
+
+    def parent
+      @parent
+    end
+
+    def descendants
+      enum = to_enum(:each_child)
+      TreeNodeCollection.new(enum.respond_to?(:lazy) ? enum.lazy : enum)
+    end
+
+    def ancestors
+      enum = ::Enumerator.new do |yielder|
+        parent_node = parent
+        while parent_node
+          yielder << parent_node
+          parent_node = parent_node.parent
+        end
+      end
+      TreeNodeCollection.new(enum.respond_to?(:lazy) ? enum.lazy : enum)
+    end
+
+    def parent=(node)
+      raise InvalidNodeException.new("`parent' can only be a TreeNode") unless node.is_a?(TreeNode)
+      raise InvalidNodeException.new("`parent' can not be a descendant") if descendants.member?(node)
+      @parent = node.tap { |n| n.children.add(self) }
+    end
+  end
+end

data/lib/tooled/tree_node_collection.rb

@@ -0,0 +1,14 @@
+require "tooled/tree_node"
+
+module Tooled
+  class InvalidNodeException < Exception; end
+
+  class TreeNodeCollection < ::Set
+    def add(node)
+      raise InvalidNodeException.new("Only TreeNode objects can be added to collection") unless node.is_a?(TreeNode)
+      @hash[node] = true
+      self
+    end
+    alias_method :<<, :add
+  end
+end

data/lib/tooled/version.rb

@@ -1,3 +1,3 @@
 module Tooled
-  VERSION = "0.0.2"
+  VERSION = "0.1.0"
 end

data/spec/tree_node_collection_spec.rb

@@ -0,0 +1,18 @@
+require "tooled/tree_node_collection"
+
+describe Tooled::TreeNodeCollection do
+  let(:object) { Tooled::TreeNodeCollection.new }
+  subject { object }
+
+  describe "add" do
+    context "TreeNode to collection" do
+      specify { expect { object.add(Tooled::TreeNode.new) }.to change { object.size }.by(1) }
+      specify { expect { object << Tooled::TreeNode.new }.to change { object.size }.by(1) }
+    end
+
+    context "raises if not provided a TreeNode" do
+      specify { expect { object.add(String.new) }.to raise_error(Tooled::InvalidNodeException) }
+      specify { expect { object << String.new }.to raise_error(Tooled::InvalidNodeException) }
+    end
+  end
+end

data/spec/tree_node_spec.rb

@@ -0,0 +1,66 @@
+require "tooled/tree_node"
+
+describe Tooled::TreeNode do
+  let(:object) { Tooled::TreeNode.new }
+  subject { object }
+
+  its(:children) { should be_instance_of(Tooled::TreeNodeCollection) }
+
+  describe "reports all descendants" do
+    subject { object.descendants }
+
+    before do
+      object.children << Tooled::TreeNode.new
+      object.children << Tooled::TreeNode.new
+      object.children << Tooled::TreeNode.new.tap do |node_a|
+        node_a.children << Tooled::TreeNode.new.tap do |node_b|
+          node_b.children << Tooled::TreeNode.new
+        end
+      end
+    end
+
+    it { should be_instance_of(Tooled::TreeNodeCollection) }
+    its(:size) { should eql(5) }
+  end
+
+  describe "reports all ancestors" do
+    subject { object.ancestors }
+
+    before do
+      object.parent = Tooled::TreeNode.new.tap do |node_a|
+        node_a.parent = Tooled::TreeNode.new.tap do |node_b|
+          node_b.parent = Tooled::TreeNode.new.tap do |node_c|
+            node_c.parent = Tooled::TreeNode.new
+          end
+        end
+      end
+    end
+
+    it { should be_instance_of(Tooled::TreeNodeCollection) }
+    its(:size) { should eql(4) }
+  end
+
+  describe "setting the parent" do
+    context "to an invalid object" do
+      specify { expect { object.parent = String.new }.to raise_error(Tooled::InvalidNodeException) }
+    end
+
+    context "to a TreeNode" do
+      let(:parent) { Tooled::TreeNode.new }
+      before { object.parent = parent }
+
+      it "adds self to parents children" do
+        parent.children.to_a.should include(object)
+      end
+
+      its(:parent) { should eql(parent) }
+    end
+
+    context "to a child" do
+      let(:child) { Tooled::TreeNode.new }
+      before { object.children.add(child) }
+
+      specify { expect { object.parent = child }.to raise_error(Tooled::InvalidNodeException) }
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tooled
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-05-24 00:00:00.000000000 Z
+date: 2013-06-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -76,12 +76,16 @@ files:
 - lib/tooled/publisher.rb
 - lib/tooled/queue.rb
 - lib/tooled/stack.rb
+- lib/tooled/tree_node.rb
+- lib/tooled/tree_node_collection.rb
 - lib/tooled/vector2d.rb
 - lib/tooled/vector3d.rb
 - lib/tooled/version.rb
 - spec/publisher_spec.rb
 - spec/queue_spec.rb
 - spec/stack_spec.rb
+- spec/tree_node_collection_spec.rb
+- spec/tree_node_spec.rb
 - spec/vector2d_spec.rb
 - spec/vector3d_spec.rb
 - tooled.gemspec
@@ -114,5 +118,7 @@ test_files:
 - spec/publisher_spec.rb
 - spec/queue_spec.rb
 - spec/stack_spec.rb
+- spec/tree_node_collection_spec.rb
+- spec/tree_node_spec.rb
 - spec/vector2d_spec.rb
 - spec/vector3d_spec.rb