import scipy.spatial

from interp.grid import grid as basegrid
from interp.grid import get_simplex_extra_points


class dgrid(basegrid):
    def __init__(self, points, values):
        self.points = points
        self.values = values
        self.triangulation = scipy.spatial.Delaunay(points)
        self.kdtree = scipy.spatial.KDTree(points)

    def get_simplex_extra_points(self, X, extra_points=8):
        return get_simplex_extra_points(X, self.points, self.triangulation,
                                self.kdtree, extra_points=extra_points)