ClusterCandidate.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 <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;
    };
 
  }
 
}