similie 0.2.0 → 0.2.1

data/VERSION

@@ -1 +1 @@
-0.2.0
+0.2.1

data/ext/similie.c

@@ -52,20 +52,22 @@ int popcount(uint64_t x) {
     return (x * h01)>>56;           //returns left 8 bits of x + (x<<8) + (x<<16) + (x<<24) + ...
 }
 
+#define DCT_SIZE 32
+
 uint64_t image_phash(IplImage *img) {
     uint64_t phash = 0;
     int x, y;
     double avg;
 
-    IplImage *small = cvCreateImage(cvSize(32, 32), 8, img->nChannels);
-    IplImage *mono  = cvCreateImage(cvSize(32, 32), 8, 1);
+    IplImage *small = cvCreateImage(cvSize(64, 64), 8, img->nChannels);
+    IplImage *mono  = cvCreateImage(cvSize(64, 64), 8, 1);
 
     cvResize(img, small, CV_INTER_CUBIC);
     cvCvtColor(small, mono, CV_RGB2GRAY);
 
-    CvMat *dct = cvCreateMat(8, 8, CV_32FC1);
-    for (x = 0; x < 8; x++) {
-        for (y = 0; y < 8; y++) {
+    CvMat *dct = cvCreateMat(DCT_SIZE, DCT_SIZE, CV_32FC1);
+    for (x = 0; x < DCT_SIZE; x++) {
+        for (y = 0; y < DCT_SIZE; y++) {
             cvSet2D(dct, x, y, cvScalarAll(cvGetMonoPixel(mono, x, y)));
         }
     }

data/test/test_basic.rb

@@ -12,12 +12,12 @@ describe 'Similie image load' do
   it 'should hash image' do
     img = Similie.new(File.join($testdir, 'lena1.png'))
     assert img
-    assert_equal 216455360913932544, img.hash
+    assert_equal 72342380518637824, img.hash
   end
 
   it 'should hash image using class method' do
     hash = Similie.phash(File.join($testdir, 'lena1.png'))
     assert hash
-    assert_equal 216455360913932544, hash
+    assert_equal 72342380518637824, hash
   end
 end

data/test/test_distance.rb

@@ -4,14 +4,14 @@ describe 'Similie image distance' do
   it 'should work for similar images' do
     images = (1..5).map {|n| Similie.new(File.join($testdir, 'lena%d.png' % n ))}
 
-    assert_equal 1, images[0].distance(images[1])
-    assert_equal 2, images[1].distance(images[2])
-    assert_equal 0, images[2].distance(images[3])
-    assert_equal 1, images[0].distance(images[3])
-    assert_equal 6, images[3].distance(images[4])
+    assert_equal 2,  images[0].distance(images[1])
+    assert_equal 3,  images[1].distance(images[2])
+    assert_equal 2,  images[2].distance(images[3])
+    assert_equal 3,  images[0].distance(images[3])
+    assert_equal 18, images[3].distance(images[4])
   end
 
   it 'should work using the singleton method' do
-    assert_equal 6, Similie.distance(File.join($testdir, 'lena4.png'), File.join($testdir, 'lena5.png'))
+    assert_equal 18, Similie.distance(File.join($testdir, 'lena4.png'), File.join($testdir, 'lena5.png'))
   end
 end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 0
   - 2
-  - 0
-  version: 0.2.0
+  - 1
+  version: 0.2.1
 platform: ruby
 authors: 
 - Bharanee Rathna