ThreeDTracksBaseAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_THREE_D_TRACKS_BASE_ALGORITHM_H
#define LAR_THREE_D_TRACKS_BASE_ALGORITHM_H 1

#include "larpandoracontent/LArObjects/LArTwoDSlidingFitResult.h"

#include "larpandoracontent/LArThreeDReco/LArThreeDBase/ThreeDBaseAlgorithm.h"

#include <unordered_map>

namespace lar_content
{

typedef std::unordered_map<const pandora::Cluster*, pandora::CartesianPointVector> SplitPositionMap;

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

template<typename T>
class ThreeDTracksBaseAlgorithm : public ThreeDBaseAlgorithm<T>
{
public:
    ThreeDTracksBaseAlgorithm();

    virtual ~ThreeDTracksBaseAlgorithm();

    const TwoDSlidingFitResult &GetCachedSlidingFitResult(const pandora::Cluster *const pCluster) const;

    unsigned int GetSlidingFitWindow() const;

    virtual bool MakeClusterSplits(const SplitPositionMap &splitPositionMap);

    virtual bool MakeClusterSplit(const pandora::CartesianVector &splitPosition, const pandora::Cluster *&pCurrentCluster,
        const pandora::Cluster *&pLowXCluster, const pandora::Cluster *&pHighXCluster) const;

    static bool SortSplitPositions(const pandora::CartesianVector &lhs, const pandora::CartesianVector &rhs);

    virtual void UpdateForNewCluster(const pandora::Cluster *const pNewCluster);
    virtual void UpdateUponDeletion(const pandora::Cluster *const pDeletedCluster);
    virtual void SelectInputClusters(const pandora::ClusterList *const pInputClusterList, pandora::ClusterList &selectedClusterList) const;
    virtual void SetPfoParameters(const ProtoParticle &protoParticle, PandoraContentApi::ParticleFlowObject::Parameters &pfoParameters) const;

protected:
    virtual void PreparationStep();

    virtual void PreparationStep(pandora::ClusterList &clusterList);

    virtual void TidyUp();

    void AddToSlidingFitCache(const pandora::Cluster *const pCluster);

    void RemoveFromSlidingFitCache(const pandora::Cluster *const pCluster);

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

    unsigned int                m_slidingFitWindow;             
    TwoDSlidingFitResultMap     m_slidingFitResultMap;          

    unsigned int                m_minClusterCaloHits;           
    float                       m_minClusterLengthSquared;      
};

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

template<typename T>
inline unsigned int ThreeDTracksBaseAlgorithm<T>::GetSlidingFitWindow() const
{
    return m_slidingFitWindow;
}

} // namespace lar_content

#endif // #ifndef LAR_THREE_D_TRACKS_BASE_ALGORITHM_H