Merged Cvetans RowHoughTransformer, Anders latest developments in comp

[u/mrichter/AliRoot.git] / HLT / hough / AliL3HoughMaxFinder.h

// @(#) $Id$

#ifndef ALIL3_HOUGH_MaxFinder
#define ALIL3_HOUGH_MaxFinder

#include "AliL3RootTypes.h"

class AliL3Histogram;
class AliL3TrackArray;
class AliL3HoughTrack;
class TNtuple;

struct AxisWindow
{
  Int_t ymin;
  Int_t ymax;
  Int_t xbin;
  Int_t weight;
};

class AliL3HoughMaxFinder {
  
 private:

  Int_t fThreshold;
  AliL3Histogram *fCurrentHisto;  //!
  
  Float_t fGradX;
  Float_t fGradY;
  Float_t *fXPeaks; //!
  Float_t *fYPeaks; //!
  Int_t *fWeight;   //!
  Int_t fNPeaks;
  Int_t fNMax;
  
  Char_t fHistoType;

#ifndef no_root
  TNtuple *fNtuppel; //!
#endif

 public:
  AliL3HoughMaxFinder(); 
  AliL3HoughMaxFinder(Char_t *histotype,Int_t nmax,AliL3Histogram *hist=0);
  virtual ~AliL3HoughMaxFinder();
  void Reset();

  void CreateNtuppel();
  void WriteNtuppel(Char_t *filename);

  //Simple maxima finders:
  void FindAbsMaxima();
  void FindBigMaxima();
  void FindMaxima(Int_t threshold=0);
  void FindAdaptedPeaks(Int_t nkappawindow,Float_t cut_ratio);
  //Peak finder for HoughTransformerRow
  void FindAdaptedRowPeaks(Int_t kappawindow,Int_t xsize,Int_t ysize);
  //More sophisticated peak finders:
  void FindPeak(Int_t t1,Double_t t2,Int_t t3);
  void FindPeak1(Int_t y_window=2,Int_t x_bin_sides=1);
  void SortPeaks(struct AxisWindow **a,Int_t first,Int_t last);
  Int_t PeakCompare(struct AxisWindow *a,struct AxisWindow *b);
  
  //Setters:
  void SetGradient(Float_t x,Float_t y) {fGradX=x; fGradY=y;}
  void SetThreshold(Int_t f) {fThreshold = f;}
  void SetHistogram(AliL3Histogram *hist) {fCurrentHisto = hist;}
  
  //Getters:
  Float_t GetXPeak(Int_t i);
  Float_t GetYPeak(Int_t i);
  Int_t GetWeight(Int_t i);
  Int_t GetEntries() {return fNPeaks;}

  ClassDef(AliL3HoughMaxFinder,1) //Maximum finder class

};

inline Float_t AliL3HoughMaxFinder::GetXPeak(Int_t i)
{
  if(i<0 || i>fNMax)
    {
      STDCERR<<"AliL3HoughMaxFinder::GetXPeak : Invalid index "<<i<<STDENDL;
      return 0;
    }
  return fXPeaks[i];
}

inline Float_t AliL3HoughMaxFinder::GetYPeak(Int_t i)
{
  if(i<0 || i>fNMax)
    {
      STDCERR<<"AliL3HoughMaxFinder::GetYPeak : Invalid index "<<i<<STDENDL;
      return 0;
    }
  return fYPeaks[i];

}

inline Int_t AliL3HoughMaxFinder::GetWeight(Int_t i)
{
  if(i<0 || i>fNMax)
    {
      STDCERR<<"AliL3HoughMaxFinder::GetWeight : Invalid index "<<i<<STDENDL;
      return 0;
    }
  return fWeight[i];
}

#endif