/[PAMELA software]/DarthVader/TriggerLevel2/src/TrigCore.cpp
ViewVC logotype

Contents of /DarthVader/TriggerLevel2/src/TrigCore.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.29 - (show annotations) (download)
Tue Oct 14 14:07:45 2014 UTC (10 years, 1 month ago) by mocchiut
Branch: MAIN
CVS Tags: v10RED, v10REDr01, HEAD
Changes since 1.28: +2 -1 lines
10RED: lost sync bug fixed

1 //
2 // C/C++ headers
3 //
4 #include <fstream>
5 #include <string.h>
6 //
7 // ROOT headers
8 //
9 #include <TTree.h>
10 #include <TClassEdit.h>
11 #include <TObject.h>
12 #include <TList.h>
13 #include <TArrayL.h>
14 #include <TSystem.h>
15 #include <TSystemDirectory.h>
16 #include <TString.h>
17 #include <TFile.h>
18 #include <TClass.h>
19 #include <TCanvas.h>
20 #include <TH1.h>
21 #include <TH1F.h>
22 #include <TH2D.h>
23 #include <TLatex.h>
24 #include <TPad.h>
25 #include <TSQLServer.h>
26 #include <TSQLRow.h>
27 #include <TSQLResult.h>
28 #include <TClonesArray.h>
29 //
30 // RunInfo header
31 //
32 #include <RunInfo.h>
33 //
34 // YODA headers
35 //
36 #include <PamelaRun.h>
37 #include <physics/trigger/TriggerEvent.h>
38 //
39 // This program headers
40 //
41 #include <TrigCore.h>
42 #include <TrigLevel2.h>
43 #include <TrigVerl2.h>
44 //
45 using namespace std;
46 //
47 //
48 // CORE ROUTINE
49 //
50 //
51 int TrigCore(UInt_t run, TFile *file, GL_TABLES *glt, Int_t Trigargc, char *Trigargv[]){
52 //
53 TString processFolder = Form("TrigFolder_%u",run);
54 //
55 // Set these to true to have a very verbose output.
56 //
57 Bool_t verbose = false;
58 Bool_t debug = false;
59 //
60 if ( Trigargc > 0 ){
61 Int_t i = 0;
62 while ( i < Trigargc ){
63 if ( !strcmp(Trigargv[i],"-processFolder") ) {
64 if ( Trigargc < i+1 ){
65 throw -3;
66 };
67 processFolder = (TString)Trigargv[i+1];
68 i++;
69 };
70 if ( !strcmp(Trigargv[i],"-v") || !strcmp(Trigargv[i],"--verbose") ) {
71 verbose = true;
72 };
73 if ( !strcmp(Trigargv[i],"-g") || !strcmp(Trigargv[i],"--debug") ) {
74 verbose = true;
75 debug = true;
76 };
77 i++;
78 };
79 };
80 //
81 //
82 // Output directory is the working directoy.
83 //
84 const char* outdir = gSystem->DirName(gSystem->DirName(file->GetPath()));
85 //
86 // Variables for level2
87 //
88 Long64_t maxsize = 10000000000LL;
89 TTree::SetMaxTreeSize(maxsize);
90 //
91 TTree *trigt = 0;
92 UInt_t nevents = 0;
93 //
94 // variables needed to reprocess data
95 //
96 TString trigversion;
97 ItoRunInfo *runinfo = 0;
98 TArrayI *runlist = 0;
99 TTree *trigtclone = 0;
100 Bool_t reproc = false;
101 Bool_t reprocall = false;
102 UInt_t nobefrun = 0;
103 UInt_t noaftrun = 0;
104 UInt_t numbofrun = 0;
105 stringstream ftmpname;
106 TString fname;
107 UInt_t totfileentries = 0;
108 UInt_t idRun = 0;
109 //
110 // variables needed to handle error signals
111 //
112 Int_t code = 0;
113 Int_t sgnl;
114 //
115 // trigger level2 classes
116 //
117 TrigLevel2 *trig = new TrigLevel2();
118 TrigLevel2 *trigclone = new TrigLevel2();
119 //
120 // define variables for opening and reading level0 file
121 //
122 TFile *l0File = 0;
123 TTree *l0tr = 0;
124 TBranch *l0head = 0;
125 TBranch *l0trig = 0;
126 pamela::EventHeader *eh = 0;
127 pamela::PscuHeader *ph = 0;
128 pamela::trigger::TriggerEvent *triggerEvent = 0;
129 //
130 // Define other basic variables
131 //
132 UInt_t procev = 0;
133 stringstream file2;
134 stringstream file3;
135 stringstream qy;
136 Int_t totevent = 0;
137 UInt_t atime = 0;
138 UInt_t re = 0;
139 //
140 // Working filename
141 //
142 TString outputfile;
143 stringstream name;
144 name.str("");
145 name << outdir << "/";
146 //
147 // temporary file and folder
148 //
149 TFile *tempfile = 0;
150 TTree *temptrig = 0;
151 stringstream tempname;
152 stringstream Trigfolder;
153 Bool_t myfold = false;
154 tempname.str("");
155 tempname << outdir;
156 tempname << "/" << processFolder.Data();
157 Trigfolder.str("");
158 Trigfolder << tempname.str().c_str();
159 tempname << "/trigtree_run";
160 tempname << run << ".root";
161 UInt_t totnorun = 0;
162 //
163 // DB classes
164 //
165 GL_ROOT *glroot = new GL_ROOT();
166 GL_TIMESYNC *dbtime = 0;
167 //
168 // Let's start!
169 //
170 //
171 // As a first thing we must check what we have to do: if run = 0 we must process all events in the file has been passed
172 // if run != 0 we must process only that run but first we have to check if the tree Trigger already exist in the file
173 // if it exists we are reprocessing data and we must delete that entries, if not we must create it.
174 //
175 if ( run == 0 ) reproc = true;
176 //
177 //
178 // Output file is "outputfile"
179 //
180 if ( !file->IsOpen() ){
181 if ( verbose ) printf(" Trigger - ERROR: cannot open file for writing\n");
182 throw -401;
183 };
184 //
185 // Retrieve GL_RUN variables from the level2 file
186 //
187 trigversion = TrigInfo(false); // we should decide how to handle versioning system
188 //
189 // create an interface to RunInfo called "runinfo"
190 //
191 runinfo = new ItoRunInfo(file);
192 //
193 // open "Run" tree in level2 file, if not existing return an error (sngl != 0)
194 //
195 sgnl = 0;
196 sgnl = runinfo->Update(run,"TRIG",trigversion);
197 if ( sgnl ){
198 if ( verbose ) printf(" Trigger - ERROR: RunInfo exited with non-zero status\n");
199 code = sgnl;
200 goto closeandexit;
201 } else {
202 sgnl = 0;
203 };
204 //
205 // number of events in the file BEFORE the first event of our run
206 //
207 nobefrun = runinfo->GetFirstEntry();
208 //
209 // total number of events in the file
210 //
211 totfileentries = runinfo->GetFileEntries();
212 //
213 // first file entry AFTER the last event of our run
214 //
215 noaftrun = runinfo->GetLastEntry() + 1;
216 //
217 // number of run to be processed
218 //
219 numbofrun = runinfo->GetNoRun();
220 totnorun = runinfo->GetRunEntries();
221 //
222 // Try to access the Trigger tree in the file, if it exists we are reprocessing data if not we are processing a new run
223 //
224 trigtclone = (TTree*)file->Get("Trigger");
225 //
226 if ( !trigtclone ){
227 //
228 // tree does not exist, we are not reprocessing
229 //
230 reproc = false;
231 if ( run == 0 && verbose ) printf(" Trigger - WARNING: you are reprocessing data but Trigger tree does not exist!\n");
232 if ( runinfo->IsReprocessing() && run != 0 && verbose ) printf(" Trigger - WARNING: it seems you are not reprocessing data but Trigger\n versioning information already exists in RunInfo.\n");
233
234 } else {
235 //
236 // tree exists, we are reprocessing data. Are we reprocessing a single run or all the file?
237 //
238 trigtclone->SetAutoSave(900000000000000LL);
239 reproc = true;
240 //
241 // update versioning information
242 //
243 if ( verbose ) printf("\n Preparing the pre-processing...\n");
244 //
245 if ( run == 0 || totnorun == 1 ){
246 //
247 // we are reprocessing all the file
248 // if we are reprocessing everything we don't need to copy any old event and we can just work with the new tree and delete the old one immediately
249 //
250 reprocall = true;
251 //
252 if ( verbose ) printf("\n Trigger - WARNING: Reprocessing all runs\n");
253 //
254 } else {
255 //
256 // we are reprocessing a single run, we must copy to the new tree the events in the file which preceed the first event of the run
257 //
258 reprocall = false;
259 //
260 if ( verbose ) printf("\n Trigger - WARNING: Reprocessing run number %u \n",run);
261 //
262 // copying old tree to a new file
263 //
264 gSystem->MakeDirectory(Trigfolder.str().c_str());
265 myfold = true;
266 tempfile = new TFile(tempname.str().c_str(),"RECREATE");
267 temptrig = trigtclone->CloneTree(-1,"fast");
268 temptrig->SetName("Trigger-old");
269 tempfile->Write();
270 tempfile->Close();
271 }
272 //
273 // Delete the old tree from old file and memory
274 //
275 trigtclone->Delete("all");
276 //
277 if ( verbose ) printf(" ...done!\n");
278 //
279 };
280 //
281 // create Trigger tree trigt
282 //
283 file->cd();
284 trigt = new TTree("Trigger-new","PAMELA Level2 Trigger data");
285 trigt->SetAutoSave(900000000000000LL);
286 trigt->Branch("TrigLevel2","TrigLevel2",&trig);
287 //
288 if ( reproc && !reprocall ){
289 //
290 // open new file and retrieve alo tree informations
291 //
292 tempfile = new TFile(tempname.str().c_str(),"READ");
293 trigtclone = (TTree*)tempfile->Get("Trigger-old");
294 trigtclone->SetAutoSave(900000000000000LL);
295 trigtclone->SetBranchAddress("TrigLevel2",&trigclone);
296 //
297 if ( nobefrun > 0 ){
298 if ( verbose ) printf("\n Pre-processing: copying events from the old tree before the processed run\n");
299 if ( verbose ) printf(" Copying %u events in the file which are before the beginning of the run %u \n",nobefrun,run);
300 if ( verbose ) printf(" Start copying at event number 0, end copying at event number %u \n",nobefrun);
301 for (UInt_t j = 0; j < nobefrun; j++){
302 //
303 if ( trigtclone->GetEntry(j) <= 0 ) throw -36;
304 //
305 // copy trigclone to trig
306 //
307 // trig = new TrigLevel2();
308 // trig->Clear();
309 memcpy(&trig,&trigclone,sizeof(trigclone));
310 //
311 // Fill entry in the new tree
312 //
313 trigt->Fill();
314 //
315 };
316 if ( verbose ) printf(" Finished successful copying!\n");
317 };
318 };
319 //
320 // Get the list of run to be processed, if only one run has to be processed the list will contain one entry only.
321 //
322 runlist = runinfo->GetRunList();
323 //
324 // Loop over the run to be processed
325 //
326 for (UInt_t irun=0; irun < numbofrun; irun++){
327 //
328 // retrieve the first run ID to be processed using the RunInfo list
329 //
330 idRun = runlist->At(irun);
331 if ( verbose ) printf("\n\n\n ####################################################################### \n");
332 if ( verbose ) printf(" PROCESSING RUN NUMBER %u \n",idRun);
333 if ( verbose ) printf(" ####################################################################### \n\n\n");
334 //
335 runinfo->ID_ROOT_L0 = 0;
336 //
337 // store in the runinfo class the GL_RUN variables for our run
338 //
339 sgnl = 0;
340 sgnl = runinfo->GetRunInfo(idRun);
341 if ( sgnl ){
342 if ( verbose ) printf(" Trigger - ERROR: RunInfo exited with non-zero status\n");
343 code = sgnl;
344 goto closeandexit;
345 } else {
346 sgnl = 0;
347 };
348 //
349 // now you can access that variables using the RunInfo class this way runinfo->ID_ROOT_L0
350 //
351 if ( runinfo->ID_ROOT_L0 == 0 ){
352 if ( verbose ) printf("\n Trigger - ERROR: no run with ID_RUN = %u \n\n Exiting... \n\n",idRun);
353 code = -5;
354 goto closeandexit;
355 };
356 //
357 // prepare the timesync for the db
358 //
359 TString host = glt->CGetHost();
360 TString user = glt->CGetUser();
361 TString psw = glt->CGetPsw();
362 TSQLServer *dbc = TSQLServer::Connect(host.Data(),user.Data(),psw.Data());
363 if ( !dbc->IsConnected() ) throw -403;
364 stringstream myquery;
365 myquery.str("");
366 myquery << "SET time_zone='+0:00';";
367 delete dbc->Query(myquery.str().c_str());
368 delete dbc->Query("SET sql_mode = 'NO_UNSIGNED_SUBTRACTION';");
369 dbtime = new GL_TIMESYNC(runinfo->ID_ROOT_L0,"ID",dbc);
370 //
371 // Search in the DB the path and name of the LEVEL0 file to be processed.
372 //
373 // if ( !dbc->IsConnected() ) throw -403;
374 glroot->Query_GL_ROOT(runinfo->ID_ROOT_L0,dbc);
375 //
376 ftmpname.str("");
377 ftmpname << glroot->PATH.Data() << "/";
378 ftmpname << glroot->NAME.Data();
379 fname = ftmpname.str().c_str();
380 //
381 // print out informations
382 //
383 totevent = runinfo->NEVENTS;
384 if ( verbose ) printf("\n LEVEL0 data file: %s \n",fname.Data());
385 if ( verbose ) printf(" RUN HEADER absolute time is: %u \n",runinfo->RUNHEADER_TIME);
386 if ( verbose ) printf(" RUN TRAILER absolute time is: %u \n",runinfo->RUNTRAILER_TIME);
387 if ( verbose ) printf(" %i events to be processed for run %u: from %i to %i \n\n",totevent,idRun,runinfo->EV_FROM,runinfo->EV_FROM+totevent);
388 //
389 // if ( !totevent ) goto closeandexit;
390 //
391 // Open Level0 file
392 //
393 if ( l0File ) l0File->Close();
394 l0File = new TFile(fname.Data());
395 if ( !l0File ) {
396 if ( verbose ) printf(" Trigger - ERROR: problems opening Level0 file\n");
397 code = -6;
398 goto closeandexit;
399 };
400 l0tr = (TTree*)l0File->Get("Physics");
401 if ( !l0tr ) {
402 if ( verbose ) printf(" Trigger - ERROR: no Physics tree in Level0 file\n");
403 l0File->Close();
404 code = -7;
405 goto closeandexit;
406 };
407 l0head = l0tr->GetBranch("Header");
408 if ( !l0head ) {
409 if ( verbose ) printf(" Trigger - ERROR: no Header branch in Level0 tree\n");
410 l0File->Close();
411 code = -8;
412 goto closeandexit;
413 };
414 l0trig = l0tr->GetBranch("Trigger");
415 if ( !l0trig ) {
416 if ( verbose ) printf(" Trigger - ERROR: no Trigger branch in Level0 tree\n");
417 l0File->Close();
418 code = -402;
419 goto closeandexit;
420 };
421 //
422 l0tr->SetBranchAddress("Trigger", &triggerEvent);
423 l0tr->SetBranchAddress("Header", &eh);
424 //
425 nevents = l0trig->GetEntries();
426 //
427 if ( nevents < 1 && totevent ) {
428 if ( verbose ) printf(" Trigger - ERROR: Level0 file is empty\n\n");
429 l0File->Close();
430 code = -11;
431 goto closeandexit;
432 };
433 //
434 if ( runinfo->EV_TO > nevents-1 && totevent ) {
435 if ( verbose ) printf(" Trigger - ERROR: too few entries in the tree\n");
436 l0File->Close();
437 code = -12;
438 goto closeandexit;
439 };
440 //
441 // run over all the events of the run
442 //
443 if ( verbose ) printf("\n Ready to start! \n\n Processed events: \n\n");
444 //
445 if ( dbc ){
446 dbc->Close();
447 delete dbc;
448 dbc = 0;
449 };
450 //
451 for ( re = runinfo->EV_FROM; re < (runinfo->EV_FROM+runinfo->NEVENTS); re++){
452 //
453 if ( procev%1000 == 0 && procev > 0 && verbose ) printf(" %iK \n",procev/1000);
454 //
455 if ( l0head->GetEntry(re) <= 0 ) throw -36;
456 //
457 // absolute time of this event
458 //
459 ph = eh->GetPscuHeader();
460 atime = dbtime->DBabsTime(ph->GetOrbitalTime());
461 //
462 // paranoid check
463 //
464 if ( atime > (runinfo->RUNTRAILER_TIME+1) || atime < (runinfo->RUNHEADER_TIME-1) ) {
465 if ( verbose || debug ) printf(" Trigger - WARNING: event at time outside the run time window, skipping it\n");
466 goto jumpev;
467 };
468 ///
469 if ( l0trig->GetEntry(re) <= 0 ) throw -36;
470 ///
471 //
472 procev++;
473 //
474 // start processing
475 //
476 trig->Clear();
477 // trig = new TrigLevel2();
478 //
479 // now we must copy from the output structure to the level2 class:
480 //
481 trig->evcount = triggerEvent->evcount;
482 for (Int_t kk=0; kk<3;kk++){
483 trig->pmtpl[kk] = triggerEvent->pmtpl[kk];
484 trig->patternbusy[kk] = triggerEvent->patternbusy[kk];
485 }
486
487 for (Int_t kk=0; kk<6;kk++){
488 trig->trigrate[kk] = triggerEvent->trigrate[kk];
489 trig->patterntrig[kk] = triggerEvent->patterntrig[kk];
490 }
491
492 for (Int_t kk=0; kk<2;kk++){
493 trig->dltime[kk] = triggerEvent->dltime[kk];
494 trig->s4calcount[kk] = triggerEvent->s4calcount[kk];
495 }
496
497 for (Int_t kk=0; kk<24;kk++){
498 trig->pmtcount1[kk] = triggerEvent->pmtcount1[kk];
499 trig->pmtcount2[kk] = triggerEvent->pmtcount2[kk];
500 }
501
502 trig->trigconf = triggerEvent->trigconf;
503
504 trig->unpackError = triggerEvent->unpackError;
505
506 trigt->Fill();
507 //
508 //
509 jumpev:
510 debug = false;
511 //
512 };
513 //
514 // Here you may want to clear some variables before processing another run
515 //
516 delete dbtime;
517 }; // process all the runs
518 //
519 if ( verbose ) printf("\n Finished processing data \n");
520 //
521 closeandexit:
522 //
523 // we have finished processing the run(s). If we processed a single run now we must copy all the events after our run from the old tree to the new one and delete the old tree.
524 //
525 if ( !reprocall && reproc && code >= 0 ){
526 if ( totfileentries > noaftrun ){
527 if ( verbose ) printf("\n Post-processing: copying events from the old tree after the processed run\n");
528 if ( verbose ) printf(" Copying %i events in the file which are after the end of the run %i \n",(int)(totfileentries-noaftrun),(int)run);
529 if ( verbose ) printf(" Start copying at event number %i end copying at event number %i \n",(int)noaftrun,(int)totfileentries);
530 for (UInt_t j = noaftrun; j < totfileentries; j++ ){
531 //
532 // Get entry from old tree
533 //
534 if ( trigtclone->GetEntry(j) <= 0 ) throw -36;
535 //
536 // copy trigclone to trig
537 //
538 // trig->Clear();
539 //
540 memcpy(&trig,&trigclone,sizeof(trigclone));
541 //
542 // Fill entry in the new tree
543 //
544 trigt->Fill();
545 };
546 if ( verbose ) printf(" Finished successful copying!\n");
547 };
548 };
549 //
550 // Close files, delete old tree(s), write and close level2 file
551 //
552 if ( l0File ) l0File->Close();
553 if ( tempfile ) tempfile->Close();
554 if ( myfold ) gSystem->Unlink(tempname.str().c_str());
555 //
556 if ( code < 0 && verbose ) printf("\n Trigger - ERROR: an error occurred, try to save anyway...\n");
557 if ( verbose ) printf("\n Writing and closing rootple\n");
558 if ( trigt ) trigt->SetName("Trigger");
559 if ( file ){
560 file->cd();
561 if ( trigt ) trigt->Write("Trigger", TObject::kOverwrite); // 10RED bug fixed
562 // file->Write("Trigger");
563 };
564 //
565 if ( myfold ) gSystem->Unlink(Trigfolder.str().c_str());
566 //
567 // the end
568 //
569 if ( verbose ) printf("\n Exiting...\n");
570 //
571 if ( glroot ) delete glroot;
572 if ( runinfo ) runinfo->Close();
573 if ( runinfo ) delete runinfo;
574 //
575 if ( code < 0 ) throw code;
576 return(code);
577 }
578
579
580

  ViewVC Help
Powered by ViewVC 1.1.23