ATLAS Offline Software
Loading...
Searching...
No Matches
MuonR4::TruthSegmentCsvDumperAlg Class Reference

#include <TruthSegmentCsvDumperAlg.h>

Inheritance diagram for MuonR4::TruthSegmentCsvDumperAlg:
Collaboration diagram for MuonR4::TruthSegmentCsvDumperAlg:

Public Member Functions

 ~TruthSegmentCsvDumperAlg ()=default
StatusCode initialize () override
StatusCode execute () override
 AthAlgorithm (const std::string &name, ISvcLocator *pSvcLocator)
 Constructor with parameters:
virtual StatusCode sysInitialize () override
 Override sysInitialize.
virtual const DataObjIDColl & extraOutputDeps () const override
 Return the list of extra output dependencies.
ServiceHandle< StoreGateSvc > & evtStore ()
 The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.
const ServiceHandle< StoreGateSvc > & detStore () const
 The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.
virtual StatusCode sysStart () override
 Handle START transition.
virtual std::vector< Gaudi::DataHandle * > inputHandles () const override
 Return this algorithm's input handles.
virtual std::vector< Gaudi::DataHandle * > outputHandles () const override
 Return this algorithm's output handles.
Gaudi::Details::PropertyBase & declareProperty (Gaudi::Property< T, V, H > &t)
void updateVHKA (Gaudi::Details::PropertyBase &)
MsgStream & msg () const
bool msgLvl (const MSG::Level lvl) const

Protected Member Functions

void renounceArray (SG::VarHandleKeyArray &handlesArray)
 remove all handles from I/O resolution
std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > renounce (T &h)
void extraDeps_update_handler (Gaudi::Details::PropertyBase &ExtraDeps)
 Add StoreName to extra input/output deps as needed.

Private Types

typedef ServiceHandle< StoreGateSvcStoreGateSvc_t

Private Member Functions

const MuonGMR4::SpectrometerSectormsSector (const xAOD::MuonSegment &segment) const
Gaudi::Details::PropertyBase & declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
 specialization for handling Gaudi::Property<SG::VarHandleKey>

Private Attributes

SG::ReadHandleKey< ActsTrk::GeometryContextm_geoCtxKey {this, "AlignmentKey", "ActsAlignment", "cond handle key"}
SG::ReadHandleKey< xAOD::MuonSegmentContainerm_inSegmentKey {this, "MuonTruthSegmentsKey", "MuonTruthSegments"}
ServiceHandle< Muon::IMuonIdHelperSvcm_idHelperSvc {this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"}
ServiceHandle< Muon::IMuonEDMHelperSvcm_edmHelperSvc {this, "edmHelperSvc", "Muon::MuonEDMHelperSvc/MuonEDMHelperSvc"}
const MuonGMR4::MuonDetectorManagerm_detMgr {nullptr}
size_t m_event {0}
DataObjIDColl m_extendedExtraObjects
StoreGateSvc_t m_evtStore
 Pointer to StoreGate (event store by default)
StoreGateSvc_t m_detStore
 Pointer to StoreGate (detector store by default)
std::vector< SG::VarHandleKeyArray * > m_vhka
bool m_varHandleArraysDeclared

Detailed Description

Definition at line 17 of file TruthSegmentCsvDumperAlg.h.

Member Typedef Documentation

◆ StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< Algorithm > >::StoreGateSvc_t
privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

◆ ~TruthSegmentCsvDumperAlg()

MuonR4::TruthSegmentCsvDumperAlg::~TruthSegmentCsvDumperAlg ( )
default

Member Function Documentation

◆ AthAlgorithm()

AthAlgorithm::AthAlgorithm ( const std::string & name,
ISvcLocator * pSvcLocator )

Constructor with parameters:

Definition at line 51 of file AthAlgorithm.cxx.

25 :
27{
28 // Set up to run AthAlgorithmDHUpdate in sysInitialize before
29 // merging dependency lists. This extends the output dependency
30 // list with any symlinks implied by inheritance relations.
31 m_updateDataHandles =
32 std::make_unique<AthenaBaseComps::AthAlgorithmDHUpdate>
34 std::move (m_updateDataHandles));
35}
DataObjIDColl m_extendedExtraObjects
AthCommonDataStore(const std::string &name, T... args)

◆ declareGaudiProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< Algorithm > >::declareGaudiProperty ( Gaudi::Property< T, V, H > & hndl,
const SG::VarHandleKeyType &  )
inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleKey>

Definition at line 156 of file AthCommonDataStore.h.

158 {
160 hndl.value(),
161 hndl.documentation());
162
163 }
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T, V, H > &t)

◆ declareProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty ( Gaudi::Property< T, V, H > & t)
inlineinherited

Definition at line 145 of file AthCommonDataStore.h.

145 {
146 typedef typename SG::HandleClassifier<T>::type htype;
148 }
Gaudi::Details::PropertyBase & declareGaudiProperty(Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
specialization for handling Gaudi::Property<SG::VarHandleKey>

◆ detStore()

const ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< Algorithm > >::detStore ( ) const
inlineinherited

The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 95 of file AthCommonDataStore.h.

◆ evtStore()

ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< Algorithm > >::evtStore ( )
inlineinherited

The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 85 of file AthCommonDataStore.h.

◆ execute()

StatusCode MuonR4::TruthSegmentCsvDumperAlg::execute ( )
override

Segment information

Local direction

Definition at line 54 of file TruthSegmentCsvDumperAlg.cxx.

54 {
55 const EventContext & ctx = Gaudi::Hive::currentContext();
56 std::ofstream file{std::format("event-{:09}-MuonTruthSegment.csv", ++m_event)};
57 constexpr std::string_view delim = ",";
58 file<<"sectorId"<<delim;
59 file<<"globalPositionX"<<delim;
60 file<<"globalPositionY"<<delim;
61 file<<"globalPositionZ"<<delim;
62
63 file<<"globalDirectionX"<<delim;
64 file<<"globalDirectionY"<<delim;
65 file<<"globalDirectionZ"<<delim;
66
67
68 file<<"localPositionX"<<delim;
69 file<<"localPositionY"<<delim;
70 file<<"localPositionZ"<<delim;
71
72 file<<"localDirectionX"<<delim;
73 file<<"localDirectionY"<<delim;
74 file<<"localDirectionZ"<<delim;
75
76 file<<"time"<<delim;
77 file<<"timeError"<<delim;
78 file<<"chiSquared"<<delim;
79 file<<"nDoF"<<delim;
80 file<<"precisionHits"<<delim;
81 file<<"phiLayers"<<delim;
82 file<<"trigEtaLayers"<<delim;
83 file<<std::endl;
84
85 const xAOD::MuonSegmentContainer* readTruthSegment{nullptr};
86 ATH_CHECK(SG::get(readTruthSegment, m_inSegmentKey, ctx));
87
88 const ActsTrk::GeometryContext* gctxHandle{nullptr};
89 ATH_CHECK(SG::get(gctxHandle, m_geoCtxKey, ctx));
90
91 for (const xAOD::MuonSegment* segment : *readTruthSegment) {
92 const MuonGMR4::SpectrometerSector* sector = msSector(*segment);
93 if (!sector) {
94 continue;
95 }
96 const Amg::Transform3D globToLoc{sector->globalToLocalTrans(*gctxHandle)};
98 const Amg::Vector3D globPos = segment->position();
99 const Amg::Vector3D globDir = segment->direction();
101 const Amg::Vector3D locPos = globToLoc * globPos;
102 const Amg::Vector3D locDir = globToLoc.linear() * globDir;
103
104 const int secId = encodeId(static_cast<int8_t>(sector->chamberIndex()),
105 sector->side(),
106 sector->sector());
107 // time information
108 float seg_t0 = segment->t0();
109 float seg_t0error = segment->t0error();
110
111 // Fit quality information
112 float seg_chiSquared = segment->chiSquared();
113 float seg_numberDoF = segment->numberDoF();
114
115 // nHits
116 int seg_PrecisionHits = segment->nPrecisionHits();
117 int seg_PhiLayers = segment->nPhiLayers();
118 int seg_TrigEtaLayers = segment->nTrigEtaLayers();
119
120 // Detector information
121
122
123 // verbose some information to check
124 ATH_MSG_VERBOSE("Segment global position "<<Amg::toString(globPos)<<", direction: "<<Amg::toString(globDir));
125 ATH_MSG_VERBOSE("Segment local position: "<<Amg::toString(locPos)<<", direction: "<<Amg::toString(locDir));
126 ATH_MSG_VERBOSE("t0: "<<seg_t0<<" t0error: "<<seg_t0error);
127 ATH_MSG_VERBOSE("chiSquared: "<<seg_chiSquared<<" numberDoF: "<<seg_numberDoF);
128 ATH_MSG_VERBOSE("nPrecisionHits: "<<seg_PrecisionHits<<" nPhiLayers: "<<seg_PhiLayers<<" nTrigEtaLayers: "<<seg_TrigEtaLayers);
129
130 // save the segment information to the csv file
131 file<<secId<<delim;
132 file<<precCutOff(globPos.x())<<delim;
133 file<<precCutOff(globPos.y())<<delim;
134 file<<precCutOff(globPos.z())<<delim;
135 file<<precCutOff(globDir.x())<<delim;
136 file<<precCutOff(globDir.y())<<delim;
137 file<<precCutOff(globDir.z())<<delim;
138
139 file<<precCutOff(locPos.x())<<delim;
140 file<<precCutOff(locPos.y())<<delim;
141 file<<precCutOff(locPos.z())<<delim;
142 file<<precCutOff(locDir.x())<<delim;
143 file<<precCutOff(locDir.y())<<delim;
144 file<<precCutOff(locDir.z())<<delim;
145
146
147 file<<precCutOff(seg_t0)<<delim;
148 file<<precCutOff(seg_t0error)<<delim;
149 file<<precCutOff(seg_chiSquared)<<delim;
150 file<<seg_numberDoF<<delim;
151 file<<seg_PrecisionHits<<delim;
152 file<<seg_PhiLayers<<delim;
153 file<<seg_TrigEtaLayers<<delim;
154 file<<std::endl;
155 }
156 return StatusCode::SUCCESS;
157}
#define ATH_CHECK
Evaluate an expression and check for errors.
#define ATH_MSG_VERBOSE(x)
int8_t side() const
Returns the side of the MS-sector 1 -> A side ; -1 -> C side.
Amg::Transform3D globalToLocalTrans(const ActsTrk::GeometryContext &gctx) const
Returns the global -> local transformation from the ATLAS global.
int sector() const
Returns the sector of the MS-sector.
Muon::MuonStationIndex::ChIndex chamberIndex() const
Returns the chamber index scheme.
const MuonGMR4::SpectrometerSector * msSector(const xAOD::MuonSegment &segment) const
SG::ReadHandleKey< xAOD::MuonSegmentContainer > m_inSegmentKey
SG::ReadHandleKey< ActsTrk::GeometryContext > m_geoCtxKey
std::string toString(const Translation3D &translation, int precision=4)
GeoPrimitvesToStringConverter.
Eigen::Affine3d Transform3D
Eigen::Matrix< double, 3, 1 > Vector3D
const T * get(const ReadCondHandleKey< T > &key, const EventContext &ctx)
Convenience function to retrieve an object given a ReadCondHandleKey.
MuonSegmentContainer_v1 MuonSegmentContainer
Definition of the current "MuonSegment container version".
MuonSegment_v1 MuonSegment
Reference the current persistent version:
TFile * file

◆ extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< Algorithm > >::extraDeps_update_handler ( Gaudi::Details::PropertyBase & ExtraDeps)
protectedinherited

Add StoreName to extra input/output deps as needed.

use the logic of the VarHandleKey to parse the DataObjID keys supplied via the ExtraInputs and ExtraOuputs Properties to add the StoreName if it's not explicitly given

◆ extraOutputDeps()

const DataObjIDColl & AthAlgorithm::extraOutputDeps ( ) const
overridevirtualinherited

Return the list of extra output dependencies.

This list is extended to include symlinks implied by inheritance relations.

Definition at line 50 of file AthAlgorithm.cxx.

51{
52 // If we didn't find any symlinks to add, just return the collection
53 // from the base class. Otherwise, return the extended collection.
54 if (!m_extendedExtraObjects.empty()) {
56 }
57 return Algorithm::extraOutputDeps();
58}

◆ initialize()

StatusCode MuonR4::TruthSegmentCsvDumperAlg::initialize ( )
override

Definition at line 33 of file TruthSegmentCsvDumperAlg.cxx.

33 {
34 ATH_CHECK(m_inSegmentKey.initialize());
35 ATH_CHECK(m_geoCtxKey.initialize());
36 ATH_CHECK(m_idHelperSvc.retrieve());
37 ATH_CHECK(m_edmHelperSvc.retrieve());
39 return StatusCode::SUCCESS;
40}
const ServiceHandle< StoreGateSvc > & detStore() const
ServiceHandle< Muon::IMuonIdHelperSvc > m_idHelperSvc
ServiceHandle< Muon::IMuonEDMHelperSvc > m_edmHelperSvc
const MuonGMR4::MuonDetectorManager * m_detMgr
retrieve(aClass, aKey=None)
Definition PyKernel.py:110

◆ inputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< Algorithm > >::inputHandles ( ) const
overridevirtualinherited

Return this algorithm's input handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ msg()

MsgStream & AthCommonMsg< Algorithm >::msg ( ) const
inlineinherited

Definition at line 24 of file AthCommonMsg.h.

24 {
25 return this->msgStream();
26 }

◆ msgLvl()

bool AthCommonMsg< Algorithm >::msgLvl ( const MSG::Level lvl) const
inlineinherited

Definition at line 30 of file AthCommonMsg.h.

30 {
31 return this->msgLevel(lvl);
32 }

◆ msSector()

const MuonGMR4::SpectrometerSector * MuonR4::TruthSegmentCsvDumperAlg::msSector ( const xAOD::MuonSegment & segment) const
private

Definition at line 42 of file TruthSegmentCsvDumperAlg.cxx.

42 {
43 const auto truthHits = getMatchingSimHits(segment);
44 if (truthHits.size()) {
45 return m_detMgr->getSectorEnvelope((*truthHits.begin())->identify());
46 }
47 if (segment.muonSegment().isValid()) {
48 return m_detMgr->getSectorEnvelope(m_edmHelperSvc->chamberId(static_cast<const Muon::MuonSegment&>(**segment.muonSegment())));
49 }
50 ATH_MSG_WARNING("No matching hits were found. Neither the simulated ones or the reconstructed");
51 return nullptr;
52}
#define ATH_MSG_WARNING(x)
const ElementLink< ::Trk::SegmentCollection > & muonSegment() const
std::unordered_set< const xAOD::MuonSimHit * > getMatchingSimHits(const xAOD::MuonSegment &segment)
: Returns all sim hits matched to a xAOD::MuonSegment

◆ outputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< Algorithm > >::outputHandles ( ) const
overridevirtualinherited

Return this algorithm's output handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ renounce()

std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > AthCommonDataStore< AthCommonMsg< Algorithm > >::renounce ( T & h)
inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

381 {
382 h.renounce();
384 }
std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > renounce(T &h)

◆ renounceArray()

void AthCommonDataStore< AthCommonMsg< Algorithm > >::renounceArray ( SG::VarHandleKeyArray & handlesArray)
inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

364 {
366 }

◆ sysInitialize()

StatusCode AthAlgorithm::sysInitialize ( )
overridevirtualinherited

Override sysInitialize.

Override sysInitialize from the base class.

Loop through all output handles, and if they're WriteCondHandles, automatically register them and this Algorithm with the CondSvc

Scan through all outputHandles, and if they're WriteCondHandles, register them with the CondSvc

Reimplemented from AthCommonDataStore< AthCommonMsg< Algorithm > >.

Reimplemented in AthAnalysisAlgorithm, AthFilterAlgorithm, AthHistogramAlgorithm, and PyAthena::Alg.

Definition at line 66 of file AthAlgorithm.cxx.

66 {
68
69 if (sc.isFailure()) {
70 return sc;
71 }
72 ServiceHandle<ICondSvc> cs("CondSvc",name());
73 for (auto h : outputHandles()) {
74 if (h->isCondition() && h->mode() == Gaudi::DataHandle::Writer) {
75 // do this inside the loop so we don't create the CondSvc until needed
76 if ( cs.retrieve().isFailure() ) {
77 ATH_MSG_WARNING("no CondSvc found: won't autoreg WriteCondHandles");
78 return StatusCode::SUCCESS;
79 }
80 if (cs->regHandle(this,*h).isFailure()) {
81 sc = StatusCode::FAILURE;
82 ATH_MSG_ERROR("unable to register WriteCondHandle " << h->fullKey()
83 << " with CondSvc");
84 }
85 }
86 }
87 return sc;
88}
#define ATH_MSG_ERROR(x)
static Double_t sc
virtual StatusCode sysInitialize() override
Override sysInitialize.
virtual std::vector< Gaudi::DataHandle * > outputHandles() const override
::StatusCode StatusCode
StatusCode definition for legacy code.

◆ sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< Algorithm > >::sysStart ( )
overridevirtualinherited

Handle START transition.

We override this in order to make sure that conditions handle keys can cache a pointer to the conditions container.

◆ updateVHKA()

void AthCommonDataStore< AthCommonMsg< Algorithm > >::updateVHKA ( Gaudi::Details::PropertyBase & )
inlineinherited

Definition at line 308 of file AthCommonDataStore.h.

308 {
309 // debug() << "updateVHKA for property " << p.name() << " " << p.toString()
310 // << " size: " << m_vhka.size() << endmsg;
311 for (auto &a : m_vhka) {
313 for (auto k : keys) {
314 k->setOwner(this);
315 }
316 }
317 }
std::vector< SG::VarHandleKeyArray * > m_vhka

Member Data Documentation

◆ m_detMgr

const MuonGMR4::MuonDetectorManager* MuonR4::TruthSegmentCsvDumperAlg::m_detMgr {nullptr}
private

Definition at line 33 of file TruthSegmentCsvDumperAlg.h.

33{nullptr};

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Algorithm > >::m_detStore
privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

◆ m_edmHelperSvc

ServiceHandle<Muon::IMuonEDMHelperSvc> MuonR4::TruthSegmentCsvDumperAlg::m_edmHelperSvc {this, "edmHelperSvc", "Muon::MuonEDMHelperSvc/MuonEDMHelperSvc"}
private

Definition at line 31 of file TruthSegmentCsvDumperAlg.h.

31{this, "edmHelperSvc", "Muon::MuonEDMHelperSvc/MuonEDMHelperSvc"};

◆ m_event

size_t MuonR4::TruthSegmentCsvDumperAlg::m_event {0}
private

Definition at line 34 of file TruthSegmentCsvDumperAlg.h.

34{0};

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Algorithm > >::m_evtStore
privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

◆ m_extendedExtraObjects

DataObjIDColl AthAlgorithm::m_extendedExtraObjects
privateinherited

Definition at line 79 of file AthAlgorithm.h.

◆ m_geoCtxKey

SG::ReadHandleKey<ActsTrk::GeometryContext> MuonR4::TruthSegmentCsvDumperAlg::m_geoCtxKey {this, "AlignmentKey", "ActsAlignment", "cond handle key"}
private

Definition at line 27 of file TruthSegmentCsvDumperAlg.h.

27{this, "AlignmentKey", "ActsAlignment", "cond handle key"};

◆ m_idHelperSvc

ServiceHandle<Muon::IMuonIdHelperSvc> MuonR4::TruthSegmentCsvDumperAlg::m_idHelperSvc {this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"}
private

Definition at line 30 of file TruthSegmentCsvDumperAlg.h.

30{this, "MuonIdHelperSvc", "Muon::MuonIdHelperSvc/MuonIdHelperSvc"};

◆ m_inSegmentKey

SG::ReadHandleKey<xAOD::MuonSegmentContainer> MuonR4::TruthSegmentCsvDumperAlg::m_inSegmentKey {this, "MuonTruthSegmentsKey", "MuonTruthSegments"}
private

Definition at line 28 of file TruthSegmentCsvDumperAlg.h.

28{this, "MuonTruthSegmentsKey", "MuonTruthSegments"};

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< Algorithm > >::m_varHandleArraysDeclared
privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vhka

std::vector<SG::VarHandleKeyArray*> AthCommonDataStore< AthCommonMsg< Algorithm > >::m_vhka
privateinherited

Definition at line 398 of file AthCommonDataStore.h.


The documentation for this class was generated from the following files: