corrects mapping sensitivity error

waveygang · Oct 12, 2024 · 60b1f03 · 60b1f03
1 parent 64a2b47
commit 60b1f03
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/src/map/include/computeMap.hpp b/src/map/include/computeMap.hpp
@@ -780,9 +780,9 @@ namespace skch
             std::this_thread::sleep_for(std::chrono::milliseconds(10));
         }
 
-        if (param.mergeMappings && param.split) {
-            output->results = mergeMappingsInRange(output->results, param.chain_gap);
-        }
+        //if (param.mergeMappings && param.split) {
+        //    output->results = mergeMappingsInRange(output->results, param.chain_gap);
+        //}
 
         mappingBoundarySanityCheck(input, output->results);
 
@@ -817,7 +817,19 @@ namespace skch
 
       void processAggregatedMappings(const std::string& queryName, MappingResultsVector_t& mappings, std::ofstream& outstrm) {
           if (param.mergeMappings && param.split) {
-              filterMaximallyMerged(mappings, param);
+              auto maximallyMergedMappings = mergeMappingsInRange(mappings, param.chain_gap);
+              filterMaximallyMerged(maximallyMergedMappings, param);
+              robin_hood::unordered_set<offset_t> kept_chains;
+              for (auto &mapping : maximallyMergedMappings) {
+                  kept_chains.insert(mapping.splitMappingId);
+              }
+              mappings.erase(
+                  std::remove_if(mappings.begin(), mappings.end(),
+                                 [&kept_chains](const MappingResult &mapping) {
+                                     return !kept_chains.count(mapping.splitMappingId);
+                                 }),
+                  mappings.end());
+            //filterMaximallyMerged(mappings, param);
           } else {
               filterNonMergedMappings(mappings, param);
           }