Browse Source

Optimize LSF for size, efficiency (#21443)

vanilla_fb_2.0.x
Ramiro Polla 4 years ago
committed by Scott Lahteine
parent
commit
3d70b645b6
  1. 26
      Marlin/src/libs/least_squares_fit.cpp
  2. 4
      Marlin/src/libs/least_squares_fit.h

26
Marlin/src/libs/least_squares_fit.cpp

@ -46,23 +46,23 @@ int finish_incremental_LSF(struct linear_fit_data *lsf) {
if (N == 0.0) if (N == 0.0)
return 1; return 1;
lsf->xbar /= N; const float RN = 1.0f / N,
lsf->ybar /= N; xbar = lsf->xbar * RN,
lsf->zbar /= N; ybar = lsf->ybar * RN,
lsf->x2bar = lsf->x2bar / N - sq(lsf->xbar); zbar = lsf->zbar * RN,
lsf->y2bar = lsf->y2bar / N - sq(lsf->ybar); x2bar = lsf->x2bar * RN - sq(xbar),
lsf->z2bar = lsf->z2bar / N - sq(lsf->zbar); y2bar = lsf->y2bar * RN - sq(ybar),
lsf->xybar = lsf->xybar / N - lsf->xbar * lsf->ybar; xybar = lsf->xybar * RN - xbar * ybar,
lsf->yzbar = lsf->yzbar / N - lsf->ybar * lsf->zbar; yzbar = lsf->yzbar * RN - ybar * zbar,
lsf->xzbar = lsf->xzbar / N - lsf->xbar * lsf->zbar; xzbar = lsf->xzbar * RN - xbar * zbar,
const float DD = lsf->x2bar * lsf->y2bar - sq(lsf->xybar); DD = x2bar * y2bar - sq(xybar);
if (ABS(DD) <= 1e-10 * (lsf->max_absx + lsf->max_absy)) if (ABS(DD) <= 1e-10 * (lsf->max_absx + lsf->max_absy))
return 1; return 1;
lsf->A = (lsf->yzbar * lsf->xybar - lsf->xzbar * lsf->y2bar) / DD; lsf->A = (yzbar * xybar - xzbar * y2bar) / DD;
lsf->B = (lsf->xzbar * lsf->xybar - lsf->yzbar * lsf->x2bar) / DD; lsf->B = (xzbar * xybar - yzbar * x2bar) / DD;
lsf->D = -(lsf->zbar + lsf->A * lsf->xbar + lsf->B * lsf->ybar); lsf->D = -(zbar + lsf->A * xbar + lsf->B * ybar);
return 0; return 0;
} }

4
Marlin/src/libs/least_squares_fit.h

@ -37,7 +37,7 @@
struct linear_fit_data { struct linear_fit_data {
float xbar, ybar, zbar, float xbar, ybar, zbar,
x2bar, y2bar, z2bar, x2bar, y2bar,
xybar, xzbar, yzbar, xybar, xzbar, yzbar,
max_absx, max_absy, max_absx, max_absy,
A, B, D, N; A, B, D, N;
@ -56,7 +56,6 @@ inline void incremental_WLSF(struct linear_fit_data *lsf, const float &x, const
lsf->zbar += wz; lsf->zbar += wz;
lsf->x2bar += wx * x; lsf->x2bar += wx * x;
lsf->y2bar += wy * y; lsf->y2bar += wy * y;
lsf->z2bar += wz * z;
lsf->xybar += wx * y; lsf->xybar += wx * y;
lsf->xzbar += wx * z; lsf->xzbar += wx * z;
lsf->yzbar += wy * z; lsf->yzbar += wy * z;
@ -74,7 +73,6 @@ inline void incremental_LSF(struct linear_fit_data *lsf, const float &x, const f
lsf->zbar += z; lsf->zbar += z;
lsf->x2bar += sq(x); lsf->x2bar += sq(x);
lsf->y2bar += sq(y); lsf->y2bar += sq(y);
lsf->z2bar += sq(z);
lsf->xybar += x * y; lsf->xybar += x * y;
lsf->xzbar += x * z; lsf->xzbar += x * z;
lsf->yzbar += y * z; lsf->yzbar += y * z;

Loading…
Cancel
Save