EnergyKickFeatureTool.h

Go to the documentation of this file.

#ifndef LAR_ENERGY_KICK_FEATURE_TOOL_H
#define LAR_ENERGY_KICK_FEATURE_TOOL_H 1

#include "larpandoracontent/LArVertex/VertexSelectionBaseAlgorithm.h"

namespace lar_content
{

class EnergyKickFeatureTool : public VertexSelectionBaseAlgorithm::VertexFeatureTool
{
public:
    EnergyKickFeatureTool();

    void Run(LArMvaHelper::MvaFeatureVector &featureVector, const VertexSelectionBaseAlgorithm *const pAlgorithm, const pandora::Vertex *const pVertex,
        const VertexSelectionBaseAlgorithm::SlidingFitDataListMap &slidingFitDataListMap, const VertexSelectionBaseAlgorithm::ClusterListMap &,
        const VertexSelectionBaseAlgorithm::KDTreeMap &, const VertexSelectionBaseAlgorithm::ShowerClusterListMap &, const float, float &);

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

    float GetEnergyKickForView(
        const pandora::CartesianVector &vertexPosition2D, const VertexSelectionBaseAlgorithm::SlidingFitDataList &slidingFitDataList) const;

    void IncrementEnergyKickParameters(const pandora::Cluster *const pCluster, const pandora::CartesianVector &clusterDisplacement,
        const pandora::CartesianVector &clusterDirection, float &totEnergyKick, float &totEnergy, float &totHitKick, unsigned int &totHits) const;

    float m_rOffset; 
    float m_xOffset; 
};

} // namespace lar_content

#endif // #ifndef LAR_ENERGY_KICK_FEATURE_TOOL_H