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