color_extract 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: fcdf8c9b4f5cae0c0df68f6253ac81af6435457a
+  data.tar.gz: 9ff5cc7c5c43e25a1183a440e18e7bf8cd2c27d1
+SHA512:
+  metadata.gz: 79583d8b9e76d7f89ff0abd2f8a4b7646d72d807337516e7b0f2480ec8cee6b926d07e6eb30ae858cd0b2513c407678e31aa44677709549604acd77cbe581434
+  data.tar.gz: 4199a9b1d150627ac1b7a9b68cee415f46364cfd07e231faf8423c21602e97e0abd38ebe9ddb36f7652af923cdfab2ed5adc6fd9a7dadc6a7f93dec789b4648a

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in color_extract.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 qhwa
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,5 @@
+# ColorExtract
+
+This gem extract important colors from an image.
+
+on construction

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/color_extract.gemspec

@@ -0,0 +1,25 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'color_extract/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "color_extract"
+  spec.version       = ColorExtract::VERSION
+  spec.authors       = ["qhwa"]
+  spec.email         = ["qhwa@163.com"]
+  spec.summary       = %q{extract color information from images}
+  spec.description   = %q{extract color information from images}
+  spec.homepage      = "https://github.com/qhwa/color_extract"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "colorscore", "~> 0.0.4"
+
+  spec.add_development_dependency "bundler", "~> 1.5"
+  spec.add_development_dependency "rake"
+end

data/example/extract.rb

@@ -0,0 +1,18 @@
+$:.unshift File.expand_path( '../../lib', __FILE__ )
+require 'color_extract'
+
+file = ARGV[0]
+if file
+  puts "File: #{file}"
+  ColorExtract::Analytics.new( file ).valuable_colors.each do |per, color|
+    print "  - %#.2f%% %s %s" % [per*100, color.html, color.css_hsl]
+    print "\n"
+  end
+
+  puts "palette:"
+  ColorExtract::Palette.new( file ).palette.each do |name, color|
+    next unless color
+    print "  - %s : %s %s" % [ name.to_s.ljust(6), color.html, color.css_hsl ]
+    print "\n"
+  end
+end

data/lib/color_extract/analytics.rb

@@ -0,0 +1,178 @@
+require 'colorscore'
+
+module ColorExtract
+
+  MAX_VISIBLE_COLORS = 128
+
+  # Public: 分析图片的色彩信息，相似的颜色会合并在一起
+  #
+  # Example:
+  # 
+  #   img = "http://distilleryimage2...jpg"
+  #   colors = ColorExtract::Analytics.new( img ).valuable_colors
+  #   colors.each do |percent, color|
+  #     puts "#{color.html} : #{percent}"
+  #   end
+  class Analytics
+
+    include ColorUtil
+
+    attr_reader :merge_factor
+
+    def initialize( img )
+      @img = img
+    end
+
+    # Public: 获取颜色数组
+    #
+    # - merge_factor: 颜色合并系数，数字越大，相似的颜色越容易被合并在一起
+    #
+    # Returns 颜色数组，数组的每个元素都是 [percent, color] 结构
+    def valuable_colors( merge_factor: 5 )
+      @merge_factor = merge_factor
+      @colors ||= calc_valuable_colors
+    end
+
+    private
+      
+      def calc_valuable_colors
+        @colors = @raw_colors = visible_colors
+
+        # 黑白灰是不具有色相的颜色，可以和任何色相的颜色搭配
+        # 所以不是我们关心的颜色
+        remove_gray_colors
+
+        # 去掉灰色之后重新计算一下百分比，这样后期就不会有
+        # 太多比例特别少的颜色，防止色彩丢失过多
+        update_percents
+
+        # 视觉上相近的颜色都合并成一种进行统计
+        # 这样得到的颜色比例会更加精确
+        merge_similar_colors
+
+        # 数量太少的颜色很有可能是主色的过渡
+        # 可以去掉和主色同色系的低比例色彩
+        remove_few_colors
+
+        return @colors
+      end
+
+    public
+
+      # Public: 获取可见的颜色，这里的颜色是没有经过相似合并的
+      # 
+      # Returns 可见颜色数组，数组每个元素都是 [percent, color] 结构
+      def visible_colors
+        Colorscore::Histogram.new( @img, MAX_VISIBLE_COLORS).scores
+      end
+
+    private
+  
+      def remove_gray_colors
+        @colors.reject! do |per, color|
+          hsl = color.to_hsl
+          # 灰色系颜色
+          l, a, b = *rgb2lab(color.to_rgb)
+          hsl.s < 0.1 or
+            # 接近白色
+            l > 83 or
+            # 接近黑色
+            l < 25
+        end
+      end
+
+      def update_percents
+        total = @raw_colors.reduce(0) {|sum, info| sum + info[0]}
+        @colors.each do |info|
+          per, c = *info
+          info[0] = per / total
+        end
+      end
+
+      # TODO: 重构这个函数
+      # 这里应该可以逻辑更加简单一些的
+      def merge_similar_colors
+
+        @colors.each do |info|
+          per, c = *info
+          info[1] = pure( dither( c ), s: nil, l:nil )
+        end
+
+        auto_link_colors!
+
+        new_colors = {}
+        @colors.each do |per, color|
+          link_to = parent_color(color)
+          if link_to
+            new_colors[link_to] ||= 0
+            new_colors[link_to] += per
+          else
+            new_colors[color.html] = per
+          end
+        end
+
+        @colors = new_colors.map do |color_html, per|
+          [per, Color::RGB.from_html(color_html)]
+        end
+      end
+
+      def auto_link_colors!
+        @merge_link = {}
+        colors      = @colors.sort_by {|per, c| c.to_hsl.h }
+        len         = colors.size - 1
+        
+        0.upto(len) do |i|
+          per, color, lab = *colors[i]
+          hsl = color.to_hsl
+
+          (i+1).upto(len) do |j|
+            per2, color2, lab2 = *colors[j]
+
+            if should_merge?(color, color2)
+              # 相近的颜色，合并到面积更大的那种颜色
+              # 这里尝试过另外的策略，比如合并到更鲜艳
+              # 的那种颜色，但是效果不是很好
+              if per >= per2
+                @merge_link[color.html]  = parent_color(color2)
+              else
+                @merge_link[color2.html] = parent_color(color)
+              end
+            end
+          end
+        end
+      end
+
+      def should_merge?( color1, color2 )
+        hue_similarity( color1, color2 ) <= 20 && similarity( color1, color2, ignore_lightness: true ) < @merge_factor
+      end
+
+      def parent_color color
+        parent = @merge_link[color.html]
+        if parent && parent != color.html
+          parent_color( Color::RGB.from_html(parent) )
+        else
+          color.html
+        end
+      end
+
+      def remove_few_colors
+        sorted_colors = @colors.sort_by {|per,c| -per}
+        if sorted_colors.size > 5
+          sorted_colors.reject! {|per, c| per < 0.0005 }
+        end
+        common_colors = sorted_colors
+        little_colors = sorted_colors.select {|per, c| per < 0.05 }
+
+        little_colors.select! do |per, c|
+          common_colors.any? do |common_per, common_color|
+            common_per > per && hue_similarity( c, common_color ) <= 10
+          end
+        end
+
+        @colors = sorted_colors - little_colors
+      end
+
+  end
+
+end
+

data/lib/color_extract/color_util.rb

@@ -0,0 +1,131 @@
+module ColorExtract
+
+  module ColorUtil
+
+    # Corresponds roughly to RGB brighter/darker
+    K = 18
+
+    # D65 standard referent
+    X = 0.950470
+    Y = 1
+    Z = 1.088830
+
+    # 使用 CIE76 公式计算颜色的相似程度
+    # 参考: 
+    # - [CIELAB](http://en.wikipedia.org/wiki/CIELAB)
+    # - [Color Difference](http://en.wikipedia.org/wiki/Color_difference)
+    def similarity( color1, color2, ignore_lightness: false )
+      l1, a1, b1 = *rgb2lab(color1.to_rgb)
+      l2, a2, b2 = *rgb2lab(color2.to_rgb)
+      Math.sqrt( (ignore_lightness ? 0 : (l1-l2)**2) + (a1-a2)**2 + (b1-b2)**2 )
+    end
+
+    # Public: 判断两种颜色的色系相似程度
+    #         在色盘上越接近，返回值越小
+    #
+    # Returns 色相距离，0-180之间的角度值
+    def hue_similarity( color1, color2 )
+      deg1   = color1.to_hsl.h
+      deg2   = color2.to_hsl.h
+      deg1  += 1 if deg1 < 0
+      deg2  += 1 if deg2 < 0
+      delta  = (deg1 - deg2).abs
+      delta  = 1 - delta if delta > 0.5
+      delta * 360
+    end
+  
+    # 来自 github 上的开源项目 chroma
+    # https://github.com/gka/chroma.js/blob/master/src/conversions/rgb2lab.coffee
+    #
+    # color - RGB颜色
+    #
+    # Returns [l*, a*, b*] 值。
+    #         亮度（l*）的范围是（0-100）
+    def rgb2lab(color)
+      r, g, b = color.r * 255, color.g * 255, color.b * 255
+      r = rgb_xyz r
+      g = rgb_xyz g
+      b = rgb_xyz b
+      x = xyz_lab (0.4124564 * r + 0.3575761 * g + 0.1804375 * b) / X
+      y = xyz_lab (0.2126729 * r + 0.7151522 * g + 0.0721750 * b) / Y
+      z = xyz_lab (0.0193339 * r + 0.1191920 * g + 0.9503041 * b) / Z
+      [116 * y - 16, 500 * (x - y), 200 * (y - z)]
+    end
+
+    def rgb_xyz (r)
+      if (r /= 255) <= 0.04045 then r / 12.92 else ((r + 0.055) / 1.055) ** 2.4 end
+    end
+
+    def xyz_lab (x)
+      if x > 0.008856 then x ** (1.0/3) else 7.787037 * x + 4 / 29 end
+    end
+
+    # Public: 根据背景色，计算适用于显示其上的文字的颜色
+    def readable_textcolor( bg: nil, accent: nil, lock_accent: false)
+      l1, a1, b1 = *rgb2lab(bg)
+      if accent
+        l2, a2, b2 = *rgb2lab(accent)
+        return accent if (l1 - l2).abs > 255 * 0.8
+      end
+      
+      # TODO: ajust accent unless lock_accent
+      # 白色会显得品质高档，因此尽量使用白色
+      if l1 < 75
+        Color::RGB.from_html( '#FFFFFF' )
+      else
+        bg.to_hsl.tap { |c| c.l = 0.1 }.to_rgb
+      end
+    end
+
+    def pure( color, s: 1, l: 1 )
+      color.to_hsl.tap do |c|
+        if s
+          c.s = s
+        else
+          c.s += (1 - c.s) * 0.382
+        end
+        c.l = l if l
+      end.to_rgb
+    end
+
+    def dither( color )
+      color.to_hsl.tap do |c|
+        if c.s > 0.8
+          c.s -= (c.s) * 0.3
+          c.l -= 0.1
+        else
+          c.s += (1-c.s) * 0.5
+        end
+      end
+    end
+
+    def reverse_color( color )
+      color.to_hsl.tap do |c|
+        # 色盘旋转60度
+        h = c.h + 1.0/6
+        h -= 1 if h > 1
+        c.h = h
+      end.to_rgb
+    end
+
+    def darken( hsl_color, percent )
+      hsl_color.tap { |c| darken!( c, percent ) }
+    end
+
+    def darken!( hsl_color, percent )
+      l = hsl_color.l
+      hsl_color.l -= (1-l) * percent
+    end
+
+    def lighten( hsl_color, percent )
+      hsl_color.tap { |c| lighten!( c, percent ) }
+    end
+
+    def lighten!( hsl_color, percent )
+      l = hsl_color.l
+      hsl_color.l += l * percent
+    end
+
+  end
+
+end

data/lib/color_extract/palette.rb

@@ -0,0 +1,188 @@
+module ColorExtract
+
+  class Palette
+
+    include ColorUtil
+
+    class << self
+      def from_colors( colors )
+        new.tap do |palette|
+          palette.raw_colors = colors.to_a
+        end
+      end
+    end
+
+    attr_accessor :raw_colors, :main_color, :max_pri_brightness
+
+    def initialize( file = nil )
+      @analytics = Analytics.new( file ) if file
+    end
+
+    # 目前只生成一种配色方案
+    def palette( accent_seed: nil )
+      palettes( count: 1, accent_seed: accent_seed ).first
+    end
+
+    def palettes( opts={} )
+      @raw_colors ||= @analytics.valuable_colors.map {|per, c| c }
+      @palettes   ||= gen_palettes( opts )
+    end
+
+    private
+
+      # 目前采用最简单
+      def gen_palettes( count: 1, accent_seed: nil, max_pri_brightness: 1 )
+        count               = 1 if accent_seed
+        @max_pri_brightness = max_pri_brightness
+        @main_color         = most_possible_main_color
+        @accent_colors      = []
+
+        palettes = count.times.map do |i|
+          gen_palette( accent_seed: accent_seed, fewest_index: i )
+        end.compact.sort_by do |pal|
+          pal[:primary].to_hsl.h
+        end
+
+        if palettes.size < count
+
+          if palettes.empty?
+            return preset_palettes( count: count )
+          end
+
+          last_primary_color = palettes.last[:primary].to_hsl
+          main_color         = palettes.last[:back].to_hsl
+          puts last_primary_color.html
+          (count - palettes.size ).times do |i|
+            if i.even?
+              color = last_primary_color.dup.tap do |c|
+                h   = c.h
+                h  += ((i+1)/2.0).ceil / 10.0
+                h  += 1 if h < 0 
+                c.h = h
+                c.l = 0.33
+              end.to_rgb
+            else
+              color = main_color.dup.tap do |c|
+                h   = c.h
+                h  -= ((i+1)/2.0).ceil / 10.0
+                h  += 1 if h < 0 
+                c.h = h
+                c.l = 0.33
+              end.to_rgb
+            end
+            next if already_have_similar_accent_color?( color )
+            palettes << gen_palette( accent_seed: color )
+          end
+        end
+
+        palettes.compact
+      end
+
+      def preset_palettes( count: 1 )
+        0.upto(count).map do |i|
+          accent = Color::HSL.from_fraction( i*(1.0 /(count+1)), 0.95, 0.4 )
+          main   = Color::RGB.from_html( '#ffffff' )
+          { 
+            :primary      => accent.to_rgb,
+            :back         => main,
+            :"pri-dark"   => darken( accent, 0.2 ).to_rgb,
+            :"pri-light"  => lighten( accent, 0.2 ).to_rgb,
+            :"text"       => readable_textcolor( bg: accent.to_rgb, accent: main.to_rgb )
+          }
+        end
+      end
+          
+      def most_possible_main_color
+        raw_colors.size > 0 ? raw_colors.first : Color::RGB.from_html( '#CCCCCC' )
+      end
+
+      # Private: 生成配色方案
+      #
+      # accent_seed:  指定的 primary 颜色
+      # fewest_index: 指定倒数第几个最少的颜色
+      #               0 - 最少的颜色;
+      #               1 - 倒数第二少的颜色;
+      #               以此类推
+      #
+      # Returns 一种配色方案
+      def gen_palette( accent_seed: nil, fewest_index: 0 )
+        accent_color = accent_seed || most_possible_accent_color( fewest_index )
+        if !accent_color || already_have_similar_accent_color?( accent_color )
+          return nil
+        else
+          @accent_colors << accent_color
+        end
+
+        if accent_color
+          light, dark  = most_possible_around_colors( accent_color )
+          text_color   = readable_textcolor bg: accent_color, accent: main_color
+          {
+            primary:        accent_color,
+            :'pri-light' => light,
+            :'pri-dark'  => dark,
+            back:           @main_color,
+            text:           text_color
+          }
+        end
+      end
+
+      def most_possible_accent_color( fewest_index=0, &block )
+
+        color = if raw_colors.size > fewest_index + 1
+          # 除去最多的一种颜色
+          # 因为最多的颜色是作为 back
+          raw_colors.reverse[0..-2].compact.sort_by do |c|
+            hue_simi   = hue_similarity(c, main_color)
+            simi       = hue_simi > 60 ? -0.5*hue_simi + 120 : 1.5*hue_simi
+            l,a,b      = *rgb2lab(c)
+            sat        = c.to_hsl.s
+            # 按和主要颜色的差异系数由小到大排
+            # 饱和度越高、亮度越低的越有可能成为主色
+            # simi: 0-90; l: 0-100; sat:  0-1
+            simi * 2 + sat * 10 - l * 7.01
+          end.reverse[fewest_index]
+        end
+
+        if max_pri_brightness && color
+          color.to_hsl.tap do |c|
+            c.l = max_pri_brightness if c.l > max_pri_brightness
+          end.to_rgb
+        else
+          color
+        end
+      end
+
+      def already_have_similar_accent_color?( color )
+        @accent_colors.any? do |c|
+          similarity( c, color, ignore_lightness: false ) < 20
+        end
+      end
+
+      def most_possible_around_colors( accent_color )
+        sub_color1 = nil
+        sub_color2 = nil
+
+        if raw_colors.size > 4
+          raw_colors[1..-1].each do |c|
+            if hue_similarity(c, accent_color ) < 30
+              if sub_color1.nil?
+                sub_color1 = c.to_hsl.tap do |c|
+                  c.l = accent_color.to_hsl.l + 0.2
+                end.to_rgb
+              else
+                sub_color2 = c.to_hsl.tap do |c|
+                  c.l = accent_color.to_hsl.l - 0.2
+                end.to_rgb
+              end
+            end
+          end
+        end
+
+        sub_color1 ||= accent_color.to_hsl.tap {|c| lighten!(c, 0.2) }.to_rgb
+        sub_color2 ||= accent_color.to_hsl.tap {|c| darken!(c, 0.2) }.to_rgb
+        [sub_color1, sub_color2]
+      end
+
+  end
+
+end

data/lib/color_extract/version.rb

@@ -0,0 +1,3 @@
+module ColorExtract
+  VERSION = "0.0.1"
+end

data/lib/color_extract.rb

@@ -0,0 +1,4 @@
+require "color_extract/version"
+require 'color_extract/color_util'
+require "color_extract/analytics"
+require 'color_extract/palette'

metadata

@@ -0,0 +1,98 @@
+--- !ruby/object:Gem::Specification
+name: color_extract
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- qhwa
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-04-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: colorscore
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.4
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.0.4
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: extract color information from images
+email:
+- qhwa@163.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- color_extract.gemspec
+- example/extract.rb
+- lib/color_extract.rb
+- lib/color_extract/analytics.rb
+- lib/color_extract/color_util.rb
+- lib/color_extract/palette.rb
+- lib/color_extract/version.rb
+homepage: https://github.com/qhwa/color_extract
+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.2.2
+signing_key: 
+specification_version: 4
+summary: extract color information from images
+test_files: []