f8245b6edea733a5b9e1b51fa617bbeab7949dec
[u/mrichter/AliRoot.git] / MUON / AliMUONRejectList.h
1 #ifndef ALIMUONREJECTLIST_H
2 #define ALIMUONREJECTLIST_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
9 /// \ingroup rec
10 /// \class AliMUONRejectList
11 /// \brief Object to hold the list of elements we want to reject
12 /// from the reconstruction
13 /// 
14 // Author Laurent Aphecetche
15
16 #ifndef ROOT_TObject
17 #  include "TObject.h"
18 #endif
19
20 class AliMUONVStore;
21
22 class AliMUONRejectList : public TObject
23 {
24 public:
25   AliMUONRejectList(TRootIOCtor* ioCtor);
26   AliMUONRejectList();
27   AliMUONRejectList(const AliMUONRejectList& rl);
28   AliMUONRejectList& operator=(const AliMUONRejectList& rl);
29   virtual ~AliMUONRejectList();
30
31   /// Clone this object
32   virtual TObject* Clone(const char* /*name*/="") const { return new AliMUONRejectList(*this); }
33   
34   /// Whether or not all our probabilities are 0.0 or 1.0
35   Bool_t IsBinary() const { return fIsBinary; }
36   
37   Float_t DetectionElementProbability(Int_t detElemId) const;
38   Float_t BusPatchProbability(Int_t busPatchId) const;
39   Float_t ManuProbability(Int_t detElemId, Int_t manuId) const;
40   Float_t ChannelProbability(Int_t detElemId, Int_t manuId, Int_t manuChannel) const;
41   
42   void SetDetectionElementProbability(Int_t detElemId, Float_t proba=1.0);
43   void SetBusPatchProbability(Int_t busPatchId, Float_t proba=1.0);
44   void SetManuProbability(Int_t detElemId, Int_t manuId, Float_t proba=1.0);
45   void SetChannelProbability(Int_t detElemId, Int_t manuId, Int_t manuChannel, Float_t proba=1.0);
46   
47   void Print(Option_t* opt="") const;
48   
49 private:
50   void ZeroOrOne(Float_t proba);
51   
52 private:
53   
54   Bool_t fIsBinary; ///< Whether or not we only store zeros and ones for probabilities  
55
56   UInt_t fMaxNofDEs; ///< max number of detection elements (for allocation purposes)  
57   UInt_t fMaxNofBPs; ///< max number of bus patches (for allocation purposes)
58   UInt_t fMaxNofManus; ///< max number of manus (for allocation purposes)
59   
60   UInt_t fNofDEs; ///< actual number of detection elements for which we have probabilities
61   UInt_t fNofBPs; ///< actual number of bus patches for which we have probabilities
62   UInt_t fNofManus; ///< actual number of manus for which we have probabilities
63
64   /// array of detection element ids
65   UInt_t* fDEIds; //[fMaxNofDEs] 
66
67   /// array of probabilities of DEs
68   Float_t* fDEProbas; //[fMaxNofDEs] 
69
70   /// array of bus patch ids
71   UInt_t* fBPIds; //[fMaxNofBPs] 
72   
73   /// array of proba for bus patches
74   Float_t* fBPProbas; //[fMaxNofBPs]
75
76   /// array of manu ids
77   UInt_t* fManuIds; //[fMaxNofManus]
78   
79   /// array of proba for manus
80   Float_t* fManuProbas; //[fMaxNofManus]
81
82   AliMUONVStore* fChannels; ///< probabilities for all channels
83   
84   ClassDef(AliMUONRejectList,1) // (probabilistic) Reject list for MUON Tracker
85 };
86
87 #endif