identiconify 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 437de306876ca2063bb4da1d21c071a22fc2ab6f
-  data.tar.gz: 10db2215c471afbef61efb9cec4c1260fb04dc5e
+  metadata.gz: eee400110cb4bc5725c78c80660d096234dc8b66
+  data.tar.gz: f18b0408a27534063be413e461bc387713ab7cfc
 SHA512:
-  metadata.gz: 36d54596ba82a24cd92ef29a75fa4c1ffa856f30331a42ad93e9162c4324240a3e23df780543bcb8898fd1a358564f62b2bb63cf2f7b0f4c6a5ae580f913c2da
-  data.tar.gz: 1edda89f727358415794d332c20b5f9b35cf8d212c4a5aa00eea2165638ea0b1450f2360c2b4a425b519640c5d85358f24b82a9feccc6f06dae2a25efcbaf57f
+  metadata.gz: 2c1250f3827a233091811db3277c5120cab852fed2e9f9618ddec02d92f2938bab4d8d5166db8aac4af435b43894e50f1c4dd0c8c11ee0269b4c9fcc155c355f
+  data.tar.gz: dd5c602bfcb43c70137f577e3f0b6c93d02916fae2b977244eea1ae912fa6704c4010ce93cec07c0b2026db6baab02cb41fc4f71141c2566a016176866613f74

data/lib/identiconify/version.rb

@@ -1,3 +1,3 @@
 module Identiconify
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/lib/identiconify.rb

@@ -6,38 +6,69 @@ module Identiconify
   class Identicon
     HASH_KEY = "61616f73646a6173646a616f7369646a"
 
-    attr_reader :string, :square_size, :row_count, :width, :inverse_offset
+    attr_reader :string,
+                :square_size,
+                :row_count,
+                :size,
+                :inverse_offset,
+                :colors
 
     def initialize(string, options={})
       @string = string
-      @width = options.fetch(:width) { 250 }
+      @size = options.fetch(:size) { 250 }
       @row_count = 5
-      @square_size = @width / @row_count
+      @square_size = @size / @row_count
       # Since we can't draw subpixels we need to calculate how much we have to
       # offset the inverted version of the identicon to not create gaps or
       # overlaps in the middle of the image.
-      @inverse_offset = @width - @square_size * @row_count
+      @inverse_offset = @size - @square_size * @row_count
+      @colors = options.fetch(:colors) { :default }.to_sym
     end
 
     def column_count
       row_count.even? ? row_count/2 : row_count/2+1
     end
 
-    def to_png_blob
-      hash = SipHash.digest(HASH_KEY, string)
-
+    def color_for_hash(hash)
       # Use the three first bytes of the hash to generate a color
       r = hash & 0xff
       g = (hash >> 8) & 0xff
       b = (hash >> 16) & 0xff
-      a = 0xff
-      color = ChunkyPNG::Color.rgba(r,g,b,a)
+      transform_color ChunkyPNG::Color.rgb(r,g,b)
+    end
+
+    def greyscale(color)
+      ChunkyPNG::Color.to_grayscale(color)
+    end
+
+    def tint(color)
+      r = ChunkyPNG::Color.r(color)
+      g = ChunkyPNG::Color.g(color)
+      b = ChunkyPNG::Color.b(color)
+      r += (0.3*(255-r)).round
+      g += (0.3*(255-g)).round
+      b += (0.3*(255-b)).round
+      ChunkyPNG::Color.rgb(r,g,b)
+    end
+
+    def transform_color(color)
+      case colors
+      when :bw then greyscale(color)
+      when :tinted then tint(color)
+      else color
+      end
+    end
+
+    def to_png_blob
+      hash = SipHash.digest(HASH_KEY, string)
+
+      color = color_for_hash(hash)
       bg_color = ChunkyPNG::Color::TRANSPARENT
 
-      # Remove the used three bytes
+      # Remove the used three color bytes
       hash >>= 24
 
-      png = ChunkyPNG::Image.new(width, width, bg_color)
+      png = ChunkyPNG::Image.new(size, size, bg_color)
       0.upto(row_count-1).each do |row|
         0.upto(column_count-1).each do |column|
           if hash & 1 == 1
@@ -48,14 +79,14 @@ module Identiconify
             png.rect(x0, y0, x1, y1, color, color)
 
             # Inverse the x coordinates making the image mirrored vertically
-            x0 = width-(column+1)*square_size-inverse_offset
-            x1 = width-column*square_size-inverse_offset-1
+            x0 = size-(column+1)*square_size-inverse_offset
+            x1 = size-column*square_size-inverse_offset-1
             png.rect(x0, y0, x1, y1, color, color)
           end
           hash >>= 1
         end
       end
-      png.to_blob :color_mode => ChunkyPNG::COLOR_INDEXED
+      png.to_blob :fast_rgba
     end
   end
 end

data/test/test.rb

@@ -0,0 +1,12 @@
+require_relative '../lib/identiconify'
+
+string = ARGV[0]
+identicon = Identiconify::Identicon.new(string, size: 512, colors: :tinted)
+png_data = identicon.to_png_blob
+tmpdir = File.join(File.dirname(__FILE__), "tmp")
+Dir.mkdir(tmpdir) unless File.directory?(tmpdir)
+filename = File.join(tmpdir, "image.png")
+File.open(filename, "w") do |file|
+  file.write(png_data)
+end
+system "open #{filename}"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: identiconify
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - calleerlandsson
@@ -94,6 +94,7 @@ files:
 - identiconify.gemspec
 - lib/identiconify.rb
 - lib/identiconify/version.rb
+- test/test.rb
 homepage: https://github.com/calleerlandsson/identiconify
 licenses:
 - MIT
@@ -118,4 +119,5 @@ rubygems_version: 2.0.6
 signing_key: 
 specification_version: 4
 summary: Super simple identicons
-test_files: []
+test_files:
+- test/test.rb