HoughPlane.h

//-----------------------------------------------------------------------------
// $Id$
//-----------------------------------------------------------------------------
// Filename : HoughPlane.h
// Section  : TRG CDC
// Owner    : Yoshihito Iwasaki
// Email    : yoshihito.iwasaki@kek.jp
//-----------------------------------------------------------------------------
// Description : A class to represent a Hough parameter plane
//-----------------------------------------------------------------------------
// $Log$
//-----------------------------------------------------------------------------
 
#ifndef TRGCDCHoughPlane_FLAG_
#define TRGCDCHoughPlane_FLAG_
 
#include <string.h>
#include "trg/cdc/HoughPlaneBase.h"
 
#ifdef TRGCDC_SHORT_NAMES
#define TCHPlane TRGCDCHoughPlane
#endif
 
namespace Belle2 {
  class TRGCDCHoughPlane : public TRGCDCHoughPlaneBase {
 
  public:
 
    TRGCDCHoughPlane(const std::string& name,
                     const TRGCDCHoughTransformation& transformation,
                     unsigned nX,
                     float xMin,
                     float xMax,
                     unsigned nY,
                     float yMin,
                     float yMax);
 
    virtual ~TRGCDCHoughPlane();
 
  public:// Selectors
 
    unsigned entry(unsigned id) const override;
 
    unsigned entry(unsigned x, unsigned y) const override;
 
    int maxEntry(void) const override;
 
    virtual unsigned nActiveCellsInPattern(void) const;
 
  public:// Modifiers
 
    unsigned setEntry(unsigned serialId, unsigned n) override;
 
    void clear(void) override;
 
    void clearCells(void);
 
    void registerPattern(unsigned id) override;
 
    virtual void voteByPattern(float xOffset, int weight = 1);
 
  protected:
 
    void add(unsigned cellId, int weight) override;
 
  private:
 
    int* const _cell;
 
    unsigned* _patterns;
 
    unsigned _nPatterns;
 
    friend class TRGCDCHoughPlaneMulti;
  };
 
//-----------------------------------------------------------------------------
 
  inline
  unsigned
  TRGCDCHoughPlane::setEntry(unsigned serialId, unsigned n)
  {
    return _cell[serialId] = n;
  }
 
  inline
  void
  TRGCDCHoughPlane::clear(void)
  {
    memset(_cell, 0, nX() * nY() * sizeof(int));
    TRGCDCHoughPlaneBase::clear();
  }
 
  inline
  void
  TRGCDCHoughPlane::clearCells(void)
  {
    memset(_cell, 0, nX() * nY() * sizeof(int));
  }
 
  inline
  unsigned
  TRGCDCHoughPlane::entry(unsigned id) const
  {
    if (id < nX() * nY()) return _cell[id];
    return 0;
  }
 
  inline
  unsigned
  TRGCDCHoughPlane::entry(unsigned x, unsigned y) const
  {
    if (x < nX())
      if (y < nY())
        return _cell[nY() * x + y];
    return 0;
  }
 
  inline
  int
  TRGCDCHoughPlane::maxEntry(void) const
  {
    int m = 0;
    for (unsigned i = 0; i < nX() * nY(); i++)
      if (_cell[i] > m)
        m = _cell[i];
    return m;
  }
 
  inline
  unsigned
  TRGCDCHoughPlane::nActiveCellsInPattern(void) const
  {
    return _nPatterns;
  }
 
  inline
  void
  TRGCDCHoughPlane::add(unsigned a, int w)
  {
    _cell[a] += w;
    if (_cell[a] < 0)
      _cell[a] = 0;
  }
 
} // namespace Belle2
 
#endif