ClusterAssociation.h

Go to the documentation of this file.

 
#ifndef LAR_CLUSTER_ASSOCIATION_H
#define LAR_CLUSTER_ASSOCIATION_H 1
 
#include "larpandoracontent/LArHelpers/LArClusterHelper.h"
 
namespace lar_content
{
class ClusterAssociation
{
public:
    ClusterAssociation();
 
    ClusterAssociation(const pandora::CartesianVector &upstreamMergePoint, const pandora::CartesianVector &upstreamMergeDirection,
        const pandora::CartesianVector &downstreamMergePoint, const pandora::CartesianVector &downstreamMergeDirection);
 
    const pandora::CartesianVector GetUpstreamMergePoint() const;
 
    const pandora::CartesianVector GetUpstreamMergeDirection() const;
 
    const pandora::CartesianVector GetDownstreamMergePoint() const;
 
    const pandora::CartesianVector GetDownstreamMergeDirection() const;
 
    const pandora::CartesianVector GetConnectingLineDirection() const;
 
    void SetUpstreamMergePoint(const pandora::CartesianVector &upstreamMergePoint);
 
    void SetDownstreamMergePoint(const pandora::CartesianVector &downstreamMergePoint);
 
    bool operator==(const ClusterAssociation &clusterAssociation) const;
    bool operator<(const ClusterAssociation &clusterAssociation) const;
 
protected:
    void UpdateConnectingLine();
 
    pandora::CartesianVector m_upstreamMergePoint; 
    pandora::CartesianVector m_upstreamMergeDirection; 
    pandora::CartesianVector m_downstreamMergePoint; 
    pandora::CartesianVector m_downstreamMergeDirection; 
    pandora::CartesianVector m_connectingLineDirection; 
};
 
//------------------------------------------------------------------------------------------------------------------------------------------
//------------------------------------------------------------------------------------------------------------------------------------------
 
class ClusterPairAssociation : public ClusterAssociation
{
public:
    ClusterPairAssociation();
 
    ClusterPairAssociation(const pandora::CartesianVector &upstreamMergePoint, const pandora::CartesianVector &upstreamMergeDirection,
        const pandora::CartesianVector &downstreamMergePoint, const pandora::CartesianVector &downstreamMergeDirection,
        const pandora::Cluster *pUpstreamCluster, const pandora::Cluster *pDownstreamCluster);
 
    const pandora::Cluster *GetUpstreamCluster() const;
 
    const pandora::Cluster *GetDownstreamCluster() const;
 
private:
    const pandora::Cluster *m_pUpstreamCluster;   
    const pandora::Cluster *m_pDownstreamCluster; 
};
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline ClusterAssociation::ClusterAssociation() :
    m_upstreamMergePoint(pandora::CartesianVector(0.f, 0.f, 0.f)),
    m_upstreamMergeDirection(pandora::CartesianVector(0.f, 0.f, 0.f)),
    m_downstreamMergePoint(pandora::CartesianVector(0.f, 0.f, 0.f)),
    m_downstreamMergeDirection(pandora::CartesianVector(0.f, 0.f, 0.f)),
    m_connectingLineDirection(pandora::CartesianVector(0.f, 0.f, 0.f))
{
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline ClusterAssociation::ClusterAssociation(const pandora::CartesianVector &upstreamMergePoint, const pandora::CartesianVector &upstreamMergeDirection,
    const pandora::CartesianVector &downstreamMergePoint, const pandora::CartesianVector &downstreamMergeDirection) :
    m_upstreamMergePoint(upstreamMergePoint),
    m_upstreamMergeDirection(upstreamMergeDirection),
    m_downstreamMergePoint(downstreamMergePoint),
    m_downstreamMergeDirection(downstreamMergeDirection),
    m_connectingLineDirection(0.f, 0.f, 0.f)
{
    const pandora::CartesianVector connectingLineDirection(
        m_downstreamMergePoint.GetX() - m_upstreamMergePoint.GetX(), 0.f, m_downstreamMergePoint.GetZ() - m_upstreamMergePoint.GetZ());
    m_connectingLineDirection = connectingLineDirection.GetUnitVector();
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline bool ClusterAssociation::operator==(const ClusterAssociation &clusterAssociation) const
{
    return (m_upstreamMergePoint == clusterAssociation.GetUpstreamMergePoint() &&
            m_upstreamMergeDirection == clusterAssociation.GetUpstreamMergeDirection() &&
            m_downstreamMergePoint == clusterAssociation.GetDownstreamMergePoint() &&
            m_downstreamMergeDirection == clusterAssociation.GetDownstreamMergeDirection());
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline bool ClusterAssociation::operator<(const ClusterAssociation &clusterAssociation) const
{
    return (LArClusterHelper::SortCoordinatesByPosition(m_upstreamMergePoint, clusterAssociation.GetUpstreamMergePoint()));
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const pandora::CartesianVector ClusterAssociation::GetUpstreamMergePoint() const
{
    return m_upstreamMergePoint;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const pandora::CartesianVector ClusterAssociation::GetUpstreamMergeDirection() const
{
    return m_upstreamMergeDirection;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const pandora::CartesianVector ClusterAssociation::GetDownstreamMergePoint() const
{
    return m_downstreamMergePoint;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const pandora::CartesianVector ClusterAssociation::GetDownstreamMergeDirection() const
{
    return m_downstreamMergeDirection;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const pandora::CartesianVector ClusterAssociation::GetConnectingLineDirection() const
{
    return m_connectingLineDirection;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline void ClusterAssociation::SetUpstreamMergePoint(const pandora::CartesianVector &upstreamMergePoint)
{
    m_upstreamMergePoint = upstreamMergePoint;
    this->UpdateConnectingLine();
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline void ClusterAssociation::SetDownstreamMergePoint(const pandora::CartesianVector &downstreamMergePoint)
{
    m_downstreamMergePoint = downstreamMergePoint;
    this->UpdateConnectingLine();
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline void ClusterAssociation::UpdateConnectingLine()
{
    const pandora::CartesianVector connectingLineDirection(
        m_downstreamMergePoint.GetX() - m_upstreamMergePoint.GetX(), 0.f, m_downstreamMergePoint.GetZ() - m_upstreamMergePoint.GetZ());
    m_connectingLineDirection = connectingLineDirection.GetUnitVector();
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline ClusterPairAssociation::ClusterPairAssociation(const pandora::CartesianVector &upstreamMergePoint,
    const pandora::CartesianVector &upstreamMergeDirection, const pandora::CartesianVector &downstreamMergePoint,
    const pandora::CartesianVector &downstreamMergeDirection, const pandora::Cluster *pUpstreamCluster, const pandora::Cluster *pDownstreamCluster) :
    ClusterAssociation(upstreamMergePoint, upstreamMergeDirection, downstreamMergePoint, downstreamMergeDirection),
    m_pUpstreamCluster(pUpstreamCluster),
    m_pDownstreamCluster(pDownstreamCluster)
{
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline ClusterPairAssociation::ClusterPairAssociation() : ClusterAssociation(), m_pUpstreamCluster(nullptr), m_pDownstreamCluster(nullptr)
{
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const pandora::Cluster *ClusterPairAssociation::GetUpstreamCluster() const
{
    return m_pUpstreamCluster;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
inline const pandora::Cluster *ClusterPairAssociation::GetDownstreamCluster() const
{
    return m_pDownstreamCluster;
}
 
} // namespace lar_content
 
#endif // #ifndef LAR_CLUSTER_ASSOCIATION_H