pipeline_unbinned_mem.py
3.53 KB
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
#! /usr/bin/env python
# ==========================================================================
# Perform unbinned in-memory analysis of simulated CTA data
#
# Copyright (C) 2015-2016 Juergen Knoedlseder
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
# ==========================================================================
import gammalib
import ctools
import cscripts
# ================================ #
# Simulation and analysis pipeline #
# ================================ #
def run_pipeline(obs, ra=83.63, dec=22.01, rad=3.0,
emin=0.1, emax=100.0,
tmin=0.0, tmax=0.0,
debug=False):
"""
Simulation and binned analysis pipeline
Parameters
----------
obs : `~gammalib.GObservations`
Observation container
ra : float, optional
Right Ascension of Region of Interest centre (deg)
dec : float, optional
Declination of Region of Interest centre (deg)
rad : float, optional
Radius of Region of Interest (deg)
emin : float, optional
Minimum energy (TeV)
emax : float, optional
Maximum energy (TeV)
tmin : float, optional
Start time (s)
tmax : float, optional
Stop time (s)
debug : bool, optional
Debug function
"""
# Simulate events
sim = ctools.ctobssim(obs)
sim['debug'] = debug
sim.run()
# Select events
select = ctools.ctselect(sim.obs())
select['ra'] = ra
select['dec'] = dec
select['rad'] = rad
select['emin'] = emin
select['emax'] = emax
select['tmin'] = tmin
select['tmax'] = tmax
select['debug'] = debug
select.run()
# Perform maximum likelihood fitting
like = ctools.ctlike(select.obs())
like['debug'] = True # Switch this always on for results in console
like.run()
# Return
return
# =============================== #
# Run unbinned in-memory pipeline #
# =============================== #
def pipeline_unbinned_mem():
"""
Run unbinned in-memory pipeline
"""
# Set usage string
usage = 'pipeline_unbinned_mem.py [-d datadir]'
# Set default options
options = [{'option': '-d', 'value': 'data'}]
# Get arguments and options from command line arguments
args, options = cscripts.ioutils.get_args_options(options, usage)
# Extract script parameters from options
datadir = options[0]['value']
# Setup observations
obs = cscripts.obsutils.set_observations(83.63, 22.01, 5.0, 0.0, 180.0,
0.1, 100.0, 'South_0.5h', 'prod2',
pattern='four', offset=1.5)
# Setup model
obs.models(gammalib.GModels(datadir+'/crab.xml'))
# Run analysis pipeline
run_pipeline(obs)
# Return
return
# ======================== #
# Main routine entry point #
# ======================== #
if __name__ == '__main__':
# Run unbinned in-memory pipeline
pipeline_unbinned_mem()