MUON/AliMUONData.h

   1 #ifndef ALIMUONDATA_H
   2 #define ALIMUONDATA_H
   3 //
   4 // AliMUONData
   5 // Class containing MUON data: hits, digits, rawclusters, globaltrigger, localtrigger, etc ...
   6 // Gines Martinez, Subatech,  September 2003
   7 //
   8
   9 #include "AliLoader.h"
  10
  11 class TClonesArray;
  12 class TNamed;
  13 class TObjArray;
  14 class TTree;
  15
  16
  17 class AliMUONConstants;
  18 class AliMUONRawCluster;
  19 class AliMUONTrack;
  20
  21 //__________________________________________________________________
  22 /////////////////////////////////////////////////////////////////////
  23 //                                                                 //
  24 //  class AliMUONData                                              //
  25 //                                                                 //
  26 /////////////////////////////////////////////////////////////////////
  27
  28 class AliMUONData : public TNamed {
  29  public:
  30     AliMUONData();
  31     AliMUONData(AliLoader * loader, const char* name, const char* title);
  32     AliMUONData(const AliMUONData& rMUONData);
  33     virtual ~AliMUONData();
  34     virtual void   AddDigit(Int_t id, Int_t* tracks, Int_t* charges,
  35                              Int_t* digits);
  36     virtual void   AddHit(Int_t fIshunt, Int_t track, Int_t iChamber,
  37                           Int_t idpart, Float_t X, Float_t Y, Float_t Z,
  38                           Float_t tof, Float_t momentum, Float_t theta,
  39                           Float_t phi, Float_t length, Float_t destep);
  40     virtual void   AddGlobalTrigger(Int_t *singlePlus, Int_t *singleMinus,
  41                                     Int_t *singleUndef, Int_t *pairUnlike,
  42                                     Int_t *pairLike);
  43     virtual void   AddLocalTrigger(Int_t* ltrigger);
  44     virtual void   AddRawCluster(Int_t id, const AliMUONRawCluster& clust);
  45     virtual void   AddRecTrack(const AliMUONTrack& track);
  46
  47     TClonesArray*  Hits() {return fHits;}
  48     TClonesArray*  Digits(Int_t DetectionPlane)
  49       {return ( (TClonesArray*) fDigits->At(DetectionPlane) );}
  50     TClonesArray*  LocalTrigger() {return fLocalTrigger;}
  51     TClonesArray*  GlobalTrigger() {return fGlobalTrigger;}
  52     TClonesArray*  RawClusters(Int_t DetectionPlane)
  53       {return ( (TClonesArray*) fRawClusters->At(DetectionPlane) );}
  54     TClonesArray*  RecTracks() {return fRecTracks;}
  55
  56     void           GetTrack(Int_t it) {fLoader->TreeH()->GetEvent(it);}
  57     Int_t          GetNtracks()       {return (Int_t) fLoader->TreeH()->GetEntries();}
  58     void           GetCathode(Int_t ic) {fLoader->TreeD()->GetEvent(ic);}
  59     void           GetRawClusters() {fLoader->TreeR()->GetEvent(0);}
  60     void           GetTrigger() {fLoader->TreeR()->GetEvent(0);}
  61
  62     virtual AliLoader* GetLoader() {return fLoader;}
  63     virtual void       SetLoader(AliLoader * loader) {fLoader=loader;}
  64
  65     virtual void   Fill(Option_t* opt=" ");
  66     virtual void   MakeBranch(Option_t *opt=" ");
  67     virtual void   SetTreeAddress(Option_t *opt=" ");
  68
  69     virtual void   ResetHits();
  70     virtual void   ResetDigits();
  71     virtual void   ResetTrigger();
  72     virtual void   ResetRawClusters();
  73     virtual void   ResetRecTracks();
  74
  75     TTree*         TreeH() {return fLoader->TreeH(); }
  76     TTree*         TreeD() {return fLoader->TreeD(); }
  77     TTree*         TreeR() {return fLoader->TreeR(); }
  78     TTree*         TreeT() {return fLoader->TreeT(); }
  79     TTree*         TreeP() {return fLoader->TreeP(); }
  80
  81  private:
  82     //descendant classes should
  83     //use protected interface methods to access these folders
  84
  85
  86  protected:
  87     AliLoader*  fLoader;
  88
  89     // One event in treeH per primary track
  90     TClonesArray*   fHits;
  91     // One event in treeD and one branch per detection plane
  92     TObjArray*      fDigits;
  93     //One event in TreeR/Rawcluster and one branch per tracking detection plane
  94     TObjArray*      fRawClusters;
  95     //! List of Global Trigger One event in TreeR/GlobalTriggerBranch
  96     TClonesArray*   fGlobalTrigger;
  97     //! List of Local Trigger, One event in TreeR/LocalTriggerBranch
  98     TClonesArray*   fLocalTrigger;
  99     // pointer to array of reconstructed tracks
 100     TClonesArray*   fRecTracks;
 101
 102     Int_t           fNhits;
 103     Int_t*          fNdigits;
 104     Int_t*          fNrawclusters;
 105     Int_t           fNglobaltrigger;
 106     Int_t           fNlocaltrigger;
 107     Int_t           fNrectracks;
 108
 109     ClassDef(AliMUONData,1)
 110  };
 111 #endif