Added DirichletSpace

Author: fverdugo

src/Arrays/IdentityVectors.jl

@@ -26,3 +26,6 @@ function reindex(values::AbstractArray, indices::IdentityVector)
   values
 end
 
+function reindex(a::AppliedArray,b::IdentityVector)
+  a
+end

src/FESpaces/DirichletFESpaces.jl

@@ -0,0 +1,68 @@
+
+struct DirichletFESpace <: SingleFieldFESpace
+  space::SingleFieldFESpace
+end
+
+function num_free_dofs(f::DirichletFESpace)
+  num_dirichlet_dofs(f.space)
+end
+
+function zero_free_values(::Type{T},f::DirichletFESpace) where T
+  zeros(T,num_free_dofs(f))
+end
+
+function get_cell_dofs(f::DirichletFESpace)
+  apply(elem(-),get_cell_dofs(f.space))
+end
+
+function num_dirichlet_dofs(f::DirichletFESpace)
+  num_free_dofs(f.space)
+end
+
+function zero_dirichlet_values(f::DirichletFESpace)
+  T = Float64
+  zero_free_values(T,f.space)
+end
+
+function num_dirichlet_tags(f::DirichletFESpace)
+  1
+end
+
+function get_dirichlet_dof_tag(f::DirichletFESpace)
+  ones(Int8,num_dirichlet_dofs(f))
+end
+
+function scatter_free_and_dirichlet_values(f::DirichletFESpace,fv,dv)
+  scatter_free_and_dirichlet_values(f.space,dv,fv)
+end
+
+function gather_free_and_dirichlet_values(f::DirichletFESpace,cv)
+  dv, fv = gather_free_and_dirichlet_values(f.space,cv)
+  (fv, dv)
+end
+
+function TrialFESpace(f::DirichletFESpace)
+  U = TrialFESpace(f.space)
+  DirichletFESpace(U)
+end
+
+function get_cell_basis(f::DirichletFESpace)
+  get_cell_basis(f.space)
+end
+
+function get_cell_dof_basis(f::DirichletFESpace)
+  get_cell_dof_basis(f.space)
+end
+
+function apply_constraints_matrix_cols(f::DirichletFESpace,cm,cids)
+  apply_constraints_matrix_cols(f.space,cm,cids)
+end
+
+function apply_constraints_matrix_rows(f::DirichletFESpace,cm,cids)
+  apply_constraints_matrix_rows(f.space,cm,cids)
+end
+
+function apply_constraints_vector(f::DirichletFESpace,cm,cids)
+  apply_constraints_vector(f.space,cm,cids)
+end
+

src/FESpaces/FESpaces.jl

@@ -157,6 +157,7 @@ export FESpaceWithLastDofRemoved
 export ZeroMeanFESpace
 export CLagrangianFESpace
 export DivConformingFESpace
+export DirichletFESpace
 
 export @law
 export operate
@@ -204,6 +205,8 @@ include("ZeroMeanFESpaces.jl")
 
 include("CLagrangianFESpaces.jl")
 
+include("DirichletFESpaces.jl")
+
 include("FESpaceFactories.jl")
 
 end # module

test/FESpacesTests/DirichletFESpacesTests.jl

@@ -0,0 +1,45 @@
+module DirichletFESpacesTests
+
+using Test
+using Gridap.Algebra
+using Gridap.Geometry
+using Gridap.FESpaces
+
+n = 2
+mesh = (n,n)
+domain = (0,1,0,1)
+order = 1
+dirichlet_tags = [1,3,4,6,7]
+model = CartesianDiscreteModel(domain, mesh)
+labels = get_face_labeling(model)
+add_tag_from_tags!(labels,"dirichlet",dirichlet_tags)
+
+Vf = TestFESpace(
+  reffe=:Lagrangian, order=order, valuetype=Float64,
+  conformity=:H1, model=model, labels=labels, dirichlet_tags="dirichlet")
+
+Vd = DirichletFESpace(Vf)
+test_single_field_fe_space(Vd,[],[],[],[])
+
+Uf = TrialFESpace(Vf,1)
+
+Ud = TrialFESpace(Vd)
+test_single_field_fe_space(Ud,[],[],[],[])
+
+trian = Triangulation(model)
+degree = 2*order
+quad = CellQuadrature(trian,degree)
+
+a(v,u) = v*u
+l(v) = v*0
+t_Ω = AffineFETerm(a,l,trian,quad)
+
+op_ff = AffineFEOperator(Vf,Uf,t_Ω)
+op_fd = AffineFEOperator(Vf,Ud,t_Ω)
+op_df = AffineFEOperator(Vd,Uf,t_Ω)
+
+xd = get_dirichlet_values(Uf)
+
+@test maximum(abs.(get_vector(op_ff) + get_matrix(op_fd)*xd)) < 1.0e-10
+
+end # module

test/FESpacesTests/runtests.jl

@@ -44,6 +44,8 @@ using Test
 
 @testset "CLagrangianFESpaces" begin include("CLagrangianFESpacesTests.jl") end
 
+@testset "DirichletFESpaces" begin include("DirichletFESpacesTests.jl") end
+
 @testset "FESpaceFactories" begin include("FESpaceFactoriesTests.jl") end
 
 end # module