void GradGrad<D>::T_CalcElementMatrix (const FiniteElement & base_fel,
const ElementTransformation & eltrans,
FlatMatrix<SCAL> elmat,
LocalHeap & lh) const {
const CompoundFiniteElement & cfel // product space
= dynamic_cast<const CompoundFiniteElement&> (base_fel);
const ScalarFiniteElement<D> & fel_u = // u space
dynamic_cast<const ScalarFiniteElement<D>&> (cfel[GetInd1()]);
const ScalarFiniteElement<D> & fel_e = // e space
dynamic_cast<const ScalarFiniteElement<D>&> (cfel[GetInd2()]);
elmat = SCAL(0.0);
// u dofs [ru.First() : ru.Next()-1], e dofs [re.First() : re.Next()-1]
IntRange ru = cfel.GetRange(GetInd1());
IntRange re = cfel.GetRange(GetInd2());
int ndofe = re.Size();
int ndofu = ru.Size();
FlatMatrixFixWidth<D> dum(ndofu,lh); // to store grad(u-basis)
FlatMatrixFixWidth<D> dem(ndofe,lh); // to store grad(e-basis)
ELEMENT_TYPE eltype // get the type of element:
= fel_u.ElementType(); // ET_TRIG in 2d, ET_TET in 3d.
const IntegrationRule & // Note: p = fel_u.Order()-1
ir = SelectIntegrationRule(eltype, fel_u.Order()+fel_e.Order()-2);
FlatMatrix<SCAL> submat(ndofe,ndofu,lh);
submat = 0.0;
for(int k=0; k<ir.GetNIP(); k++) {
MappedIntegrationPoint<D,D> mip (ir[k],eltrans);
// set grad(u-basis) and grad(e-basis) at mapped pts in dum and dem.
fel_u.CalcMappedDShape( mip, dum );
fel_e.CalcMappedDShape( mip, dem );
// evaluate coefficient
SCAL fac = coeff_a -> T_Evaluate<SCAL>(mip);
fac *= mip.GetWeight() ;
// [ndofe x D] * [D x ndofu]
submat += fac * dem * Trans(dum) ;
}
elmat.Rows(re).Cols(ru) += submat;
if (GetInd1() != GetInd2())
elmat.Rows(ru).Cols(re) += Conj(Trans(submat));
}
void DGFiniteElement<D>::
CalcTraceMatrix (int facet, FlatMatrix<> trace) const
{
ELEMENT_TYPE ftype = ElementTopology::GetFacetType (this->ElementType(), facet);
Facet2ElementTrafo f2el(this->ElementType(), FlatArray<int> (8, const_cast<int*> (vnums)) );
const IntegrationRule & ir = SelectIntegrationRule (ftype, 2*order);
ScalarFiniteElement<0> * facetfe0 = NULL;
ScalarFiniteElement<1> * facetfe1 = NULL;
ScalarFiniteElement<2> * facetfe2 = NULL;
switch (ftype)
{
case ET_POINT : facetfe0 = new FE_Point; break;
case ET_SEGM : facetfe1 = new L2HighOrderFE<ET_SEGM> (order); break;
case ET_TRIG : facetfe2 = new L2HighOrderFE<ET_TRIG> (order); break;
case ET_QUAD : facetfe2 = new L2HighOrderFE<ET_QUAD> (order); break;
default:
;
}
int ndof_facet = trace.Height();
Vector<> shape(ndof);
Vector<> fshape(ndof_facet);
Vector<> norms(ndof_facet);
trace = 0.0;
norms = 0.0;
for (int i = 0; i < ir.Size(); i++)
{
if (D == 1)
facetfe0 -> CalcShape (ir[i], fshape);
else if (D == 2)
facetfe1 -> CalcShape (ir[i], fshape);
else
facetfe2 -> CalcShape (ir[i], fshape);
this -> CalcShape (f2el (facet, ir[i]), shape);
trace += ir[i].Weight() * fshape * Trans (shape);
for (int j = 0; j < norms.Size(); j++)
norms(j) += ir[i].Weight() * sqr (fshape(j));
}
for (int j = 0; j < fshape.Size(); j++)
trace.Row(j) /= norms(j);
delete facetfe0;
delete facetfe1;
delete facetfe2;
}
void EyeEye<D>::T_CalcElementMatrix (const FiniteElement & base_fel,
const ElementTransformation & eltrans,
FlatMatrix<SCAL> elmat,
LocalHeap & lh) const {
const CompoundFiniteElement & cfel // product space
= dynamic_cast<const CompoundFiniteElement&> (base_fel);
const ScalarFiniteElement<D> & fel_u = // u space
dynamic_cast<const ScalarFiniteElement<D>&> (cfel[GetInd1()]);
const ScalarFiniteElement<D> & fel_e = // e space
dynamic_cast<const ScalarFiniteElement<D>&> (cfel[GetInd2()]);
elmat = SCAL(0.0);
IntRange ru = cfel.GetRange(GetInd1());
IntRange re = cfel.GetRange(GetInd2());
int ndofe = re.Size();
int ndofu = ru.Size();
Vector<> ushape(ndofu);
Vector<> eshape(ndofe);
ELEMENT_TYPE eltype = fel_u.ElementType();
const IntegrationRule &
ir = SelectIntegrationRule(eltype, fel_u.Order()+fel_e.Order());
FlatMatrix<SCAL> submat(ndofe,ndofu,lh);
submat = SCAL(0.0);
for(int k=0; k<ir.GetNIP(); k++) {
MappedIntegrationPoint<D,D> mip (ir[k],eltrans);
fel_u.CalcShape( ir[k], ushape );
fel_e.CalcShape( ir[k], eshape );
SCAL fac = (coeff_a -> T_Evaluate<SCAL>(mip))* mip.GetWeight() ;
// [ndofe x D] * [D x ndofu]
submat += fac * eshape * Trans(ushape) ;
}
elmat.Rows(re).Cols(ru) += submat;
if (GetInd1() != GetInd2())
elmat.Rows(ru).Cols(re) += Conj(Trans(submat));
}
void FluxTrace<D>::T_CalcElementMatrix (const FiniteElement & base_fel,
const ElementTransformation & eltrans,
FlatMatrix<SCAL> elmat,
LocalHeap & lh) const {
const CompoundFiniteElement & cfel // product space
= dynamic_cast<const CompoundFiniteElement&> (base_fel);
const HDivFiniteElement<D> & fel_q = // q space
dynamic_cast<const HDivFiniteElement<D>& > (cfel[GetInd1()]);
const ScalarFiniteElement<D> & fel_e = // e space
dynamic_cast<const ScalarFiniteElement<D>&> (cfel[GetInd2()]);
elmat = SCAL(0.0);
IntRange rq = cfel.GetRange(GetInd1());
IntRange re = cfel.GetRange(GetInd2());
int ndofq = rq.Size();
int ndofe = re.Size();
FlatMatrix<SCAL> submat(ndofe,ndofq,lh);
FlatMatrixFixWidth<D> shapeq(ndofq,lh); // q-basis (vec) values
FlatVector<> shapee(ndofe,lh); // e-basis basis
ELEMENT_TYPE eltype // get the type of element:
= fel_q.ElementType(); // ET_TRIG in 2d, ET_TET in 3d.
// transform facet integration points to volume integration points
Facet2ElementTrafo transform(eltype);
int nfa = ElementTopology::GetNFacets(eltype); /* nfa = number of facets
of an element */
submat = 0.0;
for(int k = 0; k<nfa; k++) {
// type of geometry of k-th facet
ELEMENT_TYPE eltype_facet = ElementTopology::GetFacetType(eltype, k);
const IntegrationRule & facet_ir =
SelectIntegrationRule (eltype_facet, fel_q.Order()+fel_e.Order());
// reference element normal vector
FlatVec<D> normal_ref = ElementTopology::GetNormals(eltype) [k];
for (int l = 0; l < facet_ir.GetNIP(); l++) {
// map 1D facet points to volume integration points
IntegrationPoint volume_ip = transform(k, facet_ir[l]);
MappedIntegrationPoint<D,D> mip (volume_ip, eltrans);
// compute normal on physcial element
Mat<D> inv_jac = mip.GetJacobianInverse();
double det = mip.GetJacobiDet();
Vec<D> normal = fabs(det) * Trans(inv_jac) * normal_ref;
double len = L2Norm(normal);
normal /= len;
double weight = facet_ir[l].Weight()*len;
// mapped H(div) basis fn values and DG fn (no need to map) values
fel_q.CalcMappedShape(mip,shapeq);
fel_e.CalcShape(volume_ip,shapee);
// evaluate coefficient
SCAL dd = coeff_d -> T_Evaluate<SCAL>(mip);
// [ndofe x 1] [ndofq x D] * [D x 1]
submat += (dd*weight) * shapee * Trans( shapeq * normal ) ;
}
}
elmat.Rows(re).Cols(rq) += submat;
elmat.Rows(rq).Cols(re) += Conj(Trans(submat));
}
void TraceTrace<D>::T_CalcElementMatrix (const FiniteElement & base_fel,
const ElementTransformation & eltrans,
FlatMatrix<SCAL> elmat,
LocalHeap & lh) const {
const CompoundFiniteElement & cfel // product space
= dynamic_cast<const CompoundFiniteElement&> (base_fel);
const ScalarFiniteElement<D> & fel_u = // u space
dynamic_cast<const ScalarFiniteElement<D>&> (cfel[GetInd1()]);
const ScalarFiniteElement<D> & fel_e = // e space
dynamic_cast<const ScalarFiniteElement<D>&> (cfel[GetInd2()]);
elmat = SCAL(0.0);
IntRange ru = cfel.GetRange(GetInd1());
IntRange re = cfel.GetRange(GetInd2());
int ndofe = re.Size();
int ndofu = ru.Size();
FlatVector<> shapee(ndofe,lh);
FlatVector<> shapeu(ndofu,lh);
FlatMatrix<SCAL> submat(ndofe,ndofu, lh);
submat = SCAL(0.0);
ELEMENT_TYPE eltype = fel_u.ElementType();
Facet2ElementTrafo transform(eltype);
int nfa = ElementTopology :: GetNFacets(eltype);
for(int k = 0; k<nfa; k++) {
// type of geometry of k-th facet
ELEMENT_TYPE eltype_facet = ElementTopology::GetFacetType(eltype, k);
const IntegrationRule & facet_ir =
SelectIntegrationRule (eltype_facet, fel_u.Order()+fel_e.Order());
// reference element normal vector
FlatVec<D> normal_ref = ElementTopology::GetNormals(eltype) [k];
for (int l = 0; l < facet_ir.GetNIP(); l++) {
// map 1D facet points to volume integration points
IntegrationPoint volume_ip = transform(k, facet_ir[l]);
MappedIntegrationPoint<D,D> mip (volume_ip, eltrans);
// compute normal on physcial element
Mat<D> inv_jac = mip.GetJacobianInverse();
double det = mip.GetJacobiDet();
Vec<D> normal = fabs(det) * Trans(inv_jac) * normal_ref;
double len = L2Norm(normal);
normal /= len;
double weight = facet_ir[l].Weight()*len;
fel_e.CalcShape(volume_ip,shapee);
fel_u.CalcShape(volume_ip,shapeu);
SCAL cc = coeff_c -> T_Evaluate<SCAL>(mip);
// [ndofe x 1] [1 x ndofu]
submat += (cc*weight) * shapee * Trans(shapeu);
}
}
elmat.Rows(re).Cols(ru) += submat;
if (GetInd1() != GetInd2())
elmat.Rows(ru).Cols(re) += Conj(Trans(submat));
}
