import time import glob import argparse import os from collections import defaultdict import composition.support_functions.simfunctions as simfunctions import composition.support_functions.paths as paths from composition.support_functions.checkdir import checkdir # from ShowerLLH_scripts.analysis.zfix import zfix if __name__ == "__main__": # Setup global path names mypaths = paths.Paths() checkdir(mypaths.comp_data_dir) p = argparse.ArgumentParser( description='Runs extra modules over a given fileList') p.add_argument('-o', '--outfile', dest='outfile', help='Output file') args = p.parse_args() dataframe_dict = defaultdict(list) # Get simulation information t_sim = time.time() print('Loading simulation information...') file_list = sorted(glob.glob(mypaths.comp_data_dir + '/IT73_sim/files/sim_????.hdf5')) value_keys = ['IceTopMaxSignal',
dest='outdir', default='/home/jbourbeau/public_html/figures/composition', help='Output directory') p.add_argument('-e', '--energy', dest='energy', default='MC', choices=['MC', 'reco'], help='Option for a variety of preset bin values') p.add_argument('--extended', dest='extended', default=False, action='store_true', help='Use extended energy range') args = p.parse_args() checkdir(args.outdir + '/') # Import ShowerLLH sim reconstructions and cuts to be made df, cut_dict = load_sim(return_cut_dict=True) selection_mask = np.array([True] * len(df)) standard_cut_keys = [ 'reco_exists', 'reco_zenith', 'reco_IT_containment', 'IceTopMaxSignalInEdge', 'IceTopMaxSignal', 'NChannels', 'LF_InIce_containment' ] # standard_cut_keys = ['reco_exists', 'reco_zenith', 'reco_IT_containment', # 'IceTopMaxSignalInEdge', 'IceTopMaxSignal', 'NChannels', 'InIce_containment'] for key in standard_cut_keys: selection_mask *= cut_dict[key] df = df[selection_mask]