operator id(u, c) and direct broken scalarplot! (#66)

* Add id(u, c) high level operator

* Add direct scalarplot! dispatch for broken plots

Now the user can manually plot broken functions into a GridVisualizer

Author: pjaap

CHANGELOG.md

@@ -3,8 +3,9 @@
 ## v1.3.0
 
 ### Added
-  - `plot` for broken FE Spaces are now plotted discontinuously to avoid averaging of values at the grid nodes.
+  - `plot` and `scalarplot!` for broken FE Spaces are now plotted discontinuously to avoid averaging of values at the grid nodes.
      This can be disabled by passing the kwarg `average_broken_plots = true` to the `plot` call.
+  - new `id(u, c)` operator for the component `c` of a FE function `u` wrapping `IdentityComponent{c}` from ExtendableFEMBase
 
 ## v1.2.0 June 4, 2025
 

src/ExtendableFEM.jl

@@ -18,7 +18,7 @@ using ExtendableFEMBase: ExtendableFEMBase, AbstractFiniteElement,
     FaceDofs, Gradient, H1BR, H1BUBBLE, H1CR, H1MINI,
     H1P1, H1P1TEB, H1P2, H1P2B, H1P3, H1Pk, H1Q1, H1Q2,
     HCURLN0, HCURLN1, HDIVBDM1, HDIVBDM2, HDIVRT0,
-    HDIVRT1, HDIVRTk, HDIVRTkENRICH, Hessian, Identity,
+    HDIVRT1, HDIVRTk, HDIVRTkENRICH, Hessian, Identity, IdentityComponent,
     L2P0, L2P1, Laplacian, Length4Operator,
     NeededDerivative4Operator, NormalFlux,
     ParentDofmap4Dofmap, PointEvaluator, QPInfos,

src/plots.jl

@@ -1,9 +1,20 @@
-function GridVisualize.scalarplot!(p, op::Tuple{Unknown, DataType}, sol; abs = false, component = 1, title = String(op[1].identifier), kwargs...)
-    return GridVisualize.scalarplot!(p, sol[op[1]].FES.dofgrid, view(nodevalues(sol[op[1]], op[2]; abs = abs), component, :); title = title, kwargs...)
-end
-function GridVisualize.scalarplot!(p, op::Tuple{Int, DataType}, sol; abs = false, component = 1, title = sol[op[1]].name, kwargs...)
-    return GridVisualize.scalarplot!(p, sol[op[1]].FES.dofgrid, view(nodevalues(sol[op[1]], op[2]; abs = abs), component, :); title = title, kwargs...)
+function GridVisualize.scalarplot!(
+        p,
+        op::Union{Tuple{Unknown, DataType}, Tuple{Int, DataType}},
+        sol;
+        abs = false,
+        component = 1,
+        title = typeof(op) <: Tuple{Unknown, DataType} ? String(op[1].identifier) : sol[op[1]].name,
+        average_broken_plots = false,
+        kwargs...
+    )
+    if !average_broken_plots && ExtendableFEMBase.broken(sol[op[1]].FES)
+        broken_scalarplot!(p, sol[op[1]], op[2]; title, average_broken_plots, kwargs...)
+    else
+        return GridVisualize.scalarplot!(p, sol[op[1]].FES.dofgrid, view(nodevalues(sol[op[1]], op[2]; abs = abs), component, :); title = title, kwargs...)
+    end
 end
+
 function GridVisualize.vectorplot!(p, op::Tuple{Unknown, DataType}, sol; title = String(op[1].identifier), kwargs...)
     return GridVisualize.vectorplot!(p, sol[op[1]].FES.dofgrid, eval_func_bary(PointEvaluator([op], sol)); title = title, kwargs...)
 end

src/unknowns.jl

@@ -101,6 +101,13 @@ alias for (u, Identity)
 """
 id(u) = (u, Identity)
 
+"""
+	id(u, c::Int)
+
+alias for (u, IdentityComponent{c})
+"""
+id(u, c::Int) = (u, IdentityComponent{c})
+
 """
 	curl1(u)