test_interp.py

#!/usr/bin/env python
"""Test Cython interpolation"""

from __future__ import division, print_function
import argparse
import os
import sys

from viscid_test_common import next_plot_fname

import numpy as np
import viscid
from viscid.plot import vpyplot as vlt
from matplotlib import pyplot as plt


def run_test(fld, seeds, kind, show=False):
    plt.clf()
    vlt.plot(viscid.interp(fld, seeds, kind=kind))
    plt.title(kind)

    plt.savefig(next_plot_fname(__file__))
    if show:
        vlt.show()

def _main():
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("--show", "--plot", action="store_true")
    args = viscid.vutil.common_argparse(parser, default_verb=0)

    img = np.load(os.path.join(viscid.sample_dir, "logo.npy"))
    x = np.linspace(-1, 1, img.shape[0])
    y = np.linspace(-1, 1, img.shape[1])
    z = np.linspace(-1, 1, img.shape[2])
    logo = viscid.arrays2field([x, y, z], img)

    seeds = viscid.Volume([-0.8, -0.8, 0.0], [0.8, 0.8, 0.0],
                          n=[64, 64, 1])

    run_test(logo, seeds, "Nearest", show=args.show)
    run_test(logo, seeds, "Trilinear", show=args.show)

    return 0

if __name__ == "__main__":
    sys.exit(_main())

##
## EOF
##