ATLAS Offline Software
Loading...
Searching...
No Matches
MuonValR4::TrackVisualizationTool Class Reference

#include <TrackVisualizationTool.h>

Inheritance diagram for MuonValR4::TrackVisualizationTool:
Collaboration diagram for MuonValR4::TrackVisualizationTool:

Classes

struct  PlotLegend
 Helper struct to administer the markers & colors that are added to the legend. More...

Public Member Functions

virtual ~TrackVisualizationTool ()=default
 Destructor.
virtual StatusCode initialize () override final
virtual void displaySeeds (const EventContext &ctx, const MuonR4::MsTrackSeeder &seederObj, const xAOD::MuonSegmentContainer &segments, const MuonR4::MsTrackSeedContainer &seeds) const override final
 Displays all segments on the representative cylinder in the R-Z & X-Y plane and draws the markers of the successfully built seeds & truth segments.
virtual void displaySeeds (const EventContext &ctx, const MuonR4::MsTrackSeeder &seederObj, const xAOD::MuonSegmentContainer &segments, const MuonR4::MsTrackSeedContainer &seeds, PrimitivesVec_t &&extPrimitives) const override final
 Displays all segments on the representative cylinder in the R-Z & X-Y plane and draws the markers of the successfully built seeds & truth segments.
virtual void displayTrackSeedObj (const EventContext &ctx, const MuonR4::MsTrackSeed &seed, const OptBoundPars_t &parsToExt, const std::string &objName="") const override final
 Visualizes the measurements of the segments on the track seed together with their predicted local line parameters as an obj file.

Private Types

enum class  DisplayView { RZ , XY }
 Enumeration to toggle whether the seeds shall be painted in x-y or R-z view. More...
using Canvas_t = IRootVisualizationService::ICanvasObject
using Edges = Canvas_t::AxisRanges
using Location = MuonR4::MsTrackSeed::Location
 Enumeration to toggle whether the segment shall displayed on the barrel/endcap.

Private Member Functions

void displaySeeds (const EventContext &ctx, const MuonR4::MsTrackSeeder &seederObj, const DisplayView view, const xAOD::MuonSegmentContainer &segments, const MuonR4::MsTrackSeedContainer &seeds, PrimitivesVec_t &&extPrimitives) const
 Actual implementation of the display seeds method with the augmentation to define the view.
void fillTruthSeedPoints (const EventContext &ctx, const MuonR4::MsTrackSeeder &seeder, const DisplayView view, PlotLegend &legend, Canvas_t &canvas) const
 Add the truth segments to the canvas.
void displaySeedSegmentsGlobalWithTruth (const EventContext &ctx, const DisplayView view, const MuonR4::MsTrackSeedContainer &seeds) const

Static Private Member Functions

static Amg::Vector2D viewVector (const double phi, const Amg::Vector2D &posOnCylinder, const DisplayView view)
 Transforms the projected vector into the actual view, if it's xy then an external phi is needed to place the marker.

Private Attributes

ToolHandle< ActsTrk::IExtrapolationToolm_extrapolationTool {this, "ExtrapolationTool" ,"" }
 Track extrapolation tool.
ServiceHandle< IRootVisualizationServicem_visualSvc {this, "VisualSvc", "MuonValR4::RootVisualizationService"}
 Service handle of the visualization service.
IRootVisualizationService::ClientToken m_clientToken {}
 Token to present to the visualization service such that the display froms this tool are grouped together.
SG::ReadHandleKey< xAOD::MuonSegmentContainerm_truthSegKey {this, "TruthSegkey", "MuonTruthSegments"}
 Key to the truth segment selection to draw the segment parameters.
ActsTrk::GeoContextReadKey_t m_geoCtxKey {this, "AlignmentKey", "ActsAlignment", "cond handle key"}
 Dependency on the geometry alignment.
Gaudi::Property< unsigned > m_canvasLimit {this, "CanvasLimit", 5000}
 Maximum canvases to draw.
Gaudi::Property< bool > m_saveSinglePDFs {this, "saveSinglePDFs", true}
 If set to true each canvas is saved into a dedicated pdf file.
Gaudi::Property< bool > m_saveSummaryPDF {this, "saveSummaryPDF", true}
 If set to true a summary Canvas is created.
Gaudi::Property< std::string > m_subDir {this, "outSubDir", ""}
 Define the subdirectory in which the plots shall be saved.
Gaudi::Property< std::string > m_canvasPrefix {this, "CanvasPreFix", "MsTrackValid"}
 Prefix of the individual canvas file names <MANDATORY>.
std::atomic< bool > m_plotsDone {false}
 Flag toggling whether all Canvases have been exhausted.
std::atomic< unsigned > m_objCounter {0}
 How many obj have been produced.

Detailed Description

Definition at line 21 of file TrackVisualizationTool.h.

Member Typedef Documentation

◆ Canvas_t

◆ Edges

◆ Location

Enumeration to toggle whether the segment shall displayed on the barrel/endcap.

Definition at line 98 of file TrackVisualizationTool.h.

Member Enumeration Documentation

◆ DisplayView

Enumeration to toggle whether the seeds shall be painted in x-y or R-z view.

Enumerator
RZ 
XY 

Definition at line 100 of file TrackVisualizationTool.h.

100{RZ, XY};

Constructor & Destructor Documentation

◆ ~TrackVisualizationTool()

virtual MuonValR4::TrackVisualizationTool::~TrackVisualizationTool ( )
virtualdefault

Destructor.

Member Function Documentation

◆ displaySeeds() [1/3]

void MuonValR4::TrackVisualizationTool::displaySeeds ( const EventContext & ctx,
const MuonR4::MsTrackSeeder & seederObj,
const DisplayView view,
const xAOD::MuonSegmentContainer & segments,
const MuonR4::MsTrackSeedContainer & seeds,
PrimitivesVec_t && extPrimitives ) const
private

Actual implementation of the display seeds method with the augmentation to define the view.

Parameters
ctxEventContext to access store gate & conditions
seederObjConfigured instance of the track seeder which actually constructed the seeds.
viewActual view to display
segmentsContainer of all MS segments in the event
seedsThe constructed track seeds from the event
extPrimitivesExtra TObjects that should be additionally painted onto the Canvases

First add the truth points

Draw the sector map

Definition at line 299 of file TrackVisualizationTool.cxx.

304 {
305
306 auto canvas = m_visualSvc->prepareCanvas(ctx, m_clientToken, std::format("SeedDisplay{:}",
307 view == DisplayView::XY ? "XY" : "RZ" ));
308
309 if (!canvas) {
310 m_plotsDone = true;
311 return;
312 }
313 canvas->setAxisTitles(view == DisplayView::XY ? "x [m]" : "z [m]",
314 view == DisplayView::XY ? "y [m]" : "R [m]");
315
316 PlotLegend legend{0.005,0.005, 0.6,0.1};
318 fillTruthSeedPoints(ctx, seeder, view, legend, *canvas);
319
320 auto onSeed = [&seeds](const xAOD::MuonSegment* segment,
321 const Location loc) {
322 return std::ranges::any_of(seeds,[segment, loc](const MsTrackSeed& seed){
323 return seed.location() == loc && std::ranges::find(seed.segments(), segment) != seed.segments().end();
324 });
325 };
326
327 const ActsTrk::GeometryContext* gctx{nullptr};
328 if (!SG::get(gctx, m_geoCtxKey, ctx).isSuccess()) {
329 THROW_EXCEPTION("Failed to fetch the geometry context "<<m_geoCtxKey.fullKey());
330 }
331 bool drawnPoint{false};
332
333 for (const xAOD::MuonSegment* segment: segments) {
334 using enum Location;
336 using namespace Muon;
337 const MuonGMR4::SpectrometerSector* msSector = detailedSegment(*segment)->msSector();
338 const auto chIdx = segment->chamberIndex();
339 const int mColor = msSector->barrel() ? ColorBarrel : (msSector->side() > 0 ? ColorEndcapA : ColorEndcapC);
340
341 for (const auto secProj : {leftOverlap, center, rightOverlap}) {
342 const ExpandedSector sector{static_cast<unsigned>(segment->sector()), secProj};
343 if (ExpandedSector{segment->position().phi()} != sector){
344 continue;
345 }
346
347 for (const Location loc : {Barrel, Endcap}) {
348 const Amg::Vector2D projPos{seeder.expressOnCylinder(*gctx, *segment, loc, sector)};
349 if (!seeder.withinBounds(projPos, loc)) {
350 continue;
351 }
352 drawnPoint = true;
353 const bool isGood = onSeed(segment, loc);
354 const int mStyle = stationMarkerSyle(chIdx, true, isGood);
355 const Amg::Vector2D markerPos{viewVector(sector.phi(), projPos, view)};
356 extPrimitives.emplace_back( drawMarker(markerPos, mStyle, mColor)); //Is this necessary?
357 canvas->add( drawMarker(markerPos, mStyle, mColor));
358
359 if (view == DisplayView::XY) {
360 const double r = markerPos.mag() + extraMargin;
361 canvas->expandPad(r, r);
362 canvas->expandPad(-r, -r);
363 } else {
364 canvas->expandPad(markerPos[0] - extraMargin, markerPos[1] - extraMargin);
365 canvas->expandPad(markerPos[0] + extraMargin, markerPos[1] + extraMargin);
366 }
367 legend.addMarker(mStyle, std::format("{:}{:}", MuonStationIndex::layerName(MuonStationIndex::toLayerIndex(chIdx)),
368 isGood ? "" : " (discarded)"), mColor);
369 legend.addColor(mColor, msSector->barrel() ? "Barrel" : msSector->side() > 0 ? "Endcap A" : "Endcap C");
370 }
371 }
372 }
373 if (!drawnPoint) {
374 canvas->trash();
375 return;
376 }
377 for (const MsTrackSeed& seed : seeds) {
378 const Amg::Vector3D& pos{seed.position()};
379 canvas->add(drawMarker(viewVector(pos.phi(), Amg::Vector2D{pos.perp(), pos.z()}, view),
380 kFullDiamond, kBlack));
381 legend.addMarker(kFullDiamond, "track seed");
382 }
384 for (unsigned int s = 1; s<=16 ; ++s) {
385 if (view != DisplayView::XY) {
386 break;
387 }
388 const double phi = sectorMap.sectorPhi(s);
389 const double dPhi = sectorMap.sectorWidth(s);
390
391 const int lStyle = s%2 ? kDashed : kDotted;
392 const double r = canvas->corner(Edges::xHigh);
393 const CxxUtils::sincos phiM{phi-dPhi}, phiP{phi+dPhi};
394 const Amg::Vector2D e1{r * phiM.cs, r * phiM.sn};
395 const Amg::Vector2D e2{r * phiP.cs, r * phiP.sn};
396
397 auto theLine = std::make_unique<TLine>(0.,0., e1.x(), e1.y());
398 theLine->SetLineStyle(lStyle);
399 extPrimitives.insert(extPrimitives.begin(), std::move(theLine));
400 theLine = std::make_unique<TLine>(0.,0., e2.x(), e2.y());
401 theLine->SetLineStyle(lStyle);
402 extPrimitives.insert(extPrimitives.begin(), std::move(theLine));
403 theLine = std::make_unique<TLine>(e1.x(), e1.y(), e2.x(), e2.y());
404 theLine->SetLineStyle(lStyle);
405 extPrimitives.insert(extPrimitives.begin(), std::move(theLine));
406 }
407 legend.fillPrimitives(*canvas);
408 }
Scalar phi() const
phi method
int8_t side() const
Returns the side of the MS-sector 1 -> A side ; -1 -> C side.
bool barrel() const
Returns whether the sector is placed in the barrel.
double phi() const
Returns the phi angle of the expanded sector.
ExpandedSector::SectorProjector SectorProjector
Recycle the expanded sector.
const MuonGMR4::SpectrometerSector * msSector() const
Returns the associated MS sector.
ServiceHandle< IRootVisualizationService > m_visualSvc
Service handle of the visualization service.
std::atomic< bool > m_plotsDone
Flag toggling whether all Canvases have been exhausted.
IRootVisualizationService::ClientToken m_clientToken
Token to present to the visualization service such that the display froms this tool are grouped toget...
static Amg::Vector2D viewVector(const double phi, const Amg::Vector2D &posOnCylinder, const DisplayView view)
Transforms the projected vector into the actual view, if it's xy then an external phi is needed to pl...
ActsTrk::GeoContextReadKey_t m_geoCtxKey
Dependency on the geometry alignment.
MuonR4::MsTrackSeed::Location Location
Enumeration to toggle whether the segment shall displayed on the barrel/endcap.
void fillTruthSeedPoints(const EventContext &ctx, const MuonR4::MsTrackSeeder &seeder, const DisplayView view, PlotLegend &legend, Canvas_t &canvas) const
Add the truth segments to the canvas.
::Muon::MuonStationIndex::ChIndex chamberIndex() const
Returns the chamber index.
Amg::Vector3D position() const
Returns the position as Amg::Vector.
int r
Definition globals.cxx:22
Eigen::Matrix< double, 2, 1 > Vector2D
Eigen::Matrix< double, 3, 1 > Vector3D
@ Endcap
Definition DetType.h:14
const Segment * detailedSegment(const xAOD::MuonSegment &seg)
Helper function to navigate from the xAOD::MuonSegment to the MuonR4::Segment.
const T * get(const ReadCondHandleKey< T > &key, const EventContext &ctx)
Convenience function to retrieve an object given a ReadCondHandleKey.
bool dPhi(const xAOD::TauJet &tau, const xAOD::CaloVertexedTopoCluster &cluster, float &out)
double e2(const xAOD::CaloCluster &cluster)
return the uncorrected cluster energy in 2nd sampling
double e1(const xAOD::CaloCluster &cluster)
return the uncorrected cluster energy in 1st sampling
Muon_v1 Muon
Reference the current persistent version:
MuonSegment_v1 MuonSegment
Reference the current persistent version:
Helper struct to administer the markers & colors that are added to the legend.
#define THROW_EXCEPTION(MESSAGE)
Definition throwExcept.h:10

◆ displaySeeds() [2/3]

void MuonValR4::TrackVisualizationTool::displaySeeds ( const EventContext & ctx,
const MuonR4::MsTrackSeeder & seederObj,
const xAOD::MuonSegmentContainer & segments,
const MuonR4::MsTrackSeedContainer & seeds ) const
finaloverridevirtual

Displays all segments on the representative cylinder in the R-Z & X-Y plane and draws the markers of the successfully built seeds & truth segments.

Parameters
ctxEventContext to access store gate & conditions
seederObjConfigured instance of the track seeder which actually constructed the seeds.
segmentsContainer of all MS segments in the event
seedsThe constructed track seeds from the event

Definition at line 146 of file TrackVisualizationTool.cxx.

149 {
150 displaySeeds(ctx, seederObj, segments, seeds, PrimitivesVec_t{});
151 }
virtual void displaySeeds(const EventContext &ctx, const MuonR4::MsTrackSeeder &seederObj, const xAOD::MuonSegmentContainer &segments, const MuonR4::MsTrackSeedContainer &seeds) const override final
Displays all segments on the representative cylinder in the R-Z & X-Y plane and draws the markers of ...

◆ displaySeeds() [3/3]

void MuonValR4::TrackVisualizationTool::displaySeeds ( const EventContext & ctx,
const MuonR4::MsTrackSeeder & seederObj,
const xAOD::MuonSegmentContainer & segments,
const MuonR4::MsTrackSeedContainer & seeds,
PrimitivesVec_t && extPrimitives ) const
finaloverridevirtual

Displays all segments on the representative cylinder in the R-Z & X-Y plane and draws the markers of the successfully built seeds & truth segments.

Parameters
ctxEventContext to access store gate & conditions
seederObjConfigured instance of the track seeder which actually constructed the seeds.
segmentsContainer of all MS segments in the event
extPrimitivesExtra TObjects that should be additionally painted onto the Canvases

Definition at line 152 of file TrackVisualizationTool.cxx.

156 {
157 if (m_plotsDone || segments.empty()) {
158 return;
159 }
160
161
162
163 displaySeeds(ctx, seederObj, DisplayView::RZ, segments, seeds, clone(extPrimitives));
164 displaySeeds(ctx, seederObj, DisplayView::XY, segments, seeds, std::move(extPrimitives));
165
166 //For each seed plot respective segment parameters
167
170
171 }
bool empty() const noexcept
Returns true if the collection is empty.
void displaySeedSegmentsGlobalWithTruth(const EventContext &ctx, const DisplayView view, const MuonR4::MsTrackSeedContainer &seeds) const
std::vector< std::unique_ptr< TObject > > clone(const std::vector< std::unique_ptr< TObject > > &cloneMe)

◆ displaySeedSegmentsGlobalWithTruth()

void MuonValR4::TrackVisualizationTool::displaySeedSegmentsGlobalWithTruth ( const EventContext & ctx,
const DisplayView view,
const MuonR4::MsTrackSeedContainer & seeds ) const
private

Definition at line 173 of file TrackVisualizationTool.cxx.

176 {
177
178 auto canvas = m_visualSvc->prepareCanvas(
179 ctx, m_clientToken,
180 std::format("SeedSegsGlobalTruth{:}", view == DisplayView::XY ? "XY" : "RZ"));
181
182 if (!canvas) { m_plotsDone = true; return; }
183
184 canvas->setAxisTitles(view == DisplayView::XY ? "x [m]" : "z [m]",
185 view == DisplayView::XY ? "y [m]" : "R [m]");
186
187 PlotLegend legend{0.005, 0.005, 0.6, 0.1};
188
189 PrimitivesVec_t primitives{};
190 bool drewAny{false};
191
192 // -----------------------
193 // (A) Draw TRUTH segments
194 // -----------------------
195 const xAOD::MuonSegmentContainer* truthSegs{nullptr};
196 if (!SG::get(truthSegs, m_truthSegKey, ctx).isSuccess()) {
197 THROW_EXCEPTION("Failed to fetch the truth segment container");
198 }
199 if (truthSegs) {
200 for (const xAOD::MuonSegment* seg : *truthSegs) {
201 if (!seg) continue;
202
203 // Same station-based marker, but OPEN marker to visually separate truth vs reco
204 const auto chIdx = seg->chamberIndex();
205 const int mStyleTruth = stationMarkerSyle(chIdx, /*openMarker=*/true, /*onSeed=*/true);
206
207 const Amg::Vector3D p = seg->position();
208 const double x = p.x() * inM;
209 const double y = p.y() * inM;
210 const double z = p.z() * inM;
211 const double r = std::hypot(x, y);
212 const Amg::Vector2D plotPos =
214 : Amg::Vector2D{z, r};
215 primitives.emplace_back(drawMarker(plotPos, mStyleTruth, truthColor, /*mSize=*/3));
216
217 //Plot direction of the segment as well
218 const double dir_x = seg->direction().x() * inM;
219 const double dir_y = seg->direction().y() * inM;
220 const double dir_z = seg->direction().z() * inM;
221 const Amg::Vector2D plotDir =
222 (view == DisplayView::XY) ? Amg::Vector2D{dir_x, dir_y}
223 : Amg::Vector2D{dir_z, std::hypot(dir_x, dir_y)};
224 primitives.emplace_back(drawArrow2D(plotPos, plotDir, truthColor));
225
226
227 legend.addMarker(
228 mStyleTruth,
229 std::string(Muon::MuonStationIndex::layerName(Muon::MuonStationIndex::toLayerIndex(chIdx))) + " (truth)", truthColor);
230
231 canvas->expandPad(plotPos.x() - extraMargin, plotPos.y() - extraMargin);
232 canvas->expandPad(plotPos.x() + extraMargin, plotPos.y() + extraMargin);
233 drewAny = true;
234 }
235 }
236
237 // ---------------------------------
238 // (B) Draw RECO seed-associated segs
239 // ---------------------------------
240 for (std::size_t iSeed = 0; iSeed < seeds.size(); ++iSeed) {
241 const MsTrackSeed& seed = seeds[iSeed];
242 const int color = seedColorIdx(iSeed);
243
244 std::string seedLabel = std::format("s{:d}: ", static_cast<int>(iSeed));
245
246 for (const xAOD::MuonSegment* seg : seed.segments()) {
247 if (!seg) continue;
248
249 const auto chIdx = seg->chamberIndex();
250 const int mStyle = stationMarkerSyle(chIdx, /*openMarker=*/false, /*onSeed=*/true);
251
252 const Amg::Vector3D p = seg->position();
253 const double x = p.x() * inM;
254 const double y = p.y() * inM;
255 const double z = p.z() * inM;
256 const double r = std::hypot(x, y);
257 const Amg::Vector2D plotPos =
259 : Amg::Vector2D{z, r};
260
261 primitives.emplace_back(drawMarker(plotPos, mStyle, color));
262
263 //Plot direction of the segment as well
264 const double dir_x = seg->direction().x() * inM;
265 const double dir_y = seg->direction().y() * inM;
266 const double dir_z = seg->direction().z() * inM;
267 const Amg::Vector2D plotDir =
268 (view == DisplayView::XY) ? Amg::Vector2D{dir_x, dir_y}
269 : Amg::Vector2D{dir_z, std::hypot(dir_x, dir_y)};
270 primitives.emplace_back(drawArrow2D(plotPos, plotDir, color));
271
272 std::string label = printID(*seg);
273 seedLabel += label + "_";
274
275 primitives.emplace_back(MuonValR4::drawLabel(label, plotPos.x() + 0.01, plotPos.y() + 0.01, 10, false, color));
276
277 // Marker legend: only one entry per station style (PlotLegend dedups)
278 legend.addMarker(
279 mStyle,
281
282 canvas->expandPad(plotPos.x() - extraMargin, plotPos.y() - extraMargin);
283 canvas->expandPad(plotPos.x() + extraMargin, plotPos.y() + extraMargin);
284 drewAny = true;
285 }
286 //Add text which shows segment names associated to the respective seeed
287 primitives.emplace_back(MuonValR4::drawLabel(seedLabel, 0.15, 0.9 - (0.03*iSeed), 10, true, color));
288 }
289
290 if (!drewAny) { canvas->trash(); return; }
291
292 for (auto& p : primitives) canvas->add(std::move(p));
293 legend.fillPrimitives(*canvas);
294}
#define y
#define x
#define z
SG::ReadHandleKey< xAOD::MuonSegmentContainer > m_truthSegKey
Key to the truth segment selection to draw the segment parameters.
std::string label(const std::string &format, int i)
Definition label.h:19
std::string printID(const xAOD::MuonSegment &seg)
Print the chamber ID of a segment, e.g.
std::unique_ptr< TLatex > drawLabel(const std::string &text, const double xPos, const double yPos, const double textSize=18, const bool useNDC=true, const int color=kBlack)
Create a TLatex label,.
int seedColorIdx(std::size_t iSeed)
const std::string & layerName(LayerIndex index)
convert LayerIndex into a string
LayerIndex toLayerIndex(ChIndex index)
convert ChIndex into LayerIndex
MuonSegmentContainer_v1 MuonSegmentContainer
Definition of the current "MuonSegment container version".

◆ displayTrackSeedObj()

void MuonValR4::TrackVisualizationTool::displayTrackSeedObj ( const EventContext & ctx,
const MuonR4::MsTrackSeed & seed,
const OptBoundPars_t & parsToExt,
const std::string & objName = "" ) const
finaloverridevirtual

Visualizes the measurements of the segments on the track seed together with their predicted local line parameters as an obj file.

If parameters to extrapolate are parsed, then they're extrapolated to the end of the world and the trajectory is added to the obj

Parameters
ctxEventContext to fetch the conditions data & the event information
seedMsTrack to visualize
parsToExtParameters to extrapolate on top
objNameExtra token to be added to the file name

Definition at line 448 of file TrackVisualizationTool.cxx.

451 {
453 return;
454 }
455 const ActsTrk::GeometryContext* gctx{nullptr};
456 SG::get(gctx, m_geoCtxKey, ctx).ignore();
457
458 Acts::ObjVisualization3D visualHelper{};
459 const std::string subDir = std::format("./ObjDisplays/{:}/", m_subDir.value());
460 ensureDirectory(subDir);
461 if (parsToExt.ok() && m_extrapolationTool.isEnabled()) {
462 auto stepsResult = m_extrapolationTool->propagationSteps(ctx, *parsToExt);
463 if (stepsResult.ok()) {
464 MuonValR4::drawPropagation(stepsResult->first, visualHelper);
465 } else {
466 ATH_MSG_WARNING("Failed to extrapolate the seed for visualization: " << stepsResult.error().message());
467 }
468 }
469 std::string segStr{removeNonAlphaNum(objName)};
470 for (const xAOD::MuonSegment* seg : seed.segments()) {
471 MuonValR4::drawSegmentMeasurements(*gctx,* seg, visualHelper);
472 MuonValR4::drawSegmentLine(*gctx,*seg, visualHelper);
473 segStr += std::format("_{:}", MuonR4::printID(*seg));
474 }
475
476 unsigned fileVersion{0};
477 std::string finalStr{};
478 while (finalStr.empty() || std::filesystem::exists(finalStr)) {
479 finalStr = std::format("{:}/{:}_{:}_{:}_{:}.obj", subDir,
480 m_clientToken.preFixName, ctx.eventID().event_number(), ++fileVersion, segStr);
481 }
482 visualHelper.write(finalStr);
483 ++m_objCounter;
484 }
#define ATH_MSG_WARNING(x)
Gaudi::Property< unsigned > m_canvasLimit
Maximum canvases to draw.
ToolHandle< ActsTrk::IExtrapolationTool > m_extrapolationTool
Track extrapolation tool.
std::atomic< unsigned > m_objCounter
How many obj have been produced.
Gaudi::Property< std::string > m_subDir
Define the subdirectory in which the plots shall be saved.
void ensureDirectory(const std::string &path)
Ensures that the subdirectory in the path is created.
void drawPropagation(const std::vector< Acts::detail::Step > &steps, Acts::ObjVisualization3D &visualHelper, const Acts::ViewConfig &viewConfig=Acts::s_viewLine)
Draws the recorded propagation steps as a polygon line.
std::string removeNonAlphaNum(std::string str)
Removes all non-alpha numerical characters from a string.
void drawSegmentMeasurements(const ActsTrk::GeometryContext &gctx, const xAOD::MuonSegment &segment, Acts::ObjVisualization3D &visualHelper, const Acts::ViewConfig &viewConfig=Acts::s_viewSensitive)
Draw all uncalibrated measurements associated to the segment.
void drawSegmentLine(const ActsTrk::GeometryContext &gctx, const xAOD::MuonSegment &segment, Acts::ObjVisualization3D &visualHelper, const Acts::ViewConfig &viewConfig=Acts::s_viewLine, const double standardLength=1.*Gaudi::Units::m)
Draw a segment line inside the obj file.

◆ fillTruthSeedPoints()

void MuonValR4::TrackVisualizationTool::fillTruthSeedPoints ( const EventContext & ctx,
const MuonR4::MsTrackSeeder & seeder,
const DisplayView view,
PlotLegend & legend,
Canvas_t & canvas ) const
private

Add the truth segments to the canvas.

Parameters
ctxEventContext to access store gate
seederObjConfigured instance of the track seeder which actually constructed the seeds.
viewActual view to display
legendReference to the legend object to add the marker / truth legends
canvasReference to the canvas to which the drawn markers are appended to

Definition at line 409 of file TrackVisualizationTool.cxx.

413 {
414 const xAOD::MuonSegmentContainer* truthSegs{nullptr};
415 if (!SG::get(truthSegs,m_truthSegKey, ctx).isSuccess()) {
416 THROW_EXCEPTION("Failed to fetch the truth segment container");
417 }
418 if (!truthSegs) {
419 return;
420 }
421 const ActsTrk::GeometryContext* gctx{nullptr};
422 if (!SG::get(gctx, m_geoCtxKey, ctx).isSuccess()) {
423 THROW_EXCEPTION("Failed to fetch the geometry context "<<m_geoCtxKey.fullKey());
424 }
425
426 bool addedEntry{false};
427 for (const xAOD::MuonSegment* segment: *truthSegs) {
428 const auto chIdx = segment->chamberIndex();
429 const int mStyle = stationMarkerSyle(chIdx, true, true);
430
431 using enum Location;
433 ExpandedSector sector{segment->position().phi()};
434 for (const Location loc : {Barrel, Endcap}) {
435 const Amg::Vector2D projected{seeder.expressOnCylinder(*gctx, *segment, loc, sector)};
436 if (!seeder.withinBounds(projected, loc)) {
437 continue;
438 }
439 canvas.add(drawMarker(viewVector(sector.phi(), projected, view), mStyle, truthColor, 3));
441 addedEntry = true;
442 }
443 }
444 if (addedEntry) {
445 legend.addColor(truthColor, "truth");
446 }
447 }
bool withinBounds(const Amg::Vector2D &projPos, const Location loc) const
Returns whether the expression on the cylinder is within the surface bounds.
Amg::Vector2D expressOnCylinder(const ActsTrk::GeometryContext &gctx, const xAOD::MuonSegment &segment, const Location loc, const ExpandedSector sector) const
Expresses the segment on the cylinder surface.

◆ initialize()

StatusCode MuonValR4::TrackVisualizationTool::initialize ( )
finaloverridevirtual

Definition at line 129 of file TrackVisualizationTool.cxx.

129 {
130 if (m_canvasLimit > 0) {
131 m_clientToken.canvasLimit = m_canvasLimit;
132 m_clientToken.preFixName = m_canvasPrefix;
133 m_clientToken.saveSinglePlots = m_saveSinglePDFs;
134 m_clientToken.saveSummaryPlot = m_saveSummaryPDF;
135 m_clientToken.subDirectory = m_subDir;
136 ATH_CHECK(m_visualSvc.retrieve());
137 ATH_CHECK(m_visualSvc->registerClient(m_clientToken));
138 } else {
139 m_plotsDone = true;
140 }
142 ATH_CHECK(m_geoCtxKey.initialize());
143 ATH_CHECK(m_extrapolationTool.retrieve(EnableTool{!m_extrapolationTool.empty()}));
144 return StatusCode::SUCCESS;
145 }
#define ATH_CHECK
Evaluate an expression and check for errors.
Gaudi::Property< std::string > m_canvasPrefix
Prefix of the individual canvas file names <MANDATORY>.
Gaudi::Property< bool > m_saveSinglePDFs
If set to true each canvas is saved into a dedicated pdf file.
Gaudi::Property< bool > m_saveSummaryPDF
If set to true a summary Canvas is created.

◆ viewVector()

Amg::Vector2D MuonValR4::TrackVisualizationTool::viewVector ( const double phi,
const Amg::Vector2D & posOnCylinder,
const DisplayView view )
staticprivate

Transforms the projected vector into the actual view, if it's xy then an external phi is needed to place the marker.

Parameters
phiAngle for the xy view - taken from the segment direction
posOnCylinderProjection of the segment's position onto the seed cylinder
viewActual view to display

Definition at line 119 of file TrackVisualizationTool.cxx.

121 {
122 using enum DisplayView;
123 if (view == RZ) {
124 return Amg::Vector2D{ posOnCylinder[1]*inM, posOnCylinder[0]*inM};
125 }
126 const CxxUtils::sincos phi{phiV};
127 return posOnCylinder[0] * inM * Amg::Vector2D{phi.cs, phi.sn};
128 }
DisplayView
Enumeration to toggle whether the seeds shall be painted in x-y or R-z view.

Member Data Documentation

◆ m_canvasLimit

Gaudi::Property<unsigned> MuonValR4::TrackVisualizationTool::m_canvasLimit {this, "CanvasLimit", 5000}
private

Maximum canvases to draw.

Definition at line 151 of file TrackVisualizationTool.h.

151{this, "CanvasLimit", 5000};

◆ m_canvasPrefix

Gaudi::Property<std::string> MuonValR4::TrackVisualizationTool::m_canvasPrefix {this, "CanvasPreFix", "MsTrackValid"}
private

Prefix of the individual canvas file names <MANDATORY>.

Definition at line 159 of file TrackVisualizationTool.h.

159{this, "CanvasPreFix", "MsTrackValid"};

◆ m_clientToken

IRootVisualizationService::ClientToken MuonValR4::TrackVisualizationTool::m_clientToken {}
private

Token to present to the visualization service such that the display froms this tool are grouped together.

Definition at line 145 of file TrackVisualizationTool.h.

145{};

◆ m_extrapolationTool

ToolHandle<ActsTrk::IExtrapolationTool> MuonValR4::TrackVisualizationTool::m_extrapolationTool {this, "ExtrapolationTool" ,"" }
private

Track extrapolation tool.

Definition at line 140 of file TrackVisualizationTool.h.

140{this, "ExtrapolationTool" ,"" };

◆ m_geoCtxKey

ActsTrk::GeoContextReadKey_t MuonValR4::TrackVisualizationTool::m_geoCtxKey {this, "AlignmentKey", "ActsAlignment", "cond handle key"}
private

Dependency on the geometry alignment.

Definition at line 149 of file TrackVisualizationTool.h.

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

◆ m_objCounter

std::atomic<unsigned> MuonValR4::TrackVisualizationTool::m_objCounter {0}
mutableprivate

How many obj have been produced.

Definition at line 163 of file TrackVisualizationTool.h.

163{0};

◆ m_plotsDone

std::atomic<bool> MuonValR4::TrackVisualizationTool::m_plotsDone {false}
mutableprivate

Flag toggling whether all Canvases have been exhausted.

Definition at line 161 of file TrackVisualizationTool.h.

161{false};

◆ m_saveSinglePDFs

Gaudi::Property<bool> MuonValR4::TrackVisualizationTool::m_saveSinglePDFs {this, "saveSinglePDFs", true}
private

If set to true each canvas is saved into a dedicated pdf file.

Definition at line 153 of file TrackVisualizationTool.h.

153{this, "saveSinglePDFs", true};

◆ m_saveSummaryPDF

Gaudi::Property<bool> MuonValR4::TrackVisualizationTool::m_saveSummaryPDF {this, "saveSummaryPDF", true}
private

If set to true a summary Canvas is created.

Definition at line 155 of file TrackVisualizationTool.h.

155{this, "saveSummaryPDF", true};

◆ m_subDir

Gaudi::Property<std::string> MuonValR4::TrackVisualizationTool::m_subDir {this, "outSubDir", ""}
private

Define the subdirectory in which the plots shall be saved.

Definition at line 157 of file TrackVisualizationTool.h.

157{this, "outSubDir", ""};

◆ m_truthSegKey

SG::ReadHandleKey<xAOD::MuonSegmentContainer> MuonValR4::TrackVisualizationTool::m_truthSegKey {this, "TruthSegkey", "MuonTruthSegments"}
private

Key to the truth segment selection to draw the segment parameters.

Definition at line 147 of file TrackVisualizationTool.h.

147{this, "TruthSegkey", "MuonTruthSegments"};

◆ m_visualSvc

ServiceHandle<IRootVisualizationService> MuonValR4::TrackVisualizationTool::m_visualSvc {this, "VisualSvc", "MuonValR4::RootVisualizationService"}
private

Service handle of the visualization service.

Definition at line 142 of file TrackVisualizationTool.h.

142{this, "VisualSvc", "MuonValR4::RootVisualizationService"};

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