Protected Member Functions | Private Attributes | List of all members
lar_content::CustomParticleCreationAlgorithm Class Referenceabstract

CustomParticleCreationAlgorithm class. More...

#include <CustomParticleCreationAlgorithm.h>

Inheritance diagram for lar_content::CustomParticleCreationAlgorithm:
lar_content::PcaShowerParticleBuildingAlgorithm lar_content::TrackParticleBuildingAlgorithm

Protected Member Functions

virtual pandora::StatusCode Run ()
 
virtual pandora::StatusCode ReadSettings (const pandora::TiXmlHandle xmlHandle)
 
virtual void CreatePfo (const pandora::ParticleFlowObject *const pInputPfo, const pandora::ParticleFlowObject *&pOutputPfo) const =0
 Create specialised Pfo from an generic input Pfo. More...
 

Private Attributes

std::string m_pfoListName
 The name of the input pfo list. More...
 
std::string m_vertexListName
 The name of the input vertex list. More...
 

Detailed Description

CustomParticleCreationAlgorithm class.

Definition at line 19 of file CustomParticleCreationAlgorithm.h.

Member Function Documentation

virtual void lar_content::CustomParticleCreationAlgorithm::CreatePfo ( const pandora::ParticleFlowObject *const  pInputPfo,
const pandora::ParticleFlowObject *&  pOutputPfo 
) const
protectedpure virtual

Create specialised Pfo from an generic input Pfo.

Parameters
pInputPfothe address of the input Pfo
pOutputPfothe address of the output Pfo

Implemented in lar_content::PcaShowerParticleBuildingAlgorithm, and lar_content::TrackParticleBuildingAlgorithm.

StatusCode lar_content::CustomParticleCreationAlgorithm::ReadSettings ( const pandora::TiXmlHandle  xmlHandle)
protectedvirtual

Reimplemented in lar_content::PcaShowerParticleBuildingAlgorithm, and lar_content::TrackParticleBuildingAlgorithm.

Definition at line 121 of file CustomParticleCreationAlgorithm.cc.

122 {
123  PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, XmlHelper::ReadValue(xmlHandle, "PfoListName", m_pfoListName));
124  PANDORA_RETURN_RESULT_IF(STATUS_CODE_SUCCESS, !=, XmlHelper::ReadValue(xmlHandle, "VertexListName", m_vertexListName));
125 
126  return STATUS_CODE_SUCCESS;
127 }
std::string m_vertexListName
The name of the input vertex list.
std::string m_pfoListName
The name of the input pfo list.
StatusCode lar_content::CustomParticleCreationAlgorithm::Run ( )
protectedvirtual

Definition at line 20 of file CustomParticleCreationAlgorithm.cc.

21 {
22  // Get input Pfo List
23  const PfoList *pPfoList(NULL);
24 
25  if (STATUS_CODE_SUCCESS != PandoraContentApi::GetList(*this, m_pfoListName, pPfoList))
26  {
27  if (PandoraContentApi::GetSettings(*this)->ShouldDisplayAlgorithmInfo())
28  std::cout << "CustomParticleCreationAlgorithm: cannot find pfo list " << m_pfoListName << std::endl;
29 
30  return STATUS_CODE_SUCCESS;
31  }
32 
33  // Get input Vertex List
34  const VertexList *pVertexList(NULL);
35 
36  if (STATUS_CODE_SUCCESS != PandoraContentApi::GetList(*this, m_vertexListName, pVertexList))
37  {
38  if (PandoraContentApi::GetSettings(*this)->ShouldDisplayAlgorithmInfo())
39  std::cout << "CustomParticleCreationAlgorithm: cannot find vertex list " << m_vertexListName << std::endl;
40 
41  return STATUS_CODE_SUCCESS;
42  }
43 
44  // Create temporary lists
45  const PfoList *pTempPfoList = NULL;
46  std::string tempPfoListName;
47  PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::CreateTemporaryListAndSetCurrent(*this, pTempPfoList, tempPfoListName));
48 
49  const VertexList *pTempVertexList = NULL;
50  std::string tempVertexListName;
51  PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::CreateTemporaryListAndSetCurrent(*this, pTempVertexList, tempVertexListName));
52 
53  // Loop over input Pfos
54  PfoList pfoList(pPfoList->begin(), pPfoList->end());
55  VertexList vertexList(pVertexList->begin(), pVertexList->end());
56 
57  for (PfoList::const_iterator iter = pfoList.begin(), iterEnd = pfoList.end(); iter != iterEnd; ++iter)
58  {
59  const ParticleFlowObject *const pInputPfo = *iter;
60 
61  if (pInputPfo->GetVertexList().empty())
62  continue;
63 
64  const Vertex *const pInputVertex = LArPfoHelper::GetVertex(pInputPfo);
65 
66  if (vertexList.end() == std::find(vertexList.begin(), vertexList.end(), pInputVertex))
67  throw StatusCodeException(STATUS_CODE_FAILURE);
68 
69  // Build a new pfo and vertex from the old pfo
70  const ParticleFlowObject *pOutputPfo(NULL);
71 
72  this->CreatePfo(pInputPfo, pOutputPfo);
73 
74  if (NULL == pOutputPfo)
75  continue;
76 
77  if (pOutputPfo->GetVertexList().empty())
78  throw StatusCodeException(STATUS_CODE_FAILURE);
79 
80  // Transfer clusters and hierarchy information to new pfo, and delete old pfo and vertex
81  ClusterList clusterList(pInputPfo->GetClusterList().begin(), pInputPfo->GetClusterList().end());
82  PfoList parentList(pInputPfo->GetParentPfoList().begin(), pInputPfo->GetParentPfoList().end());
83  PfoList daughterList(pInputPfo->GetDaughterPfoList().begin(), pInputPfo->GetDaughterPfoList().end());
84 
85  PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::Delete<Pfo>(*this, pInputPfo, m_pfoListName));
86  PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::Delete<Vertex>(*this, pInputVertex, m_vertexListName));
87 
88  for (ClusterList::const_iterator cIter = clusterList.begin(), cIterEnd = clusterList.end(); cIter != cIterEnd; ++cIter)
89  {
90  PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::AddToPfo<Cluster>(*this, pOutputPfo, *cIter));
91  }
92 
93  for (PfoList::const_iterator pIter = parentList.begin(), pIterEnd = parentList.end(); pIter != pIterEnd; ++pIter)
94  {
95  PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::SetPfoParentDaughterRelationship(*this, *pIter, pOutputPfo));
96  }
97 
98  for (PfoList::const_iterator dIter = daughterList.begin(), dIterEnd = daughterList.end(); dIter != dIterEnd; ++dIter)
99  {
100  PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::SetPfoParentDaughterRelationship(*this, pOutputPfo, *dIter));
101  }
102  }
103 
104  if (!pTempPfoList->empty())
105  {
106  PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::SaveList<Pfo>(*this, m_pfoListName));
107  PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::ReplaceCurrentList<Pfo>(*this, m_pfoListName));
108  }
109 
110  if (!pTempVertexList->empty())
111  {
112  PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::SaveList<Vertex>(*this, m_vertexListName));
113  PANDORA_THROW_RESULT_IF(STATUS_CODE_SUCCESS, !=, PandoraContentApi::ReplaceCurrentList<Vertex>(*this, m_vertexListName));
114  }
115 
116  return STATUS_CODE_SUCCESS;
117 }
std::string m_vertexListName
The name of the input vertex list.
std::string string
Definition: nybbler.cc:12
static const pandora::Vertex * GetVertex(const pandora::ParticleFlowObject *const pPfo)
Get the pfo vertex.
intermediate_table::const_iterator const_iterator
std::string m_pfoListName
The name of the input pfo list.
boost::graph_traits< ModuleGraph >::vertex_descriptor Vertex
Definition: ModuleGraph.h:25
std::list< Vertex > VertexList
Definition: DCEL.h:182
virtual void CreatePfo(const pandora::ParticleFlowObject *const pInputPfo, const pandora::ParticleFlowObject *&pOutputPfo) const =0
Create specialised Pfo from an generic input Pfo.
QTextStream & endl(QTextStream &s)

Member Data Documentation

std::string lar_content::CustomParticleCreationAlgorithm::m_pfoListName
private

The name of the input pfo list.

Definition at line 34 of file CustomParticleCreationAlgorithm.h.

std::string lar_content::CustomParticleCreationAlgorithm::m_vertexListName
private

The name of the input vertex list.

Definition at line 35 of file CustomParticleCreationAlgorithm.h.


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