MvaPfoCharacterisationAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_MVA_PFO_CHARACTERISATION_ALGORITHM_H
#define LAR_MVA_PFO_CHARACTERISATION_ALGORITHM_H 1

#include "larpandoracontent/LArHelpers/LArMCParticleHelper.h"

#include "larpandoracontent/LArObjects/LArAdaBoostDecisionTree.h"
#include "larpandoracontent/LArObjects/LArSupportVectorMachine.h"

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

#include "Pandora/PandoraInternal.h"

namespace lar_content
{

template <typename T>
class MvaPfoCharacterisationAlgorithm : public PfoCharacterisationBaseAlgorithm
{
public:
    MvaPfoCharacterisationAlgorithm();

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::FeatureToolMap m_featureToolMap; 

    PfoCharacterisationFeatureTool::FeatureToolMap m_featureToolMapThreeD;       
    PfoCharacterisationFeatureTool::FeatureToolMap m_featureToolMapNoChargeInfo; 

    pandora::StringVector m_algorithmToolNames; 
    pandora::StringVector m_algorithmToolNamesNoChargeInfo; 

    T m_mva;             
    T m_mvaNoChargeInfo; 

    bool m_persistFeatures;               
    bool m_trainingSetMode;               
    bool m_testBeamMode;                  
    bool m_enableProbability;             
    bool m_useThreeDInformation;          
    float m_minProbabilityCut;            
    unsigned int m_minCaloHitsCut;        
    bool m_applyFiducialCut;              
    float m_fiducialMinX;                 
    float m_fiducialMaxX;                 
    float m_fiducialMinY;                 
    float m_fiducialMaxY;                 
    float m_fiducialMinZ;                 
    float m_fiducialMaxZ;                 
    bool m_applyReconstructabilityChecks; 

    std::string m_caloHitListName;    
    std::string m_mcParticleListName; 

    std::string m_trainingOutputFile;          
    std::string m_filePathEnvironmentVariable; 
    std::string m_mvaFileName;                 
    std::string m_mvaName;                     
    std::string m_mvaFileNameNoChargeInfo;     
    std::string m_mvaNameNoChargeInfo;         

    LArMCParticleHelper::PrimaryParameters m_primaryParameters; 

private:
    bool PassesFiducialCut(const pandora::CartesianVector &vertex) const;
};

typedef MvaPfoCharacterisationAlgorithm<AdaBoostDecisionTree> BdtPfoCharacterisationAlgorithm;
typedef MvaPfoCharacterisationAlgorithm<SupportVectorMachine> SvmPfoCharacterisationAlgorithm;

} // namespace lar_content

#endif // #ifndef LAR_MVA_PFO_CHARACTERISATION_ALGORITHM_H