source: trunk/MagicSoft/Mars/mjobs/MJStar.cc@ 7196

Last change on this file since 7196 was 7196, checked in by tbretz, 19 years ago
*** empty log message ***
File size: 13.6 KB
Line 
1/* ======================================================================== *\
2!
3! *
4! * This file is part of MARS, the MAGIC Analysis and Reconstruction
5! * Software. It is distributed to you in the hope that it can be a useful
6! * and timesaving tool in analysing Data of imaging Cerenkov telescopes.
7! * It is distributed WITHOUT ANY WARRANTY.
8! *
9! * Permission to use, copy, modify and distribute this software and its
10! * documentation for any purpose is hereby granted without fee,
11! * provided that the above copyright notice appear in all copies and
12! * that both that copyright notice and this permission notice appear
13! * in supporting documentation. It is provided "as is" without express
14! * or implied warranty.
15! *
16!
17!
18! Author(s): Thomas Bretz, 1/2004 <mailto:tbretz@astro.uni-wuerzburg.de>
19!
20! Copyright: MAGIC Software Development, 2000-2004
21!
22!
23\* ======================================================================== */
24
25/////////////////////////////////////////////////////////////////////////////
26//
27// MJStar
28//
29// Resource file entries are case sensitive!
30//
31/////////////////////////////////////////////////////////////////////////////
32#include "MJStar.h"
33
34#include <TEnv.h>
35#include <TFile.h>
36
37#include "MLog.h"
38#include "MLogManip.h"
39
40#include "MDirIter.h"
41#include "MParList.h"
42#include "MTaskList.h"
43#include "MEvtLoop.h"
44
45#include "MStatusDisplay.h"
46
47#include "MH3.h"
48#include "MHVsTime.h"
49#include "MHCamEvent.h"
50#include "MHCamEventRot.h"
51#include "MBinning.h"
52
53#include "MReadReports.h"
54#include "MReadMarsFile.h"
55#include "MF.h"
56#include "MFDataMember.h"
57#include "MFDeltaT.h"
58#include "MFSoftwareTrigger.h"
59#include "MContinue.h"
60#include "MGeomApply.h"
61#include "MEventRateCalc.h"
62#include "MImgCleanStd.h"
63#include "MSrcPosCalc.h"
64#include "MSrcPosCorrect.h"
65#include "MHillasCalc.h"
66#include "MMuonSearchParCalc.h"
67#include "MMuonCalibParCalc.h"
68#include "MFillH.h"
69#include "MWriteRootFile.h"
70
71#include "MMuonSetup.h"
72#include "MObservatory.h"
73#include "MPointingPosCalc.h"
74
75ClassImp(MJStar);
76
77using namespace std;
78
79// --------------------------------------------------------------------------
80//
81// Default constructor.
82//
83// Sets fRuns to 0, fExtractor to NULL, fDataCheck to kFALSE
84//
85MJStar::MJStar(const char *name, const char *title) : fMuonAnalysis(kTRUE)
86{
87 fName = name ? name : "MJStar";
88 fTitle = title ? title : "Standard analysis and reconstruction";
89}
90
91Bool_t MJStar::CheckEnvLocal()
92{
93 DisableMuonAnalysis(!GetEnv("MuonAnalysis", fMuonAnalysis));
94 return kTRUE;
95}
96
97Bool_t MJStar::WriteResult()
98{
99 if (fPathOut.IsNull())
100 {
101 *fLog << inf << "No output path specified via SetPathOut - no output written." << endl;
102 return kTRUE;
103 }
104
105 const TString oname = Form("%s/star%08d.root", (const char*)fPathOut, fSequence.GetSequence());
106
107 *fLog << inf << "Writing to file: " << oname << endl;
108
109 TFile file(oname, "RECREATE");
110
111 *fLog << inf << " - MStatusDisplay..." << flush;
112 if (fDisplay && fDisplay->Write()<=0)
113 {
114 *fLog << err << "Unable to write MStatusDisplay to " << oname << endl;
115 return kFALSE;
116 }
117 *fLog << inf << "ok." << endl;
118
119 return kTRUE;
120}
121
122Bool_t MJStar::Process(Bool_t ismc)
123{
124 if (!fSequence.IsValid())
125 {
126 *fLog << err << "ERROR - Sequence invalid!" << endl;
127 return kFALSE;
128 }
129
130 //if (!CheckEnv())
131 // return kFALSE;
132
133 CheckEnv();
134
135 // --------------------------------------------------------------------------------
136
137 *fLog << inf;
138 fLog->Separator(GetDescriptor());
139 *fLog << "Calculate image parameters of sequence ";
140 *fLog << fSequence.GetName() << endl;
141 *fLog << endl;
142
143 // --------------------------------------------------------------------------------
144
145 MDirIter iter;
146 const Int_t n0 = fSequence.SetupDatRuns(iter, MSequence::kCalibrated, fPathData);
147 const Int_t n1 = fSequence.GetNumDatRuns();
148 if (n0==0)
149 {
150 *fLog << err << "ERROR - No input files of sequence found!" << endl;
151 return kFALSE;
152 }
153 if (n0!=n1)
154 {
155 *fLog << err << "ERROR - Number of files found (" << n0 << ") doesn't match number of files in sequence (" << n1 << ")" << endl;
156 if (fLog->GetDebugLevel()>4)
157 {
158 *fLog << dbg << "Files which are searched:" << endl;
159 iter.Print();
160 }
161 return kFALSE;
162 }
163
164 // Setup Parlist
165 MParList plist;
166 plist.AddToList(this); // take care of fDisplay!
167
168 MObservatory obs;
169 plist.AddToList(&obs);
170
171 MMuonSetup muonsetup;
172 plist.AddToList(&muonsetup);
173
174 // Setup binnings for muon analysis
175 MBinning bins1("BinningRadius");
176 MBinning bins2("BinningArcWidth");
177 MBinning bins3("BinningRingBroadening");
178 MBinning bins4("BinningSizeVsArcRadius");
179 MBinning bins5("BinningMuonWidth");
180 MBinning bins6("BinningArcPhi");
181 plist.AddToList(&bins1);
182 plist.AddToList(&bins2);
183 plist.AddToList(&bins3);
184 plist.AddToList(&bins4);
185 plist.AddToList(&bins5);
186 plist.AddToList(&bins6);
187
188
189 // Setup Tasklist
190 MTaskList tlist;
191 plist.AddToList(&tlist);
192
193 MReadReports readreal;
194 readreal.AddTree("Events", "MTime.", MReadReports::kMaster);
195 readreal.AddTree("Drive", MReadReports::kRequired);
196 readreal.AddTree("Starguider", MReadReports::kRequired);
197 readreal.AddTree("CC");
198 readreal.AddFiles(iter);
199
200 MReadMarsFile readmc("Events");
201 readmc.DisableAutoScheme();
202 readmc.AddFiles(iter);
203
204 // ------------------ Setup general tasks ----------------
205
206 MFDeltaT fdeltat;
207 MContinue cont(&fdeltat, "FilterDeltaT", "Filter events with wrong timing");
208 cont.SetInverted();
209
210 MGeomApply apply; // Only necessary to craete geometry
211 MEventRateCalc rate;
212 rate.SetNumEvents(1200);
213
214 MFSoftwareTrigger swtrig;
215 MContinue contsw(&swtrig, "FilterSwTrigger", "Software trigger");
216 contsw.SetInverted();
217
218 MImgCleanStd clean;
219 clean.SetNamePedPhotCam("MPedPhotFromExtractorRndm");
220
221 //MSrcPosCalc poscalc;
222 //MSrcPosCorrect poscorrect;
223 MHillasCalc hcalc;
224 hcalc.Disable(MHillasCalc::kCalcConc);
225
226 // ------------------ Setup histograms and fill tasks ----------------
227 MHCamEvent evt0a(0, "Cleaned", "Signal after Cleaning;;S [\\gamma]");
228 MHCamEvent evt0b(0, "UsedPix", "Fraction of Events in which Pixels are used;;Fraction");
229 evt0b.SetThreshold(0);
230
231 MFillH fillvs("MHRate", "MTime", "FillEventRate");
232 MFillH fillp1("MHPointing", "MTimeDrive", "FillDrive");
233 MFillH fillp2("MHPointing", "MTimeStarguider", "FillStarguider");
234 fillp1.SetBit(MFillH::kDoNotDisplay);
235 //fillp2.SetNameTab("Drive");
236
237 MFillH fill0a(&evt0a, "MSignalCam", "FillSignalCam");
238 MFillH fill0b(&evt0b, "MSignalCam", "FillCntUsedPixels");
239 MFillH fill1("MHHillas", "MHillas", "FillHillas");
240 MFillH fill2("MHHillasExt", "", "FillHillasExt");
241 MFillH fill3("MHHillasSrc", "MHillasSrc", "FillHillasSrc");
242 MFillH fill4("MHImagePar", "MImagePar", "FillImagePar");
243 MFillH fill5("MHNewImagePar", "MNewImagePar", "FillNewImagePar");
244 MFillH fill9("MHEffectiveOnTime", "MTime", "FillEffOnTime");
245
246 //fillvs.SetNameTab("Rate");
247 fill9.SetNameTab("EffOnTime");
248
249 // ------------------ Setup write task ----------------
250
251 // Effective on-time need its own not to be skipped by (eg) image cleaning
252 // Muons needs its own to have a unique SetReadyToSave
253 const TString rule(Form("%s{s/_Y_/_I_}", fPathOut.Data()));
254 MWriteRootFile write( 2, rule, fOverwrite?"RECREATE":"NEW");
255 MWriteRootFile writet(2, rule, fOverwrite?"RECREATE":"NEW"); // EffectiveOnTime
256 MWriteRootFile writem(2, rule, fOverwrite?"RECREATE":"NEW"); // Muons
257
258 // Data
259 write.AddContainer("MHillas", "Events");
260 write.AddContainer("MHillasExt", "Events");
261 write.AddContainer("MHillasSrc", "Events");
262 write.AddContainer("MImagePar", "Events");
263 write.AddContainer("MNewImagePar", "Events");
264 write.AddContainer("MRawEvtHeader", "Events");
265 write.AddContainer("MPointingPos", "Events");
266
267 // Run Header
268 write.AddContainer("MRawRunHeader", "RunHeaders");
269 write.AddContainer("MBadPixelsCam", "RunHeaders");
270 write.AddContainer("MGeomCam", "RunHeaders");
271 write.AddContainer("MObservatory", "RunHeaders");
272
273 // Muon Setup
274 write.AddContainer("BinningRadius", "RunHeaders");
275 write.AddContainer("BinningArcWidth", "RunHeaders");
276 write.AddContainer("BinningRingBroadening", "RunHeaders");
277 write.AddContainer("BinningSizeVsArcRadius", "RunHeaders");
278 write.AddContainer("MMuonSetup", "RunHeaders");
279
280 if (ismc)
281 {
282 // Monte Carlo Data
283 write.AddContainer("MMcEvt", "Events");
284 write.AddContainer("MMcTrig", "Events");
285 // Monte Carlo Run Headers
286 write.AddContainer("MMcRunHeader", "RunHeaders");
287 write.AddContainer("MMcTrigHeader", "RunHeaders");
288 write.AddContainer("MMcFadcHeader", "RunHeaders");
289 write.AddContainer("MMcConfigRunHeader", "RunHeaders");
290 write.AddContainer("MMcCorsikaRunHeader", "RunHeaders");
291 }
292 else
293 {
294 write.AddContainer("MTime", "Events");
295 // Drive
296 write.AddContainer("MReportDrive", "Drive");
297 write.AddContainer("MTimeDrive", "Drive");
298 // Starguider
299 write.AddContainer("MReportStarguider", "Starguider");
300 write.AddContainer("MTimeStarguider", "Starguider");
301 // Effective On Time
302 writet.AddContainer("MEffectiveOnTime", "EffectiveOnTime");
303 writet.AddContainer("MTimeEffectiveOnTime", "EffectiveOnTime");
304 }
305
306 // What to write in muon tree
307 writem.AddContainer("MMuonSearchPar", "Muons");
308 writem.AddContainer("MMuonCalibPar", "Muons");
309 writem.AddContainer("MHillas", "Muons");
310 writem.AddContainer("MHillasExt", "Muons");
311 writem.AddContainer("MHillasSrc", "Muons");
312 writem.AddContainer("MImagePar", "Muons");
313 writem.AddContainer("MNewImagePar", "Muons");
314 writem.AddContainer("MRawEvtHeader", "Muons");
315 writem.AddContainer("MPointingPos", "Muons");
316 if (ismc)
317 {
318 // Monte Carlo Data
319 writem.AddContainer("MMcEvt", "Muons");
320 writem.AddContainer("MMcTrig", "Muons");
321 }
322
323 if (ismc)
324 if (fMuonAnalysis)
325 writem.AddCopySource("OriginalMC");
326 else
327 write.AddCopySource("OriginalMC");
328
329 MTaskList tlist2("Events");
330 tlist2.AddToList(&apply);
331 if (!ismc)
332 tlist2.AddToList(&cont);
333 tlist2.AddToList(&contsw);
334 if (!ismc)
335 {
336 // Calibration events don't enter star at all.
337 tlist2.AddToList(&rate);
338 tlist2.AddToList(&fillvs);
339 tlist2.AddToList(&fill9);
340 tlist2.AddToList(&writet);
341 }
342 tlist2.AddToList(&clean);
343 tlist2.AddToList(&fill0a);
344 tlist2.AddToList(&fill0b);
345 //tlist2.AddToList(&poscalc);
346 //tlist2.AddToList(&poscorrect);
347 tlist2.AddToList(&hcalc);
348 tlist2.AddToList(&fill1);
349 tlist2.AddToList(&fill2);
350 tlist2.AddToList(&fill3);
351 tlist2.AddToList(&fill4);
352 tlist2.AddToList(&fill5);
353
354 // ----------------------- Muon Analysis ----------------------
355 MF fmuon1("MHillas.fSize>150", "MuonPreCut");
356 MF fmuon2("(MMuonSearchPar.fRadius>180) && (MMuonSearchPar.fRadius<400) &&"
357 "(MMuonSearchPar.fDeviation<45)", "MuonSearchCut");
358 MF fmuon3("(MMuonCalibPar.fArcPhi>190) && (MMuonSearchPar.fDeviation<35) &&"
359 "(MMuonCalibPar.fArcWidth<0.20) && (MMuonCalibPar.fArcWidth>0.04)",
360 "MuonFinalCut");
361 MFDataMember fmuon4("MMuonCalibPar.fArcPhi", '>', -0.5, "MuonWriteCut");
362 writem.SetFilter(&fmuon4);
363
364 MMuonSearchParCalc muscalc;
365 muscalc.SetFilter(&fmuon1);
366
367 MMuonCalibParCalc mcalc;
368 mcalc.SetFilter(&fmuon2);
369
370 MFillH fillmuon("MHSingleMuon", "", "FillMuon");
371 MFillH fillmpar("MHMuonPar", "", "FillMuonPar");
372 fillmuon.SetFilter(&fmuon2);
373 fillmpar.SetFilter(&fmuon3);
374 fillmuon.SetBit(MFillH::kDoNotDisplay);
375
376 if (fMuonAnalysis)
377 {
378 tlist2.AddToList(&fmuon1);
379 tlist2.AddToList(&muscalc);
380 tlist2.AddToList(&fmuon2);
381 tlist2.AddToList(&fillmuon);
382 tlist2.AddToList(&mcalc);
383 tlist2.AddToList(&fmuon3);
384 tlist2.AddToList(&fillmpar);
385 tlist2.AddToList(&fmuon4);
386 tlist2.AddToList(&writem);
387 }
388 // ------------------------------------------------------------
389
390 MFillH fillw("MHWeather", "MTimeCC", "FillWeather");
391 //tlist2.AddToList(&fill6);
392 //tlist2.AddToList(&fill7);
393 //tlist2.AddToList(&fill9);
394
395 MPointingPosCalc pcalc;
396
397 tlist.AddToList(ismc ? (MTask*)&readmc : (MTask*)&readreal);
398 tlist.AddToList(&pcalc, ismc ? "Events" : "Drive");
399 tlist.AddToList(&tlist2, "Events");
400 if (!ismc)
401 {
402 tlist.AddToList(&fillw, "CC");
403 tlist.AddToList(&fillp1, "Drive");
404 tlist.AddToList(&fillp2, "Starguider");
405 }
406 tlist.AddToList(&write);
407
408 // Create and setup the eventloop
409 MEvtLoop evtloop(fName);
410 evtloop.SetParList(&plist);
411 evtloop.SetDisplay(fDisplay);
412 evtloop.SetLogStream(fLog);
413 if (!SetupEnv(evtloop))
414 return kFALSE;
415
416 // Execute first analysis
417 if (!evtloop.Eventloop(fMaxEvents))
418 {
419 *fLog << err << GetDescriptor() << ": Failed." << endl;
420 return kFALSE;
421 }
422
423 if (!WriteResult())
424 return kFALSE;
425
426 *fLog << all << GetDescriptor() << ": Done." << endl;
427 *fLog << endl << endl;
428
429 return kTRUE;
430}
Note: See TracBrowser for help on using the repository browser.