DosiSimHit.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010-2014  Belle II Collaboration                         *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Ritter, Igal Jaegle                               *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef DosiSIMHIT_H
#define DosiSIMHIT_H
 
#include <simulation/dataobjects/SimHitBase.h>
 
// ROOT
#include <TVector3.h>
 
namespace Belle2 {
  class DosiSimHit : public SimHitBase {
  public:
    DosiSimHit(): 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) {;}
    DosiSimHit(
      int CellId,            
      int TrackId,           
      int Pdg,               
      float FlightTime,      
      float Edep,            
      TVector3 Momentum,     
      TVector3 Position        
    ): SimHitBase()
    {
      m_CellId = CellId;
      m_TrackId = TrackId;
      m_Pdg = Pdg;
      m_FlightTime = FlightTime;
      m_Edep = Edep;
      m_Momentum = Momentum;
      m_Position = Position;
 
    }
 
 
 
 
    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; }
 
 
    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; }
 
 
    void shiftInTime(float delta) override {  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;        
    ClassDefOverride(DosiSimHit, 1)
  };
 
} // end namespace Belle2
 
#endif