From 706e51f878d0d39e81a683f2f7504a4b8cb46b2f Mon Sep 17 00:00:00 2001
From: Andreas Lauser <and@poware.org>
Date: Tue, 6 Dec 2016 19:49:06 +0100
Subject: [PATCH] ebos: store the names wells that are disabled on the local
 process due to load balancing

---
 ebos/eclcpgridmanager.hh | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/ebos/eclcpgridmanager.hh b/ebos/eclcpgridmanager.hh
index 7c860fd69..555851b16 100644
--- a/ebos/eclcpgridmanager.hh
+++ b/ebos/eclcpgridmanager.hh
@@ -182,7 +182,7 @@ public:
             }
 
             //distribute the grid and switch to the distributed view.
-            grid_->loadBalance(&this->eclState(), faceTrans.data());
+            defunctWellNames_ = std::get<1>(grid_->loadBalance(&this->eclState(), faceTrans.data()));
             grid_->switchToDistributedView();
 
             delete cartesianIndexMapper_;
@@ -206,6 +206,9 @@ public:
     const CartesianIndexMapper& equilCartesianIndexMapper() const
     { return *equilCartesianIndexMapper_; }
 
+    std::unordered_set<std::string> defunctWellNames() const
+    { return defunctWellNames_; }
+
 protected:
     void createGrids_()
     {
@@ -236,6 +239,8 @@ protected:
     EquilGrid* equilGrid_;
     CartesianIndexMapper* cartesianIndexMapper_;
     CartesianIndexMapper* equilCartesianIndexMapper_;
+
+    std::unordered_set<std::string> defunctWellNames_;
 };
 
 } // namespace Ewoms