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;
    typedef std::set<unsigned int> UIntSet;

    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;      
    unsigned int m_clusterMinHits; 

    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