/*
   Copyright (C) 2017  Equinor ASA, Norway.

   The file 'test_ecl_nnc_data_equinor.c' is part of ERT - Ensemble based Reservoir Tool.

   ERT 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.

   ERT 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 at <http://www.gnu.org/licenses/gpl.html>
   for more details.
*/

#include <ert/ecl/ecl_nnc_data.hpp>
#include <ert/ecl/ecl_kw_magic.hpp>
#include <ert/ecl/ecl_nnc_geometry.hpp>

#include <ert/ecl/ecl_endian_flip.hpp>
#include <ert/ecl/ecl_file.hpp>
#include <ert/ecl/ecl_grid.hpp>
#include <ert/ecl/ecl_kw.hpp>

#include <ert/util/util.h>
#include <ert/util/test_util.hpp>
#include <ert/util/test_work_area.hpp>



int find_index(ecl_nnc_geometry_type * nnc_geo, int grid1, int grid2, int indx1, int indx2) {
   int index = -1;
   int nnc_size = ecl_nnc_geometry_size( nnc_geo );
   for (int n = 0; n < nnc_size; n++) {
      const ecl_nnc_pair_type * pair = ecl_nnc_geometry_iget( nnc_geo, n );
      if (pair->grid_nr1 == grid1 && pair->grid_nr2 == grid2)
          if (pair->global_index1 == indx1 && pair->global_index2 ==indx2) {
          index = n;
          break;
      }
   }
   return index;
}


void test_alloc_file_tran(char * filename) {
   char * grid_file_name = ecl_util_alloc_filename(NULL , filename , ECL_EGRID_FILE , false  , -1);
   char * init_file_name = ecl_util_alloc_filename(NULL , filename , ECL_INIT_FILE , false  , -1);
   ecl_file_type * init_file = ecl_file_open( init_file_name , 0 );
   ecl_grid_type * grid = ecl_grid_alloc( grid_file_name );
   ecl_nnc_geometry_type * nnc_geo = ecl_nnc_geometry_alloc( grid );
   ecl_file_view_type * view_file = ecl_file_get_global_view( init_file );

   ecl_nnc_data_type * nnc_geo_data = ecl_nnc_data_alloc_tran(grid, nnc_geo, view_file);
   test_assert_true( ecl_nnc_geometry_size(nnc_geo) == ecl_nnc_data_get_size(nnc_geo_data) );

   //These numerical values are hand-tuned the specific input file at:
   //${_eclpath}/Troll/MSW_LGR/2BRANCHES-CCEWELLPATH-NEW-SCH-TUNED-AR3
   int index;

   index = find_index( nnc_geo, 0, 0, 541, 14507);
   test_assert_double_equal(13.784438, ecl_nnc_data_iget_value( nnc_geo_data, index) );

   index = find_index( nnc_geo, 0, 0, 48365, 118191);
   test_assert_double_equal(0.580284 , ecl_nnc_data_iget_value( nnc_geo_data, index) );

   index = find_index( nnc_geo, 0, 19, 42830, 211);
   test_assert_double_equal(0.571021 , ecl_nnc_data_iget_value( nnc_geo_data, index) );

   index = find_index( nnc_geo, 0, 79, 132406, 76);
   test_assert_double_equal(37.547710 , ecl_nnc_data_iget_value( nnc_geo_data, index) );

   index = find_index( nnc_geo, 18, 12, 303, 115);
   test_assert_double_equal(0.677443 , ecl_nnc_data_iget_value( nnc_geo_data, index) );

   index = find_index( nnc_geo, 72, 71, 255, 179);
   test_assert_double_equal(0.045813 , ecl_nnc_data_iget_value( nnc_geo_data, index) );

   index = find_index( nnc_geo, 110, 109, 271, 275);
   test_assert_double_equal(16.372242 , ecl_nnc_data_iget_value( nnc_geo_data, index) );

   ecl_nnc_data_free(nnc_geo_data);
   ecl_nnc_geometry_free(nnc_geo);
   ecl_grid_free(grid);
   ecl_file_close(init_file);
   free(grid_file_name);
   free(init_file_name);

}


void test_alloc_file_flux(char * filename, int file_num) {
   char * grid_file_name = ecl_util_alloc_filename(NULL , filename , ECL_EGRID_FILE , false  , -1);
   char * restart_file_name = ecl_util_alloc_filename(NULL , filename , ECL_RESTART_FILE , false  , file_num);

   ecl_file_type * restart_file = ecl_file_open( restart_file_name , 0 );
   ecl_grid_type * grid = ecl_grid_alloc( grid_file_name );
   ecl_nnc_geometry_type * nnc_geo = ecl_nnc_geometry_alloc( grid );
   {
      ecl_file_view_type * view_file = ecl_file_get_global_view( restart_file );

      ecl_nnc_data_type * nnc_flux_data = ecl_nnc_data_alloc_wat_flux(grid, nnc_geo, view_file);
      test_assert_not_NULL(nnc_flux_data);
      ecl_nnc_data_free( nnc_flux_data );
   }
   ecl_nnc_geometry_free(nnc_geo);
   ecl_grid_free(grid);
   ecl_file_close(restart_file);
   free(grid_file_name);
   free(restart_file_name);
}


int main(int argc , char ** argv) {
   test_alloc_file_tran(argv[1]);
   test_alloc_file_flux(argv[2], 0);
   test_alloc_file_flux(argv[2], 6);
   return 0;
}
