Allow for ::AbstractVector (e.g. views) in write_solution (#1177)

Author: AbdAlazezAhmed

src/Export/VTK.jl

@@ -175,7 +175,7 @@ function component_names(::Type{S}) where {S}
 end
 
 """
-    write_solution(vtk::VTKGridFile, dh::AbstractDofHandler, u::Vector, suffix="")
+    write_solution(vtk::VTKGridFile, dh::AbstractDofHandler, u::AbstractVector, suffix="")
 
 Save the values at the nodes in the degree of freedom vector `u` to `vtk`.
 Each field in `dh` will be saved separately, and `suffix` can be used to append
@@ -186,7 +186,7 @@ degree of freedom in `dh`, see [`write_node_data`](@ref write_node_data) for det
 Use `write_node_data` directly when exporting values that are already
 sorted by the nodes in the grid.
 """
-function write_solution(vtk::VTKGridFile, dh::AbstractDofHandler, u::Vector, suffix = "")
+function write_solution(vtk::VTKGridFile, dh::AbstractDofHandler, u::AbstractVector, suffix = "")
     fieldnames = getfieldnames(dh)  # all primary fields
     for name in fieldnames
         data = _evaluate_at_grid_nodes(dh, u, name, #=vtk=# Val(true))

test/test_vtk_export.jl

@@ -78,4 +78,29 @@
             end
         end
     end
+    @testset "write_solution view" begin
+        grid = generate_grid(Hexahedron, (5, 5, 5))
+        dofhandler = DofHandler(grid)
+        ip = geometric_interpolation(Hexahedron)
+        add!(dofhandler, :temperature, ip)
+        add!(dofhandler, :displacement, ip^3)
+        close!(dofhandler)
+        u = rand(ndofs(dofhandler))
+        dofhandlerfilename = "dofhandler-no-views"
+        VTKGridFile(dofhandlerfilename, grid) do vtk::VTKGridFile
+            @test write_solution(vtk, dofhandler, u) === vtk
+        end
+        dofhandler_views_filename = "dofhandler-views"
+        VTKGridFile(dofhandler_views_filename, grid) do vtk::VTKGridFile
+            @test write_solution(vtk, dofhandler, (@view u[1:end])) === vtk
+        end
+
+        # test the sha of the file
+        sha = bytes2hex(open(SHA.sha1, dofhandlerfilename * ".vtu"))
+        sha_views = bytes2hex(open(SHA.sha1, dofhandler_views_filename * ".vtu"))
+
+        @test sha == sha_views
+        rm(dofhandlerfilename * ".vtu")
+        rm(dofhandler_views_filename * ".vtu")
+    end
 end