X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=MUON%2FAliMUONDataInterface.h;h=52a085f975092b3ee956b9444f523b2ce0ba0537;hb=61678c5e00f7eff81b5f3ecbbe9d10d24d202a38;hp=936a63e1cde028ca1cdb3fea5f7000ae8a955c62;hpb=e9bef706f5b4d2d59b367c01dbce377b36cbe2a7;p=u%2Fmrichter%2FAliRoot.git

diff --git a/MUON/AliMUONDataInterface.h b/MUON/AliMUONDataInterface.h
index 936a63e1cde..52a085f9750 100644
--- a/MUON/AliMUONDataInterface.h
+++ b/MUON/AliMUONDataInterface.h
@@ -1,5 +1,5 @@
-#ifndef ALI_MUON_DATA_INTERFACE_H
-#define ALI_MUON_DATA_INTERFACE_H
+#ifndef ALIMUONDATAINTERFACE_H
+#define ALIMUONDATAINTERFACE_H
 /*  Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
  * See cxx source for full Copyright notice                               */
 
@@ -10,45 +10,47 @@
 /// \class AliMUONDataInterface
 /// \brief An easy to use interface to MUON data
 
-// Author: Artur Szostak
-//  email: artur@alice.phy.uct.ac.za
+// Author: Artur Szostak (University of Cape Town)
+//  email: artursz@iafrica.com
 //
 // Updated to MUON module w/o MUONData by Laurent Aphecetche, Subatech
 //
 
 #include <TObject.h>
-#include <TString.h>
 
+class TIterator;
 class AliLoader;
-class AliMUONVClusterStore;
-class AliMUONVDigitStore;
 class AliMUONVStore;
+class AliMUONVDigitStore;
+class AliMUONVClusterStore;
 class AliMUONVTrackStore;
 class AliMUONVTriggerStore;
 class AliMUONVTriggerTrackStore;
-
-// >>LA should not needed (once we remove deprecated methods)
-class AliMUONHit; 
-class AliMUONDigit;
+class AliMUONVDigit;
+class AliMUONVCluster;
+class AliMUONTrack;
 class AliMUONLocalTrigger;
+class AliMUONRegionalTrigger;
 class AliMUONGlobalTrigger;
-class AliMUONRawCluster;
-class AliMUONTrack;
-class TParticle;
-// <<LA
+class AliMUONTriggerTrack;
+
 
 class AliMUONDataInterface : public TObject
 {
- public:
+public:
   
   AliMUONDataInterface(const char* filename="galice.root");
   virtual ~AliMUONDataInterface();
   
-  Bool_t IsValid() const;
+  /// Returns true if the data interface was able to open the root file correctly.
+  Bool_t IsValid() const { return fIsValid; };
 
   void Open(const char* filename);
 
   Int_t NumberOfEvents() const;
+  
+  /// Returns the index number of the current event as used int GetEvent(Int_t).
+  Int_t   CurrentEvent() const { return fCurrentEvent; }
 
   AliMUONVDigitStore* DigitStore(Int_t event);  
   AliMUONVClusterStore* ClusterStore(Int_t event);
@@ -64,95 +66,63 @@ class AliMUONDataInterface : public TObject
   void DumpTrigger(Int_t event, const char* treeLetter="R");  
   void DumpTriggerTracks(Int_t event, Bool_t sorted=kFALSE);
   
-  // all the methods below are deprecated.
-  
-  // Sets all internal pointers to NULL without releasing the current runloader.
-  void Reset();
-  
-  Bool_t UseCurrentRunLoader();
-  
-  Int_t NumberOfEvents(TString filename, TString foldername);
-  
-  Int_t NumberOfParticles(TString filename, TString foldername, Int_t event);
-  TParticle* Particle(TString filename, TString foldername, Int_t event, Int_t particle);
-  
-  Int_t NumberOfTracks(TString filename, TString foldername, Int_t event);
-  Int_t NumberOfHits(TString filename, TString foldername, Int_t event, Int_t track);
-  AliMUONHit* Hit(TString filename, TString foldername, Int_t event, Int_t track, Int_t hit);
-  
-  Int_t NumberOfSDigits(TString filename, TString foldername, Int_t event, Int_t chamber, Int_t cathode);
-  AliMUONDigit* SDigit(TString filename, TString foldername, Int_t event, Int_t chamber, Int_t cathode, Int_t sdigit);
-  
-  Int_t NumberOfDigits(TString filename, TString foldername, Int_t event, Int_t chamber, Int_t cathode);
-  AliMUONDigit* Digit(TString filename, TString foldername, Int_t event, Int_t chamber, Int_t cathode, Int_t digit);
-  
-  Int_t NumberOfRawClusters(TString filename, TString foldername, Int_t event, Int_t chamber);
-  AliMUONRawCluster* RawCluster(TString filename, TString foldername, Int_t event, Int_t chamber, Int_t cluster);
-  
-  Int_t NumberOfLocalTriggers(TString filename, TString foldername, Int_t event);
-  AliMUONLocalTrigger* LocalTrigger(TString filename, TString foldername, Int_t event, Int_t trigger);
-  
-  Bool_t SetFile(TString filename = "galice.root", TString foldername = "MUONFolder");
   Bool_t GetEvent(Int_t event = 0);
   
-  Int_t NumberOfParticles();
-  TParticle* Particle(Int_t particle);
-  
-  Int_t NumberOfTracks();
-  Int_t NumberOfHits(Int_t track);
-  AliMUONHit* Hit(Int_t track, Int_t hit);
-  
-  Int_t NumberOfSDigits(Int_t chamber, Int_t cathode);
-  AliMUONDigit* SDigit(Int_t chamber, Int_t cathode, Int_t sdigit);
-  
+  // Note the following methods can be extremely slow. Remember they are only
+  // here for end user convenience for his/her small tests and macros.
+  // If you want speed then don't use these methods. If you really want peak
+  // performance then you should be talking to the AliRunLoader and Store
+  // objects directly.
+  Int_t NumberOfDigits(Int_t detElemId);
+  AliMUONVDigit* Digit(Int_t detElemId, Int_t index);
   Int_t NumberOfDigits(Int_t chamber, Int_t cathode);
-  AliMUONDigit* Digit(Int_t chamber, Int_t cathode, Int_t digit);
-  
+  AliMUONVDigit* Digit(Int_t chamber, Int_t cathode, Int_t index);
   Int_t NumberOfRawClusters(Int_t chamber);
-  AliMUONRawCluster* RawCluster(Int_t chamber, Int_t cluster);
-  
+  AliMUONVCluster* RawCluster(Int_t chamber, Int_t index);
+  Int_t NumberOfTracks();
+  AliMUONTrack* Track(Int_t index);
   Int_t NumberOfLocalTriggers();
-  AliMUONLocalTrigger* LocalTrigger(Int_t trigger);
-  
-  Int_t NumberOfGlobalTriggers();
-  AliMUONGlobalTrigger* GlobalTrigger(Int_t trigger);
-  // Returns the name of the currently selected file.
-  
-  Int_t NumberOfRecTracks();
-  AliMUONTrack* RecTrack(Int_t rectrack);
-  
-  /// Returns the file name from which we are fetching data
-  TString CurrentFile() const { return ""; }
-  
-  /// Returns the name of the currently selected folder.
-  TString CurrentFolder() const { return ""; }
-  
-  /// Returns the number of the currently selected event.
-  Int_t   CurrentEvent() const { return 0; }
-  
-  /// Returns the currently selected track.
-  Int_t   CurrentTrack() const { return 0; }
-  
-  /// Returns the currently selected cathode in TreeS.
-  Int_t   CurrentSCathode() const { return 0; }
+  AliMUONLocalTrigger* LocalTrigger(Int_t index);
+  Int_t NumberOfRegionalTriggers();
+  AliMUONRegionalTrigger* RegionalTrigger(Int_t index);
+  AliMUONGlobalTrigger* GlobalTrigger();
+  Int_t NumberOfTriggerTracks();
+  AliMUONTriggerTrack* TriggerTrack(Int_t index);
   
-  /// Returns the currently selected cathode in TreeD.
-  Int_t   CurrentDCathode() const { return 0; }
-  
- private:
+private:
+
+  /// The various identifiers for the type of iterator constructed.
+  enum IteratorType
+  {
+    kNoIterator,  ///< No iterator was constructed.
+    kDigitIteratorByDetectorElement,  ///< A digit iterator for iterating over detector elements.
+    kDigitIteratorByChamberAndCathode,  ///< A digit iterator for iterating over chambers and cathodes.
+    kRawClusterIterator,  ///< A raw cluster iterator.
+    kTrackIterator,  ///< An iterator for iterating over reconstructed tracks.
+    kLocalTriggerIterator,  ///< An iterator for iterating over reconstructed local triggers.
+    kRegionalTriggerIterator,  ///< An iterator for iterating over reconstructed regional triggers.
+    kTriggerTrackIterator  ///< An iterator for iterating over reconstructed trigger tracks.
+  };
     
   void DumpSorted(const AliMUONVStore& store) const;
 
-  Int_t LoadEvent(Int_t event);
+  Bool_t LoadEvent(Int_t event);
 
   void NtupleTrigger(const char* treeLetter);
   
+  void ResetStores();
+  
+  TIterator* GetIterator(IteratorType type, Int_t x = 0, Int_t y = 0);
+  void ResetIterator();
+  
+  Int_t CountObjects(TIterator* iter);
+  TObject* FetchObject(TIterator* iter, Int_t index);
+  
   /// Not implemented
   AliMUONDataInterface(const AliMUONDataInterface& rhs);
   /// Not implemented
   AliMUONDataInterface& operator=(const AliMUONDataInterface& rhs);
-
-private:
+  
   
   AliLoader* fLoader; //!< Tree accessor
   AliMUONVDigitStore* fDigitStore; //!< current digit store (owner)
@@ -163,10 +133,16 @@ private:
   Int_t fCurrentEvent; //!< Current event we've read in
   Bool_t fIsValid; //!< whether we were initialized properly or not
   
+  IteratorType fCurrentIteratorType;  //!< The type of iterator that is currently set.
+  Int_t fCurrentIndex;  //!< A current index number maintained for certain iteration operations.
+  Int_t fDataX; //!< Extra data parameter about the iterator, can be the chamber number or detector element.
+  Int_t fDataY; //!< Extra data parameter about the iterator, can be the cathode number.
+  TIterator* fIterator; //!< Iterator for various iteration operations.
+  
   static Int_t fgInstanceCounter; //!< To build unique folder name for each instance
   
-  ClassDef(AliMUONDataInterface, 0)  // An easy to use interface to MUON data
+  ClassDef(AliMUONDataInterface, 0)  // An easy to use interface to MUON reconstructed data
 };
     
 
-#endif // ALI_MUON_DATA_INTERFACE_H
+#endif // ALIMUONDATAINTERFACE_H