projects.mako.cc - selectricity/blob - vendor/plugins/gruff/lib/gruff/line.rb

   1
   2 require File.dirname(__FILE__) + '/base'
   3
   4 class Gruff::Line < Gruff::Base
   5
   6   # Draw a dashed line at the given value
   7   attr_accessor :baseline_value
   8
   9   # Color of the baseline
  10   attr_accessor :baseline_color
  11
  12   # Hide parts of the graph to fit more datapoints, or for a different appearance.
  13   attr_accessor :hide_dots, :hide_lines
  14
  15   # Call with target pixel width of graph (800, 400, 300), and/or 'false' to omit lines (points only).
  16   #
  17   #  g = Gruff::Line.new(400) # 400px wide with lines
  18   #
  19   #  g = Gruff::Line.new(400, false) # 400px wide, no lines (for backwards compatibility)
  20   #
  21   #  g = Gruff::Line.new(false) # Defaults to 800px wide, no lines (for backwards compatibility)
  22   #
  23   # The preferred way is to call hide_dots or hide_lines instead.
  24   def initialize(*args)
  25     raise ArgumentError, "Wrong number of arguments" if args.length > 2
  26     if args.empty? or ((not Numeric === args.first) && (not String === args.first)) then
  27       super()
  28     else
  29       super args.shift
  30     end
  31
  32     @hide_dots = @hide_lines = false
  33     @baseline_color = 'red'
  34     @baseline_value = nil
  35   end
  36
  37   def draw
  38     super
  39
  40     return unless @has_data
  41
  42     # Check to see if more than one datapoint was given. NaN can result otherwise.
  43     @x_increment = (@column_count > 1) ? (@graph_width / (@column_count - 1).to_f) : @graph_width
  44
  45     if (defined?(@norm_baseline)) then
  46       level = @graph_top + (@graph_height - @norm_baseline * @graph_height)
  47       @d = @d.push
  48       @d.stroke_color @baseline_color
  49       @d.fill_opacity 0.0
  50       @d.stroke_dasharray(10, 20)
  51       @d.stroke_width 5
  52       @d.line(@graph_left, level, @graph_left + @graph_width, level)
  53       @d = @d.pop
  54     end
  55
  56     @norm_data.each do |data_row|
  57       prev_x = prev_y = nil
  58
  59       data_row[1].each_with_index do |data_point, index|
  60         new_x = @graph_left + (@x_increment * index)
  61         next if data_point.nil?
  62
  63         draw_label(new_x, index)
  64
  65         new_y = @graph_top + (@graph_height - data_point * @graph_height)
  66
  67         # Reset each time to avoid thin-line errors
  68         @d = @d.stroke data_row[DATA_COLOR_INDEX]
  69         @d = @d.fill data_row[DATA_COLOR_INDEX]
  70         @d = @d.stroke_opacity 1.0
  71         @d = @d.stroke_width clip_value_if_greater_than(@columns / (@norm_data.first[1].size * 4), 5.0)
  72
  73         if !@hide_lines and !prev_x.nil? and !prev_y.nil? then
  74           @d = @d.line(prev_x, prev_y, new_x, new_y)
  75         end
  76         circle_radius = clip_value_if_greater_than(@columns / (@norm_data.first[1].size * 2.5), 5.0)
  77         @d = @d.circle(new_x, new_y, new_x - circle_radius, new_y) unless @hide_dots
  78
  79         prev_x = new_x
  80         prev_y = new_y
  81       end
  82
  83     end
  84
  85     @d.draw(@base_image)
  86   end
  87
  88   def normalize
  89     @maximum_value = [@maximum_value.to_f, @baseline_value.to_f].max
  90     super
  91     @norm_baseline = (@baseline_value.to_f / @maximum_value.to_f) if @baseline_value
  92   end
  93
  94 end