Speedup transform(::Grid) by using inplace map!. (#675)

Ferrite-FEM · Apr 11, 2023 · aa26130 · aa26130
1 parent 247d7c3
commit aa26130
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/src/Grid/grid.jl b/src/Grid/grid.jl
@@ -620,13 +620,9 @@ end
 
 Transform all nodes of the `grid` based on some transformation function `f`.
 """
-function transform!(g::AbstractGrid, f::Function)
- c = similar(g.nodes)
- for i in 1:length(c)
- c[i] = Node(f(g.nodes[i].x))
- end
- copyto!(g.nodes, c)
- g
+function transform!(g::Grid, f::Function)
+ map!(n -> Node(f(getcoordinates(n))), g.nodes, g.nodes)
+ return g
 end
 
 # Sets