@ -1,7 +1,7 @@
#!/usr/bin/env python3
#!/usr/bin/env python3
from ctypes import *
from ctypes import *
from sys import exit , platform
from sys import platform , exit
large_image1_path = " ../test_images/sample_01_large.jpg " . encode ( encoding = " utf-8 " )
large_image1_path = " ../test_images/sample_01_large.jpg " . encode ( encoding = " utf-8 " )
medium_image1_path = " ../test_images/sample_01_medium.jpg " . encode ( encoding = " utf-8 " )
medium_image1_path = " ../test_images/sample_01_medium.jpg " . encode ( encoding = " utf-8 " )
@ -19,31 +19,31 @@ large_image4_path = "../test_images/sample_04_large.jpg".encode(encoding="utf-8"
medium_image4_path = " ../test_images/sample_04_medium.jpg " . encode ( encoding = " utf-8 " )
medium_image4_path = " ../test_images/sample_04_medium.jpg " . encode ( encoding = " utf-8 " )
small_image4_path = " ../test_images/sample_04_small.jpg " . encode ( encoding = " utf-8 " )
small_image4_path = " ../test_images/sample_04_small.jpg " . encode ( encoding = " utf-8 " )
test_images = [
test_images = [
large_image1_path , medium_image1_path , small_image1_path ,
large_image1_path , medium_image1_path , small_image1_path ,
large_image2_path , medium_image2_path , small_image2_path ,
large_image2_path , medium_image2_path , small_image2_path ,
large_image3_path , medium_image3_path , small_image3_path ,
large_image3_path , medium_image3_path , small_image3_path ,
large_image4_path , medium_image4_path , small_image4_path
large_image4_path , medium_image4_path , small_image4_path
]
]
def unsigned64 ( number ) :
def unsigned64 ( number ) :
return ( number & 0xfffffffffffffffff )
return ( number & 0xfffffffffffffffff )
print ( " starting ffi test " )
print ( " starting ffi test " )
# Load the shared library
# Load the shared library
lib = None
lib = None
if platform == ' win32 ' :
if platform == ' win32 ' :
path = os . path . dirname ( __file__ ) + " \\ pihash.dll "
print ( path )
lib = CDLL ( path )
lib = cdll . pihash
elif platform == ' darwin ' :
elif platform == ' darwin ' :
lib = cdll . LoadLibrary ( " libpihash.dylib " )
lib = cdll . LoadLibrary ( " libpihash.dylib " )
elif platform == ' linux ' or platform == ' linux2 ' :
elif platform == ' linux ' or platform == ' linux2 ' :
lib = cdll . LoadLibrary ( " libpihash.so " )
lib = cdll . LoadLibrary ( " libpihash.so " )
else :
else :
print ( ' Unrecognized platform ' , platform )
print ( ' Unrecognized platform ' , platform )
sys . exit ( - 1 )
exit ( - 1 )
class PIHashes ( Structure ) :
class PIHashes ( Structure ) :
@ -70,26 +70,26 @@ lib.ext_free_pihashes.argtypes = [c_void_p]
# Takes a pointer and frees the struct at that memory location
# Takes a pointer and frees the struct at that memory location
lib . ext_free . argtypes = [ c_void_p ]
lib . ext_free . argtypes = [ c_void_p ]
#initialize the library
# initialize the library
lib_struct = lib . ext_init ( " ./.hash_cache " . encode ( ' utf-8 ' ) )
lib_struct = lib . ext_init ( " ./.hash_cache " . encode ( ' utf-8 ' ) )
#print("Pointer to lib_struct: ", lib_struct)
# print("Pointer to lib_struct: ", lib_struct)
# Print the path and the bytes as hex for debugging
# Print the path and the bytes as hex for debugging
#print(large_image_path)
#print('\\x'+'\\x'.join('{:02x}'.format(x) for x in large_image_path))
# print(large_image_path)
# print('\\x'+'\\x'.join('{:02x}'.format(x) for x in large_image_path))
for image in test_images :
for image in test_images :
print ( " Requesting hashes for: %s " % image )
print ( " Requesting hashes for: %s " % image )
phashes = lib . ext_get_pihashes ( lib_struct , image )
phashes = lib . ext_get_pihashes ( lib_struct , image )
pihashes = PIHashes . from_address ( phashes )
pihashes = PIHashes . from_address ( phashes )
lib . ext_free_pihashes ( phashes )
lib . ext_free_pihashes ( phashes )
print ( " ahash: %i " % unsigned64 ( pihashes . ahash ) )
print ( " dhash: %i " % unsigned64 ( pihashes . dhash ) )
print ( " phash: %i " % unsigned64 ( pihashes . phash ) )
# print("ahash: %i"% unsigned64(lib.ext_get_ahash(lib_struct, image)))
# print("dhash: %i"% unsigned64(lib.ext_get_dhash(lib_struct, image)))
# print("phash: %i"% unsigned64(lib.ext_get_phash(lib_struct, image)))
print ( " ahash: %i " % unsigned64 ( pihashes . ahash ) )
print ( " dhash: %i " % unsigned64 ( pihashes . dhash ) )
print ( " phash: %i " % unsigned64 ( pihashes . phash ) )
# print("ahash: %i"% unsigned64(lib.ext_get_ahash(lib_struct, image)))
# print("dhash: %i"% unsigned64(lib.ext_get_dhash(lib_struct, image)))
# print("phash: %i"% unsigned64(lib.ext_get_phash(lib_struct, image)))
# Do cleanup
# Do cleanup
# Makes sure that the heap is cleaned up
# Makes sure that the heap is cleaned up