Source code for xoak.index.scipy_adapters

from scipy.spatial import cKDTree

from .base import IndexAdapter, register_default


@register_default('scipy_kdtree')
class ScipyKDTreeAdapter(IndexAdapter):
    """Xoak index adapter for :class:`scipy.spatial.cKDTree`."""

    def __init__(self, **kwargs):
        self.index_options = kwargs

    def build(self, points):
        return cKDTree(points, **self.index_options)

    def query(self, kdtree, points):
        return kdtree.query(points)