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