ThreeDReclusteringAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_THREE_D_RECLUSTERING_ALGORITHM_H
#define LAR_THREE_D_RECLUSTERING_ALGORITHM_H 1

#include "Objects/ParticleFlowObject.h"
#include "Pandora/Algorithm.h"
#include "Pandora/AlgorithmTool.h"
#include "Pandora/PandoraInternal.h"

namespace lar_content
{

class ClusteringTool;

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

class ThreeDReclusteringAlgorithm : public pandora::Algorithm
{
public:
    enum FigureOfMeritType
    {
        CHEATED
    };

    ThreeDReclusteringAlgorithm();

    ~ThreeDReclusteringAlgorithm();

private:
    typedef std::vector<ClusteringTool *> ClusteringToolVector;

    pandora::StatusCode Run();

    pandora::StatusCode RebuildPfo(const pandora::Pfo *pPfoToRebuild, pandora::ClusterList &newClustersList);

    pandora::StatusCode BuildNewTwoDClusters(const pandora::Pfo *pPfoToRebuild, pandora::ClusterList &newClustersList);

    pandora::StatusCode BuildNewPfos(const pandora::Pfo *pPfoToRebuild, pandora::ClusterList &newClustersList);

    float GetFigureOfMerit(const pandora::CaloHitList &mergedClusterCaloHitList3D);

    float GetFigureOfMerit(const std::string &figureOfMeritName, const pandora::CaloHitList &mergedClusterCaloHitList3D);

    float GetFigureOfMerit(const std::vector<pandora::CaloHitList> &newClustersCaloHitList3D);

    float GetFigureOfMerit(const std::string &figureOfMeritName, const std::vector<pandora::CaloHitList> &newClustersCaloHitLists3D);

    float GetCheatedFigureOfMerit(const pandora::CaloHitList &mergedClusterCaloHitList3D);

    bool PassesCutsForReclustering(const pandora::ParticleFlowObject *const pPfo);

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

    ClusteringToolVector m_algorithmToolVector; 
    std::string m_pfoListName;                  
    std::string m_clusterListName;              
    pandora::StringVector m_figureOfMeritNames; 
    std::string m_PfosForReclusteringListName;  
    std::string m_mcParticleListName;           
    float m_fomThresholdForReclustering;        
    long unsigned int m_minNumCaloHitsForReclustering;
    std::string m_uClustersListName;
    std::string m_vClustersListName;
    std::string m_wClustersListName;                                                                 //Names of U,V and W cluster lists
    std::map<int, const pandora::Cluster *> m_newClustersUMap, m_newClustersVMap, m_newClustersWMap; 

    static const std::unordered_map<std::string, ThreeDReclusteringAlgorithm::FigureOfMeritType> m_stringToEnumMap; //Figure of merit type enum to string map
};

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

class ClusteringTool : public pandora::AlgorithmTool
{
public:
    ClusteringTool() = default;
    ~ClusteringTool() = default;
    virtual bool Run(const pandora::CaloHitList &inputCaloHitList, std::vector<pandora::CaloHitList> &outputCaloHitListsVector) = 0;
};

} // namespace lar_content

#endif // #ifndef LAR_THREE_D_RECLUSTERING_ALGORITHM_H