/[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.7 by mocchiut, Thu Nov 9 17:05:45 2006 UTC revision 1.20 by mocchiut, Mon Sep 1 16:58:42 2008 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    //    //
# Line 221  int NDCore(UInt_t run, TFile *file, TSQL Line 231  int NDCore(UInt_t run, TFile *file, TSQL
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");
# Line 238  int NDCore(UInt_t run, TFile *file, TSQL Line 252  int NDCore(UInt_t run, TFile *file, TSQL
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;
# Line 246  int NDCore(UInt_t run, TFile *file, TSQL Line 262  int NDCore(UInt_t run, TFile *file, TSQL
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
# Line 265  int NDCore(UInt_t run, TFile *file, TSQL Line 281  int NDCore(UInt_t run, TFile *file, TSQL
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");
# Line 284  int NDCore(UInt_t run, TFile *file, TSQL Line 302  int NDCore(UInt_t run, TFile *file, TSQL
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 ){
# Line 292  int NDCore(UInt_t run, TFile *file, TSQL Line 311  int NDCore(UInt_t run, TFile *file, TSQL
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 ){
# Line 360  int NDCore(UInt_t run, TFile *file, TSQL Line 380  int NDCore(UInt_t run, TFile *file, TSQL
380      //      //
381      // prepare the timesync for the db      // prepare the timesync for the db
382      //      //
383        TString host = glt->CGetHost();
384        TString user = glt->CGetUser();
385        TString psw = glt->CGetPsw();
386        TSQLServer *dbc = TSQLServer::Connect(host.Data(),user.Data(),psw.Data());    
387        if ( !dbc->IsConnected() ) throw -604;
388        stringstream myquery;
389        myquery.str("");
390        myquery << "SET time_zone='+0:00'";
391        dbc->Query(myquery.str().c_str());
392      dbtime = new GL_TIMESYNC(runinfo->ID_ROOT_L0,"ID",dbc);      dbtime = new GL_TIMESYNC(runinfo->ID_ROOT_L0,"ID",dbc);
393      //      //
394      // 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.
395      //      //
396        //    if ( !dbc->IsConnected() ) throw -604;
397      glroot->Query_GL_ROOT(runinfo->ID_ROOT_L0,dbc);      glroot->Query_GL_ROOT(runinfo->ID_ROOT_L0,dbc);
398      //      //
399      ftmpname.str("");      ftmpname.str("");
# Line 381  int NDCore(UInt_t run, TFile *file, TSQL Line 411  int NDCore(UInt_t run, TFile *file, TSQL
411        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);
412      }      }
413      //      //
414        if ( !totevent ) goto closeandexit;
415        //
416      // Open Level0 file      // Open Level0 file
417      //      //
418      l0File = new TFile(fname.Data());      l0File = new TFile(fname.Data());
# Line 435  int NDCore(UInt_t run, TFile *file, TSQL Line 467  int NDCore(UInt_t run, TFile *file, TSQL
467      //      //
468      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");
469      //      //
470        if ( dbc ){
471          dbc->Close();
472          //      delete dbc;
473        };
474        //
475      for ( re = runinfo->EV_FROM; re < (runinfo->EV_FROM+runinfo->NEVENTS); re++){      for ( re = runinfo->EV_FROM; re < (runinfo->EV_FROM+runinfo->NEVENTS); re++){
476        //        //
477        if ( procev%1000 == 0 && procev > 0 && verbose ) printf(" %iK \n",procev/1000);          if ( procev%1000 == 0 && procev > 0 && verbose ) printf(" %iK \n",procev/1000);  
# Line 528  int NDCore(UInt_t run, TFile *file, TSQL Line 565  int NDCore(UInt_t run, TFile *file, TSQL
565    //    //
566    if ( l0File ) l0File->Close();    if ( l0File ) l0File->Close();
567    if ( tempfile ) tempfile->Close();                if ( tempfile ) tempfile->Close();            
568    gSystem->Unlink(tempname.str().c_str());    if ( myfold ) gSystem->Unlink(tempname.str().c_str());
569    //    //
570    if ( runinfo ) runinfo->Close();        if ( runinfo ) runinfo->Close();    
571    if ( NDtr ) NDtr->SetName("NeutronD");        if ( NDtr ) NDtr->SetName("NeutronD");    
# Line 537  int NDCore(UInt_t run, TFile *file, TSQL Line 574  int NDCore(UInt_t run, TFile *file, TSQL
574      file->Write();      file->Write();
575    };    };
576    //    //
577    gSystem->Unlink(NDfolder.str().c_str());    if ( myfold ) gSystem->Unlink(NDfolder.str().c_str());
578    //    //
579    // the end    // the end
580    //    //
Legend:
Removed from v.1.7  
changed lines
  Added in v.1.20
  ViewVC Help
Powered by ViewVC 1.1.23