LArTwoDSlidingShowerFitResult.h

Go to the documentation of this file.

#ifndef LAR_TWO_D_SLIDING_SHOWER_FIT_RESULT_H
#define LAR_TWO_D_SLIDING_SHOWER_FIT_RESULT_H 1

#include "Api/PandoraApi.h"

#include "larpandoracontent/LArObjects/LArTwoDSlidingFitResult.h"

#include <unordered_map>

namespace lar_content
{

enum ShowerEdge
{
    POSITIVE_SHOWER_EDGE,
    NEGATIVE_SHOWER_EDGE
};

//------------------------------------------------------------------------------------------------------------------------------------------

class TwoDSlidingShowerFitResult
{
public:
    template <typename T>
    TwoDSlidingShowerFitResult(const T *const pT, const unsigned int slidingFitWindow, const float slidingFitLayerPitch,
        const float showerEdgeMultiplier = 1.f);

    const TwoDSlidingFitResult &GetShowerFitResult() const;

    const TwoDSlidingFitResult &GetNegativeEdgeFitResult() const;

    const TwoDSlidingFitResult &GetPositiveEdgeFitResult() const;

    void GetShowerEdges(const float x, const bool widenIfAmbiguity, pandora::FloatVector &edgePositions) const;

private:
    static TwoDSlidingFitResult LArTwoDShowerEdgeFit(const pandora::Cluster *const pCluster, const TwoDSlidingFitResult &fullShowerFit,
        const ShowerEdge showerEdge, const float showerEdgeMultiplier);

    static TwoDSlidingFitResult LArTwoDShowerEdgeFit(const pandora::CartesianPointVector *const pPointVector, const TwoDSlidingFitResult &fullShowerFit,
        const ShowerEdge showerEdge, const float showerEdgeMultiplier);

    typedef std::pair<float, float> FitCoordinate;
    typedef std::vector<FitCoordinate> FitCoordinateList;
    typedef std::map<int, FitCoordinateList> FitCoordinateMap;

    TwoDSlidingFitResult    m_showerFitResult;              
    TwoDSlidingFitResult    m_negativeEdgeFitResult;        
    TwoDSlidingFitResult    m_positiveEdgeFitResult;        
};

typedef std::vector<TwoDSlidingShowerFitResult> TwoDSlidingShowerFitResultList;
typedef std::unordered_map<const pandora::Cluster*, TwoDSlidingShowerFitResult> TwoDSlidingShowerFitResultMap;

//------------------------------------------------------------------------------------------------------------------------------------------

class ShowerExtent
{
public:
    ShowerExtent(const float xCoordinate, const float highEdgeZ, const float lowEdgeZ);

    float GetXCoordinate() const;

    float GetHighEdgeZ() const;

    float GetLowEdgeZ() const;

private:
    float   m_xCoordinate;      
    float   m_highEdgeZ;        
    float   m_lowEdgeZ;         
};

typedef std::map<int, ShowerExtent> ShowerPositionMap;

//------------------------------------------------------------------------------------------------------------------------------------------
//------------------------------------------------------------------------------------------------------------------------------------------

inline const TwoDSlidingFitResult &TwoDSlidingShowerFitResult::GetShowerFitResult() const
{
    return m_showerFitResult;
}

//------------------------------------------------------------------------------------------------------------------------------------------

inline const TwoDSlidingFitResult &TwoDSlidingShowerFitResult::GetNegativeEdgeFitResult() const
{
    return m_negativeEdgeFitResult;
}

//------------------------------------------------------------------------------------------------------------------------------------------

inline const TwoDSlidingFitResult &TwoDSlidingShowerFitResult::GetPositiveEdgeFitResult() const
{
    return m_positiveEdgeFitResult;
}

//------------------------------------------------------------------------------------------------------------------------------------------
//------------------------------------------------------------------------------------------------------------------------------------------

inline ShowerExtent::ShowerExtent(const float xCoordinate, const float edge1, const float edge2) :
    m_xCoordinate(xCoordinate),
    m_highEdgeZ(std::max(edge1, edge2)),
    m_lowEdgeZ(std::min(edge1, edge2))
{
}

//------------------------------------------------------------------------------------------------------------------------------------------

inline float ShowerExtent::GetXCoordinate() const
{
    return m_xCoordinate;
}

//------------------------------------------------------------------------------------------------------------------------------------------

inline float ShowerExtent::GetHighEdgeZ() const
{
    return m_highEdgeZ;
}

//------------------------------------------------------------------------------------------------------------------------------------------

inline float ShowerExtent::GetLowEdgeZ() const
{
    return m_lowEdgeZ;
}

} // namespace lar_content

#endif // #ifndef LAR_TWO_D_SLIDING_SHOWER_FIT_RESULT_H