BmnEventCentrality.cxx

Go to the documentation of this file.

#include "BmnEventCentrality.h"
 
BmnEventCentrality::BmnEventCentrality()
    : TNamed()
    , fClassProbability()
{}
 
BmnEventCentrality::BmnEventCentrality(const BmnEventCentrality& other)
    : TNamed(other)
    , fClassProbability(other.fClassProbability)
{
    // Copy constructor: copies the class probability set
}
 
BmnEventCentrality& BmnEventCentrality::operator=(const BmnEventCentrality& other)
{
    if (this != &other) {
        TNamed::operator=(other);
        fClassProbability = other.fClassProbability;   // Copy the class probability set
    }
    return *this;
}
 
BmnEventCentrality::~BmnEventCentrality()
{
    // Destructor: nothing to do as the set is cleaned up automatically
}
 
int BmnEventCentrality::GetClass() const
{
    return GetClass(BmnCentralityClass::Method::FHCalHodo);   // Delegate to the method-specific function
}
 
float BmnEventCentrality::GetProbability() const
{
    return GetProbability(BmnCentralityClass::Method::FHCalHodo);   // Delegate to the method-specific function
}
 
const std::vector<float>& BmnEventCentrality::GetSoftCentrality() const
{
    return GetSoftCentrality(BmnCentralityClass::Method::FHCalHodo);   // Delegate to the method-specific function
}
 
std::pair<int, float> BmnEventCentrality::GetClassProbabilityPair(BmnCentralityClass::Method method) const
{
    int candida = -1;
    float proba = -1.0;
    for (const auto& cls : fClassProbability) {
        if (cls.GetMethod() == method) {
            if (cls.GetProbability() > proba) {
                proba = cls.GetProbability();
                candida = cls.GetIndex();
            }
        }
    }
    return std::make_pair(candida, proba);
}
 
int BmnEventCentrality::GetClass(BmnCentralityClass::Method method) const
{
    return GetClassProbabilityPair(method).first;
}
 
float BmnEventCentrality::GetProbability(BmnCentralityClass::Method method) const
{
    return GetClassProbabilityPair(method).second;
}
 
const std::vector<float>& BmnEventCentrality::GetSoftCentrality(BmnCentralityClass::Method method) const
{
    // Convert fClassProbability to a std::vector<float>
    static std::vector<float> probabilities;
    probabilities.clear();
    for (const auto& cls : fClassProbability) {
        probabilities.push_back(cls.GetProbability());
    }
    return probabilities;
}
 
void BmnEventCentrality::SetCentrality(int cluster_id, float probability, BmnCentralityClass::Method method)
{
    // Add or update the class probability for the specified method
    fClassProbability.insert({cluster_id, probability, method});
}
 
void BmnEventCentrality::Reset()
{
    fClassProbability.clear();   // Clear the class probability set
}