/*
  Copyright 2015 Statoil ASA.

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


#define BOOST_TEST_MODULE TuningTests

#include <boost/test/unit_test.hpp>

#include <opm/parser/eclipse/Deck/Deck.hpp>
#include <opm/parser/eclipse/Parser/Parser.hpp>
#include <opm/parser/eclipse/EclipseState/Grid/EclipseGrid.hpp>
#include <opm/parser/eclipse/EclipseState/Schedule/Schedule.hpp>
#include <opm/parser/eclipse/EclipseState/Schedule/Tuning.hpp>
#include <opm/parser/eclipse/Units/Units.hpp>


using namespace Opm;


const std::string& deckStr =  "START\n"
                              " 21 MAY 1981 /\n"
                              "\n"
                              "SCHEDULE\n"
                              "TSTEP\n"
                              " 1 2 3 4 5 /\n"
                              "\n"
                              "TUNING\n"
                              "2 300 0.3 0.30 6 0.6 0.2 2.25 2E20 /\n"
                              "0.2 0.002 2E-7 0.0002 11 0.02 2.0E-6 0.002 0.002 0.035 66 0.02 2/\n"
                              "13 2 26 2 9 9 4.0E6 4.0E6 4.0E6 1/\n"
                              "DATES\n"
                              " 1 JAN 1982 /\n"
                              " 1 JAN 1982 13:55:44 /\n"
                              " 3 JAN 1982 14:56:45.123 /\n"
                              "/\n"
                              "TSTEP\n"
                              " 9 10 /\n"
                              "\n"
                              "TUNING\n"
                              "2 300 0.3 0.30 6 0.6 0.2 2.25 2E20 10.0/\n"
                              "/\n"
                              "/\n";



static Deck createDeck(const std::string& input) {
    Opm::Parser parser;
    return parser.parseString(input);
}



BOOST_AUTO_TEST_CASE(TuningTest) {

  auto deck = createDeck(deckStr);
  EclipseGrid grid(10,10,10);
  TableManager table ( deck );
  Eclipse3DProperties eclipseProperties ( deck , table, grid);
  Runspec runspec (deck);
  Schedule schedule( deck, grid , eclipseProperties, runspec);
  auto tuning = schedule.getTuning();
  auto event = schedule.getEvents();

  const double diff = 1.0e-14;

  /*** TIMESTEP 4***/
  /********* Record 1 ***********/
  size_t timestep = 4;
  BOOST_CHECK(!event.hasEvent(ScheduleEvents::TUNING_CHANGE, timestep));

  double TSINIT_default = tuning.getTSINIT(timestep);
  BOOST_CHECK_CLOSE(TSINIT_default, 1 * Metric::Time, diff);

  double TSMAXZ_default = tuning.getTSMAXZ(timestep);
  BOOST_CHECK_CLOSE(TSMAXZ_default, 365 * Metric::Time, diff);

  double TSMINZ_default = tuning.getTSMINZ(timestep);
  BOOST_CHECK_CLOSE(TSMINZ_default, 0.1 * Metric::Time, diff);

  double TSMCHP_default = tuning.getTSMCHP(timestep);
  BOOST_CHECK_CLOSE(TSMCHP_default, 0.15 * Metric::Time, diff);

  double TSFMAX_default = tuning.getTSFMAX(timestep);
  BOOST_CHECK_CLOSE(TSFMAX_default, 3.0, diff);

  double TSFMIN_default = tuning.getTSFMIN(timestep);
  BOOST_CHECK_CLOSE(TSFMIN_default, 0.3, diff);

  double TSFCNV_default = tuning.getTSFCNV(timestep);
  BOOST_CHECK_CLOSE(TSFCNV_default, 0.1, diff);

  double TFDIFF_default = tuning.getTFDIFF(timestep);
  BOOST_CHECK_CLOSE(TFDIFF_default, 1.25, diff);

  double THRUPT_default = tuning.getTHRUPT(timestep);
  BOOST_CHECK_CLOSE(THRUPT_default, 1E20, diff);

  bool TMAXWC_has_value = tuning.getTMAXWChasValue(timestep);
  double TMAXWC_default = tuning.getTMAXWC(timestep);
  BOOST_CHECK_EQUAL(false, TMAXWC_has_value);
  BOOST_CHECK_CLOSE(TMAXWC_default, 0.0 * Metric::Time, diff);


  /********* Record 2 ************/
  double TRGTTE_default = tuning.getTRGTTE(timestep);
  BOOST_CHECK_CLOSE(TRGTTE_default, 0.1, diff);

  double TRGCNV_default = tuning.getTRGCNV(timestep);
  BOOST_CHECK_CLOSE(TRGCNV_default, 0.001, diff);

  double TRGMBE_default = tuning.getTRGMBE(timestep);
  BOOST_CHECK_CLOSE(TRGMBE_default, 1.0E-7, diff);

  double TRGLCV_default = tuning.getTRGLCV(timestep);
  BOOST_CHECK_CLOSE(TRGLCV_default, 0.0001, diff);

  double XXXTTE_default = tuning.getXXXTTE(timestep);
  BOOST_CHECK_CLOSE(XXXTTE_default, 10.0, diff);

  double XXXCNV_default = tuning.getXXXCNV(timestep);
  BOOST_CHECK_CLOSE(XXXCNV_default, 0.01, diff);

  double XXXMBE_default = tuning.getXXXMBE(timestep);
  BOOST_CHECK_CLOSE(XXXMBE_default, 1.0E-6, diff);

  double XXXLCV_default = tuning.getXXXLCV(timestep);
  BOOST_CHECK_CLOSE(XXXLCV_default, 0.001, diff);

  double XXXWFL_default = tuning.getXXXWFL(timestep);
  BOOST_CHECK_CLOSE(XXXWFL_default, 0.001, diff);

  double TRGFIP_default = tuning.getTRGFIP(timestep);
  BOOST_CHECK_CLOSE(TRGFIP_default, 0.025, diff);

  bool TRGSFT_has_value = tuning.getTRGSFThasValue(timestep);
  double TRGSFT_default = tuning.getTRGSFT(timestep);
  BOOST_CHECK_EQUAL(false, TRGSFT_has_value);
  BOOST_CHECK_CLOSE(TRGSFT_default, 0.0, diff);

  double THIONX_default = tuning.getTHIONX(timestep);
  BOOST_CHECK_CLOSE(THIONX_default, 0.01, diff);

  int TRWGHT_default = tuning.getTRWGHT(timestep);
  BOOST_CHECK_EQUAL(TRWGHT_default, 1);


  /********* Record 3 ************/
  int NEWTMX_default = tuning.getNEWTMX(timestep);
  BOOST_CHECK_EQUAL(NEWTMX_default, 12);

  int NEWTMN_default = tuning.getNEWTMN(timestep);
  BOOST_CHECK_EQUAL(NEWTMN_default, 1);

  int LITMAX_default = tuning.getLITMAX(timestep);
  BOOST_CHECK_EQUAL(LITMAX_default, 25);

  int LITMIN_default = tuning.getLITMIN(timestep);
  BOOST_CHECK_EQUAL(LITMIN_default, 1);

  int MXWSIT_default = tuning.getMXWSIT(timestep);
  BOOST_CHECK_EQUAL(MXWSIT_default, 8);

  int MXWPIT_default = tuning.getMXWPIT(timestep);
  BOOST_CHECK_EQUAL(MXWPIT_default, 8);

  double DDPLIM_default = tuning.getDDPLIM(timestep);
  BOOST_CHECK_CLOSE(DDPLIM_default, 1.0E6 * Metric::Pressure, diff);

  double DDSLIM_default = tuning.getDDSLIM(timestep);
  BOOST_CHECK_CLOSE(DDSLIM_default, 1.0E6, diff);

  double TRGDPR_default = tuning.getTRGDPR(timestep);
  BOOST_CHECK_CLOSE(TRGDPR_default, 1.0E6 * Metric::Pressure, diff);

  bool XXXDPR_has_value = tuning.getXXXDPRhasValue(timestep);
  double XXXDPR_default = tuning.getXXXDPR(timestep);
  BOOST_CHECK_EQUAL(false, XXXDPR_has_value);
  BOOST_CHECK_CLOSE(XXXDPR_default, 0.0, diff);




  /*** TIMESTEP 5***/
  /********* Record 1 ***********/
  timestep = 5;
  BOOST_CHECK(event.hasEvent(ScheduleEvents::TUNING_CHANGE, timestep));
  double TSINIT = tuning.getTSINIT(timestep);
  BOOST_CHECK_CLOSE(TSINIT, 2 * Metric::Time, diff);

  double TSMAXZ = tuning.getTSMAXZ(timestep);
  BOOST_CHECK_CLOSE(TSMAXZ, 300 * Metric::Time, diff);

  double TSMINZ = tuning.getTSMINZ(timestep);
  BOOST_CHECK_CLOSE(TSMINZ, 0.3 * Metric::Time, diff);

  double TSMCHP = tuning.getTSMCHP(timestep);
  BOOST_CHECK_CLOSE(TSMCHP, 0.30 * Metric::Time, diff);

  double TSFMAX = tuning.getTSFMAX(timestep);
  BOOST_CHECK_CLOSE(TSFMAX, 6.0, 1.0);

  double TSFMIN = tuning.getTSFMIN(timestep);
  BOOST_CHECK_CLOSE(TSFMIN, 0.6, 1.0);

  double TSFCNV = tuning.getTSFCNV(timestep);
  BOOST_CHECK_CLOSE(TSFCNV, 0.2, diff);

  double TFDIFF = tuning.getTFDIFF(timestep);
  BOOST_CHECK_CLOSE(TFDIFF, 2.25, diff);

  double THRUPT = tuning.getTHRUPT(timestep);
  BOOST_CHECK_CLOSE(THRUPT, 2E20, diff);

  TMAXWC_has_value = tuning.getTMAXWChasValue(timestep);
  TMAXWC_default = tuning.getTMAXWC(timestep);
  BOOST_CHECK_EQUAL(false, TMAXWC_has_value);
  BOOST_CHECK_CLOSE(TMAXWC_default, 0.0 * Metric::Time, diff);

  /********* Record 2 ***********/
  double TRGTTE = tuning.getTRGTTE(timestep);
  BOOST_CHECK_CLOSE(TRGTTE, 0.2, diff);

  double TRGCNV = tuning.getTRGCNV(timestep);
  BOOST_CHECK_CLOSE(TRGCNV, 0.002, diff);

  double TRGMBE = tuning.getTRGMBE(timestep);
  BOOST_CHECK_CLOSE(TRGMBE, 2.0E-7, diff);

  double TRGLCV = tuning.getTRGLCV(timestep);
  BOOST_CHECK_CLOSE(TRGLCV, 0.0002, diff);

  double XXXTTE = tuning.getXXXTTE(timestep);
  BOOST_CHECK_CLOSE(XXXTTE, 11.0, diff);

  double XXXCNV = tuning.getXXXCNV(timestep);
  BOOST_CHECK_CLOSE(XXXCNV, 0.02, diff);

  double XXXMBE = tuning.getXXXMBE(timestep);
  BOOST_CHECK_CLOSE(XXXMBE, 2.0E-6, diff);

  double XXXLCV = tuning.getXXXLCV(timestep);
  BOOST_CHECK_CLOSE(XXXLCV, 0.002, diff);

  double XXXWFL = tuning.getXXXWFL(timestep);
  BOOST_CHECK_CLOSE(XXXWFL, 0.002, diff);

  double TRGFIP = tuning.getTRGFIP(timestep);
  BOOST_CHECK_CLOSE(TRGFIP, 0.035, diff);

  TRGSFT_has_value = tuning.getTRGSFThasValue(timestep);
  double TRGSFT = tuning.getTRGSFT(timestep);
  BOOST_CHECK_EQUAL(true, TRGSFT_has_value);
  BOOST_CHECK_CLOSE(TRGSFT, 66.0, diff);

  double THIONX = tuning.getTHIONX(timestep);
  BOOST_CHECK_CLOSE(THIONX, 0.02, diff);

  int TRWGHT = tuning.getTRWGHT(timestep);
  BOOST_CHECK_EQUAL(TRWGHT, 2);

  /********* Record 3 ***********/
  int NEWTMX = tuning.getNEWTMX(timestep);
  BOOST_CHECK_EQUAL(NEWTMX, 13);

  int NEWTMN = tuning.getNEWTMN(timestep);
  BOOST_CHECK_EQUAL(NEWTMN, 2);

  int LITMAX = tuning.getLITMAX(timestep);
  BOOST_CHECK_EQUAL(LITMAX, 26);

  int LITMIN = tuning.getLITMIN(timestep);
  BOOST_CHECK_EQUAL(LITMIN, 2);

  int MXWSIT = tuning.getMXWSIT(timestep);
  BOOST_CHECK_EQUAL(MXWSIT, 9);

  int MXWPIT = tuning.getMXWPIT(timestep);
  BOOST_CHECK_EQUAL(MXWPIT, 9);

  double DDPLIM= tuning.getDDPLIM(timestep);
  BOOST_CHECK_CLOSE(DDPLIM, 4.0E6 * Metric::Pressure, diff);

  double DDSLIM= tuning.getDDSLIM(timestep);
  BOOST_CHECK_CLOSE(DDSLIM, 4.0E6, diff);

  double TRGDPR = tuning.getTRGDPR(timestep);
  BOOST_CHECK_CLOSE(TRGDPR, 4.0E6 * Metric::Pressure, diff);

  XXXDPR_has_value = tuning.getXXXDPRhasValue(timestep);
  double XXXDPR = tuning.getXXXDPR(timestep);
  BOOST_CHECK_EQUAL(true, XXXDPR_has_value);
  BOOST_CHECK_CLOSE(XXXDPR, 1.0 * Metric::Pressure, diff);


  /*** TIMESTEP 7 ***/
  BOOST_CHECK(!event.hasEvent(ScheduleEvents::TUNING_CHANGE, 7));

  /*** TIMESTEP 10 ***/
  /********* Record 1 ***********/
  timestep = 10;
  BOOST_CHECK(event.hasEvent(ScheduleEvents::TUNING_CHANGE, timestep));
  TMAXWC_has_value = tuning.getTMAXWChasValue(timestep);
  TMAXWC_default = tuning.getTMAXWC(timestep);
  BOOST_CHECK_EQUAL(true, TMAXWC_has_value);
  BOOST_CHECK_CLOSE(TMAXWC_default, 10.0 * Metric::Time, diff);



}


BOOST_AUTO_TEST_CASE(TuningInitTest) {

  auto deck = createDeck(deckStr);
  EclipseGrid grid(10,10,10);
  TableManager table ( deck );
  Eclipse3DProperties eclipseProperties ( deck , table, grid);
  Runspec runspec (deck);
  Schedule schedule(deck , grid , eclipseProperties, runspec);
  auto tuning = schedule.getTuning();


  const double diff = 1.0e-14;

  /*** TIMESTEP 4***/
  /********* Record 1 ***********/
  size_t timestep = 0;
  double value = 100.00;
  tuning.setTuningInitialValue("TSINIT",value, false);
  double TSINIT_default = tuning.getTSINIT(timestep);
  BOOST_CHECK_CLOSE(TSINIT_default, 100.00, diff);

  timestep = 10;
  bool TMAXWC_has_value = tuning.getTMAXWChasValue(timestep);
  double TMAXWC_default = tuning.getTMAXWC(timestep);
  BOOST_CHECK_EQUAL(true, TMAXWC_has_value);
  BOOST_CHECK_CLOSE(TMAXWC_default, 10.0 * Metric::Time, diff);


  }


BOOST_AUTO_TEST_CASE(TuningResetTest) {

  auto deck = createDeck(deckStr);
  EclipseGrid grid(10,10,10);
  TableManager table ( deck );
  Eclipse3DProperties eclipseProperties ( deck , table, grid);
  Runspec runspec (deck);
  Schedule schedule(deck, grid , eclipseProperties, runspec);
  auto tuning = schedule.getTuning();


  const double diff = 1.0e-14;

  /*** TIMESTEP 4***/
  /********* Record 1 ***********/
  size_t timestep = 0;
  double value = 666.00;
  tuning.setTuningInitialValue("TSINIT",value, true);
  double TSINIT_default = tuning.getTSINIT(timestep);
  BOOST_CHECK_CLOSE(TSINIT_default, 666.00, diff);

  timestep = 10;
  TSINIT_default = tuning.getTSINIT(timestep);
  BOOST_CHECK_CLOSE(TSINIT_default, 666.00, diff);

}