ruby-treemap 0.0.1

data/lib/treemap/svg_output.rb

@@ -0,0 +1,96 @@
+#
+# svg_output.rb - RubyTreemap
+#
+# Copyright (c) 2006 by Andrew Bruno <aeb@qnot.org>
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+#
+
+require 'cgi'
+require 'RMagick'
+
+require File.dirname(__FILE__) + "/output_base"
+
+class Treemap::SvgOutput < Treemap::OutputBase
+
+    def initialize
+        super
+
+        yield self if block_given?
+    end
+
+    def node_label(node)
+        CGI.escapeHTML(node.label)
+    end
+
+    def node_color(node)
+        color = "#CCCCCC"
+
+        if(!node.color.nil?)
+            if(not Numeric === node.color)
+                color = node.color
+            else
+                color = "#" + @color.get_hex_color(node.color)
+            end
+        end
+
+        color
+    end
+
+    def to_png(node, filename="treemap.png")
+        svg = to_svg(node)
+        img = Magick::Image.from_blob(svg) { self.format = "SVG" }
+        img[0].write(filename)
+    end
+
+    def to_svg(node)
+        bounds = self.bounds
+
+        @layout.process(node, bounds)
+
+        draw_map(node)
+    end
+
+    def draw_map(node)
+        svg = "<svg"
+        svg += " xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\""
+        svg += " xmlns:ev=\"http://www.w3.org/2001/xml-events\"" 
+        svg += " width=\"" + (node.bounds.width).to_s + "\" height=\"" + (node.bounds.height).to_s + "\">"
+
+        svg += draw_node(node)
+        svg += "</svg>"
+        svg
+    end
+
+    def draw_node(node)
+        return "" if node.bounds.nil?
+
+        svg = ""
+        svg += "<rect"
+        svg += " id=\"rect-" + node.id.to_s + "\""
+        svg += " x=\"" + node.bounds.x1.to_s + "\""
+        svg += " y=\"" + node.bounds.y1.to_s + "\""
+        svg += " width=\"" + node.bounds.width.to_s + "\""
+        svg += " height=\"" + node.bounds.height.to_s + "\""
+        #svg += " style=\""
+        #svg += " fill: " + node_color(node) + ";"
+        #svg += " stroke: #000000;"
+        #svg += " stroke-width: 1px;"
+        svg += " fill=\"" + node_color(node) + "\""
+        svg += " stroke=\"#000000\""
+        svg += " stroke-width=\"1px\""
+        svg += " />"
+
+        if(!node.children.nil? and node.children.size > 0)
+            node.children.each do |c|
+                svg += draw_node(c)
+            end
+        end
+
+        svg
+    end
+end

data/test/tc_color.rb

@@ -0,0 +1,17 @@
+
+require "test_base"
+
+class TestColor < Test::Unit::TestCase
+    def test_color
+        g = Treemap::GradientColor.new do |g|
+            g.min_color = "FC0404"
+            g.mean_color = "DBD9DC"
+            g.max_color = "0092C8"
+            g.increment = 0.5 
+        end
+
+        File.open(File.dirname(__FILE__) + '/color_test.html', "w") do |f|
+            f.puts g.to_html
+        end
+    end
+end

data/test/tc_html.rb

@@ -0,0 +1,22 @@
+
+require "test_base"
+
+class TestHTMLOutput < Test::Unit::TestCase
+    include TestTreemapBase
+
+    def setup
+        @root = setup_root
+    end
+
+    def test_html_output
+        output = Treemap::HtmlOutput.new do |o|
+            o.width = 800
+            o.height = 600
+            o.center_labels_at_depth = 1
+        end
+
+        File.open(File.dirname(__FILE__) + '/html_test.html', "w") do |f|
+            f.puts output.to_html(@root)
+        end
+    end
+end

data/test/tc_simple.rb

@@ -0,0 +1,30 @@
+
+require "test_base"
+
+class TestSimple < Test::Unit::TestCase
+    def setup 
+        @root = Treemap::Node.new
+        @root.new_child(:size => 6)
+        @root.new_child(:size => 6)
+        @root.new_child(:size => 4)
+        @root.new_child(:size => 3)
+        @root.new_child(:size => 2)
+        @root.new_child(:size => 2)
+        @root.new_child(:size => 1)
+    end
+
+    def test_simple
+        output = Treemap::ImageOutput.new do |o|
+            o.width = 600
+            o.height = 400
+            o.layout = Treemap::SquarifiedLayout.new
+        end
+
+        bounds = Treemap::Rectangle.new(0, 0, 600, 400)
+        output.layout.process(@root, bounds)
+
+        Treemap::dump_tree(@root)
+
+        output.to_png(@root, File.dirname(__FILE__) + '/simple_test.png')
+    end
+end

data/test/tc_slice.rb

@@ -0,0 +1,25 @@
+
+require "test_base"
+
+class TestSlice < Test::Unit::TestCase
+    include TestTreemapBase
+
+    def setup
+        @root = setup_root
+    end
+
+    def test_slice
+        output = Treemap::ImageOutput.new do |o|
+            o.width = 800
+            o.height = 600
+            o.layout = Treemap::SliceLayout.new
+        end
+
+        output.to_png(@root, File.dirname(__FILE__) + '/slice_test.png')
+
+        bounds = Treemap::Rectangle.new(0, 0, 800, 600)
+        output.layout.process(@root, bounds)
+
+        Treemap::dump_tree(@root)
+    end
+end

data/test/tc_squarified.rb

@@ -0,0 +1,25 @@
+
+require "test_base"
+
+class TestSquarified < Test::Unit::TestCase
+    include TestTreemapBase
+
+    def setup
+        @root = setup_root
+    end
+
+    def test_squarified
+        output = Treemap::ImageOutput.new do |o|
+            o.width = 800
+            o.height = 600
+            o.layout = Treemap::SquarifiedLayout.new
+        end
+
+        output.to_png(@root, File.dirname(__FILE__) + '/squarified_test.png')
+
+        bounds = Treemap::Rectangle.new(0, 0, 800, 600)
+        output.layout.process(@root, bounds)
+
+        Treemap::dump_tree(@root)
+    end
+end

data/test/tc_svg.rb

@@ -0,0 +1,21 @@
+
+require "test_base"
+
+class TestSvgOutput < Test::Unit::TestCase
+    include TestTreemapBase
+
+    def setup
+        @root = setup_root
+    end
+
+    def test_svg_output
+        output = Treemap::SvgOutput.new do |o|
+            o.width = 800
+            o.height = 600
+        end
+
+        File.open(File.dirname(__FILE__) + '/svg_test.svg', "w") do |f|
+            f.puts output.to_svg(@root)
+        end
+    end
+end

data/test/test_base.rb

@@ -0,0 +1,39 @@
+
+require "test/unit"
+require "treemap"
+begin
+    require "RMagick"
+    require "treemap/image_output" 
+    require "treemap/svg_output" 
+rescue LoadError
+end
+
+module TestTreemapBase
+    def setup_root
+        root = Treemap::Node.new(:size => 100, :label => "all")
+
+        root.new_child(:size => 12, :label => "Child1")
+        root.new_child(:size => 8, :label => "Child2")
+
+        child3 = Treemap::Node.new(:size => 10, :label => "Child3")
+        [2,2,2,3,1].each do |n|
+            child3.new_child(:size => n, :label => "Child3-"+n.to_s)
+        end
+        root.add_child(child3)
+
+        child4 = Treemap::Node.new(:size => 70, :label => "Child4")
+        [5, 40].each do |n|
+            child4.new_child(:size => n, :label => "Child4-"+n.to_s)
+        end
+
+        child5 = Treemap::Node.new(:size => 25, :label => "Child5")
+        [6,8,11].each do |n|
+            child5.new_child(:size => n, :label => "Child5-"+n.to_s)
+        end
+
+        child4.add_child(child5)
+        root.add_child(child4)
+
+        root
+    end
+end

metadata

@@ -0,0 +1,73 @@
+--- !ruby/object:Gem::Specification 
+rubygems_version: 0.9.0
+specification_version: 1
+name: ruby-treemap
+version: !ruby/object:Gem::Version 
+  version: 0.0.1
+date: 2006-07-17 00:00:00 -07:00
+summary: Treemap visualization in ruby
+require_paths: 
+- lib
+email: aeb@qnot.org
+homepage: http://rubytreemap.rubyforge.org/
+rubyforge_project: 
+description: Treemap visualization in ruby
+autorequire: treemap
+default_executable: 
+bindir: bin
+has_rdoc: true
+required_ruby_version: !ruby/object:Gem::Version::Requirement 
+  requirements: 
+  - - ">"
+    - !ruby/object:Gem::Version 
+      version: 0.0.0
+  version: 
+platform: ruby
+signing_key: 
+cert_chain: 
+post_install_message: 
+authors: 
+- Andrew Bruno
+files: 
+- Rakefile
+- TODO
+- EXAMPLES
+- README
+- ChangeLog
+- COPYING
+- lib/treemap
+- lib/treemap.rb
+- lib/treemap/squarified_layout.rb
+- lib/treemap/svg_output.rb
+- lib/treemap/image_output.rb
+- lib/treemap/html_output.rb
+- lib/treemap/rectangle.rb
+- lib/treemap/node.rb
+- lib/treemap/layout_base.rb
+- lib/treemap/color_base.rb
+- lib/treemap/gradient_color.rb
+- lib/treemap/output_base.rb
+- lib/treemap/slice_layout.rb
+- test/tc_squarified.rb
+- test/tc_simple.rb
+- test/tc_color.rb
+- test/tc_slice.rb
+- test/tc_svg.rb
+- test/test_base.rb
+- test/tc_html.rb
+test_files: []
+
+rdoc_options: 
+- --main
+- README
+extra_rdoc_files: 
+- README
+- EXAMPLES
+executables: []
+
+extensions: []
+
+requirements: 
+- none
+dependencies: []
+