Simulation/BOOST/TruSim/include/TruSim/BesTruthTrack.hh

Go to the documentation of this file.

//---------------------------------------------------------------------------//
////////      BOOST --- BESIII Object_Oriented Simulation Tool                     //
////////---------------------------------------------------------------------------//
////////Description:
////////Author : Dengzy
////
//// ////Created:  Aug, 2004
//// ////Modified:
//// ////Comment:
//// ////---------------------------------------------------------------------------//
//// //// $Id:BesTruthTrack.hh
 
#ifndef BesTruthTrack_h
#define BesTruthTrack_h 1
 
#include <iostream>
#include <vector>
using namespace std;
#include "BesTruthVertex.hh"
#include "CLHEP/Vector/LorentzVector.h"
 
using namespace CLHEP;
 
class BesTruthTrack {
public:
  BesTruthTrack();
  ~BesTruthTrack(){};
 
  friend ostream& operator<<( ostream&, const BesTruthTrack& );
  friend ostream& operator<<( ostream&, const BesTruthTrack* );
 
  enum { unassigned = -1 };
 
  // return four momentum at origin
  HepLorentzVector GetP4() const { return m_fourMomentum; }
  void             SetP4( const HepLorentzVector& p4 ) { m_fourMomentum = p4; }
 
  // return particle ID (PDG)
  G4int GetPDGCode() const { return m_PDGCode; }
  void  SetPDGCode( G4int code ) { m_PDGCode = code; }
 
  // return particle charge
  G4double GetPDGCharge() const { return m_PDGCharge; }
  void     SetPDGCharge( G4double charge ) { m_PDGCharge = charge; }
 
  // return particle name of this track
  G4String GetParticleName() const { return m_particleName; }
  void     SetParticleName( G4String name ) { m_particleName = name; }
 
  // Access vertex associated with the beginning of the track
  BesTruthVertex* GetVertex() const { return m_vertex; }
  void            SetVertex( BesTruthVertex* vertex ) { m_vertex = vertex; }
 
  G4int GetBarcodeEndVtx() { return m_barcodeEndVtx; }
  void  SetBarcodeEndVtx( G4int vtx ) { m_barcodeEndVtx = vtx; }
 
  // Access vertex associated with the death of the track,
  // or return zero if the track exited the detector.
  BesTruthVertex* GetTerminalVertex() const { return m_terminalVertex; }
  void            SetTerminalVertex( BesTruthVertex* vertex ) { m_terminalVertex = vertex; }
 
  // Returns the index of this track
  G4int GetIndex() const { return m_index; }
  void  SetIndex( G4int index ) { m_index = index; }
 
  // return the Geant4 trackId of this track
  G4int GetG4TrackId() const { return m_g4TrackId; }
  void  SetG4TrackId( G4int trackId ) { m_g4TrackId = trackId; }
 
  // Return the youngest parent saved to GTrack. This may or
  // may not be the immediate parent of the track.
  BesTruthTrack* GetParentTrack() {
    if ( m_vertex == 0 ) return 0;
    return m_vertex->GetParentTrack();
  }
 
  // add an index of one daughter of this track
  void AddDaughterIndex( G4int index ) { m_daughterIndexes.push_back( index ); }
 
  // return indexes of daughters of this track
  vector<int> GetDaughterIndexes() const { return m_daughterIndexes; }
 
  void   Found() { m_found = true; }
  G4bool NotFound() { return !m_found; }
 
  G4String GetSource() { return m_source; }
  void     SetSource( G4String source ) { m_source = source; }
 
private:
  // the four-momentum of the particle in the
  // lab frame at the point of origin (as stored in GVertex)
  HepLorentzVector m_fourMomentum;
 
  // vertex representing the origin of this particle
  BesTruthVertex* m_vertex;
 
  // vertex representing the end of this particle
  BesTruthVertex* m_terminalVertex;
 
  G4int m_barcodeEndVtx;
 
  // particle ID
  G4int m_PDGCode;
 
  // particle charge;
  G4double m_PDGCharge;
 
  // particle name
  G4String m_particleName;
 
  // index in trackList
  G4int m_index;
 
  // Geant4 track ID
  G4int m_g4TrackId;
 
  // indexes of daughter tracks
  vector<int> m_daughterIndexes;
 
  G4bool   m_found;
  G4String m_source;
};
 
#endif