#!/usr/bin/env python3

import numpy as np
import ndpolator

class Pyterpolmu():
    def __init__(self):
 
        axes = np.loadtxt("gridlist", usecols=[1, 2], unpack=True)
        files = np.loadtxt("gridlist", usecols=[0], dtype="str")

        print("axes[0] = ", axes[0])
        print("axes[1] = ", axes[1])

        self.ndp = ndpolator.Ndpolator(basic_axes=(axes[0], axes[1]))

        grid = np.empty((len(axes[0]), len(axes[1]), 20001))

        grid[:, :, :] = None

        for i, f in enumerate(files):
            grid[i, i, :] = 1.0

        print("grid = ", grid)

        self.ndp.register(table='main', associated_axes=None, grid=grid)

        query_pts = [[6000.0, 4.25]]

        print("query_pts = ", query_pts)

        interps = self.ndp.ndpolate(table='main', query_pts=query_pts, extrapolation_method='linear')

        print("interps = ", interps['interps'])

def main():
    pyterpolmu = Pyterpolmu()


if __name__ == "__main__":
    main()