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