CosmicRayTrackRecoveryAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_COSMIC_RAY_TRACK_RECOVERY_ALGORITHM_H
#define LAR_COSMIC_RAY_TRACK_RECOVERY_ALGORITHM_H 1

#include "Pandora/Algorithm.h"

#include "larpandoracontent/LArObjects/LArTwoDSlidingFitResult.h"

namespace lar_content
{

class CosmicRayTrackRecoveryAlgorithm : public pandora::Algorithm

{
public:
    CosmicRayTrackRecoveryAlgorithm();

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

    class Particle
    {
    public:
        pandora::ClusterList m_clusterList;
    };

    typedef std::vector<Particle> ParticleList;
    typedef std::unordered_map<const pandora::Cluster*, pandora::ClusterList> ClusterAssociationMap;

    pandora::StatusCode GetAvailableClusters(const std::string &inputClusterListName, pandora::ClusterVector &clusterVector) const;

    void SelectCleanClusters(const pandora::ClusterVector &inputVector, pandora::ClusterVector &outputVector) const;

    void BuildSlidingFitResultMap(const pandora::ClusterVector &clusterVector, TwoDSlidingFitResultMap &slidingFitResultMap) const;

    void MatchViews(const pandora::ClusterVector &clusterVector1, const pandora::ClusterVector &clusterVector2, 
        const TwoDSlidingFitResultMap &slidingFitResultMap, ClusterAssociationMap &clusterAssociationMap) const;

    void MatchClusters(const pandora::Cluster* const pSeedCluster, const pandora::ClusterVector &targetClusters, 
        const TwoDSlidingFitResultMap &slidingFitResultMap, ClusterAssociationMap &clusterAssociationMap) const;
   
    void MatchThreeViews(const pandora::ClusterVector &clusterVectorU, const pandora::ClusterVector &clusterVectorV,
        const pandora::ClusterVector &clusterVectorW, const ClusterAssociationMap &clusterAssociationMapUV, 
        const ClusterAssociationMap &clusterAssociationMapVW, const ClusterAssociationMap &clusterAssociationMapWU,
        ParticleList &particleList) const;

    void MatchTwoViews(const pandora::ClusterVector &clusterVectorU,  const pandora::ClusterVector &clusterVectorV,
        const pandora::ClusterVector &clusterVectorW, const ClusterAssociationMap &clusterAssociationMapUV, 
        const ClusterAssociationMap &clusterAssociationMapVW, const ClusterAssociationMap &clusterAssociationMapWU,
        ParticleList &particleList) const;
 
    void MatchOneView(const pandora::ClusterVector &clusterVectorU,  const pandora::ClusterVector &clusterVectorV,
        const pandora::ClusterVector &clusterVectorW, const ClusterAssociationMap &clusterAssociationMapUV, 
        const ClusterAssociationMap &clusterAssociationMapVW, const ClusterAssociationMap &clusterAssociationMapWU,
        ParticleList &particleList) const;

    void BuildVetoList(const ParticleList &particleList, pandora::ClusterSet &vetoList) const;

    void RemoveAmbiguities(const ParticleList &inputParticleList, ParticleList &outputParticleList) const; 

    void MergeClusters(const pandora::ClusterList &inputClusterList, pandora::ClusterList &outputClusterList) const; 

    void BuildParticles(const ParticleList &particleList);

    float          m_clusterMinLength;          
    float          m_clusterMinSpanZ;           
    float          m_clusterMinOverlapX;        
    float          m_clusterMaxDeltaX;          

    std::string    m_inputClusterListNameU;     
    std::string    m_inputClusterListNameV;     
    std::string    m_inputClusterListNameW;     
    std::string    m_outputPfoListName;         
};

} // namespace lar_content

#endif // #ifndef LAR_COSMIC_RAY_TRACK_RECOVERY_ALGORITHM_H