LArMvaInterface.h

Go to the documentation of this file.

 
#ifndef LAR_MVA_INTERFACE_H
#define LAR_MVA_INTERFACE_H 1
 
#include "Pandora/PandoraInputTypes.h"
 
#include <vector>
 
namespace lar_content
{
 
class MvaTypes
{
public:
    class InitializedDouble
    {
    public:
        InitializedDouble();
 
        InitializedDouble(const double number);
 
        InitializedDouble(const InitializedDouble &rhs);
 
        InitializedDouble &operator=(const double number);
 
        InitializedDouble &operator=(const InitializedDouble rhs);
 
        double Get() const;
 
        bool IsInitialized() const;
 
    private:
        double m_number;      
        bool m_isInitialized; 
    };
 
    typedef InitializedDouble MvaFeature;
    typedef std::vector<MvaFeature> MvaFeatureVector;
    typedef std::map<std::string, MvaFeature> MvaFeatureMap;
};
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
class MvaInterface
{
public:
    virtual bool Classify(const MvaTypes::MvaFeatureVector &features) const = 0;
 
    virtual double CalculateClassificationScore(const MvaTypes::MvaFeatureVector &features) const = 0;
 
    virtual double CalculateProbability(const MvaTypes::MvaFeatureVector &features) const = 0;
 
    virtual ~MvaInterface() = default;
};
 
//------------------------------------------------------------------------------------------------------------------------------------------
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline MvaTypes::InitializedDouble::InitializedDouble() : m_number(0.), m_isInitialized(false)
{
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline MvaTypes::InitializedDouble::InitializedDouble(const double number) : m_number(number), m_isInitialized(true)
{
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline MvaTypes::InitializedDouble::InitializedDouble(const InitializedDouble &rhs) :
    m_number(rhs.m_number),
    m_isInitialized(rhs.m_isInitialized)
{
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline MvaTypes::InitializedDouble &MvaTypes::InitializedDouble::operator=(const double number)
{
    m_number = number;
    m_isInitialized = true;
 
    return *this;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline MvaTypes::InitializedDouble &MvaTypes::InitializedDouble::operator=(const InitializedDouble rhs)
{
    if (this != &rhs)
    {
        m_number = rhs.m_number;
        m_isInitialized = rhs.m_isInitialized;
    }
 
    return *this;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline double MvaTypes::InitializedDouble::Get() const
{
    if (!m_isInitialized)
        throw pandora::StatusCodeException(pandora::STATUS_CODE_NOT_INITIALIZED);
 
    return m_number;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline bool MvaTypes::InitializedDouble::IsInitialized() const
{
    return m_isInitialized;
}
 
} // namespace lar_content
 
#endif // #ifndef LAR_MVA_INTERFACE_H