CheatingPfoCharacterisationAlgorithm class. More...

#include "CheatingPfoCharacterisationAlgorithm.h"

Inheritance diagram for lar_content::CheatingPfoCharacterisationAlgorithm:

Protected Member Functions
pandora::StatusCode	Run ()

virtual bool	IsClearTrack3x2D (const pandora::ParticleFlowObject *const pPfo) const
	Whether pfo is identified as a clear track using its three clusters. More...

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

Protected Attributes
std::string	m_trackPfoListName
	The track pfo list name. More...

std::string	m_showerPfoListName
	The shower pfo list name. More...

pandora::StringVector	m_inputPfoListNames
	The names of the input pfo lists. More...

bool	m_updateClusterIds
	Whether to update daughter cluster particle id labels to match pfo id. More...

bool	m_postBranchAddition
	Whether to use configuration for shower clusters post branch addition. More...

bool	m_useThreeDInformation
	Whether to use PFO and 3D information or clusters for characterisation. More...

unsigned int	m_minTrackLikeViews
	The minimum number of track-like views to declare a pfo as track-like. More...

Private Member Functions
bool	IsClearTrack (const pandora::ParticleFlowObject *const pPfo) const
	Whether pfo is identified as a clear track. More...

bool	IsClearTrack (const pandora::Cluster *const pCluster) const
	Whether cluster is identified as a clear track. More...

Detailed Description

CheatingPfoCharacterisationAlgorithm class.

Definition at line 19 of file CheatingPfoCharacterisationAlgorithm.h.

Member Function Documentation

bool lar_content::CheatingPfoCharacterisationAlgorithm::IsClearTrack ( const pandora::ParticleFlowObject *const pPfo ) const

privatevirtual

Whether pfo is identified as a clear track.

Parameters

pPfo	address of the relevant pfo

Returns: boolean

Implements lar_content::PfoCharacterisationBaseAlgorithm.

Definition at line 21 of file CheatingPfoCharacterisationAlgorithm.cc.

References util::abs(), f, and weight.

 {
     CaloHitList caloHitList;
     LArPfoHelper::GetCaloHits(pPfo, TPC_VIEW_U, caloHitList);
     LArPfoHelper::GetCaloHits(pPfo, TPC_VIEW_V, caloHitList);
     LArPfoHelper::GetCaloHits(pPfo, TPC_VIEW_W, caloHitList);
 
     MCParticleWeightMap mcParticleWeightMap;
 
     for (const CaloHit *const pCaloHit : caloHitList)
     {
         for (const MCParticleWeightMap::value_type &mapEntry : pCaloHit->GetMCParticleWeightMap())
             mcParticleWeightMap[mapEntry.first] += mapEntry.second;
     }
 
     float bestWeight(0.f);
     const MCParticle *pBestMCParticle(nullptr);
 
     MCParticleList mcParticleList;
     for (const auto &mapEntry : mcParticleWeightMap)
         mcParticleList.push_back(mapEntry.first);
     mcParticleList.sort(LArMCParticleHelper::SortByMomentum);
 
     for (const MCParticle *const pMCParticle : mcParticleList)
     {
         const float weight(mcParticleWeightMap.at(pMCParticle));
 
         if (weight > bestWeight)
         {
             pBestMCParticle = pMCParticle;
             bestWeight = weight;
         }
     }
 
     if (!pBestMCParticle)
         return false;
 
     return ((PHOTON != pBestMCParticle->GetParticleId()) && (E_MINUS != std::abs(pBestMCParticle->GetParticleId())));
 }

bool lar_content::CheatingPfoCharacterisationAlgorithm::IsClearTrack ( const pandora::Cluster *const pCluster ) const

privatevirtual

Whether cluster is identified as a clear track.

Parameters

pCluster address of the relevant cluster

Returns: boolean

Implements lar_content::PfoCharacterisationBaseAlgorithm.

bool lar_content::PfoCharacterisationBaseAlgorithm::IsClearTrack3x2D ( const pandora::ParticleFlowObject *const pPfo ) const

protectedvirtualinherited

Whether pfo is identified as a clear track using its three clusters.

Parameters

pPfo	address of the relevant pfo

Returns: boolean

Definition at line 108 of file PfoCharacterisationBaseAlgorithm.cc.

References lar_content::LArClusterHelper::GetClusterHitType(), lar_content::LArPfoHelper::GetTwoDClusterList(), lar_content::PfoCharacterisationBaseAlgorithm::IsClearTrack(), and lar_content::PfoCharacterisationBaseAlgorithm::m_minTrackLikeViews.

Referenced by lar_content::PfoCharacterisationBaseAlgorithm::Run().

 {
     ClusterList twoDClusterList;
     LArPfoHelper::GetTwoDClusterList(pPfo, twoDClusterList);
 
     typedef std::set<pandora::HitType> HitTypeSet;
     HitTypeSet hitTypeSet;
 
     unsigned int nTrackLikeViews(0);
     for (const Cluster *const pCluster : twoDClusterList)
     {
         const HitType hitType(LArClusterHelper::GetClusterHitType(pCluster));
         if (!hitTypeSet.insert(hitType).second)
             continue;
 
         if (this->IsClearTrack(pCluster))
             ++nTrackLikeViews;
 
         if (nTrackLikeViews >= m_minTrackLikeViews)
             return true;
     }
 
     return false;
 }

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

protectedinherited

Definition at line 135 of file PfoCharacterisationBaseAlgorithm.cc.

References lar_content::PfoCharacterisationBaseAlgorithm::m_inputPfoListNames, lar_content::PfoCharacterisationBaseAlgorithm::m_minTrackLikeViews, lar_content::PfoCharacterisationBaseAlgorithm::m_showerPfoListName, lar_content::PfoCharacterisationBaseAlgorithm::m_trackPfoListName, lar_content::PfoCharacterisationBaseAlgorithm::m_updateClusterIds, and lar_content::PfoCharacterisationBaseAlgorithm::m_useThreeDInformation.

Referenced by lar_content::CutPfoCharacterisationAlgorithm::ReadSettings(), and lar_content::MvaPfoCharacterisationAlgorithm< T >::ReadSettings().

 {
     PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, XmlHelper::ReadValue(xmlHandle, "TrackPfoListName", m_trackPfoListName));
     m_inputPfoListNames.push_back(m_trackPfoListName);
 
     PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, XmlHelper::ReadValue(xmlHandle, "ShowerPfoListName", m_showerPfoListName));
     m_inputPfoListNames.push_back(m_showerPfoListName);
 
     PANDORA_RETURN_RESULT_IF_AND_IF(
         STATUS_CODE_SUCCESS, STATUS_CODE_NOT_FOUND, !=, XmlHelper::ReadValue(xmlHandle, "UpdateClusterIds", m_updateClusterIds));
 
     PANDORA_RETURN_RESULT_IF_AND_IF(
         STATUS_CODE_SUCCESS, STATUS_CODE_NOT_FOUND, !=, XmlHelper::ReadValue(xmlHandle, "MinTrackLikeViews", m_minTrackLikeViews));
 
     PANDORA_RETURN_RESULT_IF_AND_IF(
         STATUS_CODE_SUCCESS, STATUS_CODE_NOT_FOUND, !=, XmlHelper::ReadValue(xmlHandle, "UseThreeDInformation", m_useThreeDInformation));
 
     return STATUS_CODE_SUCCESS;
 }

StatusCode lar_content::PfoCharacterisationBaseAlgorithm::Run ( )

protectedinherited

Definition at line 39 of file PfoCharacterisationBaseAlgorithm.cc.

References lar_content::LArPfoHelper::GetTwoDClusterList(), lar_content::PfoCharacterisationBaseAlgorithm::IsClearTrack(), lar_content::PfoCharacterisationBaseAlgorithm::IsClearTrack3x2D(), lar_content::PfoCharacterisationBaseAlgorithm::m_inputPfoListNames, lar_content::PfoCharacterisationBaseAlgorithm::m_showerPfoListName, lar_content::PfoCharacterisationBaseAlgorithm::m_trackPfoListName, lar_content::PfoCharacterisationBaseAlgorithm::m_updateClusterIds, and lar_content::PfoCharacterisationBaseAlgorithm::m_useThreeDInformation.

 {
     PfoList tracksToShowers, showersToTracks;
 
     for (const std::string &pfoListName : m_inputPfoListNames)
     {
         const PfoList *pPfoList(nullptr);
         PANDORA_RETURN_RESULT_IF_AND_IF(STATUS_CODE_SUCCESS, STATUS_CODE_NOT_INITIALIZED, !=, PandoraContentApi::GetList(*this, pfoListName, pPfoList));
 
         if (!pPfoList || pPfoList->empty())
         {
             if (PandoraContentApi::GetSettings(*this)->ShouldDisplayAlgorithmInfo())
                 std::cout << "PfoCharacterisationBaseAlgorithm: unable to find pfo list " << pfoListName << std::endl;
 
             continue;
         }
 
         for (const ParticleFlowObject *const pPfo : *pPfoList)
         {
             PandoraContentApi::ParticleFlowObject::Metadata pfoMetadata;
             const bool isTrackLike(m_useThreeDInformation ? this->IsClearTrack(pPfo) : this->IsClearTrack3x2D(pPfo));
 
             if (isTrackLike)
             {
                 pfoMetadata.m_particleId = MU_MINUS;
 
                 if (m_showerPfoListName == pfoListName)
                     showersToTracks.push_back(pPfo);
             }
             else
             {
                 pfoMetadata.m_particleId = E_MINUS;
 
                 if (m_trackPfoListName == pfoListName)
                     tracksToShowers.push_back(pPfo);
             }
 
             if (pPfo->GetParticleId() != pfoMetadata.m_particleId.Get())
                 PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::ParticleFlowObject::AlterMetadata(*this, pPfo, pfoMetadata));
 
             if (!m_updateClusterIds)
                 continue;
 
             ClusterList twoDClusterList;
             LArPfoHelper::GetTwoDClusterList(pPfo, twoDClusterList);
 
             for (const Cluster *const pCluster : twoDClusterList)
             {
                 if (pCluster->GetParticleId() == pfoMetadata.m_particleId.Get())
                     continue;
 
                 PandoraContentApi::Cluster::Metadata clusterMetadata;
                 clusterMetadata.m_particleId = pfoMetadata.m_particleId.Get();
                 PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::Cluster::AlterMetadata(*this, pCluster, clusterMetadata));
             }
         }
     }
 
     if (!tracksToShowers.empty())
         PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::SaveList(*this, m_trackPfoListName, m_showerPfoListName, tracksToShowers));
 
     if (!showersToTracks.empty())
         PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::SaveList(*this, m_showerPfoListName, m_trackPfoListName, showersToTracks));
 
     return STATUS_CODE_SUCCESS;
 }

Member Data Documentation

pandora::StringVector lar_content::PfoCharacterisationBaseAlgorithm::m_inputPfoListNames

protectedinherited

The names of the input pfo lists.

Definition at line 66 of file PfoCharacterisationBaseAlgorithm.h.

Referenced by lar_content::PfoCharacterisationBaseAlgorithm::ReadSettings(), and lar_content::PfoCharacterisationBaseAlgorithm::Run().

unsigned int lar_content::PfoCharacterisationBaseAlgorithm::m_minTrackLikeViews

protectedinherited

The minimum number of track-like views to declare a pfo as track-like.

Definition at line 71 of file PfoCharacterisationBaseAlgorithm.h.

Referenced by lar_content::PfoCharacterisationBaseAlgorithm::IsClearTrack3x2D(), and lar_content::PfoCharacterisationBaseAlgorithm::ReadSettings().

bool lar_content::PfoCharacterisationBaseAlgorithm::m_postBranchAddition

protectedinherited

Whether to use configuration for shower clusters post branch addition.

Definition at line 69 of file PfoCharacterisationBaseAlgorithm.h.

std::string lar_content::PfoCharacterisationBaseAlgorithm::m_showerPfoListName

protectedinherited

The shower pfo list name.

Definition at line 65 of file PfoCharacterisationBaseAlgorithm.h.

Referenced by lar_content::PfoCharacterisationBaseAlgorithm::ReadSettings(), and lar_content::PfoCharacterisationBaseAlgorithm::Run().

std::string lar_content::PfoCharacterisationBaseAlgorithm::m_trackPfoListName

protectedinherited

The track pfo list name.

Definition at line 64 of file PfoCharacterisationBaseAlgorithm.h.

Referenced by lar_content::PfoCharacterisationBaseAlgorithm::ReadSettings(), and lar_content::PfoCharacterisationBaseAlgorithm::Run().

bool lar_content::PfoCharacterisationBaseAlgorithm::m_updateClusterIds

protectedinherited

Whether to update daughter cluster particle id labels to match pfo id.

Definition at line 68 of file PfoCharacterisationBaseAlgorithm.h.

Referenced by lar_content::PfoCharacterisationBaseAlgorithm::ReadSettings(), and lar_content::PfoCharacterisationBaseAlgorithm::Run().

bool lar_content::PfoCharacterisationBaseAlgorithm::m_useThreeDInformation

protectedinherited

Whether to use PFO and 3D information or clusters for characterisation.

Definition at line 70 of file PfoCharacterisationBaseAlgorithm.h.

Referenced by lar_content::PfoCharacterisationBaseAlgorithm::ReadSettings(), and lar_content::PfoCharacterisationBaseAlgorithm::Run().

The documentation for this class was generated from the following files:

larpandoracontent/v04_08_01/source/larpandoracontent/LArCheating/CheatingPfoCharacterisationAlgorithm.h
larpandoracontent/v04_08_01/source/larpandoracontent/LArCheating/CheatingPfoCharacterisationAlgorithm.cc

Protected Member Functions

Protected Attributes

Private Member Functions

Detailed Description

Member Function Documentation

Member Data Documentation