Update build

bench/benchmark.py

@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 import timeit
 import sys
 
@@ -6,9 +6,11 @@ IMAGE = sys.argv[1]
 COUNT = 20
 SIZE = 8
 
+
 def print_result(method, time):
     print("%s_%s,%d" % (IMAGE, method, time / COUNT * 1000000000))
 
+
 print_result("phash", timeit.timeit(
     setup="from imagehash import phash \n"
           "from PIL import Image",
@@ -36,4 +38,3 @@ print_result("ahash", timeit.timeit(
     stmt="average_hash(Image.open('%s'), hash_size=%d)" % (IMAGE, SIZE),
     number=COUNT
 ))
-

bench/run.py

@@ -15,7 +15,7 @@ files = (
 )
 
 for f in files:
-    out = check_output(["python", "benchmark.py", f])
+    out = check_output(["python3", "benchmark.py", f])
     print(out.decode(), end="")
 
 print()

cmake/FindFFTW.cmake

@@ -85,6 +85,14 @@ if( FFTW_ROOT )
             NO_DEFAULT_PATH
     )
 
+    find_library(
+            FFTW_DOUBLE_MPI_LIB
+            NAMES "fftw3_mpi"
+            PATHS ${FFTW_ROOT}
+            PATH_SUFFIXES "lib" "lib64"
+            NO_DEFAULT_PATH
+    )
+
     find_library(
             FFTW_FLOAT_LIB
             NAMES "fftw3f" libfftw3f-3
@@ -109,6 +117,14 @@ if( FFTW_ROOT )
             NO_DEFAULT_PATH
     )
 
+    find_library(
+            FFTW_FLOAT_MPI_LIB
+            NAMES "fftw3f_mpi"
+            PATHS ${FFTW_ROOT}
+            PATH_SUFFIXES "lib" "lib64"
+            NO_DEFAULT_PATH
+    )
+
     find_library(
             FFTW_LONGDOUBLE_LIB
             NAMES "fftw3l" libfftw3l-3
@@ -133,6 +149,14 @@ if( FFTW_ROOT )
             NO_DEFAULT_PATH
     )
 
+    find_library(
+            FFTW_LONGDOUBLE_MPI_LIB
+            NAMES "fftw3l_mpi"
+            PATHS ${FFTW_ROOT}
+            PATH_SUFFIXES "lib" "lib64"
+            NO_DEFAULT_PATH
+    )
+
     #find includes
     find_path(FFTW_INCLUDE_DIRS
             NAMES "fftw3.h"
@@ -161,6 +185,12 @@ else()
             PATHS ${PKG_FFTW_LIBRARY_DIRS} ${LIB_INSTALL_DIR}
     )
 
+    find_library(
+            FFTW_DOUBLE_MPI_LIB
+            NAMES "fftw3_mpi"
+            PATHS ${PKG_FFTW_LIBRARY_DIRS} ${LIB_INSTALL_DIR}
+    )
+
     find_library(
             FFTW_FLOAT_LIB
             NAMES "fftw3f"
@@ -179,6 +209,12 @@ else()
             PATHS ${PKG_FFTW_LIBRARY_DIRS} ${LIB_INSTALL_DIR}
     )
 
+    find_library(
+            FFTW_FLOAT_MPI_LIB
+            NAMES "fftw3f_mpi"
+            PATHS ${PKG_FFTW_LIBRARY_DIRS} ${LIB_INSTALL_DIR}
+    )
+
     find_library(
             FFTW_LONGDOUBLE_LIB
             NAMES "fftw3l"
@@ -196,6 +232,11 @@ else()
             PATHS ${PKG_FFTW_LIBRARY_DIRS} ${LIB_INSTALL_DIR}
             )
 
+    find_library(FFTW_LONGDOUBLE_MPI_LIB
+            NAMES "fftw3l_mpi"
+            PATHS ${PKG_FFTW_LIBRARY_DIRS} ${LIB_INSTALL_DIR}
+            )
+
     find_path(FFTW_INCLUDE_DIRS
             NAMES "fftw3.h"
             PATHS ${PKG_FFTW_INCLUDE_DIRS} ${INCLUDE_INSTALL_DIR}
@@ -247,7 +288,7 @@ if (FFTW_DOUBLE_THREADS_LIB)
     add_library(FFTW::DoubleThreads INTERFACE IMPORTED)
     set_target_properties(FFTW::DoubleThreads
             PROPERTIES INTERFACE_INCLUDE_DIRECTORIES "${FFTW_INCLUDE_DIRS}"
-               INTERFACE_LINK_LIBRARIES "${FFTW_DOUBLETHREADS_LIB}"
+            INTERFACE_LINK_LIBRARIES "${FFTW_DOUBLE_THREADS_LIB}"
             )
 else ()
     set(FFTW_DOUBLE_THREADS_LIB_FOUND FALSE)
@@ -313,6 +354,42 @@ else()
     set(FFTW_LONGDOUBLE_OPENMP_LIB_FOUND FALSE)
 endif ()
 
+if (FFTW_DOUBLE_MPI_LIB)
+    set(FFTW_DOUBLE_MPI_LIB_FOUND TRUE)
+    set(FFTW_LIBRARIES ${FFTW_LIBRARIES} ${FFTW_DOUBLE_MPI_LIB})
+    add_library(FFTW::DoubleMPI INTERFACE IMPORTED)
+    set_target_properties(FFTW::DoubleMPI
+            PROPERTIES INTERFACE_INCLUDE_DIRECTORIES "${FFTW_INCLUDE_DIRS}"
+            INTERFACE_LINK_LIBRARIES "${FFTW_DOUBLE_MPI_LIB}"
+            )
+else ()
+    set(FFTW_DOUBLE_MPI_LIB_FOUND FALSE)
+endif ()
+
+if (FFTW_FLOAT_MPI_LIB)
+    set(FFTW_FLOAT_MPI_LIB_FOUND TRUE)
+    set(FFTW_LIBRARIES ${FFTW_LIBRARIES} ${FFTW_FLOAT_MPI_LIB})
+    add_library(FFTW::FloatMPI INTERFACE IMPORTED)
+    set_target_properties(FFTW::FloatMPI
+            PROPERTIES INTERFACE_INCLUDE_DIRECTORIES "${FFTW_INCLUDE_DIRS}"
+            INTERFACE_LINK_LIBRARIES "${FFTW_FLOAT_MPI_LIB}"
+            )
+else ()
+    set(FFTW_FLOAT_MPI_LIB_FOUND FALSE)
+endif ()
+
+if (FFTW_LONGDOUBLE_MPI_LIB)
+    set(FFTW_LONGDOUBLE_MPI_LIB_FOUND TRUE)
+    set(FFTW_LIBRARIES ${FFTW_LIBRARIES} ${FFTW_LONGDOUBLE_MPI_LIB})
+    add_library(FFTW::LongDoubleMPI INTERFACE IMPORTED)
+    set_target_properties(FFTW::LongDoubleMPI
+            PROPERTIES INTERFACE_INCLUDE_DIRECTORIES "${FFTW_INCLUDE_DIRS}"
+            INTERFACE_LINK_LIBRARIES "${FFTW_LONGDOUBLE_MPI_LIB}"
+            )
+else ()
+    set(FFTW_LONGDOUBLE_MPI_LIB_FOUND FALSE)
+endif ()
+
 #--------------------------------------- end components
 
 set(CMAKE_FIND_LIBRARY_SUFFIXES ${CMAKE_FIND_LIBRARY_SUFFIXES_SAV})
@@ -336,4 +413,7 @@ mark_as_advanced(
         FFTW_FLOAT_OPENMP_LIB
         FFTW_DOUBLE_OPENMP_LIB
         FFTW_LONGDOUBLE_OPENMP_LIB
+        FFTW_FLOAT_MPI_LIB
+        FFTW_DOUBLE_MPI_LIB
+        FFTW_LONGDOUBLE_MPI_LIB
 )

thirdparty/benchmark

@@ -1 +1 @@
-Subproject commit 0811f1d782455b3c80285bebf934a7045d845ed3
+Subproject commit 604f6fd3f4b34a84ec4eb4db81d842fa4db829cd