You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							501 lines
						
					
					
						
							12 KiB
						
					
					
				
			
		
		
	
	
							501 lines
						
					
					
						
							12 KiB
						
					
					
				| 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
 |