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>
182 elOp.allocate_result(sizes[elOp.size_type()],
patch_arena_);
207 for (
uint i_col=0; i_col<coords.n_cols; ++i_col)
208 for (
uint i_row=0; i_row<coords.n_rows; ++i_row) {
210 coords_mat(i_row, i_col)(i_elem) = coords(i_row, i_col);
226 for (
uint i_col=0; i_col<elm_coords.n_cols; ++i_col)
227 for (
uint i_row=0; i_row<elm_coords.n_rows; ++i_row) {
233 for (
uint i_col=0; i_col<side_coords.n_cols; ++i_col)
234 for (
uint i_row=0; i_row<side_coords.n_rows; ++i_row) {
368 for (
uint i=0; i<3; ++i)
374 const auto &op_matrix =
operations_[op_idx].result_matrix();
375 for (
uint i=0; i<3; ++i)
388 for (
uint i=0; i<3; ++i)
389 for (
uint j=0; j<3; ++j)
395 const auto &op_matrix =
operations_[op_idx].result_matrix();
396 for (
uint i=0; i<3; ++i)
397 for (
uint j=0; j<3; ++j)
413 const auto &
mat = op.result_matrix();
414 for (
uint i_mat=0; i_mat<
mat.rows()*
mat.cols(); ++i_mat) {
415 if (
mat(i_mat).data_size()==0) stream <<
"<empty>";
417 const double *vals =
mat(i_mat).data_ptr();
418 for (
size_t i_val=0; i_val<
mat(i_mat).data_size(); ++i_val)
419 stream << vals[i_val] <<
" ";
429 for (
uint i_col=0; i_col<3; ++i_col)
430 stream <<
int_vals_(i_col)(i_row) <<
" ";
446 {
"el_coords",
"jacobian",
"inv_jac",
"jac_det",
"pt_coords",
"weights",
"JxW",
"",
"",
"",
"",
"" },
447 {
"el_coords",
"el_jac",
"el_inv_jac",
"side_coords",
"side_jac",
"side_jac_det",
"exp_el_coords",
"exp_el_jac",
"exp_el_inv_jac",
448 "exp_side_coords",
"exp_side_jac",
"exp_side_jac_det",
"pt_coords",
"weights",
"JxW",
"normal_vec",
"",
"",
"",
"",
"" }
450 stream << std::setfill(
' ') <<
" Operation" << setw(12) <<
"" <<
"Shape" << setw(2) <<
""
451 <<
"Result row" << setw(2) <<
"" <<
"n DOFs" << setw(2) <<
"" <<
"Input operations" << endl;
453 stream <<
" " << std::left << setw(20) << op_names[bulk_side][i] <<
"" <<
" " << setw(6) <<
operations_[i].format_shape() <<
"" <<
" "
454 << setw(11) <<
operations_[i].result_row() <<
"" <<
" " << setw(7) <<
operations_[i].n_dofs() <<
"" <<
" ";
456 for (
auto i_o : input_ops) stream << op_names[bulk_side][i_o] <<
" ";
495 friend class ElOp<spacedim>;
496 template<
unsigned int dim>
498 template<
unsigned int dim>
500 template<
unsigned int dim>
507 template<
unsigned int spacedim = 3>
522 if (shape_vec.size() == 1) shape_vec.push_back(1);
583 result_ = Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic>(
shape_[0],
shape_[1]);
589 Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic> &
result_matrix() {
597 const Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic> &
result_matrix()
const {
605 template<
unsigned int dim1,
unsigned int dim2>
606 Eigen::Map<Eigen::Matrix<ArrayDbl, dim1, dim2>>
value(
TableDbl &op_results,
uint i_dof = 0)
const {
607 return Eigen::Map<Eigen::Matrix<ArrayDbl, dim1, dim2>>(op_results.data() +
result_row_ + i_dof *
n_comp(), dim1, dim2);
612 return Eigen::Map<Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic>>(op_results(
result_row_).data(), dim1, dim2);
617 return Eigen::Map<Eigen::Vector<double, Eigen::Dynamic>>(op_results(
result_row_).data(), op_results(
result_row_).rows());
625 Eigen::Matrix<ArenaVec<double>, Eigen::Dynamic, Eigen::Dynamic>
result_;
646 uint size = op_results(row_begin).rows();
647 for (
int i_pt=size-1; i_pt>=0; --i_pt) {
648 uint el_table_idx = el_table(1)(i_pt);
649 for (
uint i_q=row_begin; i_q<row_end; ++i_q)
650 op_results(i_q)(i_pt) = op_results(i_q)(el_table_idx);
658 template<
unsigned int dim>
662 auto &jac_value = op.result_matrix();
663 const auto &coords_value = operations[ op.input_ops()[0] ].result_matrix();
664 for (
unsigned int i=0; i<3; i++)
665 for (
unsigned int j=0; j<dim; j++)
666 jac_value(i,j) = coords_value(i,j+1) - coords_value(i,0);
668 template<
unsigned int dim>
672 auto &inv_jac_value = op.result_matrix();
673 const auto &jac_value = operations[ op.input_ops()[0] ].result_matrix();
674 inv_jac_value = eigen_arena_tools::inverse<3, dim>(jac_value);
676 template<
unsigned int dim>
680 auto &jac_det_value = op.result_matrix();
681 const auto &jac_value = operations[ op.input_ops()[0] ].result_matrix();
682 jac_det_value(0,0) = eigen_arena_tools::determinant<3, dim>(jac_value).abs();
691 op.allocate_result(point_weights.size(), arena);
692 auto &weights_value = op.result_matrix();
693 for (
uint i=0; i<point_weights.size(); ++i)
694 weights_value(0,0)(i) = point_weights[i];
698 auto &weights_value = operations[ op.input_ops()[0] ].result_matrix();
699 auto &jac_det_value = operations[ op.input_ops()[1] ].result_matrix();
702 std::cout <<
"ptop_JxW " << weights_ovec.
data_size() <<
"x" << jac_det_ovec.
data_size() << std::endl;
704 auto &jxw_value = op.result_matrix();
705 jxw_value(0,0) = jxw_ovec.
get_vec();
709 auto &op = operations[scalar_shape_op_idx];
710 uint n_points = shape_values.size();
712 for (
uint i_row=0; i_row<shape_values[0].size(); ++i_row) {
713 ArrayDbl &result_row = op_results( op.result_row()+i_row );
714 for (
uint i_pt=0; i_pt<result_row.rows(); ++i_pt)
715 result_row(i_pt) = shape_values[i_pt % n_points][i_row];
718 template<
unsigned int dim>
721 auto &op = operations[scalar_shape_grads_op_idx];
722 uint n_points = ref_shape_grads.size();
723 uint n_dofs = ref_shape_grads[0].size();
725 Eigen::Vector<ArrayDbl, Eigen::Dynamic> ref_shape_grads_expd;
726 ref_shape_grads_expd.resize(dim*n_dofs);
727 for (
uint i=0; i<ref_shape_grads_expd.rows(); ++i)
728 ref_shape_grads_expd(i).resize(op_results(0).
rows());
730 for (
uint i_dof=0; i_dof<n_dofs; ++i_dof)
731 for (
uint i_c=0; i_c<dim; ++i_c) {
732 ArrayDbl &shape_grad_row = ref_shape_grads_expd(i_dof*dim+i_c);
733 for (
uint i_pt=0; i_pt<shape_grad_row.rows(); ++i_pt)
734 shape_grad_row(i_pt) = ref_shape_grads[i_pt % n_points][i_dof][i_c];
737 auto inv_jac_value = operations[ op.input_ops()[0] ].value<dim, 3>(op_results);
738 for (
uint i_dof=0; i_dof<n_dofs; ++i_dof) {
739 auto shape_grad_value = op.value<3, 1>(op_results, i_dof);
740 Eigen::Map<Eigen::Matrix<ArrayDbl, dim, 1>> ref_shape_grads_dof_value(ref_shape_grads_expd.data() + dim*i_dof, dim, 1);
741 shape_grad_value = inv_jac_value.transpose() * ref_shape_grads_dof_value;
751 template<
unsigned int dim>
755 auto jac_value = op.value<3, dim>(op_results);
756 auto coords_value = operations[ op.input_ops()[0] ].value<3, dim+1>(op_results);
757 jac_value = eigen_tools::jacobian<3,dim>(coords_value);
759 template<
unsigned int dim>
762 auto inv_jac_value = op.value<dim, 3>(op_results);
763 auto jac_value = operations[ op.input_ops()[0] ].value<3, dim>(op_results);
764 inv_jac_value = eigen_tools::inverse<3, dim>(jac_value);
766 template<
unsigned int dim>
770 auto jac_value = op.value<3, dim-1>(op_results);
771 auto coords_value = operations[ op.input_ops()[0] ].value<3, dim>(op_results);
774 template<
unsigned int dim>
778 ArrayDbl &det_value = op_results( op.result_row() );
779 auto jac_value = operations[ op.input_ops()[0] ].value<3, dim-1>(op_results);
801 template<
unsigned int dim>
817 ArrayDbl &result_row = op_results( op.result_row() );
818 auto n_points = point_weights.size();
819 for (
uint i=0; i<result_row.rows(); ++i)
820 result_row(i) = point_weights[i%n_points];
824 ArrayDbl &weights_row = op_results( operations[op.input_ops()[0]].result_row() );
825 ArrayDbl &jac_det_row = op_results( operations[op.input_ops()[1]].result_row() );
826 ArrayDbl &result_row = op_results( op.result_row() );
827 result_row = jac_det_row * weights_row;
829 template<
unsigned int dim>
832 auto normal_value = op.value<3, 1>(op_results);
833 auto inv_jac_mat_value = operations[ op.input_ops()[0] ].value<dim, 3>(op_results);
837 norm_vec.resize(normal_value(0).
rows());
838 Eigen::VectorXd A(3);
840 for (
uint i=0; i<normal_value(0).rows(); ++i) {
841 A(0) = normal_value(0)(i);
842 A(1) = normal_value(1)(i);
843 A(2) = normal_value(2)(i);
844 norm_vec(i) = A.norm();
846 for (
uint i=0; i<3; ++i) {
847 normal_value(i) /= norm_vec;
852 auto &op = operations[scalar_shape_op_idx];
853 uint n_points = shape_values[0].size();
855 for (
uint i_row=0; i_row<shape_values[0][0].size(); ++i_row) {
856 ArrayDbl &result_row = op_results( op.result_row()+i_row );
857 for (
uint i_pt=0; i_pt<result_row.rows(); ++i_pt)
858 result_row(i_pt) = shape_values[el_table(3)(i_pt)][i_pt % n_points][i_row];
861 template<
unsigned int dim>
864 auto &op = operations[scalar_shape_grads_op_idx];
865 uint n_points = ref_shape_grads[0].size();
866 uint n_dofs = ref_shape_grads[0][0].size();
868 Eigen::Vector<ArrayDbl, Eigen::Dynamic> ref_shape_grads_expd;
869 ref_shape_grads_expd.resize(dim*n_dofs);
870 for (
uint i=0; i<ref_shape_grads_expd.rows(); ++i)
871 ref_shape_grads_expd(i).resize(op_results(0).
rows());
873 for (
uint i_dof=0; i_dof<n_dofs; ++i_dof)
874 for (
uint i_c=0; i_c<dim; ++i_c) {
875 ArrayDbl &shape_grad_row = ref_shape_grads_expd(i_dof*dim+i_c);
876 for (
uint i_pt=0; i_pt<shape_grad_row.rows(); ++i_pt)
877 shape_grad_row(i_pt) = ref_shape_grads[el_table(3)(i_pt)][i_pt % n_points][i_dof][i_c];
880 auto inv_jac_value = operations[ op.input_ops()[0] ].value<dim, 3>(op_results);
881 for (
uint i_dof=0; i_dof<n_dofs; ++i_dof) {
882 auto shape_grad_value = op.value<3, 1>(op_results, i_dof);
883 Eigen::Map<Eigen::Matrix<ArrayDbl, dim, 1>> ref_shape_grads_dof_value(ref_shape_grads_expd.data() + dim*i_dof, dim, 1);
884 shape_grad_value = inv_jac_value.transpose() * ref_shape_grads_dof_value;
898 ArrayDbl &result_vec = op_results( op.result_row() );
899 for (
uint i=0;i<result_vec.size(); ++i) {
913 template<
unsigned int spacedim = 3>
935 template<
unsigned int dim>
967 template<
unsigned int spacedim = 3>
989 template<
unsigned int dim>
1013 this->
make_expansion( sd_jac, {spacedim, this->
dim_-1}, &side_reinit::expd_sd_jac<dim> );
#define ASSERT_EQ(a, b)
Definition of comparative assert macro (EQual) only for debug mode.
Outer product - only proposal of multi operator.
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.
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.
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.
void allocate_result(size_t data_size, AssemblyArena &arena)
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_.
void reinit_function(std::vector< ElOp< spacedim >> &operations, TableDbl &data_table, TableInt &int_table)
Call reinit function on element table if function is defined.
Bulk data specialization, order of item in operations_ vector corresponds to the BulkOps enum.
PatchPointValues(uint dim, uint quad_order, AssemblyArena &patch_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 &patch_arena)
Constructor.
void init()
Initialize operations vector.
void print_data_tables(ostream &stream, bool points, bool ints) const
ElOp< spacedim > & make_expansion(ElOp< spacedim > &el_op, std::initializer_list< uint > shape, ReinitFunction reinit_f)
std::vector< OpSizeType > row_sizes_
hold sizes of rows by type of operation
void initialize(uint int_cols)
uint n_rows() const
Getter for n_rows_.
std::vector< uint > points_map_
Map of point patch indices to point_vals_ and int_vals_ tables.
uint register_side_point(uint elem_table_row, uint value_patch_idx, uint elem_idx, uint side_idx)
uint n_rows_
Number of columns of point_vals table.
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.
Tensor tensor_val(uint result_row, uint point_idx) const
uint register_side(arma::mat elm_coords, arma::mat side_coords)
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_.
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)
Scalar scalar_value(uint op_idx, uint point_idx) const
std::vector< uint > elements_map_
Map of element patch indices to el_vals_ table.
uint dim() const
Getter for dim_.
Vector vector_value(uint op_idx, uint point_idx) const
Vector vector_val(uint result_row, uint point_idx) const
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.
PatchPointValues(uint dim, AssemblyArena &patch_arena)
Scalar scalar_val(uint result_row, uint point_idx) const
ElOp< spacedim > & make_new_op(std::initializer_list< uint > shape, ReinitFunction reinit_f, std::vector< uint > input_ops_vec, OpSizeType size_type=pointOp)
uint register_bulk_point(uint elem_table_row, uint value_patch_idx, uint elem_idx)
Tensor tensor_value(uint op_idx, uint point_idx) const
uint n_elems_
Number of elements in patch.
AssemblyArena & patch_arena_
Reference to global Arena of PatchFeValues.
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::Vector< ArrayDbl, dim > normal_vector_array(Eigen::Array< uint, Eigen::Dynamic, 1 > loc_side_idx_array)
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.
@ opExpansionElCoords
operation executed expansion to quadrature point (value of element / side > values on quadrature poin...
@ opExpansionElJac
expands Jacobian on element
@ opNormalVec
normal vector of quadrature point
@ opSideCoords
operations evaluated on sides
@ opExpansionSideCoords
expands coordinates on side
@ opJxW
JxW value of quadrature point.
@ opCoords
operations evaluated on quadrature points
@ opExpansionElInvJac
expands inverse Jacobian on element
@ opExpansionSideJac
expands Jacobian on side
@ 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 > > &, TableDbl &, TableInt &)> 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_jac(std::vector< ElOp< 3 >> &operations, FMT_UNUSED TableDbl &op_results, FMT_UNUSED TableInt &el_table)
static void ptop_scalar_shape_grads(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, std::vector< std::vector< arma::mat > > ref_shape_grads, uint scalar_shape_grads_op_idx)
static void ptop_JxW(std::vector< ElOp< 3 >> &operations, FMT_UNUSED TableDbl &op_results, FMT_UNUSED TableInt &el_table)
static void ptop_coords(FMT_UNUSED std::vector< ElOp< 3 >> &operations, FMT_UNUSED TableDbl &op_results, FMT_UNUSED TableInt &el_table)
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_weights(std::vector< ElOp< 3 >> &operations, AssemblyArena &arena, const std::vector< double > &point_weights)
static void elop_jac_det(std::vector< ElOp< 3 >> &operations, FMT_UNUSED TableDbl &op_results, FMT_UNUSED TableInt &el_table)
static void elop_inv_jac(std::vector< ElOp< 3 >> &operations, FMT_UNUSED TableDbl &op_results, FMT_UNUSED TableInt &el_table)
Defines common functionality of reinit operations.
static void expand_data(ElOp< 3 > &op, TableDbl &op_results, TableInt &el_table)
static void op_base(FMT_UNUSED std::vector< ElOp< 3 >> &operations, FMT_UNUSED TableDbl &op_results, FMT_UNUSED TableInt &el_table)
Defines reinit operations on side points.
static void ptop_normal_vec(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, TableInt &el_table)
static void elop_el_jac(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, FMT_UNUSED TableInt &el_table)
static void ptop_weights(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, std::vector< double > point_weights)
static void elop_sd_jac_det(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, FMT_UNUSED TableInt &el_table)
static void expd_sd_coords(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, TableInt &el_table)
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)
static void expd_el_inv_jac(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, TableInt &el_table)
static void expd_sd_jac_det(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, TableInt &el_table)
static void ptop_JxW(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, FMT_UNUSED TableInt &el_table)
static void expd_el_jac(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, TableInt &el_table)
static void expd_el_coords(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, TableInt &el_table)
static void elop_el_inv_jac(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, FMT_UNUSED TableInt &el_table)
static void expd_sd_jac(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, TableInt &el_table)
static void ptop_coords(FMT_UNUSED std::vector< ElOp< 3 >> &operations, FMT_UNUSED TableDbl &op_results, FMT_UNUSED TableInt &el_table)
static void elop_sd_jac(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, FMT_UNUSED TableInt &el_table)
static void ptop_scalar_shape_grads(std::vector< ElOp< 3 >> &operations, TableDbl &op_results, TableInt &el_table, std::vector< std::vector< std::vector< arma::mat > > > ref_shape_grads, uint scalar_shape_grads_op_idx)