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