ssimsort 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 9d474e8a2b445d7be76008689d6a53fd569fe07d
+  data.tar.gz: 96310e969b9347869195bc45fa8ecebae19de4f9
+SHA512:
+  metadata.gz: 3bc858ac61bf7db56e23a874e916f6c508aaee59a054795fb12cfd3cb4613c0ba05808976e6992d5fdc1db009a4eac61c57aaabb79a8b24f5026705a24976d5c
+  data.tar.gz: f3051dc4fe04deea95cf6ffdaaf1b3ea9ecd71dd37f112886c56ef0598d00fed43ecb768cbb101594fe8e839dfcb2c44b8ebd3c79f4eac88342fe4beccae5a3c

data/README.md

@@ -0,0 +1,18 @@
+#Ssimsort
+
+
+*Requires* : RMagick
+
+*Install ssimsort* :
+
+~~~
+gem install ssimsort
+~~~
+
+*Use ssim as an executable* :
+
+~~~
+$ ssim <INPUT>, <OUTPUT>, <TOLERANCE>
+
+#Tolerance is a number between 0.0 and 1.0 that defines how similar the images you want to sort will be, 1.0 being the exact same image.
+~~~

data/bin/ssimsort

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+
+root = File.expand_path("../..", __FILE__)
+require "#{root}/lib/main.rb"
+
+
+SsimSort.sort(ARGV[0],ARGV[1],*ARGV[2])

data/lib/array.rb

@@ -0,0 +1,16 @@
+class Array
+
+	def covariance
+		sx, sy, sp = 0.0, 0.0, 0.0
+		return sx if empty?
+	target = first.respond_to?(:to_ary) ? self :
+	self.class.new(size) { |i| [i + 1, at(i)] }
+	target.each { |x, y|
+		sx += x
+		sy += y
+		sp += x * y
+	}
+	(sp - sx * sy / size) / size
+	end
+
+end

data/lib/main.rb

@@ -0,0 +1,51 @@
+#!/usr/bin/env ruby
+
+module SsimSort
+
+	root = File.expand_path("../..", __FILE__)
+	require "numeric_array"
+	require "#{root}/lib/array.rb"
+	require "RMagick"
+	require "fileutils"
+	include Magick
+
+	def SsimSort.cov(x,y)
+		return x.zip(y).covariance
+	end
+
+
+	def SsimSort.get_lum(img,px=100)
+		img = img.scale(px,px)
+		lum_average = img.get_pixels(0, 0, img.columns, img.rows).map do |p|
+			p.to_HSL[2]
+		end
+		return lum_average
+	end
+
+
+	def SsimSort.ssim(file1,file2)
+		cA,cB = 0, 0
+		x,y = get_lum(ImageList.new(file1)), get_lum(ImageList.new(file2))
+		var_x, var_y = x.variance, y.variance
+		moy_x, moy_y = x.mean, y.mean
+		a = (2*moy_x*moy_y+cA)*(2*(cov(x,y))+cB)
+		b = (moy_x**2+moy_y**2+cA)*(var_x+var_y+cB)
+		(a/b) < 0 ? 0 : (a/b).round(3)
+	end
+
+
+	def SsimSort.sort(input_path,output_path,tolerance=0.8)
+		formats =  /(.jpg$|.png$|.JPG$|.PNG$|.gif$|.bmp$|.BMP$)/
+		files = Dir.entries(ARGV[0]).map {|file| File.absolute_path("#{ARGV[0]}/#{file}")}
+		files.select!{|f| formats=~ f}
+		set = files.combination(2).to_a
+		set.each do |file1,file2|
+			path = "#{ARGV[1]}/#{file1.split("/").last}/"
+			dir = FileUtils.makedirs(path) unless File.exists?(path)
+			simil = SsimSort.ssim(file1,file2)
+			simil < tolerance ? next : FileUtils.cp(file2,path)
+		end
+	end
+
+end
+