From c19c117ac076fe39f9fa4c58e43d0faa5d9af05c Mon Sep 17 00:00:00 2001
From: ageay <ageay>
Date: Thu, 8 Aug 2013 10:53:42 +0000
Subject: [PATCH] For load balancing on // interpolation

- MEDCouplingPointSet::ComputeNbOfInteractionsWithSrcCells
- DataArrayDouble::computeNbOfInteractionsWith
- DataArrayInt::splitInBalancedSlices
---
 src/MEDCoupling/MEDCouplingMemArray.cxx       |  5 ++++-
 src/MEDCoupling_Swig/MEDCouplingBasicsTest.py | 19 +++++++++++++++++++
 2 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/src/MEDCoupling/MEDCouplingMemArray.cxx b/src/MEDCoupling/MEDCouplingMemArray.cxx
index e96490c05..9c2fda8fa 100644
--- a/src/MEDCoupling/MEDCouplingMemArray.cxx
+++ b/src/MEDCoupling/MEDCouplingMemArray.cxx
@@ -9932,7 +9932,10 @@ std::vector< std::pair<int,int> > DataArrayInt::splitInBalancedSlices(int nbOfSl
       std::pair<int,int> p(pos,-1);
       int locSum(0);
       while(locSum<sumPerSlc && pos<nbOfTuples) { pos++; locSum+=*w++; }
-      p.second=pos;
+      if(i!=nbOfSlices-1)
+        p.second=pos;
+      else
+        p.second=nbOfTuples;
       ret[i]=p;
     }
   return ret;
diff --git a/src/MEDCoupling_Swig/MEDCouplingBasicsTest.py b/src/MEDCoupling_Swig/MEDCouplingBasicsTest.py
index 94f70f0b9..ae1c8c58d 100644
--- a/src/MEDCoupling_Swig/MEDCouplingBasicsTest.py
+++ b/src/MEDCoupling_Swig/MEDCouplingBasicsTest.py
@@ -13733,6 +13733,25 @@ class MEDCouplingBasicsTest(unittest.TestCase):
         self.assertAlmostEqual(16.,d.getMeasureField(False).accumulate()[0],1e-13)
         pass
 
+    def testSwig2LoadBalanceBBox1(self):
+        arr=DataArrayDouble(5) ; arr.iota()
+        t=MEDCouplingCMesh() ; t.setCoords(arr,arr)
+        arr=DataArrayDouble(16) ; arr.iota() ; arr*=2./15
+        s=MEDCouplingCMesh() ; s.setCoords(arr,arr[:]) ; s.translate([2.,1.])
+        #
+        s1=s.build1SGTUnstructured()
+        t1=t.build1SGTUnstructured()
+        w=MEDCouplingPointSet.ComputeNbOfInteractionsWithSrcCells(s1,t1,1e-12)
+        wExp=DataArrayInt([0,0,0,0,0,0,64,64,0,0,64,64,0,0,0,0])
+        self.assertTrue(w.isEqual(wExp))
+        slcs=w.splitInBalancedSlices(4)
+        self.assertEqual(len(slcs),4)
+        self.assertEqual(slcs,[slice(0,7,1),slice(7,8,1),slice(8,11,1),slice(11,16,1)])
+        bbs=s1.getBoundingBoxForBBTree()
+        bbt=t1.getBoundingBoxForBBTree()
+        self.assertTrue(bbt.computeNbOfInteractionsWith(bbs,1e-12).isEqual(wExp))
+        pass
+
     def setUp(self):
         pass
     pass
-- 
2.39.2