Add heatmap of the zoom window to the plotattributes (#90)

* Add heatmap of the zoom window to the plotattributes

* Set data_limits_ext for first iteration

* Add test for plotting a pyramid and that the colorbar updates

Author: felixcremer

ext/PyramidSchemeMakieExt.jl

@@ -81,10 +81,15 @@ function Makie.plot!(plot::Heatmap{<: Tuple{<: Pyramid}})
         pyramid_data_ext = Extent(X=xval, Y=yval)
         pyramid_ext = extent(pyramid)
 
-        data_limits_ext = Extent(X = extrema(first, datapos), Y = extrema(x -> x[2], datapos))
+        data_limits_ext = if isnothing(cached)
+            Extent(X=first(extent(pyramid, XDim)), Y=first(extent(pyramid, YDim)))
+        else 
+            Extent(X = extrema(first, datapos), Y = extrema(x -> x[2], datapos))
+        end
         pixel_widths = Point2f(abs.(pixelpos[2] .- pixelpos[1]))
 
         datalimit = switchkeys(data_limits_ext, pyramid_ext)
+        
         if intersects(pyramid_data_ext, data_limits_ext)
             # @show data_limits_ext
             return (Ref{DD.AbstractDimMatrix}(miss2nan.(
@@ -94,8 +99,9 @@ function Makie.plot!(plot::Heatmap{<: Tuple{<: Pyramid}})
             return nothing # nothing changed so the downstream computation is not marked dirty
         end
     end
-
-    heatmap!(plot, plot.attributes, plot.__pyramid_data)
+    zoomheat = heatmap!(plot, plot.attributes, plot.__pyramid_data)
+    add_input!(plot.attributes, :__pyramid_heatmap, zoomheat)
+    zoomheat
 end
 
 function Makie.data_limits(p::Heatmap{<: Tuple{<: Pyramid}})

test/runtests.jl

@@ -180,7 +180,22 @@ end
     @test pyrsum.levels[2][10,10] == 17
 end
 
-
+@testitem "Plot of Pyramid" begin
+    using CairoMakie
+    using YAXArrays
+    using DimensionalData.Dimensions
+    a = zeros(1500, 1524)
+    a[1:100, 1:100] .= 2
+    yax = YAXArray((X(1.:size(a,1)),Y(1.:size(a,2))), a)
+    pyramid = Pyramid(yax)
+
+    # test that colorbar updates
+    fig, ax, plt = plot(pyramid)
+    cb = Colorbar(fig[1,2], plt.__pyramid_heatmap[])
+    @test cb.limits[] == [0.0, 2.0]
+    limits!(ax, 200,300, 200, 300)
+    @test cb.limits[] == [-0.5, 0.5]
+end
 #=
 @testitem "Comparing zarr pyramid with tif pyramid" begin
     using PyramidScheme: PyramidScheme as PS