Refactoring of curve fitting algorithm:
removal of artificial extension at the ends of the curve removal of observation points normalization added clamping of parameter index which compensates for under-represented spline segments added parameter for level of freedom at the ends of the curve
This commit is contained in:
parent
c19770189f
commit
396d3215bd
@ -1199,9 +1199,9 @@ void SeamPlacer::align_seam_points(const PrintObject *po, const SeamPlacerImpl::
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Curve Fitting
|
// Curve Fitting
|
||||||
size_t number_of_splines = std::max(size_t(1),
|
size_t number_of_segments = std::max(size_t(1),
|
||||||
size_t(observations.size() / SeamPlacer::seam_align_seams_per_spline));
|
size_t(observations.size() / SeamPlacer::seam_align_seams_per_segment));
|
||||||
auto curve = Geometry::fit_cubic_bspline(observations, observation_points, weights, number_of_splines);
|
auto curve = Geometry::fit_cubic_bspline(observations, observation_points, weights, number_of_segments);
|
||||||
|
|
||||||
// Do alignment - compute fitted point for each point in the string from its Z coord, and store the position into
|
// Do alignment - compute fitted point for each point in the string from its Z coord, and store the position into
|
||||||
// Perimeter structure of the point; also set flag aligned to true
|
// Perimeter structure of the point; also set flag aligned to true
|
||||||
|
@ -118,7 +118,7 @@ public:
|
|||||||
// minimum number of seams needed in cluster to make alignemnt happen
|
// minimum number of seams needed in cluster to make alignemnt happen
|
||||||
static constexpr size_t seam_align_minimum_string_seams = 6;
|
static constexpr size_t seam_align_minimum_string_seams = 6;
|
||||||
// points covered by spline; determines number of splines for the given string
|
// points covered by spline; determines number of splines for the given string
|
||||||
static constexpr size_t seam_align_seams_per_spline = 30;
|
static constexpr size_t seam_align_seams_per_segment = 8;
|
||||||
|
|
||||||
//The following data structures hold all perimeter points for all PrintObject. The structure is as follows:
|
//The following data structures hold all perimeter points for all PrintObject. The structure is as follows:
|
||||||
// Map of PrintObjects (PO) -> vector of layers of PO -> vector of perimeter points of the given layer
|
// Map of PrintObjects (PO) -> vector of layers of PO -> vector of perimeter points of the given layer
|
||||||
|
@ -6,6 +6,8 @@
|
|||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
|
||||||
|
//#define LSQR_DEBUG
|
||||||
|
|
||||||
namespace Slic3r {
|
namespace Slic3r {
|
||||||
namespace Geometry {
|
namespace Geometry {
|
||||||
|
|
||||||
@ -62,33 +64,24 @@ struct PiecewiseFittedCurve {
|
|||||||
|
|
||||||
Eigen::MatrixXf coefficients;
|
Eigen::MatrixXf coefficients;
|
||||||
NumberType start;
|
NumberType start;
|
||||||
NumberType length;
|
NumberType segment_size;
|
||||||
NumberType n_segment_size;
|
size_t endpoints_level_of_freedom;
|
||||||
|
|
||||||
size_t segments() const { return this->coefficients.cols(); }
|
|
||||||
|
|
||||||
NumberType get_n_segment_start(int segment_index) const {
|
|
||||||
return n_segment_size * segment_index;
|
|
||||||
}
|
|
||||||
|
|
||||||
NumberType normalize(const NumberType &observation_point) const {
|
|
||||||
return (observation_point - start) / length;
|
|
||||||
}
|
|
||||||
|
|
||||||
Vec<Dimension, NumberType> get_fitted_value(const NumberType &observation_point) const {
|
Vec<Dimension, NumberType> get_fitted_value(const NumberType &observation_point) const {
|
||||||
Vec<Dimension, NumberType> result = Vec<Dimension, NumberType>::Zero();
|
Vec<Dimension, NumberType> result = Vec<Dimension, NumberType>::Zero();
|
||||||
NumberType t = normalize(observation_point);
|
|
||||||
|
|
||||||
int start_segment_idx = int(floor(t / this->n_segment_size)) - int(Kernel::kernel_span * 0.5f - 1.0f);
|
//find corresponding segment index; expects kernels to be centered; NOTE: shift by number of additional segments, which are outside of the valid length
|
||||||
|
int middle_right_segment_index = floor((observation_point - start) / segment_size);
|
||||||
|
//find index of first segment that is affected by the point i; this can be deduced from kernel_span
|
||||||
|
int start_segment_idx = middle_right_segment_index - Kernel::kernel_span / 2 + 1;
|
||||||
for (int segment_index = start_segment_idx; segment_index < int(start_segment_idx + Kernel::kernel_span);
|
for (int segment_index = start_segment_idx; segment_index < int(start_segment_idx + Kernel::kernel_span);
|
||||||
segment_index++) {
|
segment_index++) {
|
||||||
if (segment_index < 0 || segment_index >= int(this->coefficients.cols())) {
|
NumberType segment_start = start + segment_index * segment_size;
|
||||||
continue;
|
NumberType normalized_segment_distance = (segment_start - observation_point) / segment_size;
|
||||||
}
|
|
||||||
NumberType segment_start = this->get_n_segment_start(segment_index);
|
|
||||||
NumberType normalized_segment_distance = (segment_start - t) / this->n_segment_size;
|
|
||||||
|
|
||||||
result += Kernel::kernel(normalized_segment_distance) * coefficients.col(segment_index);
|
int parameter_index = segment_index + endpoints_level_of_freedom;
|
||||||
|
parameter_index = std::clamp(parameter_index, 0, int(coefficients.cols()) - 1);
|
||||||
|
result += Kernel::kernel(normalized_segment_distance) * coefficients.col(parameter_index);
|
||||||
}
|
}
|
||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
@ -106,96 +99,74 @@ PiecewiseFittedCurve<Dimension, NumberType, Kernel> fit_curve(
|
|||||||
const std::vector<Vec<Dimension, NumberType>> &observations,
|
const std::vector<Vec<Dimension, NumberType>> &observations,
|
||||||
const std::vector<NumberType> &observation_points,
|
const std::vector<NumberType> &observation_points,
|
||||||
const std::vector<NumberType> &weights,
|
const std::vector<NumberType> &weights,
|
||||||
size_t number_of_inner_splines) {
|
size_t segments_count,
|
||||||
|
size_t endpoints_level_of_freedom) {
|
||||||
|
|
||||||
// check to make sure inputs are correct
|
// check to make sure inputs are correct
|
||||||
assert(number_of_inner_splines >= 1);
|
assert(segments_count > 0);
|
||||||
|
assert(observations.size() > 0);
|
||||||
assert(observation_points.size() == observations.size());
|
assert(observation_points.size() == observations.size());
|
||||||
assert(observation_points.size() == weights.size());
|
assert(observation_points.size() == weights.size());
|
||||||
assert(number_of_inner_splines <= observations.size());
|
assert(segments_count <= observations.size());
|
||||||
assert(observations.size() >= 4);
|
|
||||||
|
|
||||||
size_t extremes_repetition = Kernel::kernel_span - 1; //how many (additional) times is the first and last point repeated
|
|
||||||
|
|
||||||
//prepare sqrt of weights, which will then be applied to both matrix T and observed data: https://en.wikipedia.org/wiki/Weighted_least_squares
|
//prepare sqrt of weights, which will then be applied to both matrix T and observed data: https://en.wikipedia.org/wiki/Weighted_least_squares
|
||||||
std::vector<NumberType> sqrt_weights(weights.size() + extremes_repetition * 2);
|
std::vector<NumberType> sqrt_weights(weights.size());
|
||||||
for (size_t index = 0; index < weights.size(); ++index) {
|
for (size_t index = 0; index < weights.size(); ++index) {
|
||||||
assert(weights[index] > 0);
|
assert(weights[index] > 0);
|
||||||
sqrt_weights[index + extremes_repetition] = sqrt(weights[index]);
|
sqrt_weights[index] = sqrt(weights[index]);
|
||||||
}
|
|
||||||
//repeat weights for addtional extreme segments
|
|
||||||
{
|
|
||||||
auto first = sqrt_weights[extremes_repetition];
|
|
||||||
auto last = sqrt_weights[extremes_repetition + weights.size() - 1];
|
|
||||||
for (int index = 0; index < int(extremes_repetition); ++index) {
|
|
||||||
sqrt_weights[index] = first;
|
|
||||||
sqrt_weights[sqrt_weights.size() - index - 1] = last;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// prepare result and compute metadata
|
// prepare result and compute metadata
|
||||||
PiecewiseFittedCurve<Dimension, NumberType, Kernel> result { };
|
PiecewiseFittedCurve<Dimension, NumberType, Kernel> result { };
|
||||||
|
|
||||||
NumberType orig_len = observation_points.back() - observation_points.front();
|
NumberType valid_length = observation_points.back() - observation_points.front();
|
||||||
NumberType orig_segment_size = orig_len / NumberType(number_of_inner_splines * Kernel::kernel_span);
|
NumberType segment_size = valid_length / NumberType(segments_count);
|
||||||
result.start = observation_points.front() - extremes_repetition * orig_segment_size;
|
result.start = observation_points.front();
|
||||||
result.length = observation_points.back() + extremes_repetition * orig_segment_size - result.start;
|
result.segment_size = segment_size;
|
||||||
size_t segments_count = number_of_inner_splines * Kernel::kernel_span + extremes_repetition * 2;
|
result.endpoints_level_of_freedom = endpoints_level_of_freedom;
|
||||||
result.n_segment_size = NumberType(1) / NumberType(segments_count - 1);
|
|
||||||
|
|
||||||
//normalize observations points by start and length
|
|
||||||
std::vector<NumberType> normalized_obs_points(observation_points.size() + extremes_repetition * 2);
|
|
||||||
for (size_t index = 0; index < observation_points.size(); ++index) {
|
|
||||||
normalized_obs_points[index + extremes_repetition] = result.normalize(observation_points[index]);
|
|
||||||
}
|
|
||||||
// create artificial values at the extremes for constant curve fitting
|
|
||||||
for (int index = 0; index < int(extremes_repetition); ++index) {
|
|
||||||
normalized_obs_points[extremes_repetition - 1 - index] = result.normalize(observation_points.front()
|
|
||||||
- index * orig_segment_size - NumberType(0.5) * orig_segment_size);
|
|
||||||
|
|
||||||
normalized_obs_points[normalized_obs_points.size() - extremes_repetition + index] = result.normalize(
|
|
||||||
observation_points.back() + index * orig_segment_size + NumberType(0.5) * orig_segment_size);
|
|
||||||
}
|
|
||||||
|
|
||||||
// prepare observed data
|
// prepare observed data
|
||||||
// Eigen defaults to column major memory layout.
|
// Eigen defaults to column major memory layout.
|
||||||
Eigen::MatrixXf data_points(Dimension, observations.size() + extremes_repetition * 2);
|
Eigen::MatrixXf data_points(Dimension, observations.size());
|
||||||
for (size_t index = 0; index < observations.size(); ++index) {
|
for (size_t index = 0; index < observations.size(); ++index) {
|
||||||
data_points.col(index + extremes_repetition) = observations[index]
|
data_points.col(index) = observations[index] * sqrt_weights[index];
|
||||||
* sqrt_weights[index + extremes_repetition];
|
|
||||||
}
|
|
||||||
//duplicate observed data at the extremes
|
|
||||||
for (int index = 0; index < int(extremes_repetition); index++) {
|
|
||||||
data_points.col(index) = observations.front() * sqrt_weights[index];
|
|
||||||
data_points.col(data_points.cols() - index - 1) = observations.back()
|
|
||||||
* sqrt_weights[data_points.cols() - index - 1];
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
size_t parameters_count = segments_count + 1 + 2 * endpoints_level_of_freedom;
|
||||||
//Create weight matrix T for each point and each segment;
|
//Create weight matrix T for each point and each segment;
|
||||||
Eigen::MatrixXf T(normalized_obs_points.size(), segments_count);
|
Eigen::MatrixXf T(observation_points.size(), parameters_count);
|
||||||
T.setZero();
|
T.setZero();
|
||||||
|
|
||||||
//Fill the weight matrix
|
//Fill the weight matrix
|
||||||
for (size_t i = 0; i < normalized_obs_points.size(); ++i) {
|
for (size_t i = 0; i < observation_points.size(); ++i) {
|
||||||
NumberType knot_val = normalized_obs_points[i];
|
NumberType observation_point = observation_points[i];
|
||||||
|
//find corresponding segment index; expects kernels to be centered
|
||||||
|
int middle_right_segment_index = floor((observation_point - result.start) / result.segment_size);
|
||||||
//find index of first segment that is affected by the point i; this can be deduced from kernel_span
|
//find index of first segment that is affected by the point i; this can be deduced from kernel_span
|
||||||
int start_segment_idx = int(floor(knot_val / result.n_segment_size)) - int(Kernel::kernel_span * 0.5f - 1.0f);
|
int start_segment_idx = middle_right_segment_index - Kernel::kernel_span / 2 + 1;
|
||||||
for (int segment_index = start_segment_idx; segment_index < int(start_segment_idx + Kernel::kernel_span);
|
for (int segment_index = start_segment_idx; segment_index < int(start_segment_idx + Kernel::kernel_span);
|
||||||
segment_index++) {
|
segment_index++) {
|
||||||
// skip if we overshoot segment_index - happens at the extremes
|
NumberType segment_start = result.start + segment_index * result.segment_size;
|
||||||
if (segment_index < 0 || segment_index >= int(segments_count)) {
|
NumberType normalized_segment_distance = (segment_start - observation_point) / result.segment_size;
|
||||||
continue;
|
|
||||||
}
|
|
||||||
NumberType segment_start = result.get_n_segment_start(segment_index);
|
|
||||||
NumberType normalized_segment_distance = (segment_start - knot_val) / result.n_segment_size;
|
|
||||||
// fill in kernel value with weight applied
|
|
||||||
|
|
||||||
T(i, segment_index) += Kernel::kernel(normalized_segment_distance) * sqrt_weights[i];
|
int parameter_index = segment_index + endpoints_level_of_freedom;
|
||||||
|
parameter_index = std::clamp(parameter_index, 0, int(parameters_count) - 1);
|
||||||
|
T(i, parameter_index) += Kernel::kernel(normalized_segment_distance) * sqrt_weights[i];
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#ifdef LSQR_DEBUG
|
||||||
|
std::cout << "weight matrix: " << std::endl;
|
||||||
|
for (int obs = 0; obs < observation_points.size(); ++obs) {
|
||||||
|
std::cout << std::endl;
|
||||||
|
for (int segment = 0; segment < parameters_count; ++segment) {
|
||||||
|
std::cout << T(obs, segment) << " ";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
std::cout << std::endl;
|
||||||
|
#endif
|
||||||
|
|
||||||
// Solve for linear least square fit
|
// Solve for linear least square fit
|
||||||
result.coefficients.resize(Dimension, segments_count);
|
result.coefficients.resize(Dimension, parameters_count);
|
||||||
const auto QR = T.fullPivHouseholderQr();
|
const auto QR = T.fullPivHouseholderQr();
|
||||||
for (size_t dim = 0; dim < Dimension; ++dim) {
|
for (size_t dim = 0; dim < Dimension; ++dim) {
|
||||||
result.coefficients.row(dim) = QR.solve(data_points.row(dim).transpose());
|
result.coefficients.row(dim) = QR.solve(data_points.row(dim).transpose());
|
||||||
@ -210,8 +181,10 @@ fit_cubic_bspline(
|
|||||||
const std::vector<Vec<Dimension, NumberType>> &observations,
|
const std::vector<Vec<Dimension, NumberType>> &observations,
|
||||||
std::vector<NumberType> observation_points,
|
std::vector<NumberType> observation_points,
|
||||||
std::vector<NumberType> weights,
|
std::vector<NumberType> weights,
|
||||||
size_t number_of_segments) {
|
size_t segments_count,
|
||||||
return fit_curve<CubicBSplineKernel<NumberType>>(observations, observation_points, weights, number_of_segments);
|
size_t endpoints_level_of_freedom = 0) {
|
||||||
|
return fit_curve<CubicBSplineKernel<NumberType>>(observations, observation_points, weights, segments_count,
|
||||||
|
endpoints_level_of_freedom);
|
||||||
}
|
}
|
||||||
|
|
||||||
template<int Dimension, typename NumberType>
|
template<int Dimension, typename NumberType>
|
||||||
@ -220,8 +193,10 @@ fit_catmul_rom_spline(
|
|||||||
const std::vector<Vec<Dimension, NumberType>> &observations,
|
const std::vector<Vec<Dimension, NumberType>> &observations,
|
||||||
std::vector<NumberType> observation_points,
|
std::vector<NumberType> observation_points,
|
||||||
std::vector<NumberType> weights,
|
std::vector<NumberType> weights,
|
||||||
size_t number_of_segments) {
|
size_t segments_count,
|
||||||
return fit_curveCubicCatmulRomKernel<NumberType>(observations, observation_points, weights, number_of_segments);
|
size_t endpoints_level_of_freedom = 0) {
|
||||||
|
return fit_curve<CubicCatmulRomKernel<NumberType>>(observations, observation_points, weights, segments_count,
|
||||||
|
endpoints_level_of_freedom);
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -104,8 +104,8 @@ template<typename Derived, typename Derived2>
|
|||||||
inline double angle(const Eigen::MatrixBase<Derived> &v1, const Eigen::MatrixBase<Derived2> &v2) {
|
inline double angle(const Eigen::MatrixBase<Derived> &v1, const Eigen::MatrixBase<Derived2> &v2) {
|
||||||
static_assert(Derived::IsVectorAtCompileTime && int(Derived::SizeAtCompileTime) == 2, "angle(): first parameter is not a 2D vector");
|
static_assert(Derived::IsVectorAtCompileTime && int(Derived::SizeAtCompileTime) == 2, "angle(): first parameter is not a 2D vector");
|
||||||
static_assert(Derived2::IsVectorAtCompileTime && int(Derived2::SizeAtCompileTime) == 2, "angle(): second parameter is not a 2D vector");
|
static_assert(Derived2::IsVectorAtCompileTime && int(Derived2::SizeAtCompileTime) == 2, "angle(): second parameter is not a 2D vector");
|
||||||
auto v1d = v1.typename cast<double>();
|
auto v1d = v1.template cast<double>();
|
||||||
auto v2d = v2.typename cast<double>();
|
auto v2d = v2.template cast<double>();
|
||||||
return atan2(cross2(v1d, v2d), v1d.dot(v2d));
|
return atan2(cross2(v1d, v2d), v1d.dot(v2d));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user