src/ImFilter.jl

changeset 32
41d13bf7d637
parent 30
0b25d9ef7af9
child 48
1b9e90ca81e3
--- a/src/ImFilter.jl	Fri Dec 27 21:41:52 2019 +0200
+++ b/src/ImFilter.jl	Sat Dec 28 02:10:06 2019 +0200
@@ -8,6 +8,7 @@
 module ImFilter
 
 using OffsetArrays
+using AlgTools.Util: @threadsif
 
 ##########
 # Exports
@@ -25,7 +26,8 @@
 end
 
 function simple_imfilter(b::Array{Float64,2},
-                         kernel::OffsetArray{Float64,2,Array{Float64,2}})
+                         kernel::OffsetArray{Float64,2,Array{Float64,2}};
+                         threads::Bool=true)
 
     n, m = size(b)
     k, 𝓁 = size(kernel)
@@ -38,7 +40,7 @@
 
     res = similar(b)
 
-    Threads.@threads for i=1:n
+    @threadsif threads for i=1:n
         @inbounds for j=1:m
             tmp = 0.0
             it₁ = inside(i, a₁, b₁, 1, n)

mercurial