/[PAMELA software]/PadmeAmidala/src/RunGlue.cpp
ViewVC logotype

Diff of /PadmeAmidala/src/RunGlue.cpp

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

revision 1.1.1.1 by mocchiut, Tue Dec 5 12:31:00 2006 UTC revision 1.12 by mocchiut, Wed Dec 10 13:15:59 2008 UTC
# Line 11  Line 11 
11  #include <TFile.h>  #include <TFile.h>
12  #include <TTree.h>  #include <TTree.h>
13  #include <TTimeStamp.h>  #include <TTimeStamp.h>
14    #include <TTreeCloner.h>
15  #include <RunGlue.h>  #include <RunGlue.h>
16  //  //
17  ClassImp(RunGlue);  ClassImp(RunGlue);
# Line 24  RunGlue::RunGlue() { Line 25  RunGlue::RunGlue() {
25  RunGlue::RunGlue(TSQLServer *da, UInt_t ru, TString di, TString wrkdi) {  RunGlue::RunGlue(TSQLServer *da, UInt_t ru, TString di, TString wrkdi) {
26    fDBG = false;    fDBG = false;
27    li = new PamLevel2();    li = new PamLevel2();
28      //  li = 0;
29    this->Clear();    this->Clear();
30    dbc = da;    dbc = da;
31    run = ru;    run = ru;
32    dir = di;    dir = di;
33    //    //
34    TString wd = gSystem->WorkingDirectory();      wd = gSystem->WorkingDirectory();  
35    //    //
36    outdir = wrkdi;    outdir = wrkdi;
37    if ( !strcmp(outdir.Data(),"") ) outdir = wd;    if ( !strcmp(outdir.Data(),"") ) outdir = wd;
# Line 45  RunGlue::RunGlue(TSQLServer *da, UInt_t Line 47  RunGlue::RunGlue(TSQLServer *da, UInt_t
47    //    //
48  };  };
49    
   
50  void RunGlue::Clear() {  void RunGlue::Clear() {
51    fEnd = false;    fEnd = false;
52    run = 0;    run = 0;
# Line 74  void RunGlue::SetDList(TString list) { Line 75  void RunGlue::SetDList(TString list) {
75    if (list.Contains("+ALL", TString::kIgnoreCase)) RUN = true;    if (list.Contains("+ALL", TString::kIgnoreCase)) RUN = true;
76    //    //
77    fDList = list;    fDList = list;
78    li->SetWhichTrees(fDList);    //  li->SetWhichTrees(fDList);
79    //    //
80    if ( DebugMode() ) printf(" Detector list is %s \n",fDList.Data());    if ( DebugMode() ) printf(" Detector list is %s \n",fDList.Data());
81    //    //
82  };  };
83    
84  TList *RunGlue::GetRunList(){  TList *RunGlue::GetRunList(){
85    //    //  
86    lList = new TList();    lList = new TList();
87    lList->Clear();    lList->Clear();
88    TString thisrun;    TString thisrun;
# Line 115  TList *RunGlue::GetRunList(){ Line 116  TList *RunGlue::GetRunList(){
116      UInt_t lowerlimit = llim->GetSec();      UInt_t lowerlimit = llim->GetSec();
117      UInt_t upperlimit = lowerlimit + 86401;      UInt_t upperlimit = lowerlimit + 86401;
118      //      //
     stringstream myquery;  
     myquery.str("");  
     myquery << "SELECT ID FROM GL_RUN WHERE RUNHEADER_TIME>=" << (UInt_t)lowerlimit  
             << " AND RUNHEADER_TIME<" << (UInt_t)upperlimit << " ORDER BY RUNHEADER_TIME ASC;";  
     //  
     pResult = dbc->Query(myquery.str().c_str());  
     for( UInt_t r=0; r < 1000; r++){  
       Row = pResult->Next();        
       if( Row == NULL ) break;  
       if ( DebugMode() ) printf(" %u RUN %s \n",r,Row->GetField(0));  
       //  
       thisrun=dir+(TString)Row->GetField(0)+".Level2.root";        
       //  
       if ( DebugMode() ) printf(" Filename is %s \n",thisrun.Data());  
       //  
       su = TFile::Open(thisrun);  
       if ( li->CheckLevel2File(thisrun) ){  
         lList->Add(su);  
       } else {  
         if ( DebugMode() ) printf(" RUN %s DISCARDED \n",Row->GetField(0));  
       };  
       //  
     };  
     //  
119      if ( DebugMode() ) printf(" YY %u MM %u DD %u ll %u ul %u \n",YY,MM,DD,lowerlimit,upperlimit);      if ( DebugMode() ) printf(" YY %u MM %u DD %u ll %u ul %u \n",YY,MM,DD,lowerlimit,upperlimit);
120      //      //
121      YY -= 2000;      YY -= 2000;
# Line 172  TList *RunGlue::GetRunList(){ Line 149  TList *RunGlue::GetRunList(){
149      //      //
150      printf(" Output filename is %s \n",fFilename.Data());      printf(" Output filename is %s \n",fFilename.Data());
151      //      //
152        if ( !this->OpenFile() ){
153             fEnd = true;
154             return(0);
155        };
156        //
157        stringstream myquery;
158        myquery.str("");
159        myquery << "SELECT ID FROM GL_RUN WHERE RUNHEADER_TIME>=" << (UInt_t)lowerlimit
160                << " AND RUNHEADER_TIME<" << (UInt_t)upperlimit << " ORDER BY RUNHEADER_TIME ASC;";
161        //
162        pResult = dbc->Query(myquery.str().c_str());
163        for( UInt_t r=0; r < 1000; r++){
164          Row = pResult->Next();      
165          if( Row == NULL ) break;
166          if ( DebugMode() ) printf(" %u RUN %s \n",r,Row->GetField(0));
167          //
168          thisrun=dir+(TString)Row->GetField(0)+".Level2.root";      
169          //
170          if ( DebugMode() ) printf(" Filename is %s \n",thisrun.Data());
171          //
172          su = TFile::Open(thisrun);
173          if ( li->CheckLevel2File(thisrun) ){
174            lList->Add(su);
175          } else {
176                    if ( DebugMode() ) printf(" RUN %s DISCARDED \n",Row->GetField(0));
177          };
178          //
179        };
180        //
181        //
182      delete glrun;      delete glrun;
183      delete dbtime;      delete dbtime;
184      //      //
# Line 276  TList *RunGlue::GetRunList(){ Line 283  TList *RunGlue::GetRunList(){
283      UInt_t lowerlimit = llim->GetSec();      UInt_t lowerlimit = llim->GetSec();
284      UInt_t upperlimit = lowerlimit + 86401;      UInt_t upperlimit = lowerlimit + 86401;
285      //      //
     stringstream myquery;  
     myquery.str("");  
     myquery << "SELECT ID FROM GL_RUN WHERE RUNHEADER_TIME>=" << (UInt_t)lowerlimit  
             << " AND RUNHEADER_TIME<" << (UInt_t)upperlimit << " ORDER BY RUNHEADER_TIME ASC;";  
     //  
     pResult = dbc->Query(myquery.str().c_str());  
     for( UInt_t r=0; r < 1000; r++){  
       Row = pResult->Next();        
       if( Row == NULL ) break;  
       if ( DebugMode() ) printf(" %u RUN %s \n",r,Row->GetField(0));  
       //  
       thisrun=dir+(TString)Row->GetField(0)+".Level2.root";        
       //  
       if ( DebugMode() ) printf(" Filename is %s \n",thisrun.Data());  
       //  
       su = TFile::Open(thisrun);  
       if ( su ){  
         TFile *su0 = TFile::Open(thisrun);  
         fDoneList->Add(su0);  
         fNlistdone++;  
       };  
       if ( li->CheckLevel2File(thisrun) ){  
         lList->Add(su);  
         if ( DebugMode() ) printf(" RUN %s ADDED \n",Row->GetField(0));  
       } else {  
         if ( DebugMode() ) printf(" RUN %s DISCARDED \n",Row->GetField(0));  
       };  
       //  
     };  
     //  
286      if ( DebugMode() ) printf(" YY %u MM %u DD %u ll %u ul %u \n",YY,MM,DD,lowerlimit,upperlimit);      if ( DebugMode() ) printf(" YY %u MM %u DD %u ll %u ul %u \n",YY,MM,DD,lowerlimit,upperlimit);
287      //      //
288      YY -= 2000;      YY -= 2000;
# Line 339  TList *RunGlue::GetRunList(){ Line 316  TList *RunGlue::GetRunList(){
316      //      //
317      printf(" Output filename is %s \n",fFilename.Data());      printf(" Output filename is %s \n",fFilename.Data());
318      //      //
319        if ( !this->OpenFile() ) return(0);
320        //
321        stringstream myquery;
322        myquery.str("");
323        myquery << "SELECT ID FROM GL_RUN WHERE RUNHEADER_TIME>=" << (UInt_t)lowerlimit
324                << " AND RUNHEADER_TIME<" << (UInt_t)upperlimit << " ORDER BY RUNHEADER_TIME ASC;";
325        //
326        pResult = dbc->Query(myquery.str().c_str());
327        for( UInt_t r=0; r < 1000; r++){
328          Row = pResult->Next();      
329          if( Row == NULL ) break;
330          if ( DebugMode() ) printf(" %u RUN %s \n",r,Row->GetField(0));
331          //
332          thisrun=dir+(TString)Row->GetField(0)+".Level2.root";      
333          //
334          if ( DebugMode() ) printf(" Filename is %s \n",thisrun.Data());
335          //
336          su = TFile::Open(thisrun);
337          if ( su ){
338            TFile *su0 = TFile::Open(thisrun);
339            fDoneList->Add(su0);
340            fNlistdone++;
341          };
342          if ( li->CheckLevel2File(thisrun) ){
343            lList->Add(su);
344            if ( DebugMode() ) printf(" RUN %s ADDED \n",Row->GetField(0));
345          } else {
346            if ( DebugMode() ) printf(" RUN %s DISCARDED \n",Row->GetField(0));
347          };
348          //
349        };
350        //
351      if ( DebugMode() ){      if ( DebugMode() ){
352        UInt_t ll = 0;        UInt_t ll = 0;
353        while ( (TFile*)lList->At(ll) ){        while ( (TFile*)lList->At(ll) ){
# Line 371  TList *RunGlue::GetRunList(){ Line 380  TList *RunGlue::GetRunList(){
380    
381  Bool_t RunGlue::OpenFile(){  Bool_t RunGlue::OpenFile(){
382    //    //
383      fOpen = false;
384    printf(" Check if output file already exists \n");    printf(" Check if output file already exists \n");
385    Target = TFile::Open((this->GetFilename()).Data(), "READ" );    ifstream myfile;
386    //    myfile.open((this->GetFilename()).Data());
387    if ( Target ){    if ( myfile ){
388      Target->Close();      //  Target = TFile::Open((this->GetFilename()).Data(), "READ" );
389        //
390        //  if ( Target ){
391        //    Target->Close();
392        myfile.close();
393      printf("Error opening target file, %s  already exist!\n",(this->GetFilename()).Data());      printf("Error opening target file, %s  already exist!\n",(this->GetFilename()).Data());
394      return(false);      return(false);
395    } else {    } else {
396      //      //
397      printf(" Output file does not exist, creating it\n");      printf(" Output file does not exist, creating it\n");
398      //      //
399      Long64_t maxsize = 10000000000LL;        Long64_t maxsize = 99900000000LL;  
400      //      //
401      Target = TFile::Open((this->GetFilename()).Data(), "RECREATE" );      Target = TFile::Open((this->GetFilename()).Data(), "RECREATE" );
402      fastMethod = kTRUE;      //fastMethod = kTRUE;
403      //fastMethod = kFALSE;      fastMethod = kFALSE;
404      //          //    
405      //          //    
406      if ( !Target || Target->IsZombie()) {      if ( !Target || Target->IsZombie()) {
# Line 395  Bool_t RunGlue::OpenFile(){ Line 409  Bool_t RunGlue::OpenFile(){
409      }      }
410      //      //
411      TTree::SetMaxTreeSize(maxsize);      TTree::SetMaxTreeSize(maxsize);
412        Target->SetCompressionLevel(2);
413        //Target->SetCompressionLevel(5);
414      //      //
415        fOpen = true;
416      return(true);      return(true);
417      //      //
418    };    };
# Line 434  void RunGlue::UpdateDB(TList *dlist){ Line 451  void RunGlue::UpdateDB(TList *dlist){
451    //    //
452    UInt_t ll = 0;    UInt_t ll = 0;
453    UInt_t idl2 = 0;    UInt_t idl2 = 0;
454      UInt_t idr = 0;
455      //
456      //
457      myquery.str("");
458      myquery << "select ID from GL_RAW where NAME='level2 files';";
459      if ( DebugMode() ) printf(" query is %s \n",myquery.str().c_str());
460    //    //
461      pResult = dbc->Query(myquery.str().c_str());
462      if ( !pResult ){
463        printf(" ERROR QUERYING ON DB!\n");
464        return;
465      };
466      Row = pResult->Next();
467      if( Row == NULL ){
468        printf(" ERROR QUERYING THE DB!\n");
469        return;
470      } else {
471        idr = (UInt_t)atoll(Row->GetField(0));
472      };
473    //    //
474    myquery.str("");    myquery.str("");
475    myquery << "insert into GL_ROOT (ID_RAW,PATH,NAME) values (4294967295,'" << outdir.Data() << "','" << ((TString)gSystem->BaseName(this->GetFilename())).Data() << "');";    //  myquery << "insert into GL_ROOT (ID_RAW,PATH,NAME) values (" << idr << ",'" << outdir.Data() << "','" << ((TString)gSystem->BaseName(this->GetFilename())).Data() << "');";
476      myquery << "insert into GL_ROOT (ID_RAW,PATH,NAME) values (" << idr << ",'$PAM_L2','" << ((TString)gSystem->BaseName(this->GetFilename())).Data() << "');";
477    if ( DebugMode() ) printf(" query is %s \n",myquery.str().c_str());    if ( DebugMode() ) printf(" query is %s \n",myquery.str().c_str());
478    //    //
479    pResult = dbc->Query(myquery.str().c_str());    pResult = dbc->Query(myquery.str().c_str());
# Line 493  void RunGlue::UpdateDB(TList *dlist){ Line 529  void RunGlue::UpdateDB(TList *dlist){
529  //  //
530  void RunGlue::MergeRootfile(TList *sourcelist) {  void RunGlue::MergeRootfile(TList *sourcelist) {
531    //    //
532    merged = false;    merged = true;
   //  
   TDirectory *target = Target;  
533    //    //
534    if ( DebugMode() ) printf("\nTarget path is: %s \n",target->GetPath());    if ( li ) li->Delete();
535      li = new PamLevel2(wd,sourcelist,fDList);
536    //    //
537    TString path( (char*)strstr( target->GetPath(), ":" ) );    Target->cd();
   path.Remove(0,2);  
   //  
   TDirectory *first_source = (TDirectory*)sourcelist->First();  
   THashList allNames;  
   //  
   while ( first_source ){  
     //  
     TDirectory *current_sourcedir = first_source->GetDirectory(path);  
     //  
     if ( !current_sourcedir ){  
       first_source = (TDirectory*)sourcelist->After(first_source);  
       continue;  
     }  
     //  
     // loop over all keys in this directory  
     //  
     TChain *globChain = 0;  
     TIter nextkey( current_sourcedir->GetListOfKeys() );  
     TKey *key, *oldkey=0;  
     TH1::AddDirectory(kFALSE); // gain time, do not add the objects in the list in memory  
     //  
     while ( (key = (TKey*)nextkey()) ) {  
       //  
       if ( current_sourcedir == target ) break;  
       //  
       if (oldkey && !strcmp(oldkey->GetName(),key->GetName())) continue; //keep only the highest cycle number for each key  
       //  
       if (allNames.FindObject(key->GetName())) continue;  
       //  
       if ( DebugMode() ) printf(" Key name is -%s- \n",key->GetName());  
       //  
       //  
       //  
       if ( !strcmp(key->GetName(),"Tracker") && !li->TRK2 && !li->TRK1 && !li->TRKh ) continue;  
       //  
       if ( !strcmp(key->GetName(),"Calorimeter") && !li->CAL2 && !li->CAL1 ) continue;  
       //  
       if ( !strcmp(key->GetName(),"ToF") && !li->TOF ) continue;  
       //  
       if ( !strcmp(key->GetName(),"Trigger") && !li->TRG ) continue;  
       //  
       if ( !strcmp(key->GetName(),"Anticounter") && !li->AC ) continue;  
       //  
       if ( !strcmp(key->GetName(),"S4") && !li->S4 ) continue;  
       //  
       if ( !strcmp(key->GetName(),"NeutronD") && !li->ND ) continue;  
       //  
       if ( !strcmp(key->GetName(),"OrbitalInfo") && !li->ORB ) continue;  
       //  
       if ( !strcmp(key->GetName(),"Run") && !RUN ) continue;  
       //  
       if ( strcmp(key->GetName(),"Calorimeter") && strcmp(key->GetName(),"Tracker") && strcmp(key->GetName(),"ToF") && strcmp(key->GetName(),"Trigger") && strcmp(key->GetName(),"Anticounter") && strcmp(key->GetName(),"S4") && strcmp(key->GetName(),"NeutronD") && strcmp(key->GetName(),"OrbitalInfo") && strcmp(key->GetName(),"Run") && strcmp(key->GetName(),"ProcessID0") ){  
         if ( DebugMode() ) printf(" ERROR UNKNOWN KEY %s !\n",key->GetName());  
         continue;  
       };  
       //  
       allNames.Add(new TObjString(key->GetName()));  
       //  
       // read object from first source file  
       //  
       current_sourcedir->cd();  
       TObject *obj = key->ReadObj();  
       //  
       if ( obj->IsA()->InheritsFrom("TTree") ){  
         //  
         // loop over all source files create a chain of Trees "globChain"  
         //  
         TString obj_name;  
         //  
         if ( path.Length() ) {  
           obj_name = path + "/" + obj->GetName();  
         } else {  
           obj_name = obj->GetName();  
         };  
         //  
         globChain = new TChain(obj_name);  
         //  
         globChain->Add(first_source->GetName());  
         //  
         TFile *nextsource = (TFile*)sourcelist->After( first_source );  
         //  
         while ( nextsource ) {              
           //  
           //do not add to the list a file that does not contain this Tree  
           //  
           TFile *curf = TFile::Open(nextsource->GetName());  
           //  
           if ( curf ) {  
             //  
             Bool_t mustAdd = kFALSE;  
             //  
             if (curf->FindKey(obj_name)) {  
               //  
               mustAdd = kTRUE;  
               //  
             } else {  
               //  
               //we could be more clever here. No need to import the object  
               //we are missing a function in TDirectory  
               //  
               TObject *aobj = curf->Get(obj_name);  
               //  
               if ( aobj ){  
                 mustAdd = kTRUE;  
                 delete aobj;  
               };  
             };  
             if (mustAdd) {  
               globChain->Add(nextsource->GetName());  
             };  
           };  
           delete curf;  
           nextsource = (TFile*)sourcelist->After(nextsource);  
         };  
         //  
       } else {  
         //  
         // object is of no type that we know or can handle  
         //  
         if ( DebugMode() ) cout << "Unknown object type, name: "  
              << obj->GetName() << " title: " << obj->GetTitle() << endl;  
       };  
       //  
       // now write the merged histogram (which is "in" obj) to the target file  
       // note that this will just store obj in the current directory level,  
       // which is not persistent until the complete directory itself is stored  
       // by "target->Write()" below  
       //  
       if ( obj ){  
         //  
         target->cd();  
         //  
         if( obj->IsA()->InheritsFrom("TTree") ) {  
           //  
           //  
           if ( fastMethod ){  
             globChain->Merge(target->GetFile(),0,"keep fast");  
           } else {  
             globChain->Merge(target->GetFile(),0,"keep");  
           };  
           //  
           merged = true;  
           //  
           if ( DebugMode() ) globChain->ls();  
           //  
           delete globChain;  
           //  
         } else {  
           //  
           //      obj->Write( key->GetName() ); // <==================================================================================  
           //  
         };  
       };  
       oldkey = key;  
     };  
     //  
     first_source = (TDirectory*)sourcelist->After(first_source);  
     //  
   };  
   // save modifications to target file  
   target->SaveSelf(kTRUE);  
538    //    //
539      li->CreateCloneTrees(Target);
540      ULong64_t nevents = li->GetEntries();
541      printf(" NEVENTS %llu \n",nevents);
542      for(ULong64_t iev=0; iev<nevents; iev++){
543        li->Clear();
544        if( li->GetEntry(iev) ){
545          li->FillCloneTrees();
546        };
547      };
548      Target->cd();
549      li->WriteCloneTrees();
550      printf(" file %s \n",Target->GetName());
551      //  Target->Write();
552      TTree *slist = (TTree*)Target->Get("SelectionList");
553      slist->Delete("all");
554      Target->Close();
555    
556    //   //
557    //   TDirectory *target = Target;
558    //   //
559    //   if ( DebugMode() ) printf("\nTarget path is: %s \n",target->GetPath());
560    //   //
561    //   TString path( (char*)strstr( target->GetPath(), ":" ) );
562    //   path.Remove(0,2);
563    //   //
564    //   TDirectory *first_source = (TDirectory*)sourcelist->First();
565    //   THashList allNames;
566    //   //
567    //   while ( first_source ){
568    //     //
569    //     TDirectory *current_sourcedir = first_source->GetDirectory(path);
570    //     //
571    //     if ( !current_sourcedir ){
572    //       first_source = (TDirectory*)sourcelist->After(first_source);
573    //       continue;
574    //     }
575    //     //
576    //     // loop over all keys in this directory
577    //     //
578    //     TChain *globChain = 0;    
579    //     TIter nextkey( current_sourcedir->GetListOfKeys() );
580    //     TKey *key = 0;
581    //     TKey *oldkey = 0;
582    //     TH1::AddDirectory(kFALSE); // gain time, do not add the objects in the list in memory
583    //     //
584    //     while ( (key = (TKey*)nextkey()) ) {
585    //       //
586    //       //      printf(" target ls \n");
587    //       //      Target->ls();
588    //       //
589    //       if ( current_sourcedir == target ) break;
590    //       //
591    //       if ( oldkey && !strcmp(oldkey->GetName(),key->GetName()) ) continue; //keep only the highest cycle number for each key
592    //       //
593    //       if ( allNames.FindObject(key->GetName()) ) continue;
594    //       //
595    //       if ( DebugMode() ) printf(" Key name is -%s- \n",key->GetName());
596    //       //
597    //       //
598    //       //
599    //       if ( !strcmp(key->GetName(),"Tracker") && !li->IsTRK2() && !li->IsTRK1() && !li->IsTRKh() ) continue;
600    //       //
601    //       if ( !strcmp(key->GetName(),"Calorimeter") && !li->IsCAL2() && !li->IsCAL1() ) continue;
602    //       //
603    //       if ( !strcmp(key->GetName(),"ToF") && !li->IsTOF() ) continue;
604    //       //
605    //       if ( !strcmp(key->GetName(),"Trigger") && !li->IsTRG() ) continue;
606    //       //
607    //       if ( !strcmp(key->GetName(),"Anticounter") && !li->IsAC() ) continue;
608    //       //
609    //       if ( !strcmp(key->GetName(),"S4") && !li->IsS4() ) continue;
610    //       //
611    //       if ( !strcmp(key->GetName(),"NeutronD") && !li->IsND() ) continue;
612    //       //
613    //       if ( !strcmp(key->GetName(),"OrbitalInfo") && !li->IsORB() ) continue;
614    //       //
615    //       if ( !strcmp(key->GetName(),"Run") && !li->IsRUN() ) continue;
616    //       //
617    //       if ( strcmp(key->GetName(),"Calorimeter") && strcmp(key->GetName(),"Tracker") && strcmp(key->GetName(),"ToF") && strcmp(key->GetName(),"Trigger") && strcmp(key->GetName(),"Anticounter") && strcmp(key->GetName(),"S4") && strcmp(key->GetName(),"NeutronD") && strcmp(key->GetName(),"OrbitalInfo") && strcmp(key->GetName(),"Run") && strcmp(key->GetName(),"ProcessID0") ){
618    //      if ( DebugMode() ) printf(" ERROR UNKNOWN KEY %s !\n",key->GetName());
619    //      continue;
620    //       };
621    //       //
622    //       printf(" 1 \n");
623    //       allNames.Add(new TObjString(key->GetName()));
624    //       printf(" 2 \n");
625    //       //
626    //       // read object from first source file
627    //       //
628    //       current_sourcedir->cd();
629    //       TObject *obj = key->ReadObj();
630    //       printf(" 3 \n");
631    //       //
632    //       if ( obj->IsA()->InheritsFrom("TTree") ){
633    //      //
634    //              // loop over all source files create a chain of Trees "globChain"
635    //      //
636    //      TString obj_name;
637    //      printf(" 4 \n");
638    //      //
639    //      if ( path.Length() ) {
640    //        obj_name = path + "/" + obj->GetName();
641    //      } else {
642    //        obj_name = obj->GetName();
643    //      };
644    //      //
645    //      globChain = new TChain(obj_name);
646    //      //
647    //      globChain->SetCacheSize(0);
648    //      //
649    //      printf(" 5 \n");
650    //      globChain->Add(first_source->GetName());
651    //      printf(" 6 \n");
652    //      //
653    //      TFile *nextsource = (TFile*)sourcelist->After( first_source );
654    //      //
655    //      while ( nextsource ) {            
656    //        //
657    //        //do not add to the list a file that does not contain this Tree
658    //        //
659    //        printf(" 7 \n");
660    //        TFile *curf = TFile::Open(nextsource->GetName());
661    //        //
662    //        if ( curf ) {
663    //          //
664    //          Bool_t mustAdd = kFALSE;
665    //          //
666    //          if (curf->FindKey(obj_name)) {
667    //            //
668    //            mustAdd = kTRUE;
669    //            //
670    //          } else {
671    //            //
672    //            //we could be more clever here. No need to import the object
673    //            //we are missing a function in TDirectory
674    //            //
675    //            TObject *aobj = curf->Get(obj_name);
676    //            //
677    //            if ( aobj ){
678    //              mustAdd = kTRUE;
679    //              delete aobj;
680    //            };
681    //          };
682    //          if (mustAdd) {
683    //            printf(" 8 \n");
684    //            globChain->Add(nextsource->GetName());
685    //            printf(" 9 \n");
686    //          };
687    //        };
688    //        delete curf;
689    //        nextsource = (TFile*)sourcelist->After(nextsource);
690    //      };
691    //      //
692    //      printf(" 10 \n");      
693    //      delete nextsource;
694    //      printf(" 11 \n");
695    //      //
696    //       } else {
697    //      //
698    //      // object is of no type that we know or can handle
699    //      //
700    //      if ( DebugMode() ) cout << "Unknown object type, name: "
701    //           << obj->GetName() << " title: " << obj->GetTitle() << endl;
702    //       };
703    //       //
704    //       // now write the merged histogram (which is "in" obj) to the target file
705    //       // note that this will just store obj in the current directory level,
706    //       // which is not persistent until the complete directory itself is stored
707    //       // by "target->Write()" below
708    //       //
709    //       printf(" 12 \n");
710    //       if ( obj ){
711    //      //
712    //      target->cd();
713    //      //
714    //      if( obj->IsA()->InheritsFrom("TTree") ) {
715    //        //
716    //        Long64_t nfiles = 0;
717    //        if ( fastMethod ){
718    //          //      globChain->Merge(target->GetFile(),0,"C keep fast");
719    //          nfiles=this->Mergy((TChain*)globChain,target->GetFile(),0,"C keep fast");
720    //        } else {
721    //          //globChain->Merge(target->GetFile(),0,"C keep");
722    //          printf(" 13 %llu \n",globChain->GetEntries());          
723    //          nfiles=this->Mergy((TChain*)globChain,target->GetFile(),0,"C keep");
724    //          printf(" 14 \n");
725    //        };
726    //        //
727    //        merged = true;
728    //        //
729    //        if ( DebugMode() ){
730    //          printf(" Merged %i files\n",(int)nfiles);
731    //          globChain->ls();
732    //        };
733    //        //
734    //        delete globChain;
735    //        //      return;
736    //        //
737    //      } else {
738    //        //
739    //        //      obj->Write( key->GetName() ); // <==================================================================================
740    //        //
741    //      };
742    //       };
743    //       delete obj;
744    //       oldkey = key;
745    //     };
746    //     //
747    //     first_source = (TDirectory*)sourcelist->After(first_source);
748    //     //
749    //   };
750    //   // save modifications to target file
751    //   //
752    //   target->SaveSelf(kTRUE);
753    //   //
754    };
755    
756    
757    Long64_t RunGlue::Mergy(TChain *mychain, TFile* file, Int_t basketsize, Option_t* option){
758       // We must have been passed a file, we will use it
759       // later to reset the compression level of the branches.
760       if (!file) {
761          // FIXME: We need an error message here.
762         printf(" 19 \n");
763         return 0;
764       }
765       printf(" 20 \n");
766    
767       // Options
768       Bool_t fastClone = kFALSE;
769       TString opt = option;
770       opt.ToLower();
771       if (opt.Contains("fast")) {
772          fastClone = kTRUE;
773          printf(" 21 \n");
774       }
775       // The chain tree must have a list of branches
776       // because we may try to change their basket
777       // size later.
778       TObjArray* lbranches = mychain->GetListOfBranches();
779       if (!lbranches) {
780          // FIXME: We need an error message here.
781         printf(" 22 \n");
782          return 0;
783       }
784    
785       //   file->cd();
786       // The chain must have a current tree because
787       // that is the one we will clone.
788       //   if (!fTree) {
789          // -- LoadTree() has not yet been called, no current tree.
790          // FIXME: We need an error message here.
791       //      return 0;
792       //   }
793    
794       // Copy the chain's current tree without
795       // copying any entries, we will do that later.
796       printf(" 23 \n");
797       TTree* newTree = mychain->CloneTree(0);
798       if (!newTree) {
799          // FIXME: We need an error message here.
800         printf(" 24 \n");
801          return 0;
802       }
803       printf(" 25 \n");
804    
805    
806       // Strip out the (potential) directory name.
807       // FIXME: The merged chain may or may not have the
808       //        same name as the original chain.  This is
809       //        bad because the chain name determines the
810       //        names of the trees in the chain by default.
811       printf(" 26 \n");
812       newTree->SetName(gSystem->BaseName(mychain->GetName()));
813       printf(" 27 \n");
814    
815       // FIXME: Why do we do this?
816       //   newTree->SetAutoSave(-1);
817       newTree->SetAutoSave(900000000000000LL);
818    
819       printf(" 28 \n");
820       // Circularity is incompatible with merging, it may
821       // force us to throw away entries, which is not what
822       // we are supposed to do.
823       newTree->SetCircular(0);
824    
825       // Reset the compression level of the branches.
826       if (opt.Contains("c")) {
827          TBranch* branch = 0;
828          TIter nextb(newTree->GetListOfBranches());
829          while ((branch = (TBranch*) nextb())) {
830            printf(" 29 \n");
831             branch->SetCompressionLevel(file->GetCompressionLevel());
832          }
833       }
834    
835       printf(" 30 \n");
836       // Reset the basket size of the branches.
837       if (basketsize > 1000) {
838          TBranch* branch = 0;
839          TIter nextb(newTree->GetListOfBranches());
840          while ((branch = (TBranch*) nextb())) {
841            printf(" 31 \n");
842             branch->SetBasketSize(basketsize);
843          }
844       }
845    
846       printf(" 32 \n");
847       Long64_t nentries = mychain->GetEntriesFast();
848       //Long64_t nentries = mychain->GetEntries();
849    
850       // Copy the entries.
851       if (fastClone) {
852          // For each tree in the chain.
853          for (Long64_t i = 0; i < nentries; i += mychain->GetTree()->GetEntries()) {
854             if (mychain->LoadTree(i) < 0) {
855                break;
856             }
857             TTreeCloner cloner(mychain->GetTree(), newTree, option);
858             if (cloner.IsValid()) {
859                newTree->SetEntries(newTree->GetEntries() + mychain->GetTree()->GetEntries());
860                cloner.Exec();
861             } else {
862                if (mychain->GetFile()) {
863                  printf("Merge Skipped file %s\n", mychain->GetFile()->GetName());
864                   //            } else {
865                   //               Warning("Merge", "Skipped file number %d\n", fTreeNumber);
866                }
867             }
868          }
869       } else {
870         printf(" 33 %llu \n",nentries);
871         mychain->Print();
872         for (Long64_t i = 0; i < nentries; i++) {
873           printf(" i %llu \n",i);
874           // for (Long64_t i = 0; i < 1; i++) {
875             if (mychain->GetEntry(i) <= 0) {
876                break;
877             }
878             newTree->Fill();
879          }
880          printf(" 34 \n");
881       }
882    
883       // Write the new tree header.
884       printf(" 35 \n");
885       newTree->Write();
886    
887       printf(" 36 \n");
888       // Get our return value.
889       Int_t nfiles = newTree->GetFileNumber() + 1;
890    
891       // Close and delete the current file of the new tree.
892       if (!opt.Contains("keep")) {
893          // FIXME: What happens to fDirectory in newTree here?
894          delete newTree->GetCurrentFile();
895       }
896       return nfiles;
897  }  }
Legend:
Removed from v.1.1.1.1  
changed lines
  Added in v.1.12
  ViewVC Help
Powered by ViewVC 1.1.23