DLBaseHierarchyTool.h

Go to the documentation of this file.

#ifndef LAR_DL_BASE_HIERARCHY_TOOL_H
#define LAR_DL_BASE_HIERARCHY_TOOL_H 1

#include "Pandora/PandoraInternal.h"

#include "larpandoracontent/LArHelpers/LArGeometryHelper.h"

#include "larpandoradlcontent/LArHelpers/LArDLHelper.h"
#include "larpandoradlcontent/LArThreeDReco/LArEventBuilding/LArHierarchyPfo.h"

#include <torch/script.h>
#include <torch/torch.h>

namespace lar_dl_content
{

class DLBaseHierarchyTool : public pandora::AlgorithmTool
{
public:
    DLBaseHierarchyTool();

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

    void SetDetectorBoundaries();

    std::pair<float, float> GetParticleInfoAboutPfoPosition(const pandora::Algorithm *const pAlgorithm,
        const pandora::ParticleFlowObject *const pPfo, const pandora::CartesianVector &pointOfInterest) const;

    void NormaliseNetworkParam(const float minLimit, const float maxLimit, float &networkParam) const;

    float m_vertexRegionRadiusSq;                               
    pandora::StringVector m_pfoListNames;                       
    LArGeometryHelper::DetectorBoundaries m_detectorBoundaries; 
    bool m_areBoundariesSet;                                    
};

} // namespace lar_dl_content

#endif // #ifndef LAR_DL_BASE_HIERARCHY_TOOL_H