Added support for Numo arrays to plots

Author: alishdipani

lib/rubyplot/artist/plot/area.rb

@@ -15,7 +15,7 @@ def stacked(bool)
         end
 
         def data x_values, y_values
-          x_values, y_values = x_values.zip(y_values).sort.transpose if @sort_data
+          x_values, y_values = x_values.to_a.zip(y_values.to_a).sort.transpose if @sort_data
           super(x_values, y_values)
         end
 

lib/rubyplot/artist/plot/bar.rb

@@ -34,7 +34,7 @@ def spacing_ratio=(s_r)
 
         # Set Bar plot data.
         def data y_values
-          super(Array.new(y_values.size) { |i| i }, y_values)
+          super(Array.new(y_values.to_a.size) { |i| i }, y_values.to_a)
         end
 
         # Number of bars in this Bar plot

lib/rubyplot/artist/plot/base.rb

@@ -39,8 +39,8 @@ def color=(color)
         end
 
         def data(x_values, y_values)
-          @data[:x_values] = x_values
-          @data[:y_values] = y_values
+          @data[:x_values] = x_values.to_a
+          @data[:y_values] = y_values.to_a
         end
 
         def process_data

lib/rubyplot/artist/plot/box_plot.rb

@@ -39,7 +39,13 @@ def process_data
         end
 
         def data vectors
-          @vectors = vectors
+          # @vectors = vectors.to_a unless vectors.is_a? Array
+          if vectors.is_a? Array
+            vectors.each_with_index { |_, idx| vectors[idx] = vectors[idx].to_a}
+            @vectors = vectors
+          else
+            @vectors = vectors.to_a
+          end
         end
 
         def draw

lib/rubyplot/artist/plot/bubble.rb

@@ -16,8 +16,8 @@ def initialize(*)
         end
 
         def data x_values, y_values, z_values
-          super(x_values, y_values)
-          @data[:z_values] = z_values
+          super(x_values.to_a, y_values.to_a)
+          @data[:z_values] = z_values.to_a
         end
 
         def draw

lib/rubyplot/artist/plot/candle_stick.rb

@@ -2,10 +2,10 @@ module Rubyplot
   module Artist
     module Plot
       class CandleStick < Artist::Plot::Base
-        attr_accessor :lows
-        attr_accessor :highs
-        attr_accessor :opens
-        attr_accessor :closes
+        attr_reader :lows
+        attr_reader :highs
+        attr_reader :opens
+        attr_reader :closes
         attr_accessor :bar_width
         attr_accessor :x_left_candle
         attr_accessor :x_low_stick
@@ -24,6 +24,22 @@ def initialize(*)
           @border_color = :black
         end
 
+        def lows=(lows_arr)
+          @lows = lows_arr.to_a
+        end
+
+        def highs=(highs_arr)
+          @highs = highs_arr.to_a
+        end
+
+        def opens=(opens_arr)
+          @opens = opens_arr.to_a
+        end
+
+        def closes=(closes_arr)
+          @closes = closes_arr.to_a
+        end
+
         def process_data
           if @lows.size != @highs.size || @opens.size != @closes.size
             raise Rubyplot::SizeError, "all given parameters must be of equal size."

lib/rubyplot/artist/plot/error_bar.rb

@@ -2,7 +2,8 @@ module Rubyplot
   module Artist
     module Plot
       class ErrorBar < Artist::Plot::Base
-        attr_accessor :xerr, :yerr, :xuplims, :xlolims, :yuplims, :ylolims, :line_width, :xerr_width, :yerr_width, :xerr_color, :yerr_color
+        attr_accessor :xuplims, :xlolims, :yuplims, :ylolims, :line_width, :xerr_width, :yerr_width, :xerr_color, :yerr_color
+        attr_reader :xerr, :yerr
 
         def initialize(*)
           super
@@ -13,6 +14,22 @@ def initialize(*)
           @yerr_color = nil
         end
 
+        def xerr=(xerror)
+          if (xerror.is_a?(Float) || xerror.is_a?(Integer))
+            @xerr = xerror
+          else
+            @xerr = xerror.to_a
+          end
+        end
+
+        def yerr=(yerror)
+          if (yerror.is_a?(Float) || yerror.is_a?(Integer))
+            @yerr = yerror
+          else
+            @yerr = yerror.to_a
+          end
+        end
+
         def process_data
           super
           preprocess_err_values!

lib/rubyplot/artist/plot/histogram.rb

@@ -3,7 +3,7 @@ module Artist
     module Plot
       class Histogram < Artist::Plot::Base
         # Values that need to be shown as a histogram.
-        attr_accessor :x
+        attr_reader :x
         # Array of bins into which the data should be subdivided.
         attr_accessor :bins
         # Width of each bar.
@@ -13,8 +13,12 @@ def initialize(*)
           super
         end
 
+        def x=(xvals)
+          @x = xvals.to_a
+        end
+
         def data x_values
-          @x = x_values
+          @x = x_values.to_a
         end
 
         def process_data

lib/rubyplot/artist/plot/line.rb

@@ -20,7 +20,7 @@ def initialize(*)
         end
 
         def data(x_values, y_values)
-          super x_values, y_values
+          super x_values.to_a, y_values.to_a
         end
 
         def draw

lib/rubyplot/artist/plot/scatter.rb

@@ -7,12 +7,12 @@ class Scatter < Artist::Plot::Base
         # Rubyplot::MARKER_TYPES.
         attr_accessor :marker_type
         # Set a color from Rubyplot::Color::COLOR_INDEX as the color of the
-        # fill of the marker.        
+        # fill of the marker.
         attr_accessor :marker_fill_color
         # Set a color from Rubyplot::Color::COLOR_INDEX as the color of the
         # border of the marker.
         attr_accessor :marker_border_color
-        
+
         def initialize(*)
           super
           @marker_size = 1.0

lib/rubyplot/artist/plot/stacked_bar.rb

@@ -21,7 +21,7 @@ def initialize(*)
         end
 
         def data y_values
-          super(Array(0...(y_values.size)), y_values)
+          super(Array(0...(y_values.to_a.size)), y_values.to_a)
         end
 
         def y_values