mirror of
https://github.com/OPM/ResInsight.git
synced 2025-01-08 07:03:25 -06:00
116 lines
5.1 KiB
C++
116 lines
5.1 KiB
C++
/////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// 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 "RifEnsembleStatisticsReader.h"
|
|
|
|
#include "RimEnsembleCurveSet.h"
|
|
#include "RimEnsembleStatisticsCase.h"
|
|
#include "RimSummaryCaseCollection.h"
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
RifEnsembleStatisticsReader::RifEnsembleStatisticsReader( RimEnsembleStatisticsCase* ensStatCase )
|
|
{
|
|
CVF_ASSERT( ensStatCase );
|
|
|
|
m_ensembleStatCase = ensStatCase;
|
|
|
|
m_allResultAddresses =
|
|
std::set<RifEclipseSummaryAddress>( { RifEclipseSummaryAddress::ensembleStatisticsAddress( ENSEMBLE_STAT_P10_QUANTITY_NAME, "" ),
|
|
RifEclipseSummaryAddress::ensembleStatisticsAddress( ENSEMBLE_STAT_P50_QUANTITY_NAME, "" ),
|
|
RifEclipseSummaryAddress::ensembleStatisticsAddress( ENSEMBLE_STAT_P90_QUANTITY_NAME, "" ),
|
|
RifEclipseSummaryAddress::ensembleStatisticsAddress( ENSEMBLE_STAT_MEAN_QUANTITY_NAME, "" ) } );
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
std::vector<time_t> RifEnsembleStatisticsReader::timeSteps( const RifEclipseSummaryAddress& resultAddress ) const
|
|
{
|
|
if ( !validateAddress( resultAddress ) ) return {};
|
|
return m_ensembleStatCase->timeSteps();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
bool RifEnsembleStatisticsReader::values( const RifEclipseSummaryAddress& resultAddress, std::vector<double>* values ) const
|
|
{
|
|
if ( !validateAddress( resultAddress ) ) return false;
|
|
|
|
const std::vector<double>* sourceData = nullptr;
|
|
auto quantityName = resultAddress.ensembleStatisticsVectorName();
|
|
|
|
if ( quantityName == ENSEMBLE_STAT_P10_QUANTITY_NAME )
|
|
sourceData = &m_ensembleStatCase->p10();
|
|
else if ( quantityName == ENSEMBLE_STAT_P50_QUANTITY_NAME )
|
|
sourceData = &m_ensembleStatCase->p50();
|
|
else if ( quantityName == ENSEMBLE_STAT_P90_QUANTITY_NAME )
|
|
sourceData = &m_ensembleStatCase->p90();
|
|
else if ( quantityName == ENSEMBLE_STAT_MEAN_QUANTITY_NAME )
|
|
sourceData = &m_ensembleStatCase->mean();
|
|
|
|
if ( !sourceData ) return false;
|
|
|
|
values->clear();
|
|
values->reserve( sourceData->size() );
|
|
for ( auto val : *sourceData )
|
|
values->push_back( val );
|
|
return true;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
std::string RifEnsembleStatisticsReader::unitName( const RifEclipseSummaryAddress& resultAddress ) const
|
|
{
|
|
std::string retval;
|
|
|
|
// The stat case does not have a unit set, so pick up the unit from one of the input cases, if possible
|
|
auto cases = m_ensembleStatCase->curveSet()->summaryCaseCollection()->allSummaryCases();
|
|
if ( cases.size() > 0 )
|
|
{
|
|
QString qName = QString::fromStdString( resultAddress.vectorName() );
|
|
std::string orgQName = qName.split( ":" )[1].toStdString();
|
|
|
|
RifEclipseSummaryAddress address = RifEclipseSummaryAddress( resultAddress );
|
|
address.setVectorName( orgQName );
|
|
|
|
retval = cases[0]->summaryReader()->unitName( address );
|
|
}
|
|
|
|
return retval;
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
RiaDefines::EclipseUnitSystem RifEnsembleStatisticsReader::unitSystem() const
|
|
{
|
|
return m_ensembleStatCase->unitSystem();
|
|
}
|
|
|
|
//--------------------------------------------------------------------------------------------------
|
|
///
|
|
//--------------------------------------------------------------------------------------------------
|
|
bool RifEnsembleStatisticsReader::validateAddress( const RifEclipseSummaryAddress& address ) const
|
|
{
|
|
return address.category() == RifEclipseSummaryAddress::SUMMARY_ENSEMBLE_STATISTICS && !address.vectorName().empty();
|
|
}
|