PXDDigit.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Peter Kvasnicka, Zbynek Drasal, Martin Ritter            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef PXDDIGIT_H
#define PXDDIGIT_H
 
#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;
    }
 
    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
 
#endif