#4495 Active Cells : Add helper class to import active cells

2025-02-25 18:55:39 -06:00 · 2019-06-19 13:36:33 +02:00
parent df69bc2d8d
commit ada2fb1480
5 changed files with 254 additions and 0 deletions
--- a/ApplicationCode/UnitTests/RifActiveCellsReader-Test.cpp
+++ b/ApplicationCode/UnitTests/RifActiveCellsReader-Test.cpp
@@ -0,0 +1,119 @@
+////////////////////////////////////////////////////////////////////////////////
+//
+//  Copyright (C) 2019-     Equinor ASA
+//
+//  ResInsight 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.
+//
+//  ResInsight 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 at <http://www.gnu.org/licenses/gpl.html>
+//  for more details.
+//
+/////////////////////////////////////////////////////////////////////////////////
+
+#include "gtest/gtest.h"
+
+#include "RiaStringEncodingTools.h"
+#include "RiaTestDataDirectory.h"
+#include "RifActiveCellsReader.h"
+
+#include "ert/ecl/ecl_grid.hpp"
+#include "ert/ecl/ecl_file.hpp"
+
+#include <QDir>
+
+//--------------------------------------------------------------------------------------------------
+///
+//--------------------------------------------------------------------------------------------------
+TEST(RifActiveCellsReaderTest, BasicTest10k)
+{
+    QDir baseFolder(TEST_MODEL_DIR);
+    bool subFolderExists = baseFolder.cd("TEST10K_FLT_LGR_NNC");
+    EXPECT_TRUE(subFolderExists);
+
+    ecl_grid_type* mainEclGrid = nullptr;
+
+    {
+        QString filename("TEST10K_FLT_LGR_NNC.EGRID");
+        QString filePath = baseFolder.absoluteFilePath(filename);
+
+        mainEclGrid = ecl_grid_alloc(RiaStringEncodingTools::toNativeEncoded(filePath).data());
+    }
+
+    std::vector<std::vector<int>> activeCellsFromActnum;
+    std::vector<std::vector<int>> activeCellsFromPorv;
+    {
+        QString filename("TEST10K_FLT_LGR_NNC.EGRID");
+        QString filePath = baseFolder.absoluteFilePath(filename);
+
+        ecl_file_type* gridFile = ecl_file_open(RiaStringEncodingTools::toNativeEncoded(filePath).data(), ECL_FILE_CLOSE_STREAM);
+        activeCellsFromActnum   = RifActiveCellsReader::activeCellsFromActnumKeyword(mainEclGrid, gridFile);
+        EXPECT_EQ(2, activeCellsFromActnum.size());
+        ecl_file_close(gridFile);
+    }
+
+    {
+        QString filename("TEST10K_FLT_LGR_NNC.INIT");
+        QString filePath = baseFolder.absoluteFilePath(filename);
+
+        ecl_file_type* initFile = ecl_file_open(RiaStringEncodingTools::toNativeEncoded(filePath).data(), ECL_FILE_CLOSE_STREAM);
+
+        activeCellsFromPorv = RifActiveCellsReader::activeCellsFromPorvKeyword(mainEclGrid, initFile);
+        EXPECT_EQ(2, activeCellsFromPorv.size());
+
+        ecl_file_close(initFile);
+    }
+
+    for (size_t gridIndex = 0; gridIndex < activeCellsFromActnum.size(); gridIndex++)
+    {
+        for (size_t valueIndex = 0; valueIndex < activeCellsFromActnum[gridIndex].size(); valueIndex++)
+        {
+            auto actnumValue = activeCellsFromActnum[gridIndex][valueIndex];
+            auto porvValue   = activeCellsFromPorv[gridIndex][valueIndex];
+            if (actnumValue > 0)
+            {
+                EXPECT_TRUE(porvValue > 0);
+            }
+            else
+            {
+                EXPECT_EQ(0, porvValue);
+            }
+        }
+    }
+
+    std::vector<int> expectedActiveCellCountPerGrid;
+    expectedActiveCellCountPerGrid.push_back(8517);
+    expectedActiveCellCountPerGrid.push_back(2608);
+
+    for (int gridIndex = 0; gridIndex < static_cast<int>(activeCellsFromActnum.size()); gridIndex++)
+    {
+        ecl_grid_type* currentGrid = nullptr;
+        if (gridIndex == 0)
+        {
+            currentGrid = mainEclGrid;
+        }
+        else
+        {
+            currentGrid = ecl_grid_iget_lgr(mainEclGrid, gridIndex - 1);
+        }
+
+        auto activeCellsForGrid = activeCellsFromActnum[gridIndex];
+        if (ecl_grid_get_global_size(currentGrid) == static_cast<int>(activeCellsForGrid.size()))
+        {
+            int expectedCellCount = expectedActiveCellCountPerGrid[gridIndex];
+            EXPECT_EQ(expectedCellCount, ecl_grid_get_nactive(currentGrid));
+
+            int* actnum_values = activeCellsForGrid.data();
+
+            ecl_grid_reset_actnum(currentGrid, actnum_values);
+            EXPECT_EQ(expectedCellCount, ecl_grid_get_nactive(currentGrid));
+        }
+    }
+
+    ecl_grid_free(mainEclGrid);
+}