TwoViewSimpleTracksTool.cc

Go to the documentation of this file.

 
#include "Pandora/AlgorithmHeaders.h"
 
#include "larpandoracontent/LArThreeDReco/LArTwoViewMatching/TwoViewSimpleTracksTool.h"
 
using namespace pandora;
 
namespace lar_content
{
 
TwoViewSimpleTracksTool::TwoViewSimpleTracksTool() :
    m_minMatchedFraction(0.2f),
    m_minMatchingScore(0.9f),
    m_minMatchedSamplingPoints(5),
    m_minXOverlapFraction(0.5f)
{
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
bool TwoViewSimpleTracksTool::Run(TwoViewTransverseTracksAlgorithm *const pAlgorithm, MatrixType &overlapMatrix)
{
    if (PandoraContentApi::GetSettings(*pAlgorithm)->ShouldDisplayAlgorithmInfo())
        std::cout << "----> Running Algorithm Tool: " << this->GetInstanceName() << ", " << this->GetType() << std::endl;
 
    ProtoParticleVector protoParticleVector;
    this->FindBestTrack(overlapMatrix, protoParticleVector);
 
    return pAlgorithm->CreateThreeDParticles(protoParticleVector);
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
void TwoViewSimpleTracksTool::FindBestTrack(const MatrixType &overlapMatrix, ProtoParticleVector &protoParticleVector) const
{
    ClusterVector sortedKeyClusters;
    overlapMatrix.GetSortedKeyClusters(sortedKeyClusters);
 
    for (const Cluster *const pKeyCluster : sortedKeyClusters)
    {
        if (!pKeyCluster->IsAvailable())
            continue;
 
        unsigned int n0(0), n1(0);
        MatrixType::ElementList elementList;
        overlapMatrix.GetConnectedElements(pKeyCluster, true, elementList, n0, n1);
 
        if (elementList.empty())
            continue;
 
        for (MatrixType::ElementList::const_reverse_iterator iIter = elementList.rbegin(); iIter != elementList.rend(); ++iIter)
        {
            if (this->PassesElementCuts(iIter))
            {
                if ((nullptr == iIter->GetCluster1()) || (nullptr == iIter->GetCluster2()))
                    continue;
 
                ProtoParticle protoParticle;
                protoParticle.m_clusterList.push_back(iIter->GetCluster1());
                protoParticle.m_clusterList.push_back(iIter->GetCluster2());
                protoParticleVector.push_back(protoParticle);
 
                return;
            }
        }
    }
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
bool TwoViewSimpleTracksTool::PassesElementCuts(MatrixType::ElementList::const_reverse_iterator eIter) const
{
    if (!eIter->GetOverlapResult().IsInitialized())
        return false;
 
    if (eIter->GetOverlapResult().GetLocallyMatchedFraction() < m_minMatchedFraction)
        return false;
 
    if (eIter->GetOverlapResult().GetMatchingScore() < m_minMatchingScore)
        return false;
 
    if (eIter->GetOverlapResult().GetNMatchedReUpsampledSamplingPoints() < m_minMatchedSamplingPoints)
        return false;
 
    const TwoViewXOverlap &xOverlap(eIter->GetOverlapResult().GetTwoViewXOverlap());
 
    if (!((xOverlap.GetXOverlapFraction0() > m_minXOverlapFraction) && (xOverlap.GetXOverlapFraction1() > m_minXOverlapFraction)))
    {
        return false;
    }
 
    return true;
}
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
StatusCode TwoViewSimpleTracksTool::ReadSettings(const TiXmlHandle xmlHandle)
{
    PANDORA_RETURN_RESULT_IF_AND_IF(
        STATUS_CODE_SUCCESS, STATUS_CODE_NOT_FOUND, !=, XmlHelper::ReadValue(xmlHandle, "MinMatchedFraction", m_minMatchedFraction));
 
    PANDORA_RETURN_RESULT_IF_AND_IF(
        STATUS_CODE_SUCCESS, STATUS_CODE_NOT_FOUND, !=, XmlHelper::ReadValue(xmlHandle, "MinMatchingScore", m_minMatchingScore));
 
    PANDORA_RETURN_RESULT_IF_AND_IF(STATUS_CODE_SUCCESS, STATUS_CODE_NOT_FOUND, !=,
        XmlHelper::ReadValue(xmlHandle, "MinMatchedSamplingPoints", m_minMatchedSamplingPoints));
 
    PANDORA_RETURN_RESULT_IF_AND_IF(
        STATUS_CODE_SUCCESS, STATUS_CODE_NOT_FOUND, !=, XmlHelper::ReadValue(xmlHandle, "MinXOverlapFraction", m_minXOverlapFraction));
 
    return STATUS_CODE_SUCCESS;
}
 
} // namespace lar_content