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