Wrijupan Bhattacharyya / ctools

Commit 75e6edfb8a5045f05d7fdd19ecb872434ace7dd9

Authored by Jürgen Knödlseder
1 parent 1399aa6b

Adapt benchmark to new interface

Inline Side-by-side
Showing 1 changed file with 38 additions and 21 deletions
examples/benchmark_cta_analysis.py
  View file @75e6edf
@@ -167,10 +167,10 @@ def binned_pipeline(duration): @@ -167,10 +167,10 @@ def binned_pipeline(duration):
167 return telapsed, tctlike 167 return telapsed, tctlike
168 168
169 169
170 -# ============================ #  
171 -# Cube-style analysis pipeline #  
172 -# ============================ #  
173 -def cube_pipeline(duration): 170 +# ========================= #
  171 +# Stacked analysis pipeline #
  172 +# ========================= #
  173 +def stacked_pipeline(duration):
174 """ 174 """
175 Cube-style analysis pipeline. 175 Cube-style analysis pipeline.
176 """ 176 """
@@ -260,17 +260,34 @@ def cube_pipeline(duration): @@ -260,17 +260,34 @@ def cube_pipeline(duration):
260 psfcube["yref"].real(dec) 260 psfcube["yref"].real(dec)
261 psfcube.run() 261 psfcube.run()
262 262
263 - # Set exposure and PSF cube for first CTA observation  
264 - obs = bin.obs()  
265 - cta = gammalib.GCTAObservation(obs[0])  
266 - cta.response(expcube.expcube(), psfcube.psfcube())  
267 - obs[0] = cta 263 + # Create background cube
  264 + bkgcube = ctools.ctbkgcube(sim.obs())
  265 + bkgcube["incube"].filename("NONE")
  266 + bkgcube["ebinalg"].string("LOG")
  267 + bkgcube["emin"].real(emin)
  268 + bkgcube["emax"].real(emax)
  269 + bkgcube["enumbins"].integer(enumbins)
  270 + bkgcube["nxpix"].integer(10)
  271 + bkgcube["nypix"].integer(10)
  272 + bkgcube["binsz"].real(1.0)
  273 + bkgcube["coordsys"].string(coordsys)
  274 + bkgcube["proj"].string(proj)
  275 + bkgcube["xref"].real(ra)
  276 + bkgcube["yref"].real(dec)
  277 + bkgcube.run()
  278 +
  279 + # Attach background model to observation container
  280 + bin.obs().models(bkgcube.models())
  281 +
  282 + # Set Exposure and Psf cube for first CTA observation
  283 + # (ctbin will create an observation with a single container)
  284 + bin.obs()[0].response(expcube.expcube(), psfcube.psfcube(), bkgcube.bkgcube())
268 285
269 # Get ctlike start CPU time 286 # Get ctlike start CPU time
270 tctlike = time.clock() 287 tctlike = time.clock()
271 288
272 # Perform maximum likelihood fitting 289 # Perform maximum likelihood fitting
273 - like = ctools.ctlike(obs) 290 + like = ctools.ctlike(bin.obs())
274 like.run() 291 like.run()
275 292
276 # Get stop CPU time 293 # Get stop CPU time
@@ -292,16 +309,16 @@ if __name__ == '__main__': @@ -292,16 +309,16 @@ if __name__ == '__main__':
292 print("*************************************") 309 print("*************************************")
293 print("* CTA analysis benchmark *") 310 print("* CTA analysis benchmark *")
294 print("*************************************") 311 print("*************************************")
295 - print(" Duration Unbinned Binned Cube-style ctlike-UB ctlike-B ctlike-CS") 312 + print(" Duration Unbinned Binned Stacked ctlike-UB ctlike-B ctlike-S")
296 313
297 # Initialize arrays 314 # Initialize arrays
298 a_duration = [] 315 a_duration = []
299 a_unbinned = [] 316 a_unbinned = []
300 a_binned = [] 317 a_binned = []
301 - a_cube = [] 318 + a_stacked = []
302 c_unbinned = [] 319 c_unbinned = []
303 c_binned = [] 320 c_binned = []
304 - c_cube = [] 321 + c_stacked = []
305 322
306 # Perform benchmarks 323 # Perform benchmarks
307 duration = 1800.0 324 duration = 1800.0
@@ -309,22 +326,22 @@ if __name__ == '__main__': @@ -309,22 +326,22 @@ if __name__ == '__main__':
309 326
310 # Perform analyses 327 # Perform analyses
311 t_unbinned, ct_unbinned = unbinned_pipeline(duration) 328 t_unbinned, ct_unbinned = unbinned_pipeline(duration)
312 - t_binned, ct_binned = binned_pipeline(duration)  
313 - t_cube, ct_cube = cube_pipeline(duration) 329 + t_binned, ct_binned = binned_pipeline(duration)
  330 + t_stacked, ct_stacked = stacked_pipeline(duration)
314 331
315 # Collect results 332 # Collect results
316 a_duration.append(duration/3600.0) 333 a_duration.append(duration/3600.0)
317 a_unbinned.append(t_unbinned) 334 a_unbinned.append(t_unbinned)
318 a_binned.append(t_binned) 335 a_binned.append(t_binned)
319 - a_cube.append(t_cube) 336 + a_stacked.append(t_stacked)
320 c_unbinned.append(ct_unbinned) 337 c_unbinned.append(ct_unbinned)
321 c_binned.append(ct_binned) 338 c_binned.append(ct_binned)
322 - c_cube.append(ct_cube) 339 + c_stacked.append(ct_stacked)
323 340
324 # Print results 341 # Print results
325 print("%10.0f %10.4f %10.4f %10.4f %10.4f %10.4f %10.4f" % \ 342 print("%10.0f %10.4f %10.4f %10.4f %10.4f %10.4f %10.4f" % \
326 - (duration, t_unbinned, t_binned, t_cube, \  
327 - ct_unbinned, ct_binned, ct_cube)) 343 + (duration, t_unbinned, t_binned, t_stacked, \
  344 + ct_unbinned, ct_binned, ct_stacked))
328 345
329 # Increment duration 346 # Increment duration
330 duration *= 2.0 347 duration *= 2.0
@@ -335,10 +352,10 @@ if __name__ == '__main__': @@ -335,10 +352,10 @@ if __name__ == '__main__':
335 plt.title("CTA analysis benchmark") 352 plt.title("CTA analysis benchmark")
336 plt.loglog(a_duration, a_unbinned, 'ro-', label='unbinned') 353 plt.loglog(a_duration, a_unbinned, 'ro-', label='unbinned')
337 plt.loglog(a_duration, a_binned, 'bo-', label='binned') 354 plt.loglog(a_duration, a_binned, 'bo-', label='binned')
338 - plt.loglog(a_duration, a_cube, 'go-', label='cube-style') 355 + plt.loglog(a_duration, a_stacked, 'go-', label='cube-style')
339 plt.loglog(a_duration, c_unbinned, 'ro--') 356 plt.loglog(a_duration, c_unbinned, 'ro--')
340 plt.loglog(a_duration, c_binned, 'bo--') 357 plt.loglog(a_duration, c_binned, 'bo--')
341 - plt.loglog(a_duration, c_cube, 'go--') 358 + plt.loglog(a_duration, c_stacked, 'go--')
342 plt.xlabel("Duration (hours)") 359 plt.xlabel("Duration (hours)")
343 plt.ylabel("CPU time (seconds)") 360 plt.ylabel("CPU time (seconds)")
344 plt.legend(loc="lower right") 361 plt.legend(loc="lower right")