Chris@1294: require 'rexml/document'
Chris@1294: require 'SVG/Graph/Graph'
Chris@1294: require 'SVG/Graph/BarBase'
Chris@1294: 
Chris@1294: module SVG
Chris@1294:   module Graph
Chris@1294:     # === Create presentation quality SVG bar graphs easily
Chris@1294:     #
Chris@1294:     # = Synopsis
Chris@1294:     #
Chris@1294:     #   require 'SVG/Graph/Bar'
Chris@1294:     #
Chris@1294:     #   fields = %w(Jan Feb Mar);
Chris@1294:     #   data_sales_02 = [12, 45, 21]
Chris@1294:     #
Chris@1294:     #   graph = SVG::Graph::Bar.new(
Chris@1294:     #     :height => 500,
Chris@1294:     #     :width => 300,
Chris@1294:     #     :fields => fields
Chris@1294:     #   )
Chris@1294:     #
Chris@1294:     #   graph.add_data(
Chris@1294:     #     :data => data_sales_02,
Chris@1294:     #     :title => 'Sales 2002'
Chris@1294:     #   )
Chris@1294:     #
Chris@1294:     #   print "Content-type: image/svg+xml\r\n\r\n"
Chris@1294:     #   print graph.burn
Chris@1294:     #
Chris@1294:     # = Description
Chris@1294:     #
Chris@1294:     # This object aims to allow you to easily create high quality
Chris@1294:     # SVG[http://www.w3c.org/tr/svg bar graphs. You can either use the default
Chris@1294:     # style sheet or supply your own. Either way there are many options which
Chris@1294:     # can be configured to give you control over how the graph is generated -
Chris@1294:     # with or without a key, data elements at each point, title, subtitle etc.
Chris@1294:     #
Chris@1294:     # = Notes
Chris@1294:     #
Chris@1294:     # The default stylesheet handles upto 12 data sets, if you
Chris@1294:     # use more you must create your own stylesheet and add the
Chris@1294:     # additional settings for the extra data sets. You will know
Chris@1294:     # if you go over 12 data sets as they will have no style and
Chris@1294:     # be in black.
Chris@1294:     #
Chris@1294:     # = Examples
Chris@1294:     #
Chris@1294:     # * http://germane-software.com/repositories/public/SVG/test/test.rb
Chris@1294:     #
Chris@1294:     # = See also
Chris@1294:     #
Chris@1294:     # * SVG::Graph::Graph
Chris@1294:     # * SVG::Graph::BarHorizontal
Chris@1294:     # * SVG::Graph::Line
Chris@1294:     # * SVG::Graph::Pie
Chris@1294:     # * SVG::Graph::Plot
Chris@1294:     # * SVG::Graph::TimeSeries
Chris@1294:     class Bar < BarBase
Chris@1294:       include REXML
Chris@1294: 
Chris@1294:       # See Graph::initialize and BarBase::set_defaults
Chris@1294:       def set_defaults 
Chris@1294:         super
Chris@1294:         self.top_align = self.top_font = 1
Chris@1294:       end
Chris@1294: 
Chris@1294:       protected
Chris@1294: 
Chris@1294:       def get_x_labels
Chris@1294:         @config[:fields]
Chris@1294:       end
Chris@1294: 
Chris@1294:       def get_y_labels
Chris@1294:         maxvalue = max_value
Chris@1294:         minvalue = min_value
Chris@1294:         range = maxvalue - minvalue
Chris@1294: 
Chris@1294:         top_pad = range == 0 ? 10 : range / 20.0
Chris@1294:         scale_range = (maxvalue + top_pad) - minvalue
Chris@1294: 
Chris@1294:         scale_division = scale_divisions || (scale_range / 10.0)
Chris@1294: 
Chris@1294:         if scale_integers
Chris@1294:           scale_division = scale_division < 1 ? 1 : scale_division.round
Chris@1294:         end
Chris@1294: 
Chris@1294:         rv = []
Chris@1294:         maxvalue = maxvalue%scale_division == 0 ? 
Chris@1294:           maxvalue : maxvalue + scale_division
Chris@1294:         minvalue.step( maxvalue, scale_division ) {|v| rv << v}
Chris@1294:         return rv
Chris@1294:       end
Chris@1294: 
Chris@1294:       def x_label_offset( width )
Chris@1294:         width / 2.0
Chris@1294:       end
Chris@1294: 
Chris@1294:       def draw_data
Chris@1294:         minvalue = min_value
Chris@1294:         fieldwidth = field_width
Chris@1294: 
Chris@1294:         unit_size =  (@graph_height.to_f - font_size*2*top_font) / 
Chris@1294:                           (get_y_labels.max - get_y_labels.min)
Chris@1294:         bargap = bar_gap ? (fieldwidth < 10 ? fieldwidth / 2 : 10) : 0
Chris@1294: 
Chris@1294:         bar_width = fieldwidth - bargap
Chris@1294:         bar_width /= @data.length if stack == :side
Chris@1294:         x_mod = (@graph_width-bargap)/2 - (stack==:side ? bar_width/2 : 0)
Chris@1294:  
Chris@1294:         bottom = @graph_height
Chris@1294: 
Chris@1294:         field_count = 0
Chris@1294:         @config[:fields].each_index { |i|
Chris@1294:           dataset_count = 0
Chris@1294:           for dataset in @data
Chris@1294:           
Chris@1294:             # cases (assume 0 = +ve):
Chris@1294:             #   value  min  length
Chris@1294:             #    +ve   +ve  value - min
Chris@1294:             #    +ve   -ve  value - 0
Chris@1294:             #    -ve   -ve  value.abs - 0
Chris@1294:           
Chris@1294:             value = dataset[:data][i]
Chris@1294:             
Chris@1294:             left = (fieldwidth * field_count)
Chris@1294:             
Chris@1294:             length = (value.abs - (minvalue > 0 ? minvalue : 0)) * unit_size
Chris@1294:             # top is 0 if value is negative
Chris@1294:             top = bottom - (((value < 0 ? 0 : value) - minvalue) * unit_size)
Chris@1294:             left += bar_width * dataset_count if stack == :side
Chris@1294:  
Chris@1294:             @graph.add_element( "rect", {
Chris@1294:               "x" => left.to_s,
Chris@1294:               "y" => top.to_s,
Chris@1294:               "width" => bar_width.to_s,
Chris@1294:               "height" => length.to_s,
Chris@1294:               "class" => "fill#{dataset_count+1}"
Chris@1294:             })
Chris@1294: 
Chris@1294:             make_datapoint_text(left + bar_width/2.0, top - 6, value.to_s)
Chris@1294:             dataset_count += 1
Chris@1294:           end
Chris@1294:           field_count += 1
Chris@1294:         }
Chris@1294:       end
Chris@1294:     end
Chris@1294:   end
Chris@1294: end