1 files changed, 41 insertions, 4 deletions

diff --git a/cmake/modules/FindSSE.cmake b/cmake/modules/FindSSE.cmake
index d001b03..b860dca 100644
--- a/cmake/modules/FindSSE.cmake
+++ b/cmake/modules/FindSSE.cmake
@@ -40,6 +40,41 @@ if(CMAKE_SYSTEM_NAME MATCHES "Linux")
      string(COMPARE EQUAL "avx2" "${_SSE_THERE}" _AVX2_TRUE)
      CHECK_CXX_ACCEPTS_FLAG("-mavx2" _AVX2_OK)
    endif()
+elseif(CMAKE_SYSTEM_NAME MATCHES "FreeBSD")
+   if(CPU MATCHES "amd64" OR CPU MATCHES "i.86")
+     exec_program(cat ARGS "/var/run/dmesg.boot | grep Features" OUTPUT_VARIABLE CPUINFO)
+
+     string(REGEX REPLACE "^.*(SSE).*$" "\\1" _SSE_THERE ${CPUINFO})
+     string(COMPARE EQUAL "SSE" "${_SSE_THERE}" _SSE_TRUE)
+     CHECK_CXX_ACCEPTS_FLAG("-msse" _SSE_OK)
+
+     string(REGEX REPLACE "^.*(SSE2).*$" "\\1" _SSE_THERE ${CPUINFO})
+     string(COMPARE EQUAL "SSE2" "${_SSE_THERE}" _SSE2_TRUE)
+     CHECK_CXX_ACCEPTS_FLAG("-msse2" _SSE2_OK)
+
+     string(REGEX REPLACE "^.*(SSE3).*$" "\\1" _SSE_THERE ${CPUINFO})
+     string(COMPARE EQUAL "SSE3" "${_SSE_THERE}" _SSE3_TRUE)
+     CHECK_CXX_ACCEPTS_FLAG("-msse3" _SSE3_OK)
+
+     string(REGEX REPLACE "^.*(SSSE3).*$" "\\1" _SSE_THERE ${CPUINFO})
+     string(COMPARE EQUAL "SSSE3" "${_SSE_THERE}" _SSSE3_TRUE)
+     CHECK_CXX_ACCEPTS_FLAG("-mssse3" _SSSE3_OK)
+
+     string(REGEX REPLACE "^.*(SSE4.1).*$" "\\1" _SSE_THERE ${CPUINFO})
+     string(COMPARE EQUAL "SSE4.1" "${_SSE_THERE}" _SSE41_TRUE)
+     CHECK_CXX_ACCEPTS_FLAG("-msse4.1" _SSE41_OK)
+     string(REGEX REPLACE "^.*(SSE4.2).*$" "\\1" _SSE_THERE ${CPUINFO})
+     string(COMPARE EQUAL "SSE4.2" "${_SSE_THERE}" _SSE42_TRUE)
+     CHECK_CXX_ACCEPTS_FLAG("-msse4.2" _SSE42_OK)
+
+     string(REGEX REPLACE "^.*(AVX).*$" "\\1" _SSE_THERE ${CPUINFO})
+     string(COMPARE EQUAL "AVX" "${_SSE_THERE}" _AVX_TRUE)
+     CHECK_CXX_ACCEPTS_FLAG("-mavx" _AVX_OK)
+
+     string(REGEX REPLACE "^.*(AVX2).*$" "\\1" _SSE_THERE ${CPUINFO})
+     string(COMPARE EQUAL "AVX2" "${_SSE_THERE}" _AVX2_TRUE)
+     CHECK_CXX_ACCEPTS_FLAG("-mavx2" _AVX2_OK)
+   endif()  
 elseif(CMAKE_SYSTEM_NAME MATCHES "Android")
   if(CPU MATCHES "x86_64" OR CPU MATCHES "i.86")
     set(_SSE_TRUE TRUE)
@@ -94,10 +129,12 @@ elseif(CMAKE_SYSTEM_NAME MATCHES "Darwin")
    endif()
 elseif(CMAKE_SYSTEM_NAME MATCHES "Windows")
    # TODO
-   set(_SSE_TRUE true)
-   set(_SSE_OK   true)
-   set(_SSE2_TRUE true)
-   set(_SSE2_OK   true)
+   if(ARCH STREQUAL win32 OR ARCH STREQUAL x64)
+      set(_SSE_TRUE true)
+      set(_SSE_OK   true)
+      set(_SSE2_TRUE true)
+      set(_SSE2_OK   true)
+   endif()
 endif()
 
 include(FindPackageHandleStandardArgs)