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.
SG::ReadHandleKey< ActsTrk::GeometryContextm_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 if (!sectorMap.insideSector(segment->sector() + Acts::toUnderlying(secProj),
343 segment->position().phi())){
344 continue;
345 }
346
347 for (const Location loc : {Barrel, Endcap}) {
348 const Amg::Vector2D projPos{seeder.expressOnCylinder(*gctx, *segment, loc, secProj)};
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 double phi = seeder.projectedPhi(segment->sector(), secProj);
356 const Amg::Vector2D markerPos{viewVector(phi, projPos, view)};
357 extPrimitives.emplace_back( drawMarker(markerPos, mStyle, mColor)); //Is this necessary?
358 canvas->add( drawMarker(markerPos, mStyle, mColor));
359
360 if (view == DisplayView::XY) {
361 const double r = markerPos.mag() + extraMargin;
362 canvas->expandPad(r, r);
363 canvas->expandPad(-r, -r);
364 } else {
365 canvas->expandPad(markerPos[0] - extraMargin, markerPos[1] - extraMargin);
366 canvas->expandPad(markerPos[0] + extraMargin, markerPos[1] + extraMargin);
367 }
368 legend.addMarker(mStyle, std::format("{:}{:}", MuonStationIndex::layerName(MuonStationIndex::toLayerIndex(chIdx)),
369 isGood ? "" : " (discarded)"), mColor);
370 legend.addColor(mColor, msSector->barrel() ? "Barrel" : msSector->side() > 0 ? "Endcap A" : "Endcap C");
371 }
372 }
373 }
374 if (!drawnPoint) {
375 canvas->trash();
376 return;
377 }
378 for (const MsTrackSeed& seed : seeds) {
379 const Amg::Vector3D& pos{seed.position()};
380 canvas->add(drawMarker(viewVector(pos.phi(), Amg::Vector2D{pos.perp(), pos.z()}, view),
381 kFullDiamond, kBlack));
382 legend.addMarker(kFullDiamond, "track seed");
383 }
385 for (unsigned int s = 1; s<=16 ; ++s) {
386 if (view != DisplayView::XY) {
387 break;
388 }
389 const double phi = sectorMap.sectorPhi(s);
390 const double dPhi = sectorMap.sectorWidth(s);
391
392 const int lStyle = s%2 ? kDashed : kDotted;
393 const double r = canvas->corner(Edges::xHigh);
394 const CxxUtils::sincos phiM{phi-dPhi}, phiP{phi+dPhi};
395 const Amg::Vector2D e1{r * phiM.cs, r * phiM.sn};
396 const Amg::Vector2D e2{r * phiP.cs, r * phiP.sn};
397
398 auto theLine = std::make_unique<TLine>(0.,0., e1.x(), e1.y());
399 theLine->SetLineStyle(lStyle);
400 extPrimitives.insert(extPrimitives.begin(), std::move(theLine));
401 theLine = std::make_unique<TLine>(0.,0., e2.x(), e2.y());
402 theLine->SetLineStyle(lStyle);
403 extPrimitives.insert(extPrimitives.begin(), std::move(theLine));
404 theLine = std::make_unique<TLine>(e1.x(), e1.y(), e2.x(), e2.y());
405 theLine->SetLineStyle(lStyle);
406 extPrimitives.insert(extPrimitives.begin(), std::move(theLine));
407 }
408 legend.fillPrimitives(*canvas);
409 }
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.
SectorProjector
Enumeration to select the sector projection.
const MuonGMR4::SpectrometerSector * msSector() const
Returns the associated MS sector.
SG::ReadHandleKey< ActsTrk::GeometryContext > m_geoCtxKey
Dependency on the geometry alignment.
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...
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 455 of file TrackVisualizationTool.cxx.

458 {
460 return;
461 }
462 const ActsTrk::GeometryContext* gctx{nullptr};
463 SG::get(gctx, m_geoCtxKey, ctx).ignore();
464
465 Acts::ObjVisualization3D visualHelper{};
466 const std::string subDir = std::format("./ObjDisplays/{:}/", m_subDir.value());
467 ensureDirectory(subDir);
468 if (parsToExt.ok() && m_extrapolationTool.isEnabled()) {
469 auto stepsResult = m_extrapolationTool->propagationSteps(ctx, *parsToExt);
470 if (stepsResult.ok()) {
471 MuonValR4::drawPropagation(stepsResult->first, visualHelper);
472 } else {
473 ATH_MSG_WARNING("Failed to extrapolate the seed for visualization: " << stepsResult.error().message());
474 }
475 }
476 std::string segStr{removeNonAlphaNum(objName)};
477 for (const xAOD::MuonSegment* seg : seed.segments()) {
478 MuonValR4::drawSegmentMeasurements(*gctx,* seg, visualHelper);
479 MuonValR4::drawSegmentLine(*gctx,*seg, visualHelper);
480 segStr += std::format("_{:}", MuonR4::printID(*seg));
481 }
482
483 unsigned fileVersion{0};
484 std::string finalStr{};
485 while (finalStr.empty() || std::filesystem::exists(finalStr)) {
486 finalStr = std::format("{:}/{:}_{:}_{:}_{:}.obj", subDir,
487 m_clientToken.preFixName, ctx.eventID().event_number(), ++fileVersion, segStr);
488 }
489 visualHelper.write(finalStr);
490 ++m_objCounter;
491 }
#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 410 of file TrackVisualizationTool.cxx.

414 {
415 const xAOD::MuonSegmentContainer* truthSegs{nullptr};
416 if (!SG::get(truthSegs,m_truthSegKey, ctx).isSuccess()) {
417 THROW_EXCEPTION("Failed to fetch the truth segment container");
418 }
419 if (!truthSegs) {
420 return;
421 }
422 const ActsTrk::GeometryContext* gctx{nullptr};
423 if (!SG::get(gctx, m_geoCtxKey, ctx).isSuccess()) {
424 THROW_EXCEPTION("Failed to fetch the geometry context "<<m_geoCtxKey.fullKey());
425 }
426
427 bool addedEntry{false};
428 for (const xAOD::MuonSegment* segment: *truthSegs) {
429 const auto chIdx = segment->chamberIndex();
430 const int mStyle = stationMarkerSyle(chIdx, true, true);
431
432 using enum Location;
434 for (const auto secProj : {leftOverlap, center, rightOverlap}) {
435 if (!sectorMap.insideSector(segment->sector() + Acts::toUnderlying(secProj),
436 segment->position().phi())){
437 continue;
438 }
439 for (const Location loc : {Barrel, Endcap}) {
440 const Amg::Vector2D projected{seeder.expressOnCylinder(*gctx, *segment, loc, secProj)};
441 if (!seeder.withinBounds(projected, loc)) {
442 continue;
443 }
444 const double phi = MsTrackSeeder::projectedPhi(segment->sector(), secProj);
445 canvas.add(drawMarker(viewVector(phi, projected, view), mStyle, truthColor, 3));
447 addedEntry = true;
448 }
449 }
450 }
451 if (addedEntry) {
452 legend.addColor(truthColor, "truth");
453 }
454 }
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 SectorProjector proj) const
Expresses the segment on the cylinder surface.
static double projectedPhi(const int sector, const SectorProjector proj)
Returns the projected phi for a given sector and projector.

◆ 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

SG::ReadHandleKey<ActsTrk::GeometryContext> 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: