21 #ifndef PATCH_POINT_VALUES_HH_
22 #define PATCH_POINT_VALUES_HH_
24 #include <Eigen/Dense>
35 template<
unsigned int spacedim>
class ElOp;
114 template<
unsigned int spacedim = 3>
177 elOp.allocate_result(sizes[elOp.size_type()], *
patch_arena_);
191 for (
uint i_col=0; i_col<coords.n_cols; ++i_col)
192 for (
uint i_row=0; i_row<coords.n_rows; ++i_row) {
193 coords_mat(i_row, i_col)(i_elem) = coords(i_row, i_col);
212 for (
uint i_col=0; i_col<elm_coords.n_cols; ++i_col)
213 for (
uint i_row=0; i_row<elm_coords.n_rows; ++i_row) {
214 coords_mat(i_row, i_col)(i_elem) = elm_coords(i_row, i_col);
222 for (
uint i_col=0; i_col<side_coords.n_cols; ++i_col)
223 for (
uint i_row=0; i_row<side_coords.n_rows; ++i_row) {
224 coords_mat(i_row, i_col)(i_elem) = side_coords(i_row, i_col);
242 uint point_pos = i_point_on_elem *
n_elems_ + elem_table_row;
261 uint point_pos = i_point_on_side *
n_elems_ + elem_table_row;
307 ElOp<spacedim> op_accessor(this->
dim_, shape, reinit_f, size_type, input_ops_vec, n_dofs);
345 const auto &op_matrix =
operations_[op_idx].result_matrix();
347 for (
uint i=0; i<3; ++i)
348 val(i) = op_matrix(i)(op_matrix_idx);
361 const auto &op_matrix =
operations_[op_idx].result_matrix();
363 for (
uint i=0; i<3; ++i)
364 for (
uint j=0; j<3; ++j)
365 val(i,j) = op_matrix(i,j)(op_matrix_idx);
378 stream <<
"Point vals: " << std::endl;
380 const auto &
mat = op.result_matrix();
381 for (
uint i_mat=0; i_mat<
mat.rows()*
mat.cols(); ++i_mat) {
382 if (
mat(i_mat).data_size()==0) stream <<
"<empty>";
384 const double *vals =
mat(i_mat).data_ptr();
385 for (
size_t i_val=0; i_val<
mat(i_mat).data_size(); ++i_val)
386 stream << vals[i_val] <<
" ";
390 stream <<
" --- end of operation ---" << std::endl;
396 if (
int_table_(i_row).data_size()==0) stream <<
"<empty>";
399 for (
size_t i_val=0; i_val<
int_table_(i_row).data_size(); ++i_val)
400 stream << vals[i_val] <<
" ";
417 {
"el_coords",
"jacobian",
"inv_jac",
"jac_det",
"pt_coords",
"weights",
"JxW",
"",
"",
"",
"",
"" },
418 {
"el_coords",
"el_jac",
"el_inv_jac",
"side_coords",
"side_jac",
"side_jac_det",
"exp_el_coords",
"exp_el_jac",
"exp_el_inv_jac",
419 "exp_side_coords",
"exp_side_jac",
"exp_side_jac_det",
"pt_coords",
"weights",
"JxW",
"normal_vec",
"",
"",
"",
"",
"" }
421 stream << std::setfill(
' ') <<
" Operation" << setw(12) <<
"" <<
"Shape" << setw(2) <<
""
422 <<
"n DOFs" << setw(2) <<
"" <<
"Input operations" << endl;
424 stream <<
" " << std::left << setw(20) << op_names[bulk_side][i] <<
"" <<
" " << setw(6) <<
operations_[i].format_shape() <<
"" <<
" "
425 << setw(7) <<
operations_[i].n_dofs() <<
"" <<
" ";
427 for (
auto i_o : input_ops) stream << op_names[bulk_side][i_o] <<
" ";
467 friend class ElOp<spacedim>;
468 template<
unsigned int dim>
470 template<
unsigned int dim>
472 template<
unsigned int dim>
479 template<
unsigned int spacedim = 3>
494 if (shape_vec.size() == 1) shape_vec.push_back(1);
550 result_ = Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic>(
shape_[0],
shape_[1]);
556 Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic> &
result_matrix() {
561 const Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic> &
result_matrix()
const {
569 Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic>
result_;
590 template<
unsigned int dim>
594 auto &jac_value = op.result_matrix();
595 const auto &coords_value = operations[ op.input_ops()[0] ].result_matrix();
596 for (
unsigned int i=0; i<3; i++)
597 for (
unsigned int j=0; j<dim; j++)
598 jac_value(i,j) = coords_value(i,j+1) - coords_value(i,0);
600 template<
unsigned int dim>
604 auto &inv_jac_value = op.result_matrix();
605 const auto &jac_value = operations[ op.input_ops()[0] ].result_matrix();
606 inv_jac_value = eigen_arena_tools::inverse<3, dim>(jac_value);
608 template<
unsigned int dim>
612 auto &jac_det_value = op.result_matrix();
613 const auto &jac_value = operations[ op.input_ops()[0] ].result_matrix();
614 jac_det_value(0,0) = eigen_arena_tools::determinant<3, dim>(jac_value).abs();
623 op.allocate_result(point_weights.size(), *arena);
624 auto &weights_value = op.result_matrix();
625 for (
uint i=0; i<point_weights.size(); ++i)
626 weights_value(0,0)(i) = point_weights[i];
630 auto &weights_value = operations[ op.input_ops()[0] ].result_matrix();
631 auto &jac_det_value = operations[ op.input_ops()[1] ].result_matrix();
635 auto &jxw_value = op.result_matrix();
636 jxw_value(0,0) = jxw_ovec.
get_vec();
640 auto &op = operations[scalar_shape_op_idx];
641 uint n_dofs = shape_values.size();
642 uint n_points = shape_values[0].size();
643 uint n_elem = op.result_matrix()(0).data_size() / n_points;
645 auto &shape_matrix = op.result_matrix();
648 for (
uint i=0; i<n_elem; ++i) {
653 for (
uint i_dof=0; i_dof<n_dofs; ++i_dof)
654 for (
uint i_p=0; i_p<n_points; ++i_p)
655 ref_vec(i_dof * n_points + i_p) = shape_values[i_dof][i_p];
657 shape_matrix(0) = shape_ovec.
get_vec();
659 template<
unsigned int dim>
662 auto &op = operations[scalar_shape_grads_op_idx];
663 auto &inv_jac_vec = operations[ op.input_ops()[0] ].result_matrix();
664 uint n_points = ref_shape_grads.size();
665 uint n_dofs = ref_shape_grads[0].size();
667 Eigen::Matrix<ArenaVec<double>, dim, 1> ref_grads_vec;
668 Eigen::Matrix<ArenaOVec<double>, dim, 1> ref_grads_ovec;
669 for (
uint i=0; i<ref_grads_vec.rows(); ++i) {
670 ref_grads_vec(i) =
ArenaVec<double>(n_points * n_dofs, op.result_matrix()(0).arena());
671 for (
uint i_dof=0; i_dof<n_dofs; ++i_dof)
672 for (
uint i_p=0; i_p<n_points; ++i_p)
673 ref_grads_vec(i)(i_dof * n_points + i_p) = ref_shape_grads[i_dof][i_p](i);
674 ref_grads_ovec(i) =
ArenaOVec(ref_grads_vec(i));
677 Eigen::Matrix<ArenaOVec<double>, dim, 3> inv_jac_ovec;
678 for (
uint i=0; i<dim*3; ++i) {
679 inv_jac_ovec(i) =
ArenaOVec(inv_jac_vec(i));
682 auto &result_vec = op.result_matrix();
683 Eigen::Matrix<ArenaOVec<double>, 3, 1> result_ovec = inv_jac_ovec.transpose() * ref_grads_ovec;
684 for (
uint i=0; i<3; ++i) {
685 result_vec(i) = result_ovec(i).get_vec();
695 template<
unsigned int dim>
699 auto &jac_value = op.result_matrix();
700 const auto &coords_value = operations[ op.input_ops()[0] ].result_matrix();
701 for (
unsigned int i=0; i<3; i++)
702 for (
unsigned int j=0; j<dim; j++)
703 jac_value(i,j) = coords_value(i,j+1) - coords_value(i,0);
705 template<
unsigned int dim>
708 auto &inv_jac_value = op.result_matrix();
709 const auto &jac_value = operations[ op.input_ops()[0] ].result_matrix();
710 inv_jac_value = eigen_arena_tools::inverse<3, dim>(jac_value);
712 template<
unsigned int dim>
716 auto &jac_value = op.result_matrix();
717 const auto &coords_value = operations[ op.input_ops()[0] ].result_matrix();
718 for (
unsigned int i=0; i<3; i++)
719 for (
unsigned int j=0; j<dim-1; j++)
720 jac_value(i,j) = coords_value(i,j+1) - coords_value(i,0);
722 template<
unsigned int dim>
726 auto &jac_det_value = op.result_matrix();
727 const auto &jac_value = operations[ op.input_ops()[0] ].result_matrix();
737 op.allocate_result(point_weights.size(), *arena);
738 auto &weights_value = op.result_matrix();
739 for (
uint i=0; i<point_weights.size(); ++i)
740 weights_value(0,0)(i) = point_weights[i];
744 auto &weights_value = operations[ op.input_ops()[0] ].result_matrix();
745 auto &jac_det_value = operations[ op.input_ops()[1] ].result_matrix();
749 auto &jxw_value = op.result_matrix();
750 jxw_value(0,0) = jxw_ovec.
get_vec();
752 template<
unsigned int dim>
755 auto &normal_value = op.result_matrix();
756 auto &inv_jac_mat_value = operations[ op.input_ops()[0] ].result_matrix();
759 ArenaVec<double> norm_vec( normal_value(0).data_size(), normal_value(0).arena() );
760 Eigen::VectorXd A(3);
761 for (
uint i=0; i<normal_value(0).data_size(); ++i) {
762 A(0) = normal_value(0)(i);
763 A(1) = normal_value(1)(i);
764 A(2) = normal_value(2)(i);
765 norm_vec(i) = A.norm();
768 size_t points_per_side = el_table(4).
data_size() / el_table(3).data_size();
769 size_t n_points = el_table(3).data_size();
770 for (
uint i=0; i<3; ++i) {
771 normal_value(i) = normal_value(i) / norm_vec;
772 ArenaVec<double> expand_vec( normal_value(i).data_size() * points_per_side, normal_value(i).arena() );
774 expand_vec(j) = normal_value(i)(j % n_points);
776 normal_value(i) = expand_vec;
781 uint n_dofs = shape_values[0][0].size();
782 uint n_sides = el_table(3).data_size();
783 uint n_patch_points = el_table(4).data_size();
785 auto &op = operations[scalar_shape_op_idx];
786 auto &scalar_shape_value = op.result_matrix();
787 scalar_shape_value(0) =
ArenaVec<double>(n_dofs*n_patch_points, scalar_shape_value(0).arena());
789 for (
uint i_dof=0; i_dof<n_dofs; ++i_dof) {
790 uint dof_shift = i_dof * n_patch_points;
791 for (
uint i_pt=0; i_pt<n_patch_points; ++i_pt)
792 scalar_shape_value(0)(i_pt + dof_shift) = shape_values[el_table(4)(i_pt)][i_pt / n_sides][i_dof];
795 template<
unsigned int dim>
798 uint n_points = ref_shape_grads[0].size();
799 uint n_dofs = ref_shape_grads[0][0].size();
800 uint n_sides = el_table(3).data_size();
801 uint n_patch_points = el_table(4).data_size();
804 auto &op = operations[scalar_shape_grads_op_idx];
805 auto &grad_scalar_shape_value = op.result_matrix();
808 auto &inv_jac_value = operations[ op.input_ops()[0] ].result_matrix();
809 Eigen::Matrix<ArenaVec<double>, dim, 3> inv_jac_expd_value;
810 for (
uint i=0; i<dim*3; ++i) {
811 inv_jac_expd_value(i) =
ArenaVec<double>( n_dofs*n_patch_points, inv_jac_value(i).arena() );
812 for (
uint j=0; j<n_dofs*n_patch_points; ++j)
813 inv_jac_expd_value(i)(j) = inv_jac_value(i)(j%n_sides);
817 Eigen::Matrix<ArenaVec<double>, dim, 1> ref_shape_grads_expd;
818 for (
uint i=0; i<dim; ++i) {
819 ref_shape_grads_expd(i) =
ArenaVec<double>( n_dofs*n_patch_points, inv_jac_value(0).arena() );
821 for (
uint i_dof=0; i_dof<n_dofs; ++i_dof) {
822 for (
uint i_pt=0; i_pt<n_points; ++i_pt) {
823 uint i_begin = (i_dof * n_points + i_pt) * n_sides;
824 for (
uint i_sd=0; i_sd<n_sides; ++i_sd) {
825 for (
uint i_c=0; i_c<dim; ++i_c) {
826 ref_shape_grads_expd(i_c)(i_begin + i_sd) = ref_shape_grads[el_table(3)(i_sd)][i_pt][i_dof][i_c];
833 grad_scalar_shape_value = inv_jac_expd_value.transpose() * ref_shape_grads_expd;
846 auto &result_vec = op.result_matrix();
847 for (
uint i=0;i<result_vec(0,0).data_size(); ++i) {
848 result_vec(0,0)(i) = 1.0;
857 template<
unsigned int spacedim = 3>
880 template<
unsigned int dim>
912 template<
unsigned int spacedim = 3>
935 template<
unsigned int dim>
#define ASSERT_EQ(a, b)
Definition of comparative assert macro (EQual) only for debug mode.
ArenaVec< T > get_vec() const
Convert ArenaOVec to ArenaVec and its.
size_t data_size() const
Getter for data_size_.
Class represents element or FE operations.
ElOp(uint dim, std::initializer_list< uint > shape, ReinitFunction reinit_f, OpSizeType size_type, std::vector< uint > input_ops={}, uint n_dofs=1)
Eigen::Matrix< ArenaVec< double >, Eigen::Dynamic, Eigen::Dynamic > & result_matrix()
Return map referenced result as Eigen::Matrix.
void allocate_result(size_t data_size, PatchArena &arena)
std::string format_shape() const
std::vector< uint > input_ops_
Indices of operations in PatchPointValues::operations_ vector on which ElOp is depended.
std::vector< uint > set_shape_vec(std::initializer_list< uint > shape) const
Aligns shape_vec to 2 items (equal to matrix number of dimensions)
Eigen::Matrix< ArenaVec< double >, Eigen::Dynamic, Eigen::Dynamic > result_
Result matrix of operation.
std::vector< uint > shape_
Shape of stored data (size of vector or number of rows and cols of matrix)
const std::vector< uint > & shape() const
Getter for shape_.
const std::vector< uint > & input_ops() const
Getter for input_ops_.
OpSizeType size_type_
Type of operation by size of vector (element, point or fixed size)
const Eigen::Matrix< ArenaVec< double >, Eigen::Dynamic, Eigen::Dynamic > & result_matrix() const
Same as previous but return const reference.
uint n_dofs() const
Getter for n_dofs_.
ReinitFunction reinit_func
Pointer to patch reinit function of element data table specialized by operation.
void reinit_function(std::vector< ElOp< spacedim >> &operations, IntTableArena &int_table)
Call reinit function on element table if function is defined.
uint n_dofs_
Number of DOFs of FE operations (or 1 in case of element operations)
uint dim() const
Getter for dimension.
OpSizeType size_type() const
Getter for size_type_.
Bulk data specialization, order of item in operations_ vector corresponds to the BulkOps enum.
PatchPointValues(uint dim, uint quad_order, AssemblyArena &asm_arena)
Constructor.
void init()
Initialize operations vector.
Bulk Side specialization, order of item in operations_ vector corresponds to the SideOps enum.
PatchPointValues(uint dim, uint quad_order, AssemblyArena &asm_arena)
Constructor.
void init()
Initialize operations vector.
void print_data_tables(ostream &stream, bool points, bool ints) const
PatchPointValues(uint dim, AssemblyArena &asm_arena)
AssemblyArena & asm_arena_
Reference to global assembly arena of PatchFeValues.
std::vector< OpSizeType > row_sizes_
hold sizes of rows by type of operation
std::vector< uint > points_map_
Map of point patch indices to ElOp::result_ and int_table_ tables.
uint register_bulk_point(uint elem_table_row, uint value_patch_idx, uint elem_idx, uint i_point_on_elem)
uint register_element(arma::mat coords, uint element_patch_idx)
void resize_tables(uint n_elems, uint n_points)
Resize data tables. Method is called before reinit of patch.
ElOp< spacedim > & make_fe_op(std::initializer_list< uint > shape, ReinitFunction reinit_f, std::vector< uint > input_ops_vec, uint n_dofs, OpSizeType size_type=pointOp)
std::vector< ElOp< spacedim > > operations_
Vector of all defined operations.
Quadrature * get_quadrature() const
Getter for quadrature.
uint n_elems() const
Getter for n_elems_.
uint i_elem_
Index of registered element in table, helper value used during patch creating.
void reset()
Reset number of columns (points and elements)
uint n_points() const
Getter for n_points_.
ElOp< spacedim > & make_fixed_op(std::initializer_list< uint > shape, ReinitFunction reinit_f)
std::vector< uint > elements_map_
Map of element patch indices to ElOp::result_ and int_table_ tables.
uint dim() const
Getter for dim_.
Tensor tensor_value(uint op_idx, uint point_idx, uint i_dof=0) const
Scalar scalar_value(uint op_idx, uint point_idx, uint i_dof=0) const
void init_finalize(PatchArena *patch_arena)
uint register_side(arma::mat elm_coords, arma::mat side_coords, uint side_idx)
uint register_side_point(uint elem_table_row, uint value_patch_idx, uint elem_idx, uint side_idx, uint i_point_on_side)
uint n_points_
Number of points in patch.
void print_operations(ostream &stream, uint bulk_side) const
Quadrature * quad_
Quadrature of given dimension and order passed in constructor.
ElOp< spacedim > & make_new_op(std::initializer_list< uint > shape, ReinitFunction reinit_f, std::vector< uint > input_ops_vec, OpSizeType size_type=pointOp)
std::vector< OpSizeType > int_sizes_
Set size and type of rows of int_table_, value is set implicitly in constructor of descendants.
Vector vector_value(uint op_idx, uint point_idx, uint i_dof=0) const
PatchArena * patch_arena_
Pointer to global patch arena of PatchFeValues.
uint n_elems_
Number of elements in patch.
Symmetric Gauss-Legendre quadrature formulae on simplices.
Base class for quadrature rules on simplices in arbitrary dimensions.
const std::vector< double > & get_weights() const
Return a reference to the whole array of weights.
static Eigen::Matrix< ArenaVec< double >, dim, 1 > normal_vector_array(ArenaVec< uint > loc_side_idx_vec)
Class ElementValues calculates data related to transformation of reference cell to actual cell (Jacob...
ArmaMat< double, N, M > mat
@ opCoords
operations evaluated on quadrature points
@ opInvJac
inverse Jacobian
@ opWeights
weight of quadrature point
@ opJac
Jacobian of element.
@ opElCoords
operations evaluated on elements
@ opJxW
JxW value of quadrature point.
@ opNormalVec
normal vector of quadrature point
@ opSideCoords
operations evaluated on sides
@ opJxW
JxW value of quadrature point.
@ opCoords
operations evaluated on quadrature points
@ opElCoords
operations evaluated on elements
@ opElJac
Jacobian of element.
@ opWeights
weight of quadrature point
@ opSideJac
Jacobian of element.
std::function< void(std::vector< ElOp< 3 > > &, IntTableArena &)> ReinitFunction
Type for conciseness.
OpSizeType
Distinguishes operations by type and size of output rows.
@ pointOp
operation is evaluated on quadrature points
@ elemOp
operation is evaluated on elements or sides
@ fixedSizeOp
operation has fixed size and it is filled during initialization
Definitions of particular quadrature rules on simplices.
Defines reinit operations on bulk points.
static void elop_inv_jac(std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table)
static void ptop_scalar_shape_grads(std::vector< ElOp< 3 >> &operations, std::vector< std::vector< arma::mat > > ref_shape_grads, uint scalar_shape_grads_op_idx)
static void elop_jac(std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table)
static void elop_jac_det(std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table)
static void ptop_JxW(std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table)
static void ptop_weights(std::vector< ElOp< 3 >> &operations, PatchArena *arena, const std::vector< double > &point_weights)
static void ptop_scalar_shape(std::vector< ElOp< 3 >> &operations, std::vector< std::vector< double > > shape_values, uint scalar_shape_op_idx)
static void ptop_coords(FMT_UNUSED std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table)
Defines common functionality of reinit operations.
static void op_base(FMT_UNUSED std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table)
Defines reinit operations on side points.
static void elop_el_jac(std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table)
static void elop_sd_jac_det(std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table)
static void elop_sd_jac(std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table)
static void ptop_normal_vec(std::vector< ElOp< 3 >> &operations, IntTableArena &el_table)
static void ptop_weights(std::vector< ElOp< 3 >> &operations, PatchArena *arena, const std::vector< double > &point_weights)
static void ptop_scalar_shape_grads(std::vector< ElOp< 3 >> &operations, IntTableArena &el_table, std::vector< std::vector< std::vector< arma::mat > > > ref_shape_grads, uint scalar_shape_grads_op_idx)
static void ptop_scalar_shape(std::vector< ElOp< 3 >> &operations, IntTableArena &el_table, std::vector< std::vector< std::vector< double > > > shape_values, uint scalar_shape_op_idx)
static void elop_el_inv_jac(std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table)
static void ptop_JxW(std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table)
static void ptop_coords(FMT_UNUSED std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table)