WIP stacked bar

Author: v0dro

lib/rubyplot/artist/axes.rb

@@ -331,6 +331,13 @@ def consolidate_plots
           @plots.delete_if { |p| p.is_a?(Rubyplot::Artist::Plot::Bar) }
           @plots << Rubyplot::Artist::Plot::MultiBars.new(self, bar_plots: bars)
         end
+
+        stacked_bars = @plots.grep(Rubyplot::Artist::Plot::StackedBar)
+        if !stacked_bars.empty?
+          @plots.delete_if { |p| p.is_a?(Rubyplot::Artist::Plot::StackedBar) }
+          @plots << Rubyplot::Artist::Plot::MultiStackedBar.new(
+            self, stacked_bars: stacked_bars)
+        end
       end
 
       # FIXME: replace x_range and y_range with XAxis::max/min_value and YAxis::max/min_value.

lib/rubyplot/artist/plot/bar_type.rb

@@ -0,0 +1,33 @@
+module Rubyplot
+  module Artist
+    module Plot
+      class BarType < Artist::Plot::Base
+
+        def initialize(*)
+          super
+          @spacing_ratio = 0.1
+          @abs_x_left = []
+          @abs_y_left = []
+          @rectangles = []
+        end
+
+        def data y_values
+          super(Array(0...(y_values.size)), y_values)
+        end
+        
+        def num_bars
+          @data[:y_values].size
+        end
+
+        def draw
+          setup_bar_rectangles
+          @rectangles.each(&:draw)
+        end
+
+        protected
+
+        
+      end # class BarType
+    end # module Plot
+  end # module Artist
+end # module Rubyplot

lib/rubyplot/artist/plot/multi_stacked_bar.rb

@@ -0,0 +1,49 @@
+module Rubyplot
+  module Artist
+    module Plot
+      class MultiStackedBar < Artist::Plot::Base
+        def initialize(*, stacked_bars:)
+          super
+          @stacked_bars = stacked_bars
+          @x_min = @stacked_bars.map(&:x_min).min
+          @y_min = @stacked_bars.map(&:y_min).min
+          @x_max = @stacked_bars.map(&:x_max).max
+          @y_max = @stacked_bars.map(&:y_max).max
+          configure_plot_geometry_data
+          configure_x_ticks
+        end
+
+        def draw
+          
+        end
+
+        private
+
+        def configure_plot_geometry_data
+          @num_max_slots = @stacked_bars.map { |bar| bar.num_bars }.max
+          @max_slot_width = (@axes.x_axis.abs_x2 - @axes.x_axis.abs_x1).abs / @num_max_slots
+          @spacing_ratio = @stacked_bars[0].spacing_ratio
+          @padding = @spacing_ratio * @max_slot_width
+          @max_bars_width = @max_slot_width - @padding
+          @num_max_stacks = @stacked_bars.size
+          @stacked_bars.each_with_index do |bar, index|
+            set_bar_dims bar, index
+          end
+        end
+
+        def configure_x_ticks
+          
+        end
+
+        def set_bar_dims bar, plot_index
+          bar.bar_width = @max_bars_width
+          bar.num_bars.times do |i|
+            bar.abs_x_left[i] = @axes.abs_x + @axes.y_axis_margin +
+                                i * @max_bars_width  + @padding / 2
+            bar.abs_y_left[i] = 
+          end
+        end
+      end # class StackedBar
+    end # module Plot
+  end # module Artist
+end # module Rubyplot

lib/rubyplot/artist/plot/stacked_bar.rb

@@ -0,0 +1,41 @@
+module Rubyplot
+  module Artist
+    module Plot
+      class StackedBar < Artist::Plot::Base
+        # Ratio of the total avialable width that is left as padding space.
+        attr_accessor :spacing_ratio
+        # Array of X co-ordinates of the lower left corner of each stacked bar.
+        attr_accessor :abs_x_left
+        # Array of Y co-ordinates of the lower left corner of each stacked bar.
+        attr_accessor :abs_y_left
+        # Width in pixels of each bar.
+        attr_accessor :bar_width        
+        
+        def initialize(*)
+          super
+          @spacing_ratio = 0.1
+          @abs_x_left = []
+          @abs_y_left = []
+        end
+
+        def data y_values
+          super(Array(0...(y_values.size)), y_values)
+        end
+
+        def num_bars
+          @data[:y_values].size
+        end
+
+        def draw
+          setup_bar_rectangles
+          @rectangles.each(&:draw)
+        end
+
+        private
+
+        def setup_bar_rectangles
+        end
+      end # class StackedBar
+    end # module Plot
+  end # module Artist
+end # module Rubyplot

spec/axes_spec.rb

@@ -20,36 +20,7 @@
         FileUtils.mkdir_p @temp_dir
       end
 
-      it "plots a single stacked bar graph with default colors" do
-        fig = Rubyplot::Figure.new
-        axes = fig.add_subplot 0,0
-        axes.stacked_bar! do |p|
-          p.data [25, 36, 86, 39]
-          p.label = "moon"
-        end
-        axes.title = "net earnings in different months."
-        axes.x_ticks = {
-          0 => 'Jan',
-          1 => 'Feb',
-          2 => 'March',
-          3 => 'April',
-          4 => 'May',
-          5 => 'June',
-          6 => 'July',
-          7 => 'August',
-          8 => 'September',
-          9 => 'October',
-          10 => 'November',
-          11 => 'December'
-        }
-        
-        file = "/#{Rubyplot.backend}_simple_stacked_bar.png"
-        fig.write(@temp_dir + file)
-
-        #expect("temp/stacked_bar" + file).to eq_image("fixtures/stacked_bar" + file)
-      end
-
-      it "plots multiple stacked bar graphs with default colors" do
+      it "plots multiple stacked bar graphs with default colors", hell: true do
         fig = Rubyplot::Figure.new
         axes = fig.add_subplot 0,0
         [
@@ -63,20 +34,8 @@
           end          
         end
         axes.title = "net earnings in different months."
-        axes.x_ticks = {
-          0 => 'Jan',
-          1 => 'Feb',
-          2 => 'March',
-          3 => 'April',
-          4 => 'May',
-          5 => 'June',
-          6 => 'July',
-          7 => 'August',
-          8 => 'September',
-          9 => 'October',
-          10 => 'November',
-          11 => 'December'
-        }
+        axes.x_ticks = ['Jan', 'Feb', 'March', 'April', 'May', 'June',  'July',
+                        'August', 'September', 'October', 'November', 'December']
 
         file = "/#{Rubyplot.backend}_multiple_stacked_bar.png"
         fig.write(@temp_dir + file)
@@ -85,14 +44,14 @@
       end
 
       it "plots stacked bar in a small size" do
-        fig = Rubyplot::Figure.new
+        fig = Rubyplot::Figure.new(height: 400, width: 400)
         axes = fig.add_subplot 0,0
         [
           ["Car", [25, 36, 86, 39]],
           ["Bus", [80, 54, 67, 54]],
           ["Train", [22, 29, 35, 38]]
         ].each do |label, data|
-          axes.stacked_bar!(400) do |p| 
+          axes.stacked_bar! do |p| 
             p.data data
             p.label = label
           end
@@ -113,29 +72,24 @@
         FileUtils.mkdir_p @temp_dir
       end
 
-      it "plots a single dot plot" do
+      skip "plots a single dot plot" do
         fig = Rubyplot::Figure.new
         axes = fig.add_subplot 0,0
         axes.dot! do |p|
           p.data [0,5,8,15]
           p.label = "Car"
           p.color = :maroon
-          p.minimum_value = 0 # FIXME: change this!
         end
-        axes.y_ticks = {
-          0 => '5/6',
-          1 => '5/15',
-          2 => '5/24',
-          3 => '5/30'
-        }
+        axes.num_y_ticks = 4
+        axes.y_ticks = ['5/6', '5/15', '5/24', '5/30']
 
         file = "/#{Rubyplot.backend}_simple_dot.png"
         fig.write(@temp_dir + file)
 
         #expect("temp/dot" + file).to eq_image("fixtures/dot" + file)
       end
 
-      it "plots multiple dot plots" do
+      skip "plots multiple dot plots" do
         fig = Rubyplot::Figure.new
         axes = fig.add_subplot 0,0
         [