Add another binning [skip ci]

JuliaMicroscopy · Nov 16, 2021 · 9997d2f · 9997d2f · roflmaostc · Nov 23, 2021
1 parent 63dfbc2
commit 9997d2f
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/examples/benchmarks/bin.jl b/examples/benchmarks/bin.jl
@@ -1,11 +1,25 @@
 using BenchmarkTools, MicroscopyTools
 
+function bin_3(arr::AbstractArray{T, N}, factors=ntuple(x -> 2, Val(N))) where {T, N}
+    # old size
+    sz = size(arr)
+    # new size
+    sn = sz .÷ factors
+    s2 = ntuple(i -> isodd(i) ? factors[(i+1)/2] : sn[i/2], Val(2 * N)) 
+
+    dims = ntuple(i -> 2 * i - 1, Val(N))
+    reshape(sum(reshape(arr,s2),dims=dims),sn)
+end
+
+
 function f(x)
     @info "Random data with size $(size(x))"
     print("Tullio: ")
     @btime bin2($x)
     print("CartesianIndices: ")
     @btime bin($x, (2,2))
+    print("RH: ")
+    @btime bin_3($x, (2,2))
 end
 
 function main()