def do_grids(
mem=60000,
grpid_list=None,
c0_256=False,
c0_512_z0=False,
c0_512_z03=False,
c0_512_z1=False,
c0_512_z2=False,
c0_512_z25=False,
c0_512_z3=False,
c0_512_z4=False,
c2_z0=False,
c2_z03=False,
c2_z1=False,
c2_z2=False,
c2_z25=False,
c2_z3=False,
c2_z4=False,
c3_256=False,
c3_512_z0=False,
c3_512_z1=False,
c3_512_z2=False,
c3_512_z25=False,
c3_512_z3=False,
c3_512_z4=False,
c4_z03=False,
c4_z05=False,
c4_z1=False,
c4_z2=False,
c4_z25=False,
c4_z3=False,
c4_z4=False,
):
[alias, name, res_list, snapbase_list, suffix_list, snap_num_list, BH_dat_list] = list_prep(
c0_256=c0_256,
c0_512_z0=c0_512_z0,
c0_512_z03=c0_512_z03,
c0_512_z1=c0_512_z1,
c0_512_z2=c0_512_z2,
c0_512_z25=c0_512_z25,
c0_512_z3=c0_512_z3,
c0_512_z4=c0_512_z4,
c2_z0=c2_z0,
c2_z03=c2_z03,
c2_z1=c2_z1,
c2_z2=c2_z2,
c2_z25=c2_z25,
c2_z3=c2_z3,
c2_z4=c2_z4,
c3_256=c3_256,
c3_512_z0=c3_512_z0,
c3_512_z1=c3_512_z1,
c3_512_z2=c3_512_z2,
c3_512_z25=c3_512_z25,
c3_512_z3=c3_512_z3,
c3_512_z4=c3_512_z4,
c4_z03=c4_z03,
c4_z05=c4_z05,
c4_z1=c4_z1,
c4_z2=c4_z2,
c4_z25=c4_z25,
c4_z3=c4_z3,
c4_z4=c4_z4,
)
grid_radius_pkpc = 200
# elem = ["H","Mg","C","O","O"]
# ion = np.array([1,2,4,6,7])
for i in np.arange(len(res_list)):
snap_dir = '"{}"'.format(snapbase_list[i] + suffix_list[i])
snap_num = snap_num_list[i]
CGM_extract_file = '"{}"'.format(
savebase + alias[i] + "_r{}_s{}_CGM_extract.hdf5".format(res_list[i], snap_num_list[i])
)
BH_dat = BH_dat_list[i]
atten_type = "ion_out_fancy_atten" # ion_out_no_atten" #ion_out
# save_file = "\"{}\"".format(savebase + alias[i] +"_r{}_s{}_grids.hdf5".format(res_list[i],snap_num_list[i]))
# command_string = "python -c \'import grid_mass; grid_mass.Halo_Grid({},{},[\"H\",\"Mg\",\"C\",\"O\",\"O\"],[1,2,4,6,7],{},grid_radius_pkpc={},atten_type=\"{}\")\' "\
# .format(CGM_extract_file,res_list[i],save_file,grid_radius_pkpc,atten_type)
save_file = '"{}"'.format(
savebase + alias[i] + "_r{}_s{}_grids_newatten.hdf5".format(res_list[i], snap_num_list[i])
)
command_string = 'python -c \'import grid_mass; grid_mass.Halo_Grid({},{},["Mg","Si","O"],[2,3,6],{},grid_radius_pkpc={},atten_type="{}")\' '.format(
CGM_extract_file, res_list[i], save_file, grid_radius_pkpc, atten_type
)
# save_file = "\"{}\"".format(savebase + alias[i] +"_r{}_s{}_quasargrids.hdf5".format(res_list[i],snap_num_list[i]))
# command_string = "python -c \'import grid_mass; grid_mass.Halo_Grid({},{},[\"H\"],[1],{},grid_radius_pkpc={},atten_type=\"{}\",grpid_list=[0,4,8,15,17])\' "\
# .format(CGM_extract_file,res_list[i],save_file,grid_radius_pkpc,atten_type)
print "Submitting SLURM job with the following command:\n{} \n".format(command_string)
job_name = alias[i] + "_r{}_s{}_grids".format(res_list[i], snap_num_list[i])
job_submit.submit_serial_job(command_string, job_name=job_name, mem=mem, time_limit="3-0")
def do_CGM_extract(
c0_256=False,
c0_512_z0=False,
c0_512_z03=False,
c0_512_z1=False,
c0_512_z2=False,
c0_512_z25=False,
c0_512_z3=False,
c0_512_z4=False,
c2_z0=False,
c2_z03=False,
c2_z1=False,
c2_z2=False,
c2_z25=False,
c2_z3=False,
c2_z4=False,
c3_256=False,
c3_512_z0=False,
c3_512_z1=False,
c3_512_z2=False,
c3_512_z25=False,
c3_512_z3=False,
c3_512_z4=False,
c4_z03=False,
c4_z05=False,
c4_z1=False,
c4_z2=False,
c4_z25=False,
c4_z3=False,
c4_z4=False,
):
""" Submits jobs for enrichment profile calculations at z=2 for each feedback. """
[alias, name, res_list, snapbase_list, suffix_list, snap_num_list, BH_dat_list] = list_prep(
c0_256=c0_256,
c0_512_z0=c0_512_z0,
c0_512_z03=c0_512_z03,
c0_512_z1=c0_512_z1,
c0_512_z2=c0_512_z2,
c0_512_z25=c0_512_z25,
c0_512_z3=c0_512_z3,
c0_512_z4=c0_512_z4,
c2_z0=c2_z0,
c2_z03=c2_z03,
c2_z1=c2_z1,
c2_z2=c2_z2,
c2_z25=c2_z25,
c2_z3=c2_z3,
c2_z4=c2_z4,
c3_256=c3_256,
c3_512_z0=c3_512_z0,
c3_512_z1=c3_512_z1,
c3_512_z2=c3_512_z2,
c3_512_z25=c3_512_z25,
c3_512_z3=c3_512_z3,
c3_512_z4=c3_512_z4,
c4_z03=c4_z03,
c4_z05=c4_z05,
c4_z1=c4_z1,
c4_z2=c4_z2,
c4_z25=c4_z25,
c4_z3=c4_z3,
c4_z4=c4_z4,
)
rmin_pkpc = 200
Rvir_min = 2
for i in np.arange(len(res_list)):
snap_dir = '"{}"'.format(snapbase_list[i] + suffix_list[i])
snap_num = snap_num_list[i]
save_file = '"{}"'.format(
savebase + alias[i] + "_r{}_s{}_CGM_extract.hdf5".format(res_list[i], snap_num_list[i])
)
BH_dat = BH_dat_list[i]
command_string = "python -c 'import CGM_extract; CGM_extract.CGM_extract({},{},{},rmin_pkpc={},Rvir_min={},group_min_mass={},BH_dat={})' ".format(
snap_dir, snap_num, save_file, rmin_pkpc, Rvir_min, group_min_mass, BH_dat
)
print "Submitting SLURM job with the following command:\n{} \n".format(command_string)
job_name = alias[i] + "_r{}_s{}_CGM_extract".format(res_list[i], snap_num_list[i])
job_submit.submit_serial_job(command_string, job_name=job_name, mem=60000)
