ShowerGrowingAlgorithm.h

Go to the documentation of this file.

 
#ifndef LAR_SHOWER_GROWING_ALGORITHM_H
#define LAR_SHOWER_GROWING_ALGORITHM_H 1
 
#include "Pandora/Algorithm.h"
 
#include "larpandoracontent/LArHelpers/LArVertexHelper.h"
 
#include "larpandoracontent/LArObjects/LArPointingCluster.h"
 
#include "larpandoracontent/LArTrackShowerId/BranchGrowingAlgorithm.h"
 
#include <unordered_map>
 
namespace lar_content
{
 
class ShowerGrowingAlgorithm : public BranchGrowingAlgorithm
{
public:
    ShowerGrowingAlgorithm();
 
protected:
    bool IsVertexAssociated(const LArPointingCluster &pointingCluster, const pandora::CartesianVector &vertexPosition2D) const;
 
    static bool SortClusters(const pandora::Cluster *const pLhs, const pandora::Cluster *const pRhs);
 
    typedef std::unordered_map<const pandora::Cluster *, LArVertexHelper::ClusterDirection> ClusterDirectionMap;
    mutable ClusterDirectionMap m_clusterDirectionMap; 
 
private:
    pandora::StatusCode Run();
 
    void SimpleModeShowerGrowing(const pandora::ClusterList *const pClusterList, const std::string &clusterListName) const;
 
    bool GetNextSeedCandidate(const pandora::ClusterList *const pClusterList, const pandora::ClusterSet &usedClusters,
        const pandora::Cluster *&pSeedCluster) const;
 
    void GetAllVertexSeedCandidates(
        const pandora::ClusterList *const pClusterList, const pandora::Vertex *const pVertex, pandora::ClusterVector &seedClusters) const;
 
    void GetSeedAssociationList(const pandora::ClusterVector &particleSeedVector, const pandora::ClusterList *const pClusterList,
        SeedAssociationList &seedAssociationList) const;
 
    void ProcessSeedAssociationDetails(
        const SeedAssociationList &seedAssociationList, const std::string &clusterListName, pandora::ClusterSet &usedClusters) const;
 
    void ProcessBranchClusters(
        const pandora::Cluster *const pParentCluster, const pandora::ClusterVector &branchClusters, const std::string &listName) const;
 
    AssociationType AreClustersAssociated(const pandora::Cluster *const pClusterSeed, const pandora::Cluster *const pCluster) const;
 
    float GetFigureOfMerit(const SeedAssociationList &seedAssociationList) const;
 
    unsigned int GetNVertexConnections(const pandora::CartesianVector &vertexPosition2D, const LArPointingClusterList &pointingClusterList) const;
 
    pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);
 
    pandora::StringVector m_inputClusterListNames; 
 
    unsigned int m_minCaloHitsPerCluster; 
    float m_nearbyClusterDistance;        
    float m_remoteClusterDistance;        
 
    float m_directionTanAngle;  
    float m_directionApexShift; 
 
    float m_minVertexLongitudinalDistance; 
    float m_maxVertexLongitudinalDistance; 
    float m_maxVertexTransverseDistance;   
    float m_vertexAngularAllowance;        
};
 
} // namespace lar_content
 
#endif // #ifndef LAR_SHOWER_GROWING_ALGORITHM_H