#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) {

    ParseContext parseContext;
    const string filename1 = "testBlackoilState1.DATA";
    const string filename2 = "testBlackoilState2.DATA";
    Opm::ParserPtr parser(new Opm::Parser());

    Opm::DeckConstPtr deck1(parser->parseFile(filename1, parseContext));
    auto eg1 = std::make_shared<Opm::EclipseGrid>(deck1);
    std::vector<int> actnum = get_testBlackoilStateActnum();
    eg1->resetACTNUM(actnum.data());


    Opm::DeckConstPtr deck2(parser->parseFile(filename2, parseContext));
    auto eg2 = std::make_shared<Opm::EclipseGrid>(deck2);

    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_EQUAL( false , state1.equal(state2) );
}




BOOST_AUTO_TEST_CASE(EqualsNumericalDifferenceReturnFalse) {

    const string filename = "testBlackoilState1.DATA";
    Opm::ParseContext parseContext;
    Opm::ParserPtr parser(new Opm::Parser());
    Opm::DeckConstPtr deck(parser->parseFile(filename , parseContext));
    auto eg = std::make_shared<Opm::EclipseGrid>(deck);
    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_EQUAL( true , state1.equal(state2) );
    {
        std::vector<double>& p1 = state1.pressure();
        std::vector<double>& p2 = state2.pressure();
        p1[0] = p1[0] * 2 + 1;

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

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

        BOOST_CHECK_EQUAL( false , state1.equal(state2) );
        p1[0] = p2[0];
        BOOST_CHECK_EQUAL( true , 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_EQUAL( false , state1.equal(state2) );
            f1[0] = f2[0];
            BOOST_CHECK_EQUAL( true , 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_EQUAL( false , state1.equal(state2) );
            sv1[0] = sv2[0];
            BOOST_CHECK_EQUAL( true , state1.equal(state2) );
        }
    }
    {
        std::vector<double>& sat1 = state1.saturation();
        std::vector<double>& sat2 = state2.saturation();
        sat1[0] = sat1[0] * 2 + 1;

        BOOST_CHECK_EQUAL( false , state1.equal(state2) );
        sat1[0] = sat2[0];
        BOOST_CHECK_EQUAL( true , state1.equal(state2) );
    }
}