KFMCParticle.h

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

#include <vector>

#ifdef HLTCA_STANDALONE
#include "RootTypesDef.h"
#else
#include "TObject.h"
#endif

class KFMCParticle :public TObject
{
 public:
  KFMCParticle();
  ~KFMCParticle();

  void AddDaughter( int i ); 
  int  NDaughters() const { return fDaughterIds.size(); } 
  const std::vector<int>&  GetDaughterIds() const { return fDaughterIds; } 
  void CleanDaughters() { fDaughterIds.resize(0); } 

  void SetPDG(int pdg) {fPDG = pdg;} 
  void SetMCTrackID(int id) {fMCTrackID = id;} 
  void SetMotherId(int id) {fMotherId = id;} 
  
  int  GetMCTrackID()      const {return fMCTrackID;} 
  int  GetMotherId()       const {return fMotherId;}  
  int  GetPDG()            const {return fPDG;}       
  
  bool IsReconstructable(int i) const {return fIsReconstructable[i];} 
  void SetAsReconstructable(int i) { fIsReconstructable[i] = 1;} 
    
  bool IsReconstructableV0(int i) const {return fIsV0[i];} 
  void SetAsReconstructableV0(int i) { fIsV0[i] = 1;}      
  
  void SetInitialParticleId(int i) {fInitialParticleId = i;} 
  int InitialParticleId() const {return fInitialParticleId;} 
 private: //data
  std::vector<int> fDaughterIds; 
  int fMCTrackID; 
  int fMotherId;  
  int fPDG;       
  
  bool fIsReconstructable[5];
  bool fIsV0[3]; 

  int fInitialParticleId;
#ifndef KFParticleStandalone
  ClassDef( KFMCParticle, 1 )
#endif
};

#endif