21 #ifndef PATCH_FE_VALUES_HH_
22 #define PATCH_FE_VALUES_HH_
50 template <
class ValueType>
70 template <
class ValueType>
95 template <
class ValueType>
114 unsigned int begin,
unsigned int begin_side,
unsigned int n_dofs_bulk,
unsigned int n_dofs_side,
unsigned int join_idx)
174 template<
unsigned int dim>
183 template<
unsigned int FE_dim>
186 if (fe_sys !=
nullptr) {
187 return fe_sys->
fe()[component_idx];
189 ASSERT_EQ(component_idx, 0).warning(
"Non-zero component_idx can only be used for FESystem.");
201 template<
unsigned int FE_dim>
205 arma::mat shape_values(fe->n_dofs(), fe->n_components());
206 for (
unsigned int i=0; i<q->
size(); i++)
208 for (
unsigned int j=0; j<fe->n_dofs(); j++)
210 for (
unsigned int c=0; c<fe->n_components(); c++)
211 shape_values(j,c) = fe->shape_value(j, q->
point<FE_dim>(i), c);
213 ref_shape_vals[i][j] = trans(shape_values.row(j));
217 return ref_shape_vals;
228 template<
unsigned int FE_dim>
232 arma::mat shape_values(fe->n_dofs(), fe->n_components());
234 for (
unsigned int sid=0; sid<FE_dim+1; sid++) {
236 for (
unsigned int i=0; i<quad.size(); i++)
238 for (
unsigned int j=0; j<fe->n_dofs(); j++)
240 for (
unsigned int c=0; c<fe->n_components(); c++) {
241 shape_values(j,c) = fe->shape_value(j, quad.template point<FE_dim>(i), c);
244 ref_shape_vals[sid][i][j] = trans(shape_values.row(j));
249 return ref_shape_vals;
254 template<
unsigned int dim>
302 auto fe_component = this->
fe_comp(
fe_, component_idx);
303 ASSERT_EQ(fe_component->fe_type(),
FEType::FEScalar).error(
"Type of FiniteElement of scalar_shape accessor must be FEScalar!\n");
309 for (
unsigned int j = 0; j < fe_component->n_dofs(); j++) {
310 shape_values[i][j] = ref_shape_vals[i][j][0];
317 uint begin = scalar_shape_bulk_op.result_row();
336 auto fe_component = this->
fe_comp(
fe_, component_idx);
337 ASSERT_EQ(fe_component->fe_type(),
FEType::FEScalar).error(
"Type of FiniteElement of grad_scalar_shape accessor must be FEScalar!\n");
343 bulk_reinit::ptop_scalar_shape_grads<dim>(operations, op_results, ref_shape_grads, scalar_shape_grads_op_idx);
346 uint begin = grad_scalar_shape_bulk_op.result_row();
367 for (
unsigned int i_pt=0; i_pt<q->
size(); i_pt++)
369 for (
unsigned int i_dof=0; i_dof<fe->n_dofs(); i_dof++)
372 for (
unsigned int c=0; c<fe->n_components(); c++)
373 grad.col(c) += fe->shape_grad(i_dof, q->
point<dim>(i_pt), c);
375 ref_shape_grads[i_pt][i_dof] = grad;
379 return ref_shape_grads;
383 std::shared_ptr< FiniteElement<dim> >
fe_;
387 template<
unsigned int dim>
433 auto fe_component = this->
fe_comp(
fe_, component_idx);
434 ASSERT_EQ(fe_component->fe_type(),
FEType::FEScalar).error(
"Type of FiniteElement of scalar_shape accessor must be FEScalar!\n");
442 for (
unsigned int s=0; s<dim+1; ++s)
444 for (
unsigned int j = 0; j < fe_component->n_dofs(); j++) {
445 shape_values[s][i][j] = ref_shape_vals[s][i][j][0];
452 uint begin = scalar_shape_bulk_op.result_row();
460 auto fe_component = this->
fe_comp(
fe_, component_idx);
461 ASSERT_EQ(fe_component->fe_type(),
FEType::FEScalar).error(
"Type of FiniteElement of grad_scalar_shape accessor must be FEScalar!\n");
467 side_reinit::ptop_scalar_shape_grads<dim>(operations, op_results, el_table, ref_shape_grads, scalar_shape_grads_op_idx);
470 uint begin = grad_scalar_shape_side_op.result_row();
489 for (
unsigned int sid=0; sid<dim+1; sid++) {
491 for (
unsigned int i_pt=0; i_pt<quad.size(); i_pt++)
493 for (
unsigned int i_dof=0; i_dof<fe->n_dofs(); i_dof++)
496 for (
unsigned int c=0; c<fe->n_components(); c++)
497 grad.col(c) += fe->shape_grad(i_dof, quad.template point<dim>(i_pt), c);
499 ref_shape_grads[sid][i_pt][i_dof] = grad;
504 return ref_shape_grads;
508 std::shared_ptr< FiniteElement<dim> >
fe_;
512 template<
unsigned int dim>
529 ASSERT_EQ(fe_component_low->fe_type(),
FEType::FEScalar).error(
"Type of FiniteElement of scalar_shape accessor must be FEScalar!\n");
534 for (
unsigned int j = 0; j < fe_component_low->n_dofs(); j++) {
535 shape_values_bulk[i][j] = ref_shape_vals_bulk[i][j][0];
542 uint begin_bulk = grad_scalar_shape_bulk_op.result_row();
546 ASSERT_EQ(fe_component_high->fe_type(),
FEType::FEScalar).error(
"Type of FiniteElement of scalar_shape accessor must be FEScalar!\n");
553 for (
unsigned int s=0; s<dim+1; ++s)
555 for (
unsigned int j = 0; j < fe_component_high->n_dofs(); j++) {
556 shape_values_side[s][i][j] = ref_shape_vals_side[s][i][j][0];
563 uint begin_side = grad_scalar_shape_side_op.result_row();
566 begin_bulk, begin_side, fe_component_low->n_dofs(), fe_component_high->n_dofs(), 0) );
567 unsigned int end_idx = fe_component_low->n_dofs() + fe_component_high->n_dofs();
569 begin_bulk, begin_side, fe_component_low->n_dofs(), fe_component_high->n_dofs(), end_idx) );
598 template<
unsigned int spacedim = 3>
605 used_quads_[0] =
false; used_quads_[1] =
false;
616 used_quads_[0] =
false; used_quads_[1] =
false;
626 if (is_bulk)
return patch_point_vals_bulk_[dim-1].get_quadrature();
627 else return patch_point_vals_side_[dim-1].get_quadrature();
637 template<
unsigned int DIM>
640 if ( _quadrature.
dim() == DIM ) {
641 used_quads_[0] =
true;
642 patch_point_vals_bulk_[DIM-1].initialize(3);
644 used_quads_[1] =
true;
645 patch_point_vals_side_[DIM-1].initialize(4);
652 for (
unsigned int i=0; i<3; ++i) {
653 if (used_quads_[0]) patch_point_vals_bulk_[i].reset();
654 if (used_quads_[1]) patch_point_vals_side_[i].reset();
661 for (
unsigned int i=0; i<3; ++i) {
662 if (used_quads_[0]) patch_point_vals_bulk_[i].reinit_patch();
663 if (used_quads_[1]) patch_point_vals_side_[i].reinit_patch();
670 template<
unsigned int dim>
672 ASSERT((dim>=0) && (dim<=3))(dim).error(
"Dimension must be 0, 1, 2 or 3.");
677 template<
unsigned int dim>
679 ASSERT((dim>0) && (dim<=3))(dim).error(
"Dimension must be 1, 2 or 3.");
684 template<
unsigned int dim>
686 ASSERT((dim>0) && (dim<=3))(dim).error(
"Dimension must be 1, 2 or 3.");
691 template<
unsigned int dim>
694 return JoinValues<dim>(&patch_point_vals_bulk_[dim-2], &patch_point_vals_side_[dim-1], fe_);
704 for (
uint i=0; i<3; ++i) {
705 patch_point_vals_bulk_[i].resize_tables(dim_sizes[0][i]);
706 patch_point_vals_side_[i].resize_tables(dim_sizes[1][i]);
713 switch (cell.
dim()) {
716 return patch_point_vals_bulk_[0].register_element(coords, element_patch_idx);
720 return patch_point_vals_bulk_[1].register_element(coords, element_patch_idx);
724 return patch_point_vals_bulk_[2].register_element(coords, element_patch_idx);
736 for (
unsigned int n=0; n<cell_side.
dim(); n++)
737 for (
unsigned int c=0; c<spacedim; c++)
738 side_coords(c,n) = (*cell_side.
side().
node(n))[c];
742 switch (cell.
dim()) {
745 return patch_point_vals_side_[0].register_side(elm_coords, side_coords);
749 return patch_point_vals_side_[1].register_side(elm_coords, side_coords);
753 return patch_point_vals_side_[2].register_side(elm_coords, side_coords);
764 return patch_point_vals_bulk_[cell.
dim()-1].register_bulk_point(elem_table_row, value_patch_idx, cell.
elm_idx());
769 return patch_point_vals_side_[cell_side.
dim()-1].register_side_point(elem_table_row, value_patch_idx, cell_side.
elem_idx(), cell_side.
side_idx());
774 stream << endl <<
"Table of patch FE data:" << endl;
775 for (
uint i=0; i<3; ++i) {
776 stream << std::setfill(
'-') << setw(100) <<
"" << endl;
777 stream <<
"Bulk, dimension " << (i+1) << endl;
778 patch_point_vals_bulk_[i].print_data_tables(stream, points, ints);
781 for (
uint i=0; i<3; ++i) {
782 stream << std::setfill(
'-') << setw(100) <<
"" << endl;
783 stream <<
"Side, dimension " << (i+1) << endl;
784 patch_point_vals_side_[i].print_data_tables(stream, points, ints);
786 stream << std::setfill(
'=') << setw(100) <<
"" << endl;
791 stream << endl <<
"Table of patch FE operations:" << endl;
792 for (
uint i=0; i<3; ++i) {
793 stream << std::setfill(
'-') << setw(100) <<
"" << endl;
794 stream <<
"Bulk, dimension " << (i+1) <<
", n_rows " << patch_point_vals_bulk_[i].n_rows() << endl;
795 patch_point_vals_bulk_[i].print_operations(stream, 0);
797 for (
uint i=0; i<3; ++i) {
798 stream << std::setfill(
'-') << setw(100) <<
"" << endl;
799 stream <<
"Side, dimension " << (i+1) <<
", n_rows " << patch_point_vals_side_[i].n_rows() << endl;
800 patch_point_vals_side_[i].print_operations(stream, 1);
802 stream << std::setfill(
'=') << setw(100) <<
"" << endl;
812 template <
class ValueType>
814 template <
class ValueType>
816 template <
class ValueType>
821 template <
class ValueType>
824 return patch_point_vals_.scalar_val(
begin_, value_cache_idx);
830 return patch_point_vals_.vector_val(
begin_, value_cache_idx);
836 return patch_point_vals_.tensor_val(
begin_, value_cache_idx);
839 template <
class ValueType>
842 return patch_point_vals_.scalar_val(
begin_, value_cache_idx);
848 return patch_point_vals_.vector_val(
begin_, value_cache_idx);
854 return patch_point_vals_.tensor_val(
begin_, value_cache_idx);
857 template <
class ValueType>
860 return patch_point_vals_.scalar_val(
begin_+shape_idx, value_cache_idx);
866 return patch_point_vals_.vector_val(
begin_+3*shape_idx, value_cache_idx);
871 Tensor tens; tens.zeros();
875 template <
class ValueType>
878 return patch_point_vals_.scalar_val(
begin_+shape_idx, value_cache_idx);
884 return patch_point_vals_.vector_val(
begin_+3*shape_idx, value_cache_idx);
889 Tensor tens; tens.zeros();
894 template <
class ValueType>
906 Vector vect; vect.zeros();
912 Tensor tens; tens.zeros();
916 template <
class ValueType>
928 Vector vect; vect.zeros();
934 Tensor tens; tens.zeros();
#define ASSERT_EQ(a, b)
Definition of comparative assert macro (EQual) only for debug mode.
std::vector< std::vector< std::vector< arma::vec > > > ref_shape_values_side(Quadrature *q, std::shared_ptr< FiniteElement< FE_dim >> fe)
Precomputed values of basis functions at the side quadrature points.
std::shared_ptr< FiniteElement< FE_dim > > fe_comp(std::shared_ptr< FiniteElement< FE_dim > > fe, uint component_idx)
Return FiniteElement of component_idx for FESystem or fe for other types.
std::vector< std::vector< arma::vec > > ref_shape_values_bulk(Quadrature *q, std::shared_ptr< FiniteElement< FE_dim >> fe)
Precomputed values of basis functions at the bulk quadrature points.
Base point accessor class.
const ElementCacheMap * elm_cache_map() const
unsigned int elem_patch_idx() const
unsigned int eval_point_idx() const
Return index in EvalPoints object.
ElQ< Scalar > JxW()
Register the product of Jacobian determinant and the quadrature weight at bulk quadrature points.
PatchPointValues< 3 > & patch_point_vals_
ElQ< Vector > coords()
Create bulk accessor of coords entity.
std::vector< std::vector< arma::mat > > ref_shape_gradients(std::shared_ptr< FiniteElement< dim >> fe)
Precomputed gradients of basis functions at the quadrature points.
ElQ< Scalar > determinant()
Create bulk accessor of jac determinant entity.
std::shared_ptr< FiniteElement< dim > > fe_
FeQ< Vector > grad_scalar_shape(uint component_idx=0)
Return the value of the function_no-th gradient shape function at the p bulk quadrature point.
FeQ< Scalar > scalar_shape(uint component_idx=0)
Return the value of the function_no-th shape function at the p bulk quadrature point.
BulkValues(PatchPointValues< 3 > &patch_point_vals, MixedPtr< FiniteElement > fe)
Constructor.
Cell accessor allow iterate over DOF handler cells.
unsigned int dim() const
Return dimension of element appropriate to cell.
unsigned int elm_idx() const
Return serial idx to element of loc_ele_idx_.
ElementAccessor< 3 > elm() const
Return ElementAccessor to element of loc_ele_idx_.
Side accessor allows to iterate over sides of DOF handler cell.
unsigned int elem_idx() const
Side side() const
Return Side of given cell and side_idx.
const DHCellAccessor & cell() const
Return DHCellAccessor appropriate to the side.
unsigned int dim() const
Return dimension of element appropriate to the side.
unsigned int side_idx() const
ElQ()=delete
Forbidden default constructor.
PatchPointValues< 3 > & patch_point_vals_
Reference to PatchPointValues.
ValueType operator()(FMT_UNUSED const SidePoint &point)
ValueType operator()(FMT_UNUSED const BulkPoint &point)
ElQ(PatchPointValues< 3 > &patch_point_vals, unsigned int begin)
Constructor.
int element_eval_point(unsigned int i_elem_in_cache, unsigned int i_eval_point) const
Compound finite element on dim dimensional simplex.
const std::vector< std::shared_ptr< FiniteElement< dim > > > & fe() const
Bulk data specialization, order of item in operations_ vector corresponds to the BulkOps enum.
unsigned int begin_
Index of the first component of the Quantity. Size is given by ValueType.
PatchPointValues< 3 > & patch_point_vals_
Reference to PatchPointValues.
unsigned int n_dofs_
Number of DOFs.
ValueType operator()(FMT_UNUSED unsigned int shape_idx, FMT_UNUSED const BulkPoint &point)
FeQ(PatchPointValues< 3 > &patch_point_vals, unsigned int begin, unsigned int n_dofs)
FeQ()=delete
Forbidden default constructor.
ValueType operator()(FMT_UNUSED unsigned int shape_idx, FMT_UNUSED const SidePoint &point)
Bulk Side specialization, order of item in operations_ vector corresponds to the SideOps enum.
Abstract class for the description of a general finite element on a reference simplex in dim dimensio...
unsigned int n_dofs_both() const
unsigned int begin_side_
Index of the first component of the side Quantity. Size is given by ValueType.
unsigned int local_idx() const
Return local index of DOF (on low / high-dim) - should be private method.
bool operator==(const JoinShapeAccessor< ValueType > &other) const
Comparison of accessors.
unsigned int begin_
Index of the first component of the bulk Quantity. Size is given by ValueType.
PatchPointValues< 3 > * patch_point_vals_bulk_
Pointer to bulk PatchPointValues.
JoinShapeAccessor(PatchPointValues< 3 > *patch_point_vals_bulk, PatchPointValues< 3 > *patch_point_vals_side, unsigned int begin, unsigned int begin_side, unsigned int n_dofs_bulk, unsigned int n_dofs_side, unsigned int join_idx)
unsigned int n_dofs_high() const
JoinShapeAccessor()
Default constructor.
void inc()
Iterates to next item.
unsigned int join_idx_
Index of processed DOF.
unsigned int n_dofs_high_
Number of DOFs on high-dim element.
unsigned int join_idx() const
Return global index of DOF.
unsigned int n_dofs_low() const
ValueType operator()(const BulkPoint &point)
unsigned int n_dofs_low_
Number of DOFs on low-dim element.
PatchPointValues< 3 > * patch_point_vals_side_
Pointer to side PatchPointValues.
JoinValues(FMT_UNUSED PatchPointValues< 3 > *patch_point_vals_bulk, FMT_UNUSED PatchPointValues< 3 > *patch_point_vals_side, FMT_UNUSED MixedPtr< FiniteElement > fe)
Constructor.
Range< JoinShapeAccessor< Scalar > > scalar_join_shape(FMT_UNUSED uint component_idx=0)
std::shared_ptr< FiniteElement< dim > > fe_high_dim_
JoinValues(PatchPointValues< 3 > *patch_point_vals_bulk, PatchPointValues< 3 > *patch_point_vals_side, MixedPtr< FiniteElement > fe)
Constructor.
PatchPointValues< 3 > * patch_point_vals_bulk_
Range< JoinShapeAccessor< Scalar > > scalar_join_shape(uint component_idx=0)
std::shared_ptr< FiniteElement< dim-1 > > fe_low_dim_
PatchPointValues< 3 > * patch_point_vals_side_
static ElementMap element_map(ElementAccessor< 3 > elm)
uint register_element(DHCellAccessor cell, uint element_patch_idx)
Register element to patch_point_vals_ table by dimension of element.
~PatchFEValues()
Destructor.
std::array< FeBulk::PatchPointValues< spacedim >, 3 > patch_point_vals_bulk_
Sub objects of bulk data of dimensions 1,2,3.
uint register_bulk_point(DHCellAccessor cell, uint elem_table_row, uint value_patch_idx)
Register bulk point to patch_point_vals_ table by dimension of element.
uint register_side(DHCellSide cell_side)
Register side to patch_point_vals_ table by dimension of side.
void print_data_tables(ostream &stream, bool points, bool ints, bool only_bulk=true) const
Temporary development method.
JoinValues< dim > join_values()
Return JoinValue object of dimension given by template parameter.
Quadrature * get_quadrature(uint dim, bool is_bulk) const
Return bulk or side quadrature of given dimension.
BulkValues< dim > bulk_values()
Return BulkValue object of dimension given by template parameter.
void initialize(Quadrature &_quadrature)
Initialize structures and calculates cell-independent data.
void print_operations(ostream &stream) const
Temporary development method.
void resize_tables(std::vector< std::vector< uint > > dim_sizes)
Resize tables of patch_point_vals_.
std::array< FeSide::PatchPointValues< spacedim >, 3 > patch_point_vals_side_
Sub objects of side data of dimensions 1,2,3.
MixedPtr< FiniteElement > fe_
Mixed of shared pointers of FiniteElement object.
SideValues< dim > side_values()
Return SideValue object of dimension given by template parameter.
PatchFEValues(unsigned int quad_order, MixedPtr< FiniteElement > fe)
void reinit_patch()
Reinit data.
unsigned int n_dofs() const
Returns the number of shape functions.
uint register_side_point(DHCellSide cell_side, uint elem_table_row, uint value_patch_idx)
Register side point to patch_point_vals_ table by dimension of side.
void reset()
Reset PatchpointValues structures.
std::vector< ElOp< spacedim > > operations_
Vector of all defined operations.
Quadrature * get_quadrature() const
Return quadrature.
uint dim() const
Getter of dim_.
ElOp< spacedim > & make_fe_op(std::initializer_list< uint > shape, ReinitFunction reinit_f, std::vector< uint > input_ops_vec, uint n_dofs)
Add accessor to operations_ vector.
Scalar scalar_val(uint result_row, uint point_idx) const
Base class for quadrature rules on simplices in arbitrary dimensions.
Quadrature make_from_side(unsigned int sid) const
unsigned int size() const
Returns number of quadrature points.
Armor::ArmaVec< double, point_dim > point(unsigned int i) const
Returns the ith quadrature point.
General point a+ side_begin_ + ccessor allow iterate over quadrature points of given side defined in ...
unsigned int eval_point_idx() const
Return index in EvalPoints object.
SideValues(PatchPointValues< 3 > &patch_point_vals, MixedPtr< FiniteElement > fe)
Constructor.
ElQ< Vector > normal_vector()
Register the normal vector to a side at side quadrature points.
std::shared_ptr< FiniteElement< dim > > fe_
ElQ< Scalar > JxW()
Same as BulkValues::JxW but register at side quadrature points.
ElQ< Vector > coords()
Create side accessor of coords entity.
FeQ< Scalar > scalar_shape(uint component_idx=0)
Same as BulkValues::scalar_shape but register at side quadrature points.
FeQ< Vector > grad_scalar_shape(uint component_idx=0)
Same as BulkValues::grad_scalar_shape but register at side quadrature points.
std::vector< std::vector< std::vector< arma::mat > > > ref_shape_gradients(std::shared_ptr< FiniteElement< dim >> fe)
Precomputed gradients of basis functions at the quadrature points.
PatchPointValues< 3 > & patch_point_vals_
ElQ< Scalar > determinant()
Create bulk accessor of jac determinant entity.
NodeAccessor< 3 > node(unsigned int i) const
Returns node for given local index i on the side.
Class ElementValues calculates data related to transformation of reference cell to actual cell (Jacob...
Class FESystem for compound finite elements.
Class FEValues calculates finite element data on the actual cells such as shape function values,...
Iter< Object > make_iter(Object obj)
Class MappingP1 implements the affine transformation of the unit cell onto the actual cell.
ArmaMat< double, N, M > mat
@ opCoords
operations evaluated on quadrature points
@ opInvJac
inverse Jacobian
@ opJxW
JxW value of quadrature point.
@ opNormalVec
normal vector of quadrature point
std::vector< std::array< uint, 3 > > DimPointTable
Holds triplet (dim; bulk/side; idx of point in subtable)
Store finite element data on the actual patch such as shape function values, gradients,...
Definitions of particular quadrature rules on simplices.
Class RefElement defines numbering of vertices, sides, calculation of normal vectors etc.
static void ptop_scalar_shape(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, std::vector< std::vector< double > > shape_values, uint scalar_shape_op_idx)
static void ptop_scalar_shape(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, TableInt &el_table, std::vector< std::vector< std::vector< double > > > shape_values, uint scalar_shape_op_idx)
Enum type UpdateFlags indicates which quantities are to be recomputed on each finite element cell.