/*
  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>

#define NVERBOSE  // Suppress own messages when throw()ing

#define BOOST_TEST_MODULE WellsModuleTest
#include <boost/test/unit_test.hpp>

#include <opm/core/wells.h>
#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 ));
    
    {
        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;
        double alq = 88;
        int vfp = 42;

        well_controls_assert_number_of_phases( ctrls , num_phases );
        well_controls_add_new( type1 ,   target ,   alq ,   vfp , dist1 , ctrls );
        well_controls_add_new( type2 , 2*target , 2*alq , 2*vfp , dist2 , ctrls );

        BOOST_CHECK_EQUAL( target , well_controls_iget_target(ctrls , 0 ));
        BOOST_CHECK_EQUAL( alq , well_controls_iget_alq(ctrls , 0 ));
        BOOST_CHECK_EQUAL( vfp , well_controls_iget_vfp(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( 2*alq , well_controls_iget_alq(ctrls , 1 ));
        BOOST_CHECK_EQUAL( 2*vfp , well_controls_iget_vfp(ctrls , 1 ));
        BOOST_CHECK_EQUAL( type2 , well_controls_iget_type(ctrls , 1 ));
        well_controls_set_current( ctrls , 1 );
        BOOST_CHECK_EQUAL( type2 , well_controls_get_current_type( ctrls ));

        BOOST_CHECK_EQUAL( well_controls_iget_target( ctrls , 1 ) , well_controls_get_current_target( ctrls ));

        {
            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_alq( ctrls , 0 , 789);
    BOOST_CHECK_EQUAL( 789 , well_controls_iget_alq( ctrls , 0 ));
    well_controls_iset_alq( ctrls , 1 , 234);
    BOOST_CHECK_EQUAL( 234 , well_controls_iget_alq( ctrls , 1 ));

    well_controls_iset_vfp( ctrls , 0 , 567);
    BOOST_CHECK_EQUAL( 567 , well_controls_iget_vfp( ctrls , 0 ));
    well_controls_iset_vfp( ctrls , 1 , 890);
    BOOST_CHECK_EQUAL( 890 , well_controls_iget_vfp( 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 ));
    

    {
        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 );
}


BOOST_AUTO_TEST_CASE(OpenClose)
{
    struct WellControls * ctrls = well_controls_create();

    BOOST_CHECK_EQUAL( true , well_controls_well_is_open(ctrls) );
    BOOST_CHECK_EQUAL( false , well_controls_well_is_stopped(ctrls) );

    well_controls_open_well( ctrls );
    BOOST_CHECK_EQUAL( true   , well_controls_well_is_open(ctrls) );
    BOOST_CHECK_EQUAL( false  , well_controls_well_is_stopped(ctrls) );

    well_controls_stop_well( ctrls );
    BOOST_CHECK_EQUAL( false   , well_controls_well_is_open(ctrls) );
    BOOST_CHECK_EQUAL( true    , well_controls_well_is_stopped(ctrls) );
    
    well_controls_destroy( ctrls );
}


BOOST_AUTO_TEST_CASE(Clone)
{
    std::shared_ptr<WellControls>
        ctrls(well_controls_create(),
              & well_controls_destroy);

    const WellControlType type1 = BHP;
    const WellControlType type2 = SURFACE_RATE;

    const int num_phases = 3;
    const double dist1[] = { 0,  1,  2};
    const double dist2[] = {10, 11, 12};
    const double target  = 77;
    const double alq     = 88;
    const int vfp        = 42;

    well_controls_assert_number_of_phases(ctrls.get(), num_phases);
    well_controls_add_new(type1,   target,   alq,   vfp, dist1, ctrls.get());
    well_controls_add_new(type2, 2*target, 2*alq, 2*vfp, dist2, ctrls.get());

    std::shared_ptr<WellControls>
        c(well_controls_clone(ctrls.get()),
          & well_controls_destroy);

    BOOST_CHECK(well_controls_equal(ctrls.get(), c.get(), false));
}