blob: 229411ae34031510735913d1d995f53add84abb3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
// Copyright Matt Overby 2020.
// Distributed under the MIT License.
#ifndef ADMMPD_SOLVER_H_
#define ADMMPD_SOLVER_H_
#include "admmpd_types.h"
namespace admmpd {
class Solver {
public:
// Initialies solver data. If a per-vertex
// variable is resized it is initialized to zero.
// Returns true on success
bool init(
const Eigen::MatrixXd &V, // vertices
const Eigen::MatrixXi &T, // tets
const Options *options,
Data *data);
// Solve a single time step.
// Returns number of iterations.
int solve(
const Options *options,
Data *data);
protected:
void update_constraints(
const Options *options,
Data *data);
void init_solve(
const Options *options,
Data *data);
void solve_local_step(
const Options *options,
Data *data);
// Global step with CG:
// 1/2||Ax-b||^2 + k/2||Kx-l||^2
void solve_conjugate_gradients(
const Options *options,
Data *data);
bool compute_matrices(
const Options *options,
Data *data);
void compute_masses(
const Options *options,
Data *data);
void append_energies(
const Options *options,
Data *data,
std::vector<Eigen::Triplet<double> > &D_triplets);
}; // class ADMMPD_solver
} // namespace admmpd
#endif // ADMMPD_SOLVER_H_
|