In [ ]:

    

%matplotlib inline
import os
import sys
sys.path.append('/home/ychen/lib/util')
import util
import numpy as np
#import matplotlib
import yt
yt.mylog.setLevel("WARNING")
import matplotlib
matplotlib.rcParams['figure.dpi'] = 100
from matplotlib import pyplot as plt
from plotSlices import plotSliceField
from tools import calcDen0
from particles.particle_filters import *


    

In [ ]:

    

#dir = '/home/ychen/data/0only_1204_M24_b01'
#dir = '/home/ychen/data/0only_1022_h1_10Myr'
#dir ='/home/ychen/d9/FLASH4/stampede/0314_L45_M10_b1_h1_nojiggle/'
#dir ='/home/ychen/data/0only_1106_M3_h1/'
#dir ='/home/ychen/d9/FLASH4/2015_production_runs/1022_L45_M10_b1_h1_10Myr/'
#dir ='/home/ychen/d9/FLASH4/Gravity_test/0627_tree_gravity/'
dir = '/home/ychen/data/0only_0529_h1/data/'

def rescan(printlist=False):
    files = util.scan_files(dir, '*hdf5_plt_cnt_[0-9][0-9][0-9][0-9]', printlist=printlist)
    return files

files = rescan(True)


    

In [ ]:

    

ds = yt.load(files[60].fullpath)


    

In [ ]:

    

dx = ds.index.get_smallest_dx()


    

In [ ]:

    

center = (5*dx, 0, 0)
plot = yt.SlicePlot(ds, 'z', fields=['velocity_z'], width=(20, 'kpc'), center=center)
frb_p = plot.frb['velocity_z']
plot.set_zlim('velocity_z', -1E7, 1E7)
plot.set_cmap('velocity_z', 'seismic')
plot.show()


    

In [ ]:

    

center = (-5*dx, 0, 0)
plot = yt.SlicePlot(ds, 'z', fields=['velocity_z'], width=(20, 'kpc'), center=center)
frb_m = plot.frb['velocity_z']
plot.set_zlim('velocity_z', -1E7, 1E7)
plot.set_cmap('velocity_z', 'seismic')
plot.show()


    

In [ ]:

    

diff = frb_p - frb_m
im = plt.imshow(diff, vmin=-1E8, vmax=1E8)

plt.set_cmap('seismic')


    

In [ ]:

    

add = frb_p + frb_m
im = plt.imshow(add, vmin=-1E7, vmax=1E7)


    

In [ ]: