minor updates
[u/mrichter/AliRoot.git] / PWG0 / CreateESDChain.C
index 297da8f1f5ec1213b836dd55070a69439e6b6d64..d572649afc57709f6a41cc952f651c4b37376de8 100644 (file)
@@ -1,4 +1,4 @@
-TChain* CreateESDChain(const char* aDataDir, Bool_t aAddHeader = kTRUE)
+TChain* CreateESDChain(const char* aDataDir, Int_t aRuns = 20, Bool_t aAddHeader = kTRUE)
 {
   if (!aDataDir)
     return 0;
@@ -15,12 +15,17 @@ TChain* CreateESDChain(const char* aDataDir, Bool_t aAddHeader = kTRUE)
   Int_t nDirs               = dirList->GetEntries();
   gSystem->cd(execDir);
 
+  Int_t count = 0;
+
   for (Int_t iDir=0; iDir<nDirs; ++iDir)
   {
     TSystemFile* presentDir = (TSystemFile*) dirList->At(iDir);
     if (!presentDir || !presentDir->IsDirectory() || strcmp(presentDir->GetName(), ".") == 0 || strcmp(presentDir->GetName(), "..") == 0)
       continue;
 
+    if (count++ == aRuns)
+      break;
+
     TString presentDirName(aDataDir);
     presentDirName += "/";
     presentDirName += presentDir->GetName();