gf-treevisitor 0.0.13 → 0.0.15

data/VERSION

@@ -1 +1 @@
-0.0.13
+0.0.15

data/lib/treevisitor.rb

@@ -1,3 +1,3 @@
 class TreeVisitor
-  VERSION = '0.0.13'
+  VERSION = '0.0.15'
 end

data/lib/treevisitor/tree_node.rb

@@ -9,6 +9,41 @@ require 'treevisitor/leaf_node'
 #
 class TreeNode < AbsNode
   
+  class << self
+    
+    def create(class1 = TreeNode, class2 = LeafNode, &block)
+      if class1.ancestors.include?(TreeNode) and class2.ancestors.include?(LeafNode)
+        @tree_node_class = class1
+        @leaf_node_class = class2
+      elsif class1.ancestors.include?(LeafNode) and class2 == LeafNode
+        @tree_node_class = self
+        @leaf_node_class = class1
+      end
+
+      if @tree_node_class.nil? || @leaf_node_class.nil?
+        raise "Must be specified class derived from TreeNode and LeafNode"
+      end
+
+      @scope_stack = []
+      class_eval(&block)
+    end
+
+    private
+
+    def node(name, &block)
+      parent_node = @scope_stack.length > 0 ? @scope_stack[-1] : nil
+      tree_node = @tree_node_class.new(name, parent_node)
+      @scope_stack.push tree_node
+      class_eval(&block)
+      @scope_stack.pop
+    end
+
+    def leaf(name)
+      tree_node = @scope_stack[-1]
+      @leaf_node_class.new(name, tree_node)
+    end
+  end
+
   attr_reader :leaves
   attr_reader :children
 

data/test/treevisitor/tc_tree_node_dsl.rb

@@ -0,0 +1,78 @@
+require File.join(File.dirname(__FILE__), "test_helper")
+
+require 'treevisitor/tree_node.rb'
+
+
+class DTreeNode < TreeNode
+  def to_s
+    "dt: #{name}"
+  end
+end
+
+class DLeafNode < LeafNode
+  def to_s
+    "dl: #{name}"
+  end
+end
+
+class TCTreeNodeDsl < Test::Unit::TestCase
+
+  def test_dsl
+
+    tree = TreeNode.create do
+      node "root" do
+        leaf "l1"
+        leaf "l2"
+        node "sub" do
+          leaf "l3"
+        end
+      end
+    end
+
+    # puts tree.to_str
+    out =<<EOS
+root
+|-- l1
+|-- l2
+`-- sub
+    `-- l3
+EOS
+    assert_equal out, tree.to_str
+
+  end
+
+
+  def test_derivated
+    tree = TreeNode.create(DTreeNode, DLeafNode) do
+      node "root" do
+        leaf "l1"
+        leaf "l2"
+        node "sub" do
+          leaf "l3"
+        end
+      end
+    end
+
+    # puts tree.to_str
+    out =<<EOS
+dt: root
+|-- dl: l1
+|-- dl: l2
+`-- dt: sub
+    `-- dl: l3
+EOS
+    assert_equal out, tree.to_str
+
+    tree = DTreeNode.create(DLeafNode) do
+      node "root" do
+        leaf "l1"
+        leaf "l2"
+        node "sub" do
+          leaf "l3"
+        end
+      end
+    end
+    assert_equal out, tree.to_str
+  end
+
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: gf-treevisitor
 version: !ruby/object:Gem::Version 
-  version: 0.0.13
+  version: 0.0.15
 platform: ruby
 authors: 
 - gf
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-06-07 00:00:00 -07:00
+date: 2009-06-09 00:00:00 -07:00
 default_executable: tree.rb
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -96,6 +96,7 @@ test_files:
 - test/treevisitor/tc_dir_tree_walker.rb
 - test/treevisitor/tc_tree_node.rb
 - test/treevisitor/tc_tree_node_visitor.rb
+- test/treevisitor/tc_tree_node_dsl.rb
 - test/treevisitor/test_helper.rb
 - test/gf_utilities/tc_md5.rb
 - test/gf_utilities/tc_kwartz.rb