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

Annotation of /PadmeAmidala/src/RunGlue.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.13 - (hide annotations) (download)
Thu Dec 11 10:41:11 2008 UTC (15 years, 11 months ago) by mocchiut
Branch: MAIN
Changes since 1.12: +3 -2 lines
Changed compression level

1 mocchiut 1.1 /**
2     * \file RunGlue.cpp
3     * \author Emiliano Mocchiutti
4     *
5     * The file contains implementation of the methods to query the DB.
6     */
7     //
8 mocchiut 1.11 #include <sstream>
9     #include <iostream>
10     //
11     #include <TFile.h>
12     #include <TTree.h>
13     #include <TTimeStamp.h>
14     #include <TTreeCloner.h>
15 mocchiut 1.1 #include <RunGlue.h>
16     //
17     ClassImp(RunGlue);
18     //
19     using namespace std;
20    
21     RunGlue::RunGlue() {
22     this->Clear();
23     }
24    
25 mocchiut 1.11 RunGlue::RunGlue(TSQLServer *da, UInt_t ru, TString di, TString wrkdi) {
26 mocchiut 1.1 fDBG = false;
27     li = new PamLevel2();
28 mocchiut 1.12 // li = 0;
29 mocchiut 1.1 this->Clear();
30     dbc = da;
31     run = ru;
32     dir = di;
33     //
34 mocchiut 1.12 wd = gSystem->WorkingDirectory();
35 mocchiut 1.1 //
36     outdir = wrkdi;
37     if ( !strcmp(outdir.Data(),"") ) outdir = wd;
38     fList = 0;
39     if ( run ){
40     runmode = true;
41     } else {
42     runmode = false;
43     createlist = true;
44     };
45     //
46     // lList = new TList();
47     //
48     };
49    
50     void RunGlue::Clear() {
51     fEnd = false;
52     run = 0;
53     dir = "";
54     outdir = "";
55     fFilename = "";
56     };
57    
58     void RunGlue::Clean() {
59     if ( !merged ) gSystem->Unlink(fFilename);
60     lList->Delete();
61     delete Target;
62     };
63    
64     void RunGlue::SetDebug(Bool_t dbg) {
65     fDBG = dbg;
66     };
67    
68     void RunGlue::SetDList(TString list) {
69     //
70     RUN = true;
71     //
72     if (list.Contains("-RUN", TString::kIgnoreCase)) RUN = false;
73     if (list.Contains("-ALL", TString::kIgnoreCase)) RUN = false;
74     if (list.Contains("+RUN", TString::kIgnoreCase)) RUN = true;
75     if (list.Contains("+ALL", TString::kIgnoreCase)) RUN = true;
76     //
77     fDList = list;
78 mocchiut 1.12 // li->SetWhichTrees(fDList);
79 mocchiut 1.1 //
80     if ( DebugMode() ) printf(" Detector list is %s \n",fDList.Data());
81     //
82     };
83    
84     TList *RunGlue::GetRunList(){
85 mocchiut 1.12 //
86 mocchiut 1.1 lList = new TList();
87     lList->Clear();
88     TString thisrun;
89     TFile *su;
90     //
91     TSQLResult *pResult;
92     TSQLRow *Row;
93     //
94     if ( run && runmode ){
95     //
96     // Qurey the GL_RUN table to obtain RUN infos
97     //
98     GL_RUN *glrun = new GL_RUN();
99     glrun->Query_GL_RUN(run,dbc);
100     //
101     // convert RUNHEADER_TIME into a UTC date
102     //
103     GL_TIMESYNC *dbtime = new GL_TIMESYNC();
104     //
105     TString UTC="UTC";
106     TString rhdate=dbtime->ConvertTime(UTC,glrun->RUNHEADER_TIME);
107     //
108     TDatime ti = TDatime(rhdate.Data());
109     //
110     YY=(UInt_t)ti.GetYear();
111     MM=(UInt_t)ti.GetMonth();
112     DD=(UInt_t)ti.GetDay();
113     //
114     TTimeStamp *llim = new TTimeStamp(YY,MM,DD,0,0,0,0,true,0);
115     //
116     UInt_t lowerlimit = llim->GetSec();
117     UInt_t upperlimit = lowerlimit + 86401;
118     //
119     if ( DebugMode() ) printf(" YY %u MM %u DD %u ll %u ul %u \n",YY,MM,DD,lowerlimit,upperlimit);
120     //
121     YY -= 2000;
122     stringstream tmpf;
123     tmpf.str("");
124     tmpf << "L2PAM";
125     if ( YY < 10 ){
126     tmpf<< "0" << YY;
127     } else {
128     tmpf<< YY;
129     };
130     if ( MM < 10 ){
131     tmpf<< "0" << MM;
132     } else {
133     tmpf<< MM;
134     };
135     if ( DD < 10 ){
136     tmpf << "0" << DD << ".root";
137     } else {
138     tmpf << DD << ".root";
139     };
140     TString fpath = "";
141     if ( !strcmp(outdir,"") ){
142     fpath="./";
143     } else {
144     fpath += outdir;
145     fpath += "/";
146     };
147     //
148     fFilename = fpath + tmpf.str().c_str();
149     //
150     printf(" Output filename is %s \n",fFilename.Data());
151     //
152 mocchiut 1.4 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 mocchiut 1.12 if ( DebugMode() ) printf(" RUN %s DISCARDED \n",Row->GetField(0));
177 mocchiut 1.4 };
178     //
179     };
180     //
181     //
182 mocchiut 1.1 delete glrun;
183     delete dbtime;
184     //
185     };
186     //
187     if ( strcmp(dir.Data(),"" )){
188     //
189     if ( createlist ){
190     fList = new TList();
191     fDoneList = new TList();
192     fNlist = 0;
193     fNlistdone = 0;
194     //
195     // read files in the directory
196     //
197     if ( DebugMode() ) printf("\n No input run given\n Check for existing root files. \n");
198     //
199     TSystemDirectory *datadir = new TSystemDirectory(gSystem->BaseName(dir),dir);
200     TList *temp = datadir->GetListOfFiles();
201     //
202     TIter next(temp);
203     TSystemFile *questo = 0;
204     //
205     while ( (questo = (TSystemFile*)next()) ) {
206     TString name = questo->GetName();
207     if( name.EndsWith(".Level2.root") ){
208     char *fullpath0;
209     gSystem->IsFileInIncludePath(name,&fullpath0);
210     TString fullpath = fullpath0;
211     //
212     if ( DebugMode() ) printf(" fullpath = %s name %s \n",fullpath.Data(),name.Data());
213     //
214     //
215     su = TFile::Open(fullpath);
216     if ( li->CheckLevel2File((TString)fullpath) ){
217     fList->Add(su);
218     fNlist++;
219     } else {
220     if ( DebugMode() ) printf(" RUN %s DISCARDED \n",fullpath.Data());
221     };
222     //
223     };
224     };
225     createlist = false;
226     nrun = 0;
227     };
228     //
229     //
230     //
231     Bool_t stop = true;
232     UInt_t drun = 0;
233     while ( nrun < fNlist && stop ){
234     TString myrun = (TString)((TString)(gSystem->BaseName(((TFile*)fList->At(nrun))->GetName()))).ReplaceAll(".Level2.root",12,"",0);
235     if ( DebugMode() ) printf(" nrun %u myrun is %s \n",nrun,(myrun.ReplaceAll(".Level2.root",12,"",0)).Data());
236     run = (UInt_t)(myrun.Atoi());
237     nrun++;
238     //
239     TString donerun = "";
240     for (UInt_t ll=0; ll<fNlistdone; ll++){
241     donerun = "";
242     donerun = (TString)((TString)(gSystem->BaseName(((TFile*)fDoneList->At(ll))->GetName()))).ReplaceAll(".Level2.root",12,"",0);
243     if ( DebugMode() ) printf(" donerun is %s \n",(donerun.ReplaceAll(".Level2.root",12,"",0)).Data());
244     drun = (UInt_t)(donerun.Atoi());
245     //
246     if ( run == drun ) goto jump;
247     //
248     };
249     //
250     stop = false;
251     //
252     jump:
253     drun = 0;
254     //
255     };
256     //
257     if ( nrun >= fNlist && stop ){
258     fEnd = true;
259     return(0);
260     };
261     //
262     // Qurey the GL_RUN table to obtain RUN infos
263     //
264     printf(" PROCESSING RUN %u AND ITS FRIENDS\n",run);
265     GL_RUN *glrun = new GL_RUN();
266     glrun->Query_GL_RUN(run,dbc);
267     //
268     // convert RUNHEADER_TIME into a UTC date
269     //
270     GL_TIMESYNC *dbtime = new GL_TIMESYNC();
271     //
272     TString UTC="UTC";
273     TString rhdate=dbtime->ConvertTime(UTC,glrun->RUNHEADER_TIME);
274     //
275     TDatime ti = TDatime(rhdate.Data());
276     //
277     YY=(UInt_t)ti.GetYear();
278     MM=(UInt_t)ti.GetMonth();
279     DD=(UInt_t)ti.GetDay();
280     //
281     TTimeStamp *llim = new TTimeStamp(YY,MM,DD,0,0,0,0,true,0);
282     //
283     UInt_t lowerlimit = llim->GetSec();
284     UInt_t upperlimit = lowerlimit + 86401;
285     //
286     if ( DebugMode() ) printf(" YY %u MM %u DD %u ll %u ul %u \n",YY,MM,DD,lowerlimit,upperlimit);
287     //
288     YY -= 2000;
289     stringstream tmpf;
290     tmpf.str("");
291     tmpf << "L2PAM";
292     if ( YY < 10 ){
293     tmpf<< "0" << YY;
294     } else {
295     tmpf<< YY;
296     };
297     if ( MM < 10 ){
298     tmpf<< "0" << MM;
299     } else {
300     tmpf<< MM;
301     };
302     if ( DD < 10 ){
303     tmpf << "0" << DD << ".root";
304     } else {
305     tmpf << DD << ".root";
306     };
307     TString fpath = "";
308     if ( !strcmp(outdir,"") ){
309     fpath="./";
310     } else {
311     fpath += outdir;
312     fpath += "/";
313     };
314     //
315     fFilename = fpath + tmpf.str().c_str();
316     //
317     printf(" Output filename is %s \n",fFilename.Data());
318     //
319 mocchiut 1.4 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 mocchiut 1.1 if ( DebugMode() ){
352     UInt_t ll = 0;
353     while ( (TFile*)lList->At(ll) ){
354     TString donerun = "";
355     donerun = (TString)((TString)(gSystem->BaseName(((TFile*)lList->At(ll))->GetName()))).ReplaceAll(".Level2.root",12,"",0);
356     printf(" XXX file is %s \n",donerun.Data());
357     ll++;
358     //
359     };
360     };
361     //
362     delete glrun;
363     delete dbtime;
364     //
365     };
366     //
367     // is it the end?
368     //
369     if ( runmode ){
370     fEnd = true;
371     } else {
372     //
373     // did we use all the run in the directory? if so exit
374     //
375     if ( nrun >= fNlist ) fEnd = true;
376     };
377     //
378     return(lList);
379     };
380    
381     Bool_t RunGlue::OpenFile(){
382     //
383 mocchiut 1.11 fOpen = false;
384     printf(" Check if output file already exists \n");
385 mocchiut 1.10 ifstream myfile;
386 mocchiut 1.11 myfile.open((this->GetFilename()).Data());
387     if ( myfile ){
388 mocchiut 1.7 // Target = TFile::Open((this->GetFilename()).Data(), "READ" );
389     //
390     // if ( Target ){
391     // Target->Close();
392 mocchiut 1.11 myfile.close();
393     printf("Error opening target file, %s already exist!\n",(this->GetFilename()).Data());
394 mocchiut 1.1 return(false);
395     } else {
396     //
397     printf(" Output file does not exist, creating it\n");
398     //
399 mocchiut 1.5 Long64_t maxsize = 99900000000LL;
400 mocchiut 1.1 //
401 mocchiut 1.11 Target = TFile::Open((this->GetFilename()).Data(), "RECREATE" );
402 mocchiut 1.3 //fastMethod = kTRUE;
403     fastMethod = kFALSE;
404 mocchiut 1.1 //
405     //
406     if ( !Target || Target->IsZombie()) {
407 mocchiut 1.11 printf("Error opening target file (does %s exist?)\n",(this->GetFilename()).Data());
408 mocchiut 1.1 exit(1);
409     }
410     //
411     TTree::SetMaxTreeSize(maxsize);
412 mocchiut 1.13 // Target->SetCompressionLevel(2);
413     Target->SetCompressionLevel(3);
414 mocchiut 1.12 //Target->SetCompressionLevel(5);
415 mocchiut 1.1 //
416 mocchiut 1.4 fOpen = true;
417 mocchiut 1.1 return(true);
418     //
419     };
420     //
421     return(true);
422     //
423     };
424    
425    
426     void RunGlue::DeleteRunFiles(TList *dlist){
427     //
428     TString donerun = "";
429     //
430     printf(" Deleting runs: \n");
431     //
432     UInt_t ll = 0;
433     //
434     while ( (TFile*)dlist->At(ll) ){
435     donerun = (TString)(((TFile*)dlist->At(ll))->GetName());
436     gSystem->Unlink(donerun);
437     printf(" Deleted file is %s \n",donerun.Data());
438     ll++;
439     //
440     };
441     };
442    
443     void RunGlue::UpdateDB(TList *dlist){
444     //
445     TSQLResult *pResult;
446     TSQLRow *Row;
447     stringstream myquery;
448     //
449     TString donerun = "";
450     //
451     if ( DebugMode() ) printf(" Updating GL_RUN: \n");
452     //
453     UInt_t ll = 0;
454     UInt_t idl2 = 0;
455 mocchiut 1.2 UInt_t idr = 0;
456 mocchiut 1.1 //
457     //
458     myquery.str("");
459 mocchiut 1.2 myquery << "select ID from GL_RAW where NAME='level2 files';";
460     if ( DebugMode() ) printf(" query is %s \n",myquery.str().c_str());
461     //
462     pResult = dbc->Query(myquery.str().c_str());
463     if ( !pResult ){
464     printf(" ERROR QUERYING ON DB!\n");
465     return;
466     };
467     Row = pResult->Next();
468     if( Row == NULL ){
469     printf(" ERROR QUERYING THE DB!\n");
470     return;
471     } else {
472     idr = (UInt_t)atoll(Row->GetField(0));
473     };
474     //
475     myquery.str("");
476 mocchiut 1.9 // myquery << "insert into GL_ROOT (ID_RAW,PATH,NAME) values (" << idr << ",'" << outdir.Data() << "','" << ((TString)gSystem->BaseName(this->GetFilename())).Data() << "');";
477     myquery << "insert into GL_ROOT (ID_RAW,PATH,NAME) values (" << idr << ",'$PAM_L2','" << ((TString)gSystem->BaseName(this->GetFilename())).Data() << "');";
478 mocchiut 1.1 if ( DebugMode() ) printf(" query is %s \n",myquery.str().c_str());
479     //
480     pResult = dbc->Query(myquery.str().c_str());
481     if ( !pResult ){
482     printf(" ERROR WRITING ON DB!\n");
483     return;
484     };
485     myquery.str("");
486     myquery << "select ID from GL_ROOT where NAME='" << ((TString)gSystem->BaseName(this->GetFilename())).Data() << "';";
487     if ( DebugMode() ) printf(" query is %s \n",myquery.str().c_str());
488     //
489     pResult = dbc->Query(myquery.str().c_str());
490     if ( !pResult ){
491     printf(" ERROR WRITING ON DB!\n");
492     return;
493     };
494     Row = pResult->Next();
495     if( Row == NULL ){
496     printf(" ERROR QUERYING THE DB!\n");
497     return;
498     } else {
499     idl2 = (UInt_t)atoll(Row->GetField(0));
500     };
501     //
502     //
503     //
504     UInt_t drun = 0;
505     //
506     while ( (TFile*)dlist->At(ll) ){
507     donerun = "";
508     donerun = (TString)((TString)(gSystem->BaseName(((TFile*)dlist->At(ll))->GetName()))).ReplaceAll(".Level2.root",12,"",0);
509     drun = (UInt_t)(donerun.Atoi());
510     //
511     if ( DebugMode() ) printf(" Run is %s \n",donerun.Data());
512     //
513     myquery.str("");
514     myquery << "update GL_RUN set ID_ROOT_L2=" << idl2 << " where ID=" << drun << ";";
515     if ( DebugMode() ) printf(" query is %s \n",myquery.str().c_str());
516     //
517     pResult = dbc->Query(myquery.str().c_str());
518     if ( !pResult ){
519     printf(" ERROR WRITING ON DB!\n");
520     return;
521     };
522     //
523     ll++;
524     //
525     };
526     };
527    
528     //
529     // method stolen to ROOT's "hadd" and modified
530     //
531     void RunGlue::MergeRootfile(TList *sourcelist) {
532     //
533 mocchiut 1.12 merged = true;
534 mocchiut 1.1 //
535 mocchiut 1.12 if ( li ) li->Delete();
536     li = new PamLevel2(wd,sourcelist,fDList);
537 mocchiut 1.1 //
538 mocchiut 1.12 Target->cd();
539 mocchiut 1.1 //
540 mocchiut 1.12 li->CreateCloneTrees(Target);
541     ULong64_t nevents = li->GetEntries();
542     printf(" NEVENTS %llu \n",nevents);
543     for(ULong64_t iev=0; iev<nevents; iev++){
544     li->Clear();
545     if( li->GetEntry(iev) ){
546     li->FillCloneTrees();
547     };
548     };
549     Target->cd();
550     li->WriteCloneTrees();
551 mocchiut 1.13 printf("Written file %s \n",Target->GetName());
552 mocchiut 1.12 // Target->Write();
553     TTree *slist = (TTree*)Target->Get("SelectionList");
554     slist->Delete("all");
555     Target->Close();
556    
557     // //
558     // TDirectory *target = Target;
559     // //
560     // if ( DebugMode() ) printf("\nTarget path is: %s \n",target->GetPath());
561     // //
562     // TString path( (char*)strstr( target->GetPath(), ":" ) );
563     // path.Remove(0,2);
564     // //
565     // TDirectory *first_source = (TDirectory*)sourcelist->First();
566     // THashList allNames;
567     // //
568     // while ( first_source ){
569     // //
570     // TDirectory *current_sourcedir = first_source->GetDirectory(path);
571     // //
572     // if ( !current_sourcedir ){
573     // first_source = (TDirectory*)sourcelist->After(first_source);
574     // continue;
575     // }
576     // //
577     // // loop over all keys in this directory
578     // //
579     // TChain *globChain = 0;
580     // TIter nextkey( current_sourcedir->GetListOfKeys() );
581     // TKey *key = 0;
582     // TKey *oldkey = 0;
583     // TH1::AddDirectory(kFALSE); // gain time, do not add the objects in the list in memory
584     // //
585     // while ( (key = (TKey*)nextkey()) ) {
586     // //
587     // // printf(" target ls \n");
588     // // Target->ls();
589     // //
590     // if ( current_sourcedir == target ) break;
591     // //
592     // if ( oldkey && !strcmp(oldkey->GetName(),key->GetName()) ) continue; //keep only the highest cycle number for each key
593     // //
594     // if ( allNames.FindObject(key->GetName()) ) continue;
595     // //
596     // if ( DebugMode() ) printf(" Key name is -%s- \n",key->GetName());
597     // //
598     // //
599     // //
600     // if ( !strcmp(key->GetName(),"Tracker") && !li->IsTRK2() && !li->IsTRK1() && !li->IsTRKh() ) continue;
601     // //
602     // if ( !strcmp(key->GetName(),"Calorimeter") && !li->IsCAL2() && !li->IsCAL1() ) continue;
603     // //
604     // if ( !strcmp(key->GetName(),"ToF") && !li->IsTOF() ) continue;
605     // //
606     // if ( !strcmp(key->GetName(),"Trigger") && !li->IsTRG() ) continue;
607     // //
608     // if ( !strcmp(key->GetName(),"Anticounter") && !li->IsAC() ) continue;
609     // //
610     // if ( !strcmp(key->GetName(),"S4") && !li->IsS4() ) continue;
611     // //
612     // if ( !strcmp(key->GetName(),"NeutronD") && !li->IsND() ) continue;
613     // //
614     // if ( !strcmp(key->GetName(),"OrbitalInfo") && !li->IsORB() ) continue;
615     // //
616     // if ( !strcmp(key->GetName(),"Run") && !li->IsRUN() ) continue;
617     // //
618     // 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") ){
619     // if ( DebugMode() ) printf(" ERROR UNKNOWN KEY %s !\n",key->GetName());
620     // continue;
621     // };
622     // //
623     // printf(" 1 \n");
624     // allNames.Add(new TObjString(key->GetName()));
625     // printf(" 2 \n");
626     // //
627     // // read object from first source file
628     // //
629     // current_sourcedir->cd();
630     // TObject *obj = key->ReadObj();
631     // printf(" 3 \n");
632     // //
633     // if ( obj->IsA()->InheritsFrom("TTree") ){
634     // //
635     // // loop over all source files create a chain of Trees "globChain"
636     // //
637     // TString obj_name;
638     // printf(" 4 \n");
639     // //
640     // if ( path.Length() ) {
641     // obj_name = path + "/" + obj->GetName();
642     // } else {
643     // obj_name = obj->GetName();
644     // };
645     // //
646     // globChain = new TChain(obj_name);
647     // //
648     // globChain->SetCacheSize(0);
649     // //
650     // printf(" 5 \n");
651     // globChain->Add(first_source->GetName());
652     // printf(" 6 \n");
653     // //
654     // TFile *nextsource = (TFile*)sourcelist->After( first_source );
655     // //
656     // while ( nextsource ) {
657     // //
658     // //do not add to the list a file that does not contain this Tree
659     // //
660     // printf(" 7 \n");
661     // TFile *curf = TFile::Open(nextsource->GetName());
662     // //
663     // if ( curf ) {
664     // //
665     // Bool_t mustAdd = kFALSE;
666     // //
667     // if (curf->FindKey(obj_name)) {
668     // //
669     // mustAdd = kTRUE;
670     // //
671     // } else {
672     // //
673     // //we could be more clever here. No need to import the object
674     // //we are missing a function in TDirectory
675     // //
676     // TObject *aobj = curf->Get(obj_name);
677     // //
678     // if ( aobj ){
679     // mustAdd = kTRUE;
680     // delete aobj;
681     // };
682     // };
683     // if (mustAdd) {
684     // printf(" 8 \n");
685     // globChain->Add(nextsource->GetName());
686     // printf(" 9 \n");
687     // };
688     // };
689     // delete curf;
690     // nextsource = (TFile*)sourcelist->After(nextsource);
691     // };
692     // //
693     // printf(" 10 \n");
694     // delete nextsource;
695     // printf(" 11 \n");
696     // //
697     // } else {
698     // //
699     // // object is of no type that we know or can handle
700     // //
701     // if ( DebugMode() ) cout << "Unknown object type, name: "
702     // << obj->GetName() << " title: " << obj->GetTitle() << endl;
703     // };
704     // //
705     // // now write the merged histogram (which is "in" obj) to the target file
706     // // note that this will just store obj in the current directory level,
707     // // which is not persistent until the complete directory itself is stored
708     // // by "target->Write()" below
709     // //
710     // printf(" 12 \n");
711     // if ( obj ){
712     // //
713     // target->cd();
714     // //
715     // if( obj->IsA()->InheritsFrom("TTree") ) {
716     // //
717     // Long64_t nfiles = 0;
718     // if ( fastMethod ){
719     // // globChain->Merge(target->GetFile(),0,"C keep fast");
720     // nfiles=this->Mergy((TChain*)globChain,target->GetFile(),0,"C keep fast");
721     // } else {
722     // //globChain->Merge(target->GetFile(),0,"C keep");
723     // printf(" 13 %llu \n",globChain->GetEntries());
724     // nfiles=this->Mergy((TChain*)globChain,target->GetFile(),0,"C keep");
725     // printf(" 14 \n");
726     // };
727     // //
728     // merged = true;
729     // //
730     // if ( DebugMode() ){
731     // printf(" Merged %i files\n",(int)nfiles);
732     // globChain->ls();
733     // };
734     // //
735     // delete globChain;
736     // // return;
737     // //
738     // } else {
739     // //
740     // // obj->Write( key->GetName() ); // <==================================================================================
741     // //
742     // };
743     // };
744     // delete obj;
745     // oldkey = key;
746     // };
747     // //
748     // first_source = (TDirectory*)sourcelist->After(first_source);
749     // //
750     // };
751     // // save modifications to target file
752     // //
753     // target->SaveSelf(kTRUE);
754     // //
755 mocchiut 1.5 };
756    
757    
758     Long64_t RunGlue::Mergy(TChain *mychain, TFile* file, Int_t basketsize, Option_t* option){
759     // We must have been passed a file, we will use it
760     // later to reset the compression level of the branches.
761     if (!file) {
762     // FIXME: We need an error message here.
763 mocchiut 1.12 printf(" 19 \n");
764 mocchiut 1.5 return 0;
765     }
766 mocchiut 1.12 printf(" 20 \n");
767 mocchiut 1.5
768     // Options
769     Bool_t fastClone = kFALSE;
770     TString opt = option;
771     opt.ToLower();
772     if (opt.Contains("fast")) {
773     fastClone = kTRUE;
774 mocchiut 1.12 printf(" 21 \n");
775 mocchiut 1.5 }
776     // The chain tree must have a list of branches
777     // because we may try to change their basket
778     // size later.
779     TObjArray* lbranches = mychain->GetListOfBranches();
780     if (!lbranches) {
781     // FIXME: We need an error message here.
782 mocchiut 1.12 printf(" 22 \n");
783 mocchiut 1.5 return 0;
784     }
785    
786 mocchiut 1.12 // file->cd();
787 mocchiut 1.5 // The chain must have a current tree because
788     // that is the one we will clone.
789     // if (!fTree) {
790     // -- LoadTree() has not yet been called, no current tree.
791     // FIXME: We need an error message here.
792     // return 0;
793     // }
794    
795     // Copy the chain's current tree without
796     // copying any entries, we will do that later.
797 mocchiut 1.12 printf(" 23 \n");
798 mocchiut 1.5 TTree* newTree = mychain->CloneTree(0);
799     if (!newTree) {
800     // FIXME: We need an error message here.
801 mocchiut 1.12 printf(" 24 \n");
802 mocchiut 1.5 return 0;
803     }
804 mocchiut 1.12 printf(" 25 \n");
805    
806 mocchiut 1.5
807     // Strip out the (potential) directory name.
808     // FIXME: The merged chain may or may not have the
809     // same name as the original chain. This is
810     // bad because the chain name determines the
811     // names of the trees in the chain by default.
812 mocchiut 1.12 printf(" 26 \n");
813 mocchiut 1.5 newTree->SetName(gSystem->BaseName(mychain->GetName()));
814 mocchiut 1.12 printf(" 27 \n");
815 mocchiut 1.5
816     // FIXME: Why do we do this?
817     // newTree->SetAutoSave(-1);
818     newTree->SetAutoSave(900000000000000LL);
819    
820 mocchiut 1.12 printf(" 28 \n");
821 mocchiut 1.5 // Circularity is incompatible with merging, it may
822     // force us to throw away entries, which is not what
823     // we are supposed to do.
824     newTree->SetCircular(0);
825    
826     // Reset the compression level of the branches.
827     if (opt.Contains("c")) {
828     TBranch* branch = 0;
829     TIter nextb(newTree->GetListOfBranches());
830     while ((branch = (TBranch*) nextb())) {
831 mocchiut 1.12 printf(" 29 \n");
832 mocchiut 1.5 branch->SetCompressionLevel(file->GetCompressionLevel());
833     }
834     }
835    
836 mocchiut 1.12 printf(" 30 \n");
837 mocchiut 1.5 // Reset the basket size of the branches.
838     if (basketsize > 1000) {
839     TBranch* branch = 0;
840     TIter nextb(newTree->GetListOfBranches());
841     while ((branch = (TBranch*) nextb())) {
842 mocchiut 1.12 printf(" 31 \n");
843 mocchiut 1.5 branch->SetBasketSize(basketsize);
844     }
845     }
846    
847 mocchiut 1.12 printf(" 32 \n");
848 mocchiut 1.5 Long64_t nentries = mychain->GetEntriesFast();
849 mocchiut 1.12 //Long64_t nentries = mychain->GetEntries();
850 mocchiut 1.5
851     // Copy the entries.
852     if (fastClone) {
853     // For each tree in the chain.
854     for (Long64_t i = 0; i < nentries; i += mychain->GetTree()->GetEntries()) {
855     if (mychain->LoadTree(i) < 0) {
856     break;
857     }
858     TTreeCloner cloner(mychain->GetTree(), newTree, option);
859     if (cloner.IsValid()) {
860     newTree->SetEntries(newTree->GetEntries() + mychain->GetTree()->GetEntries());
861     cloner.Exec();
862     } else {
863     if (mychain->GetFile()) {
864     printf("Merge Skipped file %s\n", mychain->GetFile()->GetName());
865     // } else {
866     // Warning("Merge", "Skipped file number %d\n", fTreeNumber);
867     }
868     }
869     }
870     } else {
871 mocchiut 1.12 printf(" 33 %llu \n",nentries);
872     mychain->Print();
873     for (Long64_t i = 0; i < nentries; i++) {
874     printf(" i %llu \n",i);
875     // for (Long64_t i = 0; i < 1; i++) {
876 mocchiut 1.5 if (mychain->GetEntry(i) <= 0) {
877     break;
878     }
879     newTree->Fill();
880     }
881 mocchiut 1.12 printf(" 34 \n");
882 mocchiut 1.5 }
883    
884     // Write the new tree header.
885 mocchiut 1.12 printf(" 35 \n");
886 mocchiut 1.5 newTree->Write();
887    
888 mocchiut 1.12 printf(" 36 \n");
889 mocchiut 1.5 // Get our return value.
890     Int_t nfiles = newTree->GetFileNumber() + 1;
891    
892     // Close and delete the current file of the new tree.
893     if (!opt.Contains("keep")) {
894     // FIXME: What happens to fDirectory in newTree here?
895     delete newTree->GetCurrentFile();
896     }
897     return nfiles;
898 mocchiut 1.1 }

  ViewVC Help
Powered by ViewVC 1.1.23