Merge pull request #384 from pangenome/heaps-tweaks

tweaks to odgi heaps
pangenome · Apr 13, 2022 · d66d947 · d66d947
2 parents 6be13b5 + 111eb3c
commit d66d947
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/src/subcommand/heaps_main.cpp b/src/subcommand/heaps_main.cpp
@@ -148,13 +148,22 @@ int main_heaps(int argc, char **argv) {
                 uint64_t start = std::stoul(vals[1]);
                 uint64_t end = std::stoul(vals[2]);
                 if (!graph.has_path(path_name)) {
-                    std::cerr << "[odgi::heaps] no path '" << path_name << "' in graph" << std::endl;
-                    return 1;
+                    //std::cerr << "[odgi::heaps] warning: no path '" << path_name << "' in graph" << std::endl;
+                } else {
+                    auto path = graph.get_path_handle(path_name);
+                    intervals[path].push_back(algorithms::interval_t(start, end));
                 }
-                auto path = graph.get_path_handle(path_name);
-                intervals[path].push_back(algorithms::interval_t(start, end));
             }
         }
+        // ensure sorted input
+        std::vector<std::vector<algorithms::interval_t>*> v;
+        for (auto& i : intervals) {
+            v.push_back(&i.second);
+        }
+#pragma omp parallel for
+        for (auto& ivals : v) {
+            std::sort(ivals->begin(), ivals->end());
+        }
     }
 
     graph.set_number_of_threads(num_threads);