OilfieldToolsNet/ResInsight
4
0
Fork 0
mirror of https://github.com/OPM/ResInsight.git synced 2025-02-25 18:55:39 -06:00
Code Issues Packages Projects Releases Wiki Activity

7368 geological frk to csv (#7461)

Browse Source
This commit is contained in:
Kristian Bendiksen 2021-03-05 13:41:21 +01:00 committed by GitHub
parent c1ff038642
commit c33b41b2ff
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 117 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

109
ApplicationLibCode/FileInterface/RifStimPlanModelGeologicalFrkExporter.cpp
View File

@ -19,11 +19,15 @@
#include "RifStimPlanModelGeologicalFrkExporter.h"
#include "RiaLogging.h"
#include "RiaPreferences.h"
#include "RifCsvDataTableFormatter.h"
#include "RimStimPlanModel.h"
#include "RimStimPlanModelCalculator.h"
#include <QFile>
#include <QFileInfo>
#include <QTextStream>
//--------------------------------------------------------------------------------------------------
@ -126,6 +130,25 @@ bool RifStimPlanModelGeologicalFrkExporter::writeToFile( RimStimPlanModel* stimP
values["zonePoroElas"] = stimPlanModel->calculator()->calculatePoroElasticConstant();
values["zoneThermalExp"] = stimPlanModel->calculator()->calculateThermalExpansionCoefficient();
// Special values for csv export
auto [depthStart, depthEnd] = createDepthRanges( tvd );
values["dpthstart"] = depthStart;
values["dpthend"] = depthEnd;
std::vector<QString> csvLabels = { "dpthstart", "dpthend" };
for ( const QString& label : labels )
csvLabels.push_back( label );
return writeToFrkFile( filepath, labels, values ) && writeToCsvFile( filepath, csvLabels, values );
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
bool RifStimPlanModelGeologicalFrkExporter::writeToFrkFile( const QString& filepath,
const std::vector<QString>& labels,
const std::map<QString, std::vector<double>>& values )
{
QFile data( filepath );
if ( !data.open( QFile::WriteOnly | QFile::Truncate ) )
{
@ -137,8 +160,11 @@ bool RifStimPlanModelGeologicalFrkExporter::writeToFile( RimStimPlanModel* stimP
for ( QString label : labels )
{
warnOnInvalidData( label, values[label] );
appendToStream( stream, label, values[label] );
auto vals = values.find( label );
if ( vals == values.end() ) return false;
warnOnInvalidData( label, vals->second );
appendToStream( stream, label, vals->second );
}
appendFooterToStream( stream );
@ -146,6 +172,63 @@ bool RifStimPlanModelGeologicalFrkExporter::writeToFile( RimStimPlanModel* stimP
return true;
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
bool RifStimPlanModelGeologicalFrkExporter::writeToCsvFile( const QString& filepath,
const std::vector<QString>& labels,
const std::map<QString, std::vector<double>>& values )
{
// Create the csv in the same directory as the frk file
QFileInfo fi( filepath );
QString csvFilepath = fi.absolutePath() + "/Geological.csv";
QFile data( csvFilepath );
if ( !data.open( QFile::WriteOnly | QFile::Truncate ) )
{
return false;
}
QTextStream stream( &data );
QString fieldSeparator = RiaPreferences::current()->csvTextExportFieldSeparator;
RifCsvDataTableFormatter formatter( stream, fieldSeparator );
// Construct header
std::vector<RifTextDataTableColumn> header;
for ( auto label : labels )
{
header.push_back( RifTextDataTableColumn( label, RifTextDataTableDoubleFormat::RIF_FLOAT ) );
}
formatter.header( header );
// The length of the vectors are assumed to be equal
size_t idx = 0;
bool isDone = false;
while ( !isDone )
{
// Construct one row
for ( auto label : labels )
{
auto vals = values.find( label );
if ( vals == values.end() ) return false;
if ( idx >= vals->second.size() )
isDone = true;
else
{
formatter.add( vals->second[idx] );
}
}
formatter.rowCompleted();
idx++;
}
formatter.tableCompleted();
return true;
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
@ -238,3 +321,25 @@ bool RifStimPlanModelGeologicalFrkExporter::hasInvalidData( const std::vector<do
return false;
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
std::pair<std::vector<double>, std::vector<double>>
RifStimPlanModelGeologicalFrkExporter::createDepthRanges( const std::vector<double>& tvd )
{
std::vector<double> startTvd;
std::vector<double> endTvd;
for ( size_t i = 0; i < tvd.size(); i++ )
{
startTvd.push_back( tvd[i] );
// Special handling for last range
if ( i == tvd.size() - 1 )
endTvd.push_back( startTvd[i] );
else
endTvd.push_back( tvd[i + 1] );
}
return std::make_pair( startTvd, endTvd );
}

10
ApplicationLibCode/FileInterface/RifStimPlanModelGeologicalFrkExporter.h
View File

@ -18,6 +18,7 @@
#pragma once
#include <map>
#include <vector>
class RimStimPlanModel;
@ -38,6 +39,13 @@ public:
static bool writeToFile( RimStimPlanModel* plot, bool useDetailedFluidLoss, const QString& filepath );
private:
static bool writeToFrkFile( const QString& filepath,
const std::vector<QString>& labels,
const std::map<QString, std::vector<double>>& values );
static bool writeToCsvFile( const QString& filepath,
const std::vector<QString>& labels,
const std::map<QString, std::vector<double>>& values );
static void appendHeaderToStream( QTextStream& stream );
static void appendToStream( QTextStream& stream, const QString& label, const std::vector<double>& values );
static void appendFooterToStream( QTextStream& stream );
@ -47,6 +55,8 @@ private:
double maxStressGradient,
double defaultStressGradient );
static std::pair<std::vector<double>, std::vector<double>> createDepthRanges( const std::vector<double>& tvd );
static bool warnOnInvalidData( const QString& label, const std::vector<double>& values );
static bool hasInvalidData( const std::vector<double>& values );
};