Added new functions to set target and type and invert value of current control.

CMakeLists_files.cmake

@@ -168,6 +168,7 @@ list (APPEND TEST_SOURCE_FILES
         tests/test_units.cpp
         tests/test_blackoilstate.cpp
         tests/test_wellsmanager.cpp
+        tests/test_wellcontrols.cpp
   )
 
 # originally generated with the command:

opm/core/well_controls.h

@@ -121,12 +121,21 @@ well_controls_get_current( const struct WellControls * ctrl);
 void
 well_controls_set_current( struct WellControls * ctrl, int current);
 
+void
+well_controls_invert_current( struct WellControls * ctrl );
+
 int
 well_controls_add_new(enum WellControlType type , double target , const double * distr , struct WellControls * ctrl);
 
 enum WellControlType 
 well_controls_iget_type(const struct WellControls * ctrl, int control_index);
 
+void
+well_controls_iset_type( struct WellControls * ctrls , int control_index , enum WellControlType type);
+
+void
+well_controls_iset_target(struct WellControls * ctrl, int control_index , double target);
+
 double
 well_controls_iget_target(const struct WellControls * ctrl, int control_index);
 

opm/core/wells/well_controls.c

@@ -139,17 +139,34 @@ well_controls_set_current( struct WellControls * ctrl, int current) {
     ctrl->current = current;
 }
 
+void 
+well_controls_invert_current( struct WellControls * ctrl ) {
+    ctrl->current = ~ctrl->current;
+}
+
 
 enum WellControlType 
 well_controls_iget_type(const struct WellControls * ctrl, int control_index) {
     return ctrl->type[control_index];
 }
 
+void
+well_controls_iset_type( struct WellControls * ctrls , int control_index , enum WellControlType type) {
+    ctrls->type[control_index] = type;
+}
+
+
 double
 well_controls_iget_target(const struct WellControls * ctrl, int control_index) {
     return ctrl->target[control_index];
 }
 
+void
+well_controls_iset_target(struct WellControls * ctrl, int control_index , double target) {
+    ctrl->target[control_index] = target;
+}
+
+
 const double *
 well_controls_iget_distr(const struct WellControls * ctrl, int control_index) {
     int offset = control_index * ctrl->number_of_phases;

tests/test_wellcontrols.cpp

@@ -0,0 +1,93 @@
+/*
+  Copyright 2014 Statoil.
+
+  This file is part of the Open Porous Media project (OPM).
+
+  OPM 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.
+
+  OPM 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 for more details.
+
+  You should have received a copy of the GNU General Public License
+  along with OPM.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <config.h>
+
+#if HAVE_DYNAMIC_BOOST_TEST
+#define BOOST_TEST_DYN_LINK
+#endif
+
+#define NVERBOSE  // Suppress own messages when throw()ing
+
+#define BOOST_TEST_MODULE WellsModuleTest
+#include <boost/test/unit_test.hpp>
+
+#include <opm/core/wells.h>
+#define HAVE_WELLCONTROLS
+#include <opm/core/well_controls.h>
+
+#include <iostream>
+#include <vector>
+#include <memory>
+
+
+BOOST_AUTO_TEST_CASE(Construction)
+{
+    struct WellControls * ctrls = well_controls_create();
+
+    well_controls_set_current( ctrls , 1 );
+    BOOST_CHECK_EQUAL( 1 , well_controls_get_current( ctrls ));
+    well_controls_set_current( ctrls , 2 );
+    BOOST_CHECK_EQUAL( 2 , well_controls_get_current( ctrls ));
+    
+    well_controls_invert_current( ctrls );
+    BOOST_CHECK( well_controls_get_current( ctrls ) < 0 );
+    well_controls_invert_current( ctrls );
+    BOOST_CHECK_EQUAL( 2 , well_controls_get_current( ctrls ));
+
+    {
+        enum WellControlType type1 = BHP;
+        enum WellControlType type2 = SURFACE_RATE;
+        int num_phases = 3;
+        double dist1[3] = {0 , 1  ,  2};
+        double dist2[3] = {10, 11 , 12};
+        double target = 77;
+
+        well_controls_assert_number_of_phases( ctrls , num_phases );
+        well_controls_add_new( type1 ,   target , dist1 , ctrls );
+        well_controls_add_new( type2 , 2*target , dist2 , ctrls );
+
+        BOOST_CHECK_EQUAL( target , well_controls_iget_target(ctrls , 0 ));
+        BOOST_CHECK_EQUAL( type1 , well_controls_iget_type(ctrls , 0 ));
+
+        BOOST_CHECK_EQUAL( 2*target , well_controls_iget_target(ctrls , 1 ));
+        BOOST_CHECK_EQUAL( type2 , well_controls_iget_type(ctrls , 1 ));
+        
+        {
+            const double * d1 = well_controls_iget_distr( ctrls , 0 );
+            const double * d2 = well_controls_iget_distr( ctrls , 1 );
+            BOOST_CHECK( memcmp(d1 , dist1 , num_phases * sizeof * d1 ) == 0);
+            BOOST_CHECK( memcmp(d2 , dist2 , num_phases * sizeof * d2 ) == 0);
+        }
+    }
+    well_controls_iset_target( ctrls , 0 , 123);
+    BOOST_CHECK_EQUAL( 123 , well_controls_iget_target( ctrls , 0 ));
+    well_controls_iset_target( ctrls , 1 , 456);
+    BOOST_CHECK_EQUAL( 456 , well_controls_iget_target( ctrls , 1 ));
+
+    well_controls_iset_type( ctrls , 0 , SURFACE_RATE);
+    BOOST_CHECK_EQUAL( SURFACE_RATE , well_controls_iget_type( ctrls , 0 ));
+    well_controls_iset_type( ctrls , 1 , BHP);
+    BOOST_CHECK_EQUAL( BHP, well_controls_iget_type( ctrls , 1 ));
+    
+
+    well_controls_destroy( ctrls );
+}
+
+