ATLAS Offline Software
Loading...
Searching...
No Matches
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.
MsgStream & msg () const
 The standard message stream.
MsgStream & msg (const MSG::Level lvl) const
 The standard message stream.
void setLevel (MSG::Level lvl)
 Change the current logging level.

Private Member Functions

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

Private Attributes

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

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") {}
AthMessaging()
Default constructor:
IMessageSvc * getMessageSvc(bool quiet=false)

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}
@ Phi
Definition RPCdef.h:8
@ Eta
Definition RPCdef.h:8
size_type size() const
int barcode(const T *p)
Definition Barcode.h:16
int generations(const T &p)
Method to return how many interactions a particle has undergone during simulation (TODO migrate to be...
l
Printing final latex table to .tex output file.
constexpr ParticleHypothesis particle[PARTICLEHYPOTHESES]
the array of masses
double deltaPhi(double phiA, double phiB)
delta Phi in range [-pi,pi[

◆ 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 // If user did not set an explicit level, set a default
43 if (m_lvl == MSG::NIL) {
44 m_lvl = m_imsg ?
45 static_cast<MSG::Level>( m_imsg.load()->outputLevel(m_nm) ) :
46 MSG::INFO;
47 }
48}
std::string m_nm
Message source name.
std::atomic< IMessageSvc * > m_imsg
MessageSvc pointer.
std::atomic< MSG::Level > m_lvl
Current logging level.

◆ 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 163 of file AthMessaging.h.

164{
165 MsgStream* ms = m_msg_tls.get();
166 if (!ms) {
167 if (!m_initialized.test_and_set()) initMessaging();
168 ms = new MsgStream(m_imsg,m_nm);
169 m_msg_tls.reset( ms );
170 }
171
172 ms->setLevel (m_lvl);
173 return *ms;
174}
boost::thread_specific_ptr< MsgStream > m_msg_tls
MsgStream instance (a std::cout like with print-out levels)
void initMessaging() const
Initialize our message level and MessageSvc.

◆ 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 178 of file AthMessaging.h.

179{ return msg() << lvl; }
MsgStream & msg() const
The standard message stream.

◆ 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_lvl <= lvl) {
154 msg() << lvl;
155 return true;
156 } else {
157 return false;
158 }
159}

◆ 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.

135{ nullptr };

◆ m_lvl

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

Current logging level.

Definition at line 138 of file AthMessaging.h.

138{ MSG::NIL };

◆ 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: