require 'gruff'
require 'ruby-graphviz'

module Reconn
  class Visualizer
    def self.make_pie_chart(title, data, items_number)
      chart = Gruff::Pie.new
      chart.title = title.to_s
      data = Array.new(data)
      data.first(items_number).each do |item|
        chart.data(item[:label], item[:value])
        data.delete(item)
      end

      chart.data("other", data.map {|itm| itm[:value]}.inject(:+))

      chart.to_blob
    end

    def self.make_bar_chart(title, data, additional_line_value = nil)
      chart = Gruff::Bar.new
      chart.minimum_value = 0
      chart.maximum_value = data.first[:value]
      chart.additional_line_values = [additional_line_value] unless additional_line_value.nil?
      chart.title = title.to_s
      data.each do |item|
        chart.data(item[:label], item[:value])
      end

      chart.to_blob
    end

    def self.make_dependency_diagram(classes)
      diagram = GraphViz.new(:G, :type => :digraph)
      nodes = classes.map { |c| diagram.add_nodes(c.name) }
      external_nodes = classes.map {|c| c.external_deps}.inject(:+).uniq.map {|n| diagram.add_nodes(n)}
      classes.each do |klass|
        classes.each do |other_klass|
          if !klass.dependencies.index {|d| d == other_klass.name }.nil? || klass.dependencies.find_all {|d| other_klass.name =~ /^.*{0,}::#{d}$/}.size == 1
            node, other_node = [klass, other_klass].map {|k| nodes.find {|n| n[:label].to_s.gsub('"', '') == k.name}}
            diagram.add_edges(node, other_node)
          end
        end
        external_nodes.each {|n| n.set {|_n| _n.color = "blue"}}
        external_nodes.each do |ext_node|
          if klass.external_deps.include?(ext_node[:label].to_s.gsub('"', ''))
            node = nodes.find {|n| n[:label].to_s.gsub('"', '') == klass.name}
            diagram.add_edges(node, ext_node)
          end
        end
      end
      diagram.output(:png => String)
    end
  end
end

#Monkeypatching additional line drawing
module Gruff
  class Bar
    def draw
      @center_labels_over_point = (@labels.keys.length > @column_count ? true : false)
      super
      return unless @has_data

      draw_bars
      draw_additional_line
    end
    def draw_bars
      # Setup spacing.
      #
      # Columns sit side-by-side.
     @bar_spacing ||= @spacing_factor # space between the bars
      @bar_width = @graph_width / (@column_count * @data.length).to_f
      padding = (@bar_width * (1 - @bar_spacing)) / 2

      @d = @d.stroke_opacity 0.0

      # Setup the BarConversion Object
      conversion = Gruff::BarConversion.new()
      conversion.graph_height = @graph_height
      conversion.graph_top = @graph_top

      # Set up the right mode [1,2,3] see BarConversion for further explanation
      if @minimum_value >= 0 then
        # all bars go from zero to positiv
        conversion.mode = 1
      else
        # all bars go from 0 to negativ
        if @maximum_value <= 0 then
          conversion.mode = 2
        else
          # bars either go from zero to negativ or to positiv
          conversion.mode = 3
          conversion.spread = @spread
          conversion.minimum_value = @minimum_value
          conversion.zero = -@minimum_value/@spread
        end
      end

      # iterate over all normalised data
      @norm_data.each_with_index do |data_row, row_index|

        data_row[DATA_VALUES_INDEX].each_with_index do |data_point, point_index|
          # Use incremented x and scaled y
          # x
          left_x = @graph_left + (@bar_width * (row_index + point_index + ((@data.length - 1) * point_index))) + padding
          right_x = left_x + @bar_width * @bar_spacing
          # y
          conv = []
          conversion.get_left_y_right_y_scaled( data_point, conv )

          # create new bar
          @d = @d.fill data_row[DATA_COLOR_INDEX]
          @d = @d.rectangle(left_x, conv[0], right_x, conv[1])

          # Calculate center based on bar_width and current row
          label_center = @graph_left + 
                      (@data.length * @bar_width * point_index) + 
                      (@data.length * @bar_width / 2.0)

          # Subtract half a bar width to center left if requested
          draw_label(label_center - (@center_labels_over_point ? @bar_width / 2.0 : 0.0), point_index)
          if @show_labels_for_bar_values
            val = (@label_formatting || '%.2f') % @norm_data[row_index][3][point_index]
            draw_value_label(left_x + (right_x - left_x)/2, conv[0]-30, val.commify, true)
          end
        end
      end

      # Draw the last label if requested
      draw_label(@graph_right, @column_count) if @center_labels_over_point

      draw_additional_line
      @d.draw(@base_image)
    end
    # Draws additional horizontal line
    def draw_additional_line
       @additional_line_colors << '#f61100'
       @d = @d.stroke_opacity 100.0
       i = 0
       @additional_line_values.each do |value|
         @increment_scaled = @graph_height.to_f / (@maximum_value.to_f / value)

         y = @graph_top + @graph_height - @increment_scaled

         @d = @d.stroke(@additional_line_colors[i])
         @d = @d.line(@graph_left, y, @graph_right, y)


         @d.fill = @additional_line_colors[i]
         @d.font = @font if @font
         @d.stroke('transparent')
         @d.pointsize = scale_fontsize(@marker_font_size)
         @d.gravity = EastGravity
         @d = @d.annotate_scaled( @base_image,
                                 @graph_right - LABEL_MARGIN, 1.0,
                                 0.0, y - (@marker_font_size/2.0),
                                 label(value, value), @scale)
         i += 1
       end

      @d = @d.stroke_antialias true
    end
  end
end