technical_graph 0.0.0

data/lib/technical_graph/refactoring_backup/test/test_technical_graph_axis.rb

@@ -0,0 +1,306 @@
+require 'helper'
+
+class TestTechnicalGraphAxis < Test::Unit::TestCase
+  context 'ranges calculation' do
+    setup do
+    end
+
+    should 'provide defined ranges' do
+      x_min = 0.0
+      x_max = 10.0
+      y_min = -5.0
+      y_max = 5.0
+
+      @tg = TechnicalGraph.new(
+        {
+          :x_min => x_min,
+          :x_max => x_max,
+          :y_min => y_min,
+          :y_max => y_max,
+          :xy_behaviour => :fixed
+        }
+      )
+      @tg.render
+
+      @tg.axis.x_min.should == x_min
+      @tg.axis.x_max.should == x_max
+      @tg.axis.y_min.should == y_min
+      @tg.axis.y_max.should == y_max
+    end
+
+    should 'calculate ranges per layer' do
+      # adding simple layer
+      layer_data = [
+        { :x => -1, :y => 3.0 },
+        { :x => 3, :y => -8.0 },
+        { :x => 0, :y => 3.0 },
+        { :x => 10, :y => 10.0 }
+      ]
+      dl = DataLayer.new(layer_data)
+
+      dl.x_min.should == -1
+      dl.x_max.should == 10
+      dl.y_min.should == -8.0
+      dl.y_max.should == 10.0
+    end
+
+
+    should 'provide ranges calculated using data layer, and multiple layers' do
+      x_min = 0.0
+      x_max = 1.0
+      y_min = -1.0
+      y_max = 1.0
+
+      @tg = TechnicalGraph.new(
+        {
+          :x_min => x_min,
+          :x_max => x_max,
+          :y_min => y_min,
+          :y_max => y_max,
+          :xy_behaviour => :default
+        }
+      )
+
+      # adding simple layer
+      layer_data = [
+        { :x => -1, :y => 3.0 },
+        { :x => 3, :y => -8.0 },
+        { :x => 0, :y => 3.0 },
+        { :x => 10, :y => 10.0 }
+      ]
+      @tg.add_layer(layer_data)
+      # should be added
+      @tg.layers.last.data.size > 0
+      # checking ranger for layer
+
+      @tg.render
+
+
+      @tg.axis.x_min.should_not == x_min
+      @tg.axis.x_max.should_not == x_max
+      @tg.axis.y_min.should_not == y_min
+      @tg.axis.y_max.should_not == y_max
+
+      @tg.axis.x_min.should == -1
+      @tg.axis.x_max.should == 10
+      @tg.axis.y_min.should == -8.0
+      @tg.axis.y_max.should == 10.0
+
+
+
+      # adding another layer
+
+      # adding simple layer
+      layer_data = [
+        { :x => -21, :y => -93.0 },
+        { :x => -5, :y => 3.0 },
+        { :x => 39, :y => -8.0 },
+        { :x => 0, :y => 333.0 },
+        { :x => 10, :y => 50.0 }
+      ]
+      @tg.add_layer(layer_data)
+      # should be added
+      @tg.layers.last.data.size > 1
+
+      @tg.render
+
+      @tg.axis.x_min.should_not == x_min
+      @tg.axis.x_max.should_not == x_max
+      @tg.axis.y_min.should_not == y_min
+      @tg.axis.y_max.should_not == y_max
+
+      @tg.axis.x_min.should_not == -1.0
+      @tg.axis.x_max.should_not == 10.0
+      @tg.axis.y_min.should_not == -8.0
+      @tg.axis.y_max.should_not == 10.0
+
+      @tg.axis.x_min.should == -21.0
+      @tg.axis.x_max.should == 39.0
+      @tg.axis.y_min.should == -93.0
+      @tg.axis.y_max.should == 333.0
+    end
+
+    should 'provide ranges calculated with zoom' do
+      x_min = 0.0
+      x_max = 1.0
+      y_min = -1.0
+      y_max = 1.0
+
+      @tg = TechnicalGraph.new(
+        {
+          :x_min => x_min,
+          :x_max => x_max,
+          :y_min => y_min,
+          :y_max => y_max,
+          :xy_behaviour => :default
+        }
+      )
+
+      # adding simple layer
+      layer_data = [
+        { :x => -5, :y => 5.0 },
+        { :x => 2, :y => -5.0 },
+        { :x => 0, :y => 5.0 },
+        { :x => 5, :y => 5.0 }
+      ]
+      @tg.add_layer(layer_data)
+      # should be added
+      @tg.layers.last.data.size > 0
+      # checking ranger for layer
+
+      @tg.render
+
+
+      @tg.axis.x_min.should_not == x_min
+      @tg.axis.x_max.should_not == x_max
+      @tg.axis.y_min.should_not == y_min
+      @tg.axis.y_max.should_not == y_max
+
+      @tg.axis.x_min.should == -5
+      @tg.axis.x_max.should == 5
+      @tg.axis.y_min.should == -5.0
+      @tg.axis.y_max.should == 5.0
+
+      @tg.axis.zoom = 2.0
+
+      @tg.axis.x_min.should == -10.0
+      @tg.axis.x_max.should == 10.0
+      @tg.axis.y_min.should == -10.0
+      @tg.axis.y_max.should == 10.0
+
+      @tg.axis.x_min.should_not == -5
+      @tg.axis.x_max.should_not == 5
+      @tg.axis.y_min.should_not == -5.0
+      @tg.axis.y_max.should_not == 5.0
+
+      @tg.axis.raw_x_min.should == -5
+      @tg.axis.raw_x_max.should == 5
+      @tg.axis.raw_y_min.should == -5.0
+      @tg.axis.raw_y_max.should == 5.0
+    end
+
+
+    should 'calculate axis with fixed interval' do
+      x_min = -5.0
+      x_max = 5.0
+      y_min = -5.0
+      y_max = 5.0
+
+      @tg = TechnicalGraph.new(
+        {
+          :x_min => x_min,
+          :x_max => x_max,
+          :y_min => y_min,
+          :y_max => y_max,
+          :xy_behaviour => :fixed,
+
+          :y_axises_count => 10,
+          :x_axises_count => 10,
+          :y_axises_interval => 1.0,
+          :x_axises_interval => 4.0,
+          :x_axises_fixed_interval => true,
+          :y_axises_fixed_interval => true
+        }
+      )
+
+      # adding simple layer
+      layer_data = [
+        { :x => -1, :y => 2.0 },
+        { :x => 1, :y => -2.0 },
+        { :x => 0, :y => 2.0 },
+        { :x => 1, :y => 2.0 }
+      ]
+      @tg.add_layer(layer_data)
+      # should be added
+      @tg.layers.last.data.size > 0
+      # checking ranger for layer
+
+      @tg.render
+
+      @tg.axis.value_axises.should == [-5.0, -4.0, -3.0, -2.0, -1.0, 0.0, 1.0, 2.0, 3.0, 4.0]
+      @tg.axis.parameter_axises.should == [-5.0, -1.0, 3.0]
+    end
+
+
+    should 'calculate axis with fixed count' do
+      x_min = -8.0
+      x_max = 8.0
+      y_min = -4.0
+      y_max = 4.0
+
+      @tg = TechnicalGraph.new(
+        {
+          :x_min => x_min,
+          :x_max => x_max,
+          :y_min => y_min,
+          :y_max => y_max,
+          :xy_behaviour => :fixed,
+
+          :x_axises_count => 8,
+          :y_axises_count => 4,
+          :x_axises_interval => 2.0,
+          :y_axises_interval => 1.0,
+          :x_axises_fixed_interval => false,
+          :y_axises_fixed_interval => false
+        }
+      )
+
+      # adding simple layer
+      layer_data = [
+        { :x => -1, :y => 2.0 },
+        { :x => 1, :y => -2.0 },
+        { :x => 0, :y => 2.0 },
+        { :x => 1, :y => 2.0 }
+      ]
+      @tg.add_layer(layer_data)
+      # should be added
+      @tg.layers.last.data.size > 0
+      # checking ranger for layer
+
+      @tg.render
+
+      @tg.axis.parameter_axises.should == [-8.0, -6.0, -4.0, -2.0, 0.0, 2.0, 4.0, 6.0]
+      @tg.axis.value_axises.should == [-4.0, -2.0, 0.0, 2.0]
+
+      @tg.image.save_to_file('test1.png')
+    end
+
+
+    should 'draw simple graph' do
+      @tg = TechnicalGraph.new(
+        {
+          :x_axises_count => 10,
+          :y_axises_count => 10,
+          :x_axises_interval => 1.0,
+          :y_axises_interval => 1.0,
+          :x_axises_fixed_interval => false,
+          :y_axises_fixed_interval => false,
+
+          :x_min => -10.0,
+          :x_max => 10.0,
+          :y_min => -10.0,
+          :y_max => 10.0
+        }
+      )
+
+
+
+      # adding simple layer
+      layer_data = Array.new
+      (0..20).each do |i|
+        layer_data << {:x => Math.sin(i.to_f/10.0) * 10.0, :y => Math.cos(i.to_f/10.0) * 10.0 }
+      end
+      @tg.add_layer(layer_data)
+      # should be added
+      @tg.layers.last.data.size > 0
+      # checking ranger for layer
+
+      @tg.render
+
+      @tg.image.save_to_file('test2.png')
+    end
+
+  end
+
+end
+

data/lib/technical_graph.rb

@@ -0,0 +1,55 @@
+#encoding: utf-8
+
+require 'rubygems'
+require 'technical_graph/data_layer'
+require 'technical_graph/graph_data_processor'
+require 'technical_graph/graph_image_drawer'
+require 'technical_graph/graph_axis'
+
+# Universal class for creating graphs/charts.
+
+# options parameters:
+# :width - width of image
+# :height - height of image
+# :x_min, :x_max, :y_min, :y_max - default or fixed ranges
+# :xy_behaviour:
+# * :default - use them as default ranges
+# * :fixed - ranges will not be changed during addition of layers
+
+class TechnicalGraph
+
+  def initialize(options = { })
+    @options = options
+    @data_processor = GraphDataProcessor.new(self)
+    @image_drawer = GraphImageDrawer.new(self)
+    @axis = GraphAxis.new(self)
+    @layers = Array.new
+  end
+  attr_reader :options
+  attr_reader :data_processor
+  attr_reader :image_drawer
+  attr_reader :axis
+
+  attr_reader :layers
+
+  # Add new data layer to layer array
+  def add_layer(data = [], options = {})
+    @layers << DataLayer.new(data, options)
+  end
+
+  # Create graph
+  def render
+    @image = @image_drawer.crate_blank_graph_image
+    # recalculate ranges if needed
+    @layers.each do |l|
+      @data_processor.process_data_layer(l)
+    end
+    
+    # draw axis
+    @axis.render_on_image(@image)
+    # draw layers
+    @layers.each do |l|
+      @image_drawer.render_data_layer(l)
+    end
+  end
+end

data/test/helper.rb

@@ -0,0 +1,19 @@
+require 'rubygems'
+require 'bundler'
+begin
+  Bundler.setup(:default, :development)
+rescue Bundler::BundlerError => e
+  $stderr.puts e.message
+  $stderr.puts "Run `bundle install` to install missing gems"
+  exit e.status_code
+end
+require 'test/unit'
+require 'shoulda'
+require 'rspec'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'technical_graph'
+
+class Test::Unit::TestCase
+end

data/test/test_technical_graph.rb

@@ -0,0 +1,104 @@
+require 'helper'
+
+class TestTechnicalGraph < Test::Unit::TestCase
+  context 'initial options' do
+    setup do
+      @technical_graph = TechnicalGraph.new
+    end
+
+    should "has options with default values" do
+      @technical_graph.options.class.should == Hash
+      @technical_graph.options[:width] > 0
+      @technical_graph.options[:height] > 0
+    end
+
+    should "has options with custom values" do
+      s = 10
+      tg = TechnicalGraph.new({ :height => s, :width => s })
+      @technical_graph.options[:width] == s
+      @technical_graph.options[:height] == s
+
+      @technical_graph.image_drawer.width == s
+      @technical_graph.image_drawer.height == s
+    end
+
+    should "has changeable options" do
+      s = 20
+      tg = TechnicalGraph.new
+      tg.image_drawer.width = s
+      tg.image_drawer.height = s
+      @technical_graph.options[:width] == s
+      @technical_graph.options[:height] == s
+    end
+  end
+
+  context 'basic layer operation and saving file' do
+    setup do
+      @tg = TechnicalGraph.new
+      @data_size = 100
+
+      # sample data
+      @data = Array.new
+      @second_data = Array.new
+      (0...@data_size).each do |i|
+        @data << { :x => Time.now.to_i - 3600 + i, :y => Math.sin(i.to_f / 10.0) }
+        @second_data << { :x => Time.now.to_i - 1800 + i*2, :y => Math.cos(i.to_f / 10.0) }
+      end
+    end
+
+    should 'has ability do add new layer' do
+      layers = @tg.layers.size
+      @tg.add_layer(@data)
+      @tg.layers.size.should == layers + 1
+
+      layer = @tg.layers.last
+      layer.data.size.should == @data_size
+    end
+
+    should 'has ability to manipulate layers, add more data' do
+      @tg.add_layer(@data)
+      layer = @tg.layers.last
+      layer.class.should == DataLayer
+
+      layer.data.size.should == @data_size
+
+      # adding second data
+      layer.append_data(@second_data)
+      layer.data.size.should == 2 * @data_size
+
+      # @tg.render
+      # @tg.image.save_to_file('test1.png')
+    end
+
+    should 'has ability to filter records with similar x\'es' do
+      @tg.add_layer
+      layer = @tg.layers.last
+      layer.data.size.should == 0
+      layer.append_data([{ :x => 0, :y => 1 }])
+      layer.data.size.should == 1
+
+      # uniq check
+      layer.append_data([{ :x => 0, :y => 1 }])
+      layer.append_data([{ :x => 0, :y => 1 }])
+      layer.data.size.should == 1
+      layer.append_data([{ :x => 2, :y => 1 }])
+      layer.data.size.should == 2
+    end
+
+    should 'has ability to filter bad records' do
+      @tg.add_layer
+      layer = @tg.layers.last
+      layer.data.size.should == 0
+      layer.append_data([{ :x => 0, :y => 1 }])
+      layer.data.size.should == 1
+
+      # uniq check
+      layer.append_data([{ :z => 0, :y => 1 }])
+      layer.append_data([{}])
+      layer.data.size.should == 1
+    end
+
+  end
+
+
+end