ClusterCandidate.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Ritter                                            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef PXD_CLUSTERCANDIDATE_H
#define PXD_CLUSTERCANDIDATE_H
 
#include <pxd/reconstruction/Pixel.h>
#include <vector>
 
namespace Belle2 {
  namespace PXD {
 
    class ClusterCandidate {
    public:
      enum {
        c_defaultCapacity = 10,
        c_maxCapacity = 4 * c_defaultCapacity
      };
      ClusterCandidate():
        m_merged(0), m_charge(0), m_seed()
      {
        m_pixels.reserve(c_defaultCapacity);
      }
 
      void clear()
      {
        m_merged = 0;
        m_charge = 0;
        m_seed = Pixel();
        m_pixels.clear();
      }
 
      ClusterCandidate* merge(ClusterCandidate& cls);
 
      void add(const Pixel& pixel);
 
      float getCharge() const { return m_charge; }
      float getSeedCharge() const { return m_seed.getCharge(); }
      const Pixel& getSeed() const { return m_seed; }
      size_t size() const { return m_pixels.size(); }
      const std::vector<Pixel>& pixels() const { return m_pixels; }
 
    protected:
      ClusterCandidate* m_merged;
      float m_charge;
      Pixel m_seed;
      std::vector<Pixel> m_pixels;
    };
 
  }
 
}
 
#endif //PXD_CLUSTERCANDIDATE_H