MUON/AliMUONData.h

   1 #ifndef ALIMUONDATA_H
   2 #define ALIMUONDATA_H
   3 //
   4 /*  Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 /* $Id$ */
   8 // Revision of includes 07/05/2004
   9 //
  10 /// \ingroup base
  11 /// \class AliMUONData
  12 /// \brief MUON data
  13 ///
  14 /// Class containing MUON data: hits, digits, rawclusters, globaltrigger,
  15 /// localtrigger, etc ...
  16 ///
  17 /// Author: Gines Martinez, Subatech,  September 2003
  18
  19 #include <TNamed.h>
  20
  21 class TArrayI;
  22
  23 #include "AliLoader.h"
  24
  25 class TClonesArray;
  26 class TNamed;
  27 class TObjArray;
  28 class TTree;
  29 class TIterator;
  30
  31 class AliMUONConstants;
  32 class AliMUONRawCluster;
  33 class AliMUONTrack;
  34 class AliMUONTriggerTrack;
  35 class AliMUONDigit;
  36 class AliMUONHit;
  37 class AliMUONLocalTrigger;
  38 class AliMUONRegionalTrigger;
  39 class AliMUONGlobalTrigger;
  40
  41 //__________________________________________________________________
  42 /////////////////////////////////////////////////////////////////////
  43 //                                                                 //
  44 //  class AliMUONData                                              //
  45 //                                                                 //
  46 /////////////////////////////////////////////////////////////////////
  47
  48 class AliMUONData : public TNamed
  49 {
  50   public:
  51
  52   //  enum EChamberIteration { kAllChambers, kTrackingChambers, kTriggerChambers };
  53
  54     AliMUONData();
  55     AliMUONData(AliLoader * loader, const char* name, const char* title);
  56     virtual ~AliMUONData();
  57     virtual void   AddDigit(Int_t id, Int_t* tracks, Int_t* charges,
  58                              Int_t* digits);
  59     virtual void   AddSDigit(Int_t id, Int_t* tracks, Int_t* charges,
  60                              Int_t* digits);
  61     virtual void   AddDigit(Int_t id, const AliMUONDigit& digit); // use copy constructor
  62     virtual void   AddSDigit(Int_t id, const AliMUONDigit& digit); // use copy constructor
  63     virtual void   AddHit(Int_t fIshunt, Int_t track, Int_t iChamber,
  64                           Int_t idpart, Float_t X, Float_t Y, Float_t Z,
  65                           Float_t tof, Float_t momentum, Float_t theta,
  66                           Float_t phi, Float_t length, Float_t destep,
  67                           Float_t Xref,Float_t Yref,Float_t Zref);
  68                           // TBR
  69     virtual void   AddHit2(Int_t fIshunt, Int_t track, Int_t detElemId,
  70                           Int_t idpart, Float_t X, Float_t Y, Float_t Z,
  71                           Float_t tof, Float_t momentum, Float_t theta,
  72                           Float_t phi, Float_t length, Float_t destep,
  73                           Float_t Xref,Float_t Yref,Float_t Zref);
  74
  75     virtual void   AddGlobalTrigger(const AliMUONGlobalTrigger& trigger); // use copy constructor
  76
  77     virtual void   AddLocalTrigger(const AliMUONLocalTrigger& trigger); // use copy constructor
  78
  79     virtual void   AddRegionalTrigger(const AliMUONRegionalTrigger& trigger); // use copy constructor
  80
  81     virtual void   AddRawCluster(Int_t id, const AliMUONRawCluster& clust);
  82     virtual void   AddRecTrack(const AliMUONTrack& track);
  83     virtual void   AddRecTriggerTrack(const AliMUONTriggerTrack& triggertrack);
  84
  85     TClonesArray*  Hits() {return fHits;} ///< Return hits
  86     TClonesArray*  Digits(Int_t DetectionPlane) const;
  87     TClonesArray*  SDigits(Int_t DetectionPlane) const;
  88     TClonesArray*  LocalTrigger() const;
  89     TClonesArray*  RegionalTrigger() const;
  90     TClonesArray*  GlobalTrigger() const;
  91     TClonesArray*  RawClusters(Int_t DetectionPlane);
  92
  93                     /// Return reconstructed tracks
  94     TClonesArray*  RecTracks() {return fRecTracks;}
  95                    /// Return reconstructed trigger tracks
  96     TClonesArray*  RecTriggerTracks() {return fRecTriggerTracks;}
  97
  98                    /// Load hits for \a i th entry in hits three
  99     void           GetTrack(Int_t it) const  {
 100       if (fLoader && fLoader->TreeH())
 101         fLoader->TreeH()->GetEvent(it);
 102     }
 103                    /// Get number of entries in hits three
 104     Int_t          GetNtracks() const      {
 105       Int_t ntrk = 0;
 106       if (fLoader && fLoader->TreeH())
 107         ntrk = (Int_t) fLoader->TreeH()->GetEntries();
 108       return ntrk;
 109     }
 110     void           GetDigits() const;
 111                    /// Load sdigits tree
 112     void           GetSDigits() const {
 113       if (fLoader && fLoader->TreeS())
 114         fLoader->TreeS()->GetEvent(0);
 115     }
 116                    /// Load raw clusters tree
 117     void           GetRawClusters() const {
 118       if (fLoader && fLoader->TreeR())
 119         fLoader->TreeR()->GetEvent(0);
 120     }
 121                    /// Load trigger tree
 122     void           GetTrigger() const {
 123       if (fLoader && fLoader->TreeR())
 124         fLoader->TreeR()->GetEvent(0);
 125     }
 126                    /// Load trigger D tree
 127     void           GetTriggerD() const {
 128       if (fLoader && fLoader->TreeD())
 129         fLoader->TreeD()->GetEvent(0);
 130     }
 131                    /// Return split level
 132     Int_t          GetSplitLevel() const {return fSplitLevel;}
 133                    /// Return reconstructed tracks
 134     void           GetRecTracks() const {
 135       if (fLoader && fLoader->TreeT())
 136         fLoader->TreeT()->GetEvent(0);
 137     }
 138                    /// Return reconstructed trigger tracks
 139     void           GetRecTriggerTracks() const {
 140       if (fLoader && fLoader->TreeT())
 141         fLoader->TreeT()->GetEvent(0);
 142     }
 143
 144     Bool_t        IsRawClusterBranchesInTree();
 145     Bool_t        IsDigitsBranchesInTree();
 146     Bool_t        IsTriggerBranchesInTree();
 147     Bool_t        IsTriggerBranchesInTreeD();
 148     Bool_t        IsTrackBranchesInTree();
 149     Bool_t        IsTriggerTrackBranchesInTree();
 150
 151                        /// Get loader
 152     virtual AliLoader* GetLoader() const { return fLoader; }
 153                        /// Set loader
 154     virtual void       SetLoader(AliLoader * loader) {fLoader=loader;}
 155
 156     virtual void   Fill(Option_t* opt=" ");
 157     virtual void   MakeBranch(Option_t *opt=" ");
 158     virtual void   SetTreeAddress(Option_t *opt=" ");
 159
 160                    /// Set split level
 161     void           SetSplitLevel(Int_t SplitLevel) {fSplitLevel=SplitLevel;}
 162
 163     virtual void Print(Option_t* opt="") const;
 164
 165     virtual void   ResetHits();
 166     virtual void   ResetDigits();
 167     virtual void   ResetSDigits();
 168     virtual void   ResetTrigger();
 169     virtual void   ResetRawClusters();
 170     virtual void   ResetRecTracks();
 171     virtual void   ResetRecTriggerTracks();
 172
 173                    /// Return tree with hits
 174     TTree*         TreeH() {return fLoader->TreeH(); }
 175                    /// Return tree with digits
 176     TTree*         TreeD() {return fLoader->TreeD(); }
 177                    /// Return tree with summable digits
 178     TTree*         TreeS() {return fLoader->TreeS(); }
 179                    /// Return tree with raw clusters
 180     TTree*         TreeR() {return fLoader->TreeR(); }
 181                    /// Return tree with tracks
 182     TTree*         TreeT() {return fLoader->TreeT(); }
 183                    /// Return tree with particles
 184     TTree*         TreeP() {return fLoader->TreeP(); }
 185
 186     //    TIterator* CreateDigitIterator(AliMUONData::EChamberIteration type);
 187
 188   protected:
 189     AliMUONData(const AliMUONData& rhs);
 190     AliMUONData& operator=(const AliMUONData& rhs);
 191
 192     AliLoader*      fLoader;  //!< Detector Loader pointer
 193     TClonesArray*   fHits;    ///< One event in treeH per primary track
 194     TObjArray*      fDigits;  ///< One event in treeD and one branch per detection plane
 195     TObjArray*      fSDigits; ///< One event in treeS and one branch per detection plane
 196     TObjArray*      fRawClusters; ///< One event in TreeR/Rawcluster and one branch per tracking detection plane
 197     TClonesArray*   fGlobalTrigger; ///< List of Global Trigger One event in TreeR/GlobalTriggerBranch
 198     TClonesArray*   fLocalTrigger;  ///< List of Local Trigger, One event in TreeR/LocalTriggerBranch
 199     TClonesArray*   fRegionalTrigger;  ///< List of Regional Trigger, One event in TreeR/LocalTriggerBranch
 200
 201     TClonesArray*   fRecTracks; ///< pointer to array of reconstructed tracks
 202     TClonesArray*   fRecTriggerTracks; ///< pointer to array of reconstructed trigger tracks
 203
 204     Int_t           fNhits;   //!< Number of Hits
 205     Int_t*          fNdigits; //!< Number of Digits
 206     Int_t*          fNSdigits;//!< Number of Digits
 207     Int_t*          fNrawclusters;  //!< Number of Raw Clusters
 208     Int_t           fNglobaltrigger;//!< Number of Global trigger
 209     Int_t           fNlocaltrigger; //!< Number of Local trigger
 210     Int_t           fNregionaltrigger; //!< Number of regional trigger
 211     Int_t           fNrectracks;    //!< Number of reconstructed tracks
 212     Int_t           fNrectriggertracks; //!< Number of reconstructed tracks
 213     Int_t           fSplitLevel;   ///< Splitting of branches 0 no spitting (root files are smaller) 1 splitting (larger output files)
 214
 215     mutable Int_t fCurrentEvent; ///< Current event we're dealing with
 216
 217 private:
 218
 219   ClassDef(AliMUONData,3) // Data accessor for MUON module
 220
 221 };
 222
 223
 224 #endif
 225