load_bins(lc_dic, lc_loc, step, 'z', dtype='f4') load_bins(lc_dic, lc_loc, step, 'vx', dtype='f4') load_bins(lc_dic, lc_loc, step, 'vy', dtype='f4') load_bins(lc_dic, lc_loc, step, 'vz', dtype='f4') add_step(lc_dic, step) df = pd.DataFrame(lc_dic) dfs.append(df) lc_df = pd.concat(dfs, ignore_index=True) print lc_df.head() print lc_df.shape print lc_df.describe() exit() gal_cat = dtk.Catalog(gltcs_loc) gal_cat.add_steps(gltcs_steps, in_file_steps=gltcs_internal) gal_cat.add_var_step_replace("${step_var_name}", gltcs_steps, gltcs_str_z) gal_cat.add_var("/Outputs/Output${step}/nodeData/nodeIndex", as_name='nodeIndex') gal_cat.add_var( "/Outputs/Output${step}/nodeData/spheroidLuminositiesStellar:SDSS_u:rest:z${step_var_name}", as_name='lum_sdss_u_spheriod') gal_cat.add_var( "/Outputs/Output${step}/nodeData/spheroidLuminositiesStellar:SDSS_g:rest:z${step_var_name}", as_name='lum_sdss_g_spheriod') gal_cat.add_var( "/Outputs/Output${step}/nodeData/spheroidLuminositiesStellar:SDSS_r:rest:z${step_var_name}", as_name='lum_sdss_r_spheriod') gal_cat.add_var(
import matplotlib.colors as clrs import sys import dtk param = dtk.Param(sys.argv[1]) fof_loc = param.get_string("fof_loc") sod_loc = param.get_string("sod_loc") core_loc = param.get_string("core_loc") gal_loc = param.get_string("gal_loc") hp_loc = param.get_string("hp_loc") mt_loc = param.get_string("mt_loc") steps = param.get_int_list("steps") core_intacted_radius = param.get_float("core_intact_radius") fof_cat = dtk.Catalog(fof_loc) sod_cat = dtk.Catalog(sod_loc) core_cat = dtk.Catalog(core_loc) hp_cat = dtk.Catalog(hp_loc) gal_cat = dtk.Catalog(gal_loc) mt_cat = dtk.Catalog(mt_loc) gal_mt_cat = dtk.Catalog() step = steps[0] fof_cat.add_steps(steps) sod_cat.add_steps(steps) core_cat.add_steps(steps) hp_cat.add_steps(steps) gal_cat.add_steps(steps) mt_cat.add_steps(steps)
keys = infiles[0][str(step)].keys() for key in keys: print "key: ",key data = [] for i in range(0,nums): print "file num:",i data.append(infiles[i][step][key].value) data_tot = np.concatenate(data) var_path = "/"+str(step)+"/"+key print var_path outfile[var_path]=data_tot outfile.flush() for i in range(0,nums): infiles[i].flush() pos_cat = dtk.Catalog(gio_loc) pos_cat.add_steps(gltcs_steps) var_names = ['x','y','z','vx','vy','vz', 'nodeIndex','infall_halo_mass','infall_halo_tag', 'hostIndex','host_halo_mass','host_halo_tag', 'host_halo_x','host_halo_y','host_halo_z', 'positionType'] var_names2 = ['x','y','z','vx','vy','vz', 'nodeIndex','infallHaloMass','infallHaloTag', 'hostIndex','hostHaloMass','hostHaloTag', 'hostHaloX','hostHaloY','hostHaloZ', 'positionType'] for i in range(0,len(var_names)): pos_cat.add_var(var_names[i],as_name=var_names2[i]) print "reading positions...",
param = dtk.Param(sys.argv[1]) mt_loc = param.get_string("mt_loc") sod_loc = param.get_string("sod_loc") step = param.get_int('step') output = param.get_string("output").replace("${step}",str(step)) subfile_num = param.get_int('subfile_num') NFW_gamma = param.get_float("NFW_gamma") NFW_maxR200 = param.get_float("NFW_maxR200") set_gamma(NFW_gamma) stepz = dtk.StepZ(200,0,500) mt_cat1 = dtk.Catalog(mt_loc) mt_cat1.add_step(step) for i in range(0,subfile_num): mt_cat1.add_subfile(i) mt_cat1.add_var("/forestHalos/timestep",as_name='time_step') mt_cat1.add_var("/forestHalos/nodeIndex",as_name='nodeIndex') mt_cat1.add_var("/forestHalos/nodeMass",as_name='halo_mass') mt_cat1.add_var("/forestHalos/haloTag",as_name='halo_tag') mt_cat1.add_var("/forestHalos/redshift",as_name='redshift') mt_cat1.add_var("/forestHalos/timestep",as_name='timestep') mt_cat1.add_var("/forestHalos/descendentIndex",as_name='descn_indx') mt_cat1.add_var("/forestHalos/position",as_name='x',index=0); mt_cat1.add_var("/forestHalos/position",as_name='y',index=1); mt_cat1.add_var("/forestHalos/position",as_name='z',index=2); mt_cat1.add_var("/forestHalos/velocity",as_name='vx',index=0); mt_cat1.add_var("/forestHalos/velocity",as_name='vy',index=1);
import numpy as np import matplotlib.pyplot as plt import matplotlib.colors as clrs import h5py import sys import dtk param = dtk.Param(sys.argv[1]) mt_loc = param.get_string("mt_loc") fof_loc = param.get_string("fof_loc") step = param.get_int('step') output = param.get_string("output").replace("${step}", str(step)) subfile_num = param.get_int('subfile_num') mt_cat1 = dtk.Catalog(mt_loc) mt_cat1.add_step(step) for i in range(0, subfile_num): mt_cat1.add_subfile(i) mt_cat1.add_var("/forestHalos/timestep", as_name='time_step') mt_cat1.add_var("/forestHalos/nodeIndex", as_name='nodeIndex') mt_cat1.add_var("/forestHalos/nodeMass", as_name='halo_mass') mt_cat1.add_var("/forestHalos/haloTag", as_name='halo_tag') mt_cat1.add_var("/forestHalos/redshift", as_name='redshift') mt_cat1.add_var("/forestHalos/timestep", as_name='timestep') mt_cat1.add_var("/forestHalos/descendentIndex", as_name='descn_indx') mt_cat1.add_var("/forestHalos/position", as_name='x', index=0) mt_cat1.add_var("/forestHalos/position", as_name='y', index=1) mt_cat1.add_var("/forestHalos/position", as_name='z', index=2) mt_cat1.add_var("/forestHalos/velocity", as_name='vz', index=0) mt_cat1.add_var("/forestHalos/velocity", as_name='vy', index=1)
import matplotlib.pyplot as plt import matplotlib.colors as clrs import sys import h5py import dtk param = dtk.Param(sys.argv[1]) fof_loc = param.get_string("fof_loc") sod_loc = param.get_string("sod_loc") bhp_loc = param.get_string("bhp_loc") mt_loc = param.get_string("mt_loc") step = param.get_int("step") output = param.get_string("output").replace("${step}", str(step)) fof_cat = dtk.Catalog(fof_loc) sod_cat = dtk.Catalog(sod_loc) bhp_cat = dtk.Catalog(bhp_loc) mt_cat = dtk.Catalog(mt_loc) print step fof_cat.add_step(step) sod_cat.add_step(step) bhp_cat.add_step(step) mt_cat.add_step(step) fof_cat.add_var("fof_halo_tag") fof_cat.add_var("fof_halo_mass") fof_cat.add_var("fof_halo_center_x") fof_cat.add_var("fof_halo_center_y")
import matplotlib.pyplot as plt import matplotlib.colors as clrs import h5py import sys import dtk param = dtk.Param(sys.argv[1]) mt_loc = param.get_string("mt_loc") sod_loc = param.get_string("sod_loc") subfile_num = param.get_int('subfile_num') step = param.get_int("step") output = param.get_string("output").replace("${step}", str(step)) mt_cat = dtk.Catalog(mt_loc) mt_cat.add_step(step) for i in range(0, subfile_num): mt_cat.add_subfile(i) mt_cat.add_var("/forestHalos/timestep", as_name='time_step') mt_cat.add_var("/forestHalos/haloTag", as_name='htag') mt_cat.read_hdf5() mt_cat.apply_function('htag', dtk.frag_to_real) slct = mt_cat[step]["time_step"] == step mt_cat[step]["time_step"] = mt_cat[step]["time_step"][slct] mt_cat[step]["htag"] = mt_cat[step]["htag"][slct] sod_cat = dtk.Catalog(sod_loc) sod_cat.add_step(step) sod_cat.add_var("fof_halo_tag", as_name='htag') sod_cat.add_var("sod_halo_mass")
core_loc = param.get_string("core_loc") gal_loc = param.get_string("gal_loc") bhp_loc = param.get_string("bhp_loc") mt_loc = param.get_string("mt_loc") hp_loc = param.get_string("hp_loc") step = param.get_int("step") galaticus_step = param.get_int("galaticus_step") galaticus_z = param.get_string("galaticus_z") subfile_num = param.get_int("subfile_num") subfile_list = param.get_string_list('subfile_list') output = param.get_string("output").replace("${step}", str(step)) rnd_halo_prtcl = param.get_bool("rnd_halo_prtcl") NFW_profile = param.get_bool("NFW_profile") NFW_gamma = param.get_float("NFW_gamma") NFW_maxR200 = param.get_float("NFW_maxR200") fof_cat = dtk.Catalog(fof_loc) fof_cat.add_step(step) sod_cat = dtk.Catalog(sod_loc) sod_cat.add_step(step) gal_cat = dtk.Catalog(gal_loc) gal_cat.add_step(step, in_file_step=galaticus_step) core_cat = dtk.Catalog(core_loc) core_cat.add_step(step) mt_cat = dtk.Catalog(mt_loc) mt_cat.add_step(step) for i in range(0, subfile_num):
import dtk import sys plt.rc('text', usetex=True) plt.rc('font', family='serif') h=0.702 param = dtk.Param(sys.argv[1]) sod_loc = param.get_string("sod_loc") core_loc = param.get_string("core_loc") bhp_loc = param.get_string("bhp_loc") sbh_loc = param.get_string("sbh_loc") steps = param.get_int_list("steps") step = steps[0] sod_cat = dtk.Catalog(sod_loc) core_cat= dtk.Catalog(core_loc) bhp_cat = dtk.Catalog(bhp_loc) sbh_cat = dtk.Catalog(sbh_loc) sod_cat.add_steps(steps) core_cat.add_steps(steps) sbh_cat.add_steps(steps) bhp_cat.add_steps(steps) sod_cat.add_var("fof_halo_tag")