testtraining/simpleRun.py at master · pgiri4/testtraining · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
from __future__ import absolute_import, division, print_function
import argparse
import NuRadioReco.modules.trigger.highLowThreshold
import NuRadioReco.modules.channelBandPassFilter
import NuRadioReco.modules.RNO_G.hardwareResponseIncorporator
from NuRadioReco.utilities import units
from NuRadioMC.simulation import simulation

highLowThreshold = NuRadioReco.modules.trigger.highLowThreshold.triggerSimulator()
channelBandPassFilter = NuRadioReco.modules.channelBandPassFilter.channelBandPassFilter()
hardware_response = NuRadioReco.modules.RNO_G.hardwareResponseIncorporator.hardwareResponseIncorporator()


class mySimulation(simulation.simulation):

    def _detector_simulation_filter_amp(self, evt, station, det):
        hardware_response.run(evt, station, det, sim_to_data=True)
        channelBandPassFilter.run(evt, station, det, passband=[80 * units.MHz, 1000 * units.GHz],filter_type='butter', order=2)
        channelBandPassFilter.run(evt, station, det, passband=[0, 500 * units.MHz],filter_type='butter', order=10)
    def _detector_simulation_trigger(self, evt, station, det):
        highLowThreshold.run(
            evt,
            station,
            det,
            threshold_high=20. * units.mV,
            threshold_low=-20. * units.mV,
            triggered_channels=[0],
            number_concidences=1,
            trigger_name='main_trigger'
        )


parser = argparse.ArgumentParser(description='Run RNO-G simulation')
parser.add_argument('input_filename', type=str)
parser.add_argument('--detector_file', type=str, default='RNO_single_station.json')
parser.add_argument('--config', type=str, default='RNO_config.yaml')
parser.add_argument('--output_hdf', type=str, default='output.hdf5')
parser.add_argument('--output_nur', type=str, default='output.nur')

args = parser.parse_args()

if __name__ == "__main__":
    sim = mySimulation(
        inputfilename=args.input_filename,
        outputfilename=args.output_hdf,
        detectorfile=args.detector_file,
        outputfilenameNuRadioReco=args.output_nur,
        config_file=args.config,
        file_overwrite=True,
        write_detector=False,
        default_detector_channel=0,
        default_detector_station=11
    )
    sim.run()