Contents

require 'query_report/row'
require 'query_report/chart/themes'

module QueryReport
  module Chart
    class CustomChart
      attr_reader :title, :type, :options, :data_table, :row, :rows

      def initialize(type, title, query, options={})
        @type = type
        @title = title
        @query = query
        @options = options
        @row = []
        @rows = []
        @data_table = GoogleVisualr::DataTable.new
      end

      def add_column(title)
        @data_table.new_column('string', title)
        @row << title.humanize
      end

      def add(column_title, &block)
        val = block.call(@query)
        @data_table.new_column(val.kind_of?(String) ? 'string' : 'number', column_title)
        @row << val
        @rows << Row.new(column_title, val)
      end

      def prepare
        @data_table.add_row(@row)
        opts = {:width => 500, :height => 240, :title => @title}.merge(options)
        chart_type = "#{type}_chart".classify
        chart_type = "GoogleVisualr::Interactive::#{chart_type}".constantize
        chart_type.new(@data_table, opts)
      end

      def prepare_gruff
        @gruff = Gruff::Bar.new(options[:width] || 600)
        @gruff.title = title
        @gruff.theme = Gruff::Themes::GOOGLE_CHART
        @rows.each do |row|
          @gruff.data(row.name.humanize, row.value)
        end
      end

      def to_blob
        prepare_gruff
        @gruff.to_blob
      end

    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
query_report-0.0.8	lib/query_report/chart/custom_chart.rb
query_report-0.0.7	lib/query_report/chart/custom_chart.rb
query_report-0.0.6	lib/query_report/chart/custom_chart.rb
query_report-0.0.5	lib/query_report/chart/custom_chart.rb
query_report-0.0.4	lib/query_report/chart/custom_chart.rb
query_report-0.0.3	lib/query_report/chart/custom_chart.rb