ATLAS Offline Software
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 28 of file MMLoadVariables.h.

Constructor & Destructor Documentation

◆ MMLoadVariables()

MMLoadVariables::MMLoadVariables ( )

Definition at line 14 of file MMLoadVariables.cxx.

14 : 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 16 of file MMLoadVariables.cxx.

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

◆ 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:79
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:157
Phi
@ Phi
Definition: RPCdef.h:8
AthMessaging::m_imsg
std::atomic< IMessageSvc * > m_imsg
MessageSvc pointer.
Definition: AthMessaging.h:135
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:17
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
python.Constants.INFO
int INFO
Definition: Control/AthenaCommon/python/Constants.py:15
AthMessaging::m_nm
std::string m_nm
Message source name.
Definition: AthMessaging.h:129
AtlasHitsVector::size
size_type size() const
Definition: AtlasHitsVector.h:142
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:357
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
python.SystemOfUnits.ms
float ms
Definition: SystemOfUnits.py:148