Plot.rb
500 lines
| 12.3 KiB
| text/x-ruby
|
RubyLexer
|
r2553 | require 'SVG/Graph/Graph' | ||
module SVG | ||||
module Graph | ||||
# === For creating SVG plots of scalar data | ||||
# | ||||
# = Synopsis | ||||
# | ||||
# require 'SVG/Graph/Plot' | ||||
# | ||||
# # Data sets are x,y pairs | ||||
# # Note that multiple data sets can differ in length, and that the | ||||
# # data in the datasets needn't be in order; they will be ordered | ||||
# # by the plot along the X-axis. | ||||
# projection = [ | ||||
# 6, 11, 0, 5, 18, 7, 1, 11, 13, 9, 1, 2, 19, 0, 3, 13, | ||||
# 7, 9 | ||||
# ] | ||||
# actual = [ | ||||
# 0, 18, 8, 15, 9, 4, 18, 14, 10, 2, 11, 6, 14, 12, | ||||
# 15, 6, 4, 17, 2, 12 | ||||
# ] | ||||
# | ||||
# graph = SVG::Graph::Plot.new({ | ||||
# :height => 500, | ||||
# :width => 300, | ||||
# :key => true, | ||||
# :scale_x_integers => true, | ||||
# :scale_y_integerrs => true, | ||||
# }) | ||||
# | ||||
# graph.add_data({ | ||||
# :data => projection | ||||
# :title => 'Projected', | ||||
# }) | ||||
# | ||||
# graph.add_data({ | ||||
# :data => actual, | ||||
# :title => 'Actual', | ||||
# }) | ||||
# | ||||
# print graph.burn() | ||||
# | ||||
# = Description | ||||
# | ||||
# Produces a graph of scalar data. | ||||
# | ||||
# This object aims to allow you to easily create high quality | ||||
# SVG[http://www.w3c.org/tr/svg] scalar plots. You can either use the | ||||
# default style sheet or supply your own. Either way there are many options | ||||
# which can be configured to give you control over how the graph is | ||||
# generated - with or without a key, data elements at each point, title, | ||||
# subtitle etc. | ||||
# | ||||
# = Examples | ||||
# | ||||
# http://www.germane-software/repositories/public/SVG/test/plot.rb | ||||
# | ||||
# = Notes | ||||
# | ||||
# The default stylesheet handles upto 10 data sets, if you | ||||
# use more you must create your own stylesheet and add the | ||||
# additional settings for the extra data sets. You will know | ||||
# if you go over 10 data sets as they will have no style and | ||||
# be in black. | ||||
# | ||||
# Unlike the other types of charts, data sets must contain x,y pairs: | ||||
# | ||||
# [ 1, 2 ] # A data set with 1 point: (1,2) | ||||
# [ 1,2, 5,6] # A data set with 2 points: (1,2) and (5,6) | ||||
# | ||||
# = See also | ||||
# | ||||
# * SVG::Graph::Graph | ||||
# * SVG::Graph::BarHorizontal | ||||
# * SVG::Graph::Bar | ||||
# * SVG::Graph::Line | ||||
# * SVG::Graph::Pie | ||||
# * SVG::Graph::TimeSeries | ||||
# | ||||
# == Author | ||||
# | ||||
# Sean E. Russell <serATgermaneHYPHENsoftwareDOTcom> | ||||
# | ||||
# Copyright 2004 Sean E. Russell | ||||
# This software is available under the Ruby license[LICENSE.txt] | ||||
# | ||||
class Plot < Graph | ||||
# In addition to the defaults set by Graph::initialize, sets | ||||
# [show_data_values] true | ||||
# [show_data_points] true | ||||
# [area_fill] false | ||||
# [stacked] false | ||||
def set_defaults | ||||
init_with( | ||||
:show_data_values => true, | ||||
:show_data_points => true, | ||||
:area_fill => false, | ||||
:stacked => false | ||||
) | ||||
self.top_align = self.right_align = self.top_font = self.right_font = 1 | ||||
end | ||||
# Determines the scaling for the X axis divisions. | ||||
# | ||||
# graph.scale_x_divisions = 2 | ||||
# | ||||
# would cause the graph to attempt to generate labels stepped by 2; EG: | ||||
# 0,2,4,6,8... | ||||
attr_accessor :scale_x_divisions | ||||
# Determines the scaling for the Y axis divisions. | ||||
# | ||||
# graph.scale_y_divisions = 0.5 | ||||
# | ||||
# would cause the graph to attempt to generate labels stepped by 0.5; EG: | ||||
# 0, 0.5, 1, 1.5, 2, ... | ||||
attr_accessor :scale_y_divisions | ||||
# Make the X axis labels integers | ||||
attr_accessor :scale_x_integers | ||||
# Make the Y axis labels integers | ||||
attr_accessor :scale_y_integers | ||||
# Fill the area under the line | ||||
attr_accessor :area_fill | ||||
# Show a small circle on the graph where the line | ||||
# goes from one point to the next. | ||||
attr_accessor :show_data_points | ||||
# Set the minimum value of the X axis | ||||
attr_accessor :min_x_value | ||||
# Set the minimum value of the Y axis | ||||
attr_accessor :min_y_value | ||||
# Adds data to the plot. The data must be in X,Y pairs; EG | ||||
# [ 1, 2 ] # A data set with 1 point: (1,2) | ||||
# [ 1,2, 5,6] # A data set with 2 points: (1,2) and (5,6) | ||||
def add_data data | ||||
@data = [] unless @data | ||||
raise "No data provided by #{conf.inspect}" unless data[:data] and | ||||
data[:data].kind_of? Array | ||||
raise "Data supplied must be x,y pairs! "+ | ||||
"The data provided contained an odd set of "+ | ||||
"data points" unless data[:data].length % 2 == 0 | ||||
return if data[:data].length == 0 | ||||
x = [] | ||||
y = [] | ||||
data[:data].each_index {|i| | ||||
(i%2 == 0 ? x : y) << data[:data][i] | ||||
} | ||||
sort( x, y ) | ||||
data[:data] = [x,y] | ||||
@data << data | ||||
end | ||||
protected | ||||
def keys | ||||
@data.collect{ |x| x[:title] } | ||||
end | ||||
def calculate_left_margin | ||||
super | ||||
label_left = get_x_labels[0].to_s.length / 2 * font_size * 0.6 | ||||
@border_left = label_left if label_left > @border_left | ||||
end | ||||
def calculate_right_margin | ||||
super | ||||
label_right = get_x_labels[-1].to_s.length / 2 * font_size * 0.6 | ||||
@border_right = label_right if label_right > @border_right | ||||
end | ||||
X = 0 | ||||
Y = 1 | ||||
def x_range | ||||
max_value = @data.collect{|x| x[:data][X][-1] }.max | ||||
min_value = @data.collect{|x| x[:data][X][0] }.min | ||||
min_value = min_value<min_x_value ? min_value : min_x_value if min_x_value | ||||
range = max_value - min_value | ||||
right_pad = range == 0 ? 10 : range / 20.0 | ||||
scale_range = (max_value + right_pad) - min_value | ||||
scale_division = scale_x_divisions || (scale_range / 10.0) | ||||
if scale_x_integers | ||||
scale_division = scale_division < 1 ? 1 : scale_division.round | ||||
end | ||||
[min_value, max_value, scale_division] | ||||
end | ||||
def get_x_values | ||||
min_value, max_value, scale_division = x_range | ||||
rv = [] | ||||
min_value.step( max_value, scale_division ) {|v| rv << v} | ||||
return rv | ||||
end | ||||
alias :get_x_labels :get_x_values | ||||
def field_width | ||||
values = get_x_values | ||||
max = @data.collect{|x| x[:data][X][-1]}.max | ||||
dx = (max - values[-1]).to_f / (values[-1] - values[-2]) | ||||
(@graph_width.to_f - font_size*2*right_font) / | ||||
(values.length + dx - right_align) | ||||
end | ||||
def y_range | ||||
max_value = @data.collect{|x| x[:data][Y].max }.max | ||||
min_value = @data.collect{|x| x[:data][Y].min }.min | ||||
min_value = min_value<min_y_value ? min_value : min_y_value if min_y_value | ||||
range = max_value - min_value | ||||
top_pad = range == 0 ? 10 : range / 20.0 | ||||
scale_range = (max_value + top_pad) - min_value | ||||
scale_division = scale_y_divisions || (scale_range / 10.0) | ||||
if scale_y_integers | ||||
scale_division = scale_division < 1 ? 1 : scale_division.round | ||||
end | ||||
return [min_value, max_value, scale_division] | ||||
end | ||||
def get_y_values | ||||
min_value, max_value, scale_division = y_range | ||||
rv = [] | ||||
min_value.step( max_value, scale_division ) {|v| rv << v} | ||||
return rv | ||||
end | ||||
alias :get_y_labels :get_y_values | ||||
def field_height | ||||
values = get_y_values | ||||
max = @data.collect{|x| x[:data][Y].max }.max | ||||
if values.length == 1 | ||||
dx = values[-1] | ||||
else | ||||
dx = (max - values[-1]).to_f / (values[-1] - values[-2]) | ||||
end | ||||
(@graph_height.to_f - font_size*2*top_font) / | ||||
(values.length + dx - top_align) | ||||
end | ||||
def draw_data | ||||
line = 1 | ||||
x_min, x_max, x_div = x_range | ||||
y_min, y_max, y_div = y_range | ||||
x_step = (@graph_width.to_f - font_size*2) / (x_max-x_min) | ||||
y_step = (@graph_height.to_f - font_size*2) / (y_max-y_min) | ||||
for data in @data | ||||
x_points = data[:data][X] | ||||
y_points = data[:data][Y] | ||||
lpath = "L" | ||||
x_start = 0 | ||||
y_start = 0 | ||||
x_points.each_index { |idx| | ||||
x = (x_points[idx] - x_min) * x_step | ||||
y = @graph_height - (y_points[idx] - y_min) * y_step | ||||
x_start, y_start = x,y if idx == 0 | ||||
lpath << "#{x} #{y} " | ||||
} | ||||
if area_fill | ||||
@graph.add_element( "path", { | ||||
"d" => "M#{x_start} #@graph_height #{lpath} V#@graph_height Z", | ||||
"class" => "fill#{line}" | ||||
}) | ||||
end | ||||
@graph.add_element( "path", { | ||||
"d" => "M#{x_start} #{y_start} #{lpath}", | ||||
"class" => "line#{line}" | ||||
}) | ||||
if show_data_points || show_data_values | ||||
x_points.each_index { |idx| | ||||
x = (x_points[idx] - x_min) * x_step | ||||
y = @graph_height - (y_points[idx] - y_min) * y_step | ||||
if show_data_points | ||||
@graph.add_element( "circle", { | ||||
"cx" => x.to_s, | ||||
"cy" => y.to_s, | ||||
"r" => "2.5", | ||||
"class" => "dataPoint#{line}" | ||||
}) | ||||
add_popup(x, y, format( x_points[idx], y_points[idx] )) if add_popups | ||||
end | ||||
make_datapoint_text( x, y-6, y_points[idx] ) if show_data_values | ||||
} | ||||
end | ||||
line += 1 | ||||
end | ||||
end | ||||
def format x, y | ||||
"(#{(x * 100).to_i / 100}, #{(y * 100).to_i / 100})" | ||||
end | ||||
def get_css | ||||
return <<EOL | ||||
/* default line styles */ | ||||
.line1{ | ||||
fill: none; | ||||
stroke: #ff0000; | ||||
stroke-width: 1px; | ||||
} | ||||
.line2{ | ||||
fill: none; | ||||
stroke: #0000ff; | ||||
stroke-width: 1px; | ||||
} | ||||
.line3{ | ||||
fill: none; | ||||
stroke: #00ff00; | ||||
stroke-width: 1px; | ||||
} | ||||
.line4{ | ||||
fill: none; | ||||
stroke: #ffcc00; | ||||
stroke-width: 1px; | ||||
} | ||||
.line5{ | ||||
fill: none; | ||||
stroke: #00ccff; | ||||
stroke-width: 1px; | ||||
} | ||||
.line6{ | ||||
fill: none; | ||||
stroke: #ff00ff; | ||||
stroke-width: 1px; | ||||
} | ||||
.line7{ | ||||
fill: none; | ||||
stroke: #00ffff; | ||||
stroke-width: 1px; | ||||
} | ||||
.line8{ | ||||
fill: none; | ||||
stroke: #ffff00; | ||||
stroke-width: 1px; | ||||
} | ||||
.line9{ | ||||
fill: none; | ||||
stroke: #ccc6666; | ||||
stroke-width: 1px; | ||||
} | ||||
.line10{ | ||||
fill: none; | ||||
stroke: #663399; | ||||
stroke-width: 1px; | ||||
} | ||||
.line11{ | ||||
fill: none; | ||||
stroke: #339900; | ||||
stroke-width: 1px; | ||||
} | ||||
.line12{ | ||||
fill: none; | ||||
stroke: #9966FF; | ||||
stroke-width: 1px; | ||||
} | ||||
/* default fill styles */ | ||||
.fill1{ | ||||
fill: #cc0000; | ||||
fill-opacity: 0.2; | ||||
stroke: none; | ||||
} | ||||
.fill2{ | ||||
fill: #0000cc; | ||||
fill-opacity: 0.2; | ||||
stroke: none; | ||||
} | ||||
.fill3{ | ||||
fill: #00cc00; | ||||
fill-opacity: 0.2; | ||||
stroke: none; | ||||
} | ||||
.fill4{ | ||||
fill: #ffcc00; | ||||
fill-opacity: 0.2; | ||||
stroke: none; | ||||
} | ||||
.fill5{ | ||||
fill: #00ccff; | ||||
fill-opacity: 0.2; | ||||
stroke: none; | ||||
} | ||||
.fill6{ | ||||
fill: #ff00ff; | ||||
fill-opacity: 0.2; | ||||
stroke: none; | ||||
} | ||||
.fill7{ | ||||
fill: #00ffff; | ||||
fill-opacity: 0.2; | ||||
stroke: none; | ||||
} | ||||
.fill8{ | ||||
fill: #ffff00; | ||||
fill-opacity: 0.2; | ||||
stroke: none; | ||||
} | ||||
.fill9{ | ||||
fill: #cc6666; | ||||
fill-opacity: 0.2; | ||||
stroke: none; | ||||
} | ||||
.fill10{ | ||||
fill: #663399; | ||||
fill-opacity: 0.2; | ||||
stroke: none; | ||||
} | ||||
.fill11{ | ||||
fill: #339900; | ||||
fill-opacity: 0.2; | ||||
stroke: none; | ||||
} | ||||
.fill12{ | ||||
fill: #9966FF; | ||||
fill-opacity: 0.2; | ||||
stroke: none; | ||||
} | ||||
/* default line styles */ | ||||
.key1,.dataPoint1{ | ||||
fill: #ff0000; | ||||
stroke: none; | ||||
stroke-width: 1px; | ||||
} | ||||
.key2,.dataPoint2{ | ||||
fill: #0000ff; | ||||
stroke: none; | ||||
stroke-width: 1px; | ||||
} | ||||
.key3,.dataPoint3{ | ||||
fill: #00ff00; | ||||
stroke: none; | ||||
stroke-width: 1px; | ||||
} | ||||
.key4,.dataPoint4{ | ||||
fill: #ffcc00; | ||||
stroke: none; | ||||
stroke-width: 1px; | ||||
} | ||||
.key5,.dataPoint5{ | ||||
fill: #00ccff; | ||||
stroke: none; | ||||
stroke-width: 1px; | ||||
} | ||||
.key6,.dataPoint6{ | ||||
fill: #ff00ff; | ||||
stroke: none; | ||||
stroke-width: 1px; | ||||
} | ||||
.key7,.dataPoint7{ | ||||
fill: #00ffff; | ||||
stroke: none; | ||||
stroke-width: 1px; | ||||
} | ||||
.key8,.dataPoint8{ | ||||
fill: #ffff00; | ||||
stroke: none; | ||||
stroke-width: 1px; | ||||
} | ||||
.key9,.dataPoint9{ | ||||
fill: #cc6666; | ||||
stroke: none; | ||||
stroke-width: 1px; | ||||
} | ||||
.key10,.dataPoint10{ | ||||
fill: #663399; | ||||
stroke: none; | ||||
stroke-width: 1px; | ||||
} | ||||
.key11,.dataPoint11{ | ||||
fill: #339900; | ||||
stroke: none; | ||||
stroke-width: 1px; | ||||
} | ||||
.key12,.dataPoint12{ | ||||
fill: #9966FF; | ||||
stroke: none; | ||||
stroke-width: 1px; | ||||
} | ||||
EOL | ||||
end | ||||
end | ||||
end | ||||
end | ||||