CbmRichRingLight.h

Go to the documentation of this file.

/*
 * CbmRichRingLight.h
 *
 *  Created on: 09.03.2010
 *  Author: Semen Lebedev
 */
 
#ifndef CBMRICHRINGLIGHT_H_
#define CBMRICHRINGLIGHT_H_
 
#include <cmath>
#include <vector>
 
class CbmRichHitLight {
public:
  CbmRichHitLight() : fX(0.), fY(0.), fId(0) {}
 
  virtual ~CbmRichHitLight() {}
 
  CbmRichHitLight(float x, float y, unsigned int hitId = 0)
    : fX(x), fY(y), fId(hitId) {}
 
  float fX;          // x coordinate of the hit
  float fY;          // y coordinate of the hit
  unsigned int fId;  // hit id
};
 
class CbmRichRingLight {
public:
  CbmRichRingLight()
    : fHits()
    ,
 
    fCenterX(0.f)
    , fCenterY(0.f)
    , fRadius(0.f)
    ,
 
    fAaxis(0.f)
    , fBaxis(0.f)
    , fPhi(0.f)
    ,
 
    fAPar(0.f)
    , fBPar(0.f)
    , fCPar(0.f)
    , fDPar(0.f)
    , fEPar(0.f)
    , fFPar(0.f)
    ,
 
    fRecFlag(0)
    ,
 
    fChi2(0.f)
    , fAngle(0.f)
    , fNofHitsOnRing(0)
    , fSelectionNN(0.f)
 
  {
    fHits.reserve(40);
  }
 
  virtual ~CbmRichRingLight() { fHits.clear(); }
 
  void AddHit(CbmRichHitLight hit) { fHits.push_back(hit); }
 
  bool RemoveHit(int hitId) {
    std::vector<CbmRichHitLight>::iterator it;
    for (it = fHits.begin(); it != fHits.end(); it++) {
      if (hitId >= 0 && hitId == int(it->fId)) {
        fHits.erase(it);
        return true;
      }
    }
    return false;
  }
 
  int GetNofHits() const { return fHits.size(); }
 
  CbmRichHitLight GetHit(int ind) { return fHits[ind]; }
 
  unsigned int GetHitId(int ind) { return fHits[ind].fId; }
 
  void SetCenterX(float x) { fCenterX = x; }
  void SetCenterY(float y) { fCenterY = y; }
  void SetRadius(float r) { fRadius = r; }
 
  void SetXYR(float x, float y, float r) {
    fCenterX = x;
    fCenterY = y;
    fRadius  = r;
  }
 
  void SetXYABP(float x, float y, float a, float b, float p) {
    fCenterX = x;
    fCenterY = y;
    fAaxis   = a;
    fBaxis   = b;
    fPhi     = p;
  }
 
  void SetChi2(float chi2) { fChi2 = chi2; }
  void SetAngle(float angle) { fAngle = angle; }
  void SetNofHitsOnRing(unsigned short onring) { fNofHitsOnRing = onring; }
  void SetSelectionNN(float selectionNN) { fSelectionNN = selectionNN; }
 
  float GetCenterX() const { return fCenterX; }
  float GetCenterY() const { return fCenterY; }
  float GetRadius() const { return fRadius; }
 
  float GetAaxis() const { return fAaxis; }
  float GetBaxis() const { return fBaxis; }
  float GetPhi() const { return fPhi; }
  void SetAaxis(double a) { fAaxis = a; }
  void SetBaxis(double b) { fBaxis = b; }
  void SetPhi(double phi) { fPhi = phi; }
 
  double GetXF1() const {
    double c  = sqrt(fAaxis * fAaxis - fBaxis * fBaxis);
    double xc = c * cos(fabs(fPhi));
 
    return fCenterX + xc;
  }
 
  double GetYF1() const {
    double c  = sqrt(fAaxis * fAaxis - fBaxis * fBaxis);
    double yc = c * sin(fabs(fPhi));
    if (fPhi >= 0) {
      return fCenterY + yc;
    } else {
      return fCenterY - yc;
    }
  }
 
  double GetXF2() const {
    double c  = sqrt(fAaxis * fAaxis - fBaxis * fBaxis);
    double xc = c * cos(fabs(fPhi));
 
    return fCenterX - xc;
  }
 
  double GetYF2() const {
    double c  = sqrt(fAaxis * fAaxis - fBaxis * fBaxis);
    double yc = c * sin(fabs(fPhi));
    if (fPhi >= 0) {
      return fCenterY - yc;
    } else {
      return fCenterY + yc;
    }
  }
 
  void SetABCDEF(float a, float b, float c, float d, float e, float f) {
    fAPar = a;
    fBPar = b;
    fCPar = c;
    fDPar = d;
    fEPar = e;
    fFPar = f;
  }
 
  float GetAPar() const { return fAPar; }
  float GetBPar() const { return fBPar; }
  float GetCPar() const { return fCPar; }
  float GetDPar() const { return fDPar; }
  float GetEPar() const { return fEPar; }
  float GetFPar() const { return fFPar; }
 
  float GetSelectionNN() const { return fSelectionNN; }
  float GetChi2() const { return fChi2; }
 
  float GetRadialPosition() const {
    if (fCenterY > 0.f) {
      return sqrt(fCenterX * fCenterX
                  + (fCenterY - 110.f) * (fCenterY - 110.f));
    } else {
      return sqrt(fCenterX * fCenterX
                  + (fCenterY + 110.f) * (fCenterY + 110.f));
    };
    return 0.;
  }
 
  float GetAngle() const { return fAngle; }
 
  unsigned short GetNofHitsOnRing() const { return fNofHitsOnRing; }
 
  int GetRecFlag() const { return fRecFlag; }
  void SetRecFlag(int r) { fRecFlag = r; }
 
private:
  std::vector<CbmRichHitLight> fHits;  // STL container for CbmRichHitLight
 
  float fCenterX;
  float fCenterY;
  float fRadius;
 
  float fAaxis;
  float fBaxis;
  float fPhi;
 
  float fAPar;  // Axx+Bxy+Cyy+Dx+Ey+F
  float fBPar;
  float fCPar;
  float fDPar;
  float fEPar;
  float fFPar;
 
  int fRecFlag;
 
  float fChi2;
  float fAngle;
  unsigned short fNofHitsOnRing;
  float fSelectionNN;
};
 
#endif /* CBMRICHRINGLIGHT_H_ */