Added new function well_controls_iset_distr().

This commit is contained in:
Joakim Hove 2014-01-06 12:22:15 +01:00
parent 3d171bd961
commit 6098368d08
3 changed files with 20 additions and 0 deletions

View File

@ -142,6 +142,9 @@ well_controls_iget_target(const struct WellControls * ctrl, int control_index);
const double *
well_controls_iget_distr(const struct WellControls * ctrl, int control_index);
void
well_controls_iset_distr(const struct WellControls * ctrl, int control_index, const double * distr);
void
well_controls_assert_number_of_phases(struct WellControls * ctrl , int number_of_phases);

View File

@ -174,6 +174,14 @@ well_controls_iget_distr(const struct WellControls * ctrl, int control_index) {
}
void
well_controls_iset_distr(const struct WellControls * ctrl, int control_index, const double * distr) {
int offset = control_index * ctrl->number_of_phases;
for (int p=0; p < ctrl->number_of_phases; p++)
ctrl->distr[offset + p] = distr[p];
}
void
well_controls_assert_number_of_phases(struct WellControls * ctrl , int number_of_phases) {
if (ctrl->num == 0)

View File

@ -87,6 +87,15 @@ BOOST_AUTO_TEST_CASE(Construction)
BOOST_CHECK_EQUAL( BHP, well_controls_iget_type( ctrls , 1 ));
{
double newDist[3] = {77,78,79};
const double * tmp;
well_controls_iset_distr( ctrls , 0 , newDist );
tmp = well_controls_iget_distr( ctrls , 0);
BOOST_CHECK( memcmp(tmp , newDist , 3 * sizeof * tmp ) == 0);
}
well_controls_destroy( ctrls );
}