hjw3001-waldo 0.1.0

data/Manifest

@@ -0,0 +1,5 @@
+init.rb
+lib/waldo.rb
+Rakefile
+README.rdoc
+Manifest

data/README.rdoc

@@ -0,0 +1,36 @@
+= Uniquify
+
+Ruby gem to create a mosaic image from a series of smaller images.
+
+
+== Install
+
+  gem install hjw3001-waldo --source http://gems.github.com
+
+
+== Usage
+
+TBD
+
+== License
+
+Copyright (c) 2009 Henry Wagner
+ 
+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/Rakefile

@@ -0,0 +1,14 @@
+require 'rubygems'
+require 'rake'
+require 'echoe'
+
+Echoe.new('waldo', '0.1.0') do |p|
+  p.description    = "Ruby gem to create a mosaic image from a series of smaller images."
+  p.url            = "http://github.com/hjw3001/waldo"
+  p.author         = "Henry Wagner"
+  p.email          = "hjw3001@gmail.com"
+  p.ignore_pattern = ["tmp/*", "script/*"]
+  p.development_dependencies = []
+end
+
+Dir["#{File.dirname(__FILE__)}/tasks/*.rake"].sort.each { |ext| load ext }

data/init.rb

@@ -0,0 +1 @@
+require 'waldo'

data/lib/waldo.rb

@@ -0,0 +1,193 @@
+require 'rmagick'
+require 'fileutils'
+require 'erb'
+require 'rexml/document'
+require 'yaml'
+
+class Waldo
+
+  @@template  = <<EOF
+# .waldo
+# 
+# Please fill in fields like this:
+#
+#  cache: c:/waldo/cache
+#  thumb: c:/waldo/thumb
+#  waldoized: c:/waldo/waldoized
+#
+cache: 
+thumb: 
+waldoized: 
+EOF
+
+  attr_reader :photos, :config
+  
+  def initialize
+    @config = create_or_find_config
+    @photos = Array.new
+    load_photo_data
+  end
+  
+  # Move these to waldo gem
+  def waldoize(photo_id,size)
+    amount = config['colorize_percentage']
+    file = Magick::Image.read("#{config['thumb']}/#{photo_id}.jpg").first
+    mosaic = Magick::Image.new(file.columns * size, file.rows * size)
+      
+    file.rows.times {|y|
+      file.columns.times {|x|
+        pixel_color = file.pixel_color(x,y)
+        color = Pixel.as_int(pixel_color)
+        match = find_closest_photo(photos, color)
+        # Load the matching 75x75 version of the photo
+        thumb = Magick::Image.read("#{config['cache']}/#{match.external_id}.jpg").first
+        # Delete the thumbnail from the list once it has been used
+        delete_from_photos(match)
+        # Convert it to gray
+        gray = thumb.quantize(256, Magick::GRAYColorspace)
+        # Colorize the gray version to match the current pixel color
+        colorized = gray.colorize(amount, amount, amount, '#' + to_hex(pixel_color.red, pixel_color.green, pixel_color.blue))
+        # Add the colorized version of the photo to the final image
+        colorized.rows.times {|ty|
+          colorized.columns.times {|tx|
+            mosaic.pixel_color((x*size) + tx, (y*size) + ty, Magick::Pixel.new(colorized.pixel_color(tx,ty).red, colorized.pixel_color(tx,ty).green, colorized.pixel_color(tx,ty).blue, 0))
+          }
+        }
+      }
+    }
+    mosaic.write("waldo_#{photo_id}.jpg")
+  end
+  
+  def build_xml_cache
+    photos = loadCache
+    xml_data = Array.new
+    for photo in photos
+      photo_median_data = Array.new
+      file = Magick::Image.read("#{photo}").first
+      file.rows.times {|y|
+        file.columns.times {|x|
+          pixel_color = file.pixel_color(x,y)
+          color = Pixel.as_int(pixel_color)
+          photo_median_data << color
+        }
+      }
+      photo_median_data.sort!
+      xml_data << PhotoXML.new(photo, photo_median_data[photo_median_data.size / 2])
+    end
+    # xml = REXML::Document.new(File.open("#{@config['xml']}"))
+    File.open("#{@config['xml']}", 'wb') do |dest|
+      dest.write ERB.new(File.open('cache.xml').read, nil, '-').result(binding)
+    end
+  end
+  
+  def loadCache
+    jpgfiles = File.join(config['cache'], "*.jpg")
+    Dir.glob(jpgfiles)
+  end
+  
+  protected
+
+  def find_closest_photo(photos, color)
+    if (photos.size == 1)
+      photos[0]
+    elsif photos[photos.size/2].median_int > color
+      find_closest_photo(photos.values_at(0...((photos.size/2))), color)
+    else
+      find_closest_photo(photos.values_at(photos.size/2...photos.size), color)
+    end
+  end
+  
+  def to_hex(red, green, blue)
+    base = 16
+    "#{red.to_s(base=base).rjust(2, '0')}#{green.to_s(base=base).rjust(2, '0')}#{blue.to_s(base=base).rjust(2, '0')}"
+  end
+  
+  def create_grey(photo, size)
+    if (!File.exists?("public/images/#{size}/#{photo.external_id}.jpg"))
+      image = Magick::Image.read("public/images/cache/#{photo.external_id}.jpg").first  
+      image.resize!(size,size)
+      grey = image.quantize(256, Magick::GRAYColorspace)
+      grey.write("public/images/#{size}/#{photo.external_id}.jpg")
+    end
+  end
+  
+  def create_or_find_config
+    home = ENV['HOME'] || ENV['USERPROFILE'] || ENV['HOMEPATH']
+    begin
+      config = YAML::load open(home + "/.waldo")
+    rescue
+      open(home + '/.waldo','w').write(@@template)
+      config = YAML::load open(home + "/.waldo")
+    end
+            
+    if config['cache'] == nil or config['thumb'] == nil
+      puts "Please edit ~/.waldo to include your flickr api_key and secret\nTextmate users: mate ~/.waldo"
+      exit(0)
+    end
+            
+    config
+  end
+  
+  private
+  
+  def load_photo_data
+    xml = REXML::Document.new(File.open("#{@config['xml']}"))
+    xml.elements.each("//photo") {|photo| photos << Photo.new(photo) }
+    # Sort photos by median_int
+    photos.sort!{|x,y| x.median_int <=> y.median_int}
+  end
+  
+  def delete_from_photos(photo)
+    photos.delete(photo)
+    if (photos.size == 0)
+      load_photo_data
+    end
+  end
+
+end
+
+class Pixel
+  def self.as_int(pixel)
+    base = 16
+    "#{pixel.red.to_s(base=base).rjust(2, '0')}#{pixel.green.to_s(base=base).rjust(2, '0')}#{pixel.blue.to_s(base=base).rjust(2, '0')}".to_i(16)
+  end
+end
+
+#    xml.elements.each("//photo") {|photo| Photo.create(:external_id => photo.attributes['id'].split('.').first,
+#                                                       :mean => photo.attributes['a'],
+#                                                       :median => photo.attributes['m'],
+#                                                       :mean_red => photo.attributes['ar'],
+#                                                       :mean_green => photo.attributes['ag'],
+#                                                       :mean_blue => photo.attributes['ab'],
+#                                                       :median_red => photo.attributes['mr'],
+#                                                       :median_green => photo.attributes['mg'],
+#                                                       :median_blue => photo.attributes['mb'])}
+class Photo
+  attr_reader :external_id, :mean, :median, :mean_red, :mean_green, :mean_blue, :median_red, :median_green, :median_blue
+
+  def initialize(photo)
+    @external_id = photo.attributes['id'].split('.').first
+    @mean = photo.attributes['a']
+    @median = photo.attributes['m']
+    @mean_red = photo.attributes['ar']
+    @mean_green = photo.attributes['ag']
+    @mean_blue = photo.attributes['ab']
+    @median_red = photo.attributes['mr']
+    @median_green = photo.attributes['mg']
+    @median_blue = photo.attributes['mb']
+  end
+  
+  def median_int
+    median.to_i(16)
+  end
+end
+
+class PhotoXML
+  attr_reader :external_id, :median
+  
+  def initialize(photo, median)
+    @external_id = photo.slice(photo.rindex('/') + 1, photo.size)
+    @median = median
+  end  
+
+end

data/waldo.gemspec

@@ -0,0 +1,29 @@
+Gem::Specification.new do |s|
+  s.name = %q{waldo}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 1.2") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Henry Wagner"]
+  s.date = %q{2009-02-12}
+  s.description = %q{Ruby gem to create a mosaic image from a series of smaller images.}
+  s.email = %q{hjw3001@gmail.com}
+  s.extra_rdoc_files = ["lib/waldo.rb", "README.rdoc"]
+  s.files = ["init.rb", "lib/waldo.rb", "Rakefile", "README.rdoc", "Manifest", "waldo.gemspec"]
+  s.has_rdoc = true
+  s.homepage = %q{http://github.com/hjw3001/waldo}
+  s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Waldo", "--main", "README.rdoc"]
+  s.require_paths = ["lib"]
+  s.rubyforge_project = %q{waldo}
+  s.rubygems_version = %q{1.2.0}
+  s.summary = %q{Ruby gem to create a mosaic image from a series of smaller images.}
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 2
+
+    if current_version >= 3 then
+    else
+    end
+  else
+  end
+end

metadata

@@ -0,0 +1,64 @@
+--- !ruby/object:Gem::Specification 
+name: hjw3001-waldo
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Henry Wagner
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-02-12 00:00:00 -08:00
+default_executable: 
+dependencies: []
+
+description: Ruby gem to create a mosaic image from a series of smaller images.
+email: hjw3001@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- lib/waldo.rb
+- README.rdoc
+files: 
+- init.rb
+- lib/waldo.rb
+- Rakefile
+- README.rdoc
+- Manifest
+- waldo.gemspec
+has_rdoc: true
+homepage: http://github.com/hjw3001/waldo
+post_install_message: 
+rdoc_options: 
+- --line-numbers
+- --inline-source
+- --title
+- Waldo
+- --main
+- README.rdoc
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "1.2"
+  version: 
+requirements: []
+
+rubyforge_project: waldo
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: Ruby gem to create a mosaic image from a series of smaller images.
+test_files: []
+