mirror of
https://github.com/OPM/ResInsight.git
synced 2025-02-25 18:55:39 -06:00
#2350. Implement ElementPropertyTable file reader for single file
This commit is contained in:
parent
2dbc907934
commit
55147e5b70
@ -40,6 +40,7 @@ ${CEE_CURRENT_LIST_DIR}RifCsvUserData.h
|
||||
${CEE_CURRENT_LIST_DIR}RifCsvUserDataParser.h
|
||||
${CEE_CURRENT_LIST_DIR}RifWellPathFormationReader.h
|
||||
${CEE_CURRENT_LIST_DIR}RifWellPathFormationsImporter.h
|
||||
${CEE_CURRENT_LIST_DIR}RifElementPropertyTableReader.h
|
||||
# HDF5 file reader is directly included in ResInsight main CmakeList.txt
|
||||
#${CEE_CURRENT_LIST_DIR}RifHdf5Reader.h
|
||||
)
|
||||
@ -84,6 +85,7 @@ ${CEE_CURRENT_LIST_DIR}RifCsvUserData.cpp
|
||||
${CEE_CURRENT_LIST_DIR}RifCsvUserDataParser.cpp
|
||||
${CEE_CURRENT_LIST_DIR}RifWellPathFormationReader.cpp
|
||||
${CEE_CURRENT_LIST_DIR}RifWellPathFormationsImporter.cpp
|
||||
${CEE_CURRENT_LIST_DIR}RifElementPropertyTableReader.cpp
|
||||
|
||||
# HDF5 file reader is directly included in ResInsight main CmakeList.txt
|
||||
#${CEE_CURRENT_LIST_DIR}RifHdf5Reader.cpp
|
||||
|
196
ApplicationCode/FileInterface/RifElementPropertyTableReader.cpp
Normal file
196
ApplicationCode/FileInterface/RifElementPropertyTableReader.cpp
Normal file
@ -0,0 +1,196 @@
|
||||
/////////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// Copyright (C) 2017- Statoil 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 "RifElementPropertyTableReader.h"
|
||||
|
||||
#include "RiaLogging.h"
|
||||
#include "RiuMainWindow.h"
|
||||
|
||||
#include <QFile>
|
||||
#include <QMessageBox>
|
||||
#include <QStringList>
|
||||
#include <QTextStream>
|
||||
|
||||
#include <algorithm>
|
||||
#include <cctype>
|
||||
#include <string>
|
||||
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
/// Internal functions
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
static QFile* openFile(const QString &fileName);
|
||||
static void closeFile(QFile *file);
|
||||
static QStringList splitLineAndTrim(const QString& line, const QString& separator);
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
///
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
ElementPropertyMetadata RifElementPropertyTableReader::readMetadata(const QString& fileName)
|
||||
{
|
||||
ElementPropertyMetadata metadata;
|
||||
QFile* file = openFile(fileName);
|
||||
|
||||
if (file)
|
||||
{
|
||||
QTextStream stream(file);
|
||||
bool metadataBlockFound = false;
|
||||
int maxLinesToRead = 50;
|
||||
int lineNo = 0;
|
||||
|
||||
while (lineNo < maxLinesToRead)
|
||||
{
|
||||
QString line = stream.readLine();
|
||||
lineNo++;
|
||||
|
||||
if (line.toUpper().startsWith("*DISTRIBUTION TABLE"))
|
||||
{
|
||||
metadataBlockFound = true;
|
||||
continue;
|
||||
}
|
||||
|
||||
if (!metadataBlockFound) continue;
|
||||
|
||||
QStringList cols = splitLineAndTrim(line, ",");
|
||||
|
||||
metadata.fileName = fileName;
|
||||
for (QString s : cols)
|
||||
{
|
||||
metadata.dataColumns.push_back(s);
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
closeFile(file);
|
||||
}
|
||||
|
||||
return metadata;
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
///
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
void RifElementPropertyTableReader::readData(const ElementPropertyMetadata *metadata, ElementPropertyTable *table)
|
||||
{
|
||||
CVF_ASSERT(metadata && table);
|
||||
|
||||
QFile* file = openFile(metadata->fileName);
|
||||
int expectedColumnCount = (int)metadata->dataColumns.size() + 1;
|
||||
|
||||
if (file && expectedColumnCount > 0)
|
||||
{
|
||||
QTextStream stream(file);
|
||||
bool dataBlockFound = false;
|
||||
int lineNo = 0;
|
||||
|
||||
// Init data vectors
|
||||
table->elementIds.clear();
|
||||
table->data = std::vector<std::vector<float>>(metadata->dataColumns.size());
|
||||
|
||||
while (!stream.atEnd())
|
||||
{
|
||||
QString line = stream.readLine();
|
||||
QStringList cols = splitLineAndTrim(line, ",");
|
||||
lineNo++;
|
||||
|
||||
if (!dataBlockFound)
|
||||
{
|
||||
if (!line.startsWith("*") && cols.size() == expectedColumnCount) dataBlockFound = true;
|
||||
else continue;
|
||||
}
|
||||
|
||||
if (cols.size() != expectedColumnCount)
|
||||
{
|
||||
throw FileParseException(QString("Number of columns mismatch at %1:%2").arg(metadata->fileName).arg(lineNo));
|
||||
}
|
||||
|
||||
for (int c = 0; c < expectedColumnCount; c++)
|
||||
{
|
||||
bool parseOk;
|
||||
|
||||
if (c == 0)
|
||||
{
|
||||
// Remove elementId column prefix
|
||||
QStringList parts = cols[0].split(".");
|
||||
|
||||
int elementId = parts.last().toInt(&parseOk);
|
||||
if (!parseOk)
|
||||
{
|
||||
throw FileParseException(QString("Parse failed at %1:%2").arg(metadata->fileName).arg(lineNo));
|
||||
}
|
||||
table->elementIds.push_back(elementId);
|
||||
}
|
||||
else
|
||||
{
|
||||
float value = cols[c].toFloat(&parseOk);
|
||||
if (!parseOk)
|
||||
{
|
||||
throw FileParseException(QString("Parse failed at %1:%2").arg(metadata->fileName).arg(lineNo));
|
||||
}
|
||||
table->data[c - 1].push_back(value);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
table->hasData = true;
|
||||
}
|
||||
|
||||
closeFile(file);
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
///
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
QFile* openFile(const QString &fileName)
|
||||
{
|
||||
QFile *file;
|
||||
file = new QFile(fileName);
|
||||
if (!file->open(QIODevice::ReadOnly | QIODevice::Text))
|
||||
{
|
||||
RiaLogging::error(QString("Failed to open %1").arg(fileName));
|
||||
|
||||
delete file;
|
||||
return nullptr;
|
||||
}
|
||||
return file;
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
///
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
void closeFile(QFile *file)
|
||||
{
|
||||
if (file)
|
||||
{
|
||||
file->close();
|
||||
delete file;
|
||||
}
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
///
|
||||
//--------------------------------------------------------------------------------------------------
|
||||
QStringList splitLineAndTrim(const QString& line, const QString& separator)
|
||||
{
|
||||
QStringList cols = line.split(separator);
|
||||
for (QString& col : cols)
|
||||
{
|
||||
col = col.trimmed();
|
||||
}
|
||||
return cols;
|
||||
}
|
@ -0,0 +1,79 @@
|
||||
/////////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// Copyright (C) 2017- Statoil 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.
|
||||
//
|
||||
/////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "RigWellPathFormations.h"
|
||||
|
||||
#include <map>
|
||||
#include <utility>
|
||||
#include <vector>
|
||||
|
||||
#include "cvfBase.h"
|
||||
#include "cvfObject.h"
|
||||
|
||||
#include <QString>
|
||||
|
||||
class ElementPropertyTable;
|
||||
class ElementPropertyMetadata;
|
||||
|
||||
//==================================================================================================
|
||||
///
|
||||
//==================================================================================================
|
||||
class RifElementPropertyTableReader : cvf::Object
|
||||
{
|
||||
public:
|
||||
static ElementPropertyMetadata readMetadata(const QString& filePath);
|
||||
static void readData(const ElementPropertyMetadata *metadata, ElementPropertyTable *table);
|
||||
};
|
||||
|
||||
//==================================================================================================
|
||||
///
|
||||
//==================================================================================================
|
||||
class FileParseException
|
||||
{
|
||||
public:
|
||||
FileParseException(const QString &message) : message(message) {}
|
||||
QString message;
|
||||
};
|
||||
|
||||
//==================================================================================================
|
||||
///
|
||||
//==================================================================================================
|
||||
class ElementPropertyMetadata
|
||||
{
|
||||
public:
|
||||
QString fileName;
|
||||
std::vector<QString> dataColumns;
|
||||
};
|
||||
|
||||
|
||||
//==================================================================================================
|
||||
///
|
||||
//==================================================================================================
|
||||
class ElementPropertyTable
|
||||
{
|
||||
public:
|
||||
ElementPropertyTable() : hasData(false) {}
|
||||
|
||||
ElementPropertyMetadata metadata;
|
||||
bool hasData;
|
||||
std::vector<int> elementIds;
|
||||
std::vector<std::vector<float>> data;
|
||||
};
|
||||
|
Loading…
Reference in New Issue
Block a user