optimize createCluster

OpenGene · Dec 5, 2018 · 6ceeb9a · 6ceeb9a
1 parent c7cd590
commit 6ceeb9a
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/src/gencore.cpp b/src/gencore.cpp
@@ -295,12 +295,23 @@ void Gencore::addToUnProperCluster(bam1_t* b) {
 }
 
 void Gencore::createCluster(map<int, map<int, map<int, Cluster*>>>& clusters, int tid, int left, int right) {
-    if(clusters.count(tid) == 0)
+    map<int, map<int, map<int, Cluster*>>>::iterator iter1 = clusters.find(tid);
+
+    if(iter1 == clusters.end()) {
         clusters[tid] = map<int, map<int, Cluster*>>();
-    if(clusters[tid].count(left) == 0)
         clusters[tid][left] = map<int, Cluster*>();
-    if(clusters[tid][left].count(right) == 0)
         clusters[tid][left][right] = new Cluster(mOptions);
+    } else {
+        map<int, map<int, Cluster*>>::iterator iter2  =iter1->second.find(left);
+        if(iter2 == iter1->second.end()) {
+            clusters[tid][left] = map<int, Cluster*>();
+            clusters[tid][left][right] = new Cluster(mOptions);
+        } else {
+            map<int, Cluster*>::iterator iter3 = iter2->second.find(right);
+            if(iter3 == iter2->second.end())
+                clusters[tid][left][right] = new Cluster(mOptions);
+        }
+    }
 }
 
 void Gencore::addToCluster(bam1_t* b) {