Use the numeric type of the input nodes as the output type in grid ge…

…nerators (#973)

src/Grid/grid_generators.jl

@@ -67,9 +67,9 @@ function generate_grid(::Type{QuadraticLine}, nel::NTuple{1,Int}, left::Vec{1,T}
     return Grid(cells, nodes, facetsets=facetsets)
 end
 
-function _generate_2d_nodes!(nodes, nx, ny, LL, LR, UR, UL)
+function _generate_2d_nodes!(nodes::Vector{Node{2, T}}, nx, ny, LL, LR, UR, UL) where T
       for i in 0:ny-1
-        ratio_bounds = i / (ny-1)
+        ratio_bounds = convert(T, i) / (ny-1)
 
         x0 = LL[1] * (1 - ratio_bounds) + ratio_bounds * UL[1]
         x1 = LR[1] * (1 - ratio_bounds) + ratio_bounds * UR[1]
@@ -78,7 +78,7 @@ function _generate_2d_nodes!(nodes, nx, ny, LL, LR, UR, UL)
         y1 = LR[2] * (1 - ratio_bounds) + ratio_bounds * UR[2]
 
         for j in 0:nx-1
-            ratio = j / (nx-1)
+            ratio = convert(T, j) / (nx-1)
             x = x0 * (1 - ratio) + ratio * x1
             y = y0 * (1 - ratio) + ratio * y1
             push!(nodes, Node((x, y)))
@@ -284,9 +284,9 @@ function generate_grid(::Type{Pyramid}, nel::NTuple{3,Int}, left::Vec{3,T}=Vec{3
 
     #Center node in each "voxel"
     for k in 1:nel_z, j in 1:nel_y, i in 1:nel_x
-        midx = 0.5(coords_x[i+1] + coords_x[i])
-        midy = 0.5(coords_y[j+1] + coords_y[j])
-        midz = 0.5(coords_z[k+1] + coords_z[k])
+        midx = (coords_x[i+1] + coords_x[i]) / 2
+        midy = (coords_y[j+1] + coords_y[j]) / 2
+        midz = (coords_z[k+1] + coords_z[k]) / 2
         push!(nodes, Node((midx, midy, midz)))
     end
 

test/runtests.jl

@@ -45,6 +45,7 @@ include("test_constraints.jl")
 include("test_grid_dofhandler_vtk.jl")
 include("test_vtk_export.jl")
 include("test_abstractgrid.jl")
+include("test_grid_generators.jl")
 include("test_grid_addboundaryset.jl")
 include("test_mixeddofhandler.jl")
 include("test_l2_projection.jl")

test/test_grid_generators.jl

@@ -0,0 +1,24 @@
+# Helper function to test grid generation for a given floating-point type
+function test_generate_grid(T::Type)
+    # Define the cell types to test
+    cell_types = [
+        Line, QuadraticLine,
+        Quadrilateral, QuadraticQuadrilateral, Triangle, QuadraticTriangle,
+        Hexahedron, Wedge, Pyramid, Tetrahedron, SerendipityQuadraticHexahedron]
+
+    # Loop over all cell types and test grid generation
+    for CT in cell_types
+        rdim = Ferrite.getrefdim(CT)
+        nels = ntuple(i -> 2, rdim)
+        left = - ones(Vec{rdim,T})
+        right =  ones(Vec{rdim,T})
+        grid = generate_grid(CT, nels, left, right)
+        @test isa(grid, Grid{rdim, CT, T})
+    end
+end
+
+# Run tests for different floating-point types
+@testset "Generate Grid Tests" begin
+    test_generate_grid(Float64)
+    test_generate_grid(Float32)
+end