new changes

Author: nawazishkhan1-nk

Libs/Particles/ParticleShapeStatistics.h

@@ -135,6 +135,9 @@ class ParticleShapeStatistics {
   //! Set the meshes for each sample (used for some evaluation metrics)
   void set_meshes(const std::vector<Mesh>& meshes) { meshes_ = meshes; }
 
+  // import estimated parameters for regression
+  inline bool import_regression_parameters(Eigen::VectorXd slope, Eigen::VectorXd intercept) { slope_ = slope; intercept_ = intercept; };
+  
  private:
   unsigned int num_samples_group1_;
   unsigned int num_samples_group2_;
@@ -147,9 +150,14 @@ class ParticleShapeStatistics {
   std::vector<double> eigenvalues_;
   Eigen::VectorXd mean_;
   Eigen::VectorXd mean1_;
-  Eigen::VectorXd mean2_;
+  Eigen::VectorXd mean2_;x
   Eigen::MatrixXd points_minus_mean_;
   Eigen::MatrixXd shapes_;
+  
+  // for regression tasks
+  Eigen::VectorXd slope_;
+  Eigen::VectorXd intercept_;
+  bool regression_enabled_;
 
   std::vector<double> percent_variance_by_mode_;
   Eigen::MatrixXd principals_;

Libs/Project/Subject.h

@@ -123,7 +123,7 @@ class Subject {
   std::string display_name_;
   bool fixed_ = false;
   bool excluded_ = false;
-  double explanatory_variable_ = 0.0;
+  double explanatory_variable_ = std::numeric_limits<double>::lowest();
   StringList original_filenames_;
   StringList groomed_filenames_;
   StringList local_particle_filenames_;

Studio/Analysis/AnalysisTool.cpp

@@ -269,6 +269,11 @@ void AnalysisTool::on_reconstructionButton_clicked() {
 //---------------------------------------------------------------------------
 int AnalysisTool::get_pca_mode() { return ui_->pcaModeSpinBox->value() - 1; }
 
+//---------------------------------------------------------------------------
+bool AnalysisTool::get_regression_analysis_status() {
+  return ui_->enableRegressionCheckBox->isChecked();
+}
+
 //---------------------------------------------------------------------------
 double AnalysisTool::get_group_ratio() {
   double group_slider_value = ui_->group_slider->value();
@@ -505,6 +510,35 @@ void AnalysisTool::network_analysis_clicked() {
   app_->get_py_worker()->run_job(network_analysis_job_);
 }
 
+Eigen::VectorXd load_regression_parameters(std::string filepath) {
+  std::ifstream infile(slope_file_path);
+  if (!infile.good()) {
+    throw std::runtime_error("Unable to open regression parameter file: \"" +
+                             filepath + "\" for reading");
+  }
+  try {
+    std::vector<double> temp_values;
+    double value;
+    while (infile >> value) {
+      temp_values.push_back(value);
+    }
+    if (temp_values.empty()) {
+      std::cerr << "Error: No data found in file " << slope_file_path
+                << std::endl;
+      return Eigen::VectorXd();
+    }
+    Eigen::VectorXd param_vector(temp_values.size());
+    for (std::size_t i = 0; i < temp_values.size(); ++i) {
+      param_vector[i] = temp_values[i];
+    }
+    return param_vector;
+
+  } catch (json::exception& e) {
+    throw std::runtime_error("Unabled to parse regression parameter file " +
+                             filepath + " : " + e.what());
+  }
+}
+
 //-----------------------------------------------------------------------------
 bool AnalysisTool::compute_stats() {
   if (stats_ready_) {
@@ -627,6 +661,20 @@ bool AnalysisTool::compute_stats() {
     compute_shape_evaluations();
   }
 
+  can_run_regression_ = check_explanatory_variable_limits();
+  if (can_run_regression_) {
+    auto slope = load_regression_parameters(
+        session_->get_regression_param_file("slope"));
+    auto intercept = load_regression_parameters(
+        session_->get_regression_param_file("intercept"));
+    stats_.import_regression_parameters(slope, intercept);
+  }
+  else {
+    ui_->regression_groupbox->setVisible(false);
+    ui_->explanatoryVariableSlider->setVisible(false);
+    ui_->enableRegressionCheckBox->setVisible(false);
+  }
+
   stats_ready_ = true;
 
   ///  Set this to true to export long format sample data (e.g. for import into R)
@@ -667,6 +715,19 @@ bool AnalysisTool::compute_stats() {
   return true;
 }
 
+bool check_explanatory_variable_limits() {
+  auto subjects = session_->get_project()->get_subjects();
+  explanatory_variable_limits_.resize(2);
+  explanatory_variable_limits_[0] = std::numeric_limits<double>::max();
+  explanatory_variable_limits_[1] = std::numeric_limits<double>::lowest();
+  for (auto sub : subjects) {
+    double exp_val = sub->get_explanatory_variable();
+    if (exp_val == std::numeric_limits<double>::lowest()) return false;
+    explanatory_variable_limits_[0] = std::min(explanatory_variable_limits_[0], exp_val);
+    explanatory_variable_limits_[1] = std::max(explanatory_variable_limits_[1], exp_val);
+  }
+  return true;
+}
 //-----------------------------------------------------------------------------
 Particles AnalysisTool::get_mean_shape_points() {
   if (!compute_stats()) {
@@ -729,8 +790,8 @@ Particles AnalysisTool::get_shape_points(int mode, double value) {
     ui_->explained_variance->setText("");
     ui_->cumulative_explained_variance->setText("");
   }
-
-  temp_shape_ = stats_.get_mean() + (e * (value * lambda));
+  auto mean = !regression_enabled_ ? stats_.get_mean() : stats_.get_regression_mean(ui_->get_explanatory_variable_value());
+  temp_shape_ = mean + (e * (value * lambda));
 
   auto positions = temp_shape_;
 
@@ -829,11 +890,17 @@ ShapeHandle AnalysisTool::get_current_shape() {
   int pca_mode = get_pca_mode();
   double pca_value = get_pca_value();
   auto mca_level = get_mca_level();
-  if (mca_level == AnalysisTool::McaMode::Vanilla) {
-    return get_mode_shape(pca_mode, pca_value);
+  bool regression_analysis_enabled = get_regression_analysis_status();
+  if (!regression_analysis_enabled) {
+    if (mca_level == AnalysisTool::McaMode::Vanilla) {
+      return get_mode_shape(pca_mode, pca_value);
+    } else {
+      return get_mca_mode_shape(pca_mode, pca_value, mca_level);
+    }
   } else {
-    return get_mca_mode_shape(pca_mode, pca_value, mca_level);
+
   }
+
 }
 
 //---------------------------------------------------------------------------
@@ -1095,6 +1162,13 @@ double AnalysisTool::get_pca_value() {
   return value;
 }
 
+
+double AnalysisTool::get_explanatory_variable_value() {
+  int slider_value = ui_->explanatoryVariableSlider->value();
+  return t_min + (static_cast<double>(slider_value) / 100.0) * (t_max - t_min);
+
+}
+
 //---------------------------------------------------------------------------
 void AnalysisTool::pca_labels_changed(QString value, QString eigen, QString lambda) {
   set_labels(QString("pca"), value);

Studio/Analysis/AnalysisTool.h

@@ -80,6 +80,10 @@ class AnalysisTool : public QWidget {
   bool pca_animate();
   McaMode get_mca_level() const;
 
+  bool get_regression_analysis_status();
+
+  bool check_explanatory_variable_limits();
+
   int get_sample_number();
 
   bool compute_stats();
@@ -236,6 +240,8 @@ class AnalysisTool : public QWidget {
   void update_difference_particles();
 
   Eigen::VectorXd get_mean_shape_particles();
+  
+  Eigen::VectorXd load_regression_parameters(std::string filepath);
 
   ShapeHandle create_shape_from_points(Particles points);
 
@@ -275,6 +281,9 @@ class AnalysisTool : public QWidget {
 
   std::string feature_map_;
 
+  std::vector<double> explanatory_variable_limits_;
+  bool can_run_regression_;
+
   std::vector<std::string> current_group_names_;
   std::vector<std::string> current_group_values_;
 

Studio/Analysis/AnalysisTool.ui

@@ -1546,6 +1546,69 @@ QWidget#particles_panel {
                      </layout>
                     </widget>
                    </item>
+                   <item>
+                    <widget class="QGroupBox" name="regression_groupbox">
+                     <property name="title">
+                      <string>Regression Analysis</string>
+                     </property>
+                     <layout class="QGridLayout" name="gridLayout_regression">
+                      <property name="leftMargin">
+                       <number>0</number>
+                      </property>
+                      <property name="topMargin">
+                       <number>0</number>
+                      </property>
+                      <property name="rightMargin">
+                       <number>0</number>
+                      </property>
+                      <property name="bottomMargin">
+                       <number>0</number>
+                      </property>
+                      <item row="0" column="0">
+                       <widget class="QCheckBox" name="enableRegressionCheckBox">
+                        <property name="toolTip">
+                         <string>Enable regression analysis if explanatory variables are provided</string>
+                        </property>
+                        <property name="text">
+                         <string>Enable Regression Analysis</string>
+                        </property>
+                       </widget>
+                      </item>
+                      <item row="1" column="0">
+                        <widget class="QLabel" name="label_exp_slider">
+                         <property name="text">
+                          <string>Exp.Var.</string>
+                         </property>
+                        </widget>
+                      </item>
+                       <item row="1" column="1">
+                        <widget class="CustomSlider" name="explanatoryVariableSlider">
+                         <property name="enabled">
+                          <bool>false</bool>
+                         </property>
+                         <property name="toolTip">
+                          <string>Explanatory variable slider</string>
+                         </property>
+                         <property name="minimum">
+                          <number>-20</number>
+                         </property>
+                         <property name="maximum">
+                          <number>20</number>
+                         </property>
+                         <property name="orientation">
+                          <enum>Qt::Horizontal</enum>
+                         </property>
+                         <property name="tickPosition">
+                          <enum>QSlider::TicksBelow</enum>
+                         </property>
+                         <property name="tickInterval">
+                          <number>1</number>
+                         </property>
+                        </widget>
+                       </item>
+                     </layout>
+                    </widget>
+                   </item>
                    <item>
                     <layout class="QGridLayout" name="gridLayout_3">
                      <property name="verticalSpacing">

Studio/Data/Session.cpp

@@ -1418,4 +1418,19 @@ void Session::recompute_surfaces() {
   }
   Q_EMIT update_display();
 }
+
+std::string get_regression_param_file(std::string param_name) {
+    QFileInfo fileInfo(filename_);
+    QString baseName = fileInfo.completeBaseName();
+
+    QDir projectDir = fileInfo.absoluteDir();
+    QString particlesDir = baseName + "_particles";
+    QString paramFilePath = projectDir.filePath(particlesDir);
+    paramFilePath = QDir(paramFilePath).filePath(param_name);
+
+    if (!QFile::exists(paramFilePath)) {
+        return "";
+    }
+    return paramFilePath.toStdString();
+}
 }  // namespace shapeworks

Studio/Data/Session.h

@@ -339,6 +339,8 @@ class Session : public QObject, public QEnableSharedFromThis<Session> {
 
   void new_plane_point(PickResult result);
 
+  std::string get_regression_param_file(std::string param_name = "slope");
+
   QWidget* parent_{nullptr};
 
   Preferences& preferences_;