ph 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 88a7e900bca90ab2c70029559c46848add5f8384a2c60040f012b1ac16691024
+  data.tar.gz: 4f38f845987a7c33e6d8adb70766e0ea0a0f9758a0efe0e01655370f488f0b96
+SHA512:
+  metadata.gz: 47c53e0c8a34d0a5c1cbde470000b5de32a246dae9d3f15582ed024b9aaed938fdfcb8b141d5ec845b505765a2ccbab1b46f31b6173886d00c3ff3fcfc1fd7e2
+  data.tar.gz: 76d82783e82eb7f8d321a79dd21e6a72cafc5014381a6498dcb217f3a29c3c9ab3f1ec9fbf8b4099e31677e1919b4ee1f836e10f520c574a966f06c3762dea86

data/Makefile

@@ -0,0 +1,13 @@
+.PHONY: *
+
+default: test
+
+build:
+	rm -f *.gem
+	gem build ph.gemspec
+
+publish: build
+	gem push *.gem
+
+test:
+	ruby -Ilib:spec spec/*_spec.rb

data/lib/ph.rb

@@ -0,0 +1,125 @@
+# frozen_string_literal: true
+#
+class PH
+  attr_reader :pixels, :size
+
+  def initialize(pixels_2d, size: 64)
+    @pixels = pixels_2d
+    @size = size
+  end
+
+  def hash
+    # Binary to hex string
+    vector.join.to_i(2).to_s(16)
+  end
+
+  def vector
+    @_vector ||= begin
+      # Get DCT2D of the pixels
+      dct_pixels = dct2d(pixels)
+      # Get high frequency corner
+      sqrt = Math.sqrt(size).to_i
+      coords = Array.new(2, sqrt)
+      corner = flat1d(coords, dct_pixels)
+      corner_size = corner.length
+      # Median values
+      med = median(corner)
+
+      result = Array.new(size, 0)
+
+      corner.each.with_index do |f, i|
+        # Compare each value to the median
+        result[i] = 1 if f > med
+      end
+
+      result
+    end
+  end
+
+  private
+
+  # DCT on a bidimensional plane
+  # Thank you matlab forum for explaining that it's just transpositions
+  # https://www.mathworks.com/matlabcentral/answers/405088-how-to-implement-dct2-in-matlab-coder
+  #
+  def dct2d(pixels_2d)
+    dct_fn = -> (px) { dct(px) }
+
+    pixels_2d
+      .map(&dct_fn).transpose
+      .map(&dct_fn).transpose
+  end
+
+  # Slices 2D vector into a 1D version
+  #
+  def flat1d(coords, pixels_2d)
+    slice = []
+    x, y = coords
+
+    Array.new(x) do |i|
+      Array.new(y) do |j|
+        slice << pixels_2d[i][j]
+      end
+    end
+
+    slice
+  end
+
+  # Quickselect median.
+  # https://rcoh.me/posts/linear-time-median-finding/
+  #
+  def median(vector)
+    return nil if vector.empty?
+
+    n = vector.size
+
+    return quickselect(vector, n / 2) if n.odd?
+    0.5 * (quickselect(vector, n / 2 - 1) + quickselect(vector, n / 2))
+  end
+
+  def quickselect(vector, pos)
+    return vector.first if vector.size == 1 && pos.zero?
+
+    pivot = vector.sample
+
+    lows = vector.select { |i| i < pivot }
+    highs = vector.select { |i| i > pivot }
+    pivots = vector.select { |i| i == pivot }
+
+    return quickselect(lows, pos) if pos < lows.size
+    return pivots.first if pos < pivots.size + lows.size
+
+    quickselect(highs, pos - lows.size - pivots.size)
+  end
+
+  # 1984 Lee DCT implementation
+  #
+  def dct(vector)
+    n = vector.size
+    return vector if n == 1
+    raise StandardError, "Must be nxn" if n.zero? || n.odd?
+
+    half = n / 2
+    alpha = []
+    beta = []
+
+    Array.new(half).each.with_index do |_, i|
+      alpha << (vector[i] + vector[-(i + 1)])
+      beta  << (vector[i] - vector[-(i + 1)]) / (Math.cos((i + 0.5) * Math::PI / n) * 2.0)
+    end
+
+    alpha = dct(alpha)
+    beta = dct(beta)
+    result = []
+
+    Array.new(half - 1).each.with_index do |_, i|
+      result << alpha[i]
+      result << beta[i] + beta[i + 1]
+    end
+
+    result << alpha[-1]
+    result << beta[-1]
+
+    result
+  end
+end

data/ph.gemspec

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+Gem::Specification.new do |s|
+  s.name              = "ph"
+  s.version           = "0.0.1"
+  s.summary           = "Perceptual Hashing"
+  s.authors           = ["elcuervo"]
+  s.licenses          = %w[MIT]
+  s.email             = ["elcuervo@elcuervo.net"]
+  s.homepage          = "http://github.com/elcuervo/ph"
+  s.files             = `git ls-files`.split("\n")
+  s.test_files        = `git ls-files test`.split("\n")
+
+  s.add_development_dependency("vips", "~> 8.10")
+end

data/spec/ph_spec.rb

@@ -0,0 +1,33 @@
+require "spec_helper"
+
+describe PH do
+  let(:file) { "spec/image.jpg" }
+  let(:size) { 64 }
+
+  before do
+    img = Vips::Image.new_from_file(file)
+    width, height = img.width, img.height
+    w_scale, v_scale = size.fdiv(width), size.fdiv(height)
+
+    # Rescale to 64x64 and greyscale
+    img = img.resize(w_scale, vscale: v_scale).colourspace(:grey16)
+    # Ensure a 2D plane
+    @pixels_2d = img.to_a.map(&:flatten)
+  end
+
+  subject { PH.new(@pixels_2d) }
+
+  it "#hash" do
+    assert_equal "859091ce633aaebb", subject.hash
+  end
+
+  it "#vector" do
+    vector = [
+      1, 0, 0, 0, 0, 1, 0, 1, 1, 0, 0, 1, 0, 0, 0, 0, 1, 0, 0, 1, 0, 0, 0, 1, 1, 1, 0, 0, 1, 1, 1,
+      0, 0, 1, 1, 0, 0, 0, 1, 1, 0, 0, 1, 1, 1, 0, 1, 0, 1, 0, 1, 0, 1, 1, 1, 0, 1, 0, 1, 1, 1, 0,
+      1, 1
+    ]
+
+    assert_equal vector, subject.vector
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,3 @@
+require "minitest/autorun"
+require "vips"
+require "ph"

metadata

@@ -0,0 +1,63 @@
+--- !ruby/object:Gem::Specification
+name: ph
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- elcuervo
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2021-04-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: vips
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '8.10'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '8.10'
+description: 
+email:
+- elcuervo@elcuervo.net
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Makefile
+- lib/ph.rb
+- ph.gemspec
+- spec/image.jpg
+- spec/ph_spec.rb
+- spec/spec_helper.rb
+homepage: http://github.com/elcuervo/ph
+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: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: Perceptual Hashing
+test_files: []