finish class EclFile

this is a base class for ECL I/O classes
This commit is contained in:
Torbjørn Skille 2019-03-28 11:55:50 +01:00 committed by Arne Morten Kvarving
parent 6d1b2f3f4f
commit 46f2542487
8 changed files with 2832 additions and 442 deletions

View File

@ -161,6 +161,7 @@ if(ENABLE_ECL_INPUT)
examples/test_util/EclFilesComparator.cpp
examples/test_util/EclIntegrationTest.cpp
examples/test_util/EclRegressionTest.cpp
examples/test_util/EclFile.cpp
examples/test_util/EclOutput.cpp
examples/test_util/EclUtil.cpp
examples/test_util/summaryComparator.cpp
@ -173,9 +174,10 @@ if(ENABLE_ECL_INPUT)
# Add the tests
set(_libs testutil opmcommon
${Boost_UNIT_TEST_FRAMEWORK_LIBRARY})
foreach(test test_compareSummary test_EclFilesComparator)
foreach(test test_compareSummary test_EclFilesComparator test_EclIO)
opm_add_test(${test} CONDITION ENABLE_ECL_INPUT
LIBRARIES ${_libs})
LIBRARIES ${_libs}
WORKING_DIRECTORY ${PROJECT_BINARY_DIR}/tests)
endforeach()
install(TARGETS compareECL DESTINATION bin)
endif()

View File

@ -21,7 +21,6 @@
# the library needs it.
list (APPEND MAIN_SOURCE_FILES
examples/test_util/EclFile.cpp
src/opm/common/data/SimulationDataContainer.cpp
src/opm/common/OpmLog/CounterLog.cpp
src/opm/common/OpmLog/EclipsePRTLog.cpp
@ -190,7 +189,6 @@ if(ENABLE_ECL_OUTPUT)
endif()
list (APPEND TEST_SOURCE_FILES
tests/test_EclFile.cpp
tests/test_calculateCellVol.cpp
tests/test_cmp.cpp
tests/test_cubic.cpp
@ -320,6 +318,10 @@ endif()
list (APPEND EXAMPLE_SOURCE_FILES
)
if(ENABLE_ECL_INPUT)
list (APPEND TEST_DATA_FILES
tests/ECLFILE.INIT
tests/ECLFILE.FINIT
)
list (APPEND EXAMPLE_SOURCE_FILES
examples/opmi.cpp
examples/opmpack.cpp
@ -340,8 +342,6 @@ if(ENABLE_ECL_INPUT)
endif()
list( APPEND PUBLIC_HEADER_FILES
examples/test_util/EclFile.hpp
examples/test_util/data/EclIOdata.hpp
opm/common/ErrorMacros.hpp
opm/common/Exceptions.hpp
opm/common/data/SimulationDataContainer.hpp

File diff suppressed because it is too large Load Diff

View File

@ -16,26 +16,95 @@
along with OPM. If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef ECLFILE_HPP
#define ECLFILE_HPP
#include <opm/common/ErrorMacros.hpp>
#include <examples/test_util/data/EclIOdata.hpp>
//#include "EclIO.hpp"
#include <iostream>
#include <string>
#include <fstream>
#include <vector>
#include <ctime>
#include <map>
#include <examples/test_util/data/EclIOdata.hpp>
#include <unordered_map>
#include <stdio.h>
namespace EIOD = Opm::ecl;
class EclFile
class EclFile
{
public:
explicit EclFile(const std::string& filename);
};
bool formattedInput() { return formatted; }
void loadData(); // load all data
void loadData(const std::string& arrName); // load all arrays with array name equal to arrName
void loadData(int arrIndex); // load data based on array indices in vector arrIndex
void loadData(const std::vector<int>& arrIndex); // load data based on array indices in vector arrIndex
void clearData()
{
inte_array.clear();
real_array.clear();
doub_array.clear();
logi_array.clear();
char_array.clear();
}
using EclEntry = std::tuple<std::string, EIOD::eclArrType, int>;
std::vector<EclEntry> getList() const;
template <typename T>
const std::vector<T>& get(int arrIndex);
template <typename T>
const std::vector<T>& get(const std::string& name);
bool hasKey(const std::string &name) const;
const std::vector<std::string>& arrayNames() const { return array_name; }
protected:
bool formatted;
std::string inputFilename;
std::unordered_map<int, std::vector<int>> inte_array;
std::unordered_map<int, std::vector<bool>> logi_array;
std::unordered_map<int, std::vector<double>> doub_array;
std::unordered_map<int, std::vector<float>> real_array;
std::unordered_map<int, std::vector<std::string>> char_array;
std::vector<std::string> array_name;
std::vector<EIOD::eclArrType> array_type;
std::vector<int> array_size;
std::vector<unsigned long int> ifStreamPos;
std::map<std::string, int> array_index;
template<class T>
const std::vector<T>& getImpl(int arrIndex, EIOD::eclArrType type,
const std::unordered_map<int, std::vector<T>>& array,
const std::string& typeStr)
{
if (array_type[arrIndex] != type) {
std::string message = "Array with index " + std::to_string(arrIndex) + " is not of type " + typeStr;
OPM_THROW(std::runtime_error, message);
}
if (!arrayLoaded[arrIndex]) {
loadData(arrIndex);
}
return array.find(arrIndex)->second;
}
private:
std::vector<bool> arrayLoaded;
void loadArray(std::fstream& fileH, int arrIndex);
};
#endif

1737
tests/ECLFILE.FINIT Normal file

File diff suppressed because it is too large Load Diff

BIN
tests/ECLFILE.INIT Normal file

Binary file not shown.

View File

@ -1,42 +0,0 @@
/*
+ Copyright 2019 Equinor 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"
#include <iostream>
#include <stdio.h>
#include <iomanip>
#include <math.h>
#include <algorithm>
#include <examples/test_util/EclFile.hpp>
#define BOOST_TEST_MODULE Test EclFile
#include <boost/test/unit_test.hpp>
BOOST_AUTO_TEST_CASE(TestConstructor){
EclFile file1("SPE1CASE1.SMSPEC");
BOOST_CHECK_THROW(EclFile file1("XXXX.INIT"), std::runtime_error );
}

306
tests/test_EclIO.cpp Normal file
View File

@ -0,0 +1,306 @@
/*
+ Copyright 2019 Equinor 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"
#include <iostream>
#include <stdio.h>
#include <iomanip>
#include <math.h>
#include <algorithm>
#include <tuple>
#include <examples/test_util/EclFile.hpp>
#include <examples/test_util/EclOutput.hpp>
#define BOOST_TEST_MODULE Test EclIO
#include <boost/test/unit_test.hpp>
template<typename InputIterator1, typename InputIterator2>
bool
range_equal(InputIterator1 first1, InputIterator1 last1,
InputIterator2 first2, InputIterator2 last2)
{
while(first1 != last1 && first2 != last2)
{
if(*first1 != *first2) return false;
++first1;
++first2;
}
return (first1 == last1) && (first2 == last2);
}
bool compare_files(const std::string& filename1, const std::string& filename2)
{
std::ifstream file1(filename1);
std::ifstream file2(filename2);
std::istreambuf_iterator<char> begin1(file1);
std::istreambuf_iterator<char> begin2(file2);
std::istreambuf_iterator<char> end;
return range_equal(begin1, end, begin2, end);
}
template <typename T>
bool operator==(const std::vector<T> & t1, const std::vector<T> & t2)
{
return std::equal(t1.begin(), t1.end(), t2.begin(), t2.end());
}
BOOST_AUTO_TEST_CASE(TestEclFile_BINARY) {
std::string testFile="ECLFILE.INIT";
// check that exception is thrown when input file doesn't exist
BOOST_CHECK_THROW(EclFile file1("DUMMY.DAT") , std::runtime_error );
EclFile file1(testFile);
// check that exeption is thrown when member function get is used with wrong type
BOOST_CHECK_THROW(std::vector<int> vect1=file1.get<int>(2) , std::runtime_error );
BOOST_CHECK_THROW(std::vector<int> vect1=file1.get<int>("PORV") , std::runtime_error );
BOOST_CHECK_THROW(std::vector<float> vect1=file1.get<float>(0) , std::runtime_error );
BOOST_CHECK_THROW(std::vector<float> vect1=file1.get<float>("ICON") , std::runtime_error );
BOOST_CHECK_THROW(std::vector<double> vect1=file1.get<double>(0) , std::runtime_error );
BOOST_CHECK_THROW(std::vector<double> vect1=file1.get<double>("KEYWORDS") , std::runtime_error );
BOOST_CHECK_THROW(std::vector<bool> vect1=file1.get<bool>(0) , std::runtime_error );
BOOST_CHECK_THROW(std::vector<bool> vect1=file1.get<bool>("XCON") , std::runtime_error );
BOOST_CHECK_THROW(std::vector<std::string> vect1=file1.get<std::string>(0) , std::runtime_error );
BOOST_CHECK_THROW(std::vector<std::string> vect1=file1.get<std::string>("XCON") , std::runtime_error );
// check member function hasKey
BOOST_CHECK_EQUAL(file1.hasKey("PORV"), true);
BOOST_CHECK_EQUAL(file1.hasKey("XPORV"), false);
// test member functon get, use size of vector to confirm that vectror is ok
std::vector<int> vect1a=file1.get<int>(0);
std::vector<int> vect1b=file1.get<int>("ICON");
BOOST_CHECK_EQUAL(vect1a.size(), 1875);
BOOST_CHECK_EQUAL(vect1b.size(), 1875);
std::vector<bool> vect2a=file1.get<bool>(1);
std::vector<bool> vect2b=file1.get<bool>("LOGIHEAD");
BOOST_CHECK_EQUAL(vect2a.size(), 121);
BOOST_CHECK_EQUAL(vect2b.size(), 121);
std::vector<float> vect3a=file1.get<float>(2);
std::vector<float> vect3b=file1.get<float>("PORV");
BOOST_CHECK_EQUAL(vect3a.size(), 3146);
BOOST_CHECK_EQUAL(vect3b.size(), 3146);
std::vector<double> vect4a=file1.get<double>(3);
std::vector<double> vect4b=file1.get<double>("XCON");
BOOST_CHECK_EQUAL(vect4a.size(), 1740);
BOOST_CHECK_EQUAL(vect4b.size(), 1740);
std::vector<std::string> vect5a=file1.get<std::string>(4);
std::vector<std::string> vect5b=file1.get<std::string>("KEYWORDS");
BOOST_CHECK_EQUAL(vect5a.size(), 312);
BOOST_CHECK_EQUAL(vect5b.size(), 312);
}
BOOST_AUTO_TEST_CASE(TestEclFile_FORMATTED) {
std::string testFile1="ECLFILE.INIT";
std::string testFile2="ECLFILE.FINIT";
// loading data both from binary and formatted file. Check that
// date vectors are identical
EclFile file1(testFile1);
file1.loadData();
EclFile file2(testFile2);
file2.loadData();
std::vector<int> vect1a=file1.get<int>("ICON");
std::vector<int> vect1b=file2.get<int>("ICON");
BOOST_CHECK_EQUAL(vect1a.size(), vect1b.size());
BOOST_CHECK_EQUAL(vect1a==vect1b,true);
std::vector<float> vect2a=file1.get<float>("PORV");
std::vector<float> vect2b=file2.get<float>("PORV");
BOOST_CHECK_EQUAL(vect2a.size(), vect2b.size());
BOOST_CHECK_EQUAL(vect2a==vect2b,true);
std::vector<double> vect3a=file1.get<double>("XCON");
std::vector<double> vect3b=file2.get<double>("XCON");
BOOST_CHECK_EQUAL(vect3a.size(), vect3b.size());
BOOST_CHECK_EQUAL(vect3a==vect3b,true);
std::vector<bool> vect4a=file1.get<bool>("LOGIHEAD");
std::vector<bool> vect4b=file2.get<bool>("LOGIHEAD");
BOOST_CHECK_EQUAL(vect4a.size(), vect4b.size());
BOOST_CHECK_EQUAL(vect4a==vect4b,true);
std::vector<std::string> vect5a=file1.get<std::string>("KEYWORDS");
std::vector<std::string> vect5b=file2.get<std::string>("KEYWORDS");
BOOST_CHECK_EQUAL(vect5a.size(), vect5b.size());
BOOST_CHECK_EQUAL(vect5a==vect5b,true);
}
BOOST_AUTO_TEST_CASE(TestEcl_Write_binary) {
std::string inputFile="ECLFILE.INIT";
std::string testFile="TEST.DAT";
// loading vectors from binary file and write these back to a binary file1
// compare input and output file and delete file.
EclFile file1(inputFile);
file1.loadData();
std::vector<int> icon=file1.get<int>("ICON");
std::vector<float> porv=file1.get<float>("PORV");
std::vector<double> xcon=file1.get<double>("XCON");
std::vector<bool> logihead=file1.get<bool>("LOGIHEAD");
std::vector<std::string> keywords=file1.get<std::string>("KEYWORDS");
// writing vectors to test file (TEST.DAT) using class EclOutput
{
EclOutput eclTest(testFile, false);
eclTest.write("ICON",icon);
eclTest.write("LOGIHEAD",logihead);
eclTest.write("PORV",porv);
eclTest.write("XCON",xcon);
eclTest.write("KEYWORDS",keywords);
eclTest.write("ENDSOL",std::vector<char>());
}
BOOST_CHECK_EQUAL(compare_files(inputFile, testFile), true);
if (remove(testFile.c_str())==-1) {
std::cout << " > Warning! temporary file was not deleted" << std::endl;
};
}
BOOST_AUTO_TEST_CASE(TestEcl_Write_formatted) {
std::string inputFile="ECLFILE.FINIT";
std::string testFile="TEST.FDAT";
// loading vectors from formatted input file and write data back to a formatted file1
// compare input and output file, and delete file.
EclFile file1(inputFile);
file1.loadData();
std::vector<int> icon = file1.get<int>("ICON");
std::vector<float> porv = file1.get<float>("PORV");
std::vector<double> xcon = file1.get<double>("XCON");
std::vector<bool> logihead = file1.get<bool>("LOGIHEAD");
std::vector<std::string> keywords = file1.get<std::string>("KEYWORDS");
// writing vectors to test file (TEST.FDAT) using class EclOutput
EclOutput eclTest(testFile, true);
eclTest.write("ICON",icon);
eclTest.write("LOGIHEAD",logihead);
eclTest.write("PORV",porv);
eclTest.write("XCON",xcon);
eclTest.write("KEYWORDS",keywords);
eclTest.write("ENDSOL",std::vector<char>());
BOOST_CHECK_EQUAL(compare_files(inputFile, testFile), true);
if (remove(testFile.c_str())==-1) {
std::cout << " > Warning! temporary file was not deleted" << std::endl;
};
}
BOOST_AUTO_TEST_CASE(TestEcl_getList) {
std::string inputFile="ECLFILE.INIT";
std::string testFile="TEST.DAT";
// use EclFile to read/open a binary file
// Use API for class EclFile together with class EclOutput to write an
// identical eclfile
// EclFile::getList(), EclFile::get<T>(int)
EclFile file1(inputFile);
file1.loadData();
{
EclOutput eclTest(testFile, false);
auto arrayList = file1.getList();
int n=0;
for (auto array : arrayList) {
std::string name = std::get<0>(array);
EIOD::eclArrType arrType = std::get<1>(array);
if (arrType == EIOD::INTE) {
std::vector<int> vect = file1.get<int>(n);
eclTest.write(name, vect);
} else if (arrType == EIOD::REAL) {
std::vector<float> vect = file1.get<float>(n);
eclTest.write(name, vect);
} else if (arrType == EIOD::DOUB) {
std::vector<double> vect = file1.get<double>(n);
eclTest.write(name, vect);
} else if (arrType == EIOD::LOGI) {
std::vector<bool> vect = file1.get<bool>(n);
eclTest.write(name, vect);
} else if (arrType == EIOD::CHAR) {
std::vector<std::string> vect = file1.get<std::string>(n);
eclTest.write(name, vect);
} else if (arrType == EIOD::MESS) {
eclTest.write(name, std::vector<char>());
} else {
std::cout << "unknown type " << std::endl;
exit(1);
}
n++;
}
}
BOOST_CHECK_EQUAL(compare_files(inputFile, testFile), true);
if (remove(testFile.c_str())==-1) {
std::cout << " > Warning! temporary file was not deleted" << std::endl;
};
}