ElectronInitialRegionRefinementAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_ELECTRON_INITIAL_REGION_REFINEMENT_ALGORITHM_H
#define LAR_ELECTRON_INITIAL_REGION_REFINEMENT_ALGORITHM_H 1

#include "Pandora/Algorithm.h"

#include "larpandoracontent/LArShowerRefinement/ConnectionPathwayFeatureTool.h"
#include "larpandoracontent/LArShowerRefinement/LArProtoShower.h"
#include "larpandoracontent/LArShowerRefinement/PeakDirectionFinderTool.h"
#include "larpandoracontent/LArShowerRefinement/ProtoShowerMatchingTool.h"
#include "larpandoracontent/LArShowerRefinement/ShowerSpineFinderTool.h"
#include "larpandoracontent/LArShowerRefinement/ShowerStartFinderTool.h"

namespace lar_content
{

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

private:
    typedef std::map<const pandora::MCParticle *, pandora::CaloHitList> HitOwnershipMap;

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

    void FillShowerPfoVector(pandora::PfoVector &showerPfoVector) const;

    void RefineShower(const pandora::ParticleFlowObject *const pShowerPfo) const;

    pandora::StatusCode GetNeutrinoVertex(pandora::CartesianVector &nuVertex3D) const;

    void BuildViewProtoShowers(const pandora::ParticleFlowObject *const pShowerPfo, const pandora::CartesianVector &nuVertex3D,
        pandora::HitType hitType, ProtoShowerVector &protoShowerVector) const;

    pandora::StatusCode GetHitListOfType(const pandora::HitType hitType, const pandora::CaloHitList *&pCaloHitList) const;

    pandora::CartesianVector GetShowerVertex(const pandora::ParticleFlowObject *const pShowerPfo, const pandora::HitType hitType,
        const pandora::CartesianVector &nuVertex3D) const;

    void RefineShowerVertex(const pandora::ParticleFlowObject *const pShowerPfo, const pandora::HitType hitType,
        const pandora::CartesianVector &nuVertex3D, const pandora::CartesianVector &peakDirection, pandora::CartesianVector &showerVertexPosition) const;

    bool IsShowerConnected(const pandora::CartesianVector &showerVertexPosition, const pandora::CartesianVector &nuVertex2D,
        const pandora::CartesianVector &peakDirection) const;

    bool IsSpineCoincident(const pandora::ParticleFlowObject *const pShowerPfo, const pandora::CartesianVector &nuVertex3D,
        const pandora::HitType hitType, const pandora::CartesianVector &showerVertex, const pandora::CaloHitList &showerSpineHitList) const;

    void BuildViewPathways(const pandora::ParticleFlowObject *const pShowerPfo, const pandora::CaloHitList &protectedHits,
        const pandora::CartesianVector &nuVertex3D, pandora::HitType hitType, ConnectionPathwayVector &viewPathways) const;

    void RefineHitsToAdd(const pandora::CartesianVector &nuVertex3D, const pandora::HitType hitType,
        const ConnectionPathwayVector &viewPathways, ProtoShower &protoShower) const;

    void FindContinuousPath(const pandora::CaloHitList &refinedHitList, const pandora::CartesianVector &nuVertex2D,
        pandora::CaloHitList &continuousHitList) const;

    void SetMetadata(const pandora::ParticleFlowObject *const pShowerPfo, const LArMvaHelper::MvaFeatureMap &featureMap) const;

    void FillElectronHitMap(HitOwnershipMap &electronHitMap) const;

    bool IsElectron(const pandora::ParticleFlowObject *const pShowerPfo, const HitOwnershipMap &electronHitMap) const;

    PeakDirectionFinderTool *m_pShowerPeakDirectionFinderTool; 
    PeakDirectionFinderTool *m_pEventPeakDirectionFinderTool;  
    ShowerSpineFinderTool *m_pShowerSpineFinderTool;           
    ShowerSpineFinderTool *m_pEventPathwayFinderTool;          
    ShowerStartFinderTool *m_pShowerStartFinderTool;           
    ProtoShowerMatchingTool *m_pProtoShowerMatchingTool;       
    std::string m_showerPfoListName;                           
    std::string m_neutrinoVertexListName;                      
    std::string m_caloHitListNameU;                            
    std::string m_caloHitListNameV;                            
    std::string m_caloHitListNameW;                            
    unsigned int m_minShowerHits3D;                            
    unsigned int m_showerSlidingFitWindow;                     
    float m_maxCoincidenceTransverseSeparation; 
    float m_minSpinePurity;                     
    bool m_trainingMode;                        
    std::string m_trainingFileName;             
    float m_unambiguousThreshold;     
    float m_maxConnectionDistance;    
    unsigned int m_minNConnectedHits; 
    float m_minElectronCompleteness;  
    float m_minElectronPurity;        
    float m_maxSeparationFromHit;     
    float m_maxProjectionSeparation;  
    float m_maxXSeparation;           
    ConnectionPathwayFeatureTool::FeatureToolMap m_featureToolMap; 
    pandora::StringVector m_algorithmToolNames;                    
};

} // namespace lar_content

#endif // #ifndef LAR_ELECTRON_INITIAL_REGION_REFINEMENT