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

Annotation of /PadmeAmidala/src/RunGlue.cpp

Parent Directory Parent Directory | Revision Log Revision Log

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