SvmPfoCharacterisationAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_SVM_PFO_CHARACTERISATION_ALGORITHM_H
#define LAR_SVM_PFO_CHARACTERISATION_ALGORITHM_H 1

#include "larpandoracontent/LArTrackShowerId/PfoCharacterisationBaseAlgorithm.h"
#include "larpandoracontent/LArTrackShowerId/TrackShowerIdFeatureTool.h"

namespace lar_content
{

class SvmPfoCharacterisationAlgorithm : public PfoCharacterisationBaseAlgorithm
{
public:
    SvmPfoCharacterisationAlgorithm();

protected:
    virtual bool IsClearTrack(const pandora::ParticleFlowObject *const pPfo) const;
    virtual bool IsClearTrack(const pandora::Cluster *const pCluster) const;
    pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);

    ClusterCharacterisationFeatureTool::FeatureToolVector   m_featureToolVector;         
    PfoCharacterisationFeatureTool::FeatureToolVector       m_featureToolVectorThreeD;   
    PfoCharacterisationFeatureTool::FeatureToolVector       m_featureToolVectorNoChargeInfo; 

    SupportVectorMachine    m_supportVectorMachine;              
    SupportVectorMachine    m_supportVectorMachineNoChargeInfo;  

    bool                    m_trainingSetMode;              
    bool                    m_enableProbability;            
    bool                    m_useThreeDInformation;         
    float                   m_minProbabilityCut;            
    unsigned int            m_minCaloHitsCut;               

    std::string             m_trainingOutputFile;           
    std::string             m_filePathEnvironmentVariable;  
    std::string             m_svmFileName;                  
    std::string             m_svmName;                      
    std::string             m_svmFileNameNoChargeInfo;      
    std::string             m_svmNameNoChargeInfo;          

};

} // namespace lar_content

#endif // #ifndef LAR_SVM_PFO_CHARACTERISATION_ALGORITHM_H