TwoDSlidingFitSplittingAndSplicingAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_TWO_D_SLIDING_FIT_SPLITTING_AND_SPLICING_ALGORITHM_H
#define LAR_TWO_D_SLIDING_FIT_SPLITTING_AND_SPLICING_ALGORITHM_H 1

#include "Pandora/Algorithm.h"

#include "larpandoracontent/LArHelpers/LArClusterHelper.h"

namespace lar_content
{

class TwoDSlidingFitSplittingAndSplicingAlgorithm : public pandora::Algorithm
{
public:
    TwoDSlidingFitSplittingAndSplicingAlgorithm();

protected:
    virtual pandora::StatusCode Run();
    virtual pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);

    class ClusterExtension
    {
    public:
        ClusterExtension(const pandora::Cluster *const pBranchCluster, const pandora::Cluster *const pReplacementCluster,
            const pandora::CartesianVector &replacementVertex, const pandora::CartesianVector &branchVertex,
            const pandora::CartesianVector &branchDirection);

        const pandora::Cluster* GetBranchCluster() const;

        const pandora::Cluster* GetReplacementCluster() const;

        const pandora::CartesianVector &GetReplacementVertex() const;

        const pandora::CartesianVector &GetBranchVertex() const;

        const pandora::CartesianVector &GetBranchDirection() const;

        private:
        const pandora::Cluster     *m_pBranchCluster;            
        const pandora::Cluster     *m_pReplacementCluster;       
        pandora::CartesianVector    m_replacementVertex;         
        pandora::CartesianVector    m_branchVertex;              
        pandora::CartesianVector    m_branchDirection;           
    };

    typedef std::vector<ClusterExtension> ClusterExtensionList;

    virtual void FindBestSplitPosition(const TwoDSlidingFitResult &branchSlidingFit, 
        const TwoDSlidingFitResult &replacementSlidingFit, pandora::CartesianVector &replacementStartPosition,
        pandora::CartesianVector &branchSplitPosition, pandora::CartesianVector &branchSplitDirection) const = 0;

private:

    void GetListOfCleanClusters(const pandora::ClusterList *const pClusterList, pandora::ClusterVector &clusterVector) const;

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

    void BuildClusterExtensionList(const pandora::ClusterVector &clusterVector, const TwoDSlidingFitResultMap &branchResultMap,
        const TwoDSlidingFitResultMap &replacementResultMap, ClusterExtensionList &clusterExtensionList) const;

    void PruneClusterExtensionList(const ClusterExtensionList &inputList, const TwoDSlidingFitResultMap &branchResultMap,
        const TwoDSlidingFitResultMap &replacementResultMap, ClusterExtensionList &outputList) const;

    float CalculateBranchChi2(const pandora::Cluster* const pCluster, const pandora::CartesianVector &splitPosition, 
        const pandora::CartesianVector &splitDirection) const;

    void SplitBranchCluster(const pandora::Cluster* const pCluster, const pandora::CartesianVector &splitPosition, 
        const pandora::CartesianVector &splitDirection, pandora::CaloHitList &principalCaloHitList, 
        pandora::CaloHitList &branchCaloHitList) const;

    pandora::StatusCode RunSplitAndExtension(const ClusterExtensionList &splitList, TwoDSlidingFitResultMap &branchResultMap,
        TwoDSlidingFitResultMap &replacementResultMap) const;

    pandora::StatusCode ReplaceBranch(const pandora::Cluster *const pBranchCluster, const pandora::Cluster *const pReplacementCluster,
        const pandora::CartesianVector &branchSplitPosition, const pandora::CartesianVector &branchSplitDirection) const;

    unsigned int  m_shortHalfWindowLayers;          
    unsigned int  m_longHalfWindowLayers;           
    float         m_minClusterLength;               
    float         m_vetoDisplacement;               
    bool          m_runCosmicMode;                  
};

//------------------------------------------------------------------------------------------------------------------------------------------
//------------------------------------------------------------------------------------------------------------------------------------------

inline TwoDSlidingFitSplittingAndSplicingAlgorithm::ClusterExtension::ClusterExtension(const pandora::Cluster *const pBranchCluster,
        const pandora::Cluster *const pReplacementCluster, const pandora::CartesianVector &replacementVertex,
        const pandora::CartesianVector &branchVertex, const pandora::CartesianVector &branchDirection) :
    m_pBranchCluster(pBranchCluster),
    m_pReplacementCluster(pReplacementCluster),
    m_replacementVertex(replacementVertex),
    m_branchVertex(branchVertex),
    m_branchDirection(branchDirection)
{

}

//------------------------------------------------------------------------------------------------------------------------------------------

inline const pandora::Cluster *TwoDSlidingFitSplittingAndSplicingAlgorithm::ClusterExtension::GetBranchCluster() const
{
    return m_pBranchCluster;
}

//------------------------------------------------------------------------------------------------------------------------------------------

inline const pandora::Cluster *TwoDSlidingFitSplittingAndSplicingAlgorithm::ClusterExtension::GetReplacementCluster() const
{
    return m_pReplacementCluster;
}

//------------------------------------------------------------------------------------------------------------------------------------------

inline const pandora::CartesianVector &TwoDSlidingFitSplittingAndSplicingAlgorithm::ClusterExtension::GetReplacementVertex() const
{
    return m_replacementVertex;
}

//------------------------------------------------------------------------------------------------------------------------------------------

inline const pandora::CartesianVector &TwoDSlidingFitSplittingAndSplicingAlgorithm::ClusterExtension::GetBranchVertex() const
{
    return m_branchVertex;
}

//------------------------------------------------------------------------------------------------------------------------------------------

inline const pandora::CartesianVector &TwoDSlidingFitSplittingAndSplicingAlgorithm::ClusterExtension::GetBranchDirection() const
{
    return m_branchDirection;
}

} // namespace lar_content

#endif // #ifndef LAR_TWO_D_SLIDING_FIT_SPLITTING_AND_SPLICING_ALGORITHM_H