charty 0.1.5.dev → 0.2.0

data/lib/charty/backends/gruff.rb

@@ -1,98 +1,106 @@
-require 'gruff'
+require 'fileutils'
 
 module Charty
-  class Gruff < PlotterAdapter
-    Name = "gruff"
+  module Backends
+    class Gruff
+      Backends.register(:gruff, self)
 
-    def initialize
-      @plot = ::Gruff
-    end
+      class << self
+        def prepare
+          require 'gruff'
+        end
+      end
 
-    def label(x, y)
-    end
+      def initialize
+        @plot = ::Gruff
+      end
 
-    def series=(series)
-      @series = series
-    end
+      def label(x, y)
+      end
 
-    def render_layout(layout)
-      raise NotImplementedError
-    end
+      def series=(series)
+        @series = series
+      end
 
-    def render(context, filename="")
-      FileUtils.mkdir_p(File.dirname(filename))
-      plot(@plot, context).write(filename)
-    end
+      def render_layout(layout)
+        raise NotImplementedError
+      end
+
+      def render(context, filename="")
+        FileUtils.mkdir_p(File.dirname(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
+      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 :barh
-        p = plot::SideBar.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
-        labels = context.series.map {|data| data.xs.to_a}.flatten.uniq
-        labels.each do |label|
-          data_ys = context.series.map do |data|
-            if data.xs.to_a.index(label)
-              data.ys.to_a[data.xs.to_a.index(label)]
-            else
-              0
+        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 :barh
+          p = plot::SideBar.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
+          labels = context.series.map {|data| data.xs.to_a}.flatten.uniq
+          labels.each do |label|
+            data_ys = context.series.map do |data|
+              if data.xs.to_a.index(label)
+                data.ys.to_a[data.xs.to_a.index(label)]
+              else
+                0
+              end
             end
+            p.data(label, data_ys)
           end
-          p.data(label, data_ys)
-        end
-        p.labels = context.series.each_with_index.inject({}) do |attr, (data, i)|
-          attr[i] = data.label
-          attr
-        end
-        p
-      when :box_plot
-        # 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)
+          p.labels = context.series.each_with_index.inject({}) do |attr, (data, i)|
+            attr[i] = data.label
+            attr
+          end
+          p
+        when :box_plot
+          # 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 :error_bar
+          # refs. https://github.com/topfunky/gruff/issues/163
+          raise NotImplementedError
+        when :hist
+          raise NotImplementedError
         end
-        p
-      when :error_bar
-        # refs. https://github.com/topfunky/gruff/issues/163
-        raise NotImplementedError
-      when :hist
-        raise NotImplementedError
       end
     end
   end

data/lib/charty/backends/plotly.rb

@@ -0,0 +1,109 @@
+require 'json'
+
+module Charty
+  module Backends
+    class Plotly
+      Backends.register(:plotly, self)
+
+      attr_reader :context
+
+      class << self
+        attr_writer :chart_id, :with_api_load_tag, :plotly_src
+
+        def chart_id
+          @chart_id ||= 0
+        end
+
+        def with_api_load_tag
+          return @with_api_load_tag unless @with_api_load_tag.nil?
+
+          @with_api_load_tag = true
+        end
+
+        def plotly_src
+          @plotly_src ||= 'https://cdn.plot.ly/plotly-latest.min.js'
+        end
+      end
+
+      def initilize
+      end
+
+      def label(x, y)
+      end
+
+      def series=(series)
+        @series = series
+      end
+
+      def render(context, filename)
+        plot(nil, context)
+      end
+
+      def plot(plot, context)
+        context = context
+        self.class.chart_id += 1
+
+        case context.method
+        when :bar
+          render_graph(context, :bar)
+        when :curve
+          render_graph(context, :scatter)
+        when :scatter
+          render_graph(context, nil, options: {data: {mode: "markers"}})
+        else
+          raise NotImplementedError
+        end
+      end
+
+      private
+
+      def plotly_load_tag
+        if self.class.with_api_load_tag
+          "<script type='text/javascript' src='#{self.class.plotly_src}'></script>"
+        else
+        end
+      end
+
+      def div_id
+        "charty-plotly-#{self.class.chart_id}"
+      end
+
+      def div_style
+        "width: 100%;height: 100%;"
+      end
+
+      def render_graph(context, type, options: {})
+        data = context.series.map do |series|
+          {
+            type: type,
+            x: series.xs.to_a,
+            y: series.ys.to_a,
+            name: series.label
+          }.merge(options[:data] || {})
+        end
+        layout = {
+          title: { text: context.title },
+          xaxis: {
+            title: context.xlabel,
+            range: [context.range[:x].first, context.range[:x].last]
+          },
+          yaxis: {
+            title: context.ylabel,
+            range: [context.range[:y].first, context.range[:y].last]
+          }
+        }
+        render_html(data, layout)
+      end
+
+      def render_html(data, layout)
+        <<~FRAGMENT
+          #{plotly_load_tag unless self.class.chart_id > 1}
+          <div id="#{div_id}" style="#{div_style}"></div>
+          <script>
+            Plotly.plot('#{div_id}', #{JSON.dump(data)}, #{JSON.dump(layout)} );
+          </script>
+        FRAGMENT
+      end
+    end
+  end
+end

data/lib/charty/backends/pyplot.rb

@@ -1,109 +1,117 @@
-require 'matplotlib/pyplot'
+require 'fileutils'
 
 module Charty
-  class PyPlot < PlotterAdapter
-    Name = "pyplot"
+  module Backends
+    class Pyplot
+      Backends.register(:pyplot, self)
 
-    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)
+      class << self
+        def prepare
+          require 'matplotlib/pyplot'
         end
       end
-      @plot.show
-    end
 
-    def render(context, filename)
-      plot(@plot, context)
-      if filename
-        FileUtils.mkdir_p(File.dirname(filename))
-        @plot.savefig(filename)
+      def initialize
+        @plot = ::Matplotlib::Pyplot
       end
-      @plot.show
-    end
 
-    def save(context, filename)
-      plot(@plot, context)
-      if filename
-        FileUtils.mkdir_p(File.dirname(filename))
-        @plot.savefig(filename)
+      def self.activate_iruby_integration
+        require 'matplotlib/iruby'
+        ::Matplotlib::IRuby.activate
       end
-    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
+      def label(x, y)
+      end
 
-      plot.title(context.title) if context.title
-      if !subplot
-        plot.xlabel(context.xlabel) if context.xlabel
-        plot.ylabel(context.ylabel) if context.ylabel
+      def series=(series)
+        @series = series
       end
 
-      case context.method
-      when :bar
-        context.series.each do |data|
-          plot.bar(data.xs.to_a.map(&:to_s), data.ys.to_a, label: data.label)
-        end
-        plot.legend()
-      when :barh
-        context.series.each do |data|
-          plot.barh(data.xs.to_a.map(&:to_s), data.ys.to_a)
+      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
-      when :box_plot
-        plot.boxplot(context.data.to_a, labels: context.labels)
-      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, label: data.label)
+        @plot.show
+      end
+
+      def render(context, filename)
+        plot(@plot, context)
+        if filename
+          FileUtils.mkdir_p(File.dirname(filename))
+          @plot.savefig(filename)
         end
-        plot.legend()
-      when :curve
-        context.series.each do |data|
-          plot.plot(data.xs.to_a, data.ys.to_a)
+        @plot.show
+      end
+
+      def save(context, filename)
+        plot(@plot, context)
+        if filename
+          FileUtils.mkdir_p(File.dirname(filename))
+          @plot.savefig(filename)
         end
-      when :scatter
-        context.series.each do |data|
-          plot.scatter(data.xs.to_a, data.ys.to_a, label: data.label)
+      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
-        plot.legend()
-      when :error_bar
-        context.series.each do |data|
-          plot.errorbar(
-            data.xs.to_a,
-            data.ys.to_a,
-            data.xerr,
-            data.yerr,
-            label: data.label,
-          )
+
+        case context.method
+        when :bar
+          context.series.each do |data|
+            plot.bar(data.xs.to_a.map(&:to_s), data.ys.to_a, label: data.label)
+          end
+          plot.legend()
+        when :barh
+          context.series.each do |data|
+            plot.barh(data.xs.to_a.map(&:to_s), data.ys.to_a)
+          end
+        when :box_plot
+          plot.boxplot(context.data.to_a, labels: context.labels)
+        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, label: data.label)
+          end
+          plot.legend()
+        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 :error_bar
+          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
-        plot.legend()
-      when :hist
-        plot.hist(context.data.to_a)
       end
     end
   end