KFMCVertex.h

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

#include <iostream>
#include <vector>

class KFMCVertex
{
 public:
  KFMCVertex();

  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]; } 
  
  const float* GetPar()      const { return fPar; } 
  
  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; } 

  int NDaughterTracks() const { return fDaughterTracks.size(); } 
  int NReconstructedDaughterTracks() const { return fNReconstructedDaughters; } 
  void AddDaughterTrack( int iTr ) { fDaughterTracks.push_back(iTr); } 
  int DaughterTrack( int iTr ) const 
  { 
    if(iTr >= NDaughterTracks())
    {
      std::cout << "ERROR!!!! MC PV contains only " << NDaughterTracks() << " tracks" << std::endl; 
      return -1;
    }
    return fDaughterTracks[iTr];
  }

  bool IsMCReconstructable() const { return fIsMCReconstructable; } 
  bool IsReconstructable() const { return fIsReconstructable; }     
  bool IsReconstructed()  const { return fIsReconstructed;   }      
  
  void SetReconstructable() { fIsReconstructable = 1; }        
  void SetUnReconstructable() { fIsReconstructable = 0; }      
  
  void SetMCReconstructable() { fIsMCReconstructable = 1; }    
  void SetMCUnReconstructable() { fIsMCReconstructable = 0; }  
  
  void SetReconstructed() { fIsReconstructed = 1; }   
  void SetUnReconstructed() { fIsReconstructed = 0; } 

  void SetNReconstructedDaughters(int n) { fNReconstructedDaughters = n; } 
  
  bool IsTriggerPV() const { return fIsTriggerPV; } 
  void SetTriggerPV() { fIsTriggerPV = 1; }         
  
  friend std::ostream& operator<<(std::ostream& out, const KFMCVertex &a);
  friend std::istream& operator>>(std::istream& in, KFMCVertex &a);

 protected:

  float fPar[3];                    
  std::vector<int> fDaughterTracks; 
  bool fIsReconstructable;          
  bool fIsMCReconstructable;        
  bool fIsReconstructed;            
  int fNReconstructedDaughters;     
  bool fIsTriggerPV;                
};

#endif