MUON/AliMUONClusterFinderAZ.h

   1 #ifndef ALIMUONCLUSTERFINDERAZ_H
   2 #define ALIMUONCLUSTERFINDERAZ_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7 // Revision of includes 07/05/2004
   8
   9 /// \ingroup rec
  10 /// \class AliMUONClusterFinderAZ
  11 /// \brief Cluster finder in MUON arm of ALICE
  12
  13 #include "AliMUONClusterFinderVS.h"
  14
  15 class TH2D;
  16 class TClonesArray;
  17 class TMinuit;
  18 class TMatrixD;
  19
  20 class AliSegmentation;
  21 class AliMUONResponse;
  22 class AliMUONPixel;
  23 class AliMUONClusterDrawAZ;
  24
  25 class AliMUONClusterFinderAZ : public AliMUONClusterFinderVS
  26 {
  27 public:
  28   AliMUONClusterFinderAZ(Bool_t draw = 0); // Constructor
  29   virtual ~AliMUONClusterFinderAZ(); // Destructor
  30
  31   void     FindRawClusters(); // the same interface as for old cluster finder
  32   void     EventLoop(Int_t nev = 0, Int_t ch = 0); // first event
  33   Bool_t   TestTrack(Int_t t) const; // test if track was selected
  34   Int_t    GetNPads(Int_t cath) const { return fnPads[cath]; }
  35   Int_t    GetIJ(Int_t indx, Int_t iPad) const { return fPadIJ[indx][iPad]; }
  36   Float_t  GetXyq(Int_t indx, Int_t iPad) const { return fXyq[indx][iPad]; }
  37   Float_t  GetZpad() { return fZpad; }
  38   Bool_t GetUsed(Int_t cath, Int_t dig) const { return fUsed[cath][dig]; }
  39   void SetUsed(Int_t cath, Int_t dig) { fUsed[cath][dig] = kTRUE; } // mark used digits
  40   void SetUnused(Int_t cath, Int_t dig) { fUsed[cath][dig] = kFALSE; } // unmark digits
  41   void SetReco(Int_t iReco) { fReco = iReco; } // set reco flag
  42   void SetStart(Int_t iCath, Int_t iPad) { fCathBeg = iCath; fPadBeg[0] = fPadBeg[1] = 0; fPadBeg[fCathBeg] = iPad; } // start
  43
  44 protected:
  45   AliMUONClusterFinderAZ(const AliMUONClusterFinderAZ& rhs);
  46   AliMUONClusterFinderAZ& operator=(const AliMUONClusterFinderAZ& rhs);
  47
  48  private:
  49   // Some constants
  50   static const Int_t fgkDim = 10000; // array size
  51   static const Double_t fgkCouplMin; // threshold on coupling
  52
  53   static  AliMUONClusterFinderAZ* fgClusterFinder; // the ClusterFinderAZ instance
  54
  55   Int_t      fnPads[2];         // ! number of pads in the cluster on 2 cathodes
  56   Float_t    fXyq[7][fgkDim];   // ! pad information
  57   Int_t      fPadIJ[2][fgkDim]; // ! pad information
  58   AliMUONGeometrySegmentation *fSegmentation[2]; // ! new segmentation
  59   AliMUONResponse *fResponse;   // ! response
  60   Float_t    fZpad;             // ! z-coordinate of the hit
  61   Int_t      fNpar;             // ! number of fit parameters
  62   Double_t   fQtot;             // ! total cluster charge
  63   Int_t      fReco;             // ! !=0 if run reco with writing of reconstructed clusters
  64   Int_t fCathBeg;               // ! starting cathode (for combined cluster / track reco)
  65   Int_t fPadBeg[2];             // ! starting pads (for combined cluster / track reco)
  66
  67   static     TMinuit* fgMinuit; // ! Fitter
  68   Bool_t     fUsed[2][fgkDim]; // ! flags for used pads
  69   //TH2F*      fHist[4]; // ! histograms
  70   AliMUONClusterDrawAZ *fDraw; // ! drawing object
  71   //Int_t      fnMu; // ! number of muons passing thru the selected area
  72   //Double_t   fxyMu[2][7]; // ! muon information
  73   TObjArray* fPixArray; // ! collection of pixels
  74   Int_t fnCoupled; // ! number of coupled clusters in precluster
  75   Int_t fDebug; // ! debug level
  76
  77   // Functions
  78
  79   void   AddPad(Int_t cath, Int_t digit); // add a pad to the cluster
  80   Bool_t Overlap(Int_t cath, AliMUONDigit *dig); // check if the pad from one cathode overlaps with a pad in the cluster on the other cathode
  81   Bool_t Overlap(Float_t *xy1, Int_t iPad, Float_t *xy12, Int_t iSkip); // check if pads xy1 and iPad overlap and return overlap area
  82   Bool_t CheckPrecluster(Int_t *nShown); // check precluster to simplify it (if possible)
  83   void   BuildPixArray(); // build array of pixels
  84   void   AdjustPixel(Float_t width, Int_t ixy); // adjust size of small pixels
  85   void   AdjustPixel(Float_t wxmin, Float_t wymin); // adjust size of large pixels
  86   Bool_t MainLoop(Int_t iSimple); // repeat MLEM algorithm until pixels become sufficiently small
  87   void   Mlem(Double_t *coef, Double_t *probi, Int_t nIter); // use MLEM for cluster finding
  88   void   FindCOG(TH2D *mlem, Double_t *xyc); // find COG position around maximum bin
  89   Int_t  FindNearest(AliMUONPixel *pixPtr0); // find nearest neighbouring pixel to the given one
  90   void   Split(TH2D *mlem, Double_t *coef); // steering function for pixels
  91   void   AddBin(TH2D *mlem, Int_t ic, Int_t jc, Int_t mode, Bool_t* used, TObjArray *pix); // add a bin to the cluster
  92   TObject* BinToPix(TH2D *mlem, Int_t jc, Int_t ic); // hist. bin-to-pixel
  93   void   AddCluster(Int_t ic, Int_t nclust, TMatrixD *aijcluclu, Bool_t *used, Int_t *clustNumb, Int_t &nCoupled); // add a cluster to the group of coupled clusters
  94   Double_t MinGroupCoupl(Int_t nCoupled, Int_t *clustNumb, TMatrixD *aijcluclu, Int_t *minGroup); // find group of cluster with min. coupling to others
  95   Int_t  SelectPad(Int_t nCoupled, Int_t nForFit, Int_t *clustNumb, Int_t *clustFit, TMatrixD *aijcluclu); //select pads for fit
  96   void   Merge(Int_t nForFit, Int_t nCoupled, Int_t *clustNumb, Int_t *clustFit, TObjArray **clusters, TMatrixD *aijcluclu, TMatrixD *aijclupad); // merge clusters
  97   Int_t  Fit(Int_t nfit, Int_t *clustFit, TObjArray **clusters, Double_t *parOk); // do the fitting
  98   void  UpdatePads(Int_t nfit, Double_t *par); // subtract fitted charges from pads
  99   void  AddRawCluster(Double_t x, Double_t y, Double_t qTot, Double_t fmin, Int_t nfit, Int_t *tracks, Double_t sigx, Double_t sigy, Double_t dist); // add new reconstructed cluster
 100   Int_t FindLocalMaxima(Int_t *localMax, Double_t *maxVal); // find local maxima
 101   void  FlagLocalMax(TH2D *hist, Int_t i, Int_t j, Int_t *isLocalMax); // flag local max
 102   void  FindCluster(Int_t *localMax, Int_t iMax); // find cluster around local max
 103   void  AddVirtualPad(); // add virtual pads for some clusters (if necessary)
 104   void  PadsInXandY(Int_t &nInX, Int_t &nInY); // get number of pads in X and Y
 105   // This function is used for fitting
 106   void  Fcn1(Int_t &npar, Double_t *gin, Double_t &f, Double_t *par, Int_t iflag);
 107   void Simple(); // process simple cluster
 108
 109   void Errors(AliMUONRawCluster *clus); // correct coordinates and eval. errors
 110   void Errors(Int_t ny, Int_t nx, Int_t iby, Int_t ibx, Double_t fmin,
 111               Double_t wy, Double_t wx, Int_t iover,
 112               Double_t dyc, Double_t dxc, Double_t qtot,
 113               Double_t &yrec, Double_t &xrec, Double_t &erry, Double_t &errx);
 114
 115   // Dummy methods for overloading warnings
 116   void FindCluster(int, int, int, AliMUONRawCluster&) {return;}
 117   void FindLocalMaxima(AliMUONRawCluster*) {return;}
 118   void Split(AliMUONRawCluster*) {return;}
 119   void AddRawCluster(AliMUONRawCluster&) {return;}
 120
 121 ClassDef(AliMUONClusterFinderAZ,0) // cluster finder in MUON arm of ALICE
 122 };
 123
 124 #endif