TPCSimHit.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#ifndef TPCSIMHIT_H
#define TPCSIMHIT_H
 
#include <simulation/dataobjects/SimHitBase.h>
 
// ROOT
#include <TVector3.h>
 
namespace Belle2 {
  class TPCSimHit : public SimHitBase {
  public:
    TPCSimHit(): SimHitBase(), m_CellId(0), m_TrackId(0), m_Pdg(0), m_FlightTime(0), m_Edep(0), m_Momentum(0, 0, 0), m_Position(0, 0,
          0), m_flag(0) {;}
    TPCSimHit(
      int CellId,            
      int TrackId,           
      int Pdg,               
      float FlightTime,      
      float Edep,            
      TVector3 Momentum,     
      TVector3 Position,        
      int flag               
    ): SimHitBase()
    {
      m_CellId = CellId;
      m_TrackId = TrackId;
      m_Pdg = Pdg;
      m_FlightTime = FlightTime;
      m_Edep = Edep;
      m_Momentum = Momentum;
      m_Position = Position;
      m_flag = flag;
 
    }
 
 
 
 
    void setCellId(int CellId) { m_CellId = CellId; }
 
    void setTrackId(int TrackId) { m_TrackId = TrackId; }
 
    void setPDGCode(int Pdg) { m_Pdg = Pdg; }
 
    void setFlightTime(double FlightTime) { m_FlightTime = (float) FlightTime; }
 
    void setEnergyDep(double Edep) { m_Edep = (float) Edep; }
 
    void setMomentum(TVector3 Momentum) { m_Momentum = Momentum; }
 
    void setPosIn(TVector3 Position) { m_Position = Position; }
 
    void setPosition(TVector3 Position) { m_Position = Position; }
 
    void setFlag(int flag) { m_flag = flag; }
 
 
    int getCellId() const { return m_CellId; }
 
    int getTrackId() const { return m_TrackId; }
 
    int getPDGCode() const { return m_Pdg; }
 
    double getFlightTime() const { return (double)m_FlightTime; }
 
    double getEnergyDep() const { return (double) m_Edep; }
 
    TVector3 getMomentum() const { return m_Momentum; }
 
    TVector3 getPosIn() const { return m_Position; }
 
    TVector3 getPosition() const { return m_Position; }
 
    int getFlag() const { return m_flag; }
 
 
    void shiftInTime(float delta) {  m_FlightTime += delta; }
 
 
  private:
    int m_CellId;            
    int m_TrackId;           
    int m_Pdg;               
    float m_FlightTime;      
    float m_Edep;            
    TVector3 m_Momentum;     
    TVector3 m_Position;        
    int m_flag;              
    ClassDef(TPCSimHit, 1)
  };
 
} // end namespace Belle2
 
#endif