PndHoughSpace.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

//  PndHoughSpace
//  Defines a data structure for a hough space

#ifndef PndHoughSpace_H_
#define PndHoughSpace_H_

#include <map>
#include <set>
#include "TVector3.h"

struct PairCmp {
  bool operator()(const std::pair<int, int> &lhs, const std::pair<int, int> &rhs) const { return lhs.second >= rhs.second; }
};

class PndHoughSpace {

 public:
  PndHoughSpace();

  virtual ~PndHoughSpace() {}

  virtual void SetNBins1(double n) { fNbins1 = n; }
  virtual void SetNBins2(double n) { fNbins2 = n; }
  void Init();
  TVector3 GetCircleFromBin(Int_t bin);
  Int_t GetBin(Int_t binX, Int_t binY);
  void Fill(double x, double y);
  void FindMaxima(std::vector<TVector3> &FoundMaxima);
  void clear() { HoughSpaceMapBinToEntries.clear(); }
  Int_t GetEntries() { return HoughSpaceMapBinToEntries.size(); }
  std::vector<double> GetXBorder() { return bordersX; }
  std::vector<double> GetYBorder() { return bordersY; }

 private:
  std::map<int, int> HoughSpaceMapBinToEntries;

  std::vector<double> bordersX;
  std::vector<double> bordersY;

  double fNbins1;
  double fNbins2;

  Int_t fCounterNan;
  Int_t fCounterUnderflowXY;
  Int_t fCounterUnderflowXOverflowY;
  Int_t fCounterUnderflowX;
  Int_t fCounterOverflowXUnderflowY;
  Int_t fCounterOveflowXY;
  Int_t fCounterOverflowX;
  Int_t fCounterUnerflowY;
  Int_t fCounterOverflowY;

  ClassDef(PndHoughSpace, 1);
};

#endif /* PndHoughSpace_H_ */