ATLAS Offline Software
xAODElectronRetriever.cxx
Go to the documentation of this file.
1 /*
2  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
3 */
4 
7 #include "AthenaKernel/Units.h"
8 
10 
11 namespace JiveXML {
12 
19  xAODElectronRetriever::xAODElectronRetriever(const std::string& type,const std::string& name,const IInterface* parent):
21  {}
22 
23 
25  ATH_CHECK(m_keys.initialize());
26  return StatusCode::SUCCESS;
27  }
28 
33  StatusCode xAODElectronRetriever::retrieve(ToolHandle<IFormatTool> &FormatTool) {
34 
35  ATH_MSG_DEBUG("in retrieve()");
36 
37  // Loop through the keys and retrieve the corresponding data
38  for (const auto& key : m_keys) {
40  if (cont.isValid()) {
41  DataMap data = getData(&(*cont));
42  if (FormatTool->AddToEvent(dataTypeName(), key.key() + "_xAOD", &data).isFailure()) {
43  ATH_MSG_WARNING("Failed to add collection " << key.key());
44  } else {
45  ATH_MSG_DEBUG(" (" << key.key() << ") retrieved");
46  }
47  } else {
48  ATH_MSG_WARNING("Collection " << key.key() << " not found in SG");
49  }
50  }
51  return StatusCode::SUCCESS;
52  }
53 
54 
60 
61  ATH_MSG_DEBUG("in getData()");
62 
64 
65  DataVect pt; pt.reserve(elCont->size());
66  DataVect phi; phi.reserve(elCont->size());
67  DataVect eta; eta.reserve(elCont->size());
68  DataVect mass; mass.reserve(elCont->size());
69  DataVect energy; energy.reserve(elCont->size());
70  DataVect pdgId; energy.reserve(elCont->size());
71 
72  DataVect isEMString; isEMString.reserve(elCont->size());
73  DataVect author; author.reserve(elCont->size());
74  DataVect label; label.reserve(elCont->size());
75 
78 
79  int counter = 0;
80 
81  for (; elItr != elItrE; ++elItr) {
82 
83  std::string electronAuthor = "";
84  std::string electronIsEMString = "none";
85  std::string electronLabel = "";
86  phi.emplace_back(DataType((*elItr)->phi()));
87  eta.emplace_back(DataType((*elItr)->eta()));
88  pt.emplace_back(DataType((*elItr)->pt()/GeV));
89  mass.emplace_back(DataType((*elItr)->m()/GeV));
90  energy.emplace_back( DataType((*elItr)->e()/GeV ));
91 
92  if ((*elItr)->trackParticle()){ // ForwardElectrons have no track !
93  pdgId.emplace_back(DataType( -11.*(*elItr)->trackParticle()->charge() )); // pdgId not available anymore in xAOD
94  }else{
95  pdgId.emplace_back(DataType( 0. ) );
96  }
97 
98  ATH_MSG_DEBUG(" Electron #" << counter++ << " : eta = " << (*elItr)->eta() << ", phi = "
99  << (*elItr)->phi()
100  // << ", ntrk = " << (*elItr)->getNumberOfTrackParticles()
101  << ", author = " << (*elItr)->author()
102  // these don't work for ForwardElectrons !
103  // << ", isEM/Tight: " << (*elItr)->passSelection(passesTight, "Tight")
104  // << ", charge = " << (*elItr)->trackParticle()->charge()
105  // << ", pdgId = " << -11.*(*elItr)->trackParticle()->charge()
106  );
107 
108 
109  bool passesTight(false);
110  bool passesMedium(false);
111  bool passesLoose(false);
112  const bool tightSelectionExists = (*elItr)->passSelection(passesTight, "Tight");
113  ATH_MSG_VERBOSE("tight exists " << tightSelectionExists
114  << " and passes? " << passesTight);
115  const bool mediumSelectionExists = (*elItr)->passSelection(passesMedium, "Medium");
116  ATH_MSG_VERBOSE("medium exists " << mediumSelectionExists
117  << " and passes? " << passesMedium);
118  const bool looseSelectionExists = (*elItr)->passSelection(passesLoose, "Loose");
119  ATH_MSG_VERBOSE("loose exists " << looseSelectionExists
120  << " and passes? " << passesLoose);
121 
122  electronAuthor = "author"+DataType( (*elItr)->author() ).toString(); // for odd ones eg FWD
123  electronLabel = electronAuthor;
124  if (( (*elItr)->author()) == 0){ electronAuthor = "unknown"; electronLabel += "_unknown"; }
125  if (( (*elItr)->author()) == 8){ electronAuthor = "forward"; electronLabel += "_forward"; }
126  if (( (*elItr)->author()) == 2){ electronAuthor = "softe"; electronLabel += "_softe"; }
127  if (( (*elItr)->author()) == 1){ electronAuthor = "egamma"; electronLabel += "_egamma"; }
128 
129  if ( passesLoose ){
130  electronLabel += "_Loose";
131  electronIsEMString = "Loose"; // assume that hierarchy is obeyed !
132  }
133  if ( passesMedium ){
134  electronLabel += "_Medium";
135  electronIsEMString = "Medium"; // assume that hierarchy is obeyed !
136  }
137  if ( passesTight ){
138  electronLabel += "_Tight";
139  electronIsEMString = "Tight"; // assume that hierarchy is obeyed !
140  }
141  author.emplace_back( DataType( electronAuthor ) );
142  label.emplace_back( DataType( electronLabel ) );
143  isEMString.emplace_back( DataType( electronIsEMString ) );
144 
145  } // end ElectronIterator
146 
147  // four-vectors
148  DataMap["phi"] = phi;
149  DataMap["eta"] = eta;
150  DataMap["pt"] = pt;
151  DataMap["energy"] = energy;
152  DataMap["mass"] = mass;
153  DataMap["pdgId"] = pdgId;
154  DataMap["isEMString"] = isEMString;
155  DataMap["label"] = label;
156  DataMap["author"] = author;
157 
158  ATH_MSG_DEBUG(dataTypeName() << " retrieved with " << phi.size() << " entries");
159 
160  return DataMap;
161  }
162 
163 
164 } // JiveXML namespace
data
char data[hepevt_bytes_allocation_ATLAS]
Definition: HepEvt.cxx:11
GeV
#define GeV
Definition: PhysicsAnalysis/TauID/TauAnalysisTools/Root/HelperFunctions.cxx:18
DataModel_detail::const_iterator
Const iterator class for DataVector/DataList.
Definition: DVLIterator.h:82
Base_Fragment.mass
mass
Definition: Sherpa_i/share/common/Base_Fragment.py:59
xAODElectronRetriever.h
SG::ReadHandle
Definition: StoreGate/StoreGate/ReadHandle.h:67
JiveXML::xAODElectronRetriever::getData
const DataMap getData(const xAOD::ElectronContainer *)
Puts the variables into a DataMap.
Definition: xAODElectronRetriever.cxx:59
JiveXML::DataVect
std::vector< DataType > DataVect
Defines a map with a key and a vector of DataType objects e.g.
Definition: DataType.h:58
DataType
OFFLINE_FRAGMENTS_NAMESPACE::PointerType DataType
Definition: RoIBResultByteStreamTool.cxx:25
test_pyathena.pt
pt
Definition: test_pyathena.py:11
JiveXML::xAODElectronRetriever::dataTypeName
virtual std::string dataTypeName() const
Return the name of the data type that is generated by this retriever.
Definition: xAODElectronRetriever.h:42
ATH_MSG_VERBOSE
#define ATH_MSG_VERBOSE(x)
Definition: AthMsgStreamMacros.h:28
JiveXML::DataMap
std::map< std::string, DataVect > DataMap
Definition: DataType.h:59
python.CaloAddPedShiftConfig.type
type
Definition: CaloAddPedShiftConfig.py:42
ParticleGun_FastCalo_ChargeFlip_Config.energy
energy
Definition: ParticleGun_FastCalo_ChargeFlip_Config.py:78
ElectronContainer.h
JiveXML::xAODElectronRetriever::xAODElectronRetriever
xAODElectronRetriever(const std::string &type, const std::string &name, const IInterface *parent)
Standard Constructor.
Definition: xAODElectronRetriever.cxx:19
EL::StatusCode
::StatusCode StatusCode
StatusCode definition for legacy code.
Definition: PhysicsAnalysis/D3PDTools/EventLoop/EventLoop/StatusCode.h:22
ATH_MSG_DEBUG
#define ATH_MSG_DEBUG(x)
Definition: AthMsgStreamMacros.h:29
JiveXML::xAODElectronRetriever::initialize
virtual StatusCode initialize()
Definition: xAODElectronRetriever.cxx:24
add-xsec-uncert-quadrature-N.label
label
Definition: add-xsec-uncert-quadrature-N.py:104
test_pyathena.parent
parent
Definition: test_pyathena.py:15
ATH_CHECK
#define ATH_CHECK
Definition: AthCheckMacros.h:40
DataVector
Derived DataVector<T>.
Definition: DataVector.h:794
SG::ReadHandle::isValid
virtual bool isValid() override final
Can the handle be successfully dereferenced?
JiveXML
This header is shared inbetween the C-style server thread and the C++ Athena ServerSvc.
Definition: BadLArRetriever.cxx:22
name
std::string name
Definition: Control/AthContainers/Root/debug.cxx:240
runIDPVM.pdgId
pdgId
Definition: runIDPVM.py:91
Units.h
Wrapper to avoid constant divisions when using units.
DataVector::end
const_iterator end() const noexcept
Return a const_iterator pointing past the end of the collection.
ATH_MSG_WARNING
#define ATH_MSG_WARNING(x)
Definition: AthMsgStreamMacros.h:32
CaloCondBlobAlgs_fillNoiseFromASCII.author
string author
Definition: CaloCondBlobAlgs_fillNoiseFromASCII.py:25
JiveXML::xAODElectronRetriever::retrieve
virtual StatusCode retrieve(ToolHandle< IFormatTool > &FormatTool)
For each electron collection retrieve basic parameters.
Definition: xAODElectronRetriever.cxx:33
AthAlgTool
Definition: AthAlgTool.h:26
test_pyathena.counter
counter
Definition: test_pyathena.py:15
DataVector::size
size_type size() const noexcept
Returns the number of elements in the collection.
JiveXML::xAODElectronRetriever::m_keys
SG::ReadHandleKeyArray< xAOD::ElectronContainer > m_keys
Definition: xAODElectronRetriever.h:48
DataVector::begin
const_iterator begin() const noexcept
Return a const_iterator pointing at the beginning of the collection.
mapkey::key
key
Definition: TElectronEfficiencyCorrectionTool.cxx:37