ECLSimHit.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Torben Ferber (ferber@physics.ubc.ca)                    *
 *               Poyuan Chen                                              *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef ECLSIMHIT_H
#define ECLSIMHIT_H
 
#include <G4ThreeVector.hh>
#include <simulation/dataobjects/SimHitBase.h>
 
namespace Belle2 {
  class ECLSimHit : public SimHitBase {
  public:
    ECLSimHit(): SimHitBase(), m_CellId(0), m_TrackId(0), m_Pdg(0), m_FlightTime(0), m_Edep(0), m_Momentum{0}, m_Position{0},
      m_HadronEdep{0} {;}
 
    ECLSimHit(
      int CellId,            
      int TrackId,           
      int Pdg,               
      double FlightTime,      
      double Edep,            
      G4ThreeVector Momentum,     
      G4ThreeVector Position,       
      double Hadronedep              
    ): SimHitBase(), m_CellId(CellId), m_TrackId(TrackId), m_Pdg(Pdg), m_FlightTime(FlightTime), m_Edep(Edep), m_HadronEdep(Hadronedep)
    {
      m_Position[0] = Position.x(); m_Position[1] = Position.y(), m_Position[2] = Position.z();
      m_Momentum[0] = Momentum.x(); m_Momentum[1] = Momentum.y(), m_Momentum[2] = Momentum.z();
    }
 
    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 = static_cast<double>(FlightTime); }
 
    void setEnergyDep(double Edep) { m_Edep = static_cast<double>(Edep); }
 
    void setMomentum(const G4ThreeVector& Momentum) { m_Momentum[0] = Momentum.x(); m_Momentum[1] = Momentum.y(), m_Momentum[2] = Momentum.z(); }
 
    //KM void setPosIn(G4ThreeVector Position) { m_Position = Position; }
 
    void setPosition(const G4ThreeVector& Position) { m_Position[0] = Position.x(); m_Position[1] = Position.y(), m_Position[2] = Position.z(); }
 
    int getCellId() const { return m_CellId; }
 
    int getTrackId() const { return m_TrackId; }
 
    int getPDGCode() const { return m_Pdg; }
 
    double getFlightTime() const { return m_FlightTime; }
 
    double getEnergyDep() const { return m_Edep; }
 
    double getHadronEnergyDep() const { return m_HadronEdep; }
 
    G4ThreeVector getMomentum() const { return G4ThreeVector(m_Momentum[0], m_Momentum[1], m_Momentum[2]); }
 
    G4ThreeVector getPosIn() const { return getPosition(); }
 
    G4ThreeVector getPosition() const { return G4ThreeVector(m_Position[0], m_Position[1], m_Position[2]); }
 
    void shiftInTime(float delta) override { m_FlightTime += delta; }
 
 
  private:
    int m_CellId;            
    int m_TrackId;           
    int m_Pdg;               
    double m_FlightTime;      
    double m_Edep;            
    double m_Momentum[3];     
    double m_Position[3];        
    double m_HadronEdep;      
    ClassDefOverride(ECLSimHit, 6);
  };
} // end namespace Belle2
 
#endif