Multiple charts in a row

Author: Prakriti-nith

lib/daru/view/adapters/googlecharts.rb

@@ -15,20 +15,25 @@ module GooglechartsAdapter
         # and google_visualr : http://googlevisualr.herokuapp.com/
         #
         # TODO : this docs must be improved
+        # @example Multiple Charts in a row
+        #   TODO: Do after spreadsheet PR.
         def init(data=[], options={})
-          @table = GoogleVisualr::DataTable.new
-          @table =
-            if data.is_a?(Daru::View::Table) && data.table.is_a?(GoogleVisualr::DataTable)
-              data.table
-            elsif data.is_a?(GoogleVisualr::DataTable)
-              data
-            else
-              add_data_in_table(data)
-            end
-          @chart_type = extract_chart_type(options)
-          @chart = GoogleVisualr::Interactive.const_get(
-            @chart_type
-          ).new(@table, options)
+          # When multiple charts are shown in a row, @chart will contain the
+          #   instance of GoogleVisular::BaseChart so that its data can contain
+          #   the array of plots (this will be used in display.rb). Further,
+          #   @chart will be used to call show_in_iruby and to_html.
+          if data.is_a?(Array) &&
+             (data[0].is_a?(Daru::View::Plot) || data[0].is_a?(Daru::View::Table))
+            @chart = GoogleVisualr::BaseChart.new(GoogleVisualr::DataTable.new)
+          else
+            @table = GoogleVisualr::DataTable.new
+            @table = get_table(data)
+            @chart_type = extract_chart_type(options)
+            @chart = GoogleVisualr::Interactive.const_get(
+              @chart_type
+            ).new(@table, options)
+          end
+          @chart.data = data
           @chart
         end
 
@@ -47,10 +52,21 @@ def init_table(data=[], options={})
           # }
           # then directly DatTable is created using options. Use data=[] or nil
           @table = GoogleVisualr::DataTable.new(options)
+          @table.data = data
           add_data_in_table(data)
           @table
         end
 
+        def get_table(data)
+          if data.is_a?(Daru::View::Table) && data.table.is_a?(GoogleVisualr::DataTable)
+            data.table
+          elsif data.is_a?(GoogleVisualr::DataTable)
+            data
+          else
+            add_data_in_table(data)
+          end
+        end
+
         def init_script
           GoogleVisualr.init_script
         end

lib/daru/view/adapters/googlecharts/display.rb

@@ -36,24 +36,59 @@ def show_script(dom=SecureRandom.uuid, options={})
     end
 
     def to_html(id=nil, options={})
-      path = File.expand_path('../../templates/googlecharts/chart_div.erb', __dir__)
+      if data.is_a?(Array) &&
+         (data[0].is_a?(Daru::View::Plot) || data[0].is_a?(Daru::View::Table))
+        id, chart_script, path = extract_multiple_charts_id_script_path
+      else
+        id, chart_script, path = extract_chart_id_script_path(id)
+      end
       template = File.read(path)
-      id ||= SecureRandom.uuid
-      @html_id = id
-      chart_script = show_script(id, script_tag: false)
       ERB.new(template).result(binding)
     end
 
     def show_in_iruby(dom=SecureRandom.uuid)
       IRuby.html(to_html(dom))
     end
+
+    def extract_multiple_charts_id_script_path
+      path = File.expand_path(
+        '../../templates/googlecharts/multiple_charts_div.erb', __dir__
+      )
+      id = []
+      chart_script = []
+      data.each_with_index do |plot, index|
+        id[index] ||= SecureRandom.uuid
+        chart_script[index] = set_chart_script(plot, id[index])
+      end
+      [id, chart_script, path]
+    end
+
+    def extract_chart_id_script_path(id=nil)
+      path = File.expand_path(
+        '../../templates/googlecharts/chart_div.erb', __dir__
+      )
+      id ||= SecureRandom.uuid
+      @html_id = id
+      chart_script = show_script(id, script_tag: false)
+      [id, chart_script, path]
+    end
+
+    def set_chart_script(plot, id)
+      if plot.is_a?(Daru::View::Plot)
+        plot.chart.show_script(id, script_tag: false)
+      else
+        plot.table.show_script(id, script_tag: false)
+      end
+    end
   end
 
   class DataTable
+    attr_accessor :data
     include Display
   end
 
   class BaseChart
+    attr_accessor :data
     include Display
   end
 end

lib/daru/view/adapters/nyaplot.rb

@@ -1,4 +1,3 @@
-
 require_relative 'nyaplot/iruby_notebook'
 require 'daru'
 require 'nyaplot'

lib/daru/view/plot.rb

@@ -1,4 +1,3 @@
-
 module Daru
   module View
     class Plot

lib/daru/view/table.rb

@@ -1,4 +1,3 @@
-
 module Daru
   module View
     class Table

lib/daru/view/templates/googlecharts/multiple_charts_div.erb

@@ -0,0 +1,12 @@
+<table class='columns'>
+  <tr>
+    <% id.each do |div_id| %>  
+      <td><div id='<%= div_id %>'></div></td>
+    <% end %>
+  </tr>
+</table>
+<script>
+  <% chart_script.each do |ch_script| %>
+    <%= ch_script %>
+  <% end %>
+</script>