In [ ]:
%matplotlib inline
import pickle
import numpy as np
import matplotlib
matplotlib.rcParams['figure.dpi'] = 150
matplotlib.rcParams['font.family'] = 'stixgeneral'
import matplotlib.pyplot as plt
from matplotlib import cm
from particles_class import *

import yt
e  = yt.utilities.physical_constants.elementary_charge.v #4.803E-10 esu
me = yt.utilities.physical_constants.mass_electron.v #9.109E-28
c  = yt.utilities.physical_constants.speed_of_light.v #2.998E10

In [ ]:
with open('/home/ychen/d12/2015_production_runs/1022_L45_M10_b1_h1_10Myr/particles_trajectory.pickle', 'rb') as f:
    parts = pickle.load(f)

In [ ]:
fig = plt.figure(figsize=(4,15))
ax = fig.add_subplot(111)
for part in parts:
    if part.shok[0] > 0.5:
        continue
    if part.tag % 5 > 0:
        continue
    mask = (part.time < 50) & (part.time > 11)# & (np.abs(part.posy) < 5*3E21)
    #rr = np.sqrt(part.posx**2+part.posy**2)
    #ax.plot(rr[mask]/3.08567758E21, part.posz[mask]/3.08567758E21, lw=1)
    #xlabel = 'r'
    gamc = (part.dens/part.den1**(1./3.))/part.dtau

    B = (part.magx**2 + part.magy**2 + part.magz**2)*np.sqrt(4.0*np.pi)
    
    # Cutoff frequency
    nuc = np.log10(3.0*gamc**2*e*B/(4.0*np.pi*me*c))
    ax.plot(part.posx[mask]/3.08567758E21, part.posz[mask]/3.08567758E21, lw=1, alpha=0.5)
    xlabel = 'x'
    ylabel = 'z'
    ax.set_xlim(-23,23)
    ax.set_ylim(0, 60)
    ax.set_aspect('equal')

    #ax.set_facecolor('k')
    #ax.set_xlim(-xlim,xlim)
    #ax.set_ylim(-ylim,ylim)
    ax.set_xlabel(xlabel+' (kpc)')
    ax.set_ylabel(ylabel+' (kpc)')
    ax.set_aspect('equal')
    ax.tick_params(axis='x', color='grey', direction='in')
    ax.tick_params(axis='y', color='grey', direction='in')
    ax.grid(ls='--', alpha=0.5)

In [ ]: