/[PAMELA software]/DarthVader/NDLevel2/src/NDCore.cpp
ViewVC logotype

Diff of /DarthVader/NDLevel2/src/NDCore.cpp

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1.8 by mocchiut, Fri Nov 17 10:08:09 2006 UTC revision 1.31 by mocchiut, Tue Oct 14 14:07:17 2014 UTC
# Line 26  Line 26 
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>
# Line 34  Line 39 
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  //  //
# Line 53  using namespace std; Line 57  using namespace std;
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;  
   //  
   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 ){
# Line 79  int NDCore(UInt_t run, TFile *file, TSQL Line 84  int NDCore(UInt_t run, TFile *file, TSQL
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;
# Line 116  int NDCore(UInt_t run, TFile *file, TSQL Line 122  int NDCore(UInt_t run, TFile *file, TSQL
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();
# Line 155  int NDCore(UInt_t run, TFile *file, TSQL Line 165  int NDCore(UInt_t run, TFile *file, TSQL
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    //    //
# Line 221  int NDCore(UInt_t run, TFile *file, TSQL Line 232  int NDCore(UInt_t run, TFile *file, TSQL
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");
# Line 238  int NDCore(UInt_t run, TFile *file, TSQL Line 253  int NDCore(UInt_t run, TFile *file, TSQL
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;
# Line 246  int NDCore(UInt_t run, TFile *file, TSQL Line 263  int NDCore(UInt_t run, TFile *file, TSQL
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
# Line 265  int NDCore(UInt_t run, TFile *file, TSQL Line 282  int NDCore(UInt_t run, TFile *file, TSQL
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");
# Line 284  int NDCore(UInt_t run, TFile *file, TSQL Line 303  int NDCore(UInt_t run, TFile *file, TSQL
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 ){
# Line 292  int NDCore(UInt_t run, TFile *file, TSQL Line 312  int NDCore(UInt_t run, TFile *file, TSQL
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 ){
# Line 302  int NDCore(UInt_t run, TFile *file, TSQL Line 323  int NDCore(UInt_t run, TFile *file, TSQL
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
# Line 360  int NDCore(UInt_t run, TFile *file, TSQL Line 381  int NDCore(UInt_t run, TFile *file, TSQL
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;      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;      //    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("");
# Line 383  int NDCore(UInt_t run, TFile *file, TSQL Line 413  int NDCore(UInt_t run, TFile *file, TSQL
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");
# Line 419  int NDCore(UInt_t run, TFile *file, TSQL Line 452  int NDCore(UInt_t run, TFile *file, TSQL
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;
# Line 437  int NDCore(UInt_t run, TFile *file, TSQL Line 470  int NDCore(UInt_t run, TFile *file, TSQL
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        //        //
# Line 450  int NDCore(UInt_t run, TFile *file, TSQL Line 489  int NDCore(UInt_t run, TFile *file, TSQL
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        };        };
# Line 461  int NDCore(UInt_t run, TFile *file, TSQL Line 500  int NDCore(UInt_t run, TFile *file, TSQL
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 && l0ne->unpackError == 0 ){        if ( tmpSize && l0ne->unpackError == 0 ){
506          for (Int_t j = 0; j < tmpSize; j++){          for (Int_t j = 0; j < tmpSize; j++){
# Line 511  int NDCore(UInt_t run, TFile *file, TSQL Line 550  int NDCore(UInt_t run, TFile *file, TSQL
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
# Line 530  int NDCore(UInt_t run, TFile *file, TSQL Line 569  int NDCore(UInt_t run, TFile *file, TSQL
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;
Legend:
Removed from v.1.8  
changed lines
  Added in v.1.31
  ViewVC Help
Powered by ViewVC 1.1.23