Merge pull request #252 from gridap/simplexify_for_1d_case

Extended simplexify for SEGMENT and VERTEX
gridap · May 15, 2020 · c8cf93e · c8cf93e
2 parents 5d14172 + 42bd76e
commit c8cf93e
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/src/ReferenceFEs/ExtrusionPolytopes.jl b/src/ReferenceFEs/ExtrusionPolytopes.jl
@@ -217,10 +217,26 @@ function is_n_cube(p::ExtrusionPolytope{0})
   true
 end
 
+function is_simplex(p::ExtrusionPolytope{1})
+  true
+end
+
+function is_n_cube(p::ExtrusionPolytope{1})
+  true
+end
+
 function simplexify(p::ExtrusionPolytope)
   @notimplemented
 end
 
+function simplexify(p::ExtrusionPolytope{0})
+  [[1,],], VERTEX
+end
+
+function simplexify(p::ExtrusionPolytope{1})
+  [[1,2],], SEGMENT
+end
+
 function simplexify(p::ExtrusionPolytope{2})
   if p == QUAD
     ([[1,2,3],[2,3,4]], TRI)

diff --git a/test/ReferenceFEsTests/ExtrusionPolytopesTests.jl b/test/ReferenceFEsTests/ExtrusionPolytopesTests.jl
@@ -123,10 +123,22 @@ r = [[[1,2],[2,1]],[[1,2],[2,1]],[[1,2],[2,1]]]
 v,p = simplexify(QUAD)
 @test p == TRI
 
+v,p = simplexify(TRI)
+@test p == TRI
+
 v,p = simplexify(HEX)
 @test p == TET
 
+v,p = simplexify(TET)
+@test p == TET
+
+v,p = simplexify(SEGMENT)
+@test p == SEGMENT
+@test is_simplex(p)
 
+v,p = simplexify(VERTEX)
+@test p == VERTEX
+@test is_simplex(p)