CheatingClusterCreationAlgorithm class. More...

#include <CheatingClusterCreationAlgorithm.h>

Inheritance diagram for lar_content::CheatingClusterCreationAlgorithm:

Public Member Functions
	CheatingClusterCreationAlgorithm ()
	Default constructor. More...

Private Types
typedef std::unordered_map < const pandora::MCParticle *, pandora::CaloHitList >	MCParticleToHitListMap

Private Member Functions
pandora::StatusCode	Run ()

pandora::StatusCode	ReadSettings (const pandora::TiXmlHandle xmlHandle)

void	GetMCParticleToHitListMap (MCParticleToHitListMap &mcParticleToHitListMap) const
	Create map between each (primary) MC particle and associated calo hits. More...

void	SimpleMCParticleCollection (const pandora::CaloHit *const pCaloHit, const LArMCParticleHelper::MCRelationMap &mcPrimaryMap, MCParticleToHitListMap &mcParticleToHitListMap) const
	Simple mc particle collection, using main mc particle associated with each calo hit. More...

bool	SelectMCParticlesForClustering (const pandora::MCParticle *const pMCParticle) const
	Check whether mc particle is of a type specified for inclusion in cheated clustering. More...

void	CreateClusters (const MCParticleToHitListMap &mcParticleToHitListMap) const
	Create clusters based on information in the mc particle to hit list map. More...

Private Attributes
bool	m_collapseToPrimaryMCParticles
	Whether to collapse mc particle hierarchies to primary particles. More...

std::string	m_mcParticleListName
	The mc particle list name, required if want to collapse mc particle hierarchy. More...

pandora::IntVector	m_particleIdList
	list of particle ids of MCPFOs to be selected More...

Detailed Description

CheatingClusterCreationAlgorithm class.

Definition at line 23 of file CheatingClusterCreationAlgorithm.h.

Member Typedef Documentation

typedef std::unordered_map<const pandora::MCParticle *, pandora::CaloHitList> lar_content::CheatingClusterCreationAlgorithm::MCParticleToHitListMap

private

Definition at line 35 of file CheatingClusterCreationAlgorithm.h.

Constructor & Destructor Documentation

lar_content::CheatingClusterCreationAlgorithm::CheatingClusterCreationAlgorithm ( )

Default constructor.

Definition at line 18 of file CheatingClusterCreationAlgorithm.cc.

                                                                    : m_collapseToPrimaryMCParticles(false)
 {
 }

Member Function Documentation

void lar_content::CheatingClusterCreationAlgorithm::CreateClusters ( const MCParticleToHitListMap & mcParticleToHitListMap ) const

private

Create clusters based on information in the mc particle to hit list map.

Parameters

mcParticleToHitListMap the mc particle to hit list map

Definition at line 106 of file CheatingClusterCreationAlgorithm.cc.

 {
     MCParticleVector mcParticleVector;
     for (const auto &mapEntry : mcParticleToHitListMap)
         mcParticleVector.push_back(mapEntry.first);
     std::sort(mcParticleVector.begin(), mcParticleVector.end(), LArMCParticleHelper::SortByMomentum);
 
     for (const MCParticle *const pMCParticle : mcParticleVector)
     {
         const CaloHitList &caloHitList(mcParticleToHitListMap.at(pMCParticle));
 
         if (caloHitList.empty())
             continue;
 
         const Cluster *pCluster(nullptr);
         PandoraContentApi::Cluster::Parameters parameters;
         parameters.m_caloHitList = caloHitList;
         PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::Cluster::Create(*this, parameters, pCluster));
 
         PandoraContentApi::Cluster::Metadata metadata;
 
         switch (pMCParticle->GetParticleId())
         {
             case PHOTON:
             case E_PLUS:
             case E_MINUS:
             case MU_PLUS:
             case MU_MINUS:
                 metadata.m_particleId = pMCParticle->GetParticleId();
                 break;
             default:
                 break;
         }
 
         if (metadata.m_particleId.IsInitialized())
             PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::Cluster::AlterMetadata(*this, pCluster, metadata));
     }
 }

void lar_content::CheatingClusterCreationAlgorithm::GetMCParticleToHitListMap ( MCParticleToHitListMap & mcParticleToHitListMap ) const

private

Create map between each (primary) MC particle and associated calo hits.

Parameters

mcParticleToHitListMap to receive the mc particle to hit list map

Definition at line 35 of file CheatingClusterCreationAlgorithm.cc.

 {
     LArMCParticleHelper::MCRelationMap mcPrimaryMap;
 
     if (m_collapseToPrimaryMCParticles)
     {
         const MCParticleList *pMCParticleList(nullptr);
         PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::GetList(*this, m_mcParticleListName, pMCParticleList));
 
         LArMCParticleHelper::GetMCPrimaryMap(pMCParticleList, mcPrimaryMap);
     }
 
     const CaloHitList *pCaloHitList(nullptr);
     PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::GetCurrentList(*this, pCaloHitList));
 
     for (const CaloHit *const pCaloHit : *pCaloHitList)
     {
         try
         {
             if (!PandoraContentApi::IsAvailable(*this, pCaloHit))
                 continue;
 
             this->SimpleMCParticleCollection(pCaloHit, mcPrimaryMap, mcParticleToHitListMap);
         }
         catch (const StatusCodeException &)
         {
         }
     }
 }

StatusCode lar_content::CheatingClusterCreationAlgorithm::ReadSettings ( const pandora::TiXmlHandle xmlHandle )

private

Definition at line 147 of file CheatingClusterCreationAlgorithm.cc.

 {
     PANDORA_RETURN_RESULT_IF_AND_IF(STATUS_CODE_SUCCESS, STATUS_CODE_NOT_FOUND, !=,
         XmlHelper::ReadValue(xmlHandle, "CollapseToPrimaryMCParticles", m_collapseToPrimaryMCParticles));
 
     if (m_collapseToPrimaryMCParticles)
     {
         PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, XmlHelper::ReadValue(xmlHandle, "MCParticleListName", m_mcParticleListName));
     }
 
     PANDORA_RETURN_RESULT_IF_AND_IF(
         STATUS_CODE_SUCCESS, STATUS_CODE_NOT_FOUND, !=, XmlHelper::ReadVectorOfValues(xmlHandle, "ParticleIdList", m_particleIdList));
 
     return STATUS_CODE_SUCCESS;
 }

StatusCode lar_content::CheatingClusterCreationAlgorithm::Run ( )

private

Definition at line 24 of file CheatingClusterCreationAlgorithm.cc.

 {
     MCParticleToHitListMap mcParticleToHitListMap;
     this->GetMCParticleToHitListMap(mcParticleToHitListMap);
     this->CreateClusters(mcParticleToHitListMap);
 
     return STATUS_CODE_SUCCESS;
 }

bool lar_content::CheatingClusterCreationAlgorithm::SelectMCParticlesForClustering ( const pandora::MCParticle *const pMCParticle ) const

private

Check whether mc particle is of a type specified for inclusion in cheated clustering.

Parameters

pMCParticle the mc particle to hit list map

Returns: boolean

Definition at line 90 of file CheatingClusterCreationAlgorithm.cc.

 {
     if (m_particleIdList.empty())
         return true;
 
     for (const int particleId : m_particleIdList)
     {
         if (pMCParticle->GetParticleId() == particleId)
             return true;
     }
 
     return false;
 }

void lar_content::CheatingClusterCreationAlgorithm::SimpleMCParticleCollection	(	const pandora::CaloHit *const	pCaloHit,
		const LArMCParticleHelper::MCRelationMap &	mcPrimaryMap,
		MCParticleToHitListMap &	mcParticleToHitListMap
	)		const

private

Simple mc particle collection, using main mc particle associated with each calo hit.

Parameters

pCaloHit	address of the calo hit
mcPrimaryMap	the mapping between mc particles and their parents
mcParticleToHitListMap	the mc particle to hit list map

Definition at line 67 of file CheatingClusterCreationAlgorithm.cc.

 {
     const MCParticle *pMCParticle(MCParticleHelper::GetMainMCParticle(pCaloHit));
 
     if (!this->SelectMCParticlesForClustering(pMCParticle))
         return;
 
     if (m_collapseToPrimaryMCParticles)
     {
         LArMCParticleHelper::MCRelationMap::const_iterator primaryIter = mcPrimaryMap.find(pMCParticle);
 
         if (mcPrimaryMap.end() == primaryIter)
             throw StatusCodeException(STATUS_CODE_NOT_FOUND);
 
         pMCParticle = primaryIter->second;
     }
 
     mcParticleToHitListMap[pMCParticle].push_back(pCaloHit);
 }