MvaVertexSelectionAlgorithm.h

Go to the documentation of this file.

 
#ifndef LAR_MVA_VERTEX_SELECTION_ALGORITHM_H
#define LAR_MVA_VERTEX_SELECTION_ALGORITHM_H 1
 
#include "Api/PandoraContentApi.h"
 
#include "larpandoracontent/LArObjects/LArAdaBoostDecisionTree.h"
#include "larpandoracontent/LArObjects/LArSupportVectorMachine.h"
#include "larpandoracontent/LArObjects/LArTwoDSlidingFitResult.h"
 
#include "larpandoracontent/LArHelpers/LArMCParticleHelper.h"
 
#include "larpandoracontent/LArVertex/TrainedVertexSelectionAlgorithm.h"
 
#include <random>
 
namespace lar_content
{
 
template <typename, unsigned int>
class KDTreeLinkerAlgo;
template <typename, unsigned int>
class KDTreeNodeInfoT;
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
template <typename T>
class MvaVertexSelectionAlgorithm : public TrainedVertexSelectionAlgorithm
{
public:
    MvaVertexSelectionAlgorithm();
 
protected:
    pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);
 
private:
    void GetVertexScoreList(const pandora::VertexVector &vertexVector, const BeamConstants &beamConstants, HitKDTree2D &kdTreeU,
        HitKDTree2D &kdTreeV, HitKDTree2D &kdTreeW, VertexScoreList &vertexScoreList) const;
 
    const pandora::Vertex *CompareVertices(const pandora::VertexVector &vertexVector, const VertexFeatureInfoMap &vertexFeatureInfoMap,
        const LArMvaHelper::MvaFeatureVector &eventFeatureList, const KDTreeMap &kdTreeMap, const T &t, const bool useRPhi) const;
 
    std::string m_filePathEnvironmentVariable; 
    std::string m_mvaFileName;                 
    std::string m_regionMvaName;               
    std::string m_vertexMvaName;               
    T m_mvaRegion;                             
    T m_mvaVertex;                             
};
 
typedef MvaVertexSelectionAlgorithm<AdaBoostDecisionTree> BdtVertexSelectionAlgorithm;
typedef MvaVertexSelectionAlgorithm<SupportVectorMachine> SvmVertexSelectionAlgorithm;
 
} // namespace lar_content
 
#endif // #ifndef LAR_MVA_VERTEX_SELECTION_ALGORITHM_H