jamesmoriarty/scorched-earth-rb

include Java

import java.awt.Color

require 'scorched_earth/services/cie94'

module ScorchedEarth
  module Services
    class ColorPalette
      include Enumerable

      attr_reader :cache, :colors, :strategy

      def initialize(*colors)
        @cache    = {}
        @colors   = colors
        @strategy = Strategies::TriadMixing
      end

      def get(key)
        cache[key] ||= begin
          first
        end
      end

      def each
        return enum_for(:each) unless block_given?

        loop do
          color = strategy.color colors
          yield color unless already_exists?(color)
        end
      end

      def already_exists?(color)
        cache.values.any? do |color2|
          delta_e94 = Services::CIE94.new.call color, color2

          delta_e94 < 750 # MAX: 10_0000 e.g. black vs. white
        end
      end

      module Strategies
        module TriadMixing
          def self.color(colors)
            sum = 0.0

            colors.map do |color|
              sum += ratio = rand

              [color, ratio]
            end.inject(Color.new(0, 0, 0)) do |mix, (color, ratio)|
              scale = ratio / sum

              Color.new(
                (mix.red   + color.red   * scale).to_i,
                (mix.green + color.green * scale).to_i,
                (mix.blue  + color.blue  * scale).to_i
              )
            end
          end
        end
      end
    end
  end
end