| 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; |