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

Annotation of /PadmeAmidala/src/RunGlue.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.25 - (hide annotations) (download)
Fri Oct 17 15:58:41 2014 UTC (10 years, 1 month ago) by mocchiut
Branch: MAIN
CVS Tags: v10RED, HEAD
Changes since 1.24: +32 -23 lines
ProcessingInfo added to output file

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

  ViewVC Help
Powered by ViewVC 1.1.23