CreateFileCollection.C

Go to the documentation of this file.

void
CreateFileCollection(const TString& dir="/data/alice/data/ppb/LHC12g/pass1/188359/",
         const TString& tN="esdTree", 
         const TString& pa="AliESD*", 
         Bool_t mc=false, 
         Bool_t recursive=false)
)
{
  gROOT->LoadMacro("$ALICE_PHYSICS/PWGLF/FORWARD/trains/ChainBuilder.C+");
  
  UShort_t flags = 0;
  if (recursive) flags |= ChainBuilder::kRecursive;
  if (mc)        flags |= ChainBuilder::kMC;
  UShort_t type = ChainBuilder::CheckSource(dir, flags);
  Info("", "type=%d, dir=%s tN=%s pa=%s, flags=0x%x", type,
       dir.Data(), tN.Data(), pa.Data(), flags);
  TChain* chain = ChainBuilder::Create(type, dir, tN, pa, flags);
  if (!chain) { 
    Error("CreateFileCollection", "Failed to make chain");
    return;
  }
  Int_t port;
  TString host;
  { 
    TUrl u(Form("root://%s//foo", gSystem->HostName()));
    port = u.GetPort() * 10;
    host = u.GetHostFQDN();
  }
  

  TFileCollection* fc  = new TFileCollection("files");
  TObjArray*       cEs = chain->GetListOfFiles();
  TChainElement*   cE  = 0;
  TIter            next(cEs);
  while ((cE= static_cast<TChainElement*>(next()))) {
    TString fN(cE->GetTitle());
    TFile* f = TFile::Open(fN, "READ");
    TTree* t = static_cast<TTree*>(f->Get(tN));
    
    fN.Prepend(Form("root://%s:%d/", host.Data(), port));
    TFileInfo* fi = new TFileInfo(Form("%s tree:%s,%d", 
               fN.Data(), tN.Data(), t->GetEntries()),
          f->GetSize());
    f->Close();
    fc->Add(fi);
  }
  fc->Print("F");
  
  TFile* files = TFile::Open("files.root", "RECREATE");
  fc->Write();
  files->Close();
  
}
//
// EOF
//