Add ParticleShapeStatistics::project_new_sample and python API for it.

Author: akenmorris

Libs/Particles/ParticleShapeStatistics.cpp

@@ -497,7 +497,7 @@ ParticleShapeStatistics::ParticleShapeStatistics(std::shared_ptr<Project> projec
 
 //---------------------------------------------------------------------------
 int ParticleShapeStatistics::do_pca(std::vector<std::vector<Point>> global_pts, int domainsPerShape) {
-  this->domains_per_shape_ = domainsPerShape;
+  domains_per_shape_ = domainsPerShape;
 
   // Assumes all the same size.
   num_samples_ = global_pts.size() / domains_per_shape_;
@@ -508,11 +508,6 @@ int ParticleShapeStatistics::do_pca(std::vector<std::vector<Point>> global_pts,
   mean_.resize(num_dimensions_);
   mean_.fill(0);
 
-  std::cout << "VDimension = " << values_per_particle_ << "-------------\n";
-  std::cout << "m_numSamples = " << num_samples_ << "-------------\n";
-  std::cout << "m_domainsPerShape = " << domains_per_shape_ << "-------------\n";
-  std::cout << "global_pts.size() = " << global_pts.size() << "-------------\n";
-
   // Compile the "meta shapes"
   for (unsigned int i = 0; i < num_samples_; i++) {
     for (unsigned int k = 0; k < domains_per_shape_; k++) {
@@ -638,6 +633,13 @@ int ParticleShapeStatistics::principal_component_projections() {
   return 0;
 }
 
+//---------------------------------------------------------------------------
+Eigen::VectorXd ParticleShapeStatistics::project_new_sample(const Eigen::VectorXd& new_sample) {
+  // Subtract mean and project onto principal components
+  Eigen::VectorXd centered_sample = new_sample - mean_;
+  return eigenvectors_.transpose() * centered_sample;
+}
+
 //---------------------------------------------------------------------------
 int ParticleShapeStatistics::write_csv_file(const std::string& s) {
   // Write csv file

Libs/Particles/ParticleShapeStatistics.h

@@ -57,6 +57,9 @@ class ParticleShapeStatistics {
   //!  principal componenent axes for each of the samples.  ComputeModes must be called first.
   int principal_component_projections();
 
+  //! Projects a new sample into the PCA space defined by the original samples.
+  Eigen::VectorXd project_new_sample(const Eigen::VectorXd& new_sample);
+
   //! Returns the sample size
   int get_num_samples() const { return num_samples_; }
   int get_group1_num_samples() const { return num_samples_group1_; }

Libs/Python/ShapeworksPython.cpp

@@ -1302,7 +1302,11 @@ PYBIND11_MODULE(shapeworks_py, m) {
        components are constructed")
 
       .def("percentVarByMode", &ParticleShapeStatistics::get_percent_variance_by_mode,
-           "return the variance accounted for by the principal components");
+           "return the variance accounted for by the principal components")
+
+      .def("projectNewSample", &ParticleShapeStatistics::project_new_sample,
+           "project a new sample into the PCA space", "newSample"_a)
+      ;
 
   define_python_analyze(m);
   define_python_groom(m);