Added utility for querying (only) active cells.

This will relieve us from building up a full EclipseGrid just for filtering well connections.
2019-12-10 10:24:08 +01:00
parent 9f09cbe14a
commit 36e1ef69ef
4 changed files with 239 additions and 0 deletions
--- a/tests/test_ActiveGridCells.cpp
+++ b/tests/test_ActiveGridCells.cpp
@@ -0,0 +1,88 @@
+/*
+  Copyright 2019 Statoil ASA.
+
+  This file is part of the Open Porous Media project (OPM).
+
+  OPM is free software: you can redistribute it and/or modify
+  it under the terms of the GNU General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  OPM is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU General Public License for more details.
+
+  You should have received a copy of the GNU General Public License
+  along with OPM.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "config.h"
+
+#define BOOST_TEST_MODULE ActiveGridCells
+#include <boost/test/unit_test.hpp>
+
+#include <opm/common/utility/ActiveGridCells.hpp>
+#include <opm/output/eclipse/Summary.hpp>
+
+BOOST_AUTO_TEST_SUITE(Default)
+
+BOOST_AUTO_TEST_CASE(testLocal) {
+    std::size_t nx=2, ny = 3, nz=2;
+    Opm::GridDims dims(nx, ny, nz);
+    std::vector<int> cells(9);
+    std::set<std::array<std::size_t,3> > inactive;
+    inactive.insert({{0,0,0}});
+    inactive.insert({{1,1,0}});
+    inactive.insert({{1,0,1}});
+
+    for(std::size_t k = 0, index = 0; k < nz; ++k)
+    {
+        for(std::size_t j = 0; j < ny; ++j)
+        {
+            for(std::size_t i = 0; i < nx; ++i)
+            {
+                if ( inactive.find({{i,j,k}})==inactive.end())
+                {
+                    cells[index++] = dims.getGlobalIndex(i,j,k);
+                }
+            }
+        }
+    }
+
+    Opm::ActiveGridCells active(nx, ny, nz, cells.data(), cells.size());
+
+    const auto& actnum = active.actNum();
+
+    for(std::size_t k = 0, index = 0; k < nz; ++k)
+    {
+        for(std::size_t j = 0; j < ny; ++j)
+        {
+            for(std::size_t i = 0; i < nx; ++i)
+            {
+                auto cartIndex = dims.getGlobalIndex(i,j,k);
+
+                if ( inactive.find({{i,j,k}})!=inactive.end())
+                {
+                    BOOST_CHECK(actnum[cartIndex] == 0);
+                    BOOST_CHECK(active.localCell(cartIndex) == -1);
+                    BOOST_CHECK(active.localCell(i,j,k) == -1);
+                    BOOST_CHECK(!active.cellActive(cartIndex));
+                    BOOST_CHECK(!active.cellActive(i, j, k));
+                }
+                else
+                {
+                    BOOST_CHECK(actnum[cartIndex] == 1);
+                    BOOST_CHECK(active.localCell(cartIndex) == (int)index);
+                    BOOST_CHECK(active.localCell(i,j,k) == (int)index);
+                    BOOST_CHECK(active.cellActive(cartIndex));
+                    BOOST_CHECK(active.cellActive(i, j, k));
+                    ++index;
+                }
+            }
+        }
+    }
+}
+
+
+BOOST_AUTO_TEST_SUITE_END()