DlHitTrackShowerIdAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_DL_HIT_TRACK_SHOWER_ID_ALGORITHM_H
#define LAR_DL_HIT_TRACK_SHOWER_ID_ALGORITHM_H 1

#include "Pandora/Algorithm.h"

#include "larpandoradlcontent/LArHelpers/LArDLHelper.h"

namespace lar_dl_content
{

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

    virtual ~DlHitTrackShowerIdAlgorithm();

private:
    typedef std::map<const pandora::CaloHit *, std::tuple<int, int, int, int>> CaloHitToPixelMap;
    typedef std::map<int, int> PixelToTileMap;

    pandora::StatusCode Run();

    pandora::StatusCode Train();

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

    void GetHitRegion(const pandora::CaloHitList &caloHitList, float &xMin, float &xMax, float &zMin, float &zMax);

    void GetSparseTileMap(const pandora::CaloHitList &caloHitList, const float xMin, const float zMin, const int nTilesX, PixelToTileMap &sparseMap);

    pandora::StringVector m_caloHitListNames; 
    std::string m_modelFileNameU;             
    std::string m_modelFileNameV;             
    std::string m_modelFileNameW;             
    LArDLHelper::TorchModel m_modelU;         
    LArDLHelper::TorchModel m_modelV;         
    LArDLHelper::TorchModel m_modelW;         
    int m_imageHeight;                        
    int m_imageWidth;                         
    float m_tileSize;                         
    bool m_visualize;                         
    bool m_useTrainingMode;                   
    std::string m_trainingOutputFile;         
};

} // namespace lar_dl_content

#endif // LAR_DL_HIT_TRACK_SHOWER_ID_ALGORITHM_H