MvaLowEClusterMergingAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_MVA_LOW_E_CLUSTER_MERGING_ALGORITHM_H
#define LAR_MVA_LOW_E_CLUSTER_MERGING_ALGORITHM_H 1

#include "Pandora/Algorithm.h"
#include "larpandoracontent/LArHelpers/LArMCParticleHelper.h"

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

#include "Pandora/PandoraInternal.h"

namespace lar_content
{

template <typename T>
class MvaLowEClusterMergingAlgorithm : public pandora::Algorithm
{
public:
    MvaLowEClusterMergingAlgorithm();

    virtual ~MvaLowEClusterMergingAlgorithm();

protected:
    pandora::StatusCode Run();
    pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);

    const pandora::MCParticle *GetMCForCluster(
        const pandora::Cluster *const cluster, std::map<const pandora::Cluster *, const pandora::MCParticle *> &clusterToMCMap) const;

    bool IsValidToUse(const pandora::Cluster *const cluster, std::map<const pandora::Cluster *, bool> &clusterIsUsed) const;

    double Angle(const pandora::CartesianVector vector1, const pandora::CartesianVector vector2) const;

    const pandora::CaloHitList EdgeHitFinder(const pandora::Cluster *const cluster, pandora::CaloHitList &clusterEdgeHits) const;

    bool ClusterTool(std::vector<std::string> featureOrder, LArMvaHelper::MvaFeatureMap featureMap) const;

    pandora::StatusCode EdgeHitComparer(const pandora::ClusterList *const pClusterList, const std::string &listName) const;

    pandora::StringVector m_inputClusterListNames; 
    std::string m_mcParticleListName;              
    bool m_trainingSetMode;                        
    bool m_enableProbability;                      
    float m_minProbabilityCut;                     
    std::string m_treeName;                        
    std::string m_fileName;                        
    int m_event;                                   
    float m_maxClusterFraction;                    
    float m_minNCaloHits;                          
    bool m_writeTree;                              
    float m_upperHitThreshold;                     
    std::string m_trainingOutputFile;              
    std::string m_filePathEnvironmentVariable;     
    std::string m_mvaFileName;                     
    std::string m_mvaName;                         
    T m_mva;                                       
    float m_countHitsThreshold;                    
    std::string m_vertexListName;                  
    float m_contactThreshold;                      
    float m_proximityThreshold;                    
    float m_divisions;                             
    float m_sectorTolerance;                       
    bool m_printOut;                               
};

typedef MvaLowEClusterMergingAlgorithm<AdaBoostDecisionTree> BdtLowEClusterMergingAlgorithm;

} // namespace lar_content

#endif // #ifndef LAR_MVA_LOW_E_CLUSTER_MERGING_ALGORITHM_H