LArThreeDSlidingConeFitResult.h

Go to the documentation of this file.

#ifndef LAR_THREE_D_SLIDING_CONE_FIT_RESULT_H
#define LAR_THREE_D_SLIDING_CONE_FIT_RESULT_H 1

#include "Api/PandoraApi.h"

#include "larpandoracontent/LArObjects/LArThreeDSlidingFitResult.h"

#include <list>
#include <unordered_map>

namespace lar_content
{

enum ConeSelection
{
    CONE_FORWARD_ONLY,
    CONE_BACKWARD_ONLY,
    CONE_BOTH_DIRECTIONS
};

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

class SimpleCone
{
public:
    SimpleCone(const pandora::CartesianVector &coneApex, const pandora::CartesianVector &coneDirection, const float coneLength, const float coneTanHalfAngle);

    const pandora::CartesianVector &GetConeApex() const;

    const pandora::CartesianVector &GetConeDirection() const;

    float GetConeLength() const;

    float GetConeTanHalfAngle() const;

    float GetMeanRT(const pandora::Cluster *const pCluster) const;

    float GetBoundedHitFraction(const pandora::Cluster *const pCluster) const;

    float GetBoundedHitFraction(const pandora::Cluster *const pCluster, const float coneLength, const float coneTanHalfAngle) const;

private:
    pandora::CartesianVector m_coneApex;      
    pandora::CartesianVector m_coneDirection; 
    float m_coneLength;                       
    float m_coneTanHalfAngle;                 
};

typedef std::vector<SimpleCone> SimpleConeList;

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

typedef std::map<int, pandora::TrackState> TrackStateMap;

class ThreeDSlidingConeFitResult
{
public:
    template <typename T>
    ThreeDSlidingConeFitResult(const T *const pT, const unsigned int slidingFitWindow, const float slidingFitLayerPitch);

    const ThreeDSlidingFitResult &GetSlidingFitResult() const;

    const TrackStateMap &GetTrackStateMap() const;

    void GetSimpleConeList(const unsigned int nLayersForConeFit, const unsigned int nCones, const ConeSelection coneSelection,
        SimpleConeList &simpleConeList, const float tanHalfAngle = 0.5f, const bool legacyMode = true) const;

private:
    typedef std::list<pandora::TrackState> TrackStateLinkedList; 

    const ThreeDSlidingFitResult m_slidingFitResult; 
    TrackStateMap m_trackStateMap;                   
};

typedef std::vector<ThreeDSlidingConeFitResult> ThreeDSlidingConeFitResultList;
typedef std::unordered_map<const pandora::Cluster *, ThreeDSlidingConeFitResult> ThreeDSlidingConeFitResultMap;

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

inline SimpleCone::SimpleCone(const pandora::CartesianVector &coneApex, const pandora::CartesianVector &coneDirection,
    const float coneLength, const float coneTanHalfAngle) :
    m_coneApex(coneApex),
    m_coneDirection(coneDirection),
    m_coneLength(coneLength),
    m_coneTanHalfAngle(coneTanHalfAngle)
{
}

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

inline const pandora::CartesianVector &SimpleCone::GetConeApex() const
{
    return m_coneApex;
}

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

inline const pandora::CartesianVector &SimpleCone::GetConeDirection() const
{
    return m_coneDirection;
}

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

inline float SimpleCone::GetConeLength() const
{
    return m_coneLength;
}

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

inline float SimpleCone::GetConeTanHalfAngle() const
{
    return m_coneTanHalfAngle;
}

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

inline float SimpleCone::GetBoundedHitFraction(const pandora::Cluster *const pCluster) const
{
    return this->GetBoundedHitFraction(pCluster, this->GetConeLength(), this->GetConeTanHalfAngle());
}

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

inline const ThreeDSlidingFitResult &ThreeDSlidingConeFitResult::GetSlidingFitResult() const
{
    return m_slidingFitResult;
}

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

inline const TrackStateMap &ThreeDSlidingConeFitResult::GetTrackStateMap() const
{
    return m_trackStateMap;
}

} // namespace lar_content

#endif // #ifndef LAR_THREE_D_SLIDING_CONE_FIT_RESULT_H