Merge pull request #551 from gridap/get_cell_points_for_celldof

Implemented get_cell_points for CellDof
gridap · Mar 3, 2021 · 08fea6d · 08fea6d
2 parents ce72a9f + fbd8898
commit 08fea6d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/CellData/CellDofs.jl b/src/CellData/CellDofs.jl
@@ -33,7 +33,10 @@ function change_domain(a::CellDof,target_trian::RestrictedTriangulation,target_d
   CellDof(tcell_dof,target_trian,DomainStyle(a))
 end
 
-
+function get_cell_points(dofs::CellDof)
+  cell_to_x = lazy_map(get_nodes,get_data(dofs))
+  CellPoint(cell_to_x,get_triangulation(dofs),DomainStyle(dofs))
+end
 
 # Evaluation of CellDof
 

diff --git a/test/CellDataTests/CellDofsTests.jl b/test/CellDataTests/CellDofsTests.jl
@@ -30,6 +30,10 @@ a = s(v)
 r = fill(Matrix(I,4,4),num_cells(trian))
 test_array(a,r)
 
+x = get_cell_points(s)
+vx = v(x)
+test_array(vx,r)
+
 acell_to_cell = [1,3,5,3]
 atrian = Triangulation(trian,acell_to_cell)
 s_a = change_domain(s,atrian,DomainStyle(s))
@@ -39,4 +43,8 @@ a = s_a(v_a)
 r = fill(Matrix(I,4,4),num_cells(atrian))
 test_array(a,r)
 
+x_a = get_cell_points(s_a)
+vx_a = v(x_a)
+test_array(vx_a,r)
+
 end # module