21 #ifndef PATCH_POINT_VALUES_HH_
22 #define PATCH_POINT_VALUES_HH_
24 #include <Eigen/Dense>
35 template<
unsigned int spacedim>
class ElOp;
121 template<
unsigned int spacedim = 3>
193 elOp.allocate_result(sizes[elOp.size_type()], *
patch_arena_);
217 for (
uint i_col=0; i_col<coords.n_cols; ++i_col)
218 for (
uint i_row=0; i_row<coords.n_rows; ++i_row) {
220 coords_mat(i_row, i_col)(i_elem) = coords(i_row, i_col);
241 for (
uint i_col=0; i_col<elm_coords.n_cols; ++i_col)
242 for (
uint i_row=0; i_row<elm_coords.n_rows; ++i_row) {
244 coords_mat(i_row, i_col)(i_elem) = elm_coords(i_row, i_col);
254 for (
uint i_col=0; i_col<side_coords.n_cols; ++i_col)
255 for (
uint i_row=0; i_row<side_coords.n_rows; ++i_row) {
257 coords_mat(i_row, i_col)(i_elem) = side_coords(i_row, i_col);
276 uint point_pos = i_point_on_elem *
n_elems_ + elem_table_row;
295 uint point_pos = i_point_on_side *
n_elems_ + elem_table_row;
396 const auto &op_matrix =
operations_[op_idx].result_matrix();
398 for (
uint i=0; i<3; ++i)
399 val(i) = op_matrix(i)(op_matrix_idx);
412 const auto &op_matrix =
operations_[op_idx].result_matrix();
414 for (
uint i=0; i<3; ++i)
415 for (
uint j=0; j<3; ++j)
416 val(i,j) = op_matrix(i,j)(op_matrix_idx);
431 const auto &
mat = op.result_matrix();
432 for (
uint i_mat=0; i_mat<
mat.rows()*
mat.cols(); ++i_mat) {
433 if (
mat(i_mat).data_size()==0) stream <<
"<empty>";
435 const double *vals =
mat(i_mat).data_ptr();
436 for (
size_t i_val=0; i_val<
mat(i_mat).data_size(); ++i_val)
437 stream << vals[i_val] <<
" ";
447 if (
int_table_(i_row).data_size()==0) stream <<
"<empty>";
450 for (
size_t i_val=0; i_val<
int_table_(i_row).data_size(); ++i_val)
451 stream << vals[i_val] <<
" ";
468 {
"el_coords",
"jacobian",
"inv_jac",
"jac_det",
"pt_coords",
"weights",
"JxW",
"",
"",
"",
"",
"" },
469 {
"el_coords",
"el_jac",
"el_inv_jac",
"side_coords",
"side_jac",
"side_jac_det",
"exp_el_coords",
"exp_el_jac",
"exp_el_inv_jac",
470 "exp_side_coords",
"exp_side_jac",
"exp_side_jac_det",
"pt_coords",
"weights",
"JxW",
"normal_vec",
"",
"",
"",
"",
"" }
472 stream << std::setfill(
' ') <<
" Operation" << setw(12) <<
"" <<
"Shape" << setw(2) <<
""
473 <<
"Result row" << setw(2) <<
"" <<
"n DOFs" << setw(2) <<
"" <<
"Input operations" << endl;
475 stream <<
" " << std::left << setw(20) << op_names[bulk_side][i] <<
"" <<
" " << setw(6) <<
operations_[i].format_shape() <<
"" <<
" "
476 << setw(11) <<
operations_[i].result_row() <<
"" <<
" " << setw(7) <<
operations_[i].n_dofs() <<
"" <<
" ";
478 for (
auto i_o : input_ops) stream << op_names[bulk_side][i_o] <<
" ";
525 friend class ElOp<spacedim>;
526 template<
unsigned int dim>
528 template<
unsigned int dim>
530 template<
unsigned int dim>
537 template<
unsigned int spacedim = 3>
552 if (shape_vec.size() == 1) shape_vec.push_back(1);
613 result_ = Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic>(
shape_[0],
shape_[1]);
619 Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic> &
result_matrix() {
627 const Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic> &
result_matrix()
const {
635 template<
unsigned int dim1,
unsigned int dim2>
636 Eigen::Map<Eigen::Matrix<ArrayDbl, dim1, dim2>>
value(
TableDbl &op_results,
uint i_dof = 0)
const {
637 return Eigen::Map<Eigen::Matrix<ArrayDbl, dim1, dim2>>(op_results.data() +
result_row_ + i_dof *
n_comp(), dim1, dim2);
642 return Eigen::Map<Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic>>(op_results(
result_row_).data(), dim1, dim2);
647 return Eigen::Map<Eigen::Vector<double, Eigen::Dynamic>>(op_results(
result_row_).data(), op_results(
result_row_).rows());
655 Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic>
result_;
688 template<
unsigned int dim>
692 auto &jac_value = op.result_matrix();
693 const auto &coords_value = operations[ op.input_ops()[0] ].result_matrix();
694 for (
unsigned int i=0; i<3; i++)
695 for (
unsigned int j=0; j<dim; j++)
696 jac_value(i,j) = coords_value(i,j+1) - coords_value(i,0);
698 template<
unsigned int dim>
702 auto &inv_jac_value = op.result_matrix();
703 const auto &jac_value = operations[ op.input_ops()[0] ].result_matrix();
704 inv_jac_value = eigen_arena_tools::inverse<3, dim>(jac_value);
706 template<
unsigned int dim>
710 auto &jac_det_value = op.result_matrix();
711 const auto &jac_value = operations[ op.input_ops()[0] ].result_matrix();
712 jac_det_value(0,0) = eigen_arena_tools::determinant<3, dim>(jac_value).abs();
721 op.allocate_result(point_weights.size(), *arena);
722 auto &weights_value = op.result_matrix();
723 for (
uint i=0; i<point_weights.size(); ++i)
724 weights_value(0,0)(i) = point_weights[i];
728 auto &weights_value = operations[ op.input_ops()[0] ].result_matrix();
729 auto &jac_det_value = operations[ op.input_ops()[1] ].result_matrix();
730 ArenaOVec<double> weights_ovec( weights_value(0,0), weights_value(0,0).data_size() );
731 ArenaOVec<double> jac_det_ovec( jac_det_value(0,0), jac_det_value(0,0).data_size() );
733 auto &jxw_value = op.result_matrix();
734 jxw_value(0,0) = jxw_ovec.
get_vec();
738 auto &op = operations[scalar_shape_op_idx];
739 uint n_dofs = shape_values.size();
740 uint n_points = shape_values[0].size();
741 uint n_elem = op.result_matrix()(0).data_size() / n_points;
743 auto &shape_matrix = op.result_matrix();
746 for (
uint i=0; i<n_elem; ++i) {
751 for (
uint i_dof=0; i_dof<n_dofs; ++i_dof)
752 for (
uint i_p=0; i_p<n_points; ++i_p)
753 ref_vec(i_dof * n_points + i_p) = shape_values[i_dof][i_p];
755 shape_matrix(0) = shape_ovec.
get_vec();
757 template<
unsigned int dim>
760 auto &op = operations[scalar_shape_grads_op_idx];
761 auto &inv_jac_vec = operations[ op.input_ops()[0] ].result_matrix();
762 uint n_points = ref_shape_grads.size();
763 uint n_dofs = ref_shape_grads[0].size();
765 Eigen::Matrix<ArenaVec<double>, dim, 1> ref_grads_vec;
766 Eigen::Matrix<ArenaOVec<double>, dim, 1> ref_grads_ovec;
767 for (
uint i=0; i<ref_grads_vec.rows(); ++i) {
768 ref_grads_vec(i) =
ArenaVec<double>(n_points * n_dofs, op.result_matrix()(0).arena());
769 for (
uint i_dof=0; i_dof<n_dofs; ++i_dof)
770 for (
uint i_p=0; i_p<n_points; ++i_p)
771 ref_grads_vec(i)(i_dof * n_points + i_p) = ref_shape_grads[i_dof][i_p](i);
772 ref_grads_ovec(i) =
ArenaOVec(ref_grads_vec(i), ref_grads_vec(i).data_size());
775 Eigen::Matrix<ArenaOVec<double>, dim, 3> inv_jac_ovec;
776 for (
uint i=0; i<dim*3; ++i) {
777 inv_jac_ovec(i) =
ArenaOVec(inv_jac_vec(i), inv_jac_vec(i).data_size());
780 auto &result_vec = op.result_matrix();
781 Eigen::Matrix<ArenaOVec<double>, 3, 1> result_ovec = inv_jac_ovec.transpose() * ref_grads_ovec;
782 for (
uint i=0; i<3; ++i) {
783 result_vec(i) = result_ovec(i).get_vec();
793 template<
unsigned int dim>
797 auto &jac_value = op.result_matrix();
798 const auto &coords_value = operations[ op.input_ops()[0] ].result_matrix();
799 for (
unsigned int i=0; i<3; i++)
800 for (
unsigned int j=0; j<dim; j++)
801 jac_value(i,j) = coords_value(i,j+1) - coords_value(i,0);
803 template<
unsigned int dim>
806 auto &inv_jac_value = op.result_matrix();
807 const auto &jac_value = operations[ op.input_ops()[0] ].result_matrix();
808 inv_jac_value = eigen_arena_tools::inverse<3, dim>(jac_value);
810 template<
unsigned int dim>
814 auto &jac_value = op.result_matrix();
815 const auto &coords_value = operations[ op.input_ops()[0] ].result_matrix();
816 for (
unsigned int i=0; i<3; i++)
817 for (
unsigned int j=0; j<dim-1; j++)
818 jac_value(i,j) = coords_value(i,j+1) - coords_value(i,0);
820 template<
unsigned int dim>
824 auto &jac_det_value = op.result_matrix();
825 const auto &jac_value = operations[ op.input_ops()[0] ].result_matrix();
863 op.allocate_result(point_weights.size(), *arena);
864 auto &weights_value = op.result_matrix();
865 for (
uint i=0; i<point_weights.size(); ++i)
866 weights_value(0,0)(i) = point_weights[i];
870 auto &weights_value = operations[ op.input_ops()[0] ].result_matrix();
871 auto &jac_det_value = operations[ op.input_ops()[1] ].result_matrix();
872 ArenaOVec<double> weights_ovec( weights_value(0,0), weights_value(0,0).data_size() );
873 ArenaOVec<double> jac_det_ovec( jac_det_value(0,0), jac_det_value(0,0).data_size() );
875 auto &jxw_value = op.result_matrix();
876 jxw_value(0,0) = jxw_ovec.
get_vec();
878 template<
unsigned int dim>
910 template<
unsigned int dim>
947 auto &result_vec = op.result_matrix();
948 for (
uint i=0;i<result_vec(0,0).data_size(); ++i) {
949 result_vec(0,0)(i) = 1.0;
962 template<
unsigned int spacedim = 3>
985 template<
unsigned int dim>
1017 template<
unsigned int spacedim = 3>
1040 template<
unsigned int dim>
#define ASSERT_EQ(a, b)
Definition of comparative assert macro (EQual) only for debug mode.
Outer product - only proposal of multi operator.
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.
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)
uint result_row() const
Getter for result_row_.
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)
uint result_row_
First row to scalar, vector or matrix result TODO replace.
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.
Eigen::Map< Eigen::Matrix< ArrayDbl, dim1, dim2 > > value(TableDbl &op_results, uint i_dof=0) const
Return map referenced Eigen::Matrix of given dimension.
Eigen::Map< Eigen::Vector< double, Eigen::Dynamic > > vector_value(TableDbl &op_results) const
Return map referenced Eigen::Matrix of given dimensions.
ElOp(uint dim, std::initializer_list< uint > shape, uint result_row, ReinitFunction reinit_f, OpSizeType size_type, std::vector< uint > input_ops={}, uint n_dofs=1)
Eigen::Map< Eigen::Matrix< double, Eigen::Dynamic, Eigen::Dynamic > > matrix_value(TableDbl &op_results, uint dim1, uint dim2) const
Return map referenced Eigen::Matrix of given dimensions.
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)
ElOp< spacedim > & make_expansion(ElOp< spacedim > &el_op, std::initializer_list< uint > shape, ReinitFunction reinit_f)
AssemblyArena & asm_arena_
Reference to global assembly arena of PatchFeValues.
std::vector< OpSizeType > row_sizes_
hold sizes of rows by type of operation
uint n_rows() const
Getter for n_rows_.
std::vector< uint > points_map_
Map of point patch indices to point_vals_ and int_table_ tables.
uint n_rows_
Number of columns of point_vals table.
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 el_vals_ table.
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.
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
operation executed expansion to quadrature point (value of element / side > values on quadrature poin...
@ 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 ptop_normal_vec(FMT_UNUSED std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table)
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 ptop_scalar_shape_grads(FMT_UNUSED std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table, FMT_UNUSED std::vector< std::vector< std::vector< arma::mat > > > ref_shape_grads, FMT_UNUSED uint scalar_shape_grads_op_idx)
static void elop_sd_jac(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 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)
static void ptop_scalar_shape(FMT_UNUSED std::vector< ElOp< 3 >> &operations, FMT_UNUSED IntTableArena &el_table, FMT_UNUSED std::vector< std::vector< std::vector< double > > > shape_values, FMT_UNUSED uint scalar_shape_op_idx)