In [9]:

    

import numpy as np
import matplotlib.pyplot as plt
import subprocess

def slice_plotter(
        n0 = 0,
        n1 = 4,
        res = 128):
    ax = plt.figure(figsize = (8, 8)).add_axes([.0, .0, 1., 1.])
    ax.set_axis_off()
    for i in range(n0*4, n1*4, 4):
        print('plotting frame ', i)
        data = np.load(
            '/stuff/data/RMHD/2D_slices/data_rs2_u_t{0:0>4x}.npy'.format(i),
            mmap_mode = 'r')
        for j in range(3):
            for k in range(2):
                ax.cla()
                ax.set_axis_off()
                ax.imshow(data[j, :res, :res, k])
                plt.gcf().savefig(
                    'figs/{0}_{1}_t{2:0>4}.png'.format(j, k, i),
                    dpi = res//8,
                    format = 'png')
    return None


    

In [ ]:

    

generate_png = True
generate_gif = True

if generate_png:
    slice_plotter(
        n0 = 0,
        n1 = 0x2a8//4,
        res = 2048)

if generate_gif:
    for j in range(3):
        for k in range(2):
            subprocess.call(['convert',
                             'figs/{0}_{1}_t*.png'.format(j, k),
                             'figs/{0}_{1}.gif'.format(j, k)])


    

    




plotting frame  0
plotting frame 

In [ ]: