Skip to content

Libs/Particles/EvaluationUtil.h

Namespaces

Name
shapeworks
User usage reporting (telemetry)

Classes

Name
struct shapeworks::MultiVariateNormalRandom
struct shapeworks::Reconstruction

Source code

```cpp

pragma once

include

include

include

include

namespace shapeworks { struct MultiVariateNormalRandom { Eigen::VectorXd mean; Eigen::MatrixXd transform;

// seed set as constant 42 for test repeatability boost::mt19937 gen{42}; boost::normal_distribution<> dist;

MultiVariateNormalRandom(Eigen::MatrixXd const &covar) : MultiVariateNormalRandom(Eigen::VectorXd::Zero(covar.rows()), covar) {}

MultiVariateNormalRandom(Eigen::VectorXd const &mean, Eigen::MatrixXd const &covar) : mean(mean) { Eigen::SelfAdjointEigenSolver eigenSolver(covar); transform = eigenSolver.eigenvectors() * eigenSolver.eigenvalues().cwiseSqrt().asDiagonal(); }

Eigen::MatrixXd operator()() { return mean + transform * Eigen::VectorXd{mean.size()}.unaryExpr(& { return dist(gen); }); }

};

struct Reconstruction { double dist; int shapeIdx; Eigen::MatrixXd rec; };

// Sorts the reconstructions in place according to dist and saves them to the specified folder. // This generates XML files which can be opened in ShapeWorksStudio void SaveReconstructions(std::vector &reconstructions, const std::vector &srcPaths, const std::string &saveTo) { std::sort(reconstructions.begin(), reconstructions.end(), { return l.dist < r.dist; });

for (int i = 0; i < reconstructions.size(); i++) { const int percentile = i == reconstructions.size() - 1 ? 100 : std::floor(((double) i / reconstructions.size()) * 100.0);

// Save the reconstruction
const std::string recPath = saveTo + "/" + std::to_string(percentile) + "perc.particles";
std::ofstream recOF(recPath);
if (!recOF) { throw std::runtime_error("Unable to open file: " + recPath); }
recOF << reconstructions[i].rec << std::endl;
recOF.close();

// Create an XML file
const std::string xmlPath = saveTo + "/" + std::to_string(percentile) + "perc.xml";
std::ofstream xmlOF(xmlPath);
if (!xmlOF) { throw std::runtime_error("Unable to open file: " + xmlPath); }

xmlOF << "<point_files>"
      << srcPaths[reconstructions[i].shapeIdx] << std::endl << recPath
      << "</point_files>" << std::endl
      << "<group_ids>"
      << 1 << std::endl << 2
      << "</group_ids>";
xmlOF.close();

} } } ```

Updated on 2026-03-31 at 16:02:11 +0000