PndTrkFitter Class Reference

#include <PndTrkFitter.h>

Inheritance diagram for PndTrkFitter:

Public Member Functions
	PndTrkFitter ()
	PndTrkFitter (int verbose)
	~PndTrkFitter ()
void	Reset ()
void	SetPointToFit (double x, double y, double sigma)
Bool_t	StraightLineFit (Double_t &fitm, Double_t &fitp)
Double_t	StraightLineFitWithChi2 (Double_t &fitm, Double_t &fitp)
Bool_t	ConstrainedStraightLineFit (Double_t x0, Double_t y0, Double_t &fitm, Double_t &fitp)
Bool_t	ParabolaFit (Double_t &fita, Double_t &fitb, Double_t &fitc)
void	SetVerbose (int ver)

Private Attributes
Double_t	fSx
Double_t	fSy
Double_t	fSxy
Double_t	fSxx
Double_t	fSxxy
Double_t	fSxxx
Double_t	fSxxxx
Double_t	fS1
std::vector< double >	fX
std::vector< double >	fY
std::vector< double >	fSigma
Int_t	fVerbose

Detailed Description

Definition at line 7 of file PndTrkFitter.h.

Constructor & Destructor Documentation

PndTrkFitter::PndTrkFitter

(

)

Definition at line 23 of file PndTrkFitter.cxx.

References fSigma, fX, and fY.

                           : fSx(0), fSy(0), fSxy(0), fSxx(0), fS1(0), fVerbose(0)
{
  fX.clear();
  fY.clear();
  fSigma.clear();
}

PndTrkFitter::PndTrkFitter

(

int

verbose

)

Definition at line 30 of file PndTrkFitter.cxx.

References fSigma, fX, and fY.

                                      : fSx(0), fSy(0), fSxy(0), fSxx(0), fS1(0), fVerbose(verbose)
{
  fX.clear();
  fY.clear();
  fSigma.clear();
}

PndTrkFitter::~PndTrkFitter

(

)

Definition at line 37 of file PndTrkFitter.cxx.

                            { 

}

Member Function Documentation

Bool_t PndTrkFitter::ConstrainedStraightLineFit	(	Double_t	x0,
		Double_t	y0,
		Double_t &	fitm,
		Double_t &	fitp
	)

Definition at line 133 of file PndTrkFitter.cxx.

References Double_t, fS1, fSigma, fSx, fSxx, fSxy, fSy, fVerbose, fX, fY, Reset(), and x0.

Referenced by PndTrkLegendreSecTask::ZPhiFit(), PndTrkLegendreTask::ZPhiFit(), and PndTrkLegendreSecTask2::ZPhiFit().

                                                                                                        {
 
  // cout << "xoy0 " << x0 << " " << y0 << endl;

  int nofPoints = fX.size();
  if(nofPoints == 0) {
    if(fVerbose > 1) cout << "PndTrkFitter::StraightLineFit: no points to fit! fill the array with PndTrkFitter::SetPointToFit()" << endl;
    Reset();
    return kFALSE; // CHECK
  }

  //  cout << "fittinh " << nofPoints << endl;

  for(int ipnt = 0; ipnt < nofPoints; ipnt++)  
    { 
      // cout << fX[ipnt] << " " << fY[ipnt] << " " << fSigma[ipnt] << endl;
      fSx += fX[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      fSy +=  fY[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      
      fSxy += fX[ipnt] * fY[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      fSxx += fX[ipnt]* fX[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      
      fS1 += 1./(fSigma[ipnt] * fSigma[ipnt]);
    }
  
  Double_t den = 2 * x0 * fSx - fSxx - x0 * x0 * fS1;
  if(den == 0) {
    if(fVerbose > 1) cout << "PndTrkFitter:StraightLineFit: DEN == 0" << endl; // CHECK
    Reset();
    return kFALSE;
  }
  
  
  fitm =  (y0 * fSx + x0 * fSy - fSxy - x0 * y0 * fS1)/den;
  
  fitp =  y0 - fitm * x0;

  Reset();
  return kTRUE;
  
}

Bool_t PndTrkFitter::ParabolaFit	(	Double_t &	fita,
		Double_t &	fitb,
		Double_t &	fitc
	)

Definition at line 179 of file PndTrkFitter.cxx.

References CAMath::Abs(), Double_t, fS1, fSigma, fSx, fSxx, fSxxx, fSxxxx, fSxxy, fSxy, fSy, fVerbose, fX, fY, and Reset().

Referenced by PndTrkTrackFinder::AnalyticalParabolaFit().

                                                                               {
  
  int nofPoints = fX.size();
  if(nofPoints == 0) {
    if(fVerbose > 1) cout << "PndTrkFitter::StraightLineFit: no points to fit! fill the array with PndTrkFitter::SetPointToFit()" << endl;
    Reset();
    return kFALSE; // CHECK
  }

  for(int ipnt = 0; ipnt < nofPoints; ipnt++)  
    {    
 
      fSx += fX[ipnt]/(fSigma[ipnt] * fSigma[ipnt]); 
      fSy += fY[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      
      fSxy += fX[ipnt] * fY[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      fSxx += fX[ipnt]* fX[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      
      fSxxy += fX[ipnt] * fX[ipnt] * fY[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      fSxxx += fX[ipnt] * fX[ipnt] * fX[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);

      fSxxx += fX[ipnt] * fX[ipnt] * fX[ipnt] * fX[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      
      fS1 += 1./(fSigma[ipnt] * fSigma[ipnt]);
      
    }
  
  TMatrixD matrix(3,3);
  matrix[0][0] = fS1;
  matrix[0][1] = fSx;
  matrix[0][2] = fSxx;
  
  matrix[1][0] = fSx;
  matrix[1][1] = fSxx;
  matrix[1][2] = fSxxx;
  
  matrix[2][0] = fSxx;
  matrix[2][1] = fSxxx;
  matrix[2][2] = fSxxxx;
  
  Double_t determ;
  
  determ = matrix.Determinant();
  
  if (determ != 0) {
    matrix.Invert();
  }
  else {
    return 0;
    if(fVerbose == 2) cout << "DET 0" << endl;
  }
  
  TMatrixD column(3,1);
  column[0][0] = fSy;
  column[1][0] = fSxy;
  column[2][0] = fSxxy;
  
  TMatrixD column2(3,1);
  column2.Mult(matrix, column);
  
  //Double_t a, b, c; //[R.K. 01/2017] unused variable
  fita = column2[0][0];
  fitb = column2[1][0];
  fitc = column2[2][0];
  
  if(fVerbose == 2) {
    std::cout << "1) parabolic parameters:\n";
    std::cout << "a = " << fita <<  "\n";
    std::cout << "b = " << fitb <<  "\n";
    std::cout << "c = " << fitc <<  "\n";
  }
  if(TMath::Abs(fita)<0.000001) return kFALSE;

  return kTRUE;
}

void PndTrkFitter::Reset

(

)

Definition at line 41 of file PndTrkFitter.cxx.

References fS1, fSigma, fSx, fSxx, fSxxx, fSxxxx, fSxxy, fSxy, fSy, fX, and fY.

Referenced by PndTrkLegendreNew::AnalyticalFit(), PndTrkTrackFinder::AnalyticalFit(), PndTrkCombiLegendreTask::AnalyticalFit(), PndTrkLegendreNew::AnalyticalFit2(), PndTrkTrackFinder::AnalyticalFit2(), PndTrkCombiLegendreTask::AnalyticalFit2(), PndTrkTrackFinder::AnalyticalParabolaFit(), ConstrainedStraightLineFit(), PndTrkLegendreSecTask2::Exec(), PndTrkLegendreSecTask::Exec(), PndTrkLegendreTask::Exec(), PndTrkTrackFinder::Exec(), ParabolaFit(), StraightLineFit(), StraightLineFitWithChi2(), PndTrkLegendreTask::ZPhiFit(), PndTrkLegendreSecTask::ZPhiFit(), and PndTrkLegendreSecTask2::ZPhiFit().

                         {
  fSx = 0;
  fSy = 0;
  fSxy = 0;
  fSxx = 0;
  fSxxy = 0;
  fSxxx = 0;
  fSxxxx = 0;
  fS1 = 0;
  fX.clear();
  fY.clear();
  fSigma.clear();
}

void PndTrkFitter::SetPointToFit	(	double	x,
		double	y,
		double	sigma
	)

Definition at line 55 of file PndTrkFitter.cxx.

References fSigma, fX, and fY.

Referenced by PndTrkLegendreNew::AnalyticalFit(), PndTrkTrackFinder::AnalyticalFit(), PndTrkCombiLegendreTask::AnalyticalFit(), PndTrkLegendreNew::AnalyticalFit2(), PndTrkTrackFinder::AnalyticalFit2(), PndTrkCombiLegendreTask::AnalyticalFit2(), PndTrkTrackFinder::AnalyticalParabolaFit(), PndTrkLegendreTask::Exec(), PndTrkLegendreSecTask::Exec(), PndTrkLegendreSecTask2::Exec(), PndTrkTrackFinder::Exec(), PndTrkLegendreTask::ZPhiFit(), PndTrkLegendreSecTask::ZPhiFit(), and PndTrkLegendreSecTask2::ZPhiFit().

                                                                 {
  fX.push_back(x);
  fY.push_back(y);
  fSigma.push_back(sigma);
}

void PndTrkFitter::SetVerbose ( int  ver )

inline

Definition at line 26 of file PndTrkFitter.h.

References fVerbose.

{ fVerbose = ver; } 

Bool_t PndTrkFitter::StraightLineFit	(	Double_t &	fitm,
		Double_t &	fitp
	)

Definition at line 61 of file PndTrkFitter.cxx.

References Double_t, fS1, fSigma, fSx, fSxx, fSxy, fSy, fVerbose, fX, fY, and Reset().

Referenced by PndTrkLegendreNew::AnalyticalFit(), PndTrkTrackFinder::AnalyticalFit(), PndTrkCombiLegendreTask::AnalyticalFit(), PndTrkLegendreNew::AnalyticalFit2(), PndTrkTrackFinder::AnalyticalFit2(), PndTrkCombiLegendreTask::AnalyticalFit2(), PndTrkLegendreTask::Exec(), PndTrkLegendreSecTask::Exec(), PndTrkLegendreSecTask2::Exec(), PndTrkTrackFinder::Exec(), PndTrkLegendreTask::ZPhiFit(), PndTrkLegendreSecTask::ZPhiFit(), and PndTrkLegendreSecTask2::ZPhiFit().

                                                                   {
 
  int nofPoints = fX.size();
  if(nofPoints == 0) {
    if(fVerbose > 1) cout << "PndTrkFitter::StraightLineFit: no points to fit! fill the array with PndTrkFitter::SetPointToFit()" << endl;
    Reset();
    return kFALSE; // CHECK
  }
  for(int ipnt = 0; ipnt < nofPoints; ipnt++)  
    { 
      fSx += fX[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      fSy +=  fY[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      
      fSxy += fX[ipnt] * fY[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      fSxx += fX[ipnt]* fX[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      
      fS1 += 1./(fSigma[ipnt] * fSigma[ipnt]);
    }
  
  Double_t den = fSxx * fS1 - fSx * fSx;
  if(den == 0) {
    if(fVerbose > 1) cout << "PndTrkFitter:StraightLineFit: DEN == 0" << endl; // CHECK
    Reset();
    return kFALSE;
  }
  
  fitm = (fSxy * fS1 - fSx * fSy)/den;
  fitp =  (fSxx * fSy - fSx * fSxy)/den;
  Reset();
  return kTRUE; 

}

Double_t PndTrkFitter::StraightLineFitWithChi2	(	Double_t &	fitm,
		Double_t &	fitp
	)

Definition at line 94 of file PndTrkFitter.cxx.

References Double_t, fS1, fSigma, fSx, fSxx, fSxy, fSy, fVerbose, fX, fY, and Reset().

                                                                             {
int nofPoints = fX.size();
  if(nofPoints == 0) {
    if(fVerbose > 1) cout << "PndTrkFitter::StraightLineFit: no points to fit! fill the array with PndTrkFitter::SetPointToFit()" << endl;
    Reset();
    return kFALSE; // CHECK
  }
  double Syy = 0;
 for(int ipnt = 0; ipnt < nofPoints; ipnt++)  
    { 
      fSx += fX[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      fSy +=  fY[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      
      fSxy += fX[ipnt] * fY[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      fSxx += fX[ipnt]* fX[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
      
      fS1 += 1./(fSigma[ipnt] * fSigma[ipnt]);

      Syy += fY[ipnt] * fY[ipnt]/(fSigma[ipnt] * fSigma[ipnt]);
    }
  
  Double_t den = fSxx * fS1 - fSx * fSx;
  if(den == 0) {
    if(fVerbose > 1) cout << "PndTrkFitter:StraightLineFit: DEN == 0" << endl; // CHECK
    Reset();
    return kFALSE;
  }
  
  fitm = (fSxy * fS1 - fSx * fSy)/den;
  fitp =  (fSxx * fSy - fSx * fSxy)/den;

  double chi2 = Syy + fitm * fitm * fSxx + fitp * fitp * fS1 + 2 * fitm * fitp * fSx - 2 * fitm * fSxy - 2 * fitp * fSy;

  Reset();

  return chi2;

}

Member Data Documentation

Double_t PndTrkFitter::fS1

private

Definition at line 30 of file PndTrkFitter.h.

Referenced by ConstrainedStraightLineFit(), ParabolaFit(), Reset(), StraightLineFit(), and StraightLineFitWithChi2().

std::vector<double> PndTrkFitter::fSigma

private

Definition at line 31 of file PndTrkFitter.h.

Referenced by ConstrainedStraightLineFit(), ParabolaFit(), PndTrkFitter(), Reset(), SetPointToFit(), StraightLineFit(), and StraightLineFitWithChi2().

Double_t PndTrkFitter::fSx

private

Definition at line 30 of file PndTrkFitter.h.

Referenced by ConstrainedStraightLineFit(), ParabolaFit(), Reset(), StraightLineFit(), and StraightLineFitWithChi2().

Double_t PndTrkFitter::fSxx

private

Definition at line 30 of file PndTrkFitter.h.

Referenced by ConstrainedStraightLineFit(), ParabolaFit(), Reset(), StraightLineFit(), and StraightLineFitWithChi2().

Double_t PndTrkFitter::fSxxx

private

Definition at line 30 of file PndTrkFitter.h.

Referenced by ParabolaFit(), and Reset().

Double_t PndTrkFitter::fSxxxx

private

Definition at line 30 of file PndTrkFitter.h.

Referenced by ParabolaFit(), and Reset().

Double_t PndTrkFitter::fSxxy

private

Definition at line 30 of file PndTrkFitter.h.

Referenced by ParabolaFit(), and Reset().

Double_t PndTrkFitter::fSxy

private

Definition at line 30 of file PndTrkFitter.h.

Referenced by ConstrainedStraightLineFit(), ParabolaFit(), Reset(), StraightLineFit(), and StraightLineFitWithChi2().

Double_t PndTrkFitter::fSy

private

Definition at line 30 of file PndTrkFitter.h.

Referenced by ConstrainedStraightLineFit(), ParabolaFit(), Reset(), StraightLineFit(), and StraightLineFitWithChi2().

Int_t PndTrkFitter::fVerbose

private

Definition at line 32 of file PndTrkFitter.h.

Referenced by ConstrainedStraightLineFit(), ParabolaFit(), SetVerbose(), StraightLineFit(), and StraightLineFitWithChi2().

std::vector<double> PndTrkFitter::fX

private

Definition at line 31 of file PndTrkFitter.h.

Referenced by ConstrainedStraightLineFit(), ParabolaFit(), PndTrkFitter(), Reset(), SetPointToFit(), StraightLineFit(), and StraightLineFitWithChi2().

std::vector<double> PndTrkFitter::fY

private

Definition at line 31 of file PndTrkFitter.h.

Referenced by ConstrainedStraightLineFit(), ParabolaFit(), PndTrkFitter(), Reset(), SetPointToFit(), StraightLineFit(), and StraightLineFitWithChi2().

---

The documentation for this class was generated from the following files:

• PndTrkFitter.h
• PndTrkFitter.cxx