KFPTrack.h

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

/*
 * This file is part of KFParticle package
 * Copyright (C) 2007-2019 FIAS Frankfurt Institute for Advanced Studies
 *               2007-2019 Goethe University of Frankfurt
 *               2007-2019 Ivan Kisel <I.Kisel@compeng.uni-frankfurt.de>
 *               2007-2019 Maksym Zyzak
 *
 * KFParticle is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * KFParticle is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#ifndef KFPTrack_H
#define KFPTrack_H

#include <cmath>
#include "TObject.h"
class KFPTrack 
#ifdef __ROOT__
: public TObject
#endif
{

public:
  KFPTrack():fChi2(-1.f), fQ(0), fNDF(-1), fId(-1) { }
  ~KFPTrack() { }

  int    GetID() const { return fId; } 
  
  bool   GetXYZPxPyPz(float *p) const 
  {
    for(int i=0; i<6; i++)
      p[i] = fP[i];
    return 1;
  }
  bool   GetCovarianceXYZPxPyPz(float cv[21]) const
  {
    for (int i=0; i<21; i++)
      cv[i] = fC[i];
    return 1;
  }
  bool   GetCovarianceXYZPxPyPz(double cv[21]) const
  {
    for (int i=0; i<21; i++)
      cv[i] = fC[i];
    return 1;
  }

  void   GetXYZ(float *position)    const {position[0] = fP[0]; position[1] = fP[1]; position[2] = fP[2];}
  void   GetPxPyPz(float *position) const {position[0] = fP[3]; position[1] = fP[4]; position[2] = fP[5];}
  void   XvYvZv(float *position)    const {position[0] = fP[0]; position[1] = fP[1]; position[2] = fP[2];}
  void   PxPyPz(float *position) const {position[0] = fP[3]; position[1] = fP[4]; position[2] = fP[5];}
  void   XvYvZv(double *position)    const {position[0] = fP[0]; position[1] = fP[1]; position[2] = fP[2];}
  void   PxPyPz(double *position) const {position[0] = fP[3]; position[1] = fP[4]; position[2] = fP[5];}

  float GetX() const { return fP[0]; }  
  float GetY() const { return fP[1]; }  
  float GetZ() const { return fP[2]; }  
  float GetPx() const { return fP[3]; } 
  float GetPy() const { return fP[4]; } 
  float GetPz() const { return fP[5]; } 

  float GetPt() const { return sqrt(fP[3]*fP[3]+fP[4]*fP[4]); } 
  float GetP()  const { return sqrt(fP[3]*fP[3]+fP[4]*fP[4]+fP[5]*fP[5]); } 

  void GetCovarianceMatrix(float *covmatrix)
  {
    for (int i=0; i<21; i++)
      covmatrix[i] = fC[i];
  }
  float GetParameter(int i) const { return fP[i]; }  
  float GetCovariance(int i) const { return fC[i]; } 

  int    Charge()        const { return fQ; }        
  float GetChi2perNDF() const { return fChi2/fNDF; } 
  float GetChi2()       const { return fChi2;      } 
  int    GetNDF()        const { return fNDF; }      

  const float * GetTrack() const { return fP; }     
  const float * GetCovMatrix() const { return fC; } 

  void SetParameters(const float *position) 
  {
    for(int i=0; i<6; i++)
      fP[i] = position[i];
  }
  void SetParameters(double *position) 
  { 
    for(int i=0; i<6; i++)
      fP[i] = position[i];
  }
  void SetParameters(float x, float y, float z, float px, float py, float pz) 
  { 
    fP[0] = x;  fP[1] = y;  fP[2] = z;
    fP[3] = px; fP[4] = py; fP[5] = pz;
  }
  void SetXYZ(float x, float y, float z) 
  { 
    fP[0] = x;  fP[1] = y;  fP[2] = z;
  }
  void SetPxPyPz(float px, float py, float pz) 
  { 
    fP[3] = px; fP[4] = py; fP[5] = pz;
  }
  void SetID(int id)       {fId = id;}      

  void SetX(float x)      { fP[0] = x; }    
  void SetY(float y)      { fP[1] = y; }    
  void SetZ(float z)      { fP[2] = z; }    
  void SetPx(float px)      { fP[3] = px; } 
  void SetPy(float py)      { fP[4] = py; } 
  void SetPz(float pz)      { fP[5] = pz; } 
  void SetCharge(int q) { fQ = q; }         
  void SetChi2(float chi) { fChi2 = chi; }  
  void SetNDF(int ndf)     { fNDF = ndf; }  

  void SetCovarianceMatrix(const float *C)
  {
    for (int i=0; i<21; i++)
      fC[i] = C[i];
  }
  void SetCovarianceMatrix(const double *C)
  {
    for (int i=0; i<21; i++)
      fC[i] = C[i];
  }
  
  void SetCovariance(const int i, const float c) { fC[i]=c; } 
  
  void RotateXY( float alpha ); // rotate on alpha in XY plane. Should be useful for CS change

  int Id() const { return fId; }    
  void SetId( int id ){ fId = id; } 

#ifdef NonhomogeneousField
  const float* GetFieldCoeff() const { return fieldRegion; } 

  void SetFieldCoeff(float c, int i) { fieldRegion[i] = c; } 
#endif
 private:

  float fP[6];  
  float fC[21]; 
  float fChi2;  
  char fQ;      
  short fNDF;   
  int fId;      
  
#ifdef NonhomogeneousField

  float fieldRegion[10];
#endif
#ifdef __ROOT__
  ClassDef(KFPTrack,1)
#endif
};

#endif