Fast Hough transformer using extensivle LUT for geometry and cos/sin functions.

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

#ifndef ALIL3_HISTOGRAM
#define ALIL3_HISTOGRAM

#include <stream.h>
#include "AliL3RootTypes.h"

#ifdef use_root
#include <TH2.h>
#endif

class AliL3Histogram {
  
 protected:
  
  Int_t *fContent; //!
  Char_t fName[100];
  Int_t fNxbins;
  Int_t fNybins;
  Int_t fNcells;
  Int_t fEntries;
  Int_t fFirstXbin;
  Int_t fFirstYbin;
  Int_t fLastXbin;
  Int_t fLastYbin;
  Int_t fThreshold;

  Double_t fXmin;
  Double_t fYmin;
  Double_t fXmax;
  Double_t fYmax;
  
#ifdef use_root
  TH2F *fRootHisto;
#endif  
  
 public:
  AliL3Histogram();
  AliL3Histogram(Char_t *name,Char_t *id,Int_t nxbin,Double_t xmin,Double_t xmax,Int_t nybin,Double_t ymin,Double_t ymax);
  virtual ~AliL3Histogram();
  
  void Reset();
  virtual void Fill(Double_t x,Double_t y,Int_t weight=1);
  virtual Int_t FindBin(Double_t x,Double_t y);
  virtual Int_t FindXbin(Double_t x);
  virtual Int_t FindYbin(Double_t y);
  Int_t GetBin(Int_t xbin,Int_t ybin);
  Int_t GetBinContent(Int_t bin);
  void SetBinContent(Int_t xbin,Int_t ybin,Int_t value);
  void SetBinContent(Int_t bin,Int_t value);
  void AddBinContent(Int_t xbin,Int_t ybin,Int_t weight);
  void AddBinContent(Int_t bin,Int_t weight);
  void Add(AliL3Histogram *h1,Double_t weight=1);
  void SetThreshold(Int_t i) {fThreshold = i;}
  void CreateRootHisto();
  virtual void Draw(Char_t *option="hist");
  virtual void Print() {};

#ifdef use_root
  TH2F *GetRootHisto();
#else
  void *GetRootHisto();
#endif
    
  Double_t GetXmin() {return fXmin;}
  Double_t GetXmax() {return fXmax;}
  Double_t GetYmin() {return fYmin;}
  Double_t GetYmax() {return fYmax;}
  virtual Double_t GetBinCenterX(Int_t xbin);
  virtual Double_t GetBinCenterY(Int_t ybin);
  Int_t GetFirstXbin() {return fFirstXbin;}
  Int_t GetLastXbin() {return fLastXbin;}
  Int_t GetFirstYbin() {return fFirstYbin;}
  Int_t GetLastYbin() {return fLastYbin;}
  Int_t GetNbinsX() {return fNxbins;}
  Int_t GetNbinsY() {return fNybins;}
  Int_t GetNEntries() {return fEntries;}
  
  
  ClassDef(AliL3Histogram,1) //2D histogram class
    
};

#ifdef use_root
inline TH2F *AliL3Histogram::GetRootHisto()
{
  if(!fRootHisto)
    {
      cerr<<"AliL3Histogram::GetRootHisto() : You must first Draw histogram before accessing it"<<endl;
      return 0;
    }
  else
    return fRootHisto;
}
#else
inline void *AliL3Histogram::GetRootHisto()
{
  cerr<<"AliL3Histogram::GetRootHisto() : You must compile with ROOT in order to interface the ROOT histogram"<<endl;
  return 0;
}
#endif

#endif