/*
  Copyright 2019 Equinor.

  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 BOOST_TEST_MODULE NNCSortTest
#include <boost/test/unit_test.hpp>
#include <ebos/nncsorter.hpp>
#include <opm/parser/eclipse/EclipseState/Grid/NNC.hpp>

#include <cmath>

BOOST_AUTO_TEST_CASE(Test1) {
    std::vector<Opm::NNCdata> nncDataIn =
        { {9, 8, 10.0 }, { 1, 2, 3.0 }, { 3, 4, 2.0 }, { 2, 1, 5.0 } };
    std::vector<Opm::NNCdata> editnncData =
        { {20, 5, .001}, { 2, 1, .1}, {3, 4, .001}, {0, 0, 0.0}, {4, 3, 2.0} };
    std::vector<Opm::NNCdata> nncDataOut1 =
        { { 1, 2, 0.3 }, { 1, 2, 0.5 }, { 3, 4, 0.004 }, { 8, 9, 10.0 } };
    std::vector<Opm::NNCdata> nncDataOut2 =
        { { 1, 2, 0.5 }, { 1, 2, 0.3 }, { 3, 4, 0.4 }, { 8, 9, 10.0 } };

    auto nncDataProcessed = Opm::sortNncAndApplyEditnnc(nncDataIn, editnncData);
    BOOST_CHECK(nncDataProcessed.size() == nncDataOut1.size());
    auto expectedNnc1 = nncDataOut1.begin();
    auto expectedNnc2 = nncDataOut2.begin();

    for(const auto& entry: nncDataProcessed) {
        BOOST_CHECK( entry.cell1 == expectedNnc1->cell1);
        BOOST_CHECK( entry.cell2 == expectedNnc1->cell2);
        BOOST_CHECK( std::abs(entry.trans - expectedNnc1->trans) / std::abs(entry.trans) < 1e-5 ||
                std::abs(entry.trans - expectedNnc2->trans) / std::abs(entry.trans) < 1e-5);
        ++expectedNnc1;
        ++expectedNnc2;
    }
}