Changes to use new VDigitStore and VClusterStore (Laurent)

[u/mrichter/AliRoot.git] / MUON / AliMUONDataInterface.h

#ifndef ALI_MUON_DATA_INTERFACE_H
#define ALI_MUON_DATA_INTERFACE_H
/*  Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */
// Includes revised 07/05/2004
//
/// \ingroup evaluation
/// \class AliMUONDataInterface
/// \brief An easy to use interface to data in the MUON module

// Author: Artur Szostak
//  email: artur@alice.phy.uct.ac.za

#include <TObject.h>
#include <TString.h>

#include "AliMUONRecData.h"
#include "AliMUONSimData.h"

class TParticle;

class AliRunLoader;
class AliLoader;
class AliMUONRawCluster;
class AliMUONLocalTrigger;
class AliMUONHit;
class AliMUONDigit;
class AliMUONTrack;

class AliMUONDataInterface : public TObject
{
 public:
  
  AliMUONDataInterface();
  ~AliMUONDataInterface();
  
  // 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 NumberOfEvents();
  
  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);
  
  Int_t NumberOfDigits(Int_t chamber, Int_t cathode);
  AliMUONDigit* Digit(Int_t chamber, Int_t cathode, Int_t digit);
  
  Int_t NumberOfRawClusters(Int_t chamber);
  AliMUONRawCluster* RawCluster(Int_t chamber, Int_t cluster);
  
  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 fFilename;    };
  
  /// Returns the name of the currently selected folder.
  TString CurrentFolder() const   { return fFoldername;  };
  
  /// Returns the number of the currently selected event.
  Int_t   CurrentEvent() const    { return fEventnumber; };
  
  /// Returns the currently selected track.
  Int_t   CurrentTrack() const    { return fTrack;       };
  
  /// Returns the currently selected cathode in TreeS.
  Int_t   CurrentSCathode() const { return fSCathode;    };
  
  /// Returns the currently selected cathode in TreeD.
  Int_t   CurrentDCathode() const { return fCathode;     };
  
 private:
  /// Not implemented
  AliMUONDataInterface(const AliMUONDataInterface& rhs);
  /// Not implemented
  AliMUONDataInterface& operator=(const AliMUONDataInterface& rhs);
  
  Bool_t FetchMuonLoader(TString filename);
  Bool_t LoadLoaders(TString filename);
  Bool_t FetchLoaders(TString filename);
  Bool_t FetchEvent(Int_t event);
  Bool_t FetchTreeK();
  Bool_t FetchTreeH();
  Bool_t FetchTreeS();
  Bool_t FetchTreeD();
  Bool_t FetchTreeR();
  Bool_t FetchTreeT();
  
  Bool_t fCreatedRunLoader;   //!< If this object created the fRunloader then this flag is set.	
  Bool_t fCreatedRunLoaderSim;//!< If this object created the fRunloader then this flag is set.	
  
  Bool_t fHitAddressSet;     //!< Flag specifying if the TTree address for the hit tree was set.
  Bool_t fSDigitAddressSet;  //!< Flag specifying if the TTree address for the s-digit tree was set.
  Bool_t fDigitAddressSet;   //!< Flag specifying if the TTree address for the digit tree was set.
  Bool_t fClusterAddressSet; //!< Flag specifying if the TTree address for the cluster tree was set.
  Bool_t fTriggerAddressSet; //!< Flag specifying if the TTree address for the trigger tree was set.
  Bool_t fRecTracksAddressSet; //!< Flag specifying if the TTree address for the rec tracks tree was set.
  
  AliRunLoader* fRunloader;  //!< Pointer to the runloader object used.
  AliRunLoader* fRunloaderSim; //!< Pointer to the runloader object used.
  AliLoader* fRecLoader;    //!< Pointer to the muon loader object used.
  AliLoader* fSimLoader;    //!< Pointer to the muon loader object used.
  AliMUONRecData fRecData;  //!< Pointer to the muon raw data interface.
  AliMUONSimData fSimData;  //!< Pointer to the muon raw data interface.
  TString fFilename;         //!< The file name from which we are fetching data.
  TString fFoldername;       //!< The folder name from which we are fetching data.
  TString fFoldernameSim;    //!< The folder name from which we are fetching data.
  Int_t fEventnumber;        //!< The currently selected event.
  Int_t fTrack;              //!< The currently selected track.
  Int_t fSCathode;           //!< The currently selected cathode in TreeS.
  Int_t fCathode;            //!< The currently selected cathode in TreeD.
  
  ClassDef(AliMUONDataInterface, 0)  // A easy to use interface to data in the MUON module.
    };
    

#endif // ALI_MUON_DATA_INTERFACE_H