AliMUONClusterFinderMLEM.h

Go to the documentation of this file.

#ifndef ALIMUONCLUSTERFINDERMLEM_H
#define ALIMUONCLUSTERFINDERMLEM_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

//  Author Alexander Zinchenko, JINR Dubna; Laurent Aphecetche, SUBATECH
//

class TH2D;
class TMinuit;

#ifndef ROOT_TObjArray
#  include "TObjArray.h"
#endif
#ifndef ROOT_TVector2
#  include "TVector2.h"
#endif

class AliMUONPad;

#include "AliMUONVClusterFinder.h"

class AliMUONClusterSplitterMLEM;

class AliMUONClusterFinderMLEM : public AliMUONVClusterFinder
{
public:
  AliMUONClusterFinderMLEM(Bool_t plot, AliMUONVClusterFinder* clusterFinder); // Constructor
  virtual ~AliMUONClusterFinderMLEM(); // Destructor

  virtual Bool_t NeedSegmentation() const { return kTRUE; }
  
  using AliMUONVClusterFinder::Prepare;

  virtual Bool_t Prepare(Int_t detElemId,
                         TObjArray* pads[2],
                         const AliMpArea& area,
                         const AliMpVSegmentation* segmentations[2]);
  
  virtual AliMUONCluster* NextCluster();
  
  virtual void SetChargeHints(Double_t lowestPadCharge, Double_t lowestClusterCharge);
  
  virtual void Print(Option_t* opt="") const;

  virtual void Paint(Option_t* opt="");

  // Status flags for pads

  static Int_t GetZeroFlag()       { return fgkZero; }
  static Int_t GetMustKeepFlag()   { return fgkMustKeep; }
  static Int_t GetUseForFitFlag()  { return fgkUseForFit; }
  static Int_t GetOverFlag()       { return fgkOver; }
  static Int_t GetModifiedFlag()   { return fgkModified; }
  static Int_t GetCoupledFlag()    { return fgkCoupled; }
  
private:
  AliMUONClusterFinderMLEM(const AliMUONClusterFinderMLEM& rhs);
  AliMUONClusterFinderMLEM& operator=(const AliMUONClusterFinderMLEM& rhs);

  Bool_t WorkOnPreCluster();

  AliMUONCluster* CheckPrecluster(const AliMUONCluster& cluster); 
  AliMUONCluster* CheckPreclusterTwoCathodes(AliMUONCluster* cluster); 
  
  Bool_t Overlap(const AliMUONPad& pad, const AliMUONPad& pixel); 
  
  void BuildPixArray(AliMUONCluster& cluster); 
  void BuildPixArrayOneCathode(AliMUONCluster& cluster); 
  void PadOverHist(Int_t idir, Int_t ix0, Int_t iy0, AliMUONPad *pad,
       TH2D *hist1, TH2D *hist2);

  void RemovePixel(Int_t i);
  
  AliMUONPad* Pixel(Int_t i) const;
  
  Bool_t MainLoop(AliMUONCluster& cluster, Int_t iSimple); // repeat MLEM algorithm until pixels become sufficiently small
  
  void   Mlem(AliMUONCluster& cluster, const Double_t *coef, Double_t *probi, Int_t nIter); // use MLEM for cluster finding
  
  void   FindCOG(Double_t *xyc); // find COG position around maximum bin
  Int_t  FindNearest(const AliMUONPad *pixPtr0); // find nearest neighbouring pixel to the given one

  Int_t FindLocalMaxima(TObjArray *pixArray, Int_t *localMax, Double_t *maxVal); // find local maxima 
  void  FlagLocalMax(TH2D *hist, Int_t i, Int_t j, Int_t *isLocalMax); // flag local max
  void  FindCluster(AliMUONCluster& cluster, const Int_t *localMax, Int_t iMax); // find cluster around local max
  void  AddVirtualPad(AliMUONCluster& cluster); // add virtual pads for some clusters (if necessary)
  
  void  PadsInXandY(AliMUONCluster& cluster, Int_t &nInX, Int_t &nInY) const; // get number of pads in X and Y

  void Simple(AliMUONCluster& cluster); 
  
  void Plot(const char* outputfile);
    
  void ComputeCoefficients(AliMUONCluster& cluster, 
                           Double_t* coef, Double_t* probi);
  
  void CheckOverlaps();
  void AddBinSimple(TH2D *mlem, Int_t ic, Int_t jc);
  void MaskPeaks(Int_t mask);

private:
  // Status flags for pads
  static const Int_t fgkZero; 
  static const Int_t fgkMustKeep; 
  static const Int_t fgkUseForFit; 
  static const Int_t fgkOver; 
  static const Int_t fgkModified; 
  static const Int_t fgkCoupled; 
      
  // Some constants
  static const Double_t fgkDistancePrecision; 
  static const TVector2 fgkIncreaseSize; 
  static const TVector2 fgkDecreaseSize; 
  
  AliMUONVClusterFinder* fPreClusterFinder; 
  AliMUONCluster* fPreCluster; 
  TObjArray fClusterList; 
  
  Int_t fEventNumber; 
  Int_t fDetElemId; 
  Int_t fClusterNumber; 
  
  const AliMpVSegmentation *fkSegmentation[2]; 
  
  //Int_t fCathBeg;               //!<! starting cathode (for combined cluster / track reco)
  //Int_t fPadBeg[2];             //!<! starting pads (for combined cluster / track reco)
  
  //static     TMinuit* fgMinuit; //!<! Fitter
  TH2D *fHistMlem; 
  TH2D *fHistAnode; 
  
  TObjArray* fPixArray; 
  Int_t fDebug; 
  Bool_t fPlot; 
  
  AliMUONClusterSplitterMLEM* fSplitter; 
  Int_t fNClusters; 
  Int_t fNAddVirtualPads; 
  
  Double_t fLowestPixelCharge; 
  Double_t fLowestPadCharge; 
  Double_t fLowestClusterCharge; 
  
  ClassDef(AliMUONClusterFinderMLEM,0) // cluster finder in MUON arm of ALICE
};

#endif