rubyplot 0.0.1 → 0.1.pre.a1

data/lib/rubyplot/figure.rb

@@ -0,0 +1,2 @@
+# Alias meant for easy class initialization without knowing Artist details.
+Rubyplot::Figure = Rubyplot::Artist::Figure

data/lib/rubyplot/spi.rb

@@ -0,0 +1,8 @@
+module Rubyplot
+  # Rubyplot simple plotting interface
+  class SPI
+    def initialize; end
+
+    def scatter!(*args, &block); end
+  end
+end

data/lib/rubyplot/subplot.rb

@@ -0,0 +1,4 @@
+module Rubyplot
+  class Subplot
+  end
+end

data/lib/rubyplot/themes.rb

@@ -0,0 +1,47 @@
+module Rubyplot
+  # A set of constants to define themes constant variables
+  # to be used to make beautiful looking plots.
+  module Themes
+    # A color scheme similar to the popular presentation software.
+    BASIC = {
+      marker_color: 'white', # The color of the marker used to make marker lines on plot.
+      font_color: 'white', # Font Color used to write on the plot.
+      background_colors: %I[black gun_powder], # The Background colors that form the gradient
+      label_colors: %I[lemon bondi_blue sun dark_orchid magic_mint olive sandy_beach navy grey]
+    }.freeze
+
+    TRACKS = {
+      marker_color: 'white',
+      font_color: 'white',
+      background_colors: %I[eastern_blue eastern_blue],
+      label_colors: %I[yellow green blue red maroon grey \
+bittersweet light_pink violet bright_turquoise spring_green green_yellow orange misty_rose\
+silver falu_red royal_blue crimson]
+
+    }.freeze
+
+    OREO = {
+      marker_color: 'white',
+      font_color: 'white',
+      background_colors: %I[eastern_blue eastern_blue],
+      label_colors: %I[crimson fruit_salad lemon bondi_blue sun dark_orchid teal mauve hot_toddy]
+    }.freeze
+
+    RITA = {
+      marker_color: 'black',
+      font_color: 'black',
+      background_colors: %I[pattens_blue white],
+      label_colors: %I[turquoise razzle_dazzle_rose pear your_pink teal mauve hot_toddy lemon_chiffon maroon]
+    }.freeze
+
+    # Plain White back ground with no gradient.
+    CLASSIC_WHITE = {
+      marker_color: 'black',
+      font_color: 'black',
+      background_colors: %I[white white],
+      label_colors: %I[strong_blue vivid_orange dark_lime_green strong_red slightly_desaturated_violet \
+dark_grey strong_yellow strong_cyan yellow maroon grey]
+
+    }.freeze
+  end
+end

data/lib/rubyplot/utils.rb

@@ -0,0 +1,14 @@
+module Rubyplot
+  module Utils
+    THOUSAND_SEPARATOR = ','.freeze
+    class << self
+      def format_label label
+        if label.is_a? Numeric
+          '%0.2f' % label
+        elsif label.is_a? String
+          label
+        end
+      end
+    end
+  end # module Utils
+end # module Rubyplot

data/lib/rubyplot/version.rb

@@ -1,3 +1,3 @@
 module Rubyplot
-  VERSION = "0.0.1"
+  VERSION = '0.1-a1'.freeze
 end

data/rubyplot.gemspec

@@ -21,4 +21,14 @@ Gem::Specification.new do |spec|
   spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
   spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
   spec.require_paths = ["lib"]
+  spec.extensions    = ['ext/grruby/extconf.rb']
+
+  spec.add_development_dependency 'bundler'
+  spec.add_development_dependency 'rake'
+  spec.add_development_dependency 'rake-compiler'
+  spec.add_development_dependency 'rspec'
+  spec.add_development_dependency 'pry'
+  spec.add_development_dependency 'rubocop'
+
+  spec.add_runtime_dependency 'rmagick',  '>= 2.13.4'
 end

data/spec/axes_spec.rb

@@ -0,0 +1,477 @@
+require 'spec_helper'
+
+describe "Rubyplot::Axes b: #{Rubyplot.backend}." do
+  before do
+    @planet_data = [
+      ["Moon", [25, 36, 86, 39]],
+      ["Sun", [80, 54, 67, 54]],
+      ["Earth", [22, 29, 35, 38]],
+      ["Mars", [95, 95, 95, 90, 85, 80, 88, 100]],
+      ["Venus", [90, 34, 23, 12, 78, 89, 98, 88]]
+    ]
+  end
+
+  context "#stacked_bar!" do
+    it "plots multiple stacked bar graphs with default colors" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      [
+        ["Charles", [20, 10, 5, 12, 11, 6, 10, 7]],
+        ["Adam", [5, 10, 20, 6, 9, 12, 14, 8]],
+        ["Daniel", [19, 9, 6, 11, 12, 7, 15, 8]]
+      ].each do |label, data|
+        axes.stacked_bar! do |p|
+          p.data data
+          p.label = label
+        end          
+      end
+      axes.title = "net earnings in different months."
+      axes.x_title = "X title"
+      axes.y_title = "Y title"
+      axes.x_ticks = ['Jan', 'Feb', 'March', 'April', 'May', 'June',  'July',
+                      'August', 'September', 'October', 'November', 'December']
+      axes.y_ticks = ['5', '10', '15', '20', '25', '30']
+    end
+
+    it "plots stacked bar in a small size" do
+      @figure = Rubyplot::Figure.new(height: 400, width: 400)
+      axes = @figure.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! do |p| 
+          p.data data
+          p.label = label
+        end
+      end
+      axes.title = "stacked bar."
+    end
+  end
+
+  context "#dot!" do
+    skip "plots a single dot plot" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.dot! do |p|
+        p.data [0,5,8,15]
+        p.label = "Car"
+        p.color = :maroon
+      end
+      axes.num_y_ticks = 4
+      axes.y_ticks = ['5/6', '5/15', '5/24', '5/30']
+    end
+
+    skip "plots multiple dot plots" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      [
+        [[0, 5, 8, 15], "cars", :maroon],
+        [[10, 3, 2, 8], "buses", :grey],
+        [[2, 15, 8, 11],"science", :yellow]
+      ].each do |data, label, color|
+        axes.dot! do |p|
+          p.data data
+          p.label = label
+          p.color = color
+          p.minimum_value = 0
+        end
+      end
+      axes.num_y_ticks = 4
+      axes.y_ticks = ['5/6', '5/15', '5/24', '5/30']
+    end
+  end
+
+  context "#bubble!" do
+    it "plots a single bubble plot" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.bubble! do |p|
+        p.data [-1, 19, -4, -23], [-35, 21, 23, -4], [45, 10, 21, 9]
+        p.label = "apples"
+        p.color = :blue
+      end
+      axes.title = "simple bubble plot."
+    end
+
+    it "plots multiple bubble plots on same axes." do 
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.bubble! do |p|
+        p.data [-1, 19, -4, -23], [-35, 21, 23, -4], [45, 10, 21, 9]
+        p.label = "apples"
+      end
+      axes.bubble! do |p|
+        p.data [20, 30, -6, -3], [-1, 5, -27, -3], [13, 10, 20, 10]
+        p.label = "peaches"
+      end
+      axes.title = "simple bubble plot."
+    end
+  end
+
+  context "#area!" do
+    it "plots a single simple Area graph" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.area! do |p|
+        p.data [25, 36, 86, 39, 25, 31, 79, 88]
+        p.label = "Jimmy"
+      end
+      axes.title = "Visual simple area graph test."
+      axes.num_x_ticks = 5
+      axes.x_ticks = ['0', '22', '44', '66', '88']
+    end
+
+    it "plots multiple area plots on the same Axes" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      [
+        ["Jimmy", [25, 36, 86, 39, 25, 31, 79, 88]],
+        ["Charles", [80, 54, 67, 54, 68, 70, 90, 95]],
+        ["Julie", [22, 29, 35, 38, 36, 40, 46, 57]],
+        ["Jane", [3, 95, 95, 90, 85, 80, 88, 100]]
+      ].each do |n, data|
+        axes.area! do |p|
+          p.data data
+          p.label = n
+        end
+      end
+      axes.title = "Multiple area plots on same axes."
+      axes.num_x_ticks = 4
+      axes.x_ticks = ['0', '2', '4', '6']
+    end
+  end
+  
+  context "#line!" do
+    it "makes a simple line plot" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.line! do |p|
+        p.data [2, 4, 7, 9], [1,2,3,4]
+        p.label = "Marco"
+        p.color = :blue
+      end
+      axes.title = "A line graph."
+    end
+
+    it "plots 2 simple lines on the same axes" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.line! do |p|
+        p.data [3, 5, 10, 15]
+        p.label = "Marco"
+        p.color = :blue
+      end
+      axes.line! do |p|
+        p.data [1, 9, 13, 28]
+        p.label = "John"
+        p.color = :green
+      end
+      axes.title = "A line graph."
+    end
+
+    it "tests very small plot" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.title = "very small line chart 200px"
+      @planet_data.each do |name, d|
+        axes.line! do |p|
+          p.data d
+          p.label = name
+        end
+      end
+    end
+
+    it "plots multiple 0 data" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.title = "hand value graph test"
+      axes.line! do |p|
+        p.data [0,0,100]
+        p.label = "test"
+      end
+    end
+
+    it "plots small values" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.title = "small values"
+      [
+        [[0.1, 0.14356, 0.0, 0.5674839, 0.456], "small"],
+        [[0.2, 0.3, 0.1, 0.05, 0.9], "small2"]
+      ].each do |d, label|
+        axes.line! do |p|
+          p.data d
+          p.label = label
+        end
+      end
+    end
+
+    it "plots line starting with 0" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.title = "starting with 0"
+      [
+        [[0, 5, 10, 8, 18], "first0"],
+        [[1, 2, 3, 4, 5], "normal"]
+      ].each do |data, name|
+        axes.line! do |p|
+          p.data data
+          p.label = name
+        end
+      end
+    end
+
+    it "plots line with large values" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.title = "large values"
+      [
+        ["large", [100_005, 35_000, 28_000, 27_000]],
+        ["large2", [35_000, 28_000, 27_000, 100_005]],
+        ["large3", [28_000, 27_000, 100_005, 35_000]],
+        ["large4", [1_238, 39_092, 27_938, 48_876]]
+      ].each do |name, data|
+        axes.line! do |p|
+          p.line_width = 3
+          p.data data
+          p.label = name
+        end
+      end
+    end
+
+    it "accepts both X and Y data" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.title = "accept X and Y"
+      axes.line! do |p|
+        p.data [1, 3, 4, 5, 6, 10], [1, 2, 3, 4, 4, 3]
+        p.label = "X"
+      end
+      axes.line! do |p|
+        p.data [1, 3, 4, 5, 7, 9], [1, 1, 2, 2, 3, 3]
+        p.label = "X1"
+      end
+    end
+  end
+
+  context "#bar!" do
+    it "adds a simple bar plot" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.bar! do |p| 
+        p.data [5,12,9,6,7]
+        p.label = "data"
+        p.color = :yellow
+      end
+      axes.x_ticks = ["five", "twelve", "nine", "six", "seven"]
+      axes.title = "Random bar numbers"
+    end
+
+    it "adds bar plot with title margin" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.bar! do |p|
+        p.data [5,12,9,6,6]
+        p.label = "data"
+        p.color = :green
+      end
+      axes.title = "Bar with title margin = 100"
+    end
+
+    it "plots large numbers" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.bar! do |p|
+        p.data [7025, 1024, 40_257, 933_672, 1_560_496]
+        p.label = "data"
+      end
+      axes.title = "Large numbers"
+    end
+
+    it "adds axes with X-Y labels" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.bar! do |p|
+        p.data [40,50,60,80]
+        p.label = "students"
+      end
+      axes.title = "Plot with X-Y axes."
+      axes.x_title = "Students"
+      axes.y_title = "Score (%)"
+      axes.x_ticks = [ '5/6', '5/15', '5/24', '5/36' ]
+    end
+
+    it "adds multiple bar plots for wide graph" do
+      @figure = Rubyplot::Figure.new(height: 400, width: 800)
+      axes = @figure.add_subplot 0,0
+      @planet_data.each do |name, nums|
+        axes.bar! do |p| 
+          p.data nums
+          p.label = name
+        end
+      end
+    end
+
+    it "plots both positive and negative values" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.bar! do |p|
+        p.data [-1, 0, 4, -4]
+        p.label = "apples"
+      end
+      axes.bar! do |p|
+        p.data [10,8,6,3]
+        p.label = "peaches"
+      end
+      axes.title = "Pos/neg bar graph test."
+      axes.x_ticks = ['5/6', '5/15', '5/24', '5/30']
+    end
+
+    it "tests negative values" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.title = "all negative bar graph."
+      axes.x_ticks = ['5/6', '5/15', '5/24', '5/30']
+      axes.bar! do |p|
+        p.data [-1,-5,-4,-4]
+        p.label = "apples"
+      end
+      axes.bar! do |p|
+        p.data [-10,-8,-6,-3]
+        p.label = "peaches"
+      end
+    end
+
+    it "sets min-max range for Y axis" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.title = "nearly zero graph."
+      axes.y_range = [0,10]
+      [
+        [[1,2,3,4], "apples"],
+        [[4,3,2,1], "peaches"]
+      ].each do |nums, name|
+        axes.bar! do |p|
+          p.data nums
+          p.label = name
+        end
+      end
+      axes.x_ticks = ['5/6', '5/15', '5/24','5/30']
+    end
+
+    it "adjust legends if there are too many" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.title = "My graph."
+      [
+        [[1, 2, 3, 4, 4, 3], 'Apples oranges Watermelon'],
+        [[4, 8, 7, 9, 8, 9], "Oranges"],
+        [[2, 3, 1, 5, 6, 8], "Watermelon"],
+        [[9, 9, 10, 8, 7, 9], "Peaches"]
+      ].each do |d, name|
+        axes.bar! do |p|
+          p.data d
+          p.label = name
+        end
+      end
+      axes.x_ticks = ['2003', '', '2004', '', '2005']
+    end
+  end
+
+  context "#scatter!" do
+    before do
+      @x1 = [1, 2, 3, 4, 5]
+      @y1 = [11, 2, 33, 4, 65]
+    end
+    
+    it "adds a simple scatter plot." do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.scatter! do |p|
+        p.data @x1, @y1
+        p.label = "data1"
+        p.color = :plum_purple
+      end
+      axes.title = "Nice plot"
+      axes.x_title = "X data"
+      axes.y_title = "Y data"
+    end
+
+    it "adds scatter with all negative values" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.scatter! do |p|
+        p.data [-1, -1, -4, -4], [-5, -1, -3, -4]
+        p.label = "apples"
+      end
+      axes.title = "all negative scatter graph test."
+    end
+
+    it "adds scatter with positive and negative values" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.scatter! do |p|
+        p.data [-2,0,2,4,6], [-3,-1, 1, 4, 8]
+        p.label = "values"
+      end
+      axes.title = "positive + negative test."
+    end
+  end
+
+  context "#top_margin=" do
+    it "sets the top margin in pixels" do
+      
+    end
+  end
+
+  context "#left_margin=" do
+    it "sets the left margin in pixels" do
+      
+    end
+  end
+
+  context "#bottom_margin=" do
+    it "sets the bottom margin in pixels" do
+      
+    end
+  end
+
+  context "#right_margin=" do
+    it "sets the right margin in pixels" do
+      
+    end
+  end
+
+  context "#num_x_ticks=" do
+    it "assigns number of X ticks" do
+      fig = Rubyplot::Figure.new
+      axes = fig.add_subplot 0,0
+      axes.scatter! do |p|
+        p.data [0,1,2,3], [4, 7, 11, 15]
+      end
+      axes.num_x_ticks = 3
+      axes.x_ticks = ["4", "10", "15"]
+      fig.write("dummy.png", output: false)
+
+      expect(
+        axes.instance_variable_get(:@inter_x_ticks_distance)).to eq(
+                                                                   axes.x_axis.length /
+                                                                   (axes.num_x_ticks-1))
+    end
+  end
+  
+  context "#x_ticks=" do
+    it "assigns strings to X ticks" do
+      @figure = Rubyplot::Figure.new
+      axes = @figure.add_subplot 0,0
+      axes.scatter! do |p|
+        p.data [1,2,3,4], [1,2,3,4]
+        p.label = "apples"
+      end
+      axes.x_ticks = ["hello0", "hello1"]
+    end
+  end # context "#x_ticks="
+end # Rubyplot::Axes
+