Add 'convert' for 'GeometrySet' (#774)

JuliaGeometry · Feb 19, 2024 · 7e61398 · 7e61398
1 parent 6a50bfc
commit 7e61398
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 10 deletions.
diff --git a/src/domains.jl b/src/domains.jl
@@ -142,3 +142,15 @@ end
 include("sets.jl")
 include("mesh.jl")
 include("trajectories.jl")
+
+# ------------
+# CONVERSIONS
+# ------------
+
+Base.convert(::Type{GeometrySet}, d::Domain) = GeometrySet(collect(d))
+
+Base.convert(::Type{SimpleMesh}, m::Mesh) = SimpleMesh(vertices(m), topology(m))
+
+Base.convert(::Type{StructuredGrid}, g::Grid) = StructuredGrid(XYZ(g))
+
+Base.convert(::Type{RectilinearGrid}, g::CartesianGrid) = RectilinearGrid(xyz(g))
diff --git a/src/mesh.jl b/src/mesh.jl
@@ -222,13 +222,3 @@ include("mesh/rectilineargrid.jl")
 include("mesh/structuredgrid.jl")
 include("mesh/simplemesh.jl")
 include("mesh/transformedmesh.jl")
-
-# ------------
-# CONVERSIONS
-# ------------
-
-Base.convert(::Type{<:SimpleMesh}, m::Mesh) = SimpleMesh(vertices(m), topology(m))
-
-Base.convert(::Type{<:StructuredGrid}, g::Grid) = StructuredGrid(XYZ(g))
-
-Base.convert(::Type{<:RectilinearGrid}, g::CartesianGrid) = RectilinearGrid(xyz(g))
diff --git a/test/sets.jl b/test/sets.jl
@@ -22,6 +22,13 @@
     push!(geoms, Segment(P2(1, 1), P2(0, 0)))
     gset = GeometrySet(geoms)
     @test eltype(gset) <: Segment{2,T}
+
+    # conversion
+    grid = CartesianGrid{T}(10, 10)
+    gset = convert(GeometrySet, grid)
+    @test gset isa GeometrySet
+    @test nelements(gset) == 100
+    @test eltype(gset) <: Quadrangle
   end
 
   @testset "PointSet" begin