def start(self): output_parameters = {'MuonEff': [], 'ImpactP': [], 'RingWidth': []} numev = 0 num_muons_found = 0 for event in hessio_event_source(self.infile): self.log.info("Event Number: %d, found %d muons", numev, num_muons_found) self.calib.calibrate(event) muon_evt = analyze_muon_event(event) numev += 1 if not muon_evt['MuonIntensityParams']: # No telescopes contained a good muon continue else: if self.display: plot_muon_event(event, muon_evt) for tid in muon_evt['TelIds']: idx = muon_evt['TelIds'].index(tid) if not muon_evt['MuonIntensityParams'][idx]: continue self.log.info("** Muon params: %s", muon_evt[idx]) output_parameters['MuonEff'].append( muon_evt['MuonIntensityParams'][idx].optical_efficiency_muon ) output_parameters['ImpactP'].append( muon_evt['MuonIntensityParams'][idx].impact_parameter.value ) output_parameters['RingWidth'].append( muon_evt['MuonIntensityParams'][idx].ring_width.value ) print_muon(muon_evt, printer=self.log.info) num_muons_found += 1 t = Table(output_parameters) t['ImpactP'].unit = 'm' t['RingWidth'].unit = 'deg' if self.outfile: t.write(self.outfile)
def test_basic_muon_reco(example_event): """ Really simplistic test: just run the analyze_muon_event code, to make sure it doesn't crash. The input event is so far not a muon, so no output is generated. Parameters ---------- test_event - a sample event (fixture) """ calib = CameraCalibrator(subarray=example_event.inst.subarray) calib(example_event) muon_params = muon.analyze_muon_event(example_event) assert muon_params is not None
def test_basic_muon_reco(example_event): """ Really simplistic test: just run the analyze_muon_event code, to make sure it doesn't crash. The input event is so far not a muon, so no output is generated. Parameters ---------- test_event - a sample event (fixture) """ calib = CameraCalibrator() calib.calibrate(example_event) muon_params = muon.analyze_muon_event(example_event) assert muon_params is not None
def start(self): numev = 0 self.num_muons_found = defaultdict(int) for event in tqdm(self.source, desc='detecting muons'): self.calib.calibrate(event) muon_evt = analyze_muon_event(event) if numev == 0: _exclude_some_columns(event.inst.subarray, self.writer) numev += 1 if not muon_evt['MuonIntensityParams']: # No telescopes contained a good muon continue else: if self.display: plot_muon_event(event, muon_evt) for tel_id in muon_evt['TelIds']: idx = muon_evt['TelIds'].index(tel_id) intens_params = muon_evt['MuonIntensityParams'][idx] if intens_params is not None: ring_params = muon_evt['MuonRingParams'][idx] cam_id = str(event.inst.subarray.tel[tel_id].camera) self.num_muons_found[cam_id] += 1 self.log.debug("INTENSITY: %s", intens_params) self.log.debug("RING: %s", ring_params) self.writer.write(table_name=cam_id, containers=[intens_params, ring_params]) self.log.info( "Event Number: %d, found %s muons", numev, dict(self.num_muons_found) )
def start(self): numev = 0 self.num_muons_found = defaultdict(int) for event in tqdm(self.source, desc='detecting muons'): self.calib.calibrate(event) muon_evt = analyze_muon_event(event) if numev == 0: _exclude_some_columns(event.inst.subarray, self.writer) numev += 1 if not muon_evt['MuonIntensityParams']: # No telescopes contained a good muon continue else: if self.display: plot_muon_event(event, muon_evt) for tel_id in muon_evt['TelIds']: idx = muon_evt['TelIds'].index(tel_id) intens_params = muon_evt['MuonIntensityParams'][idx] if intens_params is not None: ring_params = muon_evt['MuonRingParams'][idx] cam_id = str(event.inst.subarray.tel[tel_id].camera) self.num_muons_found[cam_id] += 1 self.log.debug("INTENSITY: %s", intens_params) self.log.debug("RING: %s", ring_params) self.writer.write(table_name=cam_id, containers=[intens_params, ring_params]) self.log.info( "Event Number: %d, found %s muons", numev, dict(self.num_muons_found) )
def main(): script = os.path.splitext(os.path.basename(__file__))[0] log.info("[SCRIPT] {}".format(script)) parser = argparse.ArgumentParser( description='Display each event in the file', formatter_class=argparse.ArgumentDefaultsHelpFormatter) parser.add_argument('-f', '--file', dest='input_path', action='store', default=get_dataset('gamma_test.simtel.gz'), help='path to the input file') parser.add_argument('-O', '--origin', dest='origin', action='store', #was .origin_list() default='hessio', help='origin of the file') parser.add_argument('-D', dest='display', action='store_true', default=False, help='display the camera events') parser.add_argument('-t', '--telescope', dest='tel', action='store', type=int, default=None, help='telecope to view. Default = All') parser.add_argument('--pdf', dest='output_path', action='store', default=None, help='path to store a pdf output of the plots') parser.add_argument('--calib-help', dest='calib_help', action='store_true', default=False, help='display the arguments used for the camera ' 'calibration') args, excess_args = parser.parse_known_args() #params, unknown_args = calibration_parameters(excess_args, # args.origin, # args.calib_help) r1 = HessioR1Calibrator(None, None) dl0 = CameraDL0Reducer(None, None) dl1 = CameraDL1Calibrator(None, None) log.debug("[file] Reading file") #input_file = InputFile(args.input_path, args.origin) #source = input_file.read() source = hessio_event_source(args.input_path) geom_dict = {} # plot_dict = {} muoneff = [] impactp = [] ringwidth = [] plot_dict = {'MuonEff':muoneff,'ImpactP':impactp,'RingWidth':ringwidth} numev = 0 for event in source: print("Event Number",numev) r1.calibrate(event) dl0.reduce(event) dl1.calibrate(event) muon_evt = analyze_muon_event(event) numev += 1 #Test display #Flag 1 for true (wish to display) #plot_muon_event(event,muon_evt) # display_telescope(muon_evt, muon_evt[0].tel_id, 1, geom_dict, pp, fig) if muon_evt[0] is not None and muon_evt[1] is not None: plot_muon_event(event,muon_evt,None,args) plot_dict['MuonEff'].append(muon_evt[1].optical_efficiency_muon) plot_dict['ImpactP'].append(muon_evt[1].impact_parameter.value) plot_dict['RingWidth'].append(muon_evt[1].ring_width.value) display_muon_plot(muon_evt) #Store and or Plot muon parameters here #if numev > 50: #for testing purposes - kill early # break t = Table([muoneff, impactp, ringwidth], names=('MuonEff','ImpactP','RingWidth'), meta={'name': 'muon analysis results'}) t['ImpactP'].unit = 'm' t['RingWidth'].unit = 'deg' # print('plotdict',plot_dict) t.write(str(args.output_path)+'_muontable.fits',overwrite=True) #NEED this to overwrite plot_muon_efficiency(args.output_path) log.info("[COMPLETE]")
def main(): script = os.path.splitext(os.path.basename(__file__))[0] log.info("[SCRIPT] {}".format(script)) parser = argparse.ArgumentParser( description='Display each event in the file', formatter_class=argparse.ArgumentDefaultsHelpFormatter) parser.add_argument('-f', '--file', dest='input_path', action='store', default=get_dataset('gamma_test.simtel.gz'), help='path to the input file') parser.add_argument( '-O', '--origin', dest='origin', action='store', # was .origin_list() default='hessio', help='origin of the file') parser.add_argument('-D', dest='display', action='store_true', default=False, help='display the camera events') parser.add_argument('-t', '--telescope', dest='tel_FIXME', action='store', type=int, default=None, help='telecope to view. Default = All') parser.add_argument('--pdf', dest='output_path', action='store', default=None, help='path to store a pdf output of the plots') parser.add_argument('--calib-help', dest='calib_help', action='store_true', default=False, help='display the arguments used for the camera ' 'calibration') args, excess_args = parser.parse_known_args() # params, unknown_args = calibration_parameters(excess_args, # args.origin, # args.calib_help) r1 = HessioR1Calibrator(None, None) dl0 = CameraDL0Reducer(None, None) dl1 = CameraDL1Calibrator(None, None) log.debug("[file] Reading file") # input_file = InputFile(args.input_path, args.origin) # source = input_file.read() source = hessio_event_source(args.input_path) muoneff = [] impactp = [] ringwidth = [] plot_dict = { 'MuonEff': muoneff, 'ImpactP': impactp, 'RingWidth': ringwidth } numev = 0 for event in source: print("Event Number", numev) r1.calibrate(event) dl0.reduce(event) dl1.calibrate(event) muon_evt = analyze_muon_event(event) numev += 1 # Test display #Flag 1 for true (wish to display) # plot_muon_event(event,muon_evt) # display_telescope(muon_evt, muon_evt[0].tel_id, 1, geom_dict, pp, fig) #if muon_evt[0] is not None and muon_evt[1] is not None: if not muon_evt[ 'MuonIntensityParams']: #No telescopes contained a good muon continue else: plot_muon_event(event, muon_evt, None, args) for tid in muon_evt['TelIds']: idx = muon_evt['TelIds'].index(tid) if not muon_evt['MuonIntensityParams'][idx]: continue plot_dict['MuonEff'].append(muon_evt['MuonIntensityParams'] [idx].optical_efficiency_muon) plot_dict['ImpactP'].append(muon_evt['MuonIntensityParams'] [idx].impact_parameter.value) plot_dict['RingWidth'].append( muon_evt['MuonIntensityParams'][idx].ring_width.value) display_muon_plot(muon_evt) # Store and or Plot muon parameters here # if numev > 50: #for testing purposes - kill early # break t = Table([muoneff, impactp, ringwidth], names=('MuonEff', 'ImpactP', 'RingWidth'), meta={'name': 'muon analysis results'}) t['ImpactP'].unit = 'm' t['RingWidth'].unit = 'deg' # print('plotdict',plot_dict) #t.write(str(args.output_path) + '_muontable.fits', overwrite=True) # NEED # this to overwrite #plot_muon_efficiency(args.output_path) log.info("[COMPLETE]")
def start(self): output_parameters = { 'MuonEff': [], 'ImpactP': [], 'RingWidth': [], 'RingCont': [], 'RingComp': [], 'RingPixComp': [], 'Core_x': [], 'Core_y': [], 'Impact_x_arr': [], 'Impact_y_arr': [], 'MCImpactP': [], 'ImpactDiff': [], 'RingSize': [], 'RingRadius': [], 'NTels': [] } numev = 0 num_muons_found = 0 for event in event_source(self.infile): self.log.info("Event Number: %d, found %d muons", numev, num_muons_found) self.calib.calibrate(event) muon_evt = analyze_muon_event(event) numev += 1 if not muon_evt[ 'MuonIntensityParams']: # No telescopes contained a good muon continue else: if self.display: plot_muon_event(event, muon_evt) ntels = len(event.r0.tels_with_data) #if(len( event.r0.tels_with_data) <= 1): #continue #print("event.r0.tels_with_data", event.r0.tels_with_data) for tid in muon_evt['TelIds']: idx = muon_evt['TelIds'].index(tid) if muon_evt['MuonIntensityParams'][idx] is not None: print("pos, tid", event.inst.subarray.positions[tid], tid) tel_x = event.inst.subarray.positions[tid][0] tel_y = event.inst.subarray.positions[tid][1] core_x = event.mc.core_x # MC Core x core_y = event.mc.core_y # MC Core y rec_impact_x = muon_evt['MuonIntensityParams'][ idx].impact_parameter_pos_x rec_impact_y = muon_evt['MuonIntensityParams'][ idx].impact_parameter_pos_y print("rec_impact_x, rec_impact_y", rec_impact_x, rec_impact_y) print("event.mc.core_x, event.mc.core_y", event.mc.core_x, event.mc.core_y) impact_mc = np.sqrt( np.power(core_x - tel_x, 2) + np.power(core_y - tel_y, 2)) print("simulated impact", impact_mc) # Coordinate transformation to move the impact point to array coordinates rec_impact_x_arr = tel_x + rec_impact_x rec_impact_y_arr = tel_y + rec_impact_y print("impact_x_arr, impact_y_arr", rec_impact_x_arr, rec_impact_y_arr) # Distance between core of the showe and impact parameter impact_diff = np.sqrt( np.power(core_x - rec_impact_x_arr, 2) + np.power(core_y - rec_impact_y_arr, 2)) print("impact_diff ", impact_diff) self.log.info("** Muon params: %s", muon_evt['MuonIntensityParams'][idx]) output_parameters['MuonEff'].append( muon_evt['MuonIntensityParams'] [idx].optical_efficiency_muon) output_parameters['ImpactP'].append( muon_evt['MuonIntensityParams'] [idx].impact_parameter.value) output_parameters['RingWidth'].append( muon_evt['MuonIntensityParams'] [idx].ring_width.value) output_parameters['RingCont'].append( muon_evt['MuonRingParams'][idx].ring_containment) output_parameters['RingComp'].append( muon_evt['MuonIntensityParams'] [idx].ring_completeness) output_parameters['RingPixComp'].append( muon_evt['MuonIntensityParams'] [idx].ring_pix_completeness) output_parameters['Core_x'].append( event.mc.core_x.value) output_parameters['Core_y'].append( event.mc.core_y.value) output_parameters['Impact_x_arr'].append( rec_impact_x_arr.value) output_parameters['Impact_y_arr'].append( rec_impact_y_arr.value) output_parameters['MCImpactP'].append(impact_mc.value) output_parameters['ImpactDiff'].append( impact_diff.value) output_parameters['RingSize'].append( muon_evt['MuonIntensityParams'][idx].ring_size) output_parameters['RingRadius'].append( muon_evt['MuonRingParams'][idx].ring_radius.value) output_parameters['NTels'].append(ntels) print_muon(muon_evt, printer=self.log.info) num_muons_found += 1 t = Table(output_parameters) t['ImpactP'].unit = 'm' t['RingWidth'].unit = 'deg' #t['MCImpactP'].unit = 'm' if self.outfile: t.write(self.outfile)