first draft of 2d scalarplot (requires change in ExtendableGrids.interpolate! -> PR there)

Author: chmerdon

ext/GridVisualizeUnicodePlotsExt.jl

@@ -8,7 +8,7 @@ module GridVisualizeUnicodePlotsExt
 import GridVisualize: initialize!, gridplot!, scalarplot!, bregion_cmap, region_cmap
 using GridVisualize: UnicodePlotsType
 using UnicodePlots: UnicodePlots
-using ExtendableGrids: Coordinates, ON_CELLS, ON_FACES, ON_EDGES, CellNodes, FaceNodes, BFaceNodes, CellGeometries, CellRegions, BFaceRegions, num_cells, num_nodes, local_celledgenodes, num_bfaceregions, num_cellregions, num_targets
+using ExtendableGrids: Coordinates, simplexgrid, ON_CELLS, ON_FACES, ON_EDGES, CellNodes, FaceNodes, BFaceNodes, CellGeometries, CellRegions, BFaceRegions, num_cells, num_nodes, local_celledgenodes, num_bfaceregions, num_cellregions, num_targets, interpolate!
 using Colors: Colors, RGB, RGBA
 
 initialize!(p, ::Type{UnicodePlotsType}) = nothing
@@ -291,19 +291,72 @@ function scalarplot!(
     resolution = ctx[:size]
     ylim = ctx[:limits]
 
+    if ylim[1] > ylim[2]
+        # try to find limits automatically
+        ylim = (minimum([minimum(func) for func in funcs]), maximum([maximum(func) for func in funcs]))
+    end
+
     plt = ctx[:clear] ? nothing : ctx[:figure]
     for ifunc in 1:nfuncs
         func = funcs[ifunc]
         grid = grids[ifunc]
         coord = grid[Coordinates] * ctx[:gridscale]
         if ifunc == 1
-            plt = lineplot(coord[1, :], func; ylim, xlabel = "x", name = ctx[:label], height = resolution[2], width = resolution[1])
+            plt = UnicodePlots.lineplot(coord[1, :], func; ylim, xlabel = "x", name = ctx[:label], height = resolution[2], width = resolution[1])
         else
-            lineplot!(plt, coord[1, :], func; name = ctx[:label])
+            UnicodePlots.lineplot!(plt, coord[1, :], func; name = ctx[:label])
         end
     end
 
     return plt
 end
 
+
+function scalarplot!(
+        ctx,
+        ::Type{UnicodePlotsType},
+        ::Type{Val{2}},
+        grids,
+        parentgrid,
+        funcs
+    )
+
+    func = funcs[1]
+    resolution = ctx[:size]
+    ylim = ctx[:limits]
+    colormap = ctx[:colormap]
+
+    if ylim[1] > ylim[2]
+        # try to find limits automatically
+        ylim = (minimum([minimum(func) for func in funcs]), maximum([maximum(func) for func in funcs]))
+    end
+
+    coords = grids[1][Coordinates]
+    ex = extrema(view(coords, 1, :))
+    ey = extrema(view(coords, 2, :))
+
+    X = LinRange(ex[1], ex[2], resolution[1])
+    Y = LinRange(ey[1], ey[2], resolution[2])
+    xgrid_plot = simplexgrid(X, Y)
+
+    # interpolate data onto plot_grid
+    I = zeros(Float64, num_nodes(xgrid_plot))
+    interpolate!(I, xgrid_plot, func, grids[1]; eps = 1.0e-14, not_in_domain_value = NaN, trybrute = true)
+
+    plt = ctx[:clear] ? nothing : ctx[:figure]
+    plt = UnicodePlots.heatmap(
+        reshape(I, (resolution[1], resolution[2]))',
+        xlabel = "x",
+        ylabel = "y",
+        xfact = (ex[2] - ex[1]) / (resolution[1] - 1),
+        yfact = (ey[2] - ey[1]) / (resolution[2] - 1),
+        xoffset = ex[1],
+        yoffset = ey[1],
+        title = ctx[:title],
+        colormap = colormap,
+    )
+
+    return plt
+end
+
 end # module