20 template <
typename Derived>
24 Eigen::Matrix<double, dim * dim, dim> B(size() * size(), size());
27 for (
int i = 0; i < dim; ++i)
28 for (
int j = 0; j < dim; ++j)
29 B(i * dim + j, i) = g(j);
34 template <
typename Derived>
39 template <
typename Derived>
45 const std::function<Eigen::MatrixXd(
const Eigen::MatrixXd &)> &fun)
const
47 Eigen::MatrixXd deformation_grad(size(), size());
48 Eigen::MatrixXd stress_tensor(size(), size());
52 const auto &displacement = data.
fun;
54 const auto &bs = data.
bs;
55 const auto &gbs = data.
gbs;
56 const auto el_id = data.
el_id;
58 assert(displacement.cols() == 1);
60 all.resize(local_pts.rows(), all_size);
63 Eigen::Matrix<Diff, Eigen::Dynamic, Eigen::Dynamic, 0, 3, 3> def_grad(size(), size());
66 vals.
compute(el_id, size() == 3, local_pts, bs, gbs);
68 for (
long p = 0; p < local_pts.rows(); ++p)
73 for (
int d = 0; d < size(); ++d)
74 deformation_grad(d, d) += 1;
78 all.row(p) = fun(deformation_grad);
82 for (
int d1 = 0; d1 < size(); ++d1)
84 for (
int d2 = 0; d2 < size(); ++d2)
85 def_grad(d1, d2) = Diff(d1 * size() + d2, deformation_grad(d1, d2));
88 const auto val = derived().elastic_energy(local_pts.row(p), data.
t,
vals.
element_id, def_grad);
90 for (
int d1 = 0; d1 < size(); ++d1)
92 for (
int d2 = 0; d2 < size(); ++d2)
93 stress_tensor(d1, d2) =
val.getGradient()(d1 * size() + d2);
96 stress_tensor = 1.0 / deformation_grad.determinant() * stress_tensor * deformation_grad.transpose();
103 all.row(p) = fun(stress_tensor);
107 template <
typename Derived>
110 return compute_energy_aux<double>(data);
113 template <
typename Derived>
117 return assemble_gradient_full_ad(data);
121 auto grad = assemble_gradient_stress_ad(data);
122 auto grad_full = assemble_gradient_full_ad(data);
123 assert((std::isnan(grad.norm()) && std::isnan(grad_full.norm())) || (grad - grad_full).norm() < 1e-6);
125 return assemble_gradient_stress_ad(data);
130 auto grad = assemble_gradient_stress_noad(data);
131 auto grad_full = assemble_gradient_stress_ad(data);
132 assert((std::isnan(grad.norm()) && std::isnan(grad_full.norm())) || (grad - grad_full).norm() < 1e-6);
134 return assemble_gradient_stress_noad(data);
138 template <
typename Derived>
143 size(), n_bases, data,
144 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar1<double, Eigen::Matrix<double, 6, 1>>>(data); },
145 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar1<double, Eigen::Matrix<double, 8, 1>>>(data); },
146 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar1<double, Eigen::Matrix<double, 12, 1>>>(data); },
147 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar1<double, Eigen::Matrix<double, 18, 1>>>(data); },
148 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar1<double, Eigen::Matrix<double, 24, 1>>>(data); },
149 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar1<double, Eigen::Matrix<double, 30, 1>>>(data); },
150 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar1<double, Eigen::Matrix<double, 60, 1>>>(data); },
151 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar1<double, Eigen::Matrix<double, 81, 1>>>(data); },
152 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar1<double, Eigen::Matrix<double, Eigen::Dynamic, 1, 0, SMALL_N, 1>>>(data); },
153 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar1<double, Eigen::Matrix<double, Eigen::Dynamic, 1, 0, BIG_N, 1>>>(data); },
154 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar1<double, Eigen::VectorXd>>(data); });
157 template <
typename Derived>
160 Eigen::Matrix<double, Eigen::Dynamic, 1> gradient;
169 compute_gradient_from_stress<3, 2>(data, gradient);
175 compute_gradient_from_stress<6, 2>(data, gradient);
181 compute_gradient_from_stress<10, 2>(data, gradient);
187 compute_gradient_from_stress<Eigen::Dynamic, 2>(data, gradient);
200 compute_gradient_from_stress<4, 3>(data, gradient);
206 compute_gradient_from_stress<10, 3>(data, gradient);
212 compute_gradient_from_stress<20, 3>(data, gradient);
218 compute_gradient_from_stress<Eigen::Dynamic, 3>(data, gradient);
227 template <
typename Derived>
230 Eigen::Matrix<double, Eigen::Dynamic, 1> gradient;
239 compute_gradient_from_stress_noad<3, 2>(data, gradient);
245 compute_gradient_from_stress_noad<6, 2>(data, gradient);
251 compute_gradient_from_stress_noad<10, 2>(data, gradient);
257 compute_gradient_from_stress_noad<Eigen::Dynamic, 2>(data, gradient);
270 compute_gradient_from_stress_noad<4, 3>(data, gradient);
276 compute_gradient_from_stress_noad<10, 3>(data, gradient);
282 compute_gradient_from_stress_noad<20, 3>(data, gradient);
288 compute_gradient_from_stress_noad<Eigen::Dynamic, 3>(data, gradient);
297 template <
typename Derived>
301 return assemble_hessian_full_ad(data);
305 auto hessian = assemble_hessian_stress_ad(data);
306 auto hessian_full = assemble_hessian_full_ad(data);
307 assert((std::isnan(hessian.norm()) && std::isnan(hessian_full.norm())) || (hessian - hessian_full).norm() < 1e-5);
309 return assemble_hessian_stress_ad(data);
314 auto hessian = assemble_hessian_stress_noad(data);
315 auto hessian_full = assemble_hessian_stress_ad(data);
316 assert((std::isnan(hessian.norm()) && std::isnan(hessian_full.norm())) || (hessian - hessian_full).norm() < 1e-5);
318 return assemble_hessian_stress_noad(data);
322 template <
typename Derived>
327 size(), n_bases, data,
328 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar2<double, Eigen::Matrix<double, 6, 1>, Eigen::Matrix<double, 6, 6>>>(data); },
329 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar2<double, Eigen::Matrix<double, 8, 1>, Eigen::Matrix<double, 8, 8>>>(data); },
330 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar2<double, Eigen::Matrix<double, 12, 1>, Eigen::Matrix<double, 12, 12>>>(data); },
331 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar2<double, Eigen::Matrix<double, 18, 1>, Eigen::Matrix<double, 18, 18>>>(data); },
332 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar2<double, Eigen::Matrix<double, 24, 1>, Eigen::Matrix<double, 24, 24>>>(data); },
333 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar2<double, Eigen::Matrix<double, 30, 1>, Eigen::Matrix<double, 30, 30>>>(data); },
334 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar2<double, Eigen::Matrix<double, 60, 1>, Eigen::Matrix<double, 60, 60>>>(data); },
335 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar2<double, Eigen::Matrix<double, 81, 1>, Eigen::Matrix<double, 81, 81>>>(data); },
336 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar2<double, Eigen::Matrix<double, Eigen::Dynamic, 1, 0, SMALL_N, 1>, Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, 0, SMALL_N, SMALL_N>>>(data); },
337 [&](
const NonLinearAssemblerData &data) {
return compute_energy_aux<DScalar2<double, Eigen::VectorXd, Eigen::MatrixXd>>(data); });
340 template <
typename Derived>
343 Eigen::MatrixXd hessian;
351 hessian.resize(6, 6);
353 compute_hessian_from_stress<3, 2>(data, hessian);
358 hessian.resize(12, 12);
360 compute_hessian_from_stress<6, 2>(data, hessian);
365 hessian.resize(20, 20);
367 compute_hessian_from_stress<10, 2>(data, hessian);
374 compute_hessian_from_stress<Eigen::Dynamic, 2>(data, hessian);
386 hessian.resize(12, 12);
388 compute_hessian_from_stress<4, 3>(data, hessian);
393 hessian.resize(30, 30);
395 compute_hessian_from_stress<10, 3>(data, hessian);
400 hessian.resize(60, 60);
402 compute_hessian_from_stress<20, 3>(data, hessian);
409 compute_hessian_from_stress<Eigen::Dynamic, 3>(data, hessian);
418 template <
typename Derived>
421 Eigen::MatrixXd hessian;
429 hessian.resize(6, 6);
431 compute_hessian_from_stress_noad<3, 2>(data, hessian);
436 hessian.resize(12, 12);
438 compute_hessian_from_stress_noad<6, 2>(data, hessian);
443 hessian.resize(20, 20);
445 compute_hessian_from_stress_noad<10, 2>(data, hessian);
452 compute_hessian_from_stress_noad<Eigen::Dynamic, 2>(data, hessian);
464 hessian.resize(12, 12);
466 compute_hessian_from_stress_noad<4, 3>(data, hessian);
471 hessian.resize(30, 30);
473 compute_hessian_from_stress_noad<10, 3>(data, hessian);
478 hessian.resize(60, 60);
480 compute_hessian_from_stress_noad<20, 3>(data, hessian);
487 compute_hessian_from_stress_noad<Eigen::Dynamic, 3>(data, hessian);
496 template <
typename Derived>
499 const Eigen::MatrixXd &mat,
500 Eigen::MatrixXd &stress,
501 Eigen::MatrixXd &result)
const
503 typedef DScalar2<double, Eigen::Matrix<double, Eigen::Dynamic, 1, 0, 9, 1>, Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, 0, 9, 9>> Diff;
505 const double t = data.
t;
506 const int el_id = data.
el_id;
507 const Eigen::MatrixXd &local_pts = data.
local_pts;
508 const Eigen::MatrixXd &global_pts = data.
global_pts;
509 const Eigen::MatrixXd &grad_u_i = data.
grad_u_i;
512 Eigen::Matrix<Diff, Eigen::Dynamic, Eigen::Dynamic, 0, 3, 3> def_grad(size(), size());
514 Eigen::MatrixXd
F = grad_u_i;
515 for (
int d = 0; d < size(); ++d)
518 assert(local_pts.rows() == 1);
519 for (
int i = 0; i < size(); ++i)
520 for (
int j = 0; j < size(); ++j)
521 def_grad(i, j) = Diff(i + j * size(),
F(i, j));
523 auto energy = derived().elastic_energy(global_pts, t, el_id, def_grad);
526 Eigen::MatrixXd grad = energy.getGradient().reshaped(size(), size());
528 Eigen::MatrixXd hess = energy.getHessian();
533 result = (hess * mat.reshaped(size() * size(), 1)).reshaped(size(), size());
536 template <
typename Derived>
539 Eigen::MatrixXd &stress,
540 Eigen::MatrixXd &result)
const
542 typedef DScalar2<double, Eigen::Matrix<double, Eigen::Dynamic, 1, 0, 9, 1>, Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, 0, 9, 9>> Diff;
544 const double t = data.
t;
545 const int el_id = data.
el_id;
546 const Eigen::MatrixXd &local_pts = data.
local_pts;
547 const Eigen::MatrixXd &global_pts = data.
global_pts;
548 const Eigen::MatrixXd &grad_u_i = data.
grad_u_i;
551 Eigen::Matrix<Diff, Eigen::Dynamic, Eigen::Dynamic, 0, 3, 3> def_grad(size(), size());
553 Eigen::MatrixXd
F = grad_u_i;
554 for (
int d = 0; d < size(); ++d)
557 assert(local_pts.rows() == 1);
558 for (
int i = 0; i < size(); ++i)
559 for (
int j = 0; j < size(); ++j)
560 def_grad(i, j) = Diff(i + j * size(),
F(i, j));
562 auto energy = derived().elastic_energy(global_pts, t, el_id, def_grad);
565 Eigen::MatrixXd grad = energy.getGradient().reshaped(size(), size());
567 Eigen::MatrixXd hess = energy.getHessian();
572 result = (hess * stress.reshaped(size() * size(), 1)).reshaped(size(), size());
575 template <
typename Derived>
578 const Eigen::MatrixXd &vect,
579 Eigen::MatrixXd &stress,
580 Eigen::MatrixXd &result)
const
582 typedef DScalar2<double, Eigen::Matrix<double, Eigen::Dynamic, 1, 0, 9, 1>, Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic, 0, 9, 9>> Diff;
584 const double t = data.
t;
585 const int el_id = data.
el_id;
586 const Eigen::MatrixXd &local_pts = data.
local_pts;
587 const Eigen::MatrixXd &global_pts = data.
global_pts;
588 const Eigen::MatrixXd &grad_u_i = data.
grad_u_i;
591 Eigen::Matrix<Diff, Eigen::Dynamic, Eigen::Dynamic, 0, 3, 3> def_grad(size(), size());
593 Eigen::MatrixXd
F = grad_u_i;
594 for (
int d = 0; d < size(); ++d)
597 assert(local_pts.rows() == 1);
598 for (
int i = 0; i < size(); ++i)
599 for (
int j = 0; j < size(); ++j)
600 def_grad(i, j) = Diff(i + j * size(),
F(i, j));
602 auto energy = derived().elastic_energy(global_pts, t, el_id, def_grad);
605 Eigen::MatrixXd grad = energy.getGradient().reshaped(size(), size());
607 Eigen::MatrixXd hess = energy.getHessian();
611 result.resize(hess.rows(), vect.size());
612 for (
int i = 0; i < hess.rows(); ++i)
613 if (vect.rows() == 1)
615 result.row(i) = vect * hess.row(i).reshaped(size(), size());
618 result.row(i) = (hess.row(i).reshaped(size(), size()) * vect).transpose();
ElementAssemblyValues vals
stores per element basis values at given quadrature points and geometric mapping
std::vector< AssemblyValues > basis_values
void compute(const int el_index, const bool is_volume, const Eigen::MatrixXd &pts, const basis::ElementBases &basis, const basis::ElementBases &gbasis)
computes the per element values at the local (ref el) points (pts) sets basis_values,...
void compute_stress_grad_multiply_stress(const OptAssemblerData &data, Eigen::MatrixXd &stress, Eigen::MatrixXd &result) const override
void assign_stress_tensor(const OutputData &data, const int all_size, const ElasticityTensorType &type, Eigen::MatrixXd &all, const std::function< Eigen::MatrixXd(const Eigen::MatrixXd &)> &fun) const override
Eigen::Matrix< double, dim *dim, dim > compute_B_block(const Eigen::Matrix< double, 1, dim > &g) const
Eigen::VectorXd assemble_gradient_full_ad(const NonLinearAssemblerData &data) const
Eigen::MatrixXd assemble_hessian_full_ad(const NonLinearAssemblerData &data) const
Eigen::MatrixXd assemble_hessian_stress_noad(const NonLinearAssemblerData &data) const
Eigen::VectorXd assemble_gradient_stress_noad(const NonLinearAssemblerData &data) const
Eigen::MatrixXd assemble_hessian_stress_ad(const NonLinearAssemblerData &data) const
Eigen::MatrixXd assemble_hessian(const NonLinearAssemblerData &data) const override
void compute_stress_grad_multiply_vect(const OptAssemblerData &data, const Eigen::MatrixXd &vect, Eigen::MatrixXd &stress, Eigen::MatrixXd &result) const override
Eigen::VectorXd assemble_gradient(const NonLinearAssemblerData &data) const override
Eigen::VectorXd assemble_gradient_stress_ad(const NonLinearAssemblerData &data) const
double compute_energy(const NonLinearAssemblerData &data) const override
void compute_stress_grad_multiply_mat(const OptAssemblerData &data, const Eigen::MatrixXd &mat, Eigen::MatrixXd &stress, Eigen::MatrixXd &result) const override
const ElementAssemblyValues & vals
const Eigen::MatrixXd & local_pts
const Eigen::MatrixXd & global_pts
const Eigen::MatrixXd & grad_u_i
const basis::ElementBases & bs
const Eigen::MatrixXd & fun
const Eigen::MatrixXd & local_pts
const basis::ElementBases & gbs
Eigen::MatrixXd pk2_from_cauchy(const Eigen::MatrixXd &stress, const Eigen::MatrixXd &F)
Eigen::MatrixXd hessian_from_energy(const int size, const int n_bases, const assembler::NonLinearAssemblerData &data, const std::function< DScalar2< double, Eigen::Matrix< double, 6, 1 >, Eigen::Matrix< double, 6, 6 > >(const assembler::NonLinearAssemblerData &)> &fun6, const std::function< DScalar2< double, Eigen::Matrix< double, 8, 1 >, Eigen::Matrix< double, 8, 8 > >(const assembler::NonLinearAssemblerData &)> &fun8, const std::function< DScalar2< double, Eigen::Matrix< double, 12, 1 >, Eigen::Matrix< double, 12, 12 > >(const assembler::NonLinearAssemblerData &)> &fun12, const std::function< DScalar2< double, Eigen::Matrix< double, 18, 1 >, Eigen::Matrix< double, 18, 18 > >(const assembler::NonLinearAssemblerData &)> &fun18, const std::function< DScalar2< double, Eigen::Matrix< double, 24, 1 >, Eigen::Matrix< double, 24, 24 > >(const assembler::NonLinearAssemblerData &)> &fun24, const std::function< DScalar2< double, Eigen::Matrix< double, 30, 1 >, Eigen::Matrix< double, 30, 30 > >(const assembler::NonLinearAssemblerData &)> &fun30, const std::function< DScalar2< double, Eigen::Matrix< double, 60, 1 >, Eigen::Matrix< double, 60, 60 > >(const assembler::NonLinearAssemblerData &)> &fun60, const std::function< DScalar2< double, Eigen::Matrix< double, 81, 1 >, Eigen::Matrix< double, 81, 81 > >(const assembler::NonLinearAssemblerData &)> &fun81, const std::function< DScalar2< double, Eigen::Matrix< double, Eigen::Dynamic, 1, 0, SMALL_N, 1 >, Eigen::Matrix< double, Eigen::Dynamic, Eigen::Dynamic, 0, SMALL_N, SMALL_N > >(const assembler::NonLinearAssemblerData &)> &funN, const std::function< DScalar2< double, Eigen::VectorXd, Eigen::MatrixXd >(const assembler::NonLinearAssemblerData &)> &funn)
void compute_diplacement_grad(const int size, const ElementAssemblyValues &vals, const Eigen::MatrixXd &local_pts, const int p, const Eigen::MatrixXd &displacement, Eigen::MatrixXd &displacement_grad)
Eigen::MatrixXd pk1_from_cauchy(const Eigen::MatrixXd &stress, const Eigen::MatrixXd &F)
Eigen::VectorXd gradient_from_energy(const int size, const int n_bases, const assembler::NonLinearAssemblerData &data, const std::function< DScalar1< double, Eigen::Matrix< double, 6, 1 > >(const assembler::NonLinearAssemblerData &)> &fun6, const std::function< DScalar1< double, Eigen::Matrix< double, 8, 1 > >(const assembler::NonLinearAssemblerData &)> &fun8, const std::function< DScalar1< double, Eigen::Matrix< double, 12, 1 > >(const assembler::NonLinearAssemblerData &)> &fun12, const std::function< DScalar1< double, Eigen::Matrix< double, 18, 1 > >(const assembler::NonLinearAssemblerData &)> &fun18, const std::function< DScalar1< double, Eigen::Matrix< double, 24, 1 > >(const assembler::NonLinearAssemblerData &)> &fun24, const std::function< DScalar1< double, Eigen::Matrix< double, 30, 1 > >(const assembler::NonLinearAssemblerData &)> &fun30, const std::function< DScalar1< double, Eigen::Matrix< double, 60, 1 > >(const assembler::NonLinearAssemblerData &)> &fun60, const std::function< DScalar1< double, Eigen::Matrix< double, 81, 1 > >(const assembler::NonLinearAssemblerData &)> &fun81, const std::function< DScalar1< double, Eigen::Matrix< double, Eigen::Dynamic, 1, 0, SMALL_N, 1 > >(const assembler::NonLinearAssemblerData &)> &funN, const std::function< DScalar1< double, Eigen::Matrix< double, Eigen::Dynamic, 1, 0, BIG_N, 1 > >(const assembler::NonLinearAssemblerData &)> &funBigN, const std::function< DScalar1< double, Eigen::VectorXd >(const assembler::NonLinearAssemblerData &)> &funn)
Automatic differentiation scalar with first-order derivatives.
Automatic differentiation scalar with first- and second-order derivatives.
static void setVariableCount(size_t value)
Set the independent variable count used by the automatic differentiation layer.