RecursivePfoMopUpAlgorithm.cc

Go to the documentation of this file.

#include "Pandora/AlgorithmHeaders.h"

#include "larpandoracontent/LArHelpers/LArPfoHelper.h"

#include "larpandoracontent/LArThreeDReco/LArPfoMopUp/RecursivePfoMopUpAlgorithm.h"

using namespace pandora;

namespace lar_content
{

StatusCode RecursivePfoMopUpAlgorithm::Run()
{
    PfoMergeStatsList mergeStatsListBefore(this->GetPfoMergeStats());

    for (unsigned int iter = 0; iter < m_maxIterations; ++iter)
    {
        for (auto const &mopUpAlg : m_mopUpAlgorithms)
            PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::RunDaughterAlgorithm(*this, mopUpAlg));

        PfoMergeStatsList mergeStatsListAfter(this->GetPfoMergeStats());

        if (std::equal(mergeStatsListBefore.cbegin(), mergeStatsListBefore.cend(), mergeStatsListAfter.cbegin(), mergeStatsListAfter.cend(), PfoMergeStatsComp))
            break;

        mergeStatsListBefore = std::move(mergeStatsListAfter);
    }

    return STATUS_CODE_SUCCESS;
}

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

RecursivePfoMopUpAlgorithm::PfoMergeStatsList RecursivePfoMopUpAlgorithm::GetPfoMergeStats() const
{
    PfoMergeStatsList pfoMergeStatsList;

    for (auto const &pfoListName : m_pfoListNames)
    {
        const PfoList *pPfoList(nullptr);
        if (STATUS_CODE_SUCCESS != PandoraContentApi::GetList(*this, pfoListName, pPfoList))
            continue;

        for (const ParticleFlowObject *const pPfo : *pPfoList)
        {
            ClusterNumHitsList pfoHits;
            ClusterList clusterList;
            LArPfoHelper::GetTwoDClusterList(pPfo, clusterList);

            for (auto const &cluster : clusterList)
                pfoHits.push_back(cluster->GetNCaloHits());

            const PropertiesMap &pfoMeta(pPfo->GetPropertiesMap());
            const auto &trackScoreIter(pfoMeta.find("TrackScore"));
            const float trackScore(trackScoreIter != pfoMeta.end() ? trackScoreIter->second : -1.f);

            pfoMergeStatsList.emplace_back(PfoMergeStats{pfoHits, trackScore});
        }
    }
    return pfoMergeStatsList;
}

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

StatusCode RecursivePfoMopUpAlgorithm::ReadSettings(const pandora::TiXmlHandle xmlHandle)
{
    PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, XmlHelper::ProcessAlgorithmList(*this, xmlHandle, "MopUpAlgorithms", m_mopUpAlgorithms));

    PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, XmlHelper::ReadVectorOfValues(xmlHandle, "PfoListNames", m_pfoListNames));

    PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, XmlHelper::ReadValue(xmlHandle, "MaxIterations", m_maxIterations));

    return STATUS_CODE_SUCCESS;
}

} // namespace lar_content