Clusterizend.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 <vector>
#include <utility>
#include <array>
 
namespace Belle2 {
  struct ClustererParameters  {
    unsigned short iterations;
    unsigned short omegaTrim;
    unsigned short phiTrim;
    unsigned short nOmega;
    unsigned short nPhi;
    unsigned short nCot;
  };

  class SimpleCluster {
  public:
    // Default constructor
    SimpleCluster() = default;

    void appendCell(const cell_index& newClusterCell) { m_clusterCells.push_back(newClusterCell); }
    void addHitToCluster(unsigned short hit) { m_clusterHits.push_back(hit); }

    void setPeakCell(const cell_index& peakCell) { m_clusterPeakCell = peakCell; }
    void setPeakWeight(const unsigned int peakWeight) { m_clusterPeakWeight = peakWeight; }

    std::vector<cell_index> getCells() const { return m_clusterCells; }
    std::vector<unsigned short> getClusterHits() const { return m_clusterHits; }
    cell_index getPeakCell() const { return m_clusterPeakCell; }
    unsigned int getPeakWeight() const { return m_clusterPeakWeight; }
 
  private:
    std::vector<cell_index> m_clusterCells;
    std::vector<unsigned short> m_clusterHits;
    cell_index m_clusterPeakCell{{0, 0, 0}};
    unsigned int m_clusterPeakWeight{0};
  };

  class Clusterizend {
  public:
    struct DeletionBounds {
      c3index phiLowerBound;
      c3index phiUpperBound;
      c3index omega;
      c3index cot;
    };
 
    // Default constructor
    Clusterizend() = default;

    explicit Clusterizend(const ClustererParameters& parameters): m_clustererParams(parameters) {}

    void setNewPlane(c3array& houghSpace) { m_houghSpace = &houghSpace; }
    std::vector<SimpleCluster> makeClusters();
 
  private:
    std::array<c3index, 2> getSectionBounds(const unsigned short quadrant, const unsigned section);
    void iterateOverSection(const std::array<c3index, 2>& sectionBounds, std::vector<SimpleCluster>& candidateClusters);
    std::pair<cell_index, unsigned int> getSectionPeak(const std::array<c3index, 2>& sectionBounds);
    SimpleCluster createCluster(const cell_index& peakCell);
    void deletePeakSurroundings(const cell_index& peakCell);
    void clearHoughSpaceRow(const DeletionBounds& bounds);
    ClustererParameters m_clustererParams;
    c3array* m_houghSpace{0};
  };

}