ATLAS Offline Software
PhotonRetriever.cxx
Go to the documentation of this file.
1 /*
2  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
3 */
4 
5 #include "AnalysisJiveXML/PhotonRetriever.h"
6 
7 #include "egammaEvent/PhotonContainer.h"
8 #include "egammaEvent/EMShower.h"
9 
10 #include "CLHEP/Units/SystemOfUnits.h"
11 
12 namespace JiveXML {
13 
20  PhotonRetriever::PhotonRetriever(const std::string& type,const std::string& name,const IInterface* parent):
21  AthAlgTool(type,name,parent),
22  m_typeName("Photon"){
23 
24  //Only declare the interface
25  declareInterface<IDataRetriever>(this);
26 
27  declareProperty("StoreGateKey", m_sgKey = "PhotonAODCollection",
28  "Collection to be first in output, shown in Atlantis without switching");
29  }
30 
35  StatusCode PhotonRetriever::retrieve(ToolHandle<IFormatTool> &FormatTool) {
36 
37  if (msgLvl(MSG::DEBUG)) msg(MSG::DEBUG) << "in retrieveAll()" << endmsg;
38 
39  SG::ConstIterator<PhotonContainer> iterator, end;
40  const PhotonContainer* photons;
41 
42  //obtain the default collection first
43  ATH_MSG_DEBUG( "Trying to retrieve " << dataTypeName() << " (" << m_sgKey << ")" );
44  StatusCode sc = evtStore()->retrieve(photons, m_sgKey);
45  if (sc.isFailure() ) {
46  ATH_MSG_WARNING( "Collection " << m_sgKey << " not found in SG " );
47  }else{
48  DataMap data = getData(photons);
49  if ( FormatTool->AddToEvent(dataTypeName(), m_sgKey, &data).isFailure()){
50  ATH_MSG_WARNING( "Collection " << m_sgKey << " not found in SG " );
51  }else{
52  ATH_MSG_DEBUG( dataTypeName() << " (" << m_sgKey << ") Photon retrieved" );
53  }
54  }
55 
56  //obtain all other collections from StoreGate
57  if (( evtStore()->retrieve(iterator, end)).isFailure()){
58  if (msgLvl(MSG::WARNING)) msg(MSG::WARNING) << "Unable to retrieve iterator for Jet collection" << endmsg;
59 // return StatusCode::WARNING;
60  }
61 
62  for (; iterator!=end; ++iterator) {
63  if (iterator.key()!=m_sgKey) {
64  ATH_MSG_DEBUG( "Trying to retrieve all " << dataTypeName() << " (" << iterator.key() << ")" );
65  DataMap data = getData(&(*iterator));
66  if ( FormatTool->AddToEvent(dataTypeName(), iterator.key(), &data).isFailure()){
67  ATH_MSG_WARNING( "Collection " << iterator.key() << " not found in SG " );
68  }else{
69  ATH_MSG_DEBUG( dataTypeName() << " (" << iterator.key() << ") Photon retrieved" );
70  }
71  }
72  }
73  //All collections retrieved okay
74  return StatusCode::SUCCESS;
75  }
76 
77 
82  const DataMap PhotonRetriever::getData(const PhotonContainer* photcont) {
83 
84  ATH_MSG_DEBUG( "retrieve()" );
85 
86  DataMap DataMap;
87 
88  DataVect phi; phi.reserve(photcont->size());
89  DataVect eta; eta.reserve(photcont->size());
90  DataVect pt; pt.reserve(photcont->size());
91  DataVect mass; mass.reserve(photcont->size());
92  DataVect energy; energy.reserve(photcont->size());
93  DataVect px; px.reserve(photcont->size());
94  DataVect py; py.reserve(photcont->size());
95  DataVect pz; pz.reserve(photcont->size());
96 
97  DataVect isEM; isEM.reserve(photcont->size());
98  DataVect f1Vec; f1Vec.reserve(photcont->size());
99  DataVect etConeVec; etConeVec.reserve(photcont->size());
100  DataVect fracs1Vec; fracs1Vec.reserve(photcont->size());
101  DataVect et37Vec; et37Vec.reserve(photcont->size());
102 
103  DataVect author; author.reserve(photcont->size());
104  DataVect label; label.reserve(photcont->size());
105  DataVect isEMString; isEMString.reserve(photcont->size());
106 
107  // for associations:
108  DataVect clusterKeyVec; clusterKeyVec.reserve(photcont->size());
109  DataVect clusterIndexVec; clusterIndexVec.reserve(photcont->size());
110 
111  PhotonContainer::const_iterator photonItr = photcont->begin();
112  PhotonContainer::const_iterator photonItrE = photcont->end();
113 
114  int MCdataType = 1;
115  std::string clusterKey = "none"; // Storegate key of container
116  int clusterIndex = -1; // index number inside the container
117 
118  std::string photonAuthor = "";
119  std::string photonIsEMString = "none";
120  std::string photonLabel = "";
121 
122  // reference for authors: Reconstruction/egamma/egammaEvent/egammaParamDefs.h
123  // https://twiki.cern.ch/twiki/bin/view/AtlasProtected/PhotonIdentification
124  for (; photonItr != photonItrE; ++photonItr) {
125  photonIsEMString = "none";
126  photonAuthor = "author"+DataType( (*photonItr)->author() ).toString(); // for odd ones eg FWD
127  photonLabel = photonAuthor;
128  if (( (*photonItr)->author()) == 0x0){ photonAuthor = "unknown"; photonLabel += "_unknown"; }
129  if (( (*photonItr)->author()) == 0x8){ photonAuthor = "forward"; photonLabel += "_forward"; }
130  if (( (*photonItr)->author()) == 0x10){ photonAuthor = "rconv"; photonLabel += "_recoveredconversion"; }
131  if (( (*photonItr)->author()) == 0x4){ photonAuthor = "photon"; photonLabel += "_photon"; }
132  if ( (*photonItr)->isem(egammaPIDObs::PhotonLoose)==0){
133  photonLabel += "_Loose";
134  photonIsEMString = "Loose"; // assume that hierarchy is obeyed !
135  }
136  if ( (*photonItr)->isem(egammaPIDObs::PhotonTight)==0){
137  photonLabel += "_Tight";
138  photonIsEMString = "Tight"; // assume that hierarchy is obeyed !
139  }
140  if ( (*photonItr)->isem(egammaPIDObs::PhotonLooseAR)==0){
141  photonLabel += "_LooseAR";
142  }
143  if ( (*photonItr)->isem(egammaPIDObs::PhotonTightAR)==0){
144  photonLabel += "_TightAR";
145  }
146  if ( (*photonItr)->isem(egammaPIDObs::PhotonTightARIso)==0){
147  photonLabel += "_TightARIso";
148  }
149  if ( (*photonItr)->isem(egammaPIDObs::PhotonTightIso)==0){
150  photonLabel += "_TightIso";
151  }
152 
153  phi.emplace_back((*photonItr)->phi());
154  eta.emplace_back((*photonItr)->eta());
155  pt.emplace_back((*photonItr)->pt()/CLHEP::GeV);
156  mass.emplace_back((*photonItr)->m()/CLHEP::GeV);
157  energy.emplace_back((*photonItr)->e()/CLHEP::GeV );
158  px.emplace_back((*photonItr)->px()/CLHEP::GeV );
159  py.emplace_back((*photonItr)->py()/CLHEP::GeV );
160  pz.emplace_back((*photonItr)->pz()/CLHEP::GeV );
161 
162  MCdataType = (*photonItr)->dataType();
163  if (MCdataType != 3){ // full simulation
164  isEM.emplace_back((**photonItr).isem() );
165 // do associations:
166  const ElementLink<CaloClusterContainer> clusterLink = (*photonItr)->clusterElementLink();
167  if (clusterLink.isValid()) {
168  clusterKey = clusterLink.dataID(); // Storegate key of container
169  clusterIndex = clusterLink.index(); // index number inside the container
170  clusterKeyVec.emplace_back( clusterKey );
171  clusterIndexVec.emplace_back( clusterIndex );
172  } else { // no clusterLink
173  clusterKeyVec.emplace_back( "none" );
174  clusterIndexVec.emplace_back( -1 );
175  }
177  // emWeight|et37|etCone|etHad1|f1|fracs1|pionWeight
178  const EMShower* emShower = (*photonItr)->detail<EMShower>("egDetailAOD");
179  if (emShower) {
180  f1Vec.emplace_back( emShower->parameter(egammaParameters::f1) );
181  etConeVec.emplace_back( emShower->parameter(egammaParameters::etcone20));
182  fracs1Vec.emplace_back( emShower->parameter(egammaParameters::fracs1));
183  et37Vec.emplace_back( emShower->parameter(egammaParameters::e237));
184  }else{ //placeholders if no shower available
185  f1Vec.emplace_back( -1.);
186  etConeVec.emplace_back( -1. );
187  fracs1Vec.emplace_back( -1. );
188  et37Vec.emplace_back( -1.);
189  }
190  } else { // fast simulation: just placeholders
191  isEM.emplace_back( 0 );
192  clusterKeyVec.emplace_back( "none" );
193  clusterIndexVec.emplace_back( -1 );
194  f1Vec.push_back( -1.);
195  etConeVec.emplace_back( -1. );
196  fracs1Vec.emplace_back( -1. );
197  et37Vec.emplace_back( -1. );
198  photonLabel += "_fastSim";
199  photonIsEMString = "fastSim";
200  }
201  author.emplace_back(photonAuthor );
202  label.emplace_back( photonLabel );
203  isEMString.emplace_back( photonIsEMString );
204  }
205  // four-vectors
206  const auto nEntries = phi.size();
207  DataMap["phi"] = std::move(phi);
208  DataMap["eta"] = std::move(eta);
209  DataMap["pt"] = std::move(pt);
210  DataMap["energy"] = std::move(energy);
211  DataMap["mass"] = std::move(mass);
212  DataMap["px"] = std::move(px);
213  DataMap["py"] = std::move(py);
214  DataMap["pz"] = std::move(pz);
215 
216  // further details and associations
217  DataMap["isEM"] = std::move(isEM);
218  DataMap["clusterKey"] = std::move(clusterKeyVec);
219  DataMap["clusterIndex"] = std::move(clusterIndexVec);
220  // shower details
221  DataMap["f1"] = std::move(f1Vec);
222  DataMap["etCone"] = std::move(etConeVec);
223  DataMap["fracs1"] = std::move(fracs1Vec);
224  DataMap["et37"] = std::move(et37Vec);
225 
226  DataMap["author"] = std::move(author);
227  DataMap["isEMString"] = std::move(isEMString);
228  DataMap["label"] = std::move(label);
229 
230  ATH_MSG_DEBUG( dataTypeName() << " retrieved with " << nEntries << " entries");
231 
232 
233  //All collections retrieved okay
234  return DataMap;
235 
236  } // retrieve
237 
238  //--------------------------------------------------------------------------
239 
240 } // JiveXML namespace
xAOD::iterator
JetConstituentVector::iterator iterator
Definition: JetConstituentVector.cxx:68
egammaPIDObs::PhotonLooseAR
const unsigned int PhotonLooseAR
Loose photon selection with Ambiguity resolver.
Definition: egammaPIDdefsObs.h:577
data
char data[hepevt_bytes_allocation_ATLAS]
Definition: HepEvt.cxx:11
GeV
#define GeV
Definition: PhysicsAnalysis/TauID/TauAnalysisTools/Root/HelperFunctions.cxx:17
DataModel_detail::const_iterator
Const iterator class for DataVector/DataList.
Definition: DVLIterator.h:82
test_pyathena.px
px
Definition: test_pyathena.py:18
JiveXML::PhotonRetriever::PhotonRetriever
PhotonRetriever(const std::string &type, const std::string &name, const IInterface *parent)
Standard Constructor.
Definition: PhotonRetriever.cxx:20
JiveXML::PhotonRetriever::getData
const DataMap getData(const PhotonContainer *)
Retrieve basic parameters, mainly four-vectors, for each collection.
Definition: PhotonRetriever.cxx:82
Base_Fragment.mass
mass
Definition: Sherpa_i/share/common/Base_Fragment.py:59
AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T > &t)
Definition: AthCommonDataStore.h:145
PlotCalibFromCool.label
label
Definition: PlotCalibFromCool.py:78
egammaPIDObs::PhotonLoose
const unsigned int PhotonLoose
Loose photon selection.
Definition: egammaPIDdefsObs.h:591
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
AthCommonMsg< AlgTool >::msgLvl
bool msgLvl(const MSG::Level lvl) const
Definition: AthCommonMsg.h:30
egammaPIDObs::PhotonTightIso
const unsigned int PhotonTightIso
Tight photon selection with isolation.
Definition: egammaPIDdefsObs.h:602
JiveXML::DataMap
std::map< std::string, DataVect > DataMap
Definition: DataType.h:59
JiveXML::PhotonRetriever::m_sgKey
std::string m_sgKey
Definition: PhotonRetriever.h:50
EMShower::parameter
virtual double parameter(egammaParameters::ParamDef) const
Definition: EMShower.cxx:71
PhotonContainer
Definition: Reconstruction/egamma/egammaEvent/egammaEvent/PhotonContainer.h:30
AthenaPoolTestRead.sc
sc
Definition: AthenaPoolTestRead.py:27
mergePhysValFiles.end
end
Definition: DataQuality/DataQualityUtils/scripts/mergePhysValFiles.py:93
EMShower
Definition: EMShower.h:24
egammaParameters::fracs1
@ fracs1
shower shape in the shower core : [E(+/-3)-E(+/-1)]/E(+/-1), where E(+/-n) is the energy in +- n stri...
Definition: egammaParamDefs.h:161
AthCommonDataStore< AthCommonMsg< AlgTool > >::evtStore
ServiceHandle< StoreGateSvc > & evtStore()
The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.
Definition: AthCommonDataStore.h:85
ParticleGun_FastCalo_ChargeFlip_Config.energy
energy
Definition: ParticleGun_FastCalo_ChargeFlip_Config.py:78
egammaParameters::etcone20
@ etcone20
ET in a cone with half-opening angle 0.2, with exclusion of a window of size 7x5 in electromagnetic c...
Definition: egammaParamDefs.h:104
endmsg
#define endmsg
Definition: AnalysisConfig_Ntuple.cxx:63
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::PhotonRetriever::dataTypeName
virtual std::string dataTypeName() const
Return the name of the data type.
Definition: PhotonRetriever.h:44
Amg::pz
@ pz
Definition: GeoPrimitives.h:40
test_pyathena.parent
parent
Definition: test_pyathena.py:15
egammaParameters::e237
@ e237
uncalibrated energy (sum of cells) of the middle sampling in a rectangle of size 3x7
Definition: egammaParamDefs.h:139
egammaPIDObs::PhotonTightAR
const unsigned int PhotonTightAR
Tight photon selection with Ambiguity resolver.
Definition: egammaPIDdefsObs.h:586
PhotonRetriever.h
JiveXML
This header is shared inbetween the C-style server thread and the C++ Athena ServerSvc.
Definition: BadLArRetriever.cxx:22
Amg::py
@ py
Definition: GeoPrimitives.h:39
name
std::string name
Definition: Control/AthContainers/Root/debug.cxx:228
EMShower.h
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
python.CaloScaleNoiseConfig.type
type
Definition: CaloScaleNoiseConfig.py:78
DEBUG
#define DEBUG
Definition: page_access.h:11
CaloCondBlobAlgs_fillNoiseFromASCII.author
string author
Definition: CaloCondBlobAlgs_fillNoiseFromASCII.py:26
AthCommonMsg< AlgTool >::msg
MsgStream & msg() const
Definition: AthCommonMsg.h:24
egammaPIDObs::PhotonTight
const unsigned int PhotonTight
Tight photon selection.
Definition: egammaPIDdefsObs.h:600
AthAlgTool
Definition: AthAlgTool.h:26
egammaPIDObs::PhotonTightARIso
const unsigned int PhotonTightARIso
Tight photon selection with isolation and Ambiguity resolver.
Definition: egammaPIDdefsObs.h:588
dqBeamSpot.nEntries
int nEntries
Definition: dqBeamSpot.py:73
DataVector::size
size_type size() const noexcept
Returns the number of elements in the collection.
SG::ConstIterator
Definition: SGIterator.h:163
JiveXML::PhotonRetriever::retrieve
virtual StatusCode retrieve(ToolHandle< IFormatTool > &FormatTool)
Retrieve all the data.
Definition: PhotonRetriever.cxx:35
egammaParameters::f1
@ f1
E1/E = fraction of energy reconstructed in the first sampling, where E1 is energy in all strips belon...
Definition: egammaParamDefs.h:125
PhotonContainer.h
DataVector::begin
const_iterator begin() const noexcept
Return a const_iterator pointing at the beginning of the collection.
Generated on Sun Dec 22 2024 21:16:11 for ATLAS Offline Software by   doxygen 1.8.18