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_);
197 for (
uint i_col=0; i_col<coords.n_cols; ++i_col)
198 for (
uint i_row=0; i_row<coords.n_rows; ++i_row) {
199 coords_mat(i_row, i_col)(i_elem) = coords(i_row, i_col);
218 for (
uint i_col=0; i_col<elm_coords.n_cols; ++i_col)
219 for (
uint i_row=0; i_row<elm_coords.n_rows; ++i_row) {
220 coords_mat(i_row, i_col)(i_elem) = elm_coords(i_row, i_col);
228 for (
uint i_col=0; i_col<side_coords.n_cols; ++i_col)
229 for (
uint i_row=0; i_row<side_coords.n_rows; ++i_row) {
230 coords_mat(i_row, i_col)(i_elem) = side_coords(i_row, i_col);
248 uint point_pos = i_point_on_elem *
n_elems_ + elem_table_row;
267 uint point_pos = i_point_on_side *
n_elems_ + elem_table_row;
313 ElOp<spacedim> op_accessor(this->
dim_, shape, reinit_f, size_type, input_ops_vec, n_dofs);
351 const auto &op_matrix =
operations_[op_idx].result_matrix();
353 for (
uint i=0; i<3; ++i)
354 val(i) = op_matrix(i)(op_matrix_idx);
367 const auto &op_matrix =
operations_[op_idx].result_matrix();
369 for (
uint i=0; i<3; ++i)
370 for (
uint j=0; j<3; ++j)
371 val(i,j) = op_matrix(i,j)(op_matrix_idx);
384 stream <<
"Point vals: " << std::endl;
386 const auto &
mat = op.result_matrix();
387 for (
uint i_mat=0; i_mat<
mat.rows()*
mat.cols(); ++i_mat) {
388 if (
mat(i_mat).data_size()==0) stream <<
"<empty>";
390 const double *vals =
mat(i_mat).data_ptr();
391 for (
size_t i_val=0; i_val<
mat(i_mat).data_size(); ++i_val)
392 stream << vals[i_val] <<
" ";
396 stream <<
" --- end of operation ---" << std::endl;
402 if (
int_table_(i_row).data_size()==0) stream <<
"<empty>";
405 for (
size_t i_val=0; i_val<
int_table_(i_row).data_size(); ++i_val)
406 stream << vals[i_val] <<
" ";
423 {
"el_coords",
"jacobian",
"inv_jac",
"jac_det",
"pt_coords",
"weights",
"JxW",
"",
"",
"",
"",
"" },
424 {
"el_coords",
"el_jac",
"el_inv_jac",
"side_coords",
"side_jac",
"side_jac_det",
"exp_el_coords",
"exp_el_jac",
"exp_el_inv_jac",
425 "exp_side_coords",
"exp_side_jac",
"exp_side_jac_det",
"pt_coords",
"weights",
"JxW",
"normal_vec",
"",
"",
"",
"",
"" }
427 stream << std::setfill(
' ') <<
" Operation" << setw(12) <<
"" <<
"Shape" << setw(2) <<
""
428 <<
"n DOFs" << setw(2) <<
"" <<
"Input operations" << endl;
430 stream <<
" " << std::left << setw(20) << op_names[bulk_side][i] <<
"" <<
" " << setw(6) <<
operations_[i].format_shape() <<
"" <<
" "
431 << setw(7) <<
operations_[i].n_dofs() <<
"" <<
" ";
433 for (
auto i_o : input_ops) stream << op_names[bulk_side][i_o] <<
" ";
473 friend class ElOp<spacedim>;
474 template<
unsigned int dim>
476 template<
unsigned int dim>
478 template<
unsigned int dim>
485 template<
unsigned int spacedim = 3>
500 if (shape_vec.size() == 1) shape_vec.push_back(1);
556 result_ = Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic>(
shape_[0],
shape_[1]);
562 Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic> &
result_matrix() {
567 const Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic> &
result_matrix()
const {
575 Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic>
result_;
596 template<
unsigned int dim>
600 auto &jac_value = op.result_matrix();
601 const auto &coords_value = operations[ op.input_ops()[0] ].result_matrix();
602 for (
unsigned int i=0; i<3; i++)
603 for (
unsigned int j=0; j<dim; j++)
604 jac_value(i,j) = coords_value(i,j+1) - coords_value(i,0);
606 template<
unsigned int dim>
610 auto &inv_jac_value = op.result_matrix();
611 const auto &jac_value = operations[ op.input_ops()[0] ].result_matrix();
612 inv_jac_value = eigen_arena_tools::inverse<3, dim>(jac_value);
614 template<
unsigned int dim>
618 auto &jac_det_value = op.result_matrix();
619 const auto &jac_value = operations[ op.input_ops()[0] ].result_matrix();
620 jac_det_value(0,0) = eigen_arena_tools::determinant<3, dim>(jac_value).abs();
629 op.allocate_result(point_weights.size(), *arena);
630 auto &weights_value = op.result_matrix();
631 for (
uint i=0; i<point_weights.size(); ++i)
632 weights_value(0,0)(i) = point_weights[i];
636 auto &weights_value = operations[ op.input_ops()[0] ].result_matrix();
637 auto &jac_det_value = operations[ op.input_ops()[1] ].result_matrix();
641 auto &jxw_value = op.result_matrix();
642 jxw_value(0,0) = jxw_ovec.
get_vec();
646 auto &op = operations[scalar_shape_op_idx];
647 uint n_dofs = shape_values.size();
648 uint n_points = shape_values[0].size();
649 uint n_elem = op.result_matrix()(0).data_size() / n_points;
651 auto &shape_matrix = op.result_matrix();
654 for (
uint i=0; i<n_elem; ++i) {
659 for (
uint i_dof=0; i_dof<n_dofs; ++i_dof)
660 for (
uint i_p=0; i_p<n_points; ++i_p)
661 ref_vec(i_dof * n_points + i_p) = shape_values[i_dof][i_p];
663 shape_matrix(0) = shape_ovec.
get_vec();
665 template<
unsigned int dim>
668 auto &op = operations[scalar_shape_grads_op_idx];
669 auto &inv_jac_vec = operations[ op.input_ops()[0] ].result_matrix();
670 uint n_points = ref_shape_grads.size();
671 uint n_dofs = ref_shape_grads[0].size();
673 Eigen::Matrix<ArenaVec<double>, dim, 1> ref_grads_vec;
674 Eigen::Matrix<ArenaOVec<double>, dim, 1> ref_grads_ovec;
675 for (
uint i=0; i<ref_grads_vec.rows(); ++i) {
676 ref_grads_vec(i) =
ArenaVec<double>(n_points * n_dofs, op.result_matrix()(0).arena());
677 for (
uint i_dof=0; i_dof<n_dofs; ++i_dof)
678 for (
uint i_p=0; i_p<n_points; ++i_p)
679 ref_grads_vec(i)(i_dof * n_points + i_p) = ref_shape_grads[i_p][i_dof](i);
680 ref_grads_ovec(i) =
ArenaOVec(ref_grads_vec(i));
683 Eigen::Matrix<ArenaOVec<double>, dim, 3> inv_jac_ovec;
684 for (
uint i=0; i<dim*3; ++i) {
685 inv_jac_ovec(i) =
ArenaOVec(inv_jac_vec(i));
688 auto &result_vec = op.result_matrix();
689 Eigen::Matrix<ArenaOVec<double>, 3, 1> result_ovec = inv_jac_ovec.transpose() * ref_grads_ovec;
690 for (
uint i=0; i<3; ++i) {
691 result_vec(i) = result_ovec(i).get_vec();
701 template<
unsigned int dim>
705 auto &jac_value = op.result_matrix();
706 const auto &coords_value = operations[ op.input_ops()[0] ].result_matrix();
707 for (
unsigned int i=0; i<3; i++)
708 for (
unsigned int j=0; j<dim; j++)
709 jac_value(i,j) = coords_value(i,j+1) - coords_value(i,0);
711 template<
unsigned int dim>
714 auto &inv_jac_value = op.result_matrix();
715 const auto &jac_value = operations[ op.input_ops()[0] ].result_matrix();
716 inv_jac_value = eigen_arena_tools::inverse<3, dim>(jac_value);
718 template<
unsigned int dim>
722 auto &jac_value = op.result_matrix();
723 const auto &coords_value = operations[ op.input_ops()[0] ].result_matrix();
724 for (
unsigned int i=0; i<3; i++)
725 for (
unsigned int j=0; j<dim-1; j++)
726 jac_value(i,j) = coords_value(i,j+1) - coords_value(i,0);
728 template<
unsigned int dim>
732 auto &jac_det_value = op.result_matrix();
733 const auto &jac_value = operations[ op.input_ops()[0] ].result_matrix();
743 op.allocate_result(point_weights.size(), *arena);
744 auto &weights_value = op.result_matrix();
745 for (
uint i=0; i<point_weights.size(); ++i)
746 weights_value(0,0)(i) = point_weights[i];
750 auto &weights_value = operations[ op.input_ops()[0] ].result_matrix();
751 auto &jac_det_value = operations[ op.input_ops()[1] ].result_matrix();
755 auto &jxw_value = op.result_matrix();
756 jxw_value(0,0) = jxw_ovec.
get_vec();
758 template<
unsigned int dim>
761 auto &normal_value = op.result_matrix();
762 auto &inv_jac_mat_value = operations[ op.input_ops()[0] ].result_matrix();
765 ArenaVec<double> norm_vec( normal_value(0).data_size(), normal_value(0).arena() );
766 Eigen::VectorXd A(3);
767 for (
uint i=0; i<normal_value(0).data_size(); ++i) {
768 A(0) = normal_value(0)(i);
769 A(1) = normal_value(1)(i);
770 A(2) = normal_value(2)(i);
771 norm_vec(i) = A.norm();
774 size_t points_per_side = el_table(4).
data_size() / el_table(3).data_size();
775 size_t n_points = el_table(3).data_size();
776 for (
uint i=0; i<3; ++i) {
777 normal_value(i) = normal_value(i) / norm_vec;
778 ArenaVec<double> expand_vec( normal_value(i).data_size() * points_per_side, normal_value(i).arena() );
780 expand_vec(j) = normal_value(i)(j % n_points);
782 normal_value(i) = expand_vec;
787 uint n_dofs = shape_values[0][0].size();
788 uint n_sides = el_table(3).data_size();
789 uint n_patch_points = el_table(4).data_size();
791 auto &op = operations[scalar_shape_op_idx];
792 auto &scalar_shape_value = op.result_matrix();
793 scalar_shape_value(0) =
ArenaVec<double>(n_dofs*n_patch_points, scalar_shape_value(0).arena());
795 for (
uint i_dof=0; i_dof<n_dofs; ++i_dof) {
796 uint dof_shift = i_dof * n_patch_points;
797 for (
uint i_pt=0; i_pt<n_patch_points; ++i_pt)
798 scalar_shape_value(0)(i_pt + dof_shift) = shape_values[el_table(4)(i_pt)][i_pt / n_sides][i_dof];
801 template<
unsigned int dim>
804 uint n_points = ref_shape_grads[0].size();
805 uint n_dofs = ref_shape_grads[0][0].size();
806 uint n_sides = el_table(3).data_size();
807 uint n_patch_points = el_table(4).data_size();
810 auto &op = operations[scalar_shape_grads_op_idx];
811 auto &grad_scalar_shape_value = op.result_matrix();
814 auto &inv_jac_value = operations[ op.input_ops()[0] ].result_matrix();
815 Eigen::Matrix<ArenaVec<double>, dim, 3> inv_jac_expd_value;
816 for (
uint i=0; i<dim*3; ++i) {
817 inv_jac_expd_value(i) =
ArenaVec<double>( n_dofs*n_patch_points, inv_jac_value(i).arena() );
818 for (
uint j=0; j<n_dofs*n_patch_points; ++j)
819 inv_jac_expd_value(i)(j) = inv_jac_value(i)(j%n_sides);
823 Eigen::Matrix<ArenaVec<double>, dim, 1> ref_shape_grads_expd;
824 for (
uint i=0; i<dim; ++i) {
825 ref_shape_grads_expd(i) =
ArenaVec<double>( n_dofs*n_patch_points, inv_jac_value(0).arena() );
827 for (
uint i_dof=0; i_dof<n_dofs; ++i_dof) {
828 for (
uint i_pt=0; i_pt<n_points; ++i_pt) {
829 uint i_begin = (i_dof * n_points + i_pt) * n_sides;
830 for (
uint i_sd=0; i_sd<n_sides; ++i_sd) {
831 for (
uint i_c=0; i_c<dim; ++i_c) {
832 ref_shape_grads_expd(i_c)(i_begin + i_sd) = ref_shape_grads[el_table(3)(i_sd)][i_pt][i_dof][i_c];
839 grad_scalar_shape_value = inv_jac_expd_value.transpose() * ref_shape_grads_expd;
852 auto &result_vec = op.result_matrix();
853 for (
uint i=0;i<result_vec(0,0).data_size(); ++i) {
854 result_vec(0,0)(i) = 1.0;
863 template<
unsigned int spacedim = 3>
886 template<
unsigned int dim>
918 template<
unsigned int spacedim = 3>
941 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)