CbmRichRingFitterEllipseBase.h

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

#ifndef CBM_RICH_RING_FITTER_ELLIPSE_BASE
#define CBM_RICH_RING_FITTER_ELLIPSE_BASE

#include "CbmRichRingFitterBase.h"

class CbmRichRingFitterEllipseBase: public CbmRichRingFitterBase {
public:
   CbmRichRingFitterEllipseBase(){ };

   virtual ~CbmRichRingFitterEllipseBase(){ };

protected:

   virtual void CalcChi2(
         CbmRichRingLight* ring)
   {
      int nofHits = ring->GetNofHits();
      if (nofHits <= 5){
         ring->SetChi2(-1.);
         return;
      }

      double axisA = ring->GetAaxis();
      double axisB = ring->GetBaxis();

      if (axisA < axisB){
         ring->SetChi2(-1.);
         return;
      }

      // calculate ellipse focuses
      double xf1 = ring->GetXF1();
      double yf1 = ring->GetYF1();
      double xf2 = ring->GetXF2();
      double yf2 = ring->GetYF2();

      // calculate chi2
      double chi2 = 0.;
      for(int iHit = 0; iHit < nofHits; iHit++){
         double x = ring->GetHit(iHit).fX;
         double y = ring->GetHit(iHit).fY;

         double d1 = sqrt( (x-xf1)*(x-xf1) + (y-yf1)*(y-yf1) );
         double d2 = sqrt( (x-xf2)*(x-xf2) + (y-yf2)*(y-yf2) );

         chi2 += (d1 + d2 - 2.*axisA)*(d1 + d2 - 2.*axisA);
      }
      ring->SetChi2(chi2);
   }

   virtual void CalcChi2(
         double A,
         double B,
         double C,
         double D,
         double E,
         double F,
         CbmRichRingLight* ring)
   {
      int nofHits =  ring->GetNofHits();
      if (nofHits <= 5){
         ring->SetChi2(-1.);
         return;
      }
      double chi2 = 0.;
      for(int iHit = 0; iHit < nofHits; iHit++){
         double x = ring->GetHit(iHit).fX;
         double y = ring->GetHit(iHit).fY;

         double d1 = fabs(A*x*x + B*x*y + C*y*y + D*x + E*y + F);
         double d2 = sqrt( pow(2*A*x + B*y + D, 2) + pow(B*x + 2*C*y + E, 2) );

         chi2 += (d1*d1)/(d2*d2);
      }
      ring->SetChi2(chi2);
   }

};

#endif /* CBMRICHRINGFITTERELLIPSEBASEH */