17 lines
554 B
Python
17 lines
554 B
Python
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)
|