def random_color():
return [random.random(),random.random(),random.random()]
#sim1 = 'a08_post_sbc'
if 'frame' not in dir():
frame = 3
def prnt(normlist):
keys = sorted(normlist.keys())
for ke in keys:
values = normlist[ke]
print "%15s"%ke, "%9.2e "*len(values)%tuple(values)
#dir_name1 = '%s/%s'%(basedir,sim1)
#dir_name2 = '%s/%s'%(basedir,sim2)
if 0:
ds_name = dsdiff_helpers.get_ds_name(dir_name,frame)
print ds_name
ds = yt.load(ds_name)
ad = ds.all_data()
den = ad['density']
stat(den.in_units('code_density').v-1.0,'den')
proj = yt.ProjectionPlot(ds,0,'density')
proj.annotate_grids()
outname = sim1+'%04d'%frame
print proj.save(outname)
if 0:
dir1 = dir_name1
dir2 = dir_name2
fields = adiabatic_hydro
frames=[frame]
execfile('go_lite')
import yt
import dsdiff_helpers
base_dir = '/scratch1/dcollins/EnzoProjects/E22_DEF_NonCosmo/'
plot_sequence_name = 'E22_Turb_full'
thelist = ['f01_ppm_nodef', 'f02_ct_nodef', 'f03_ded_nodef']
thelist = ['f01c_ppm_probM1']
thelist = [
'f01_ppm_nodef', 'f03_ded_nodef', 'f05_ct_newversion_newparam',
'f01b_ppm_problem0'
]
thelist += ['f06_ppm_hll']
plot_sequence_name = 'E22_Turb_g'
field = ("enzo", "TotalEnergy")
ofield = 'enzo_TotalEnergy'
field_list = ['temperature', 'density']
for frame in [1] + range(
10, 60, 10
): # range(50): #[0,1,2,5,10,15,20]: #framelist.get(dirname, default_frames):
plt.clf()
for dirname in thelist:
print "frame", frame
short_name = dirname[0:6]
setname = dsdiff_helpers.get_ds_name("%s/%s" % (base_dir, dirname),
frame)
print setname
ds = yt.load(setname)
for field in field_list:
print yt.ProjectionPlot(ds, 0, field).save(
'%s_%s_%04d' % (plot_sequence_name, short_name, frame))
frame = 266
name = 'f02'
#if len(sys.argv) < 3:
# print "plotter.py frame name"
# sys.exit(0)
#frame = int( sys.argv[1])
#name = sys.argv[2]
field_list = ['density', 'temperature'] #,'metallicity']
do_the_vectors = False
if len(sys.argv) >= 4:
print "her"
field_list += ['magnetic_energy']
if len(sys.argv) >= 5:
do_the_vectors = True
ds = yt.load(dsdiff_helpers.get_ds_name("%s/%s" % (basedir, dirname), frame))
for field in field_list:
print field
try:
proj = ds.proj(field, 2)
pw = proj.to_pw() #width=(0.1,'code_length'))
if do_the_vectors:
pw.annotate_streamlines('Bx', 'By')
print pw.save('%s_zoom_%04d' % (name, frame))
except:
print "Problem with Field", field
#end
#end