import sys
import os
import math

sys.path.insert(1, os.path.join(sys.path[0], '../../'))
import rips

import dataroot

def check_corner(actual, expected):
    assert(math.isclose(actual.x, expected[0], abs_tol=0.1))
    assert(math.isclose(actual.y, expected[1], abs_tol=0.1))
    assert(math.isclose(actual.z, expected[2], abs_tol=0.1))

def test_10k(rips_instance, initialize_test):
    casePath = dataroot.PATH + "/TEST10K_FLT_LGR_NNC/TEST10K_FLT_LGR_NNC.EGRID"
    case = rips_instance.project.load_case(path=casePath)
    assert(len(case.grids()) == 2)
    grid = case.grid(index=0)
    dimensions = grid.dimensions()
    assert(dimensions.i == 90)
    assert(dimensions.j == 96)
    assert(dimensions.k == 36)

    cell_centers = grid.cell_centers()
    assert(len(cell_centers) == (dimensions.i * dimensions.j * dimensions.k))

    # Test a specific cell (results from ResInsight UI)
    cell_index = 168143
    assert(math.isclose(3627.17, cell_centers[cell_index].x, abs_tol=0.1))
    assert(math.isclose(5209.75, cell_centers[cell_index].y, abs_tol=0.1))
    assert(math.isclose(4179.6, cell_centers[cell_index].z, abs_tol=0.1))

    cell_corners = grid.cell_corners()
    assert(len(cell_corners) == (dimensions.i * dimensions.j * dimensions.k))

    # Expected values from ResInsight UI
    expected_corners = [[ 3565.22, 5179.02, 4177.18],
                        [ 3655.67, 5145.34, 4176.63],
                        [ 3690.07, 5240.69, 4180.02],
                        [ 3599.87, 5275.16, 4179.32],
                        [ 3564.13, 5178.61, 4179.75],
                        [ 3654.78, 5144.79, 4179.23],
                        [ 3688.99, 5239.88, 4182.7],
                        [ 3598.62, 5274.48, 4181.96]]
    check_corner(cell_corners[cell_index].c0, expected_corners[0])
    check_corner(cell_corners[cell_index].c1, expected_corners[1])
    check_corner(cell_corners[cell_index].c2, expected_corners[2])
    check_corner(cell_corners[cell_index].c3, expected_corners[3])
    check_corner(cell_corners[cell_index].c4, expected_corners[4])
    check_corner(cell_corners[cell_index].c5, expected_corners[5])
    check_corner(cell_corners[cell_index].c6, expected_corners[6])
    check_corner(cell_corners[cell_index].c7, expected_corners[7])