CbmRichRingLight.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CbmRichRingLight.h

/*
 * CbmRichRingLight.h
 *
 *  Created on: 09.03.2010
 *  Author: Semen Lebedev
 */

#ifndef CBMRICHRINGLIGHT_H_
#define CBMRICHRINGLIGHT_H_

#include <vector>
#include <cmath>

using std::vector;

class CbmRichHitLight
{
public:
   CbmRichHitLight():
      fX(0.),
      fY(0.){}

   virtual ~CbmRichHitLight(){}

   CbmRichHitLight(
         float x,
         float y):
      fX(x),
      fY(y) {}

  float fX; // x coordinate of the hit
  float fY; // y coordinate of the hit
};

class CbmRichRingLight
{
public:

   CbmRichRingLight():
      fHits(),
      fHitIds(),

      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(30);
      fHitIds.reserve(30);
   }

   virtual ~CbmRichRingLight()
   {
      fHits.clear();
      fHitIds.clear();
   }

   void AddHit(
         CbmRichHitLight hit,
         unsigned short id = -1)
   {
      fHits.push_back(hit);
      fHitIds.push_back(id);
   }

   bool RemoveHit(
         int hitId)
   {
      vector<unsigned short>::iterator it;
      for (it = fHitIds.begin(); it!= fHitIds.end(); it++){
         if (hitId == *it){
            fHitIds.erase(it);
            return true;
         }
      }
      return false;
   }

  int GetNofHits() const {return fHitIds.size(); }

  CbmRichHitLight GetHit(int ind) {return fHits[ind];}

  unsigned short GetHitId(int ind) {return fHitIds[ind];}

  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:
   vector<CbmRichHitLight> fHits; // STL container for CbmRichHitLight
   vector<unsigned short> fHitIds; // STL container for hit indexes

   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_ */