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