CDCSimHit.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
 
#include <cdc/dataobjects/WireID.h>
#include <simulation/dataobjects/SimHitBase.h>
#include <framework/geometry/B2Vector3.h>
 
namespace Belle2 {
  class CDCSimHit : public SimHitBase {
 
  private:
    float m_momentum[3];

    float m_posWire[3];

    float m_posIn[3];

    float m_posOut[3];

    float m_posTrack[3];

    WireID m_wireID;

    float m_globalTime;

    float m_driftLength;

    float m_flightTime;

    float m_edep;

    float m_stepLength;

    int m_pdg;

    unsigned short m_trackId;

    unsigned char m_leftRight;
 
 
  public:
    void setWireID(int iCLayerID, int iWireID)
    {
      m_wireID.setWireID(iCLayerID, iWireID);
    }

    void setTrackId(int trackId) { m_trackId = (unsigned short)trackId; }

    void setPDGCode(int pdg) { m_pdg = pdg; }

    void setDriftLength(double driftLength) { m_driftLength = driftLength; }

    void setFlightTime(double flightTime) { m_flightTime = flightTime; }

    void setGlobalTime(double globalTime) { m_globalTime = globalTime; }

    void setEnergyDep(double edep) { m_edep = edep; }

    void setStepLength(double stepLength) { m_stepLength = stepLength; }

    void setMomentum(const B2Vector3D& momentum)
    {
      m_momentum[0] = momentum.X();
      m_momentum[1] = momentum.Y();
      m_momentum[2] = momentum.Z();
    }

    void setPosWire(const B2Vector3D& posWire)
    {
      m_posWire[0] = posWire.X();
      m_posWire[1] = posWire.Y();
      m_posWire[2] = posWire.Z();
    }

    void setPosIn(const B2Vector3D& posIn)
    {
      m_posIn[0] = posIn.X();
      m_posIn[1] = posIn.Y();
      m_posIn[2] = posIn.Z();
    }

    void setPosOut(const B2Vector3D& posOut)
    {
      m_posOut[0] = posOut.X();
      m_posOut[1] = posOut.Y();
      m_posOut[2] = posOut.Z();
    }

    void setPosTrack(const B2Vector3D& posTrack)
    {
      m_posTrack[0] = posTrack.X();
      m_posTrack[1] = posTrack.Y();
      m_posTrack[2] = posTrack.Z();
    }

 
    void setPosFlag(int zeroOrOne)
    {
      m_leftRight &= 0x6;
      m_leftRight = (m_leftRight | (unsigned char)zeroOrOne);
    }

    void setLeftRightPassageRaw(int minusOneOrZeroOrOne)
    {
      int zeroOrOne = (minusOneOrZeroOrOne <= 0) ? 0 : 1;
      m_leftRight &= 0x5;
      m_leftRight = (m_leftRight | ((unsigned char)zeroOrOne << 1));
    }

    void setLeftRightPassage(int minusOneOrZeroOrOne)
    {
      int zeroOrOne = (minusOneOrZeroOrOne <= 0) ? 0 : 1;
      m_leftRight &= 0x3;
      m_leftRight = (m_leftRight | ((unsigned char)zeroOrOne << 2));
    }

    WireID getWireID() const {return m_wireID;}

    int getTrackId() const { return m_trackId; }

    int getPDGCode() const { return m_pdg; }

    double getDriftLength() const { return m_driftLength; }

    double getFlightTime() const { return m_flightTime; }

    double getEnergyDep() const { return m_edep; }

    double getStepLength() const { return m_stepLength; }

    B2Vector3D getMomentum() const
    {
      return B2Vector3D(m_momentum[0], m_momentum[1], m_momentum[2]);
    }

    B2Vector3D getPosWire() const
    {
      return B2Vector3D(m_posWire[0], m_posWire[1], m_posWire[2]);
    }

    B2Vector3D getPosIn() const
    {
      return B2Vector3D(m_posIn[0], m_posIn[1], m_posIn[2]);
    }

    B2Vector3D getPosOut() const
    {
      return B2Vector3D(m_posOut[0], m_posOut[1], m_posOut[2]);
    }

    B2Vector3D getPosTrack() const
    {
      return B2Vector3D(m_posTrack[0], m_posTrack[1], m_posTrack[2]);
    }

    int getPosFlag() const
    {
      return (int)(m_leftRight & 0x1);
    }

    int getLeftRightPassageRaw() const
    {
      //      int minusOneOrOne = (int((m_leftRight & 0x2) >> 1) == 0) ? -1 : 1;
      //      return minusOneOrOne;
      int zeroOrOne = (int((m_leftRight & 0x2) >> 1) == 0) ? 0 : 1;
      return zeroOrOne;
    }

    int getLeftRightPassage() const
    {
      //      int minusOneOrOne = (int((m_leftRight & 0x4) >> 2) == 0) ? -1 : 1;
      //      return minusOneOrOne;
      int zeroOrOne = (int((m_leftRight & 0x4) >> 2) == 0) ? 0 : 1;
      return zeroOrOne;
    }
 

    float getGlobalTime() const override { return m_globalTime; }


 
    CDCSimHit(): SimHitBase(),
      m_momentum(), m_posWire(),
      m_posIn(), m_posOut(), m_posTrack(),
      m_globalTime(0.0), m_driftLength(0.0), m_flightTime(0.0),
      m_edep(0.0), m_stepLength(1.0),
      m_pdg(0), m_trackId(0),
      m_leftRight(0)
    {}

    CDCSimHit(int layerId,
              int wireId,
              int trackId,
              int pdg,
              double driftLength,
              double flightTime,
              double edep,
              double stepLength,
              const B2Vector3D& momentum,
              const B2Vector3D& posWire,
              const B2Vector3D& posIn,
              const B2Vector3D& posOut,
              const B2Vector3D& posTrack,
              int leftRight,
              double globalTime): SimHitBase()
    {
 
      m_wireID.setWireID(layerId, wireId);
      m_trackId = trackId;
      m_pdg = pdg;
      m_driftLength = driftLength;
      m_flightTime = flightTime;
      m_edep = edep;
      m_stepLength = stepLength;
      m_momentum[0] = momentum.X();
      m_momentum[1] = momentum.Y();
      m_momentum[2] = momentum.Z();
      m_posWire[0] = posWire.X();
      m_posWire[1] = posWire.Y();
      m_posWire[2] = posWire.Z();
      m_posIn[0] = posIn.X();
      m_posIn[1] = posIn.Y();
      m_posIn[2] = posIn.Z();
      m_posOut[0] = posOut.X();
      m_posOut[1] = posOut.Y();
      m_posOut[2] = posOut.Z();
      m_posTrack[0] = posTrack.X();
      m_posTrack[1] = posTrack.Y();
      m_posTrack[2] = posTrack.Z();
      m_leftRight = (unsigned char)leftRight & 0x7; // mask 3 bits (LSB).
      m_globalTime = globalTime;
    }

    virtual void shiftInTime(float delta) override
    {
      m_globalTime += delta;
    }
 

    ClassDefOverride(CDCSimHit, 7);
  };

} // end namespace Belle2