ctobssim.cpp
53.8 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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
1001
1002
1003
1004
1005
1006
1007
1008
1009
1010
1011
1012
1013
1014
1015
1016
1017
1018
1019
1020
1021
1022
1023
1024
1025
1026
1027
1028
1029
1030
1031
1032
1033
1034
1035
1036
1037
1038
1039
1040
1041
1042
1043
1044
1045
1046
1047
1048
1049
1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
1060
1061
1062
1063
1064
1065
1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
1080
1081
1082
1083
1084
1085
1086
1087
1088
1089
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101
1102
1103
1104
1105
1106
1107
1108
1109
1110
1111
1112
1113
1114
1115
1116
1117
1118
1119
1120
1121
1122
1123
1124
1125
1126
1127
1128
1129
1130
1131
1132
1133
1134
1135
1136
1137
1138
1139
1140
1141
1142
1143
1144
1145
1146
1147
1148
1149
1150
1151
1152
1153
1154
1155
1156
1157
1158
1159
1160
1161
1162
1163
1164
1165
1166
1167
1168
1169
1170
1171
1172
1173
1174
1175
1176
1177
1178
1179
1180
1181
1182
1183
1184
1185
1186
1187
1188
1189
1190
1191
1192
1193
1194
1195
1196
1197
1198
1199
1200
1201
1202
1203
1204
1205
1206
1207
1208
1209
1210
1211
1212
1213
1214
1215
1216
1217
1218
1219
1220
1221
1222
1223
1224
1225
1226
1227
1228
1229
1230
1231
1232
1233
1234
1235
1236
1237
1238
1239
1240
1241
1242
1243
1244
1245
1246
1247
1248
1249
1250
1251
1252
1253
1254
1255
1256
1257
1258
1259
1260
1261
1262
1263
1264
1265
1266
1267
1268
1269
1270
1271
1272
1273
1274
1275
1276
1277
1278
1279
1280
1281
1282
1283
1284
1285
1286
1287
1288
1289
1290
1291
1292
1293
1294
1295
1296
1297
1298
1299
1300
1301
1302
1303
1304
1305
1306
1307
1308
1309
1310
1311
1312
1313
1314
1315
1316
1317
1318
1319
1320
1321
1322
1323
1324
1325
1326
1327
1328
1329
1330
1331
1332
1333
1334
1335
1336
1337
1338
1339
1340
1341
1342
1343
1344
1345
1346
1347
1348
1349
1350
1351
1352
1353
1354
1355
1356
1357
1358
1359
1360
1361
1362
1363
1364
1365
1366
1367
1368
1369
1370
1371
1372
1373
1374
1375
1376
1377
1378
1379
1380
1381
1382
1383
1384
1385
1386
1387
1388
1389
1390
1391
1392
1393
1394
1395
1396
1397
1398
1399
1400
1401
1402
1403
1404
1405
1406
1407
1408
1409
1410
1411
1412
1413
1414
1415
1416
1417
1418
1419
1420
1421
1422
1423
1424
1425
1426
1427
1428
1429
1430
1431
1432
1433
1434
1435
1436
1437
1438
1439
1440
1441
1442
1443
1444
1445
1446
1447
1448
1449
1450
1451
1452
1453
1454
1455
1456
1457
1458
1459
1460
1461
1462
1463
1464
1465
1466
1467
1468
1469
1470
1471
1472
1473
1474
1475
1476
1477
1478
1479
1480
1481
1482
1483
1484
1485
1486
1487
1488
1489
1490
1491
1492
1493
1494
1495
1496
1497
1498
1499
1500
1501
1502
/***************************************************************************
* ctobssim - Observation simulator tool *
* ----------------------------------------------------------------------- *
* copyright (C) 2011-2016 by 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/>. *
* *
***************************************************************************/
/**
* @file ctobssim.cpp
* @brief Observation simulator tool implementation
* @author Juergen Knoedlseder
*/
/* __ Includes ___________________________________________________________ */
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <cstdio>
#include <typeinfo>
#include "ctobssim.hpp"
#include "GTools.hpp"
/* __ Method name definitions ____________________________________________ */
#define G_GET_PARAMETERS "ctobssim::get_parameters()"
#define G_SIMULATE_SOURCE "ctobssim::simulate_source(GCTAObservation*, "\
"GModels&, GRan&, GLog*)"
#define G_SIMULATE_INTERVAL "ctobssim::simulate_interval(GCTAObservation*"\
", GCTAEventList*, GCTAResponseIrf*, GModels&, GTime&"\
", GTime&, GEnergy&, GEnergy&, GEnergy&, GEnergy&"\
", GSkyDir&, double&, double&, GRan&, GLog*, int&)"
#define G_GET_AREA "ctobssim::get_area(GCTAObservation* obs, GEnergy&, "\
"GEnergy&)"
/* __ Constants __________________________________________________________ */
const double g_roi_margin = 0.5; //!< Simulation radius margin (degrees)
/* __ Debug definitions __________________________________________________ */
/* __ Coding definitions _________________________________________________ */
/*==========================================================================
= =
= Constructors/destructors =
= =
==========================================================================*/
/***********************************************************************//**
* @brief Void constructor
*
* Constructs an empty ctobssim tool.
***************************************************************************/
ctobssim::ctobssim(void) : ctobservation(CTOBSSIM_NAME, CTOBSSIM_VERSION)
{
// Initialise members
init_members();
// Return
return;
}
/***********************************************************************//**
* @brief Observations constructor
*
* @param[in] obs Observation container.
*
* Constructs ctobssim tool from an observation container.
***************************************************************************/
ctobssim::ctobssim(const GObservations& obs) :
ctobservation(CTOBSSIM_NAME, CTOBSSIM_VERSION, obs)
{
// Initialise members
init_members();
// Return
return;
}
/***********************************************************************//**
* @brief Command line constructor
*
* @param[in] argc Number of arguments in command line.
* @param[in] argv Array of command line arguments.
*
* Constructs ctobssim tool using command line arguments for user parameter
* setting.
***************************************************************************/
ctobssim::ctobssim(int argc, char *argv[]) :
ctobservation(CTOBSSIM_NAME, CTOBSSIM_VERSION, argc, argv)
{
// Initialise members
init_members();
// Return
return;
}
/***********************************************************************//**
* @brief Copy constructor
*
* @param[in] app Application.
*
* Constructs ctobssim tool from another ctobssim instance.
***************************************************************************/
ctobssim::ctobssim(const ctobssim& app) : ctobservation(app)
{
// Initialise members
init_members();
// Copy members
copy_members(app);
// Return
return;
}
/***********************************************************************//**
* @brief Destructor
*
* Destructs ctobssim tool.
***************************************************************************/
ctobssim::~ctobssim(void)
{
// Free members
free_members();
// Return
return;
}
/*==========================================================================
= =
= Operators =
= =
==========================================================================*/
/***********************************************************************//**
* @brief Assignment operator
*
* @param[in] app ctobssim tool.
* @return ctobssim tool.
*
* Assigns ctobssim tool.
***************************************************************************/
ctobssim& ctobssim::operator=(const ctobssim& app)
{
// Execute only if object is not identical
if (this != &app) {
// Copy base class members
this->ctobservation::operator=(app);
// Free members
free_members();
// Initialise members
init_members();
// Copy members
copy_members(app);
} // endif: object was not identical
// Return this object
return *this;
}
/*==========================================================================
= =
= Public methods =
= =
==========================================================================*/
/***********************************************************************//**
* @brief Clear ctobssim tool
*
* Clears ctobssim tool.
***************************************************************************/
void ctobssim::clear(void)
{
// Free members
free_members();
this->ctobservation::free_members();
this->ctool::free_members();
// Clear base class (needed to conserve tool name and version)
this->GApplication::clear();
// Initialise members
this->ctool::init_members();
this->ctobservation::init_members();
init_members();
// Write header into logger
log_header();
// Return
return;
}
/***********************************************************************//**
* @brief Run the ctobssim tool.
*
* Gets the user parameters, loops over all CTA observations in the
* observation container, and simulate events for each observation.
***************************************************************************/
void ctobssim::run(void)
{
// Switch screen logging on in debug mode
if (logDebug()) {
log.cout(true);
}
// Get parameters
get_parameters();
// Special mode: if read ahead is specified we know that we called
// the execute() method, hence files are saved immediately and event
// lists are disposed afterwards.
if (read_ahead()) {
m_save_and_dispose = true;
}
// Set energy dispersion flags of all CTA observations and save old
// values in save_edisp vector
std::vector<bool> save_edisp = set_edisp(m_obs, m_apply_edisp);
// Determine the number of valid CTA observations
int n_observations = 0;
for (int i = 0; i < m_obs.size(); ++i) {
GCTAObservation* obs = dynamic_cast<GCTAObservation*>(m_obs[i]);
if (obs != NULL) {
n_observations++;
}
}
// If more than a single observation has been handled then make sure that
// an XML file will be used for storage
if (n_observations > 1) {
m_use_xml = true;
}
// Write execution model into logger
log_header1(NORMAL, "Execution mode");
std::string mode = (m_save_and_dispose)
? "Save and dispose (reduces memory needs)"
: "Keep events in memory";
std::string xml = (m_use_xml)
? "Write Observation Definition XML file"
: "Write single event list FITS file";
log_value(NORMAL, "Event list management", mode);
log_value(NORMAL, "Output format", xml);
// Write seed values into logger
log_header1(NORMAL, "Seed values");
for (int i = 0; i < m_rans.size(); ++i) {
log_value(NORMAL, "Seed "+gammalib::str(i),
gammalib::str(m_rans[i].seed()));
}
// Write input observation container into logger
log_observations(NORMAL, m_obs, "Input observation");
// Write header
log_header1(TERSE, gammalib::number("Simulate observation", m_obs.size()));
// From here on the code can be parallelized if OpenMP support
// is enabled. The code in the following block corresponds to the
// code that will be executed in each thread
#pragma omp parallel
{
// Each thread will have it's own logger to avoid conflicts
GLog wrklog;
if (logDebug()) {
wrklog.cout(true);
}
// Allocate and initialize copies for multi-threading
GModels models(m_obs.models());
// Copy configuration from application logger to thread logger
wrklog.date(log.date());
wrklog.name(log.name());
// Set a big value to avoid flushing
wrklog.buffer_size(10000000);
// Loop over all observation in the container. If OpenMP support
// is enabled, this loop will be parallelized.
#pragma omp for
for (int i = 0; i < m_obs.size(); ++i) {
// Write header for observation
if (logTerse()) {
wrklog.header3(get_obs_header(m_obs[i]));
}
// Get pointer on CTA observation
GCTAObservation* obs = dynamic_cast<GCTAObservation*>(m_obs[i]);
// Skip observation if it's not CTA
if (obs == NULL) {
if (logTerse()) {
wrklog << " Skipping ";
wrklog << m_obs[i]->instrument();
wrklog << " observation" << std::endl;
}
continue;
}
// Skip observation if we have a binned observation
if (obs->eventtype() == "CountsCube") {
if (logTerse()) {
wrklog << " Skipping binned ";
wrklog << obs->instrument();
wrklog << " observation" << std::endl;
}
continue;
}
// Remove now all events from the event list but keep the
// event list information such as ROI, Good Time Intervals,
// energy boundaries. This will also keep additional columns
// in an event list file.
GCTAEventList* events = static_cast<GCTAEventList*>(obs->events());
events->remove(0, events->size());
// Work on a clone of the CTA observation. This makes sure that
// any memory allocated for computing (for example a response
// cache) is properly de-allocated on exit of this run
GCTAObservation obs_clone = *obs;
// Save number of events before entering simulation
int events_before = obs_clone.events()->size();
// Simulate source events
simulate_source(&obs_clone, models, m_rans[i], &wrklog);
// Simulate source events
simulate_background(&obs_clone, models, m_rans[i], &wrklog);
// Dump simulation results
if (logTerse()) {
wrklog << gammalib::parformat("MC events");
wrklog << obs_clone.events()->size() - events_before;
wrklog << " (all models)";
wrklog << std::endl;
}
// Append the event list to the original observation
obs->events(*(obs_clone.events()));
// If requested, event lists are saved immediately
if (m_save_and_dispose) {
// Set event output file name. If multiple observations are
// handled, build the filename from prefix and observation
// index. Otherwise use the outfile parameter.
std::string outfile;
if (m_use_xml) {
m_prefix = (*this)["prefix"].string();
outfile = m_prefix + gammalib::str(i) + ".fits";
}
else {
outfile = (*this)["outevents"].filename();
}
// Store output file name in original observation
obs->eventfile(outfile);
// Save observation into FITS file. This is a critical zone
// to avoid multiple threads writing simultaneously
#pragma omp critical(ctobssim_run)
{
//obs_clone.save(outfile, clobber());
obs->save(outfile, clobber());
}
// Dispose events
obs->dispose_events();
} // endif: save and dispose requested
} // endfor: looped over observations
// At the end, the content of the thread logger is added to
// the application logger
#pragma omp critical (log)
{
log << wrklog;
}
} // end pragma omp parallel
// Restore energy dispersion flags of all CTA observations
restore_edisp(m_obs, save_edisp);
// Return
return;
}
/***********************************************************************//**
* @brief Save the selected event list(s)
*
* This method saves the selected event list(s) into FITS file(s). There are
* two modes, depending on the m_use_xml flag.
*
* If m_use_xml is true, all selected event list(s) will be saved into FITS
* files, where the output filenames are constructued from the input
* filenames by prepending the m_prefix string to name. Any path information
* will be stripped form the input name, hence event files will be written
* into the local working directory (unless some path information is present
* in the prefix). In addition, an XML file will be created that gathers
* the filename information for the selected event list(s). If an XML file
* was present on input, all metadata information will be copied from this
* input file.
*
* If m_use_xml is false, the selected event list will be saved into a FITS
* file.
***************************************************************************/
void ctobssim::save(void)
{
// Write header into logger
log_header1(TERSE, gammalib::number("Save observation", m_obs.size()));
// Case A: Save event file(s) and XML metadata information
if (m_use_xml) {
save_xml();
}
// Case B: Save event file as FITS file
else {
save_fits();
}
// Return
return;
}
/*==========================================================================
= =
= Private methods =
= =
==========================================================================*/
/***********************************************************************//**
* @brief Initialise class members
*
* The thrown area is fixed to pi*(2500^2) m2, which is the same value that
* is used in the Monte Carlo simulations (information from Konrad Bernloehr)
***************************************************************************/
void ctobssim::init_members(void)
{
// Initialise user parameters
m_outevents.clear();
m_prefix.clear();
m_seed = 1;
m_eslices = 10;
m_apply_edisp = false;
m_max_rate = 1.0e6;
// Initialise protected members
m_rans.clear();
m_save_and_dispose = false;
// Set fixed parameters
m_max_photons = 1000000; //!< Maximum number of photons / time slice
// Initialise first event identifier
m_event_id = 1;
// Return
return;
}
/***********************************************************************//**
* @brief Copy class members
*
* @param[in] app Application.
***************************************************************************/
void ctobssim::copy_members(const ctobssim& app)
{
// Copy user parameters
m_outevents = app.m_outevents;
m_prefix = app.m_prefix;
m_seed = app.m_seed;
m_eslices = app.m_eslices;
m_apply_edisp = app.m_apply_edisp;
m_max_rate = app.m_max_rate;
// Copy protected members
m_max_photons = app.m_max_photons;
m_rans = app.m_rans;
m_save_and_dispose = app.m_save_and_dispose;
m_event_id = app.m_event_id;
// Return
return;
}
/***********************************************************************//**
* @brief Delete class members
***************************************************************************/
void ctobssim::free_members(void)
{
// Return
return;
}
/***********************************************************************//**
* @brief Get application parameters
*
* Get all task parameters from parameter file or (if required) by querying
* the user. Most parameters are only required if no observation exists so
* far in the observation container. In this case, a single CTA observation
* will be added to the container, using the definition provided in the
* parameter file.
***************************************************************************/
void ctobssim::get_parameters(void)
{
// Initialise seed vector
m_rans.clear();
// If there are no observations in container then load them via user
// parameters
if (m_obs.size() == 0) {
// Throw exception if counts cube is given
require_inobs_nocube(G_GET_PARAMETERS);
// Get observation container
m_obs = get_observations();
}
// ... otherwise add response information and energy boundaries in case
// that they are missing
else {
setup_observations(m_obs);
}
// Read model definition file if required
if (m_obs.models().size() == 0) {
std::string inmodel = (*this)["inmodel"].filename();
m_obs.models(inmodel);
}
// Get other parameters
m_seed = (*this)["seed"].integer();
m_eslices = (*this)["eslices"].integer();
m_apply_edisp = (*this)["edisp"].boolean();
m_max_rate = (*this)["maxrate"].real();
// Optionally read ahead parameters so that they get correctly
// dumped into the log file
if (read_ahead()) {
m_outevents = (*this)["outevents"].filename();
m_prefix = (*this)["prefix"].string();
}
// Initialise random number generators. We initialise here one random
// number generator per observation so that each observation will
// get it's own random number generator. This will lead to identical
// results independently of code parallelization with OpenMP. The
// seeds for all random number generators are derived randomly but
// fully deterministacally from the seed parameter, so that a given
// seed parameter leads always to the same set of simulated events, and
// this independently of parallelization.
// Get a random number generator for seed determination
GRan master(m_seed);
// Allocate vector of random number generator seeds
std::vector<unsigned long long int> seeds;
// Loop over all observations in the container
for (int i = 0; i < m_obs.size(); ++i) {
// Allocate new seed value
unsigned long long int new_seed;
// Determine new seed value. We make sure that the new seed
// value has not been used already for another observation.
bool repeat = false;
do {
new_seed = (unsigned long long int)(master.uniform() * 1.0e10) +
m_seed;
repeat = false;
for (int j = 0; j < seeds.size(); ++j) {
if (new_seed == seeds[j]) {
repeat = true;
break;
}
}
} while(repeat);
// Add the seed to the vector for bookkeeping
seeds.push_back(new_seed);
// Use the seed to create a random number generator for the
// actual observation
m_rans.push_back(GRan(new_seed));
} // endfor: looped over observations
// Write parameters into logger
log_parameters(TERSE);
// Return
return;
}
/***********************************************************************//**
* @brief Simulate source events from photon list
*
* @param[in] obs Pointer on CTA observation.
* @param[in] models Model list.
* @param[in,out] ran Random number generator.
* @param[in] wrklog Pointer to logger.
*
* Simulate source events from a photon list for a given CTA observation and
* all source models. The events are stored in form of an event list in the
* observation.
*
* The method will loop over all Good Time Intervals (GTI) to perform the
* simulations. Within a given GTI the requested energy range is split into
* a number of energy slices so that the simulation area can be adapted to
* the effective area of the instrument (this avoids simulating a large
* number of sources photons at low energies while accepting only few of
* them do to the small effective area). In case that the flux of a source
* is large, the simulation may be done within time slices so that the
* memory requirements won't get too large.
*
* This method does nothing if the observation pointer is NULL. It verifies
* if the observation has a CTA IRF response.
***************************************************************************/
void ctobssim::simulate_source(GCTAObservation* obs,
const GModels& models,
GRan& ran,
GLog* wrklog)
{
// Continue only if observation pointer is valid
if (obs != NULL) {
// If no logger is specified then use the default logger
if (wrklog == NULL) {
wrklog = &log;
}
// Get pointer on event list
GCTAEventList* events = static_cast<GCTAEventList*>(obs->events());
// Get CTA response
const GCTAResponseIrf* rsp =
dynamic_cast<const GCTAResponseIrf*>(obs->response());
if (rsp == NULL) {
std::string cls = std::string(typeid(obs->response()).name());
std::string msg = "Response of type \""+cls+"\" is not a CTA "
"IRF response. Please make sure that a CTA "
"IRF response is contained in the CTA "
"observation.";
throw GException::invalid_value(G_SIMULATE_SOURCE, msg);
}
// Extract simulation region from event list ROI
GSkyDir dir = events->roi().centre().dir();
double rad = events->roi().radius() + g_roi_margin;
// Determine energy boundaries for simulation
GEbounds ebounds = get_ebounds(events->ebounds());
// Log simulation cone information
if (logTerse()) {
*wrklog << gammalib::parformat("Simulation cone");
*wrklog << "RA=" << dir.ra_deg() << " deg";
*wrklog << ", Dec=" << dir.dec_deg() << " deg";
*wrklog << ", radius=" << rad << " deg" << std::endl;
}
// Initialise indentation for logging
int indent = 0;
// Initialise photon and event counters
std::vector<int> nphotons(models.size(),0);
std::vector<int> nevents(models.size(),0);
// Loop over all Good Time Intervals
for (int it = 0; it < events->gti().size(); ++it) {
// Extract time interval
GTime tmin = events->gti().tstart(it);
GTime tmax = events->gti().tstop(it);
// Log time interval. Increment indentation if there are
// several Good Time Intervals.
if (logNormal()) {
if (events->gti().size() > 1) {
indent++;
wrklog->indent(indent);
}
*wrklog << gammalib::parformat("Time interval", indent);
*wrklog << tmin.convert(m_cta_ref);
*wrklog << " - ";
*wrklog << tmax.convert(m_cta_ref);
*wrklog << " s" << std::endl;
}
// Loop over all energy boundaries
for (int ie = 0; ie < ebounds.size(); ++ie) {
// Set reconstructed energy interval
GEnergy ereco_min = ebounds.emin(ie);
GEnergy ereco_max = ebounds.emax(ie);
// Set true photon energy limits for simulation. If the
// observation has energy dispersion then add a margin.
GEnergy etrue_min = ereco_min;
GEnergy etrue_max = ereco_max;
if (rsp->use_edisp()) {
etrue_min = rsp->ebounds(etrue_min).emin();
etrue_max = rsp->ebounds(etrue_max).emax();
}
// Determine simulation area
double area = get_area(obs, ereco_min, ereco_max);
// Log energy range and simulation area
if (logNormal()) {
*wrklog << gammalib::parformat("Photon energy range", indent);
*wrklog << etrue_min << " - " << etrue_max << std::endl;
*wrklog << gammalib::parformat("Event energy range", indent);
*wrklog << ereco_min << " - " << ereco_max << std::endl;
}
// Increment indentation if there are several energy
// boundaries
if (logNormal()) {
if (ebounds.size() > 1) {
indent++;
wrklog->indent(indent);
}
}
// Log simulation area
if (logNormal()) {
*wrklog << gammalib::parformat("Simulation area", indent);
*wrklog << area << " cm2" << std::endl;
}
// Save state of event counter before doing the simulation
int nevents_before = events->size();
// Simulate events for this time and energy interval
simulate_interval(obs, rsp, events, models, tmin, tmax,
etrue_min, etrue_max, ereco_min, ereco_max,
dir, rad, area,
ran, wrklog, indent, nphotons, nevents);
// Log simulation results
if (logNormal()) {
*wrklog << gammalib::parformat("MC source events", indent);
*wrklog << events->size() - nevents_before;
*wrklog << " (all source models)";
*wrklog << std::endl;
}
// Reset indentation if there were several energy boundaries
if (logNormal()) {
if (ebounds.size() > 1) {
indent--;
wrklog->indent(indent);
}
}
} // endfor: looped over all energy boundaries
// Reset indentation if there were several Good Time Intervals
if (logNormal()) {
if (events->gti().size() > 1) {
indent--;
wrklog->indent(indent);
}
}
} // endfor: looped over all time intervals
// Reset indentation
wrklog->indent(0);
// Log simulation summary
if (logTerse()) {
for (int i = 0; i < models.size(); ++i) {
const GModelSky* model = dynamic_cast<const GModelSky*>(models[i]);
if (model == NULL) {
continue;
}
if (!model->is_valid(obs->instrument(), obs->id())) {
continue;
}
*wrklog << gammalib::parformat("MC source photons");
*wrklog << nphotons[i];
if (model->name().length() > 0) {
*wrklog << " [" << model->name() << "]";
}
*wrklog << std::endl;
*wrklog << gammalib::parformat("MC source events");
*wrklog << nevents[i];
if (model->name().length() > 0) {
*wrklog << " [" << model->name() << "]";
}
*wrklog << std::endl;
}
}
} // endif: observation pointer was valid
// Return
return;
}
/***********************************************************************//**
* @brief Simulate source events for a time and energy interval
*
* @param[in] obs Pointer on CTA observation.
* @param[in] rsp Pointer on CTA IRF response.
* @param[in,out] events Pointer on CTA event list.
* @param[in] models Model list.
* @param[in] tmin Start time.
* @param[in] tmax Stop time.
* @param[in] etrue_min Minimum true energy.
* @param[in] etrue_max Maximum true energy.
* @param[in] ereco_min Minimum reconstructed energy.
* @param[in] ereco_max Maximum reconstructed energy.
* @param[in] dir Simulation cone centre.
* @param[in] rad Simulation cone radius (degrees).
* @param[in] area Simulation area (cm^2).
* @param[in,out] ran Random number generator.
* @param[in] wrklog Pointer to logger.
* @param[in,out] indent Logger indent.
* @param[in,out] nphotons Number of photons for all models.
* @param[in,out] nevents Number of events for all models.
*
* Simulate source events for a time and energy interval.
***************************************************************************/
void ctobssim::simulate_interval(GCTAObservation* obs,
const GCTAResponseIrf* rsp,
GCTAEventList* events,
const GModels& models,
const GTime& tmin,
const GTime& tmax,
const GEnergy& etrue_min,
const GEnergy& etrue_max,
const GEnergy& ereco_min,
const GEnergy& ereco_max,
const GSkyDir& dir,
const double& rad,
const double& area,
GRan& ran,
GLog* wrklog,
int& indent,
std::vector<int>& nphotons,
std::vector<int>& nevents)
{
// Loop over all models
for (int i = 0; i < models.size(); ++i) {
// Get sky model (NULL if not a sky model)
const GModelSky* model = dynamic_cast<const GModelSky*>(models[i]);
// If the model is not a sky model then skip the model
if (model == NULL) {
continue;
}
// If the model does not apply to the instrument and observation
// identifier then skip the model
if (!model->is_valid(obs->instrument(), obs->id())) {
continue;
}
// Determine duration of a time slice by limiting the number of
// simulated photons to m_max_photons. The photon rate is estimated
// from the model flux and used to set the duration of the time
// slice.
double flux = get_model_flux(model, etrue_min, etrue_max, dir, rad,
indent, wrklog);
double rate = flux * area;
double duration = 1800.0; // default: 1800 sec
if (rate > 0.0) {
duration = m_max_photons / rate;
if (duration < 1.0) { // not <1 sec
duration = 1.0;
}
else if (duration > 180000.0) { // not >50 hr
duration = 180000.0;
}
}
// Skip model if photon rate is 0
if (rate <= 0.0) {
continue;
}
// Log photon rate
if (logNormal()) {
*wrklog << gammalib::parformat("Photon rate", indent);
*wrklog << rate << " photons/s";
if (model->name().length() > 0) {
*wrklog << " [" << model->name() << "]";
}
*wrklog << std::endl;
}
// If photon rate exceeds the maximum photon rate that is allowed
// then throw an exception
if (rate > m_max_rate) {
std::string mod = (model->name().length() > 0) ?
model->name() : "Unknown";
std::string msg = "Photon rate "+gammalib::str(rate)+
" photons/s for model \""+mod+"\" exceeds "
"maximum allowed photon rate of "+
gammalib::str(m_max_rate)+" photons/s. "
"Please check the parameters of model "
"\""+mod+"\" or increase the value of the "
"\"maxrate\" parameter.";
throw GException::invalid_value(G_SIMULATE_INTERVAL, msg);
}
// To reduce memory requirements we split long time intervals into
// several time slices
GTime tstart = tmin;
GTime tstop = tstart + duration;
// Save state of photon and event counters before doing the
// simulation
int nphotons_before = nphotons[i];
int nevents_before = nevents[i];
// Loop over time slices
while (tstart < tmax) {
// Make sure that tstop <= tmax
if (tstop > tmax) {
tstop = tmax;
}
// Log time slice
if (logExplicit()) {
if (tmax - tmin > duration) {
indent++;
wrklog->indent(indent);
}
*wrklog << gammalib::parformat("Time slice", indent);
*wrklog << tstart.convert(m_cta_ref) << " - ";
*wrklog << tstop.convert(m_cta_ref) << " s";
if (model->name().length() > 0) {
*wrklog << " [" << model->name() << "]";
}
*wrklog << std::endl;
}
// Simulate time slice
simulate_time_slice(obs, rsp, events, model, tstart, tstop,
etrue_min, etrue_max, ereco_min, ereco_max,
dir, rad, area,
ran, wrklog, indent,
nphotons[i], nevents[i]);
// Go to next time slice
tstart = tstop;
tstop = tstart + duration;
// Reset indentation
if (logExplicit()) {
if (tmax - tmin > duration) {
indent--;
wrklog->indent(indent);
}
}
} // endwhile: looped over time slices
// Log simulation results
if (logNormal()) {
*wrklog << gammalib::parformat("MC source photons", indent);
*wrklog << nphotons[i] - nphotons_before;
if (model->name().length() > 0) {
*wrklog << " [" << model->name() << "]";
}
*wrklog << std::endl;
*wrklog << gammalib::parformat("MC source events", indent);
*wrklog << nevents[i] - nevents_before;
if (model->name().length() > 0) {
*wrklog << " [" << model->name() << "]";
}
*wrklog << std::endl;
}
} // endfor: looped over models
// Return
return;
}
/***********************************************************************//**
* @brief Simulate source events for a time slice
*
* @param[in] obs Pointer on CTA observation.
* @param[in] rsp Pointer on CTA response.
* @param[in,out] events Pointer on CTA event list.
* @param[in] model Model.
* @param[in] tstart Start time.
* @param[in] tstop Stop time.
* @param[in] etrue_min Minimum true energy.
* @param[in] etrue_max Maximum true energy.
* @param[in] ereco_min Minimum reconstructed energy.
* @param[in] ereco_max Maximum reconstructed energy.
* @param[in] dir Simulation cone centre.
* @param[in] rad Simulation cone radius (degrees).
* @param[in] area Simulation area (cm^2).
* @param[in,out] ran Random number generator.
* @param[in] wrklog Pointer to logger.
* @param[in,out] indent Logger indent.
* @param[in,out] nphotons Number of photons.
* @param[in,out] nevents Number of events.
*
* Simulate source events for a time slice.
***************************************************************************/
void ctobssim::simulate_time_slice(GCTAObservation* obs,
const GCTAResponseIrf* rsp,
GCTAEventList* events,
const GModelSky* model,
const GTime& tstart,
const GTime& tstop,
const GEnergy& etrue_min,
const GEnergy& etrue_max,
const GEnergy& ereco_min,
const GEnergy& ereco_max,
const GSkyDir& dir,
const double& rad,
const double& area,
GRan& ran,
GLog* wrklog,
int& indent,
int& nphotons,
int& nevents)
{
// Get photons
GPhotons photons = model->mc(area, dir, rad, etrue_min, etrue_max,
tstart, tstop, ran);
// Dump number of simulated photons
if (logExplicit()) {
*wrklog << gammalib::parformat("MC source photons/slice", indent);
*wrklog << photons.size();
if (model->name().length() > 0) {
*wrklog << " [" << model->name() << "]";
}
*wrklog << std::endl;
}
// Simulate events from photons
for (int i = 0; i < photons.size(); ++i) {
// Increment photon counter
nphotons++;
// Simulate event. Note that this method includes the deadtime
// correction.
GCTAEventAtom* event = rsp->mc(area, photons[i], *obs, ran);
// Use event only if it exists and if it falls within ROI, the
// reconstructed energy interval and the time slice
if (event != NULL) {
if (events->roi().contains(*event) &&
event->energy() >= ereco_min &&
event->energy() <= ereco_max &&
event->time() >= tstart &&
event->time() <= tstop) {
event->event_id(m_event_id);
events->append(*event);
m_event_id++;
nevents++;
}
delete event;
}
} // endfor: looped over events
// Return
return;
}
/***********************************************************************//**
* @brief Get energy boundaries
*
* @param[in] ebounds Energy boundaries of events.
* @return Energy boundaries for simulation.
*
* Get the energy boundaries for the source simulation.
*
* @todo Need to take care of the (rare) case that there are distinct
* energy boundaries in the event list. In that case we need to
* create mixed energy boundaries.
***************************************************************************/
GEbounds ctobssim::get_ebounds(const GEbounds& ebounds) const
{
// Set energy bins
GEbounds ebins(m_eslices, ebounds.emin(), ebounds.emax());
// Return energy bins
return (ebins);
}
/***********************************************************************//**
* @brief Get simulation area (cm^2)
*
* @param[in] obs Pointer on CTA observation.
* @param[in] emin Minimum energy.
* @param[in] emax Maximum energy.
* @return Simulation area (cm^2).
*
* Get the simulation area for an energy interval in units of cm^2.
***************************************************************************/
double ctobssim::get_area(GCTAObservation* obs,
const GEnergy& emin,
const GEnergy& emax) const
{
// Get CTA response
const GCTAResponseIrf* rsp =
dynamic_cast<const GCTAResponseIrf*>(obs->response());
if (rsp == NULL) {
std::string cls = std::string(typeid(rsp).name());
std::string msg = "Response of type \""+cls+"\" is not a CTA IRF "
"response. Please make sure that a CTA IRF "
"response is contained in the CTA observation.";
throw GException::invalid_value(G_GET_AREA, msg);
}
// Compute effective area at minimum and maximum energy
double area_emin = rsp->aeff()->max(emin.log10TeV(), 0.0, 0.0);
double area_emax = rsp->aeff()->max(emax.log10TeV(), 0.0, 0.0);
// Use maximum of both
double area = (area_emin > area_emax) ? area_emin : area_emax;
// Multiply by security factor
area *= 2.0;
// Return simulation area
return area;
}
/***********************************************************************//**
* @brief Determine sky model flux
*
* @param[in] model Sky model.
* @param[in] emin Minimum energy.
* @param[in] emax Maximum energy.
* @param[in] centre Centre of region for photon rate determination.
* @param[in] radius Radius of region for photon rate determination (degrees).
* @param[in] indent Indent for logging.
* @param[in,out] wrklog Pointer to logger.
* @return Model flux (photons/cm2/sec).
***************************************************************************/
double ctobssim::get_model_flux(const GModelSky* model,
const GEnergy& emin,
const GEnergy& emax,
const GSkyDir& centre,
const double& radius,
const int& indent,
GLog* wrklog)
{
// Initialise flux
double flux = 0.0;
// Determine the spatial model normalization within the simulation
// cone and check whether the model will produce any photons in that
// cone.
double norm = model->spatial()->mc_norm(centre, radius);
bool use_model = (norm > 0.0) ? true : false;
if (logNormal()) {
if (use_model) {
*wrklog << gammalib::parformat("Use model", indent);
}
else {
*wrklog << gammalib::parformat("Skip model", indent);
}
if (model->name().length() > 0) {
*wrklog << model->name();
}
*wrklog << std::endl;
*wrklog << gammalib::parformat("Normalization", indent);
*wrklog << norm;
if (model->name().length() > 0) {
*wrklog << " [" << model->name() << "]";
}
*wrklog << std::endl;
}
// Continue only if model overlaps with simulation region
if (use_model) {
// Initialise de-allocation flag
bool free_spectral = false;
// Get pointer to spectral model
const GModelSpectral* spectral = model->spectral();
// If the spatial model is a diffuse cube then create a node
// function spectral model that is the product of the diffuse
// cube node function and the spectral model evaluated at the
// energies of the node function
GModelSpatialDiffuseCube* cube = dynamic_cast<GModelSpatialDiffuseCube*>(model->spatial());
if (cube != NULL) {
// Set MC cone
cube->set_mc_cone(centre, radius);
// Allocate node function to replace the spectral component
GModelSpectralNodes* nodes = new GModelSpectralNodes(cube->spectrum());
for (int i = 0; i < nodes->nodes(); ++i) {
GEnergy energy = nodes->energy(i);
GTime time; // Dummy time
double intensity = nodes->intensity(i);
double value = spectral->eval(energy, time);
nodes->intensity(i, value * intensity);
}
// Signal that node function needs to be de-allocated later
free_spectral = true;
// Set the spectral model pointer to the node function
spectral = nodes;
} // endif: spatial model was a diffuse cube
// Compute flux within [emin, emax] in model from spectral
// component (units: ph/cm2/s)
double flux0 = spectral->flux(emin, emax);
flux = flux0 * norm;
// Dump flux
if (logNormal()) {
*wrklog << gammalib::parformat("Flux", indent);
*wrklog << flux0;
if (model->name().length() > 0) {
*wrklog << " [" << model->name() << "]";
}
*wrklog << " photons/cm2/s" << std::endl;
*wrklog << gammalib::parformat("Normalized flux", indent);
*wrklog << flux;
if (model->name().length() > 0) {
*wrklog << " [" << model->name() << "]";
}
*wrklog << " photons/cm2/s" << std::endl;
}
// Free spectral model if required
if (free_spectral) delete spectral;
} // endif: model overlaps with simulation region
// Return model flux
return flux;
}
/***********************************************************************//**
* @brief Simulate background events from model
*
* @param[in] obs Pointer on CTA observation.
* @param[in] models Models.
* @param[in] ran Random number generator.
* @param[in] wrklog Pointer to logger.
*
* Simulate background events from models. The events are stored as event
* list in the observation.
*
* This method does nothing if the observation pointer is NULL.
***************************************************************************/
void ctobssim::simulate_background(GCTAObservation* obs,
const GModels& models,
GRan& ran,
GLog* wrklog)
{
// Continue only if observation pointer is valid
if (obs != NULL) {
// If no logger is specified then use the default logger
if (wrklog == NULL) {
wrklog = &log;
}
// Get pointer on event list (circumvent const correctness)
GCTAEventList* events =
static_cast<GCTAEventList*>(const_cast<GEvents*>(obs->events()));
// Loop over all models
for (int i = 0; i < models.size(); ++i) {
// Get data model (NULL if not a data model)
const GModelData* model =
dynamic_cast<const GModelData*>(models[i]);
// If we have a data model that applies to the present observation
// then simulate events
if (model != NULL &&
model->is_valid(obs->instrument(), obs->id())) {
// Get simulated CTA event list. Note that this method
// includes the deadtime correction.
GCTAEventList* list =
dynamic_cast<GCTAEventList*>(model->mc(*obs, ran));
// Continue only if we got a CTA event list
if (list != NULL) {
// Reserves space for events
events->reserve(list->size()+events->size());
// Initialise statistics
int n_appended = 0;
int n_outside_roi = 0;
// Append events
for (int k = 0; k < list->size(); k++) {
// Get event pointer
GCTAEventAtom* event = (*list)[k];
// Use event only if it falls within ROI
if (events->roi().contains(*event)) {
// Set event identifier
event->event_id(m_event_id);
m_event_id++;
// Append event
events->append(*event);
// Increment number of appended events
n_appended++;
} // endif: event was within ROI
// ... otherwise increment outside ROI counter
else {
n_outside_roi++;
}
} // endfor: looped over all events
// Dump simulation results
if (logNormal()) {
*wrklog << gammalib::parformat("MC events outside ROI");
*wrklog << n_outside_roi << std::endl;
*wrklog << gammalib::parformat("MC background events");
*wrklog << n_appended << std::endl;
}
// Free event list
delete list;
} // endif: we had a CTA event list
} // endif: model was valid
} // endfor: looped over all models
} // endif: observation pointer was valid
// Return
return;
}
/***********************************************************************//**
* @brief Save event list in FITS format.
*
* Save the event list as a FITS file. The filename of the FITS file is
* specified by the outfile parameter.
***************************************************************************/
void ctobssim::save_fits(void)
{
// Save only if event list has not yet been saved and disposed and if
// there are observations
if (!m_save_and_dispose && m_obs.size() > 0) {
// Get output filename
m_outevents = (*this)["outevents"].filename();
// Get CTA observation from observation container
GCTAObservation* obs = dynamic_cast<GCTAObservation*>(m_obs[0]);
// Log filename
log_value(NORMAL, "Event list file", m_outevents);
// Save observation into FITS file
obs->save(m_outevents, clobber());
} // endif: event list has not yet been saved and disposed
// Return
return;
}
/***********************************************************************//**
* @brief Save event list(s) in XML format.
*
* Save the event list(s) into FITS files and write the file path information
* into a XML file. The filename of the XML file is specified by the outfile
* parameter, the filename(s) of the event lists are built by prepending a
* prefix to the input event list filenames. Any path present in the input
* filename will be stripped, i.e. the event list(s) will be written in the
* local working directory (unless a path is specified in the prefix).
***************************************************************************/
void ctobssim::save_xml(void)
{
// Get output filename and prefix
m_outevents = (*this)["outevents"].filename();
m_prefix = (*this)["prefix"].string();
// Issue warning if output filename has no .xml suffix
log_string(TERSE, warn_xml_suffix(m_outevents));
// Save only if event lists have not yet been saved and disposed
if (!m_save_and_dispose) {
// Loop over all observation in the container
for (int i = 0; i < m_obs.size(); ++i) {
// Get CTA observation
GCTAObservation* obs = dynamic_cast<GCTAObservation*>(m_obs[i]);
// Handle only CTA observations
if (obs != NULL) {
// Continue only if there is an event list (it may have been disposed)
if (obs->events()->size() != 0) {
// Set event output file name
std::string outfile = m_prefix + gammalib::str(i) +
".fits";
// Store output file name in observation
obs->eventfile(outfile);
// Log filename
log_value(NORMAL, "Event list file", outfile);
// Save observation into FITS file
obs->save(outfile, clobber());
}
} // endif: observation was a CTA observations
} // endfor: looped over observations
} // endif: event list has not yet been saved and disposed
// Save observations in XML file
m_obs.save(m_outevents);
// Return
return;
}