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