def run_mpl_testA(show=False):
logger.info("2D cell centered tests")
x = np.array(np.linspace(-10, 10, 100), dtype=dtype)
y = np.array(np.linspace(-10, 10, 120), dtype=dtype)
z = np.array(np.linspace(-1, 1, 2), dtype=dtype)
fld_s = viscid.empty([x, y, z], center='cell')
Xcc, Ycc, Zcc = fld_s.get_crds_cc(shaped=True) # pylint: disable=unused-variable
fld_s[:, :, :] = ne.evaluate("(sin(Xcc) + cos(Ycc))")
nrows = 4
ncols = 1
plt.subplot2grid((nrows, ncols), (0, 0))
mpl.plot(fld_s, "y=20f", show=False, plot_opts="lin_0")
plt.subplot2grid((nrows, ncols), (1, 0))
mpl.plot(fld_s, "x=0f:20f,y=0f:5f", earth=True, show=False,
plot_opts="x_-10_0,y_0_7")
plt.subplot2grid((nrows, ncols), (2, 0))
mpl.plot(fld_s, "y=0f", show=False, plot_opts="lin_-1_1")
plt.subplot2grid((nrows, ncols), (3, 0))
mpl.plot(fld_s, "z=0f,x=-20f:0f", earth=True, show=False, plot_opts="lin_-5_5")
mpl.tighten()
if show:
mpl.mplshow()
def run_mpl_testB(show=False):
logger.info("3D node centered tests")
x = np.array(np.linspace(-10, 10, 100), dtype=dtype)
y = np.array(np.linspace(-10, 10, 120), dtype=dtype)
z = np.array(np.linspace(-10, 10, 140), dtype=dtype)
fld_s = viscid.empty([x, y, z], center='node')
X, Y, Z = fld_s.get_crds_nc(shaped=True) #pylint: disable=W0612
fld_s[:, :, :] = ne.evaluate("sin(X) + cos(Y) - cos(Z)")
# print("shape: ", fld_s.data.shape)
nrows = 4
ncols = 1
plt.subplot2grid((nrows, ncols), (0, 0))
mpl.plot(fld_s, "z=0,x=:30", earth=True, plot_opts="lin_0")
plt.subplot2grid((nrows, ncols), (1, 0))
mpl.plot(fld_s, "z=0.75f,x=-4:-1,y=-3f:3f", earth=True)
plt.subplot2grid((nrows, ncols), (2, 0))
mpl.plot(fld_s, "x=-0.5f:,y=-3f:3f,z=0f", earth=True)
plt.subplot2grid((nrows, ncols), (3, 0))
mpl.plot(fld_s, "x=0.0f,y=-5.0f:5.0f", earth=True, plot_opts="log,g")
mpl.tighten()
if show:
mpl.mplshow()
def main():
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument("--show", "--plot", action="store_true")
args = vutil.common_argparse(parser)
####### test binary files
f_bin = viscid.load_file(_viscid_root + "/../sample/ath_sample.*.bin")
for i, grid in enumerate(f_bin.iter_times(":")):
plt.subplot2grid((2, 2), (0, i))
mpl.plot(grid["bx"])
plt.subplot2grid((2, 2), (1, i))
mpl.plot(grid["by"])
if args.show:
mpl.tighten()
mpl.mplshow()
plt.clf()
####### test ascii files
f_tab = viscid.load_file(_viscid_root + "/../sample/ath_sample.*.tab")
for i, grid in enumerate(f_tab.iter_times(":")):
plt.subplot2grid((2, 2), (0, i))
mpl.plot(grid["bx"])
plt.subplot2grid((2, 2), (1, i))
mpl.plot(grid["by"])
if args.show:
mpl.tighten()
mpl.mplshow()
plt.clf()
