Pass operator to broken_scalarplot!

Author: pjaap

src/plots.jl

@@ -62,7 +62,7 @@ function plot!(
             end
             if resultdim == 1
                 if !average_broken_plots && ExtendableFEMBase.broken(sol[op[1]].FES)
-                    broken_scalarplot!(p[row, col], sol[op[1]]; title = title * title_add, kwargs...)
+                    broken_scalarplot!(p[row, col], sol[op[1]], op[2]; title = title * title_add, kwargs...)
                 else
                     GridVisualize.scalarplot!(p[row, col], sol[op[1]].FES.dofgrid, view(nodevalues(sol[op[1]], op[2]; abs = false), 1, :), title = title * title_add; kwargs...)
                 end
@@ -90,21 +90,21 @@ end
 
 
 """
-    broken_scalarplot!(vis, feVectorBlock; kwargs...)
+    broken_scalarplot!(vis, feVectorBlock::FEVectorBlock, operator = Identity; kwargs...)
 
 A "broken" scalarplot of a broken finite element vector.
 Instead of averaging the discontinuous values on the grid nodes, each grid cell is plotted
 independently. Thus, a discontinuous plot is generated.
 
 All kwargs of the calling method are transferred to the scalarplot in this method.
 """
-function broken_scalarplot!(vis, feVectorBlock::FEVectorBlock; kwargs...)
+function broken_scalarplot!(vis, feVectorBlock::FEVectorBlock, operator = Identity; kwargs...)
 
     dofgrid = feVectorBlock.FES.dofgrid
     cell_nodes = dofgrid[CellNodes]
     coords = dofgrid[Coordinates]
 
-    all_values = nodevalues(feVectorBlock; cellwise = true) # cellwise evaluation of the FE
+    all_values = nodevalues(feVectorBlock, operator; cellwise = true) # cellwise evaluation of the FE
     all_coords = @views coords[:, cell_nodes[:]]
     all_cells = reshape(1:length(all_values), size(all_values))