ResInsight/ApplicationLibCode/ReservoirDataModel/RigResultModifierFactory.cpp
2023-05-11 08:37:58 +02:00

84 lines
3.1 KiB
C++

/////////////////////////////////////////////////////////////////////////////////
//
// Copyright (C) Statoil ASA
// Copyright (C) Ceetron Solutions AS
//
// 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 "RigResultModifierFactory.h"
#include "RigCaseCellResultsData.h"
#include "RigEclipseCaseData.h"
#include "RigEclipseResultAddress.h"
#include "RigResultModifier.h"
#include <cmath>
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
cvf::ref<RigResultModifier> RigResultModifierFactory::createResultModifier( RigEclipseCaseData* eclipseCase,
size_t gridIndex,
RiaDefines::PorosityModelType porosityModel,
size_t timeStepIndex,
const RigEclipseResultAddress& resVarAddr )
{
if ( !eclipseCase ) return nullptr;
if ( !resVarAddr.isValid() )
{
return nullptr;
}
RigGridBase* grid = eclipseCase->grid( gridIndex );
if ( !grid )
{
return nullptr;
}
if ( !eclipseCase->results( porosityModel ) || !eclipseCase->activeCellInfo( porosityModel ) )
{
return nullptr;
}
auto cellResultsData = eclipseCase->results( porosityModel );
if ( !cellResultsData->hasResultEntry( resVarAddr ) ) return nullptr;
auto scalarSetResults = cellResultsData->modifiableCellScalarResultTimesteps( resVarAddr );
if ( timeStepIndex >= scalarSetResults->size() )
{
return nullptr;
}
std::vector<double>* resultValues = nullptr;
if ( timeStepIndex < scalarSetResults->size() )
{
resultValues = &( scalarSetResults->at( timeStepIndex ) );
}
if ( resultValues->empty() ) return nullptr;
bool useGlobalActiveIndex = cellResultsData->isUsingGlobalActiveIndex( resVarAddr );
if ( useGlobalActiveIndex )
{
cvf::ref<RigResultModifier> object =
new RigActiveCellsResultModifier( grid, eclipseCase->activeCellInfo( porosityModel ), resultValues );
return object;
}
cvf::ref<RigResultModifier> object = new RigAllGridCellsResultModifier( grid, resultValues );
return object;
}