PXDClusterShapeClassifierPar.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 <TObject.h>
#include <map>
#include <vector>
 
#include <pxd/dbobjects/PXDClusterOffsetPar.h>
 
namespace Belle2 {
 

 
  class PXDClusterShapeClassifierPar: public TObject {
  public:
    PXDClusterShapeClassifierPar() {}

    void addShapeLikelyhood(int shape_index, float likelyhood) {m_shape_likelyhoods[shape_index] = likelyhood;}

    const std::map<int, float>& getShapeLikelyhoodMap() const { return m_shape_likelyhoods; }

    const PXDClusterOffsetPar* getOffset(int shape_index, float eta) const
    {
      if (m_offsets.find(shape_index) == m_offsets.end()) {
        return nullptr;
      }
      auto eta_index = getEtaIndex(shape_index, eta);
      return &m_offsets.at(shape_index)[eta_index];
    }

    void addShape(int shape_index)
    {
      m_percentiles[shape_index] = std::vector<float>();
      m_likelyhoods[shape_index] = std::vector<float>();
      m_offsets[shape_index] = std::vector<PXDClusterOffsetPar>();
    }

    void addEtaPercentile(int shape_index, float percentile) {m_percentiles[shape_index].push_back(percentile);}

    void addEtaLikelyhood(int shape_index, float likelyhood) {m_likelyhoods[shape_index].push_back(likelyhood);}

    void addEtaOffset(int shape_index, PXDClusterOffsetPar& offset) { m_offsets[shape_index].push_back(offset);}

    unsigned int getEtaIndex(int shape_index, float eta) const
    {
      auto etaPercentiles = m_percentiles.at(shape_index);
      for (int i = etaPercentiles.size() - 1; i >= 0; --i) {
        if (eta >= etaPercentiles[i])
          return i;
      }
      return 0;
    }

    const std::map<int, std::vector<float>>& getPercentilesMap() const { return m_percentiles; }

    const std::map<int, std::vector<float>>& getLikelyhoodMap() const { return m_likelyhoods; }

    const std::map<int, std::vector<PXDClusterOffsetPar> >& getOffsetMap() const { return m_offsets; }
 
  private:
    std::map<int, std::vector<PXDClusterOffsetPar> > m_offsets;
    std::map<int, std::vector<float>>  m_percentiles;
    std::map<int, std::vector<float>>  m_likelyhoods;
    std::map<int, float> m_shape_likelyhoods;
 
    ClassDef(PXDClusterShapeClassifierPar, 2);   
  };

} // end of namespace Belle2