26 |
#include <stdlib.h> |
#include <stdlib.h> |
27 |
#include <math.h> |
#include <math.h> |
28 |
// |
// |
29 |
|
// RunInfo header |
30 |
|
// |
31 |
|
#include <RunInfo.h> |
32 |
|
#include <GLTables.h> |
33 |
|
// |
34 |
// YODA headers |
// YODA headers |
35 |
// |
// |
36 |
#include <PamelaRun.h> |
#include <PamelaRun.h> |
39 |
#include <EventHeader.h> |
#include <EventHeader.h> |
40 |
#include <physics/neutronDetector/NeutronEvent.h> |
#include <physics/neutronDetector/NeutronEvent.h> |
41 |
#include <physics/neutronDetector/NeutronRecord.h> |
#include <physics/neutronDetector/NeutronRecord.h> |
|
//#include <yodaUtility.h> |
|
42 |
// |
// |
43 |
// RunInfo header |
// RunInfo header |
44 |
// |
// |
45 |
#include <RunInfo.h> |
//#include <RunInfo.h> |
46 |
#include <GLTables.h> |
//#include <GLTables.h> |
47 |
// |
// |
48 |
// This program headers |
// This program headers |
49 |
// |
// |
57 |
// CORE ROUTINE |
// CORE ROUTINE |
58 |
// |
// |
59 |
// |
// |
60 |
int NDCore(UInt_t run, TFile *file, TSQLServer *dbc, Int_t NDargc, char *NDargv[]){ |
int NDCore(UInt_t run, TFile *file, GL_TABLES *glt, Int_t NDargc, char *NDargv[]){ |
|
// |
|
|
Int_t i = 0; |
|
61 |
// |
// |
62 |
TString processFolder = "NDFolder"; |
// Set these to true to have a verbose output. |
|
// |
|
|
// Set these to true to have a very verbose output. |
|
63 |
// |
// |
64 |
Bool_t debug = false; |
Bool_t debug = false; |
|
// |
|
65 |
Bool_t verbose = false; |
Bool_t verbose = false; |
66 |
// |
// |
67 |
|
// Output directory is the working directoy. |
68 |
|
// |
69 |
|
const char* outDir = gSystem->DirName(gSystem->DirName(file->GetPath())); |
70 |
|
// |
71 |
|
Int_t i = 0; |
72 |
|
TString processFolder = Form("NDFolder_%u",run); |
73 |
if ( NDargc > 0 ){ |
if ( NDargc > 0 ){ |
74 |
i = 0; |
i = 0; |
75 |
while ( i < NDargc ){ |
while ( i < NDargc ){ |
84 |
verbose = true; |
verbose = true; |
85 |
}; |
}; |
86 |
if ( (!strcmp(NDargv[i],"--debug")) || (!strcmp(NDargv[i],"-g")) ) { |
if ( (!strcmp(NDargv[i],"--debug")) || (!strcmp(NDargv[i],"-g")) ) { |
87 |
|
debug = true; |
88 |
verbose= true; |
verbose= true; |
89 |
}; |
}; |
90 |
i++; |
i++; |
91 |
}; |
}; |
92 |
}; |
}; |
93 |
// |
// |
94 |
const char* outDir = gSystem->DirName(gSystem->DirName(file->GetPath())); |
|
95 |
// Variables for level2 |
// Variables for level2 |
96 |
// |
// |
97 |
TTree *NDtr = 0; |
TTree *NDtr = 0; |
98 |
UInt_t nevents = 0; |
UInt_t nevents = 0; |
99 |
// |
// |
100 |
// variables needed to reprocess data |
// Variables needed to reprocess data |
101 |
// |
// |
102 |
TString NDversion; |
TString NDversion; |
103 |
ItoRunInfo *runinfo = 0; |
ItoRunInfo *runinfo = 0; |
122 |
Int_t code = 0; |
Int_t code = 0; |
123 |
Int_t sgnl; |
Int_t sgnl; |
124 |
// |
// |
125 |
|
Long64_t maxsize = 10000000000LL; |
126 |
|
TTree::SetMaxTreeSize(maxsize); |
127 |
|
// |
128 |
|
// |
129 |
// ND level2 classes |
// ND level2 classes |
130 |
// |
// |
131 |
NDLevel2 *nd = new NDLevel2(); |
NDLevel2 *nd = new NDLevel2(); |
165 |
TTree *tempND = 0; |
TTree *tempND = 0; |
166 |
stringstream tempname; |
stringstream tempname; |
167 |
stringstream NDfolder; |
stringstream NDfolder; |
168 |
|
Bool_t myfold = false; |
169 |
tempname.str(""); |
tempname.str(""); |
170 |
tempname << outDir; |
tempname << outDir; |
171 |
tempname << "/" << processFolder.Data(); |
tempname << "/" << processFolder.Data(); |
172 |
NDfolder.str(""); |
NDfolder.str(""); |
173 |
NDfolder << tempname.str().c_str(); |
NDfolder << tempname.str().c_str(); |
|
gSystem->MakeDirectory(NDfolder.str().c_str()); |
|
174 |
tempname << "/NDtree_run"; |
tempname << "/NDtree_run"; |
175 |
tempname << run << ".root"; |
tempname << run << ".root"; |
176 |
|
UInt_t totnorun = 0; |
177 |
// |
// |
178 |
// DB classes |
// DB classes |
179 |
// |
// |
232 |
// |
// |
233 |
numbofrun = runinfo->GetNoRun(); |
numbofrun = runinfo->GetNoRun(); |
234 |
// |
// |
235 |
|
// |
236 |
|
// |
237 |
|
totnorun = runinfo->GetRunEntries(); |
238 |
|
// |
239 |
// Try to access the S4 tree in the file, if it exists we are reprocessing data if not we are processing a new run |
// Try to access the S4 tree in the file, if it exists we are reprocessing data if not we are processing a new run |
240 |
// |
// |
241 |
NDtrclone = (TTree*)file->Get("NeutronD"); |
NDtrclone = (TTree*)file->Get("NeutronD"); |
253 |
} |
} |
254 |
} else { |
} else { |
255 |
// |
// |
256 |
|
NDtrclone->SetAutoSave(900000000000000LL); |
257 |
|
// |
258 |
// tree exists, we are reprocessing data. Are we reprocessing a single run or all the file? |
// tree exists, we are reprocessing data. Are we reprocessing a single run or all the file? |
259 |
// |
// |
260 |
reproc = true; |
reproc = true; |
263 |
// |
// |
264 |
if (verbose) printf("\n Preparing the pre-processing...\n"); |
if (verbose) printf("\n Preparing the pre-processing...\n"); |
265 |
// |
// |
266 |
if ( run == 0 ){ |
if ( run == 0 || totnorun == 1 ){ |
267 |
// |
// |
268 |
// we are reprocessing all the file |
// we are reprocessing all the file |
269 |
// 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 |
// 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 |
282 |
// |
// |
283 |
// copying old tree to a new file |
// copying old tree to a new file |
284 |
// |
// |
285 |
|
gSystem->MakeDirectory(NDfolder.str().c_str()); |
286 |
|
myfold = true; |
287 |
tempfile = new TFile(tempname.str().c_str(),"RECREATE"); |
tempfile = new TFile(tempname.str().c_str(),"RECREATE"); |
288 |
tempND = NDtrclone->CloneTree(-1,"fast"); |
tempND = NDtrclone->CloneTree(-1,"fast"); |
289 |
tempND->SetName("NeutronD-old"); |
tempND->SetName("NeutronD-old"); |
303 |
// |
// |
304 |
file->cd(); |
file->cd(); |
305 |
NDtr = new TTree("NeutronD-new","PAMELA Level2 NeutronD data"); |
NDtr = new TTree("NeutronD-new","PAMELA Level2 NeutronD data"); |
306 |
|
NDtr->SetAutoSave(900000000000000LL); |
307 |
NDtr->Branch("NDLevel2","NDLevel2",&nd); |
NDtr->Branch("NDLevel2","NDLevel2",&nd); |
308 |
// |
// |
309 |
if ( reproc && !reprocall ){ |
if ( reproc && !reprocall ){ |
312 |
// |
// |
313 |
tempfile = new TFile(tempname.str().c_str(),"READ"); |
tempfile = new TFile(tempname.str().c_str(),"READ"); |
314 |
NDtrclone = (TTree*)tempfile->Get("NeutronD-old"); |
NDtrclone = (TTree*)tempfile->Get("NeutronD-old"); |
315 |
|
NDtrclone->SetAutoSave(900000000000000LL); |
316 |
NDtrclone->SetBranchAddress("NDLevel2",&ndclone); |
NDtrclone->SetBranchAddress("NDLevel2",&ndclone); |
317 |
// |
// |
318 |
if ( nobefrun > 0 ){ |
if ( nobefrun > 0 ){ |
323 |
} |
} |
324 |
for (UInt_t j = 0; j < nobefrun; j++){ |
for (UInt_t j = 0; j < nobefrun; j++){ |
325 |
// |
// |
326 |
NDtrclone->GetEntry(j); |
if ( NDtrclone->GetEntry(j) <= 0 ) throw -36; |
327 |
// |
// |
328 |
// copy ndclone to mydec |
// copy ndclone to mydec |
329 |
// |
// |
330 |
// nd = new NDLevel2(); |
// nd = new NDLevel2(); |
331 |
nd->Clear(); |
// nd->Clear(); |
332 |
memcpy(&nd,&ndclone,sizeof(ndclone)); |
memcpy(&nd,&ndclone,sizeof(ndclone)); |
333 |
// |
// |
334 |
// Fill entry in the new tree |
// Fill entry in the new tree |
381 |
// |
// |
382 |
// prepare the timesync for the db |
// prepare the timesync for the db |
383 |
// |
// |
384 |
|
TString host = glt->CGetHost(); |
385 |
|
TString user = glt->CGetUser(); |
386 |
|
TString psw = glt->CGetPsw(); |
387 |
|
TSQLServer *dbc = TSQLServer::Connect(host.Data(),user.Data(),psw.Data()); |
388 |
|
if ( !dbc->IsConnected() ) throw -604; |
389 |
|
stringstream myquery; |
390 |
|
myquery.str(""); |
391 |
|
myquery << "SET time_zone='+0:00';"; |
392 |
|
delete dbc->Query(myquery.str().c_str()); |
393 |
|
delete dbc->Query("SET sql_mode = 'NO_UNSIGNED_SUBTRACTION';"); |
394 |
dbtime = new GL_TIMESYNC(runinfo->ID_ROOT_L0,"ID",dbc); |
dbtime = new GL_TIMESYNC(runinfo->ID_ROOT_L0,"ID",dbc); |
395 |
// |
// |
396 |
// Search in the DB the path and name of the LEVEL0 file to be processed. |
// Search in the DB the path and name of the LEVEL0 file to be processed. |
397 |
// |
// |
398 |
|
// if ( !dbc->IsConnected() ) throw -604; |
399 |
glroot->Query_GL_ROOT(runinfo->ID_ROOT_L0,dbc); |
glroot->Query_GL_ROOT(runinfo->ID_ROOT_L0,dbc); |
400 |
// |
// |
401 |
ftmpname.str(""); |
ftmpname.str(""); |
413 |
printf(" %i events to be processed for run %u: from %i to %i \n\n",totevent,idRun,runinfo->EV_FROM,runinfo->EV_FROM+totevent); |
printf(" %i events to be processed for run %u: from %i to %i \n\n",totevent,idRun,runinfo->EV_FROM,runinfo->EV_FROM+totevent); |
414 |
} |
} |
415 |
// |
// |
416 |
|
// if ( !totevent ) goto closeandexit; |
417 |
|
// |
418 |
// Open Level0 file |
// Open Level0 file |
419 |
// |
// |
420 |
|
if ( l0File ) l0File->Close(); |
421 |
l0File = new TFile(fname.Data()); |
l0File = new TFile(fname.Data()); |
422 |
if ( !l0File ) { |
if ( !l0File ) { |
423 |
if ( debug ) printf(" ND - ERROR: problems opening Level0 file\n"); |
if ( debug ) printf(" ND - ERROR: problems opening Level0 file\n"); |
452 |
// |
// |
453 |
nevents = l0ND->GetEntries(); |
nevents = l0ND->GetEntries(); |
454 |
// |
// |
455 |
if ( nevents < 1 ) { |
if ( nevents < 1 && totevent ) { |
456 |
if ( debug ) printf(" ND - ERROR: Level0 file is empty\n\n"); |
if ( debug ) printf(" ND - ERROR: Level0 file is empty\n\n"); |
457 |
l0File->Close(); |
l0File->Close(); |
458 |
code = -11; |
code = -11; |
459 |
goto closeandexit; |
goto closeandexit; |
460 |
}; |
}; |
461 |
// |
// |
462 |
if ( runinfo->EV_TO > nevents-1 ) { |
if ( runinfo->EV_TO > nevents-1 && totevent ) { |
463 |
if ( debug ) printf(" ND - ERROR: too few entries in the tree\n"); |
if ( debug ) printf(" ND - ERROR: too few entries in the tree\n"); |
464 |
l0File->Close(); |
l0File->Close(); |
465 |
code = -12; |
code = -12; |
470 |
// |
// |
471 |
if (verbose) printf("\n Ready to start! \n\n Processed events: \n\n"); |
if (verbose) printf("\n Ready to start! \n\n Processed events: \n\n"); |
472 |
// |
// |
473 |
|
if ( dbc ){ |
474 |
|
dbc->Close(); |
475 |
|
delete dbc; |
476 |
|
dbc = 0; |
477 |
|
}; |
478 |
|
// |
479 |
for ( re = runinfo->EV_FROM; re < (runinfo->EV_FROM+runinfo->NEVENTS); re++){ |
for ( re = runinfo->EV_FROM; re < (runinfo->EV_FROM+runinfo->NEVENTS); re++){ |
480 |
// |
// |
481 |
if ( procev%1000 == 0 && procev > 0 && verbose ) printf(" %iK \n",procev/1000); |
if ( procev%1000 == 0 && procev > 0 && verbose ) printf(" %iK \n",procev/1000); |
482 |
// |
// |
483 |
l0head->GetEntry(re); |
if ( l0head->GetEntry(re) <= 0 ) throw -36; |
484 |
// |
// |
485 |
// absolute time of this event |
// absolute time of this event |
486 |
// |
// |
489 |
// |
// |
490 |
// paranoid check |
// paranoid check |
491 |
// |
// |
492 |
if ( (atime > runinfo->RUNTRAILER_TIME) || (atime < runinfo->RUNHEADER_TIME) ) { |
if ( (atime > (runinfo->RUNTRAILER_TIME+1)) || (atime < (runinfo->RUNHEADER_TIME-1)) ) { |
493 |
if (verbose) printf(" ND - WARNING: event at time outside the run time window, skipping it\n"); |
if (verbose) printf(" ND - WARNING: event at time outside the run time window, skipping it\n"); |
494 |
goto jumpev; |
goto jumpev; |
495 |
}; |
}; |
500 |
// |
// |
501 |
nd->Clear(); |
nd->Clear(); |
502 |
// |
// |
503 |
l0ND->GetEntry(re); |
if ( l0ND->GetEntry(re) <= 0 ) throw -36; |
504 |
tmpSize = l0ne->Records->GetEntries(); |
tmpSize = l0ne->Records->GetEntries(); |
505 |
if (!tmpSize) continue; |
if ( tmpSize && l0ne->unpackError == 0 ){ |
506 |
if (l0ne->unpackError == 1) continue; |
for (Int_t j = 0; j < tmpSize; j++){ |
507 |
for (Int_t j = 0; j < tmpSize; j++){ |
l0nr = (pamela::neutron::NeutronRecord*)l0ne->Records->At(j); |
508 |
l0nr = (pamela::neutron::NeutronRecord*)l0ne->Records->At(j); |
yTrig = yTrig + (float)l0nr->trigPhysics; |
509 |
yTrig = yTrig + (float)l0nr->trigPhysics; |
yUpperBackground = yUpperBackground + (float)l0nr->upperBack; |
510 |
yUpperBackground = yUpperBackground + (float)l0nr->upperBack; |
yBottomBackground = yBottomBackground + (float)l0nr->bottomBack; |
511 |
yBottomBackground = yBottomBackground + (float)l0nr->bottomBack; |
} |
512 |
} |
nd->upperBack = yUpperBackground; |
513 |
nd->upperBack = yUpperBackground; |
nd->bottomBack = yBottomBackground; |
514 |
nd->bottomBack = yBottomBackground; |
nd->trigPhysics = yTrig; |
515 |
nd->trigPhysics = yTrig; |
}; |
516 |
|
// |
517 |
|
nd->unpackError = l0ne->unpackError; |
518 |
|
// |
519 |
NDtr->Fill(); // Fill tree NDLevel2 in the root file |
NDtr->Fill(); // Fill tree NDLevel2 in the root file |
520 |
|
// |
521 |
yUpperBackground =0; |
yUpperBackground =0; |
522 |
yBottomBackground =0; |
yBottomBackground =0; |
523 |
yTrig=0; |
yTrig=0; |
550 |
// |
// |
551 |
// Get entry from old tree |
// Get entry from old tree |
552 |
// |
// |
553 |
NDtrclone->GetEntry(j); |
if ( NDtrclone->GetEntry(j) <= 0 ) throw -36; |
554 |
// |
// |
555 |
// copy ndclone to nd |
// copy ndclone to nd |
556 |
// |
// |
557 |
nd->Clear(); |
// nd->Clear(); |
558 |
memcpy(&nd,&ndclone,sizeof(ndclone)); |
memcpy(&nd,&ndclone,sizeof(ndclone)); |
559 |
// |
// |
560 |
// Fill entry in the new tree |
// Fill entry in the new tree |
569 |
// |
// |
570 |
if ( l0File ) l0File->Close(); |
if ( l0File ) l0File->Close(); |
571 |
if ( tempfile ) tempfile->Close(); |
if ( tempfile ) tempfile->Close(); |
572 |
gSystem->Unlink(tempname.str().c_str()); |
if ( myfold ) gSystem->Unlink(tempname.str().c_str()); |
573 |
// |
// |
|
if ( runinfo ) runinfo->Close(); |
|
574 |
if ( NDtr ) NDtr->SetName("NeutronD"); |
if ( NDtr ) NDtr->SetName("NeutronD"); |
575 |
if ( file ){ |
if ( file ){ |
576 |
file->cd(); |
file->cd(); |
577 |
file->Write(); |
if ( NDtr ) NDtr->Write("NeutronD", TObject::kOverwrite); // 10RED bug fixed |
578 |
}; |
}; |
579 |
// |
// |
580 |
gSystem->Unlink(NDfolder.str().c_str()); |
if ( myfold ) gSystem->Unlink(NDfolder.str().c_str()); |
581 |
// |
// |
582 |
// the end |
// the end |
583 |
// |
// |
584 |
if ( verbose ) printf("\n Exiting...\n"); |
if ( verbose ) printf("\n Exiting...\n"); |
585 |
if ( NDtr ) NDtr->Delete(); |
// if ( NDtr ) NDtr->Delete(); |
586 |
// |
// |
587 |
if ( nd ) delete nd; |
// if ( nd ) delete nd; |
588 |
if ( ndclone ) delete ndclone; |
//if ( ndclone ) delete ndclone; |
589 |
if ( glroot ) delete glroot; |
if ( glroot ) delete glroot; |
590 |
|
if ( runinfo ) runinfo->Close(); |
591 |
if ( runinfo ) delete runinfo; |
if ( runinfo ) delete runinfo; |
592 |
// |
// |
593 |
if(code < 0) throw code; |
if(code < 0) throw code; |