Add more color channel reduction presets (#6)

* Add more color channel reduction presets * Allow CI failure on nightly
Julia-XAI · Feb 16, 2024 · 9045726 · 9045726
1 parent 3414bf7
commit 9045726
Show file tree

Hide file tree

Showing 19 changed files with 37 additions and 2 deletions.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -20,6 +20,7 @@ jobs:
       matrix:
         version:
           - '1.6'
+          - '1'
           - 'nightly'
         os:
           - ubuntu-latest
@@ -33,7 +34,9 @@ jobs:
           arch: ${{ matrix.arch }}
       - uses: julia-actions/cache@v1
       - uses: julia-actions/julia-buildpkg@v1
+        continue-on-error: ${{ matrix.version == 'nightly' }}
       - uses: julia-actions/julia-runtest@v1
+        continue-on-error: ${{ matrix.version == 'nightly' }}
       - uses: julia-actions/julia-processcoverage@v1
       - uses: codecov/codecov-action@v3
         with:

diff --git a/src/heatmap.jl b/src/heatmap.jl
@@ -17,6 +17,8 @@ Visualize 4D arrays as heatmaps, assuming the WHCN convention for input array di
   - `:sum`: sum up color channels
   - `:norm`: compute 2-norm over the color channels
   - `:maxabs`: compute `maximum(abs, x)` over the color channels
+  - `:sumabs`: compute `sum(abs, x)` over the color channels
+  - `:abssum`: compute `abs(sum(x))` over the color channels
   Defaults to `:$DEFAULT_REDUCE`.
 - `rangescale::Symbol`: Selects how the color channel reduced heatmap is normalized
   before the color scheme is applied. Can be either `:extrema` or `:centered`.
@@ -85,8 +87,14 @@ function reduce_color_channel(val::AbstractArray, method::Symbol)
         return reduce((c...) -> maximum(abs.(c)), val; dims=3, init=init)
     elseif method == :norm
         return reduce((c...) -> sqrt(sum(c .^ 2)), val; dims=3, init=init)
+    elseif method == :sumabs
+        return reduce((c...) -> sum(abs, c), val; dims=3, init=init)
+    elseif method == :abssum
+        return reduce((c...) -> abs(sum(c)), val; dims=3, init=init)
     end
     throw( # else
-        ArgumentError("`reduce` :$method not supported, should be :maxabs, :sum or :norm"),
+        ArgumentError(
+            "`reduce` :$method not supported, should be :maxabs, :sum, :norm, :sumabs, or :abssum",
+        ),
     )
 end
diff --git a/test/references/abssum_centered.txt b/test/references/abssum_centered.txt
@@ -0,0 +1 @@
+[0m[38;5;160;48;5;124m▀[38;5;124;48;5;88m▀[0m
diff --git a/test/references/abssum_centered_2.txt b/test/references/abssum_centered_2.txt
@@ -0,0 +1 @@
+[0m[38;5;124;48;5;88m▀[38;5;124;48;5;88m▀[0m
diff --git a/test/references/abssum_extrema.txt b/test/references/abssum_extrema.txt
@@ -0,0 +1 @@
+[0m[38;5;17;48;5;203m▀[38;5;63;48;5;88m▀[0m
diff --git a/test/references/abssum_extrema_2.txt b/test/references/abssum_extrema_2.txt
@@ -0,0 +1 @@
+[0m[38;5;17;48;5;203m▀[38;5;63;48;5;88m▀[0m
diff --git a/test/references/overlay_abssum_centered.txt b/test/references/overlay_abssum_centered.txt
@@ -0,0 +1 @@
+[0m[38;5;160;48;5;54m▀[38;5;100;48;5;138m▀[0m
diff --git a/test/references/overlay_abssum_extrema.txt b/test/references/overlay_abssum_extrema.txt
@@ -0,0 +1 @@
+[0m[38;5;88;48;5;133m▀[38;5;31;48;5;138m▀[0m
diff --git a/test/references/overlay_rescale_abssum_centered.txt b/test/references/overlay_rescale_abssum_centered.txt
@@ -0,0 +1,3 @@
+[0m[38;5;88;48;5;124m▀[38;5;88;48;5;124m▀[38;5;88;48;5;88m▀[38;5;94;48;5;94m▀[38;5;94;48;5;94m▀[38;5;100;48;5;100m▀[0m
+[0m[38;5;124;48;5;124m▀[38;5;124;48;5;124m▀[38;5;124;48;5;124m▀[38;5;94;48;5;94m▀[38;5;94;48;5;94m▀[38;5;100;48;5;100m▀[0m
+[0m[38;5;124;48;5;124m▀[38;5;124;48;5;124m▀[38;5;124;48;5;124m▀[38;5;130;48;5;130m▀[38;5;130;48;5;130m▀[38;5;136;48;5;136m▀[0m
diff --git a/test/references/overlay_rescale_abssum_extrema.txt b/test/references/overlay_rescale_abssum_extrema.txt
@@ -0,0 +1,3 @@
+[0m[38;5;16;48;5;16m▀[38;5;16;48;5;16m▀[38;5;17;48;5;17m▀[38;5;24;48;5;60m▀[38;5;31;48;5;67m▀[38;5;31;48;5;67m▀[0m
+[0m[38;5;52;48;5;124m▀[38;5;52;48;5;130m▀[38;5;59;48;5;130m▀[38;5;60;48;5;94m▀[38;5;66;48;5;94m▀[38;5;66;48;5;100m▀[0m
+[0m[38;5;167;48;5;203m▀[38;5;167;48;5;203m▀[38;5;166;48;5;166m▀[38;5;130;48;5;130m▀[38;5;130;48;5;130m▀[38;5;136;48;5;136m▀[0m
diff --git a/test/references/overlay_rescale_sumabs_centered.txt b/test/references/overlay_rescale_sumabs_centered.txt
@@ -0,0 +1,3 @@
+[0m[38;5;88;48;5;124m▀[38;5;88;48;5;124m▀[38;5;88;48;5;88m▀[38;5;94;48;5;94m▀[38;5;94;48;5;94m▀[38;5;100;48;5;100m▀[0m
+[0m[38;5;124;48;5;124m▀[38;5;124;48;5;124m▀[38;5;124;48;5;124m▀[38;5;94;48;5;94m▀[38;5;94;48;5;94m▀[38;5;100;48;5;100m▀[0m
+[0m[38;5;124;48;5;124m▀[38;5;124;48;5;124m▀[38;5;124;48;5;124m▀[38;5;130;48;5;130m▀[38;5;130;48;5;130m▀[38;5;136;48;5;136m▀[0m
diff --git a/test/references/overlay_rescale_sumabs_extrema.txt b/test/references/overlay_rescale_sumabs_extrema.txt
@@ -0,0 +1,3 @@
+[0m[38;5;16;48;5;16m▀[38;5;16;48;5;16m▀[38;5;17;48;5;17m▀[38;5;24;48;5;60m▀[38;5;31;48;5;67m▀[38;5;31;48;5;67m▀[0m
+[0m[38;5;52;48;5;124m▀[38;5;52;48;5;130m▀[38;5;59;48;5;130m▀[38;5;60;48;5;94m▀[38;5;66;48;5;94m▀[38;5;66;48;5;100m▀[0m
+[0m[38;5;167;48;5;203m▀[38;5;167;48;5;203m▀[38;5;166;48;5;166m▀[38;5;130;48;5;130m▀[38;5;130;48;5;130m▀[38;5;136;48;5;136m▀[0m
diff --git a/test/references/overlay_sumabs_centered.txt b/test/references/overlay_sumabs_centered.txt
@@ -0,0 +1 @@
+[0m[38;5;160;48;5;54m▀[38;5;100;48;5;138m▀[0m
diff --git a/test/references/overlay_sumabs_extrema.txt b/test/references/overlay_sumabs_extrema.txt
@@ -0,0 +1 @@
+[0m[38;5;88;48;5;133m▀[38;5;31;48;5;138m▀[0m
diff --git a/test/references/sumabs_centered.txt b/test/references/sumabs_centered.txt
@@ -0,0 +1 @@
+[0m[38;5;160;48;5;124m▀[38;5;124;48;5;88m▀[0m
diff --git a/test/references/sumabs_centered_2.txt b/test/references/sumabs_centered_2.txt
@@ -0,0 +1 @@
+[0m[38;5;124;48;5;88m▀[38;5;124;48;5;88m▀[0m
diff --git a/test/references/sumabs_extrema.txt b/test/references/sumabs_extrema.txt
@@ -0,0 +1 @@
+[0m[38;5;17;48;5;203m▀[38;5;63;48;5;88m▀[0m
diff --git a/test/references/sumabs_extrema_2.txt b/test/references/sumabs_extrema_2.txt
@@ -0,0 +1 @@
+[0m[38;5;17;48;5;203m▀[38;5;63;48;5;88m▀[0m
diff --git a/test/test_heatmap.jl b/test/test_heatmap.jl
@@ -9,7 +9,7 @@ batch = reshape(collect(Float32, 1:prod(shape)), shape)
 img = [RGB(1, 0, 0) RGB(0, 1, 0); RGB(0, 0, 1) RGB(1, 1, 1)]
 img2 = [RGB(x, y, 0) for x in 0:0.2:1, y in 0:0.2:1]
 
-reducers = [:sum, :maxabs, :norm]
+reducers = [:sum, :maxabs, :norm, :sumabs, :abssum]
 rangescales = [:extrema, :centered]
 
 @testset "Single input" begin
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		[0m[38;5;160;48;5;124m▀[38;5;124;48;5;88m▀[0m
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		[0m[38;5;124;48;5;88m▀[38;5;124;48;5;88m▀[0m
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		[0m[38;5;17;48;5;203m▀[38;5;63;48;5;88m▀[0m
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		[0m[38;5;160;48;5;54m▀[38;5;100;48;5;138m▀[0m
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		[0m[38;5;88;48;5;133m▀[38;5;31;48;5;138m▀[0m