DlSecondaryVertexingAlgorithm.h

Go to the documentation of this file.

#ifndef LAR_DL_SECONDARY_VERTEXING_ALGORITHM_H
#define LAR_DL_SECONDARY_VERTEXING_ALGORITHM_H 1

#include "Pandora/Algorithm.h"
#include "Pandora/AlgorithmHeaders.h"

#include "larpandoracontent/LArHelpers/LArMCParticleHelper.h"

#include "larpandoradlcontent/LArHelpers/LArDLHelper.h"
#include "larpandoradlcontent/LArVertex/DlVertexingBaseAlgorithm.h"

#include <random>

using namespace lar_content;

namespace lar_dl_content
{
class DlSecondaryVertexingAlgorithm : public DlVertexingBaseAlgorithm
{
public:
    DlSecondaryVertexingAlgorithm();

    ~DlSecondaryVertexingAlgorithm();

private:
    class Canvas
    {
    public:
        Canvas(const pandora::HitType view, const int width, const int height, const int colOffset, const int rowOffset, const float xMin,
            const float xMax, const float zMin, const float zMax);

        virtual ~Canvas();

        pandora::HitType m_view;
        float **m_canvas;
        bool **m_visited;
        const int m_width;
        const int m_height;
        const int m_colOffset;
        const int m_rowOffset;
        const float m_xMin;
        const float m_xMax;
        const float m_zMin;
        const float m_zMax;
    };

    typedef std::map<pandora::HitType, Canvas *> CanvasViewMap;

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

    /*
     *  @brief  Create input for the network from a calo hit list
     *
     *  @param  caloHits The CaloHitList from which the input should be made
     *  @param  view The wire plane view
     *  @param  xMin The minimum x coordinate for the hits
     *  @param  xMax The maximum x coordinate for the hits
     *  @param  zMin The minimum x coordinate for the hits
     *  @param  zMax The maximum x coordinate for the hits
     *  @param  networkInput The TorchInput object to populate
     *  @param  pixelVector The output vector of populated pixels
     *
     *  @return The StatusCode resulting from the function
     **/
    pandora::StatusCode MakeNetworkInputFromHits(const pandora::CaloHitList &caloHits, const pandora::HitType view, const float xMin,
        const float xMax, const float zMin, const float zMax, LArDLHelper::TorchInput &networkInput, PixelVector &pixelVector) const;

    /*
     *  @brief  Create a list of vertices from canvases
     *
     *  @param  canvases The input canvases
     *  @param  positionVector The output vector of wire plane positions
     *
     *  @return The StatusCode resulting from the function
     **/
    pandora::StatusCode GetNetworkVertices(const CanvasViewMap &canvases, pandora::CartesianPointVector &positionVector) const;

    /*
     *  @brief  Create a list of vertices from a canvas
     *
     *  @param  canvases The input canvases
     *  @param  positionVector The output vector of wire plane positions
     *
     *  @return The StatusCode resulting from the function
     **/
    pandora::StatusCode GetVerticesFromCanvas(const Canvas &canvas, pandora::CartesianPointVector &vertices) const;

    bool GrowPeak(const Canvas &canvas, int col, int row, float intensity, std::vector<std::pair<int, int>> &peak) const;

    pandora::StatusCode MakeCandidateVertexList(const pandora::CartesianPointVector &positions);

    int m_event;                
    bool m_visualise;           
    bool m_writeTree;           
    std::string m_rootTreeName; 
    std::string m_rootFileName; 
    std::mt19937 m_rng;         
};

} // namespace lar_dl_content

#endif // LAR_DL_SECONDARY_VERTEXING_ALGORITHM_H