PXDDigit.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 <vxd/dataobjects/VxdID.h>
 
#include <framework/dataobjects/DigitBase.h>
 
namespace Belle2 {

  class PXDDigit : public DigitBase {
  public:

    PXDDigit(VxdID sensorID, unsigned short uCellID, unsigned short vCellID,
             unsigned short charge):
      m_sensorID(sensorID),
      m_uCellID(uCellID), m_vCellID(vCellID),
      m_charge(charge)
    {;}

    PXDDigit(): PXDDigit(0, 0, 0, 0) {}

    short int getFrameNumber() const { return VxdID(m_sensorID).getSegmentNumber(); }

    void setFrameNumber(unsigned short frame)
    {
      VxdID id(m_sensorID);
      id.setSegmentNumber(frame);
      m_sensorID = id;
    }

    void fillValues(unsigned short* charge, unsigned short* uCellID, unsigned short* vCellID, unsigned short* sensorID)
    {
      *charge = m_charge;
      *uCellID = m_uCellID;
      *vCellID = m_vCellID;
      *sensorID = m_sensorID;
    }

    VxdID getSensorID() const { return m_sensorID; }

    unsigned short getUCellID() const { return m_uCellID; }

    unsigned short getVCellID() const { return m_vCellID; }

    unsigned short getCharge() const { return m_charge; }

    void setCharge(unsigned short charge) { m_charge = charge; }

    unsigned int getUniqueChannelID() const override
    {
      // shift segment part of sensorID by two bits to make place for u+v
      VxdID sensorID(m_sensorID);
      sensorID.setSegmentNumber(sensorID.getSegmentNumber() << 2);
      return m_vCellID + (m_uCellID << 10) + (sensorID << 16);
    }

    DigitBase::EAppendStatus addBGDigit(const DigitBase* bg) override
    {
      m_charge += dynamic_cast<const PXDDigit*>(bg)->getCharge();
      return DigitBase::c_DontAppend;
    }
 
 
  private:
 
    unsigned short m_sensorID; 
    unsigned short m_uCellID;  
    unsigned short m_vCellID;  
    unsigned short m_charge;   
 
    ClassDefOverride(PXDDigit, 5)
 
  }; // class PXDDigit
 

} // end namespace Belle2