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)