Chris@909: begin Chris@909: require 'zlib' Chris@909: @@__have_zlib = true Chris@909: rescue Chris@909: @@__have_zlib = false Chris@909: end Chris@909: Chris@909: require 'rexml/document' Chris@909: Chris@909: module SVG Chris@909: module Graph Chris@909: VERSION = '@ANT_VERSION@' Chris@909: Chris@909: # === Base object for generating SVG Graphs Chris@909: # Chris@909: # == Synopsis Chris@909: # Chris@909: # This class is only used as a superclass of specialized charts. Do not Chris@909: # attempt to use this class directly, unless creating a new chart type. Chris@909: # Chris@909: # For examples of how to subclass this class, see the existing specific Chris@909: # subclasses, such as SVG::Graph::Pie. Chris@909: # Chris@909: # == Examples Chris@909: # Chris@909: # For examples of how to use this package, see either the test files, or Chris@909: # the documentation for the specific class you want to use. Chris@909: # Chris@909: # * file:test/plot.rb Chris@909: # * file:test/single.rb Chris@909: # * file:test/test.rb Chris@909: # * file:test/timeseries.rb Chris@909: # Chris@909: # == Description Chris@909: # Chris@909: # This package should be used as a base for creating SVG graphs. Chris@909: # Chris@909: # == Acknowledgements Chris@909: # Chris@909: # Leo Lapworth for creating the SVG::TT::Graph package which this Ruby Chris@909: # port is based on. Chris@909: # Chris@909: # Stephen Morgan for creating the TT template and SVG. Chris@909: # Chris@909: # == See Chris@909: # Chris@909: # * SVG::Graph::BarHorizontal Chris@909: # * SVG::Graph::Bar Chris@909: # * SVG::Graph::Line Chris@909: # * SVG::Graph::Pie Chris@909: # * SVG::Graph::Plot Chris@909: # * SVG::Graph::TimeSeries Chris@909: # Chris@909: # == Author Chris@909: # Chris@909: # Sean E. Russell Chris@909: # Chris@909: # Copyright 2004 Sean E. Russell Chris@909: # This software is available under the Ruby license[LICENSE.txt] Chris@909: # Chris@909: class Graph Chris@909: include REXML Chris@909: Chris@909: # Initialize the graph object with the graph settings. You won't Chris@909: # instantiate this class directly; see the subclass for options. Chris@909: # [width] 500 Chris@909: # [height] 300 Chris@909: # [show_x_guidelines] false Chris@909: # [show_y_guidelines] true Chris@909: # [show_data_values] true Chris@909: # [min_scale_value] 0 Chris@909: # [show_x_labels] true Chris@909: # [stagger_x_labels] false Chris@909: # [rotate_x_labels] false Chris@909: # [step_x_labels] 1 Chris@909: # [step_include_first_x_label] true Chris@909: # [show_y_labels] true Chris@909: # [rotate_y_labels] false Chris@909: # [scale_integers] false Chris@909: # [show_x_title] false Chris@909: # [x_title] 'X Field names' Chris@909: # [show_y_title] false Chris@909: # [y_title_text_direction] :bt Chris@909: # [y_title] 'Y Scale' Chris@909: # [show_graph_title] false Chris@909: # [graph_title] 'Graph Title' Chris@909: # [show_graph_subtitle] false Chris@909: # [graph_subtitle] 'Graph Sub Title' Chris@909: # [key] true, Chris@909: # [key_position] :right, # bottom or righ Chris@909: # [font_size] 12 Chris@909: # [title_font_size] 16 Chris@909: # [subtitle_font_size] 14 Chris@909: # [x_label_font_size] 12 Chris@909: # [x_title_font_size] 14 Chris@909: # [y_label_font_size] 12 Chris@909: # [y_title_font_size] 14 Chris@909: # [key_font_size] 10 Chris@909: # [no_css] false Chris@909: # [add_popups] false Chris@909: def initialize( config ) Chris@909: @config = config Chris@909: Chris@909: self.top_align = self.top_font = self.right_align = self.right_font = 0 Chris@909: Chris@909: init_with({ Chris@909: :width => 500, Chris@909: :height => 300, Chris@909: :show_x_guidelines => false, Chris@909: :show_y_guidelines => true, Chris@909: :show_data_values => true, Chris@909: Chris@909: # :min_scale_value => 0, Chris@909: Chris@909: :show_x_labels => true, Chris@909: :stagger_x_labels => false, Chris@909: :rotate_x_labels => false, Chris@909: :step_x_labels => 1, Chris@909: :step_include_first_x_label => true, Chris@909: Chris@909: :show_y_labels => true, Chris@909: :rotate_y_labels => false, Chris@909: :stagger_y_labels => false, Chris@909: :scale_integers => false, Chris@909: Chris@909: :show_x_title => false, Chris@909: :x_title => 'X Field names', Chris@909: Chris@909: :show_y_title => false, Chris@909: :y_title_text_direction => :bt, Chris@909: :y_title => 'Y Scale', Chris@909: Chris@909: :show_graph_title => false, Chris@909: :graph_title => 'Graph Title', Chris@909: :show_graph_subtitle => false, Chris@909: :graph_subtitle => 'Graph Sub Title', Chris@909: :key => true, Chris@909: :key_position => :right, # bottom or right Chris@909: Chris@909: :font_size =>12, Chris@909: :title_font_size =>16, Chris@909: :subtitle_font_size =>14, Chris@909: :x_label_font_size =>12, Chris@909: :x_title_font_size =>14, Chris@909: :y_label_font_size =>12, Chris@909: :y_title_font_size =>14, Chris@909: :key_font_size =>10, Chris@909: Chris@909: :no_css =>false, Chris@909: :add_popups =>false, Chris@909: }) Chris@909: Chris@909: set_defaults if respond_to? :set_defaults Chris@909: Chris@909: init_with config Chris@909: end Chris@909: Chris@909: Chris@909: # This method allows you do add data to the graph object. Chris@909: # It can be called several times to add more data sets in. Chris@909: # Chris@909: # data_sales_02 = [12, 45, 21]; Chris@909: # Chris@909: # graph.add_data({ Chris@909: # :data => data_sales_02, Chris@909: # :title => 'Sales 2002' Chris@909: # }) Chris@909: def add_data conf Chris@909: @data = [] unless defined? @data Chris@909: Chris@909: if conf[:data] and conf[:data].kind_of? Array Chris@909: @data << conf Chris@909: else Chris@909: raise "No data provided by #{conf.inspect}" Chris@909: end Chris@909: end Chris@909: Chris@909: Chris@909: # This method removes all data from the object so that you can Chris@909: # reuse it to create a new graph but with the same config options. Chris@909: # Chris@909: # graph.clear_data Chris@909: def clear_data Chris@909: @data = [] Chris@909: end Chris@909: Chris@909: Chris@909: # This method processes the template with the data and Chris@909: # config which has been set and returns the resulting SVG. Chris@909: # Chris@909: # This method will croak unless at least one data set has Chris@909: # been added to the graph object. Chris@909: # Chris@909: # print graph.burn Chris@909: def burn Chris@909: raise "No data available" unless @data.size > 0 Chris@909: Chris@909: calculations if respond_to? :calculations Chris@909: Chris@909: start_svg Chris@909: calculate_graph_dimensions Chris@909: @foreground = Element.new( "g" ) Chris@909: draw_graph Chris@909: draw_titles Chris@909: draw_legend Chris@909: draw_data Chris@909: @graph.add_element( @foreground ) Chris@909: style Chris@909: Chris@909: data = "" Chris@909: @doc.write( data, 0 ) Chris@909: Chris@909: if @config[:compress] Chris@909: if @@__have_zlib Chris@909: inp, out = IO.pipe Chris@909: gz = Zlib::GzipWriter.new( out ) Chris@909: gz.write data Chris@909: gz.close Chris@909: data = inp.read Chris@909: else Chris@909: data << ""; Chris@909: end Chris@909: end Chris@909: Chris@909: return data Chris@909: end Chris@909: Chris@909: Chris@909: # Set the height of the graph box, this is the total height Chris@909: # of the SVG box created - not the graph it self which auto Chris@909: # scales to fix the space. Chris@909: attr_accessor :height Chris@909: # Set the width of the graph box, this is the total width Chris@909: # of the SVG box created - not the graph it self which auto Chris@909: # scales to fix the space. Chris@909: attr_accessor :width Chris@909: # Set the path to an external stylesheet, set to '' if Chris@909: # you want to revert back to using the defaut internal version. Chris@909: # Chris@909: # To create an external stylesheet create a graph using the Chris@909: # default internal version and copy the stylesheet section to Chris@909: # an external file and edit from there. Chris@909: attr_accessor :style_sheet Chris@909: # (Bool) Show the value of each element of data on the graph Chris@909: attr_accessor :show_data_values Chris@909: # The point at which the Y axis starts, defaults to '0', Chris@909: # if set to nil it will default to the minimum data value. Chris@909: attr_accessor :min_scale_value Chris@909: # Whether to show labels on the X axis or not, defaults Chris@909: # to true, set to false if you want to turn them off. Chris@909: attr_accessor :show_x_labels Chris@909: # This puts the X labels at alternative levels so if they Chris@909: # are long field names they will not overlap so easily. Chris@909: # Default it false, to turn on set to true. Chris@909: attr_accessor :stagger_x_labels Chris@909: # This puts the Y labels at alternative levels so if they Chris@909: # are long field names they will not overlap so easily. Chris@909: # Default it false, to turn on set to true. Chris@909: attr_accessor :stagger_y_labels Chris@909: # This turns the X axis labels by 90 degrees. Chris@909: # Default it false, to turn on set to true. Chris@909: attr_accessor :rotate_x_labels Chris@909: # This turns the Y axis labels by 90 degrees. Chris@909: # Default it false, to turn on set to true. Chris@909: attr_accessor :rotate_y_labels Chris@909: # How many "steps" to use between displayed X axis labels, Chris@909: # a step of one means display every label, a step of two results Chris@909: # in every other label being displayed (label label label), Chris@909: # a step of three results in every third label being displayed Chris@909: # (label label label) and so on. Chris@909: attr_accessor :step_x_labels Chris@909: # Whether to (when taking "steps" between X axis labels) step from Chris@909: # the first label (i.e. always include the first label) or step from Chris@909: # the X axis origin (i.e. start with a gap if step_x_labels is greater Chris@909: # than one). Chris@909: attr_accessor :step_include_first_x_label Chris@909: # Whether to show labels on the Y axis or not, defaults Chris@909: # to true, set to false if you want to turn them off. Chris@909: attr_accessor :show_y_labels Chris@909: # Ensures only whole numbers are used as the scale divisions. Chris@909: # Default it false, to turn on set to true. This has no effect if Chris@909: # scale divisions are less than 1. Chris@909: attr_accessor :scale_integers Chris@909: # This defines the gap between markers on the Y axis, Chris@909: # default is a 10th of the max_value, e.g. you will have Chris@909: # 10 markers on the Y axis. NOTE: do not set this too Chris@909: # low - you are limited to 999 markers, after that the Chris@909: # graph won't generate. Chris@909: attr_accessor :scale_divisions Chris@909: # Whether to show the title under the X axis labels, Chris@909: # default is false, set to true to show. Chris@909: attr_accessor :show_x_title Chris@909: # What the title under X axis should be, e.g. 'Months'. Chris@909: attr_accessor :x_title Chris@909: # Whether to show the title under the Y axis labels, Chris@909: # default is false, set to true to show. Chris@909: attr_accessor :show_y_title Chris@909: # Aligns writing mode for Y axis label. Chris@909: # Defaults to :bt (Bottom to Top). Chris@909: # Change to :tb (Top to Bottom) to reverse. Chris@909: attr_accessor :y_title_text_direction Chris@909: # What the title under Y axis should be, e.g. 'Sales in thousands'. Chris@909: attr_accessor :y_title Chris@909: # Whether to show a title on the graph, defaults Chris@909: # to false, set to true to show. Chris@909: attr_accessor :show_graph_title Chris@909: # What the title on the graph should be. Chris@909: attr_accessor :graph_title Chris@909: # Whether to show a subtitle on the graph, defaults Chris@909: # to false, set to true to show. Chris@909: attr_accessor :show_graph_subtitle Chris@909: # What the subtitle on the graph should be. Chris@909: attr_accessor :graph_subtitle Chris@909: # Whether to show a key, defaults to false, set to Chris@909: # true if you want to show it. Chris@909: attr_accessor :key Chris@909: # Where the key should be positioned, defaults to Chris@909: # :right, set to :bottom if you want to move it. Chris@909: attr_accessor :key_position Chris@909: # Set the font size (in points) of the data point labels Chris@909: attr_accessor :font_size Chris@909: # Set the font size of the X axis labels Chris@909: attr_accessor :x_label_font_size Chris@909: # Set the font size of the X axis title Chris@909: attr_accessor :x_title_font_size Chris@909: # Set the font size of the Y axis labels Chris@909: attr_accessor :y_label_font_size Chris@909: # Set the font size of the Y axis title Chris@909: attr_accessor :y_title_font_size Chris@909: # Set the title font size Chris@909: attr_accessor :title_font_size Chris@909: # Set the subtitle font size Chris@909: attr_accessor :subtitle_font_size Chris@909: # Set the key font size Chris@909: attr_accessor :key_font_size Chris@909: # Show guidelines for the X axis Chris@909: attr_accessor :show_x_guidelines Chris@909: # Show guidelines for the Y axis Chris@909: attr_accessor :show_y_guidelines Chris@909: # Do not use CSS if set to true. Many SVG viewers do not support CSS, but Chris@909: # not using CSS can result in larger SVGs as well as making it impossible to Chris@909: # change colors after the chart is generated. Defaults to false. Chris@909: attr_accessor :no_css Chris@909: # Add popups for the data points on some graphs Chris@909: attr_accessor :add_popups Chris@909: Chris@909: Chris@909: protected Chris@909: Chris@909: def sort( *arrys ) Chris@909: sort_multiple( arrys ) Chris@909: end Chris@909: Chris@909: # Overwrite configuration options with supplied options. Used Chris@909: # by subclasses. Chris@909: def init_with config Chris@909: config.each { |key, value| Chris@909: self.send((key.to_s+"=").to_sym, value ) if respond_to? key.to_sym Chris@909: } Chris@909: end Chris@909: Chris@909: attr_accessor :top_align, :top_font, :right_align, :right_font Chris@909: Chris@909: KEY_BOX_SIZE = 12 Chris@909: Chris@909: # Override this (and call super) to change the margin to the left Chris@909: # of the plot area. Results in @border_left being set. Chris@909: def calculate_left_margin Chris@909: @border_left = 7 Chris@909: # Check for Y labels Chris@909: max_y_label_height_px = rotate_y_labels ? Chris@909: y_label_font_size : Chris@909: get_y_labels.max{|a,b| Chris@909: a.to_s.length<=>b.to_s.length Chris@909: }.to_s.length * y_label_font_size * 0.6 Chris@909: @border_left += max_y_label_height_px if show_y_labels Chris@909: @border_left += max_y_label_height_px + 10 if stagger_y_labels Chris@909: @border_left += y_title_font_size + 5 if show_y_title Chris@909: end Chris@909: Chris@909: Chris@909: # Calculates the width of the widest Y label. This will be the Chris@909: # character height if the Y labels are rotated Chris@909: def max_y_label_width_px Chris@909: return font_size if rotate_y_labels Chris@909: end Chris@909: Chris@909: Chris@909: # Override this (and call super) to change the margin to the right Chris@909: # of the plot area. Results in @border_right being set. Chris@909: def calculate_right_margin Chris@909: @border_right = 7 Chris@909: if key and key_position == :right Chris@909: val = keys.max { |a,b| a.length <=> b.length } Chris@909: @border_right += val.length * key_font_size * 0.6 Chris@909: @border_right += KEY_BOX_SIZE Chris@909: @border_right += 10 # Some padding around the box Chris@909: end Chris@909: end Chris@909: Chris@909: Chris@909: # Override this (and call super) to change the margin to the top Chris@909: # of the plot area. Results in @border_top being set. Chris@909: def calculate_top_margin Chris@909: @border_top = 5 Chris@909: @border_top += title_font_size if show_graph_title Chris@909: @border_top += 5 Chris@909: @border_top += subtitle_font_size if show_graph_subtitle Chris@909: end Chris@909: Chris@909: Chris@909: # Adds pop-up point information to a graph. Chris@909: def add_popup( x, y, label ) Chris@909: txt_width = label.length * font_size * 0.6 + 10 Chris@909: tx = (x+txt_width > width ? x-5 : x+5) Chris@909: t = @foreground.add_element( "text", { Chris@909: "x" => tx.to_s, Chris@909: "y" => (y - font_size).to_s, Chris@909: "visibility" => "hidden", Chris@909: }) Chris@909: t.attributes["style"] = "fill: #000; "+ Chris@909: (x+txt_width > width ? "text-anchor: end;" : "text-anchor: start;") Chris@909: t.text = label.to_s Chris@909: t.attributes["id"] = t.object_id.to_s Chris@909: Chris@909: @foreground.add_element( "circle", { Chris@909: "cx" => x.to_s, Chris@909: "cy" => y.to_s, Chris@909: "r" => "10", Chris@909: "style" => "opacity: 0", Chris@909: "onmouseover" => Chris@909: "document.getElementById(#{t.object_id}).setAttribute('visibility', 'visible' )", Chris@909: "onmouseout" => Chris@909: "document.getElementById(#{t.object_id}).setAttribute('visibility', 'hidden' )", Chris@909: }) Chris@909: Chris@909: end Chris@909: Chris@909: Chris@909: # Override this (and call super) to change the margin to the bottom Chris@909: # of the plot area. Results in @border_bottom being set. Chris@909: def calculate_bottom_margin Chris@909: @border_bottom = 7 Chris@909: if key and key_position == :bottom Chris@909: @border_bottom += @data.size * (font_size + 5) Chris@909: @border_bottom += 10 Chris@909: end Chris@909: if show_x_labels Chris@909: max_x_label_height_px = (not rotate_x_labels) ? Chris@909: x_label_font_size : Chris@909: get_x_labels.max{|a,b| Chris@909: a.to_s.length<=>b.to_s.length Chris@909: }.to_s.length * x_label_font_size * 0.6 Chris@909: @border_bottom += max_x_label_height_px Chris@909: @border_bottom += max_x_label_height_px + 10 if stagger_x_labels Chris@909: end Chris@909: @border_bottom += x_title_font_size + 5 if show_x_title Chris@909: end Chris@909: Chris@909: Chris@909: # Draws the background, axis, and labels. Chris@909: def draw_graph Chris@909: @graph = @root.add_element( "g", { Chris@909: "transform" => "translate( #@border_left #@border_top )" Chris@909: }) Chris@909: Chris@909: # Background Chris@909: @graph.add_element( "rect", { Chris@909: "x" => "0", Chris@909: "y" => "0", Chris@909: "width" => @graph_width.to_s, Chris@909: "height" => @graph_height.to_s, Chris@909: "class" => "graphBackground" Chris@909: }) Chris@909: Chris@909: # Axis Chris@909: @graph.add_element( "path", { Chris@909: "d" => "M 0 0 v#@graph_height", Chris@909: "class" => "axis", Chris@909: "id" => "xAxis" Chris@909: }) Chris@909: @graph.add_element( "path", { Chris@909: "d" => "M 0 #@graph_height h#@graph_width", Chris@909: "class" => "axis", Chris@909: "id" => "yAxis" Chris@909: }) Chris@909: Chris@909: draw_x_labels Chris@909: draw_y_labels Chris@909: end Chris@909: Chris@909: Chris@909: # Where in the X area the label is drawn Chris@909: # Centered in the field, should be width/2. Start, 0. Chris@909: def x_label_offset( width ) Chris@909: 0 Chris@909: end Chris@909: Chris@909: def make_datapoint_text( x, y, value, style="" ) Chris@909: if show_data_values Chris@909: @foreground.add_element( "text", { Chris@909: "x" => x.to_s, Chris@909: "y" => y.to_s, Chris@909: "class" => "dataPointLabel", Chris@909: "style" => "#{style} stroke: #fff; stroke-width: 2;" Chris@909: }).text = value.to_s Chris@909: text = @foreground.add_element( "text", { Chris@909: "x" => x.to_s, Chris@909: "y" => y.to_s, Chris@909: "class" => "dataPointLabel" Chris@909: }) Chris@909: text.text = value.to_s Chris@909: text.attributes["style"] = style if style.length > 0 Chris@909: end Chris@909: end Chris@909: Chris@909: Chris@909: # Draws the X axis labels Chris@909: def draw_x_labels Chris@909: stagger = x_label_font_size + 5 Chris@909: if show_x_labels Chris@909: label_width = field_width Chris@909: Chris@909: count = 0 Chris@909: for label in get_x_labels Chris@909: if step_include_first_x_label == true then Chris@909: step = count % step_x_labels Chris@909: else Chris@909: step = (count + 1) % step_x_labels Chris@909: end Chris@909: Chris@909: if step == 0 then Chris@909: text = @graph.add_element( "text" ) Chris@909: text.attributes["class"] = "xAxisLabels" Chris@909: text.text = label.to_s Chris@909: Chris@909: x = count * label_width + x_label_offset( label_width ) Chris@909: y = @graph_height + x_label_font_size + 3 Chris@909: t = 0 - (font_size / 2) Chris@909: Chris@909: if stagger_x_labels and count % 2 == 1 Chris@909: y += stagger Chris@909: @graph.add_element( "path", { Chris@909: "d" => "M#{x} #@graph_height v#{stagger}", Chris@909: "class" => "staggerGuideLine" Chris@909: }) Chris@909: end Chris@909: Chris@909: text.attributes["x"] = x.to_s Chris@909: text.attributes["y"] = y.to_s Chris@909: if rotate_x_labels Chris@909: text.attributes["transform"] = Chris@909: "rotate( 90 #{x} #{y-x_label_font_size} )"+ Chris@909: " translate( 0 -#{x_label_font_size/4} )" Chris@909: text.attributes["style"] = "text-anchor: start" Chris@909: else Chris@909: text.attributes["style"] = "text-anchor: middle" Chris@909: end Chris@909: end Chris@909: Chris@909: draw_x_guidelines( label_width, count ) if show_x_guidelines Chris@909: count += 1 Chris@909: end Chris@909: end Chris@909: end Chris@909: Chris@909: Chris@909: # Where in the Y area the label is drawn Chris@909: # Centered in the field, should be width/2. Start, 0. Chris@909: def y_label_offset( height ) Chris@909: 0 Chris@909: end Chris@909: Chris@909: Chris@909: def field_width Chris@909: (@graph_width.to_f - font_size*2*right_font) / Chris@909: (get_x_labels.length - right_align) Chris@909: end Chris@909: Chris@909: Chris@909: def field_height Chris@909: (@graph_height.to_f - font_size*2*top_font) / Chris@909: (get_y_labels.length - top_align) Chris@909: end Chris@909: Chris@909: Chris@909: # Draws the Y axis labels Chris@909: def draw_y_labels Chris@909: stagger = y_label_font_size + 5 Chris@909: if show_y_labels Chris@909: label_height = field_height Chris@909: Chris@909: count = 0 Chris@909: y_offset = @graph_height + y_label_offset( label_height ) Chris@909: y_offset += font_size/1.2 unless rotate_y_labels Chris@909: for label in get_y_labels Chris@909: y = y_offset - (label_height * count) Chris@909: x = rotate_y_labels ? 0 : -3 Chris@909: Chris@909: if stagger_y_labels and count % 2 == 1 Chris@909: x -= stagger Chris@909: @graph.add_element( "path", { Chris@909: "d" => "M#{x} #{y} h#{stagger}", Chris@909: "class" => "staggerGuideLine" Chris@909: }) Chris@909: end Chris@909: Chris@909: text = @graph.add_element( "text", { Chris@909: "x" => x.to_s, Chris@909: "y" => y.to_s, Chris@909: "class" => "yAxisLabels" Chris@909: }) Chris@909: text.text = label.to_s Chris@909: if rotate_y_labels Chris@909: text.attributes["transform"] = "translate( -#{font_size} 0 ) "+ Chris@909: "rotate( 90 #{x} #{y} ) " Chris@909: text.attributes["style"] = "text-anchor: middle" Chris@909: else Chris@909: text.attributes["y"] = (y - (y_label_font_size/2)).to_s Chris@909: text.attributes["style"] = "text-anchor: end" Chris@909: end Chris@909: draw_y_guidelines( label_height, count ) if show_y_guidelines Chris@909: count += 1 Chris@909: end Chris@909: end Chris@909: end Chris@909: Chris@909: Chris@909: # Draws the X axis guidelines Chris@909: def draw_x_guidelines( label_height, count ) Chris@909: if count != 0 Chris@909: @graph.add_element( "path", { Chris@909: "d" => "M#{label_height*count} 0 v#@graph_height", Chris@909: "class" => "guideLines" Chris@909: }) Chris@909: end Chris@909: end Chris@909: Chris@909: Chris@909: # Draws the Y axis guidelines Chris@909: def draw_y_guidelines( label_height, count ) Chris@909: if count != 0 Chris@909: @graph.add_element( "path", { Chris@909: "d" => "M0 #{@graph_height-(label_height*count)} h#@graph_width", Chris@909: "class" => "guideLines" Chris@909: }) Chris@909: end Chris@909: end Chris@909: Chris@909: Chris@909: # Draws the graph title and subtitle Chris@909: def draw_titles Chris@909: if show_graph_title Chris@909: @root.add_element( "text", { Chris@909: "x" => (width / 2).to_s, Chris@909: "y" => (title_font_size).to_s, Chris@909: "class" => "mainTitle" Chris@909: }).text = graph_title.to_s Chris@909: end Chris@909: Chris@909: if show_graph_subtitle Chris@909: y_subtitle = show_graph_title ? Chris@909: title_font_size + 10 : Chris@909: subtitle_font_size Chris@909: @root.add_element("text", { Chris@909: "x" => (width / 2).to_s, Chris@909: "y" => (y_subtitle).to_s, Chris@909: "class" => "subTitle" Chris@909: }).text = graph_subtitle.to_s Chris@909: end Chris@909: Chris@909: if show_x_title Chris@909: y = @graph_height + @border_top + x_title_font_size Chris@909: if show_x_labels Chris@909: y += x_label_font_size + 5 if stagger_x_labels Chris@909: y += x_label_font_size + 5 Chris@909: end Chris@909: x = width / 2 Chris@909: Chris@909: @root.add_element("text", { Chris@909: "x" => x.to_s, Chris@909: "y" => y.to_s, Chris@909: "class" => "xAxisTitle", Chris@909: }).text = x_title.to_s Chris@909: end Chris@909: Chris@909: if show_y_title Chris@909: x = y_title_font_size + (y_title_text_direction==:bt ? 3 : -3) Chris@909: y = height / 2 Chris@909: Chris@909: text = @root.add_element("text", { Chris@909: "x" => x.to_s, Chris@909: "y" => y.to_s, Chris@909: "class" => "yAxisTitle", Chris@909: }) Chris@909: text.text = y_title.to_s Chris@909: if y_title_text_direction == :bt Chris@909: text.attributes["transform"] = "rotate( -90, #{x}, #{y} )" Chris@909: else Chris@909: text.attributes["transform"] = "rotate( 90, #{x}, #{y} )" Chris@909: end Chris@909: end Chris@909: end Chris@909: Chris@909: def keys Chris@909: return @data.collect{ |d| d[:title] } Chris@909: end Chris@909: Chris@909: # Draws the legend on the graph Chris@909: def draw_legend Chris@909: if key Chris@909: group = @root.add_element( "g" ) Chris@909: Chris@909: key_count = 0 Chris@909: for key_name in keys Chris@909: y_offset = (KEY_BOX_SIZE * key_count) + (key_count * 5) Chris@909: group.add_element( "rect", { Chris@909: "x" => 0.to_s, Chris@909: "y" => y_offset.to_s, Chris@909: "width" => KEY_BOX_SIZE.to_s, Chris@909: "height" => KEY_BOX_SIZE.to_s, Chris@909: "class" => "key#{key_count+1}" Chris@909: }) Chris@909: group.add_element( "text", { Chris@909: "x" => (KEY_BOX_SIZE + 5).to_s, Chris@909: "y" => (y_offset + KEY_BOX_SIZE).to_s, Chris@909: "class" => "keyText" Chris@909: }).text = key_name.to_s Chris@909: key_count += 1 Chris@909: end Chris@909: Chris@909: case key_position Chris@909: when :right Chris@909: x_offset = @graph_width + @border_left + 10 Chris@909: y_offset = @border_top + 20 Chris@909: when :bottom Chris@909: x_offset = @border_left + 20 Chris@909: y_offset = @border_top + @graph_height + 5 Chris@909: if show_x_labels Chris@909: max_x_label_height_px = (not rotate_x_labels) ? Chris@909: x_label_font_size : Chris@909: get_x_labels.max{|a,b| Chris@909: a.to_s.length<=>b.to_s.length Chris@909: }.to_s.length * x_label_font_size * 0.6 Chris@909: x_label_font_size Chris@909: y_offset += max_x_label_height_px Chris@909: y_offset += max_x_label_height_px + 5 if stagger_x_labels Chris@909: end Chris@909: y_offset += x_title_font_size + 5 if show_x_title Chris@909: end Chris@909: group.attributes["transform"] = "translate(#{x_offset} #{y_offset})" Chris@909: end Chris@909: end Chris@909: Chris@909: Chris@909: private Chris@909: Chris@909: def sort_multiple( arrys, lo=0, hi=arrys[0].length-1 ) Chris@909: if lo < hi Chris@909: p = partition(arrys,lo,hi) Chris@909: sort_multiple(arrys, lo, p-1) Chris@909: sort_multiple(arrys, p+1, hi) Chris@909: end Chris@909: arrys Chris@909: end Chris@909: Chris@909: def partition( arrys, lo, hi ) Chris@909: p = arrys[0][lo] Chris@909: l = lo Chris@909: z = lo+1 Chris@909: while z <= hi Chris@909: if arrys[0][z] < p Chris@909: l += 1 Chris@909: arrys.each { |arry| arry[z], arry[l] = arry[l], arry[z] } Chris@909: end Chris@909: z += 1 Chris@909: end Chris@909: arrys.each { |arry| arry[lo], arry[l] = arry[l], arry[lo] } Chris@909: l Chris@909: end Chris@909: Chris@909: def style Chris@909: if no_css Chris@909: styles = parse_css Chris@909: @root.elements.each("//*[@class]") { |el| Chris@909: cl = el.attributes["class"] Chris@909: style = styles[cl] Chris@909: style += el.attributes["style"] if el.attributes["style"] Chris@909: el.attributes["style"] = style Chris@909: } Chris@909: end Chris@909: end Chris@909: Chris@909: def parse_css Chris@909: css = get_style Chris@909: rv = {} Chris@909: while css =~ /^(\.(\w+)(?:\s*,\s*\.\w+)*)\s*\{/m Chris@909: names_orig = names = $1 Chris@909: css = $' Chris@909: css =~ /([^}]+)\}/m Chris@909: content = $1 Chris@909: css = $' Chris@909: Chris@909: nms = [] Chris@909: while names =~ /^\s*,?\s*\.(\w+)/ Chris@909: nms << $1 Chris@909: names = $' Chris@909: end Chris@909: Chris@909: content = content.tr( "\n\t", " ") Chris@909: for name in nms Chris@909: current = rv[name] Chris@909: current = current ? current+"; "+content : content Chris@909: rv[name] = current.strip.squeeze(" ") Chris@909: end Chris@909: end Chris@909: return rv Chris@909: end Chris@909: Chris@909: Chris@909: # Override and place code to add defs here Chris@909: def add_defs defs Chris@909: end Chris@909: Chris@909: Chris@909: def start_svg Chris@909: # Base document Chris@909: @doc = Document.new Chris@909: @doc << XMLDecl.new Chris@909: @doc << DocType.new( %q{svg PUBLIC "-//W3C//DTD SVG 1.0//EN" } + Chris@909: %q{"http://www.w3.org/TR/2001/REC-SVG-20010904/DTD/svg10.dtd"} ) Chris@909: if style_sheet && style_sheet != '' Chris@909: @doc << Instruction.new( "xml-stylesheet", Chris@909: %Q{href="#{style_sheet}" type="text/css"} ) Chris@909: end Chris@909: @root = @doc.add_element( "svg", { Chris@909: "width" => width.to_s, Chris@909: "height" => height.to_s, Chris@909: "viewBox" => "0 0 #{width} #{height}", Chris@909: "xmlns" => "http://www.w3.org/2000/svg", Chris@909: "xmlns:xlink" => "http://www.w3.org/1999/xlink", Chris@909: "xmlns:a3" => "http://ns.adobe.com/AdobeSVGViewerExtensions/3.0/", Chris@909: "a3:scriptImplementation" => "Adobe" Chris@909: }) Chris@909: @root << Comment.new( " "+"\\"*66 ) Chris@909: @root << Comment.new( " Created with SVG::Graph " ) Chris@909: @root << Comment.new( " SVG::Graph by Sean E. Russell " ) Chris@909: @root << Comment.new( " Losely based on SVG::TT::Graph for Perl by"+ Chris@909: " Leo Lapworth & Stephan Morgan " ) Chris@909: @root << Comment.new( " "+"/"*66 ) Chris@909: Chris@909: defs = @root.add_element( "defs" ) Chris@909: add_defs defs Chris@909: if not(style_sheet && style_sheet != '') and !no_css Chris@909: @root << Comment.new(" include default stylesheet if none specified ") Chris@909: style = defs.add_element( "style", {"type"=>"text/css"} ) Chris@909: style << CData.new( get_style ) Chris@909: end Chris@909: Chris@909: @root << Comment.new( "SVG Background" ) Chris@909: @root.add_element( "rect", { Chris@909: "width" => width.to_s, Chris@909: "height" => height.to_s, Chris@909: "x" => "0", Chris@909: "y" => "0", Chris@909: "class" => "svgBackground" Chris@909: }) Chris@909: end Chris@909: Chris@909: Chris@909: def calculate_graph_dimensions Chris@909: calculate_left_margin Chris@909: calculate_right_margin Chris@909: calculate_bottom_margin Chris@909: calculate_top_margin Chris@909: @graph_width = width - @border_left - @border_right Chris@909: @graph_height = height - @border_top - @border_bottom Chris@909: end Chris@909: Chris@909: def get_style Chris@909: return <