Generalize VTK.jl

Author: KnutAM

src/Export/VTK.jl

@@ -223,9 +223,17 @@ end
 Project `vals` to the grid nodes with `proj` and save to `vtk`.
 """
 function write_projection(vtk::VTKGridFile, proj::L2Projector, vals, name)
-    data = _evaluate_at_grid_nodes(proj, vals, #=vtk=# Val(true))::Matrix
-    @assert size(data, 2) == getnnodes(get_grid(proj.dh))
-    _vtk_write_node_data(vtk.vtk, data, name; component_names = component_names(eltype(vals)))
+    if write_discontinuous(vtk)
+        # @assert first(vals) isa Number
+        data = evaluate_at_discontinuous_vtkgrid_nodes(proj.dh, vals, only(getfieldnames(proj.dh)), vtk.cellnodes)
+        comp_names = ["x", "y", "z"][1:size(data, 1)]
+    else
+        data = _evaluate_at_grid_nodes(proj, vals, #=vtk=# Val(true))::Matrix
+        @assert size(data, 2) == getnnodes(get_grid(proj.dh))
+        comp_names = component_names(eltype(vals))
+    end
+
+    _vtk_write_node_data(vtk.vtk, data, name; component_names = comp_names)
     return vtk
 end