Files
opm-common/test_util/rewriteEclFile.cpp
Torbjørn Skille 9fa90b2687 Update of EclIO classes.
- Enables reading of output files generated by simulator IX
 - Adding support for EclFile data type C0nn (string with length > 8 characters)
 - Update of program summary, now supporting well names with more that 8 characters
 - Updates of program convertECL, possible to write output files with IX "format"
 - updates of python bindings (EclOutput and EclFile)
2020-11-10 22:02:38 +01:00

103 lines
3.1 KiB
C++

/*
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 <cmath>
#include <iomanip>
#include <iostream>
#include <tuple>
#include <getopt.h>
#include <sstream>
#include <stdexcept>
#include <opm/io/eclipse/EclFile.hpp>
#include <opm/io/eclipse/EclOutput.hpp>
static void printHelp() {
std::cout << "\rewriteEclFile needs a minimum of one arguments which is the input file name. \n"
<< "\nIn addition, the program takes these options (which must be given before the arguments):\n\n"
<< "-h Print help and exit.\n\n";
}
int main(int argc, char **argv) {
int c = 0;
while ((c = getopt(argc, argv, "h")) != -1) {
switch (c) {
case 'h':
printHelp();
return 0;
default:
return EXIT_FAILURE;
}
}
int argOffset = optind;
Opm::EclIO::EclFile reffile(argv[argOffset]);
auto arrayList = reffile.getList();
std::string outputFile=std::string(argv[argOffset]);
int p1 = outputFile.find_last_of(".");
std::string ext = outputFile.substr(p1+1);
outputFile = outputFile.substr(0,p1) + "_REWRITE." + ext;
Opm::EclIO::EclOutput outFile(outputFile, reffile.formattedInput());
if (reffile.is_ix())
outFile.set_ix();
reffile.loadData();
std::vector<int> elementSizeList = reffile.getElementSizeList();
for (size_t n = 0; n < arrayList.size(); n++){
std::string name = std::get<0>(arrayList[n]);
auto arrType = std::get<1>(arrayList[n]);
if (arrType == Opm::EclIO::INTE) {
auto data = reffile.get<int>(n);
outFile.write(name, data);
} else if (arrType == Opm::EclIO::CHAR) {
auto data = reffile.get<std::string>(n);
outFile.write(name, data);
} else if (arrType == Opm::EclIO::C0NN) {
auto data = reffile.get<std::string>(n);
outFile.write(name, data, elementSizeList[n]);
} else if (arrType == Opm::EclIO::REAL) {
auto data = reffile.get<float>(n);
outFile.write(name, data);
} else if (arrType == Opm::EclIO::DOUB) {
auto data = reffile.get<double>(n);
outFile.write(name, data);
} else if (arrType == Opm::EclIO::LOGI) {
auto data = reffile.get<bool>(n);
outFile.write(name, data);
} else if (arrType == Opm::EclIO::MESS) {
outFile.message(name);
}
}
return 0;
}