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 |
// |
// |
58 |
// |
// |
59 |
// |
// |
60 |
int NDCore(UInt_t run, TFile *file, TSQLServer *dbc, Int_t NDargc, char *NDargv[]){ |
int NDCore(UInt_t run, TFile *file, TSQLServer *dbc, Int_t NDargc, char *NDargv[]){ |
|
// |
|
|
Int_t i = 0; |
|
|
// |
|
|
TString processFolder = "NDFolder"; |
|
61 |
// |
// |
62 |
// Set these to true to have a very verbose output. |
// Set these to true to have a 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 |
// |
// |
231 |
// |
// |
232 |
numbofrun = runinfo->GetNoRun(); |
numbofrun = runinfo->GetNoRun(); |
233 |
// |
// |
234 |
|
// |
235 |
|
// |
236 |
|
UInt_t totnorun = runinfo->GetRunEntries(); |
237 |
|
// |
238 |
// 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 |
239 |
// |
// |
240 |
NDtrclone = (TTree*)file->Get("NeutronD"); |
NDtrclone = (TTree*)file->Get("NeutronD"); |
252 |
} |
} |
253 |
} else { |
} else { |
254 |
// |
// |
255 |
|
NDtrclone->SetAutoSave(900000000000000LL); |
256 |
|
// |
257 |
// 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? |
258 |
// |
// |
259 |
reproc = true; |
reproc = true; |
262 |
// |
// |
263 |
if (verbose) printf("\n Preparing the pre-processing...\n"); |
if (verbose) printf("\n Preparing the pre-processing...\n"); |
264 |
// |
// |
265 |
if ( run == 0 ){ |
if ( run == 0 || totnorun == 1 ){ |
266 |
// |
// |
267 |
// we are reprocessing all the file |
// we are reprocessing all the file |
268 |
// 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 |
281 |
// |
// |
282 |
// copying old tree to a new file |
// copying old tree to a new file |
283 |
// |
// |
284 |
|
gSystem->MakeDirectory(NDfolder.str().c_str()); |
285 |
|
myfold = true; |
286 |
tempfile = new TFile(tempname.str().c_str(),"RECREATE"); |
tempfile = new TFile(tempname.str().c_str(),"RECREATE"); |
287 |
tempND = NDtrclone->CloneTree(-1,"fast"); |
tempND = NDtrclone->CloneTree(-1,"fast"); |
288 |
tempND->SetName("NeutronD-old"); |
tempND->SetName("NeutronD-old"); |
302 |
// |
// |
303 |
file->cd(); |
file->cd(); |
304 |
NDtr = new TTree("NeutronD-new","PAMELA Level2 NeutronD data"); |
NDtr = new TTree("NeutronD-new","PAMELA Level2 NeutronD data"); |
305 |
|
NDtr->SetAutoSave(900000000000000LL); |
306 |
NDtr->Branch("NDLevel2","NDLevel2",&nd); |
NDtr->Branch("NDLevel2","NDLevel2",&nd); |
307 |
// |
// |
308 |
if ( reproc && !reprocall ){ |
if ( reproc && !reprocall ){ |
311 |
// |
// |
312 |
tempfile = new TFile(tempname.str().c_str(),"READ"); |
tempfile = new TFile(tempname.str().c_str(),"READ"); |
313 |
NDtrclone = (TTree*)tempfile->Get("NeutronD-old"); |
NDtrclone = (TTree*)tempfile->Get("NeutronD-old"); |
314 |
|
NDtrclone->SetAutoSave(900000000000000LL); |
315 |
NDtrclone->SetBranchAddress("NDLevel2",&ndclone); |
NDtrclone->SetBranchAddress("NDLevel2",&ndclone); |
316 |
// |
// |
317 |
if ( nobefrun > 0 ){ |
if ( nobefrun > 0 ){ |
380 |
// |
// |
381 |
// prepare the timesync for the db |
// prepare the timesync for the db |
382 |
// |
// |
383 |
|
// if ( !dbc->IsConnected() ) throw -604; |
384 |
dbtime = new GL_TIMESYNC(runinfo->ID_ROOT_L0,"ID",dbc); |
dbtime = new GL_TIMESYNC(runinfo->ID_ROOT_L0,"ID",dbc); |
385 |
// |
// |
386 |
// 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. |
387 |
// |
// |
388 |
|
// if ( !dbc->IsConnected() ) throw -604; |
389 |
glroot->Query_GL_ROOT(runinfo->ID_ROOT_L0,dbc); |
glroot->Query_GL_ROOT(runinfo->ID_ROOT_L0,dbc); |
390 |
// |
// |
391 |
ftmpname.str(""); |
ftmpname.str(""); |
457 |
// |
// |
458 |
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"); |
459 |
// |
// |
460 |
|
if ( dbc ){ |
461 |
|
dbc->Close(); |
462 |
|
// delete dbc; |
463 |
|
}; |
464 |
|
// |
465 |
for ( re = runinfo->EV_FROM; re < (runinfo->EV_FROM+runinfo->NEVENTS); re++){ |
for ( re = runinfo->EV_FROM; re < (runinfo->EV_FROM+runinfo->NEVENTS); re++){ |
466 |
// |
// |
467 |
if ( procev%1000 == 0 && procev > 0 && verbose ) printf(" %iK \n",procev/1000); |
if ( procev%1000 == 0 && procev > 0 && verbose ) printf(" %iK \n",procev/1000); |
555 |
// |
// |
556 |
if ( l0File ) l0File->Close(); |
if ( l0File ) l0File->Close(); |
557 |
if ( tempfile ) tempfile->Close(); |
if ( tempfile ) tempfile->Close(); |
558 |
gSystem->Unlink(tempname.str().c_str()); |
if ( myfold ) gSystem->Unlink(tempname.str().c_str()); |
559 |
// |
// |
560 |
if ( runinfo ) runinfo->Close(); |
if ( runinfo ) runinfo->Close(); |
561 |
if ( NDtr ) NDtr->SetName("NeutronD"); |
if ( NDtr ) NDtr->SetName("NeutronD"); |
564 |
file->Write(); |
file->Write(); |
565 |
}; |
}; |
566 |
// |
// |
567 |
gSystem->Unlink(NDfolder.str().c_str()); |
if ( myfold ) gSystem->Unlink(NDfolder.str().c_str()); |
568 |
// |
// |
569 |
// the end |
// the end |
570 |
// |
// |