Merge pull request #2444 from SCIInstitute/2443-ffc-fill

Address #2443 - Studio: Add ability to fill all (inclusive or exclusive) for FFC painting

Author: akenmorris

Libs/Analyze/StudioMesh.cpp

@@ -137,6 +137,14 @@ void StudioMesh::paint_ffc(double world_pos[], double radius, bool inclusive) {
   scalars->Modified();
 }
 
+//---------------------------------------------------------------------------
+void StudioMesh::fill_ffc(bool inclusive) {
+  auto scalars = get_or_create_array(FFC_PAINT, 1.0);
+  float value = inclusive ? 1.0f : 0.0f;
+  scalars->FillComponent(0, value);
+  scalars->Modified();
+}
+
 //---------------------------------------------------------------------------
 bool StudioMesh::has_ffc_paint() {
   auto result = poly_data_->GetPointData()->GetArray(FFC_PAINT);

Libs/Analyze/StudioMesh.h

@@ -66,6 +66,9 @@ class StudioMesh {
   //! Paint free form constraint
   void paint_ffc(double world_pos[], double radius, bool inclusive);
 
+  //! Fill all vertices with specified inclusion/exclusion value
+  void fill_ffc(bool inclusive);
+
   //! Does this mesh have free form constraint paint?
   bool has_ffc_paint();
 

Studio/Data/DataTool.cpp

@@ -146,6 +146,7 @@ void DataTool::set_session(QSharedPointer<Session> session) {
   connect(ui_->ffc_brush_size_, &CustomSlider::valueChanged, session.data(), &Session::set_ffc_paint_size);
   connect(ui_->ffc_included_mode_, &QRadioButton::toggled, session.data(), &Session::set_ffc_paint_mode_inclusive);
   connect(ui_->ffc_active_, &QCheckBox::toggled, session.data(), &Session::set_ffc_paint_active);
+  connect(ui_->ffc_fill_all_button_, &QPushButton::clicked, session.data(), &Session::trigger_fill_all_ffc);
 
   update_table();
   handle_landmark_mode_changed();

Studio/Data/DataTool.ui

@@ -933,6 +933,16 @@ QWidget#specificity_panel {
                  </property>
                 </widget>
                </item>
+               <item row="3" column="0" colspan="3">
+                <widget class="QPushButton" name="ffc_fill_all_button_">
+                 <property name="text">
+                  <string>Fill All (Visible Shapes)</string>
+                 </property>
+                 <property name="toolTip">
+                  <string>Fill entire mesh of currently displayed shape(s) with selected paint type (Included/Excluded)</string>
+                 </property>
+                </widget>
+               </item>
               </layout>
              </widget>
             </item>

Studio/Data/Session.cpp

@@ -1004,6 +1004,9 @@ void Session::trigger_save() { Q_EMIT save(); }
 //---------------------------------------------------------------------------
 void Session::trigger_data_changed() { Q_EMIT data_changed(); }
 
+//---------------------------------------------------------------------------
+void Session::trigger_fill_all_ffc() { Q_EMIT fill_all_ffc_requested(); }
+
 //---------------------------------------------------------------------------
 void Session::reload_particles() {
   int num_subjects = project_->get_number_of_subjects();

Studio/Data/Session.h

@@ -172,6 +172,7 @@ class Session : public QObject, public QEnableSharedFromThis<Session> {
   void trigger_save();
   void trigger_data_changed();
   void reload_particles();
+  void trigger_fill_all_ffc();
 
   void set_active_landmark_domain(int id);
   int get_active_landmark_domain();
@@ -323,6 +324,7 @@ class Session : public QObject, public QEnableSharedFromThis<Session> {
   void save();
   void session_title_changed();
   void image_name_changed();
+  void fill_all_ffc_requested();
 
  public:
   // constants

Studio/Visualization/Viewer.cpp

@@ -594,6 +594,31 @@ void Viewer::handle_paint(double display_pos[2], double world_pos[3]) {
   session_->set_modified(true);
 }
 
+//-----------------------------------------------------------------------------
+void Viewer::fill_all_ffc() {
+  // fill all vertices of all meshes with the current paint mode (included/excluded)
+  if (!meshes_.valid()) {
+    return;
+  }
+
+  bool inclusive = session_->get_ffc_paint_mode_inclusive();
+
+  for (int domain = 0; domain < meshes_.meshes().size(); domain++) {
+    auto mesh = meshes_.meshes()[domain];
+    mesh->fill_ffc(inclusive);
+
+    auto& ffc = shape_->get_constraints(domain).getFreeformConstraint();
+    ffc.setDefinition(mesh->get_poly_data());
+    if (!ffc.isSet()) {
+      ffc.setPainted(true);
+      session_->trigger_ffc_changed();
+    }
+  }
+
+  session_->set_modified(true);
+  session_->trigger_repaint();
+}
+
 //-----------------------------------------------------------------------------
 void Viewer::update_landmarks() { landmark_widget_->update_landmarks(); }
 

Studio/Visualization/Viewer.h

@@ -145,6 +145,8 @@ class Viewer {
 
   void handle_paint(double display_pos[2], double world_pos[3]);
 
+  void fill_all_ffc();
+
   static bool is_reverse(vtkSmartPointer<vtkTransform> transform);
 
   void update_actors();

Studio/Visualization/Visualizer.cpp

@@ -53,6 +53,15 @@ void Visualizer::set_session(SessionHandle session) {
   connect(session_.data(), &Session::paint_mode_changed, this, &Visualizer::update_paint_mode);
   connect(session_.data(), &Session::repaint, this, &Visualizer::redraw);
   connect(session_.data(), &Session::annotations_changed, this, &Visualizer::update_annotations);
+  connect(session_.data(), &Session::fill_all_ffc_requested, this, &Visualizer::handle_fill_all_ffc);
+}
+
+//-----------------------------------------------------------------------------
+void Visualizer::handle_fill_all_ffc() {
+  Q_FOREACH (ViewerHandle viewer, lightbox_->get_viewers()) {
+    viewer->fill_all_ffc();
+  }
+  lightbox_->redraw();
 }
 
 //-----------------------------------------------------------------------------

Studio/Visualization/Visualizer.h

@@ -56,6 +56,7 @@ class Visualizer : public QObject {
   void update_landmarks();
   void update_planes();
   void update_paint_mode();
+  void handle_fill_all_ffc();
 
   void display_sample(int i);