[node] sfmFilter : Add new removeUnselectedPoses option

Author: Alxiice

meshroom/aliceVision/SfMFilter.py

@@ -30,6 +30,12 @@ class SfMFilter(desc.CommandLineNode):
                         r'".*\/(.*?)_.*\.\w{3}"',
             value=r'.*\/(.*?)\.\w{3}',
         ),
+        desc.BoolParam(
+            name="prunePoses",
+            label="Remove unselected poses",
+            description="Set to True to prune poses related to views that are not selected.",
+            value=False,
+        ),
         desc.ChoiceParam(
             name="verboseLevel",
             label="Verbose Level",

src/software/utils/main_sfmFilter.cpp

@@ -37,6 +37,7 @@ int aliceVision_main(int argc, char** argv)
     std::string outputSfMFilenameSelected;
     std::string outputSfMFilenameUnselected;
     std::string fileMatchingPattern;
+    bool prunePoses;
 
     // clang-format off
     po::options_description requiredParams("Required parameters");
@@ -45,6 +46,7 @@ int aliceVision_main(int argc, char** argv)
          "Path to the input SfMData file.\n")
         ("fileMatchingPattern,m", po::value<std::string>(&fileMatchingPattern)->required(),
          "Matching pattern for the from_filepath method.\n")
+        ("prunePoses,p", po::value<bool>(&prunePoses)->required(), "Prune unselected poses.\n")
         ("outputSfMData_selected,o", po::value<std::string>(&outputSfMFilenameSelected)->required(),
          "Path to the output SfMData file.\n")
          ("outputSfMData_unselected,o", po::value<std::string>(&outputSfMFilenameUnselected)->required(),
@@ -69,6 +71,7 @@ int aliceVision_main(int argc, char** argv)
 
     std::regex re(fileMatchingPattern);
     std::vector<IndexT> selectedViews;
+    std::vector<IndexT> selectedPoses;
 
     for (auto& viewIt : sfmData.getViews())
     {
@@ -77,9 +80,13 @@ int aliceVision_main(int argc, char** argv)
         if (std::regex_search(imagePath, matches, re))
         {
             selectedViews.push_back(viewIt.first);
+            if (prunePoses)
+                selectedPoses.push_back(viewIt.second->getPoseId());
         }
     }
 
+    // Split views into the selected and unselected SfmData
+
     sfmData::SfMData outputSfMData_selected = sfmData;
     sfmData::SfMData outputSfMData_unselected = sfmData;
     std::set<IndexT> viewIdsToRemove;
@@ -109,6 +116,38 @@ int aliceVision_main(int argc, char** argv)
         outputSfMData_unselected.getViews().erase(r);
     }
 
+    // Split poses into the selected and unselected SfmData
+
+    if (prunePoses)
+    {
+        std::set<IndexT> poseIdsToRemove;
+        std::set<IndexT> poseIdsToKeep;
+
+        for (auto& viewIt : outputSfMData_selected.getPoses())
+        {
+            const IndexT viewId = viewIt.first;
+            auto it = std::find(selectedPoses.begin(), selectedPoses.end(), viewId);
+            if (it == selectedPoses.end())
+            {
+                poseIdsToRemove.insert(viewId);
+            }
+            else
+            {
+                poseIdsToKeep.insert(viewId);
+            }
+        }
+
+        for (auto r : poseIdsToRemove)
+        {
+            outputSfMData_selected.getPoses().erase(r);
+        }
+
+        for (auto r : poseIdsToKeep)
+        {
+            outputSfMData_unselected.getPoses().erase(r);
+        }
+    }
+
 
     ALICEVISION_LOG_INFO("Save into '" << outputSfMFilenameSelected << "'");
     // Export the SfMData scene in the expected format