adding missing const
[u/mrichter/AliRoot.git] / STEER / AliESDInputHandlerRP.cxx
index e27a0b1..ad9b55b 100644 (file)
@@ -13,7 +13,7 @@
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
 
-/* $Id: AliESDInputHandler.cxx 24521 2008-03-14 16:43:54Z morsch $ */
+/* $Id$ */
 
 //-------------------------------------------------------------------------
 //     Event handler for ESD input reading the RecPoint Trees in parallel
 #include <TTree.h>
 #include <TList.h>
 #include <TFile.h>
+#include <TArchiveFile.h>
+#include <TSystemDirectory.h>
 #include <TString.h>
 #include <TObjString.h>
+#include <TObjArray.h>
 #include <TProcessID.h>
 
 #include "AliESDInputHandlerRP.h"
@@ -37,41 +40,44 @@ ClassImp(AliESDInputHandlerRP)
 //______________________________________________________________________________
 AliESDInputHandlerRP::AliESDInputHandlerRP() :
     AliESDInputHandler(),
-    fRTrees(new TList()),
-    fRFiles(new TList()),
+    fRTrees(   new TObjArray()),
+    fRFiles(   new TList()),
+    fDetectors(new TList()),
     fDirR(0),
     fEventNumber(-1),
-    fNEvent(-1),
     fFileNumber(0),
     fEventsPerFile(0),
     fExtension(""),
-    fPathName(new TString("./"))
+    fPathName(new TString("./")),
+    fIsArchive(kFALSE)
 {
-  // default constructor
+  // Default constructor
 }
 
-//______________________________________________________________________________
-AliESDInputHandlerRP::~AliESDInputHandlerRP() 
-{
-  // destructor
-}
 
 //______________________________________________________________________________
 AliESDInputHandlerRP::AliESDInputHandlerRP(const char* name, const char* title):
     AliESDInputHandler(name, title),
-    fRTrees(new TList()),
-    fRFiles(new TList()),
+    fRTrees(   new TObjArray()),
+    fRFiles(   new TList()),
+    fDetectors(new TList()),
     fDirR(0),
     fEventNumber(-1),
-    fNEvent(-1),
     fFileNumber(0),
     fEventsPerFile(0),
     fExtension(""),
-    fPathName(new TString("./"))
+    fPathName(new TString("./")),
+    fIsArchive(kFALSE)
 {
     // Constructor
 }
 
+//______________________________________________________________________________
+AliESDInputHandlerRP::~AliESDInputHandlerRP() 
+{
+  // Destructor
+}
+
 Bool_t AliESDInputHandlerRP::Init(Option_t* opt)
 {
     //
@@ -79,25 +85,39 @@ Bool_t AliESDInputHandlerRP::Init(Option_t* opt)
     //
     if (!(strcmp(opt, "proof")) || !(strcmp(opt, "local"))) return kTRUE;
     //
-    TFile* file;
+    TIter next(fDetectors);
+    TNamed* det;
+    TFile* file = 0;
+    while ((det = (TNamed*) next()))
+    {
+       if (!fIsArchive) {
+           file = TFile::Open(Form("%s%s.RecPoints.root", fPathName->Data(), det->GetName()));
+       } else {
+           file = TFile::Open(Form("%s#%s.RecPoints.root", fPathName->Data(), det->GetName()));
+       }
+       if (!file) AliFatal(Form("AliESDInputHandlerRP: %s.RecPoints.root not found in %s ! \n", det->GetName(), fPathName->Data()));
+       fRFiles->Add(file);
+    }
+
+    if (file) {
+       fEventsPerFile = file->GetNkeys() - file->GetNProcessIDs();
+    } else {
+       AliFatal(Form("AliESDInputHandlerRP: No file with RecPoints found in %s ! \n", fPathName->Data()));
+    }
     
-    file = TFile::Open(Form("%sTPC.RecPoints.root", fPathName->Data()));
-    if (!file) AliFatal(Form("AliESDInputHandlerRP: TPC.RecPoints.root not found in %s ! \n", fPathName->Data()));
-    fRFiles->Add(file);
-    fEventsPerFile = file->GetNkeys() - file->GetNProcessIDs();
+
     // Reset the event number
     fEventNumber      = -1;
     fFileNumber       =  0;
-    fNEvent           =  fTree->GetEntries();
-    
-    printf("AliESDInputHandler::Init() %d\n",__LINE__);
+    // Get number of events from esd tree 
+    printf("AliESDInputHandler::Init() %d %d\n",__LINE__, fNEvents);
     return kTRUE;
 }
 
 Bool_t AliESDInputHandlerRP::BeginEvent(Long64_t entry)
 {
     // Begin the next event
-    // Delegate to base class
+    // Delegate first to base class
     AliESDInputHandler::BeginEvent(entry);
 //
     if (entry == -1) {
@@ -107,8 +127,8 @@ Bool_t AliESDInputHandlerRP::BeginEvent(Long64_t entry)
        fEventNumber = entry;
     }
     
-    if (entry >= fNEvent) {
-       AliWarning(Form("AliESDInputHandlerRP: Event number out of range %5d %5d\n", entry, fNEvent));
+    if (entry >= fNEvents) {
+       AliWarning(Form("AliESDInputHandlerRP: Event number out of range %5d %5d\n", entry, fNEvents));
        return kFALSE;
     }
     return LoadEvent(entry);
@@ -130,19 +150,21 @@ Bool_t AliESDInputHandlerRP::LoadEvent(Int_t iev)
     char folder[20];
     sprintf(folder, "Event%d", iev);
     // Tree R
-    TFile* file = (TFile*) (fRFiles->At(0));
+    TIter next(fRFiles);
+    TFile* file;
+    Int_t idx = 0;
     
-    file->GetObject(folder, fDirR);
-    if (!fDirR) {
-       AliWarning(Form("AliESDInputHandlerRP: Event #%5d not found\n", iev));
-       return kFALSE;
+    while ((file = (TFile*) next()))
+    {
+       file->GetObject(folder, fDirR);
+       if (!fDirR) {
+           AliWarning(Form("AliESDInputHandlerRP: Event #%5d not found\n", iev));
+           return kFALSE;
+       }
+       TTree* tree = 0;
+       fDirR->GetObject("TreeR", tree);
+       fRTrees->AddAt(tree, idx++);
     }
-    
-    TTree* tree;
-    fDirR ->GetObject("TreeR", tree);
-    fRTrees->Add(tree);
-    tree->ls();
-    
     return kTRUE;
 }
 
@@ -157,11 +179,18 @@ Bool_t AliESDInputHandlerRP::OpenFile(Int_t i)
     }
     
     fRFiles->Delete();
+    TIter next(fDetectors);
+    TNamed* det;
     TFile* file;
-    file = TFile::Open(Form("%sTPC.RecPoints%s.root", fPathName->Data(), fExtension));
-    if (!file) {
-       AliFatal(Form("AliESDInputHandlerRP: TPC.RecPoints.root not found in %s ! \n", fPathName->Data()));
-       ok = kFALSE;
+    while ((det = (TNamed*) next()))
+    {
+       if (!fIsArchive) {
+           file = TFile::Open(Form("%s%s.RecPoints%s.root", fPathName->Data(), det->GetName(), fExtension));
+       } else {
+           file = TFile::Open(Form("%s#%s.RecPoints%s.root", fPathName->Data(), det->GetName(), fExtension));
+       }
+       if (!file) AliFatal(Form("AliESDInputHandlerRP: RecPoints.root not found in %s ! \n", fPathName->Data()));
+       fRFiles->Add(file);
     }
     return ok;
 }
@@ -170,18 +199,69 @@ Bool_t AliESDInputHandlerRP::Notify(const char *path)
 {
   // Notify about directory change
   // The directory is taken from the 'path' argument
-  // Reconnect trees
+  // 
+    // Get path to directory
     TString fileName(path);
     if(fileName.Contains("AliESDs.root")){
        fileName.ReplaceAll("AliESDs.root", "");
     }
-
+    // If this is an archive it will contain a # 
+    if(fileName.Contains("#")){
+       fileName.ReplaceAll("#", "");
+    }
+    //
+    // At this point we have a path to the directory or to the archive
     *fPathName = fileName;
+    //
+    // Now filter the files containing RecPoints *.RecPoints.*
+    fIsArchive = kFALSE;
+    if (fPathName->Contains(".zip")) fIsArchive = kTRUE;
 
-    printf("AliESDInputHandlerRP::Notify() Path: %s\n", fPathName->Data());
+    TSeqCollection* members;
+
+    
+    if (fIsArchive) {
+       // Archive
+       TFile* file = TFile::Open(fPathName->Data());
+       TArchiveFile* arch = file->GetArchive();
+       members = arch->GetMembers();
+    } else {
+       // Directory
+       TSystemDirectory dir(".", fPathName->Data());
+       members = dir.GetListOfFiles();
+    }
+
+    TIter next(members);
+    TFile* entry;
+    Int_t ien = 0;
+    fDetectors->Delete();
     
+    while ( (entry = (TFile*) next()) )
+    {
+       TString name(entry->GetName());
+       TObjArray* tokens = name.Tokenize(".");
+       Int_t ntok = tokens->GetEntries();
+       if (ntok <= 1) continue;
+       TString str = ((TObjString*) tokens->At(1))->GetString();
+       if (!(strcmp(str.Data(), "RecPoints"))){
+           TString det = ((TObjString*) tokens->At(0))->GetString();
+           printf("Found file with RecPoints for %s \n", det.Data());
+           TNamed* ent = new TNamed(det.Data(), det.Data());
+           fRTrees->AddAt(0, ien);
+           ent->SetUniqueID(ien++);
+           fDetectors->Add(ent);
+       }
+    } // loop over files
+
+
+    // Now we have the path and the list of detectors
+    
+    printf("AliESDInputHandlerRP::Notify() Path: %s\n", fPathName->Data());
+    //
     ResetIO();
     InitIO("");
+    // Some clean-up
+    members->Delete();
 
     return kTRUE;
 }
@@ -197,7 +277,19 @@ Bool_t AliESDInputHandlerRP::FinishEvent()
 void AliESDInputHandlerRP::ResetIO()
 {
 // Delete trees and files
-    fRTrees->Delete();
     fRFiles->Delete();
     fExtension="";
 }
+
+TTree* AliESDInputHandlerRP::GetTreeR(const char* det)
+{
+// Return pointer to RecPoint tree for detector det
+    TNamed* entry = (TNamed*) (fDetectors->FindObject(det));
+    if (!entry) {
+       AliWarning(Form("AliESDInputHandlerRP: No RecPoints for detector %s available \n", det));
+       return 0;
+    } else {
+       Int_t ien = entry->GetUniqueID();
+       return ((TTree*) (fRTrees->At(ien)));
+    }
+}