/*
Copyright 2017 TNO.

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 AquanconTest

#include <boost/test/unit_test.hpp>
#include <opm/parser/eclipse/Parser/Parser.hpp>
#include <opm/parser/eclipse/EclipseState/Aquancon.hpp>

using namespace Opm;

inline Deck createAQUANCONDeck() {
    const char *deckData =
        "DIMENS\n"
        "3 3 3 /\n"
        "\n"
        "GRID\n"
        "\n"
        "ACTNUM\n"
        " 0 8*1 0 8*1 0 8*1 /\n"
        "DXV\n"
        "1 1 1 /\n"
        "\n"
        "DYV\n"
        "1 1 1 /\n"
        "\n"
        "DZV\n"
        "1 1 1 /\n"
        "\n"
        "TOPS\n"
        "9*100 /\n"
        "\n"
        "SOLUTION\n"
        "\n"
        "AQUANCON\n"
        "   1      1  1  1    1   1  1  J-  1.0 1.0 NO /\n"
        "   1      1  3  1    3   3  3  I+  0.5 1.0 NO /\n"
        "   1      1  3  1    3   3  3  J+  0.75 1.0 NO /\n"
        "   1      1  3  1    3   3  3  J-  2.75 1.0 NO /\n"
        "   1      2  3  2    3   1  1  I+  2.75 1.0 NO /\n"
        "/ \n";

    Parser parser;
    return parser.parseString(deckData);
}

inline std::vector<Aquancon::AquanconOutput> init_aquancon(){
    auto deck = createAQUANCONDeck();
    EclipseState eclState( deck );
    Aquancon aqucon( eclState.getInputGrid(), deck);
    std::vector<Aquancon::AquanconOutput> aquifers = aqucon.getAquOutput();

    return aquifers;
}

inline std::vector<Aquancon::AquanconOutput> fill_result(){
    auto deck = createAQUANCONDeck();
    EclipseState eclState( deck );
    Aquancon aqucon( eclState.getInputGrid(), deck);
    std::vector<Aquancon::AquanconOutput> aquifers = aqucon.getAquOutput();

    return aquifers;
}

BOOST_AUTO_TEST_CASE(AquanconTest){
    std::vector< Aquancon::AquanconOutput > aquifers = init_aquancon();
    std::vector< Aquancon::AquanconOutput > expected_output = fill_result();

    BOOST_CHECK_EQUAL(aquifers.size(), expected_output.size());
    for (size_t i = 0; i < aquifers.size(); ++i)
    {
        BOOST_CHECK_EQUAL_COLLECTIONS( aquifers.at(i).global_index.begin(), aquifers.at(i).global_index.end(),
                                   expected_output.at(i).global_index.begin(), expected_output.at(i).global_index.end() );
        BOOST_CHECK_EQUAL_COLLECTIONS( aquifers.at(i).influx_multiplier.begin(), aquifers.at(i).influx_multiplier.end(),
                                   expected_output.at(i).influx_multiplier.begin(), expected_output.at(i).influx_multiplier.end() );
        BOOST_CHECK_EQUAL_COLLECTIONS( aquifers.at(i).reservoir_face_dir.begin(), aquifers.at(i).reservoir_face_dir.end(),
                                   expected_output.at(i).reservoir_face_dir.begin(), expected_output.at(i).reservoir_face_dir.end() );
        BOOST_CHECK_EQUAL_COLLECTIONS( aquifers.at(i).record_index.begin(), aquifers.at(i).record_index.end(),
                                   expected_output.at(i).record_index.begin(), expected_output.at(i).record_index.end() );
    }
}