KFMCTrack.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file KFMCTrack.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 KFMCTRACK_H
#define KFMCTRACK_H

#include <cmath>

class KFMCTrack
{
 public:
  KFMCTrack():fMotherId(-1),fPDG(0),fNMCPoints(0), fNMCPixelPoints(0), fIsReconstructed(0),fIsOutOfDetector(0) {};

  int   MotherId()        const { return fMotherId; } 
  int   PDG()             const { return fPDG;}       
  float Par( int i )      const { return fPar[i]; }   
  float X()               const { return fPar[0]; }   
  float Y()               const { return fPar[1]; }   
  float Z()               const { return fPar[2]; }   
  float L()               const { return sqrt(X()*X() + Y()*Y() + Z()*Z()); } 
  float Px()              const { return fPar[3]; }   
  float Py()              const { return fPar[4]; }   
  float Pz()              const { return fPar[5]; }   
  float P()               const { return sqrt(fPar[3]*fPar[3] + fPar[4]*fPar[4] + fPar[5]*fPar[5]); } 
  float Pt()              const { return sqrt(fPar[3]*fPar[3] + fPar[4]*fPar[4]); } 
  const float *Par()      const { return fPar; }             
  int   NMCPoints()       const { return fNMCPoints; }       
  int   NMCPixelPoints()  const { return fNMCPixelPoints; }  
  bool  IsReconstructed() const { return fIsReconstructed; } 
  bool  IsOutOfDetector() const { return fIsOutOfDetector; } 
  
  void SetPar( int i, float v )  { fPar[i] = v; }          
  void SetX( float v )           { fPar[0] = v; }          
  void SetY( float v )           { fPar[1] = v; }          
  void SetZ( float v )           { fPar[2] = v; }          
  void SetPx( float v )          { fPar[3] = v; }          
  void SetPy( float v )          { fPar[4] = v; }          
  void SetPz( float v )          { fPar[5] = v; }          
  void SetQP( float v )          { fPar[6] = v; }          
  void SetMotherId( int v )      { fMotherId = v; }        
  void SetPDG( int v )           { fPDG = v; }             
  void SetNMCPoints( int v )     { fNMCPoints = v; }       
  void SetNMCPixelPoints( int v ){ fNMCPixelPoints = v; }  
  void SetReconstructed()        { fIsReconstructed = 1; } 
  void SetNotReconstructed()     { fIsReconstructed = 0; } 
  void SetOutOfDetector()        { fIsOutOfDetector = 1; } 
  
 protected:

  int   fMotherId;       
  int   fPDG;            
  float fPar[7];         
  int   fNMCPoints;      
  int   fNMCPixelPoints; 
  
  bool fIsReconstructed; 
  bool fIsOutOfDetector; 
};

#endif