vibrant 0.2.0

data/lib/vibrant/vibrant.rb

@@ -0,0 +1,177 @@
+# encoding: utf-8
+module Vibrant
+
+  class Vibrant
+
+    TARGET_DARK_LUNA = 0.26
+    MAX_DARK_LUNA = 0.45
+    MIN_LIGHT_LUNA = 0.55
+    TARGET_LIGHT_LUNA = 0.74
+
+    MIN_NORMAL_LUNA = 0.3
+    TARGET_NORMAL_LUNA = 0.5
+    MAX_NORMAL_LUNA = 0.7
+
+    TARGET_MUTED_SATURATION = 0.3
+    MAX_MUTED_SATURATION = 0.4
+
+    TARGET_VIBRANT_SATURATION = 1
+    MIN_VIBRANT_SATURATION = 0.35
+
+    NORMAL_LUNA = {
+        target: TARGET_NORMAL_LUNA,
+        range: MIN_NORMAL_LUNA..MAX_NORMAL_LUNA
+    }
+    LIGHT_LUNA = {
+        target: TARGET_LIGHT_LUNA,
+        range: MIN_LIGHT_LUNA..1
+    }
+    DARK_LUNA = {
+        target: TARGET_DARK_LUNA,
+        range: 0..MAX_DARK_LUNA
+    }
+
+    VIBRANT_SATURATION = {
+        target: TARGET_VIBRANT_SATURATION,
+        range: MIN_VIBRANT_SATURATION..1
+    }
+    MUTED_SATURATION = {
+        target: TARGET_MUTED_SATURATION,
+        range: 0..MAX_MUTED_SATURATION
+    }
+
+    WEIGHT_SATURATION = 3
+    WEIGHT_LUNA = 6
+    WEIGHT_POPULATION = 1
+
+    @@highestPopulation = 0
+
+    # def initialize(sourceImage, color_count: 64, quality: 5)
+    #   @img = Magick::Image.from_blob(sourceImage.read).file.quantize(@color_count)
+    #   @process
+    # end
+
+    def initialize(sourceImage, color_count: 64, quality: 5)
+      @color_count = color_count
+      @quality = quality
+      @image = Magick::Image.read(sourceImage).first.quantize(@color_count)
+      process
+    end
+
+    def process
+
+      @_swatches = []
+
+      # histogram
+      histogram = @image.color_histogram.inject({}) { |h, pair|
+        hex = pair[0].to_color(Magick::AllCompliance, false, 8, true)
+        hsla = pair[0].to_hsla
+        count = pair[1]
+
+        h[hex] ||= {
+            hex: hex,
+            hsla: hsla,
+            count: 0
+        }
+        h[hex][:count] += count
+        h
+      }.each_pair { |key, val|
+        @_swatches.push(Swatch.new(val[:hex], val[:hsla], val[:count]))
+      }
+
+      @maxPopulation = @_swatches.collect(&:population).max
+
+      @vibrantSwatch = find_color_variation(NORMAL_LUNA, VIBRANT_SATURATION)
+      @lightVibrantSwatch = find_color_variation(LIGHT_LUNA, VIBRANT_SATURATION)
+      @darkVibrantSwatch = find_color_variation(DARK_LUNA, VIBRANT_SATURATION)
+      @mutedSwatch = find_color_variation(NORMAL_LUNA, MUTED_SATURATION)
+      @lightMutedSwatch = find_color_variation(LIGHT_LUNA, MUTED_SATURATION)
+      @darkMutedSwatch = find_color_variation(DARK_LUNA, MUTED_SATURATION)
+
+      if @vibrantSwatch.nil? && !@darkVibrantSwatch.nil?
+        hsla = @DarkVibrantSwatch.hsla.dup
+        hsla[2] = TARGET_NORMAL_LUNA
+        @vibrantSwatch = Swatch.new(@DarkVibrantSwatch.hex, hsla, 0)
+      end
+
+      if @darkVibrantSwatch.nil? && !@vibrantSwatch.nil?
+        hsla = @vibrantSwatch.hsla.dup
+        hsla[2] = TARGET_DARK_LUNA
+        @darkVibrantSwatch = Swatch.new(@vibrantSwatch.hex, hsla, 0)
+      end
+
+    end
+
+    def find_color_variation(luna, saturation)
+      max = nil
+      maxValue = 0
+
+      @_swatches.each do |swatch|
+        #h = swatch.hsla[0] / 360
+        s = swatch.hsla[1] / 255
+        l = swatch.hsla[2] / 255
+
+        if luna[:range].include?(l) && saturation[:range].include?(s) && !already_selected?(swatch)
+
+          value = create_comparison_value(s, saturation[:target], l, luna[:target], swatch.population, @maxPopulation)
+          if max.nil? || value > maxValue
+            max = swatch
+            maxValue = value
+          end
+        end
+      end
+      p max
+
+      max
+    end
+
+    def create_comparison_value(saturation, targetSaturation, luna, target_luna, population, max_population)
+      self.class.weightedMean([
+                                  [self.class.invert_diff(saturation, targetSaturation), WEIGHT_SATURATION],
+                                  [self.class.invert_diff(luna, target_luna), WEIGHT_LUNA],
+                                  [(population / max_population), WEIGHT_POPULATION]
+                              ])
+    end
+
+
+    def swatches
+      Swatches.new(
+          vibrant: @vibrantSwatch,
+          muted: @mutedSwatch,
+          dark_vibrant: @darkVibrantSwatch,
+          dark_muted: @darkMutedSwatch,
+          light_vibrant: @lightVibrantSwatch,
+          light_muted: @lightMutedSwatch
+      )
+    end
+
+    def already_selected?(swatch)
+      return @vibrantSwatch == swatch ||
+          @darkVibrantSwatch == swatch ||
+          @lightVibrantSwatch == swatch ||
+          @mutedSwatch == swatch ||
+          @darkMutedSwatch == swatch ||
+          @lightMutedSwatch == swatch
+    end
+
+    class << self
+
+      def invert_diff(value, targetValue)
+        1 - (value - targetValue).abs
+      end
+
+      def weightedMean(values)
+        sum = 0
+        sumWeight = 0
+        values.each do |v|
+          value = v[0]
+          weight = v[1]
+          sum += value * weight
+          sumWeight += weight
+        end
+        sum / sumWeight
+      end
+    end
+
+  end
+end

data/vibrant-rb.gemspec

@@ -0,0 +1,25 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'vibrant/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "vibrant"
+  spec.version       = Vibrant::VERSION
+  spec.authors       = ["dondoco7"]
+  spec.email         = ["dondoco7@gmail.com"]
+
+  spec.summary       = %q{Get color variations from an image.}
+  spec.description   = %q{Get color variations from an image. Original is 'http://jariz.github.io/vibrant.js'}
+  spec.homepage      = "https://github.com/dondoco7/vibrant-rb/"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "rmagick", ">= 2.14"
+  spec.add_development_dependency "bundler", "~> 1.10"
+  spec.add_development_dependency "rake", "~> 10.0"
+end

metadata

@@ -0,0 +1,174 @@
+--- !ruby/object:Gem::Specification
+name: vibrant
+version: !ruby/object:Gem::Version
+  version: 0.2.0
+platform: ruby
+authors:
+- dondoco7
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2015-11-22 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rmagick
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.14'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.14'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.10'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: Get color variations from an image. Original is 'http://jariz.github.io/vibrant.js'
+email:
+- dondoco7@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- CODE_OF_CONDUCT.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- examples/.gitignore
+- examples/Gemfile
+- examples/Gemfile.lock
+- examples/README.rdoc
+- examples/Rakefile
+- examples/app/assets/images/.keep
+- examples/app/assets/javascripts/application.coffee
+- examples/app/assets/javascripts/home.coffee
+- examples/app/assets/stylesheets/application.css
+- examples/app/assets/stylesheets/home.scss
+- examples/app/controllers/application_controller.rb
+- examples/app/controllers/concerns/.keep
+- examples/app/controllers/home_controller.rb
+- examples/app/helpers/application_helper.rb
+- examples/app/helpers/home_helper.rb
+- examples/app/mailers/.keep
+- examples/app/models/.keep
+- examples/app/models/concerns/.keep
+- examples/app/uploaders/photo_uploader.rb
+- examples/app/views/home/index.html.erb
+- examples/app/views/layouts/application.html.erb
+- examples/bin/bundle
+- examples/bin/rails
+- examples/bin/rake
+- examples/bin/setup
+- examples/bin/spring
+- examples/config.ru
+- examples/config/application.rb
+- examples/config/boot.rb
+- examples/config/database.yml
+- examples/config/environment.rb
+- examples/config/environments/development.rb
+- examples/config/environments/production.rb
+- examples/config/environments/test.rb
+- examples/config/initializers/assets.rb
+- examples/config/initializers/backtrace_silencers.rb
+- examples/config/initializers/cookies_serializer.rb
+- examples/config/initializers/filter_parameter_logging.rb
+- examples/config/initializers/inflections.rb
+- examples/config/initializers/mime_types.rb
+- examples/config/initializers/session_store.rb
+- examples/config/initializers/wrap_parameters.rb
+- examples/config/locales/en.yml
+- examples/config/routes.rb
+- examples/config/secrets.yml
+- examples/db/seeds.rb
+- examples/lib/assets/.keep
+- examples/lib/tasks/.keep
+- examples/log/.keep
+- examples/public/1.jpg
+- examples/public/2.jpg
+- examples/public/3.jpg
+- examples/public/4.jpg
+- examples/public/404.html
+- examples/public/422.html
+- examples/public/500.html
+- examples/public/favicon.ico
+- examples/public/octocat.png
+- examples/public/robots.txt
+- examples/test/controllers/.keep
+- examples/test/controllers/home_controller_test.rb
+- examples/test/fixtures/.keep
+- examples/test/helpers/.keep
+- examples/test/integration/.keep
+- examples/test/mailers/.keep
+- examples/test/models/.keep
+- examples/test/test_helper.rb
+- examples/vendor/assets/javascripts/.keep
+- examples/vendor/assets/stylesheets/.keep
+- images/1.jpg
+- images/2.jpg
+- images/3.jpg
+- images/4.jpg
+- images/octocat.png
+- lib/vibrant.rb
+- lib/vibrant/swatch.rb
+- lib/vibrant/version.rb
+- lib/vibrant/vibrant.rb
+- vibrant-rb.gemspec
+homepage: https://github.com/dondoco7/vibrant-rb/
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: Get color variations from an image.
+test_files: []