LArPfoObjects.h

Go to the documentation of this file.

#ifndef LAR_PFO_OBJECTS_H
#define LAR_PFO_OBJECTS_H 1

#include "Objects/CartesianVector.h"
#include "Objects/TrackState.h"

#include <vector>

namespace pandora
{
class CaloHit;
}

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

namespace lar_content
{

class LArTrackState : public pandora::TrackState
{
public:
    LArTrackState(const pandora::CartesianVector &position, const pandora::CartesianVector &direction, const pandora::CaloHit *const pCaloHit);

    LArTrackState(const pandora::CartesianVector &position, const pandora::CartesianVector &direction);

    const pandora::CartesianVector &GetDirection() const;

    const pandora::CaloHit *GetCaloHit() const;

private:
    const pandora::CaloHit *m_pCaloHit;
};

typedef std::vector<LArTrackState> LArTrackStateVector;

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

class LArTrackTrajectoryPoint : public std::pair<float, LArTrackState>
{
public:
    LArTrackTrajectoryPoint(const float projectedDistance, const LArTrackState &larTrackState);

    LArTrackTrajectoryPoint(const float projectedDistance, const LArTrackState &larTrackState, const int index);

    int GetIndex() const;

private:
    int m_index; 
};

typedef std::vector<LArTrackTrajectoryPoint> LArTrackTrajectory;

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

class LArShowerPCA
{
public:
    LArShowerPCA(const pandora::CartesianVector &centroid, const pandora::CartesianVector &primaryAxis,
        const pandora::CartesianVector &secondaryAxis, const pandora::CartesianVector &tertiaryAxis, const pandora::CartesianVector &eigenvalues);

    const pandora::CartesianVector &GetCentroid() const;

    const pandora::CartesianVector &GetPrimaryAxis() const;

    const pandora::CartesianVector &GetSecondaryAxis() const;

    const pandora::CartesianVector &GetTertiaryAxis() const;

    const pandora::CartesianVector &GetEigenValues() const;

    const pandora::CartesianVector &GetAxisLengths() const;

    float GetPrimaryLength() const;

    float GetSecondaryLength() const;

    float GetTertiaryLength() const;

private:
    const pandora::CartesianVector m_centroid;      
    const pandora::CartesianVector m_primaryAxis;   
    const pandora::CartesianVector m_secondaryAxis; 
    const pandora::CartesianVector m_tertiaryAxis;  
    const pandora::CartesianVector m_eigenValues;   
    const pandora::CartesianVector m_axisLengths;   
};

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

inline LArTrackTrajectoryPoint::LArTrackTrajectoryPoint(const float projectedDistance, const LArTrackState &larTrackState) :
    std::pair<float, LArTrackState>(projectedDistance, larTrackState),
    m_index(-1)
{
}

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

inline LArTrackTrajectoryPoint::LArTrackTrajectoryPoint(const float projectedDistance, const LArTrackState &larTrackState, const int index) :
    std::pair<float, LArTrackState>(projectedDistance, larTrackState),
    m_index(index)
{
}

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

inline int LArTrackTrajectoryPoint::GetIndex() const
{
    return m_index;
}

} // namespace lar_content

#endif // #ifndef LAR_PFO_OBJECTS_H