SoundSoftware Code Site

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