o) splitted analysis and correction of dneta

[u/mrichter/AliRoot.git] / PWG0 / CreateESDChain.C

diff --git a/PWG0/CreateESDChain.C b/PWG0/CreateESDChain.C
index d572649afc57709f6a41cc952f651c4b37376de8..2f60dfe8a9575a6f4d8446f19dd01e61453a872f 100644 (file)
--- a/PWG0/CreateESDChain.C
+++ b/PWG0/CreateESDChain.C
@@ -1,14 +1,20 @@
-TChain* CreateESDChain(const char* aDataDir, Int_t aRuns = 20, Bool_t aAddHeader = kTRUE)
+/* $Id$ */
+
+// Helper macros for creating chains
+
+TChain* CreateESDChainFromDir(const char* aDataDir, Int_t aRuns = 20, Int_t offset = 0)
 {
+  // creates chain of files in a given directory. The structure is expected as:
+  // <aDataDir>/<dir0>/AliESDs.root
+  // <aDataDir>/<dir1>/AliESDs.root
+  // ...
+
   if (!aDataDir)
     return 0;
 
   TChain* chain = new TChain("esdTree");
   TChain* chaingAlice = 0;
 
-  if (aAddHeader != kFALSE)
-    chaingAlice = new TChain("TE");
-
   TString execDir(gSystem->pwd());
   TSystemDirectory* baseDir = new TSystemDirectory(".", aDataDir);
   TList* dirList            = baseDir->GetListOfFiles();
@@ -23,6 +29,12 @@ TChain* CreateESDChain(const char* aDataDir, Int_t aRuns = 20, Bool_t aAddHeader
     if (!presentDir || !presentDir->IsDirectory() || strcmp(presentDir->GetName(), ".") == 0 || strcmp(presentDir->GetName(), "..") == 0)
       continue;
 
+    if (offset > 0)
+    {
+      --offset;
+      continue;
+    }
+
     if (count++ == aRuns)
       break;
 
@@ -31,13 +43,43 @@ TChain* CreateESDChain(const char* aDataDir, Int_t aRuns = 20, Bool_t aAddHeader
     presentDirName += presentDir->GetName();
 
     chain->Add(presentDirName + "/AliESDs.root/esdTree");
+  }
 
-    if (aAddHeader != kFALSE)
-      chaingAlice->Add(presentDirName + "/galice.root/TE");
+  return chain;
+}
+
+TChain* CreateESDChainFromList(const char* listFile, Int_t aRuns = 20)
+{
+  // Creates a chain from a file which contains a list of ESD files
+
+  if (!listFile)
+    return 0;
+
+  TChain* chain = new TChain("esdTree");
+  TChain* chaingAlice = 0;
+
+  // Open the input stream
+  ifstream in;
+  in.open(listFile);
+
+  Int_t count = 0;
+
+  // Read the input list of files and add them to the chain
+  TString esdfile;
+  while(in.good()) {
+    in >> esdfile;
+    if (!esdfile.Contains("root")) continue; // protection
+
+    if (count++ == aRuns)
+      break;
+
+      // add esd file
+    chain->Add(esdfile);
   }
 
-  if (aAddHeader != kFALSE)
-    chain->AddFriend(chaingAlice);
+  in.close();
+
+  chain->Lookup();
 
   return chain;
 }