Some additional protections
[u/mrichter/AliRoot.git] / 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 AliMUONGlobalTrigger;
39
40 //__________________________________________________________________
41 /////////////////////////////////////////////////////////////////////
42 //                                                                 //
43 //  class AliMUONData                                              //
44 //                                                                 //
45 /////////////////////////////////////////////////////////////////////
46
47 class AliMUONData : public TNamed 
48 {
49   public:
50   
51   //  enum EChamberIteration { kAllChambers, kTrackingChambers, kTriggerChambers };
52   
53     AliMUONData();
54     AliMUONData(AliLoader * loader, const char* name, const char* title);
55     virtual ~AliMUONData();  
56     virtual void   AddDigit(Int_t id, Int_t* tracks, Int_t* charges,
57                              Int_t* digits); 
58     virtual void   AddSDigit(Int_t id, Int_t* tracks, Int_t* charges,
59                              Int_t* digits); 
60     virtual void   AddDigit(Int_t id, const AliMUONDigit& digit); // use copy constructor
61     virtual void   AddSDigit(Int_t id, const AliMUONDigit& digit); // use copy constructor
62     virtual void   AddHit(Int_t fIshunt, Int_t track, Int_t iChamber, 
63                           Int_t idpart, Float_t X, Float_t Y, Float_t Z, 
64                           Float_t tof, Float_t momentum, Float_t theta, 
65                           Float_t phi, Float_t length, Float_t destep, 
66                           Float_t Xref,Float_t Yref,Float_t Zref);
67                           // TBR
68     virtual void   AddHit2(Int_t fIshunt, Int_t track, Int_t detElemId, 
69                           Int_t idpart, Float_t X, Float_t Y, Float_t Z, 
70                           Float_t tof, Float_t momentum, Float_t theta, 
71                           Float_t phi, Float_t length, Float_t destep, 
72                           Float_t Xref,Float_t Yref,Float_t Zref);
73     
74     virtual void   AddGlobalTrigger(const AliMUONGlobalTrigger& trigger); // use copy constructor
75
76     virtual void   AddLocalTrigger(const AliMUONLocalTrigger& trigger); // use copy constructor
77
78     virtual void   AddRawCluster(Int_t id, const AliMUONRawCluster& clust);
79     virtual void   AddRecTrack(const AliMUONTrack& track);
80     virtual void   AddRecTriggerTrack(const AliMUONTriggerTrack& triggertrack);
81
82     TClonesArray*  Hits() {return fHits;} ///< Return hits
83     TClonesArray*  Digits(Int_t DetectionPlane) const;
84     TClonesArray*  SDigits(Int_t DetectionPlane) const;
85     TClonesArray*  LocalTrigger() const;
86     TClonesArray*  GlobalTrigger() const;    
87     TClonesArray*  RawClusters(Int_t DetectionPlane);
88     
89                     /// Return reconstructed tracks
90     TClonesArray*  RecTracks() {return fRecTracks;} 
91                    /// Return reconstructed trigger tracks
92     TClonesArray*  RecTriggerTracks() {return fRecTriggerTracks;}
93
94                    /// Load hits for \a i th entry in hits three
95     void           GetTrack(Int_t it) const  {
96       if (fLoader && fLoader->TreeH())
97         fLoader->TreeH()->GetEvent(it);
98     }
99                    /// Get number of entries in hits three
100     Int_t          GetNtracks() const      {
101       Int_t ntrk = 0;
102       if (fLoader && fLoader->TreeH())
103         ntrk = (Int_t) fLoader->TreeH()->GetEntries();
104       return ntrk;
105     }
106     void           GetDigits() const;
107                    /// Load sdigits tree
108     void           GetSDigits() const {
109       if (fLoader && fLoader->TreeS())
110         fLoader->TreeS()->GetEvent(0);
111     }
112                    /// Load raw clusters tree
113     void           GetRawClusters() const {
114       if (fLoader && fLoader->TreeR())
115         fLoader->TreeR()->GetEvent(0);
116     }
117                    /// Load trigger tree
118     void           GetTrigger() const {
119       if (fLoader && fLoader->TreeR())
120         fLoader->TreeR()->GetEvent(0);
121     }
122                    /// Load trigger D tree
123     void           GetTriggerD() const {
124       if (fLoader && fLoader->TreeD())
125         fLoader->TreeD()->GetEvent(0);
126     }
127                    /// Return split level
128     Int_t          GetSplitLevel() const {return fSplitLevel;}
129                    /// Return reconstructed tracks
130     void           GetRecTracks() const {
131       if (fLoader && fLoader->TreeT())
132         fLoader->TreeT()->GetEvent(0);
133     }
134                    /// Return reconstructed trigger tracks
135     void           GetRecTriggerTracks() const {
136       if (fLoader && fLoader->TreeT())
137         fLoader->TreeT()->GetEvent(0);
138     }
139
140     Bool_t        IsRawClusterBranchesInTree();
141     Bool_t        IsDigitsBranchesInTree();
142     Bool_t        IsTriggerBranchesInTree();
143     Bool_t        IsTriggerBranchesInTreeD();
144     Bool_t        IsTrackBranchesInTree();
145     Bool_t        IsTriggerTrackBranchesInTree();
146
147                        /// Get loader
148     virtual AliLoader* GetLoader() const { return fLoader; }
149                        /// Set loader
150     virtual void       SetLoader(AliLoader * loader) {fLoader=loader;}    
151     
152     virtual void   Fill(Option_t* opt=" ");
153     virtual void   MakeBranch(Option_t *opt=" ");
154     virtual void   SetTreeAddress(Option_t *opt=" ");
155     
156                    /// Set split level
157     void           SetSplitLevel(Int_t SplitLevel) {fSplitLevel=SplitLevel;}
158     
159     virtual void Print(Option_t* opt="") const;
160     
161     virtual void   ResetHits();
162     virtual void   ResetDigits();
163     virtual void   ResetSDigits();
164     virtual void   ResetTrigger();
165     virtual void   ResetRawClusters();
166     virtual void   ResetRecTracks();
167     virtual void   ResetRecTriggerTracks();
168   
169                    /// Return tree with hits
170     TTree*         TreeH() {return fLoader->TreeH(); }
171                    /// Return tree with digits
172     TTree*         TreeD() {return fLoader->TreeD(); }
173                    /// Return tree with summable digits
174     TTree*         TreeS() {return fLoader->TreeS(); }
175                    /// Return tree with raw clusters
176     TTree*         TreeR() {return fLoader->TreeR(); }
177                    /// Return tree with tracks
178     TTree*         TreeT() {return fLoader->TreeT(); }
179                    /// Return tree with particles
180     TTree*         TreeP() {return fLoader->TreeP(); }
181
182     //    TIterator* CreateDigitIterator(AliMUONData::EChamberIteration type);
183     
184   protected: 
185     AliMUONData(const AliMUONData& rhs);
186     AliMUONData& operator=(const AliMUONData& rhs);
187
188     AliLoader*      fLoader;  //!< Detector Loader pointer
189     TClonesArray*   fHits;    ///< One event in treeH per primary track
190     TObjArray*      fDigits;  ///< One event in treeD and one branch per detection plane
191     TObjArray*      fSDigits; ///< One event in treeS and one branch per detection plane
192     TObjArray*      fRawClusters; ///< One event in TreeR/Rawcluster and one branch per tracking detection plane
193     TClonesArray*   fGlobalTrigger; ///< List of Global Trigger One event in TreeR/GlobalTriggerBranch
194     TClonesArray*   fLocalTrigger;  ///< List of Local Trigger, One event in TreeR/LocalTriggerBranch
195     TClonesArray*   fRecTracks; ///< pointer to array of reconstructed tracks
196     TClonesArray*   fRecTriggerTracks; ///< pointer to array of reconstructed trigger tracks
197
198     Int_t           fNhits;   //!< Number of Hits
199     Int_t*          fNdigits; //!< Number of Digits
200     Int_t*          fNSdigits;//!< Number of Digits
201     Int_t*          fNrawclusters;  //!< Number of Raw Clusters
202     Int_t           fNglobaltrigger;//!< Number of Global trigger
203     Int_t           fNlocaltrigger; //!< Number of Local trigger
204     Int_t           fNrectracks;    //!< Number of reconstructed tracks
205     Int_t           fNrectriggertracks; //!< Number of reconstructed tracks
206     Int_t           fSplitLevel;   ///< Splitting of branches 0 no spitting (root files are smaller) 1 splitting (larger output files)
207
208     mutable Int_t fCurrentEvent; ///< Current event we're dealing with
209     
210 private:  
211
212   ClassDef(AliMUONData,3) // Data accessor for MUON module
213       
214 };
215
216
217 #endif
218