ClusterLabel.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
#ifndef ALICEO2_EMCAL_CLUSTERLABEL_H_
#define ALICEO2_EMCAL_CLUSTERLABEL_H_
 
#include <fairlogger/Logger.h>
#include <gsl/span>
 
#include <vector>
 
namespace o2::emcal
{
 
 
class ClusterLabel
{
 public:
  struct labelWithE {
 
    labelWithE() : label(0), energyFraction(0.) {}
 
    labelWithE(int l, float e) : label(l), energyFraction(e) {}
 
    bool operator>=(labelWithE const& rhs) const
    {
      return energyFraction >= rhs.energyFraction;
    }
 
    float energyFraction; 
    int label;            
  };
 
  // ClusterLabel() = default;
  // ~ClusterLabel() = default;
  // ClusterLabel(const ClusterLabel& clus) = default;
  // ClusterLabel& operator=(const ClusterLabel& source) = default;
 
  void clear();
 
  void addValue(int label, float energyFraction);
 
  void normalize(float factor);
 
  std::vector<int32_t> getLabels();
 
  std::vector<float> getEnergyFractions();
 
  void orderLabels();
 
 protected:
  std::vector<labelWithE> mClusterLabels; 
};
 
} // namespace o2::emcal
#endif // ALICEO2_EMCAL_CLUSTERLABEL_H_