#include <config.h>

#include <opm/core/grid/GridManager.hpp>
#include <opm/core/grid/GridHelpers.hpp>
#include <opm/core/props/BlackoilPhases.hpp>

#include <opm/parser/eclipse/Parser/Parser.hpp>
#include <opm/parser/eclipse/Parser/ParseContext.hpp>
#include <opm/parser/eclipse/Deck/Deck.hpp>

#if HAVE_DYNAMIC_BOOST_TEST
#define BOOST_TEST_DYN_LINK
#endif
#define NVERBOSE // to suppress our messages when throwing
#define BOOST_TEST_MODULE BlackoilStateTest
#define BOOST_TEST_MAIN
#include <boost/test/unit_test.hpp>
#include <memory>
#include <iostream>
#include <iterator>
#include <vector>
#include <string>

#include "opm/core/grid/GridManager.hpp"
#include "opm/core/simulator/BlackoilState.hpp"

using namespace Opm;
using namespace std;


// ACTNUM 1 998*2 3
std::vector<int> get_testBlackoilStateActnum() {
    std::vector<int> actnum(10 * 10 * 10, 2);
    actnum.front() = 1;
    actnum.back() = 3;
    return actnum;
}

BOOST_AUTO_TEST_CASE(EqualsDifferentDeckReturnFalse) {
    const string filename1 = "testBlackoilState1.DATA";
    const string filename2 = "testBlackoilState2.DATA";

    const auto es1 = Opm::Parser::parse(filename1);
    auto eg1 = es1.getInputGridCopy();
    std::vector<int> actnum = get_testBlackoilStateActnum();
    eg1->resetACTNUM(actnum.data());

    const auto es2 = Opm::Parser::parse(filename2);
    auto eg2 = es2.getInputGrid();

    GridManager gridManager1(*eg1);
    const UnstructuredGrid& grid1 = *gridManager1.c_grid();

    GridManager gridManager2(*eg2);
    const UnstructuredGrid& grid2 = *gridManager2.c_grid();

    BlackoilState state1( UgGridHelpers::numCells( grid1 ) , UgGridHelpers::numFaces( grid1 ) , 3);
    BlackoilState state2( UgGridHelpers::numCells( grid2 ) , UgGridHelpers::numFaces( grid2 ) , 3);

    BOOST_CHECK( ! state1.equal(state2) );
}




BOOST_AUTO_TEST_CASE(EqualsNumericalDifferenceReturnFalse) {

    const string filename = "testBlackoilState1.DATA";

    const auto es = Opm::Parser::parse(filename);
    auto eg = es.getInputGridCopy();

    std::vector<int> actnum = get_testBlackoilStateActnum();
    eg->resetACTNUM(actnum.data());

    GridManager gridManager(*eg);
    const UnstructuredGrid& grid = *gridManager.c_grid();

    BlackoilState state1( UgGridHelpers::numCells( grid ) , UgGridHelpers::numFaces( grid ) , 3);
    BlackoilState state2( UgGridHelpers::numCells( grid ) , UgGridHelpers::numFaces( grid ) , 3);


    BOOST_CHECK( state1.equal(state2) );
    {
        std::vector<double>& p1 = state1.pressure();
        std::vector<double>& p2 = state2.pressure();
        p1[0] = p1[0] * 2 + 1;

        BOOST_CHECK( ! state1.equal(state2) );
        p1[0] = p2[0];
        BOOST_CHECK(   state1.equal(state2) );
    }
    {
        std::vector<double>& gor1 = state1.gasoilratio();
        std::vector<double>& gor2 = state2.gasoilratio();
        gor1[0] = gor1[0] * 2 + 1;

        BOOST_CHECK( ! state1.equal(state2) );
        gor1[0] = gor2[0];
        BOOST_CHECK(   state1.equal(state2) );
    }
    {
        std::vector<double>& p1 = state1.facepressure();
        std::vector<double>& p2 = state2.facepressure();
        p1[0] = p1[0] * 2 + 1;

        BOOST_CHECK( ! state1.equal(state2) );
        p1[0] = p2[0];
        BOOST_CHECK(   state1.equal(state2) );
    }

    {
        std::vector<double>& f1 = state1.faceflux();
        std::vector<double>& f2 = state2.faceflux();
        if (f1.size() > 0 ) {
            f1[0] = f1[0] * 2 + 1;

            BOOST_CHECK( ! state1.equal(state2) );
            f1[0] = f2[0];
            BOOST_CHECK(   state1.equal(state2) );
        }
    }
    {
        std::vector<double>& sv1 = state1.surfacevol();
        std::vector<double>& sv2 = state2.surfacevol();
        if (sv1.size() > 0) {
            sv1[0] = sv1[0] * 2 + 1;

            BOOST_CHECK( ! state1.equal(state2) );
            sv1[0] = sv2[0];
            BOOST_CHECK(   state1.equal(state2) );
        }
    }
    {
        std::vector<double>& sat1 = state1.saturation();
        std::vector<double>& sat2 = state2.saturation();
        sat1[0] = sat1[0] * 2 + 1;

        BOOST_CHECK( ! state1.equal(state2) );
        sat1[0] = sat2[0];
        BOOST_CHECK(   state1.equal(state2) );
    }
}