charty 0.1.1.dev

data/lib/charty.rb

@@ -0,0 +1,12 @@
+require_relative "charty/version"
+
+require_relative "charty/plotter"
+require_relative "charty/layout"
+require_relative "charty/linspace"
+require_relative "charty/table"
+
+module Charty
+  def self.new(*args)
+    Charty::Plotter.new(*args)
+  end
+end

data/lib/charty/gruff.rb

@@ -0,0 +1,75 @@
+require 'gruff'
+
+module Charty
+  class Gruff
+    def initialize
+      @plot = ::Gruff
+    end
+
+    def label(x, y)
+    end
+
+    def series=(series)
+      @series = series
+    end
+
+    def render_layout(layout)
+      raise NotImplementedError
+    end
+
+    def render(context, filename)
+      plot(@plot, context).write(filename)
+    end
+
+    def plot(plot, context)
+      # case
+      # when plot.respond_to?(:xlim)
+      #   plot.xlim(context.range_x.begin, context.range_x.end)
+      #   plot.ylim(context.range_y.begin, context.range_y.end)
+      # when plot.respond_to?(:set_xlim)
+      #   plot.set_xlim(context.range_x.begin, context.range_x.end)
+      #   plot.set_ylim(context.range_y.begin, context.range_y.end)
+      # end
+
+      case context.method
+      when :bar
+        p = plot::Bar.new
+        p.title = context.title if context.title
+        p.x_axis_label = context.xlabel if context.xlabel
+        p.y_axis_label = context.ylabel if context.ylabel
+        context.series.each do |data|
+          p.data(data.label, data.xs.to_a)
+        end
+        p
+      when :boxplot
+        # refs. https://github.com/topfunky/gruff/issues/155
+        raise NotImplementedError
+      when :bubble
+        raise NotImplementedError
+      when :curve
+        p = plot::Line.new
+        p.title = context.title if context.title
+        p.x_axis_label = context.xlabel if context.xlabel
+        p.y_axis_label = context.ylabel if context.ylabel
+        context.series.each do |data|
+          p.data(data.label, data.xs.to_a)
+        end
+        p
+      when :scatter
+        p = plot::Scatter.new
+        p.title = context.title if context.title
+        p.x_axis_label = context.xlabel if context.xlabel
+        p.y_axis_label = context.ylabel if context.ylabel
+        context.series.each do |data|
+          p.data(data.label, data.xs.to_a, data.ys.to_a)
+        end
+        p
+      when :errorbar
+        # refs. https://github.com/topfunky/gruff/issues/163
+        raise NotImplementedError
+      when :hist
+        raise NotImplementedError
+      end
+    end
+  end
+end

data/lib/charty/layout.rb

@@ -0,0 +1,75 @@
+module Charty
+  class Layout
+    def initialize(frontend, definition = :horizontal)
+      @frontend = frontend
+      @layout = parse_definition(definition)
+    end
+
+    def parse_definition(definition)
+      case definition
+      when :horizontal
+        ArrayLayout.new
+      when :vertical
+        ArrayLayout.new(:vertical)
+      else
+        if match = definition.to_s.match(/\Agrid(\d+)x(\d+)\z/)
+          num_cols = match[1].to_i
+          num_rows = match[2].to_i
+          GridLayout.new(num_cols, num_rows)
+        end
+      end
+    end
+
+    def <<(content)
+      if content.respond_to?(:each)
+        content.each {|c| self << c }
+      else
+        @layout << content
+      end
+      nil
+    end
+
+    def render(filename="")
+      @frontend.render_layout(@layout)
+    end
+  end
+
+  class ArrayLayout
+    def initialize(direction=:horizontal)
+      @array = []
+      @direction = direction
+    end
+
+    def <<(content)
+      @array << content
+    end
+
+    def num_rows
+      @direction == :horizontal ? 1 : @array.count
+    end
+
+    def num_cols
+      @direction == :vertical ? 1 : @array.count
+    end
+
+    def rows
+      [@array]
+    end
+  end
+
+  class GridLayout
+    attr_reader :num_rows, :num_cols, :rows
+
+    def initialize(num_cols, num_rows)
+      @rows = Array.new(num_rows) { Array.new(num_cols) }
+      @num_cols = num_cols
+      @num_rows = num_rows
+      @cursor = 0
+    end
+
+    def <<(content)
+      @rows[@cursor / @num_rows][@cursor % @num_cols] = content
+      @cursor += 1
+    end
+  end
+end

data/lib/charty/linspace.rb

@@ -0,0 +1,21 @@
+module Charty
+  class Linspace
+    include Enumerable
+
+    def initialize(range, num_step)
+      @range = range
+      @num_step = num_step
+    end
+
+    def each(&block)
+      step = (@range.end - @range.begin).to_f / @num_step
+      (@num_step - 1).times do |i|
+        block.call(@range.begin + i * step)
+      end
+
+      unless @range.exclude_end?
+        block.call(@range.end)
+      end
+    end
+  end
+end

data/lib/charty/matplot.rb

@@ -0,0 +1,91 @@
+require 'matplotlib/pyplot'
+
+module Charty
+  class Matplot
+    def initialize
+      @plot = Matplotlib::Pyplot
+    end
+
+    def self.activate_iruby_integration
+      require 'matplotlib/iruby'
+      Matplotlib::IRuby.activate
+    end
+
+    def label(x, y)
+    end
+
+    def series=(series)
+      @series = series
+    end
+
+    def render_layout(layout)
+      (fig, axes) = *@plot.subplots(nrows: layout.num_rows, ncols: layout.num_cols)
+      layout.rows.each_with_index do |row, y|
+        row.each_with_index do |cel, x|
+          plot = layout.num_rows > 1 ? axes[y][x] : axes[x]
+          plot(plot, cel, subplot: true)
+        end
+      end
+      @plot.show
+    end
+
+    def render(context, filename)
+      plot(@plot, context)
+      @plot.savefig(filename) if filename
+      @plot.show
+    end
+
+    def plot(plot, context, subplot: false)
+      # TODO: Since it is not required, research and change conditions.
+      # case
+      # when plot.respond_to?(:xlim)
+      #   plot.xlim(context.range_x.begin, context.range_x.end)
+      #   plot.ylim(context.range_y.begin, context.range_y.end)
+      # when plot.respond_to?(:set_xlim)
+      #   plot.set_xlim(context.range_x.begin, context.range_x.end)
+      #   plot.set_ylim(context.range_y.begin, context.range_y.end)
+      # end
+
+      plot.title(context.title) if context.title
+      if !subplot
+        plot.xlabel(context.xlabel) if context.xlabel
+        plot.ylabel(context.ylabel) if context.ylabel
+      end
+
+      case context.method
+      when :bar
+        context.series.each do |data|
+          plot.bar(data.xs.to_a, data.ys.to_a)
+        end
+      when :boxplot
+        plot.boxplot(context.data.to_a)
+      when :bubble
+        context.series.each do |data|
+          plot.scatter(data.xs.to_a, data.ys.to_a, s: data.zs.to_a, alpha: 0.5)
+        end
+      when :curve
+        context.series.each do |data|
+          plot.plot(data.xs.to_a, data.ys.to_a)
+        end
+      when :scatter
+        context.series.each do |data|
+          plot.scatter(data.xs.to_a, data.ys.to_a, label: data.label)
+        end
+        plot.legend()
+      when :errorbar
+        context.series.each do |data|
+          plot.errorbar(
+            data.xs.to_a,
+            data.ys.to_a,
+            data.xerr,
+            data.yerr,
+            label: data.label,
+          )
+        end
+        plot.legend()
+      when :hist
+        plot.hist(context.data.to_a)
+      end
+    end
+  end
+end

data/lib/charty/plotter.rb

@@ -0,0 +1,241 @@
+module Charty
+  class Plotter
+    def initialize(frontend)
+      @frontend =  case frontend
+      when :matplot
+        require_relative "matplot"
+        Charty::Matplot.new
+      when :gruff
+        require_relative "gruff"
+        Charty::Gruff.new
+      when :rubyplot
+        require_relative "rubyplot"
+        Charty::Rubyplot.new
+      else
+        raise NotImplementedError
+      end
+    end
+
+    def table=(data, **kwargs)
+      @table = Charty::Table.new(data)
+    end
+
+    attr_reader :table
+
+    def to_bar(x, y, **args, &block)
+      seriesx, seriesy = *table.to_a(x, y)
+      xrange = (seriesx.min)..(seriesx.max)
+      yrange = (seriesy.min)..(seriesy.max)
+      bar = bar do
+        series seriesx, seriesy
+        range x: xrange, y: yrange
+        xlabel x
+        ylabel y
+      end
+    end
+
+    def to_boxplot(x, y, **args, &block)
+      serieses = table.to_a(x, y)
+      xrange = 0..serieses.size
+      yrange = (serieses.flatten.min - 1)..(serieses.flatten.max + 1)
+      boxplot = boxplot do
+        data serieses
+        range x: xrange, y: yrange
+        xlabel x
+        ylabel y
+      end
+    end
+
+    def to_bubble(x, y, z, **args, &block)
+      seriesx, seriesy, seriesz = *table.to_a(x, y, z)
+      xrange = (seriesx.min)..(seriesx.max)
+      yrange = (seriesy.min)..(seriesy.max)
+      bubble = bubble do
+        series seriesx, seriesy, seriesz
+        range x: xrange, y: yrange
+        xlabel x
+        ylabel y
+      end
+    end
+
+    def to_curve(x, y, **args, &block)
+      seriesx, seriesy = *table.to_a(x, y)
+      xrange = (seriesx.min)..(seriesx.max)
+      yrange = (seriesy.min)..(seriesy.max)
+      curve = curve do
+        series seriesx, seriesy
+        range x: xrange, y: yrange
+        xlabel x
+        ylabel y
+      end
+    end
+
+    def to_scatter(x, y, **args, &block)
+      seriesx, seriesy = *table.to_a(x, y)
+      xrange = (seriesx.min)..(seriesx.max)
+      yrange = (seriesy.min)..(seriesy.max)
+      scatter = scatter do
+        series seriesx, seriesy
+        range x: xrange, y: yrange
+        xlabel x
+        ylabel y
+      end
+    end
+
+    def to_errorbar(x, y, **args, &block)
+      # TODO: It is not yet decided how to include data including xerror and yerror.
+      seriesx, seriesy = *table.to_a(x, y)
+      xrange = (seriesx.min)..(seriesx.max)
+      yrange = (seriesy.min)..(seriesy.max)
+      errorbar = errorbar do
+        series seriesx, seriesy
+        range x: xrange, y: yrange
+        xlabel x
+        ylabel y
+      end
+    end
+
+    def to_hst(x, y, **args, &block)
+      serieses = table.to_a(x, y)
+      xrange = (serieses.flatten.min - 1)..(serieses.flatten.max + 1)
+      yrange = 0..serieses[0].size
+      hist = hist do
+        data serieses
+        range x: xrange, y: yrange
+        xlabel x
+        ylabel y
+      end
+    end
+
+    def bar(**args, &block)
+      context = RenderContext.new :bar, **args, &block
+      context.apply(@frontend)
+    end
+
+    def boxplot(**args, &block)
+      context = RenderContext.new :boxplot, **args, &block
+      context.apply(@frontend)
+    end
+
+    def bubble(**args, &block)
+      context = RenderContext.new :bubble, **args, &block
+      context.apply(@frontend)
+    end
+
+    def curve(**args, &block)
+      context = RenderContext.new :curve, **args, &block
+      context.apply(@frontend)
+    end
+
+    def scatter(**args, &block)
+      context = RenderContext.new :scatter, **args, &block
+      context.apply(@frontend)
+    end
+
+    def errorbar(**args, &block)
+      context = RenderContext.new :errorbar, **args, &block
+      context.apply(@frontend)
+    end
+
+    def hist(**args, &block)
+      context = RenderContext.new :hist, **args, &block
+      context.apply(@frontend)
+    end
+
+    def layout(definition=:horizontal)
+      Layout.new(@frontend, definition)
+    end
+  end
+
+  Series = Struct.new(:xs, :ys, :zs, :xerr, :yerr, :label)
+
+  class RenderContext
+    attr_reader :function, :range, :series, :method, :data, :title, :xlabel, :ylabel
+
+    def initialize(method, **args, &block)
+      @method = method
+      configurator = Configurator.new(**args)
+      configurator.instance_eval &block
+      # TODO: label も外から付けられた方がよさそう
+      (@range, @series, @function, @data, @title, @xlabel, @ylabel) = configurator.to_a
+    end
+
+    class Configurator
+      def initialize(**args)
+        @args = args
+        @series = []
+      end
+
+      def function(&block)
+        @function = block
+      end
+
+      def data(data)
+        @data = data
+      end
+
+      def title(title)
+        @title = title
+      end
+
+      def xlabel(xlabel)
+        @xlabel = xlabel
+      end
+
+      def ylabel(ylabel)
+        @ylabel = ylabel
+      end
+
+      def label(x, y)
+
+      end
+
+      def series(xs, ys=nil, zs=nil, xerr: nil, yerr: nil, label: nil)
+        @series << Series.new(xs, ys, zs, xerr, yerr, label)
+      end
+
+      def range(range)
+        @range = range
+      end
+
+      def to_a
+        [@range, @series, @function, @data, @title, @xlabel, @ylabel]
+      end
+
+      def method_missing(method, *args)
+        if (@args.has_key?(method))
+          @args[name]
+        else
+          super
+        end
+      end
+    end
+
+    def range_x
+      @range[:x]
+    end
+
+    def range_y
+      @range[:y]
+    end
+
+    def render(filename=nil)
+      @frontend.render(self, filename)
+    end
+
+    def apply(frontend)
+      case
+        when !@series.empty?
+          frontend.series = @series
+        when @function
+          linspace = Linspace.new(@range[:x], 100)
+          # TODO: set label with function
+          # TODO: set ys to xs when gruff curve with function
+          @series << Series.new(linspace.to_a, linspace.map{|x| @function.call(x) }, label: "function" )
+      end
+
+      @frontend = frontend
+      self
+    end
+  end
+end