RandomClusteringAlgorithm.cc

Go to the documentation of this file.

#include "Pandora/AlgorithmHeaders.h"

#include "larpandoracontent/LArReclustering/LArExample/RandomClusteringAlgorithm.h"

#include <algorithm>
#include <random>
#include <vector>

using namespace pandora;

namespace lar_content
{

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

StatusCode RandomClusteringAlgorithm::Run()
{
    const CaloHitList *pCaloHits{nullptr};
    PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::GetCurrentList(*this, pCaloHits));

    if (pCaloHits->size() < m_numNewClusters)
        return STATUS_CODE_SUCCESS;

    std::vector<unsigned int> idxs, splitIdxs;
    for (unsigned int i = 0; i < pCaloHits->size(); i++)
        idxs.emplace_back(i);
    std::shuffle(idxs.begin(), idxs.end(), std::default_random_engine(0));
    for (unsigned int i = 0; i < m_numNewClusters - 1; i++)
        splitIdxs.emplace_back(idxs.at(i));
    std::sort(splitIdxs.begin(), splitIdxs.end());

    std::vector<unsigned int>::const_iterator splitItr{splitIdxs.cbegin()};
    PandoraContentApi::Cluster::Parameters params;
    unsigned int cntr{0};
    for (const CaloHit *const pCaloHit : *pCaloHits)
    {
        if (splitItr != splitIdxs.cend() && cntr++ > *splitItr)
        {
            const Cluster *pCluster{nullptr};
            PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::Cluster::Create(*this, params, pCluster));
            params.m_caloHitList.clear();
            splitItr++;
        }
        params.m_caloHitList.emplace_back(pCaloHit);
    }
    const Cluster *pCluster{nullptr};
    PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::Cluster::Create(*this, params, pCluster));

    return STATUS_CODE_SUCCESS;
}

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

StatusCode RandomClusteringAlgorithm::ReadSettings(const TiXmlHandle xmlHandle)
{
    PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, XmlHelper::ReadValue(xmlHandle, "NumNewClusters", m_numNewClusters));
    if (m_numNewClusters == 0)
        return STATUS_CODE_INVALID_PARAMETER;

    return STATUS_CODE_SUCCESS;
}

} // namespace lar_content