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