Add explicit Ordering for (ReferenceRegion, Int) in shuffle join

bigdatagenomics · Jan 22, 2015 · eeb251c · eeb251c
1 parent d0821c7
commit eeb251c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/adam-core/src/main/scala/org/bdgenomics/adam/rdd/ShuffleRegionJoin.scala b/adam-core/src/main/scala/org/bdgenomics/adam/rdd/ShuffleRegionJoin.scala
@@ -29,6 +29,16 @@ import scala.reflect.ClassTag
 
 object ShuffleRegionJoin {
 
+  implicit val manualRegionOrdering = new Ordering[(ReferenceRegion, Int)] {
+    override def compare(x: (ReferenceRegion, Int), y: (ReferenceRegion, Int)) = {
+      if (x._1 != y._1) {
+        x._1.compare(y._1)
+      } else {
+        x._2.compare(y._2)
+      }
+    }
+  }
+
   /**
    * Performs a region join between two RDDs (shuffle join).
    *