51 {
54 lowestEv.resize(matter->numberOfAtoms(), 3);
56
58 const int size = 3 * static_cast<int>(mobile.size());
59 if (size == 0) {
61 return;
62 }
63
64 const long maxIter = std::max(1L,
params.davidson_options().max_iterations);
65 const double tol =
params.davidson_options().tolerance;
66 const double dr =
params.main_options().finiteDifference;
68
69
72 const bool useDiagPrec =
params.davidson_options().diagonal_preconditioner;
73
76 V.setZero();
77 HV.setZero();
78
80 double beta = r.norm();
83 return;
84 }
85 r /= beta;
86
87 auto tmpMatter = std::make_unique<Matter>(*matter);
88 const long forceCallsStart = tmpMatter->getForceCalls();
89 const AtomMatrix pos0 = matter->getPositions();
90 const VectorXd force0 =
mobileForces(tmpMatter.get(), mobile);
91
92 auto applyH = [&](const VectorXd &v) -> VectorXd {
93 auto at = [&](double scale) -> VectorXd {
96 pos);
97 tmpMatter->setPositions(pos);
99 };
101 };
102
103 VectorXd diagH = VectorXd::Ones(size);
104
105 V.col(0) = r;
106 HV.col(0) = applyH(V.col(0));
107 if (useDiagPrec) {
108 for (int k = 0; k < size; ++k) {
109 const double vk = V(k, 0);
110 if (std::fabs(vk) > 1e-8) {
111 diagH(k) = std::max(std::fabs(HV(k, 0) / vk), 1e-3);
112 }
113 }
114 }
115
116 double ew = 0.0, ewOld = 0.0;
117 VectorXd evEst = V.col(0);
118 VectorXd evOldEst = evEst;
119 int subspace = 1;
120
121 for (int iter = 0; iter < maxIter; ++iter) {
124
125 MatrixXd G = V.leftCols(subspace).transpose() * HV.leftCols(subspace);
126 G = 0.5 * (G + G.transpose());
127
128 Eigen::SelfAdjointEigenSolver<MatrixXd> es(G);
129 ew = es.eigenvalues()(0);
130 VectorXd y = es.eigenvectors().col(0);
131 evEst = V.leftCols(subspace) * y;
132 eonc::safemath::safe_normalize_inplace(evEst);
133
134 VectorXd Hx = HV.leftCols(subspace) * y;
135 VectorXd resid = Hx - ew * evEst;
136 const double residNorm = resid.norm();
137 const double ewAbsRelErr =
138 (iter == 0) ? 1.0
139 : eonc::safemath::
safe_div(std::fabs(ew - ewOld),
140 std::fabs(ewOld), 1.0);
141 ewOld = ew;
142 statsTorque = std::max(ewAbsRelErr, residNorm / (std::fabs(ew) + 1e-12));
145 evOldEst = evEst;
146
148 "[Davidson] ew={:10.6f} rel_err={:10.6f} |r|={:10.6f} "
149 "angle={:7.3f} dim={:3d} iter={:3d} n_mobile={}",
150 ew, ewAbsRelErr, residNorm,
statsAngle, subspace, iter,
151 mobile.size());
152
153 if (ewAbsRelErr < tol && residNorm < tol * (std::fabs(ew) + 1.0)) {
154 QUILL_LOG_INFO(
log,
"[Davidson] Tolerance reached: {}", tol);
155 break;
156 }
157 if (subspace >= maxIter) {
158 QUILL_LOG_ERROR(
log,
"[Davidson] Max subspace dimension");
159 break;
160 }
161
162 VectorXd t(size);
163 if (useDiagPrec) {
164 for (int k = 0; k < size; ++k) {
165 const double denom = diagH(k) - ew;
166 t(k) = resid(k) / (std::fabs(denom) > 1e-12 ? denom : 1e-12);
167 }
168 } else {
169 t = resid;
170 }
171
172 for (int c = 0; c < subspace; ++c) {
173 t -= V.col(c).dot(t) * V.col(c);
174 }
175 const double tnorm = t.norm();
176 if (tnorm < 1e-14) {
177 QUILL_LOG_ERROR(
log,
"[Davidson] Linear dependence in residual");
178 break;
179 }
180 t /= tnorm;
181
182 V.col(subspace) = t;
183 HV.col(subspace) = applyH(t);
184 if (useDiagPrec) {
185 for (int k = 0; k < size; ++k) {
186 const double vk = t(k);
187 if (std::fabs(vk) > 1e-8) {
188 const double d = std::fabs(HV(k, subspace) / vk);
189 diagH(k) = std::max(diagH(k), std::max(d, 1e-3));
190 }
191 }
192 }
193 ++subspace;
194
195 if (iter >= maxIter - 1) {
196 QUILL_LOG_ERROR(
log,
"[Davidson] Max iterations");
197 break;
198 }
199 }
200
204 if (evEst.size() == size) {
206 }
207}
Eigen::Matrix< double, Eigen::Dynamic, Eigen::Dynamic, eOnStorageOrder > MatrixXd
Eigen::Matrix< double, Eigen::Dynamic, 3, eOnStorageOrder > AtomMatrix
constexpr double safe_div(double num, double denom, double fallback=0.0)
double safe_acos(double x)
FdScheme
Real finite-difference scheme for the assembled Hessian and for Lanczos/Davidson Hessian-vector produ...
VectorXd fdHessianVector(FdScheme scheme, double dr, const VectorXd &force0, Eval &&eval)
Energy Hessian-vector product -dF.
void unpackMobileRows(const VectorXd &packed, const VectorXi &mobile, AtomMatrix &full)
Write a 3*n_mobile vector into full AtomMatrix rows (other rows unchanged).
FdScheme parseFdScheme(std::string_view scheme)
VectorXd packMobileRows(const AtomMatrix &full, const VectorXi &mobile)
Pack full (n_atoms,3) rows of mobile atoms into a 3*n_mobile vector.
VectorXd mobileForces(Matter *matter, const VectorXi &mobile)
Force components on mobile atoms after Matter has a valid force cache (calls getForces under the hood...