DATCONSVDSimpleClusterCandidate.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Christian Wessel                                         *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <vxd/dataobjects/VxdID.h>
#include <vector>
 
namespace Belle2 {
  class DATCONSVDSimpleClusterCandidate {
 
  public:
 
    DATCONSVDSimpleClusterCandidate();
 
    DATCONSVDSimpleClusterCandidate(VxdID vxdID, bool isUside);
 
    DATCONSVDSimpleClusterCandidate(VxdID vxdID, bool isUside, unsigned short maxClusterSize);
 
    bool add(VxdID vxdID, bool isUside, unsigned short index, unsigned short charge, unsigned short cellID);
 
    void finalizeCluster();
 
    void finalizeSimpleCluster();
 
    bool isGoodCluster();
 
    VxdID getSensorID() {return m_vxdID;}
 
    bool isUSide() {return m_isUside;}
 
    unsigned short getCharge() const { return m_charge; }
 
    unsigned short getSeedCharge() const { return m_seedCharge; }
 
    float getPosition() const { return m_position; }
 
    unsigned short getSeedStripIndex() const { return m_seedStripIndex; }
 
    unsigned short size() const { return m_size; }
 
    std::vector<unsigned short> getStripVector() const  { return m_strips; }
 
    std::vector<unsigned short> getChargeVector() const { return m_charges; }
 
    std::vector<unsigned short> getIndexVector() const  { return m_digitIndices; }
 
  protected:
 
    VxdID m_vxdID;
 
    bool m_isUside;
 
    unsigned short m_maxClusterSize;
 
    unsigned short m_charge = 0;
 
    unsigned short m_seedCharge = 0;
 
    float m_position = 0;
 
    unsigned short m_seedStrip = 0;
 
    unsigned short m_size = 0;
 
    unsigned short m_seedStripIndex = 0;
 
    std::vector<unsigned short> m_strips;
 
    std::vector<unsigned short> m_charges;
 
    std::vector<unsigned short> m_digitIndices;
 
  }; // end class definition
 
} // end namespace Belle2