ff1ad83f08
The assumptions and interface for these two functions are distinct. Init does not have to take a preallocated vector, but post processing requires an input vector. Splits the two and changes their signatures.
84 lines
2.9 KiB
C++
84 lines
2.9 KiB
C++
#include <algorithm>
|
|
#include <limits>
|
|
#include <vector>
|
|
|
|
#include <opm/parser/eclipse/Deck/Deck.hpp>
|
|
#include <opm/parser/eclipse/EclipseState/EclipseState.hpp>
|
|
#include <opm/parser/eclipse/EclipseState/Grid/EclipseGrid.hpp>
|
|
#include <opm/parser/eclipse/EclipseState/Grid/GridProperty.hpp>
|
|
#include <opm/parser/eclipse/EclipseState/Grid/GridPropertyInitializers.hpp>
|
|
#include <opm/parser/eclipse/EclipseState/Tables/RtempvdTable.hpp>
|
|
#include <opm/parser/eclipse/EclipseState/Tables/TableManager.hpp>
|
|
|
|
namespace Opm {
|
|
|
|
template< typename T >
|
|
GridPropertyInitFunction< T >::GridPropertyInitFunction( T x ) :
|
|
constant( x )
|
|
{}
|
|
|
|
template< typename T >
|
|
GridPropertyInitFunction< T >::GridPropertyInitFunction(
|
|
signature fn,
|
|
const Deck& d,
|
|
const EclipseState& state ) :
|
|
f( fn ), deck( &d ), es( &state )
|
|
{}
|
|
|
|
template< typename T >
|
|
std::vector< T > GridPropertyInitFunction< T >::operator()( size_t size ) const {
|
|
if( !this->f ) return std::vector< T >( size, this->constant );
|
|
|
|
return (*this->f)( size, *this->deck, *this->es );
|
|
}
|
|
|
|
template< typename T >
|
|
GridPropertyPostFunction< T >::GridPropertyPostFunction(
|
|
signature fn,
|
|
const Deck& d,
|
|
const EclipseState& state ) :
|
|
f( fn ), deck( &d ), es( &state )
|
|
{}
|
|
|
|
template< typename T >
|
|
void GridPropertyPostFunction< T >::operator()( std::vector< T >& values ) const {
|
|
if( !this->f ) return;
|
|
|
|
return (*this->f)( values, *this->deck, *this->es );
|
|
}
|
|
|
|
std::vector< double > temperature_lookup(
|
|
size_t size,
|
|
const Deck& deck,
|
|
const EclipseState& eclipseState ) {
|
|
|
|
if( !deck.hasKeyword("EQLNUM") ) {
|
|
/* if values are defaulted in the TEMPI keyword, but no
|
|
* EQLNUM is specified, you will get NaNs
|
|
*/
|
|
return std::vector< double >( size, std::numeric_limits< double >::quiet_NaN() );
|
|
}
|
|
|
|
std::vector< double > values( size, 0 );
|
|
|
|
auto tables = eclipseState.getTableManager();
|
|
auto eclipseGrid = eclipseState.getEclipseGrid();
|
|
const auto& rtempvdTables = tables->getRtempvdTables();
|
|
const std::vector< int >& eqlNum = eclipseState.getIntGridProperty("EQLNUM")->getData();
|
|
|
|
for (size_t cellIdx = 0; cellIdx < eqlNum.size(); ++ cellIdx) {
|
|
int cellEquilNum = eqlNum[cellIdx];
|
|
const RtempvdTable& rtempvdTable = rtempvdTables.getTable<RtempvdTable>(cellEquilNum);
|
|
double cellDepth = std::get<2>(eclipseGrid->getCellCenter(cellIdx));
|
|
values[cellIdx] = rtempvdTable.evaluate("Temperature", cellDepth);
|
|
}
|
|
|
|
return values;
|
|
}
|
|
}
|
|
|
|
template class Opm::GridPropertyInitFunction< int >;
|
|
template class Opm::GridPropertyInitFunction< double >;
|
|
template class Opm::GridPropertyPostFunction< int >;
|
|
template class Opm::GridPropertyPostFunction< double >;
|