Loading [MathJax]/extensions/tex2jax.js
ATLAS Offline Software
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
Public Member Functions | Private Member Functions | Private Attributes | List of all members
MMLoadVariables Class Reference

#include <MMLoadVariables.h>

Inheritance diagram for MMLoadVariables:
Collaboration diagram for MMLoadVariables:

Public Member Functions

 MMLoadVariables ()
 
StatusCode getTruthInfo (const EventContext &ctx, const McEventCollection *truthContainer, const TrackRecordCollection *trackRecordCollection, std::map< std::pair< uint64_t, unsigned int >, evInf_entry > &Event_Info) const
 
bool msgLvl (const MSG::Level lvl) const
 Test the output level. More...
 
MsgStream & msg () const
 The standard message stream. More...
 
MsgStream & msg (const MSG::Level lvl) const
 The standard message stream. More...
 
void setLevel (MSG::Level lvl)
 Change the current logging level. More...
 

Private Member Functions

void initMessaging () const
 Initialize our message level and MessageSvc. More...
 

Private Attributes

std::string m_nm
 Message source name. More...
 
boost::thread_specific_ptr< MsgStream > m_msg_tls
 MsgStream instance (a std::cout like with print-out levels) More...
 
std::atomic< IMessageSvc * > m_imsg { nullptr }
 MessageSvc pointer. More...
 
std::atomic< MSG::Level > m_lvl { MSG::NIL }
 Current logging level. More...
 
std::atomic_flag m_initialized ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT
 Messaging initialized (initMessaging) More...
 

Detailed Description

Definition at line 33 of file MMLoadVariables.h.

Constructor & Destructor Documentation

◆ MMLoadVariables()

MMLoadVariables::MMLoadVariables ( )

Definition at line 8 of file MMLoadVariables.cxx.

8 : AthMessaging(Athena::getMessageSvc(), "MMLoadVariables") {}

Member Function Documentation

◆ getTruthInfo()

StatusCode MMLoadVariables::getTruthInfo ( const EventContext &  ctx,
const McEventCollection truthContainer,
const TrackRecordCollection trackRecordCollection,
std::map< std::pair< uint64_t, unsigned int >, evInf_entry > &  Event_Info 
) const

Definition at line 10 of file MMLoadVariables.cxx.

13  {
14  //*******Following MuonPRD code to access all the variables**********
15  std::vector<ROOT::Math::PtEtaPhiEVector> truthParticles, truthParticles_ent, truthParticles_pos;
16  std::vector<int> pdg;
17  std::vector<ROOT::Math::XYZVector> vertex;
18  float phiEntry_tmp = 0;
19  float phiPosition_tmp = 0;
20  float etaEntry_tmp = 0;
21  float etaPosition_tmp = 0;
22  int pdg_tmp = 0;
23  ROOT::Math::XYZVector vertex_tmp(0.,0.,0.);
24 
25  ROOT::Math::PtEtaPhiEVector thePart, theInfo;
26  auto MuEntry_Particle_n = (trackRecordCollection!=nullptr)?trackRecordCollection->size():0;
27  int j=0; // iteration of particle entries
28  if( truthContainer != nullptr ){
29  for(const auto subEvent : *truthContainer) {
30  for(const auto& particle : *subEvent){
31  const HepMC::FourVector momentum = particle->momentum();
32  if( HepMC::generations(particle) < 1 && std::abs(particle->pdg_id())==13){
33  thePart.SetCoordinates(momentum.perp(),momentum.eta(),momentum.phi(),momentum.e());
34  if(trackRecordCollection!=nullptr){
35  for(const auto & mit : *trackRecordCollection ) {
36  const CLHEP::Hep3Vector mumomentum = mit.GetMomentum();
37  const CLHEP::Hep3Vector muposition = mit.GetPosition();
38  if(!trackRecordCollection->empty() && HepMC::barcode(particle) == mit.barcode()) { // FIXME barcode-based
39  pdg_tmp = particle->pdg_id();
40  phiEntry_tmp = mumomentum.getPhi();
41  etaEntry_tmp = mumomentum.getEta();
42  phiPosition_tmp = muposition.getPhi();
43  etaPosition_tmp = muposition.getEta();
44  }
45  }//muentry loop
46  } // trackRecordCollection is not null
47 #ifdef HEPMC3
48  vertex_tmp = subEvent->vertices().front()->position();
49 #else
50  int l=0;
51  for(const auto vit : subEvent->vertex_range())
52  {
53  if(l!=0){break;}//get first vertex of iteration, may want to change this
54  l++;
55  const HepMC::GenVertex *vertex1 = vit;
56  const HepMC::FourVector& position = vertex1->position();
57  vertex_tmp.SetXYZ(position.x(),position.y(),position.z());
58  }//end vertex loop
59 #endif
60  }
61  j++;
62 
63  if(thePart.Pt() > 0. && HepMC::generations(particle) < 1){
64  bool addIt = true;
65  for(unsigned int ipart=0; ipart < truthParticles.size(); ipart++){
66  if( std::abs(thePart.Pt()-truthParticles[ipart].Pt()) < 0.001 ||
67  std::abs(thePart.Eta()-truthParticles[ipart].Eta()) < 0.001 ||
68  std::abs(xAOD::P4Helpers::deltaPhi(thePart.Phi(), truthParticles[ipart].Phi())) < 0.001 ||
69  std::abs(thePart.E()-truthParticles[ipart].E()) < 0.001 ) addIt = false;
70  }
71  if(addIt){
72  truthParticles.push_back(thePart);
73  //new stuff
74  vertex.push_back(vertex_tmp);
75  pdg.push_back(pdg_tmp);
76  truthParticles_ent.push_back(ROOT::Math::PtEtaPhiEVector(momentum.perp(),etaEntry_tmp ,phiEntry_tmp ,momentum.e()));
77  truthParticles_pos.push_back(ROOT::Math::PtEtaPhiEVector(momentum.perp(),etaPosition_tmp,phiPosition_tmp,momentum.e()));
78  }
79  }
80 
81  } //end particle loop
82  } //end truth container loop (should be only 1 container per event)
83  } // if truth container is not null
84 
85  uint64_t event = ctx.eventID().event_number();
86  for(unsigned int i=0; i<truthParticles.size(); i++) {
87  evInf_entry particle_info(event, pdg[i],
88  truthParticles[i].E(), truthParticles[i].Pt(),
89  truthParticles[i].Eta(), truthParticles_pos[i].Eta(), truthParticles_ent[i].Eta(),
90  truthParticles[i].Phi(), truthParticles_pos[i].Phi(), truthParticles_ent[i].Phi(),
91  truthParticles[i].Theta(), truthParticles_pos[i].Theta(), truthParticles_ent[i].Theta(), truthParticles_ent[i].Theta()-truthParticles_pos[i].Theta(),
92  j,MuEntry_Particle_n,vertex[i]);
93  Event_Info[std::make_pair(event,i)] = particle_info;
94  }
95 
96  return StatusCode::SUCCESS;
97 }

◆ initMessaging()

void AthMessaging::initMessaging ( ) const
privateinherited

Initialize our message level and MessageSvc.

This method should only be called once.

Definition at line 39 of file AthMessaging.cxx.

40 {
42  m_lvl = m_imsg ?
43  static_cast<MSG::Level>( m_imsg.load()->outputLevel(m_nm) ) :
44  MSG::INFO;
45 }

◆ msg() [1/2]

MsgStream & AthMessaging::msg ( ) const
inlineinherited

The standard message stream.

Returns a reference to the default message stream May not be invoked before sysInitialize() has been invoked.

Definition at line 164 of file AthMessaging.h.

165 {
166  MsgStream* ms = m_msg_tls.get();
167  if (!ms) {
168  if (!m_initialized.test_and_set()) initMessaging();
169  ms = new MsgStream(m_imsg,m_nm);
170  m_msg_tls.reset( ms );
171  }
172 
173  ms->setLevel (m_lvl);
174  return *ms;
175 }

◆ msg() [2/2]

MsgStream & AthMessaging::msg ( const MSG::Level  lvl) const
inlineinherited

The standard message stream.

Returns a reference to the default message stream May not be invoked before sysInitialize() has been invoked.

Definition at line 179 of file AthMessaging.h.

180 { return msg() << lvl; }

◆ msgLvl()

bool AthMessaging::msgLvl ( const MSG::Level  lvl) const
inlineinherited

Test the output level.

Parameters
lvlThe message level to test against
Returns
boolean Indicating if messages at given level will be printed
Return values
trueMessages at level "lvl" will be printed

Definition at line 151 of file AthMessaging.h.

152 {
153  if (!m_initialized.test_and_set()) initMessaging();
154  if (m_lvl <= lvl) {
155  msg() << lvl;
156  return true;
157  } else {
158  return false;
159  }
160 }

◆ setLevel()

void AthMessaging::setLevel ( MSG::Level  lvl)
inherited

Change the current logging level.

Use this rather than msg().setLevel() for proper operation with MT.

Definition at line 28 of file AthMessaging.cxx.

29 {
30  m_lvl = lvl;
31 }

Member Data Documentation

◆ ATLAS_THREAD_SAFE

std::atomic_flag m_initialized AthMessaging::ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT
mutableprivateinherited

Messaging initialized (initMessaging)

Definition at line 141 of file AthMessaging.h.

◆ m_imsg

std::atomic<IMessageSvc*> AthMessaging::m_imsg { nullptr }
mutableprivateinherited

MessageSvc pointer.

Definition at line 135 of file AthMessaging.h.

◆ m_lvl

std::atomic<MSG::Level> AthMessaging::m_lvl { MSG::NIL }
mutableprivateinherited

Current logging level.

Definition at line 138 of file AthMessaging.h.

◆ m_msg_tls

boost::thread_specific_ptr<MsgStream> AthMessaging::m_msg_tls
mutableprivateinherited

MsgStream instance (a std::cout like with print-out levels)

Definition at line 132 of file AthMessaging.h.

◆ m_nm

std::string AthMessaging::m_nm
privateinherited

Message source name.

Definition at line 129 of file AthMessaging.h.


The documentation for this class was generated from the following files:
AthMessaging::m_lvl
std::atomic< MSG::Level > m_lvl
Current logging level.
Definition: AthMessaging.h:138
Trk::VKContraintType::Theta
@ Theta
Trk::ParticleSwitcher::particle
constexpr ParticleHypothesis particle[PARTICLEHYPOTHESES]
the array of masses
Definition: ParticleHypothesis.h:76
xAOD::P4Helpers::deltaPhi
double deltaPhi(double phiA, double phiB)
delta Phi in range [-pi,pi[
Definition: xAODP4Helpers.h:69
UploadAMITag.l
list l
Definition: UploadAMITag.larcaf.py:158
Phi
@ Phi
Definition: RPCdef.h:8
AthMessaging::m_imsg
std::atomic< IMessageSvc * > m_imsg
MessageSvc pointer.
Definition: AthMessaging.h:135
python.SystemOfUnits.ms
int ms
Definition: SystemOfUnits.py:132
Athena::getMessageSvc
IMessageSvc * getMessageSvc(bool quiet=false)
Definition: getMessageSvc.cxx:20
AthMessaging::AthMessaging
AthMessaging()
Default constructor:
TrigConf::MSGTC::Level
Level
Definition: Trigger/TrigConfiguration/TrigConfBase/TrigConfBase/MsgStream.h:21
ParticleGun_EoverP_Config.momentum
momentum
Definition: ParticleGun_EoverP_Config.py:63
event
POOL::TEvent event(POOL::TEvent::kClassAccess)
lumiFormat.i
int i
Definition: lumiFormat.py:85
HepMC::barcode
int barcode(const T *p)
Definition: Barcode.h:16
xAOD::uint64_t
uint64_t
Definition: EventInfo_v1.cxx:123
evInf_entry
Definition: MMLoadVariables.h:22
AthMessaging::msg
MsgStream & msg() const
The standard message stream.
Definition: AthMessaging.h:164
VP1PartSpect::E
@ E
Definition: VP1PartSpectFlags.h:21
Trk::vertex
@ vertex
Definition: MeasurementType.h:21
Prompt::Def::Pt
@ Pt
Definition: VarHolder.h:76
AthMessaging::m_nm
std::string m_nm
Message source name.
Definition: AthMessaging.h:129
AtlasHitsVector::size
size_type size() const
Definition: AtlasHitsVector.h:143
AthMessaging::initMessaging
void initMessaging() const
Initialize our message level and MessageSvc.
Definition: AthMessaging.cxx:39
HepMC::generations
int generations(const T &p)
Method to return how many interactions a particle has undergone during simulation (TODO migrate to be...
Definition: MagicNumbers.h:358
AthMessaging::m_msg_tls
boost::thread_specific_ptr< MsgStream > m_msg_tls
MsgStream instance (a std::cout like with print-out levels)
Definition: AthMessaging.h:132
Eta
@ Eta
Definition: RPCdef.h:8