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")
