Updated with changes from opm-core master
This commit is contained in:
Kjell W. Kongsvik
@@ -27,6 +27,7 @@
|
||||
|
||||
#include <opm/output/eclipse/EclipseWriter.hpp>
|
||||
#include <opm/core/grid/GridManager.hpp>
|
||||
#include <opm/core/grid/GridHelpers.hpp>
|
||||
#include <opm/core/props/phaseUsageFromDeck.hpp>
|
||||
#include <opm/core/simulator/BlackoilState.hpp>
|
||||
#include <opm/core/simulator/WellState.hpp>
|
||||
@@ -57,89 +58,89 @@ void verifyWellState(const std::string& rst_filename,
|
||||
Opm::EclipseGridConstPtr ecl_grid,
|
||||
Opm::ScheduleConstPtr schedule) {
|
||||
|
||||
well_info_type * well_info = well_info_alloc(ecl_grid->c_ptr());
|
||||
well_info_load_rstfile(well_info, rst_filename.c_str(), false);
|
||||
well_info_type * well_info = well_info_alloc(ecl_grid->c_ptr());
|
||||
well_info_load_rstfile(well_info, rst_filename.c_str(), false);
|
||||
|
||||
//Verify numwells
|
||||
int numwells = well_info_get_num_wells(well_info);
|
||||
BOOST_CHECK(numwells == (int)schedule->numWells());
|
||||
//Verify numwells
|
||||
int numwells = well_info_get_num_wells(well_info);
|
||||
BOOST_CHECK(numwells == (int)schedule->numWells());
|
||||
|
||||
std::vector<Opm::WellConstPtr> wells = schedule->getWells();
|
||||
std::vector<Opm::WellConstPtr> wells = schedule->getWells();
|
||||
|
||||
for (int i = 0; i < numwells; ++i) {
|
||||
for (int i = 0; i < numwells; ++i) {
|
||||
|
||||
//Verify wellnames
|
||||
const char * wellname = well_info_iget_well_name(well_info, i);
|
||||
Opm::WellConstPtr well = wells.at(i);
|
||||
BOOST_CHECK(wellname == well->name());
|
||||
//Verify wellnames
|
||||
const char * wellname = well_info_iget_well_name(well_info, i);
|
||||
Opm::WellConstPtr well = wells.at(i);
|
||||
BOOST_CHECK(wellname == well->name());
|
||||
|
||||
// Verify well-head position data
|
||||
well_ts_type * well_ts = well_info_get_ts(well_info , wellname);
|
||||
well_state_type * well_state = well_ts_iget_state(well_ts, 0);
|
||||
const well_conn_type * well_head = well_state_get_wellhead(well_state, ECL_GRID_GLOBAL_GRID);
|
||||
BOOST_CHECK(well_conn_get_i(well_head) == well->getHeadI());
|
||||
BOOST_CHECK(well_conn_get_j(well_head) == well->getHeadJ());
|
||||
// Verify well-head position data
|
||||
well_ts_type * well_ts = well_info_get_ts(well_info , wellname);
|
||||
well_state_type * well_state = well_ts_iget_state(well_ts, 0);
|
||||
const well_conn_type * well_head = well_state_get_wellhead(well_state, ECL_GRID_GLOBAL_GRID);
|
||||
BOOST_CHECK(well_conn_get_i(well_head) == well->getHeadI());
|
||||
BOOST_CHECK(well_conn_get_j(well_head) == well->getHeadJ());
|
||||
|
||||
for (int j = 0; j < well_ts_get_size(well_ts); ++j) {
|
||||
well_state = well_ts_iget_state(well_ts, j);
|
||||
for (int j = 0; j < well_ts_get_size(well_ts); ++j) {
|
||||
well_state = well_ts_iget_state(well_ts, j);
|
||||
|
||||
//Verify welltype
|
||||
int ert_well_type = well_state_get_type(well_state);
|
||||
WellType welltype = well->isProducer(j) ? PRODUCER : INJECTOR;
|
||||
Opm::WellInjector::TypeEnum injectortype = well->getInjectionProperties(j).injectorType;
|
||||
int ecl_converted_welltype = Opm::EclipseWriter::eclipseWellTypeMask(welltype, injectortype);
|
||||
int ert_converted_welltype = well_state_translate_ecl_type_int(ecl_converted_welltype);
|
||||
BOOST_CHECK(ert_well_type == ert_converted_welltype);
|
||||
//Verify welltype
|
||||
int ert_well_type = well_state_get_type(well_state);
|
||||
WellType welltype = well->isProducer(j) ? PRODUCER : INJECTOR;
|
||||
Opm::WellInjector::TypeEnum injectortype = well->getInjectionProperties(j).injectorType;
|
||||
int ecl_converted_welltype = Opm::EclipseWriter::eclipseWellTypeMask(welltype, injectortype);
|
||||
int ert_converted_welltype = well_state_translate_ecl_type_int(ecl_converted_welltype);
|
||||
BOOST_CHECK(ert_well_type == ert_converted_welltype);
|
||||
|
||||
//Verify wellstatus
|
||||
int ert_well_status = well_state_is_open(well_state) ? 1 : 0;
|
||||
//Verify wellstatus
|
||||
int ert_well_status = well_state_is_open(well_state) ? 1 : 0;
|
||||
|
||||
Opm::WellCommon::StatusEnum status = well->getStatus(j);
|
||||
int wellstatus = Opm::EclipseWriter::eclipseWellStatusMask(status);
|
||||
Opm::WellCommon::StatusEnum status = well->getStatus(j);
|
||||
int wellstatus = Opm::EclipseWriter::eclipseWellStatusMask(status);
|
||||
|
||||
BOOST_CHECK(ert_well_status == wellstatus);
|
||||
BOOST_CHECK(ert_well_status == wellstatus);
|
||||
|
||||
//Verify number of completion connections
|
||||
const well_conn_collection_type * well_connections = well_state_get_global_connections( well_state );
|
||||
size_t num_wellconnections = well_conn_collection_get_size(well_connections);
|
||||
//Verify number of completion connections
|
||||
const well_conn_collection_type * well_connections = well_state_get_global_connections( well_state );
|
||||
size_t num_wellconnections = well_conn_collection_get_size(well_connections);
|
||||
|
||||
int report_nr = well_state_get_report_nr(well_state);
|
||||
Opm::CompletionSetConstPtr completions_set = well->getCompletions((size_t)report_nr);
|
||||
int report_nr = well_state_get_report_nr(well_state);
|
||||
Opm::CompletionSetConstPtr completions_set = well->getCompletions((size_t)report_nr);
|
||||
|
||||
BOOST_CHECK(num_wellconnections == completions_set->size());
|
||||
BOOST_CHECK(num_wellconnections == completions_set->size());
|
||||
|
||||
//Verify coordinates for each completion connection
|
||||
for (size_t k = 0; k < num_wellconnections; ++k) {
|
||||
const well_conn_type * well_connection = well_conn_collection_iget_const(well_connections , k);
|
||||
//Verify coordinates for each completion connection
|
||||
for (size_t k = 0; k < num_wellconnections; ++k) {
|
||||
const well_conn_type * well_connection = well_conn_collection_iget_const(well_connections , k);
|
||||
|
||||
Opm::CompletionConstPtr completion = completions_set->get(k);
|
||||
Opm::CompletionConstPtr completion = completions_set->get(k);
|
||||
|
||||
BOOST_CHECK(well_conn_get_i(well_connection) == completion->getI());
|
||||
BOOST_CHECK(well_conn_get_j(well_connection) == completion->getJ());
|
||||
BOOST_CHECK(well_conn_get_k(well_connection) == completion->getK());
|
||||
}
|
||||
}
|
||||
BOOST_CHECK(well_conn_get_i(well_connection) == completion->getI());
|
||||
BOOST_CHECK(well_conn_get_j(well_connection) == completion->getJ());
|
||||
BOOST_CHECK(well_conn_get_k(well_connection) == completion->getK());
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
well_info_free(well_info);
|
||||
well_info_free(well_info);
|
||||
}
|
||||
|
||||
|
||||
std::shared_ptr<Opm::BlackoilState> createBlackOilState(Opm::EclipseGridConstPtr eclGrid) {
|
||||
|
||||
std::shared_ptr<Opm::GridManager> ourFineGridManagerPtr(new Opm::GridManager(eclGrid));
|
||||
std::shared_ptr<Opm::BlackoilState> blackoilState(new Opm::BlackoilState);
|
||||
blackoilState->init(*ourFineGridManagerPtr->c_grid(), 3);
|
||||
std::shared_ptr<Opm::GridManager> grid(new Opm::GridManager(eclGrid));
|
||||
const UnstructuredGrid& ug_grid = *(grid->c_grid());
|
||||
std::shared_ptr<Opm::BlackoilState> blackoilState(new Opm::BlackoilState( Opm::UgGridHelpers::numCells(ug_grid) , Opm::UgGridHelpers::numFaces(ug_grid) , 3 ));
|
||||
|
||||
return blackoilState;
|
||||
return blackoilState;
|
||||
}
|
||||
|
||||
|
||||
Opm::DeckConstPtr createDeck(const std::string& eclipse_data_filename) {
|
||||
Opm::ParserPtr parser(new Opm::Parser());
|
||||
Opm::DeckConstPtr deck = parser->parseFile(eclipse_data_filename , Opm::ParseContext());
|
||||
Opm::ParserPtr parser(new Opm::Parser());
|
||||
Opm::DeckConstPtr deck = parser->parseFile(eclipse_data_filename , Opm::ParseContext());
|
||||
|
||||
return deck;
|
||||
return deck;
|
||||
}
|
||||
|
||||
|
||||
@@ -147,51 +148,51 @@ Opm::EclipseWriterPtr createEclipseWriter(Opm::DeckConstPtr deck,
|
||||
Opm::EclipseStatePtr eclipseState,
|
||||
std::string& eclipse_data_filename) {
|
||||
|
||||
Opm::parameter::ParameterGroup params;
|
||||
params.insertParameter("deck_filename", eclipse_data_filename);
|
||||
Opm::parameter::ParameterGroup params;
|
||||
params.insertParameter("deck_filename", eclipse_data_filename);
|
||||
|
||||
const Opm::PhaseUsage phaseUsage = Opm::phaseUsageFromDeck(deck);
|
||||
const Opm::PhaseUsage phaseUsage = Opm::phaseUsageFromDeck(deck);
|
||||
|
||||
Opm::EclipseWriterPtr eclWriter(new Opm::EclipseWriter(params,
|
||||
eclipseState,
|
||||
phaseUsage,
|
||||
eclipseState->getEclipseGrid()->getCartesianSize(),
|
||||
0));
|
||||
return eclWriter;
|
||||
Opm::EclipseWriterPtr eclWriter(new Opm::EclipseWriter(params,
|
||||
eclipseState,
|
||||
phaseUsage,
|
||||
eclipseState->getEclipseGrid()->getCartesianSize(),
|
||||
0));
|
||||
return eclWriter;
|
||||
}
|
||||
|
||||
|
||||
BOOST_AUTO_TEST_CASE(EclipseWriteRestartWellInfo)
|
||||
{
|
||||
std::string eclipse_data_filename = "testBlackoilState3.DATA";
|
||||
std::string eclipse_restart_filename = "TESTBLACKOILSTATE3.X0004";
|
||||
{
|
||||
std::string eclipse_data_filename = "testBlackoilState3.DATA";
|
||||
std::string eclipse_restart_filename = "TESTBLACKOILSTATE3.X0004";
|
||||
|
||||
|
||||
test_work_area_type * test_area = test_work_area_alloc("TEST_EclipseWriteNumWells");
|
||||
test_work_area_copy_file(test_area, eclipse_data_filename.c_str());
|
||||
test_work_area_type * test_area = test_work_area_alloc("TEST_EclipseWriteNumWells");
|
||||
test_work_area_copy_file(test_area, eclipse_data_filename.c_str());
|
||||
|
||||
Opm::ParseContext parseContext;
|
||||
Opm::DeckConstPtr deck = createDeck(eclipse_data_filename);
|
||||
Opm::EclipseStatePtr eclipseState(new Opm::EclipseState(deck , parseContext));
|
||||
Opm::EclipseWriterPtr eclipseWriter = createEclipseWriter(deck, eclipseState, eclipse_data_filename);
|
||||
Opm::ParseContext parseContext;
|
||||
Opm::DeckConstPtr deck = createDeck(eclipse_data_filename);
|
||||
Opm::EclipseStatePtr eclipseState(new Opm::EclipseState(deck , parseContext));
|
||||
Opm::EclipseWriterPtr eclipseWriter = createEclipseWriter(deck, eclipseState, eclipse_data_filename);
|
||||
|
||||
std::shared_ptr<Opm::SimulatorTimer> simTimer( new Opm::SimulatorTimer() );
|
||||
simTimer->init(eclipseState->getSchedule()->getTimeMap());
|
||||
std::shared_ptr<Opm::SimulatorTimer> simTimer( new Opm::SimulatorTimer() );
|
||||
simTimer->init(eclipseState->getSchedule()->getTimeMap());
|
||||
|
||||
eclipseWriter->writeInit(*simTimer);
|
||||
eclipseWriter->writeInit(*simTimer);
|
||||
|
||||
std::shared_ptr<Opm::WellState> wellState(new Opm::WellState());
|
||||
std::shared_ptr<Opm::BlackoilState> blackoilState = createBlackOilState(eclipseState->getEclipseGrid());
|
||||
wellState->init(0, *blackoilState);
|
||||
std::shared_ptr<Opm::WellState> wellState(new Opm::WellState());
|
||||
std::shared_ptr<Opm::BlackoilState> blackoilState = createBlackOilState(eclipseState->getEclipseGrid());
|
||||
wellState->init(0, *blackoilState);
|
||||
|
||||
int countTimeStep = eclipseState->getSchedule()->getTimeMap()->numTimesteps();
|
||||
int countTimeStep = eclipseState->getSchedule()->getTimeMap()->numTimesteps();
|
||||
|
||||
for(int timestep=0; timestep <= countTimeStep; ++timestep){
|
||||
simTimer->setCurrentStepNum(timestep);
|
||||
eclipseWriter->writeTimeStep(*simTimer, *blackoilState, *wellState, false);
|
||||
}
|
||||
for(int timestep=0; timestep <= countTimeStep; ++timestep){
|
||||
simTimer->setCurrentStepNum(timestep);
|
||||
eclipseWriter->writeTimeStep(*simTimer, *blackoilState, *wellState, false);
|
||||
}
|
||||
|
||||
verifyWellState(eclipse_restart_filename, eclipseState->getEclipseGrid(), eclipseState->getSchedule());
|
||||
verifyWellState(eclipse_restart_filename, eclipseState->getEclipseGrid(), eclipseState->getSchedule());
|
||||
|
||||
test_work_area_free(test_area);
|
||||
}
|
||||
test_work_area_free(test_area);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user