VXDSimHit.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.                  *
 **************************************************************************/
 
#pragma once
#ifndef VXD_DATAOBJECTS_VXDSIMHIT_H
#define VXD_DATAOBJECTS_VXDSIMHIT_H
 
#include <simulation/dataobjects/SimHitBase.h>
#include <vxd/dataobjects/VxdID.h>
#include <Math/Vector3D.h>
#include <vector>
#include <algorithm>
 
namespace Belle2 {
  class VXDSimHit : public SimHitBase {
  public:
    VXDSimHit(): m_posIn(), m_posOut(), m_pdg(0), m_globalTime(0), m_sensorID(0) {}

    VXDSimHit(VxdID sensorID, int pdg, float globalTime, const float* posIn, const float* posOut):
      SimHitBase(), m_pdg(pdg), m_globalTime(globalTime), m_sensorID(sensorID)
    {
      std::copy_n(posIn, 3, m_posIn);
      std::copy_n(posOut, 3, m_posOut);
    }

    void setEnergyDeposit(std::vector<unsigned int>& electronProfile)
    {
      m_electronProfile.clear();
      std::swap(m_electronProfile, electronProfile);
    }

    VxdID getSensorID() const { return m_sensorID; }
    int getPDGcode() const { return m_pdg; }
    ROOT::Math::XYZVector getPosIn() const { return ROOT::Math::XYZVector(m_posIn[0], m_posIn[1], m_posIn[2]); }
    ROOT::Math::XYZVector getPosOut() const { return ROOT::Math::XYZVector(m_posOut[0], m_posOut[1], m_posOut[2]); }
    const float* getPosArrayIn() const { return m_posIn; }
    const float* getPosArrayOut() const { return m_posOut; }
    float getGlobalTime() const override { return m_globalTime; }
    virtual void shiftInTime(float delta) override { m_globalTime += delta; }

    float getElectrons() const;

    std::vector<std::pair<float, float> > getElectronProfile() const;

    std::vector<std::pair<float, float> > getElectronsConstantDistance(double length) const;

    std::vector<std::pair<float, float> > getElectronsConstantNumber(double electronsPerStep) const;
 
  private:
    std::vector<unsigned int> m_electronProfile;
    float m_posIn[3];
    float m_posOut[3];
    int m_pdg;
    float m_globalTime;
    unsigned short m_sensorID;
 
    ClassDefOverride(VXDSimHit, 2)
  };

} // end namespace Belle2
 
#endif