prizm 0.0.1

data/.gitignore

@@ -0,0 +1,5 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*
+.DS_Store

data/Gemfile

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

data/README.md

@@ -0,0 +1,5 @@
+#Quick start
+
+require 'prizm'
+
+Prizm::Extractor.new.get_colors("http://farm3.staticflickr.com/2334/2400630645_eb571311fe.jpg", 6)

data/Rakefile

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

data/lib/prizm.rb

@@ -0,0 +1,36 @@
+require 'rubygems'
+require 'RMagick'
+
+module Prizm
+  class Extractor
+    def get_colors(img_location, nc=6, remove_bg=true)
+      begin
+        image = remove_bg ? remove_bg(img_location, nc) : reduce_size(Magick::ImageList.new(img_location), 100, 100)
+        q = image.quantize(nc)
+        colors = q.color_histogram.sort {|a, b| b[1] <=> a[1]}.map { |color, count| color }
+      rescue Magick::ImageMagickError => e
+        puts e.message
+        puts e.backtrace
+        nil
+      end
+    end
+    
+    def to_hex(pixel)
+      pixel.to_color(Magick::AllCompliance, false, 8)
+    end
+    
+    def reduce_size(image, width, height)
+      image.change_geometry!("#{width}x#{height}") { |cols, rows, img| img.resize!(cols, rows) }
+    end
+    
+    def remove_bg(img_location, nc, fuzz=20)
+      image = reduce_size Magick::ImageList.new(img_location), 100, 100
+      image = image.border(1, 1, image.pixel_color(0,0))
+      image = image.quantize(nc+4)
+      image.fuzz = "#{fuzz}%"
+      a = image.color_floodfill(0, 0, image.pixel_color(0,0))
+      a = a.matte_floodfill(0,0)
+      a
+    end
+  end
+end

data/lib/prizm/version.rb

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

data/prizm.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "prizm/version"
+
+Gem::Specification.new do |s|
+  s.name        = "prizm"
+  s.version     = Prizm::VERSION
+  s.authors     = ["Quan Nguyen"]
+  s.email       = ["mquannie@gmail.com <mailto:mquannie@gmail.com>"]
+  s.homepage    = ""
+  s.summary     = %q{Prizm is a ruby gem that extracts colors from an input image}
+  s.description = %q{Prizm uses rmagick to extract colors from images}
+
+  s.rubyforge_project = "prizm"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_dependency "rmagick"
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+  # s.add_runtime_dependency "rest-client"
+end

data/test/test.rb

@@ -0,0 +1,20 @@
+require "#{File.dirname(__FILE__)}/../lib/prizm"
+require 'test/unit'
+
+class TestExtractor < Test::Unit::TestCase
+  def test_functional
+    colors = Prizm::Extractor.new.get_colors("#{File.dirname(__FILE__)}/rainbow.jpg", 6, false)
+    assert_equal(6, colors.size)
+  end
+  
+  def test_to_hex
+    extr = Prizm::Extractor.new
+    colors = extr.get_colors("#{File.dirname(__FILE__)}/rainbow.jpg", 6, false)
+    assert_equal('#322D31', extr.to_hex(colors[0]))
+  end
+  
+  def test_error
+    c = Prizm::Extractor.new.get_colors("blahblah.jpg")
+    assert_equal(nil, c)
+  end
+end

metadata

@@ -0,0 +1,88 @@
+--- !ruby/object:Gem::Specification 
+name: prizm
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Quan Nguyen
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-12-22 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rmagick
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+description: Prizm uses rmagick to extract colors from images
+email: 
+- mquannie@gmail.com <mailto:mquannie@gmail.com>
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- lib/prizm.rb
+- lib/prizm/version.rb
+- prizm.gemspec
+- test/rainbow.jpg
+- test/test.rb
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: prizm
+rubygems_version: 1.8.6
+signing_key: 
+specification_version: 3
+summary: Prizm is a ruby gem that extracts colors from an input image
+test_files: 
+- test/rainbow.jpg
+- test/test.rb