TwoViewMatchingControl.h

Go to the documentation of this file.

#ifndef LAR_TWO_VIEW_MATCHING_CONTROL_H
#define LAR_TWO_VIEW_MATCHING_CONTROL_H 1

#include "larpandoracontent/LArObjects/LArOverlapMatrix.h"

#include "larpandoracontent/LArThreeDReco/LArThreeDBase/NViewMatchingControl.h"

#include <unordered_map>

namespace lar_content
{

template <typename T>
class TwoViewMatchingControl : public NViewMatchingControl
{
public:
    typedef OverlapMatrix<T> MatrixType;

    TwoViewMatchingControl(MatchingBaseAlgorithm *const pAlgorithm);

    virtual ~TwoViewMatchingControl();

    MatrixType &GetOverlapMatrix();

    unsigned int GetHitTypeIndex(const pandora::HitType hitType);

private:
    void UpdateForNewCluster(const pandora::Cluster *const pNewCluster);
    void UpdateUponDeletion(const pandora::Cluster *const pDeletedCluster);
    const std::string &GetClusterListName(const pandora::HitType hitType) const;
    const pandora::ClusterList &GetInputClusterList(const pandora::HitType hitType) const;
    const pandora::ClusterList &GetSelectedClusterList(const pandora::HitType hitType) const;
    void SelectAllInputClusters();
    void PrepareAllInputClusters();
    void PerformMainLoop();
    void TidyUp();
    pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);

    const pandora::ClusterList *m_pInputClusterList1; 
    const pandora::ClusterList *m_pInputClusterList2; 

    pandora::ClusterList m_clusterList1; 
    pandora::ClusterList m_clusterList2; 

    MatrixType m_overlapMatrix; 

    typedef std::unordered_map<pandora::HitType, unsigned int, std::hash<int>> HitTypeToIndexMap;
    HitTypeToIndexMap m_hitTypeToIndexMap; 

    std::string m_inputClusterListName1; 
    std::string m_inputClusterListName2; 

    template <typename U>
    friend class NViewMatchingAlgorithm;
};

} // namespace lar_content

#endif // #ifndef LAR_TWO_VIEW_MATCHING_CONTROL_H