1 #ifndef ALIMUONCLUSTERFINDERMLEM_H
2 #define ALIMUONCLUSTERFINDERMLEM_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
9 /// \class AliMUONClusterFinderMLEM
10 /// \brief Cluster finder in MUON arm of ALICE
12 // Author Alexander Zinchenko, JINR Dubna; Laurent Aphecetche, SUBATECH
19 #ifndef ROOT_TObjArray
20 # include "TObjArray.h"
23 # include "TVector2.h"
28 #include "AliMUONVClusterFinder.h"
30 class AliMUONClusterSplitterMLEM;
32 class AliMUONClusterFinderMLEM : public AliMUONVClusterFinder
35 AliMUONClusterFinderMLEM(Bool_t plot, AliMUONVClusterFinder* clusterFinder); // Constructor
36 virtual ~AliMUONClusterFinderMLEM(); // Destructor
38 virtual Bool_t NeedSegmentation() const { return kTRUE; }
40 using AliMUONVClusterFinder::Prepare;
42 virtual Bool_t Prepare(Int_t detElemId,
43 TClonesArray* pads[2],
44 const AliMpArea& area,
45 const AliMpVSegmentation* segmentations[2]);
47 virtual AliMUONCluster* NextCluster();
49 virtual void SetChargeHints(Double_t lowestPadCharge, Double_t lowestClusterCharge);
51 virtual void Print(Option_t* opt="") const;
53 virtual void Paint(Option_t* opt="");
55 // Status flags for pads
57 /// Return pad "basic" state flag
58 static Int_t GetZeroFlag() { return fgkZero; }
59 /// Return do not kill flag
60 static Int_t GetMustKeepFlag() { return fgkMustKeep; }
61 /// Return should be used for fit flag
62 static Int_t GetUseForFitFlag() { return fgkUseForFit; }
63 /// Return processing is over flag
64 static Int_t GetOverFlag() { return fgkOver; }
65 /// Return modified pad charge flag
66 static Int_t GetModifiedFlag() { return fgkModified; }
67 /// Return coupled pad flag
68 static Int_t GetCoupledFlag() { return fgkCoupled; }
72 AliMUONClusterFinderMLEM(const AliMUONClusterFinderMLEM& rhs);
74 AliMUONClusterFinderMLEM& operator=(const AliMUONClusterFinderMLEM& rhs);
76 Bool_t WorkOnPreCluster();
78 /// Check precluster to simplify it (if possible), and return the simplified cluster
79 AliMUONCluster* CheckPrecluster(const AliMUONCluster& cluster);
80 AliMUONCluster* CheckPreclusterTwoCathodes(AliMUONCluster* cluster);
82 /// Checks whether a pad and a pixel have an overlapping area.
83 Bool_t Overlap(const AliMUONPad& pad, const AliMUONPad& pixel);
85 /// build array of pixels
86 void BuildPixArray(AliMUONCluster& cluster);
87 void BuildPixArrayOneCathode(AliMUONCluster& cluster);
88 void PadOverHist(Int_t idir, Int_t ix0, Int_t iy0, AliMUONPad *pad,
89 TH2D *hist1, TH2D *hist2);
91 void RemovePixel(Int_t i);
93 AliMUONPad* Pixel(Int_t i) const;
95 Bool_t MainLoop(AliMUONCluster& cluster, Int_t iSimple); // repeat MLEM algorithm until pixels become sufficiently small
97 void Mlem(AliMUONCluster& cluster, const Double_t *coef, Double_t *probi, Int_t nIter); // use MLEM for cluster finding
99 void FindCOG(Double_t *xyc); // find COG position around maximum bin
100 Int_t FindNearest(const AliMUONPad *pixPtr0); // find nearest neighbouring pixel to the given one
102 Int_t FindLocalMaxima(TObjArray *pixArray, Int_t *localMax, Double_t *maxVal); // find local maxima
103 void FlagLocalMax(TH2D *hist, Int_t i, Int_t j, Int_t *isLocalMax); // flag local max
104 void FindCluster(AliMUONCluster& cluster, const Int_t *localMax, Int_t iMax); // find cluster around local max
105 void AddVirtualPad(AliMUONCluster& cluster); // add virtual pads for some clusters (if necessary)
107 void PadsInXandY(AliMUONCluster& cluster, Int_t &nInX, Int_t &nInY) const; // get number of pads in X and Y
109 /// Process simple cluster
110 void Simple(AliMUONCluster& cluster);
112 void Plot(const char* outputfile);
114 void ComputeCoefficients(AliMUONCluster& cluster,
115 Double_t* coef, Double_t* probi);
117 void CheckOverlaps();
118 void AddBinSimple(TH2D *mlem, Int_t ic, Int_t jc);
119 void MaskPeaks(Int_t mask);
122 // Status flags for pads
123 static const Int_t fgkZero; ///< pad "basic" state
124 static const Int_t fgkMustKeep; ///< do not kill (for pixels)
125 static const Int_t fgkUseForFit; ///< should be used for fit
126 static const Int_t fgkOver; ///< processing is over
127 static const Int_t fgkModified; ///< modified pad charge
128 static const Int_t fgkCoupled; ///< coupled pad
131 static const Double_t fgkDistancePrecision; ///< used to check overlaps and so on
132 static const TVector2 fgkIncreaseSize; ///< idem
133 static const TVector2 fgkDecreaseSize; ///< idem
135 AliMUONVClusterFinder* fPreClusterFinder; //!< the pre-clustering worker
136 AliMUONCluster* fPreCluster; //!< current pre-cluster
137 TObjArray fClusterList; //!< clusters corresponding to the current pre-cluster
139 Int_t fEventNumber; //!< current event being processed
140 Int_t fDetElemId; //!< current DE being processed
141 Int_t fClusterNumber; //!< current cluster number
143 const AliMpVSegmentation *fkSegmentation[2]; //!< new segmentation
145 //Int_t fCathBeg; //!< starting cathode (for combined cluster / track reco)
146 //Int_t fPadBeg[2]; //!< starting pads (for combined cluster / track reco)
148 //static TMinuit* fgMinuit; //!< Fitter
149 TH2D *fHistMlem; //!< histogram for MLEM procedure
150 TH2D *fHistAnode; //!< histogram for local maxima search
152 TObjArray* fPixArray; //!< collection of pixels
153 Int_t fDebug; //!< debug level
154 Bool_t fPlot; //!< whether we should plot thing (for debug only, quite slow!)
156 AliMUONClusterSplitterMLEM* fSplitter; //!< helper class to go from pixel arrays to clusters
157 Int_t fNClusters; //!< total number of clusters
158 Int_t fNAddVirtualPads; //!< number of clusters for which we added virtual pads
160 Double_t fLowestPixelCharge; //!< see AliMUONRecoParam
161 Double_t fLowestPadCharge; //!< see AliMUONRecoParam
162 Double_t fLowestClusterCharge; //!< see AliMUONRecoParam
164 ClassDef(AliMUONClusterFinderMLEM,0) // cluster finder in MUON arm of ALICE