TracccTritonTool Class Reference

TracccTritonTool sends the cells of an event to a Triton server running traccc and returns the serialized traccc EDM collections it reconstructed. More...

#include <TracccTritonTool.h>

Inheritance diagram for TracccTritonTool:

Collaboration diagram for TracccTritonTool:

Public Member Functions
	TracccTritonTool (const std::string &type, const std::string &name, const IInterface *parent)
virtual StatusCode	initialize () override
virtual StatusCode	getTracks (const EventContext &ctx, std::vector< uint8_t > &cellBytes, TracccTritonResult &result) const override
	Reconstruct the tracks of one event on the server.

Private Attributes
ToolHandle< AthInfer::IAthInferenceTool >	m_TracccTritonTool

Detailed Description

TracccTritonTool sends the cells of an event to a Triton server running traccc and returns the serialized traccc EDM collections it reconstructed.

Definition at line 19 of file TracccTritonTool.h.

Constructor & Destructor Documentation

TracccTritonTool()

TracccTritonTool::TracccTritonTool	(	const std::string &	type,
		const std::string &	name,
		const IInterface *	parent )

Definition at line 11 of file TracccTritonTool.cxx.

    : base_class(type, name, parent) {
 
    declareInterface<ITracccTritonTool>(this);
}

Member Function Documentation

getTracks()

StatusCode TracccTritonTool::getTracks ( const EventContext & ctx, std::vector< uint8_t > & cellBytes, TracccTritonResult & result ) const

overridevirtual

Reconstruct the tracks of one event on the server.

Parameters

ctx	event context for IOV sennding
cellBytes	serialized traccc silicon_cell_collection, forwarded to the server as a single UINT8 tensor.
result	the serialized collections the server returned

Definition at line 24 of file TracccTritonTool.cxx.

        {
 
    // Forward the serialized traccc silicon_cell_collection as a single UINT8
    AthInfer::InputDataMap inputData;
    inputData["CELLS"] = std::make_pair(
        std::vector<int64_t>{static_cast<int64_t>(cellBytes.size())},
        cellBytes);
 
    // The IOV the server looks the event's conditions up for: run,
    // lumiblock, time stamp [s], time stamp ns offset, conditions run (the
    // MC override of the run number)
    const EventIDBase& eventID = ctx.eventID();
    std::vector<int64_t> iov{
        eventID.run_number(), eventID.lumi_block(), eventID.time_stamp(),
        eventID.time_stamp_ns_offset(),
        Atlas::getExtendedEventContext(ctx).conditionsRun()};
    ATH_MSG_DEBUG("Sending event IOV run " << iov[0] << " lb " << iov[1]
                  << " time stamp " << iov[2] << "." << iov[3]
                  << " conditions run " << iov[4]);
    inputData["EVENT_IOV"] = std::make_pair(
        std::vector<int64_t>{static_cast<int64_t>(iov.size())}, std::move(iov));
 
    // Every output is one serialized traccc EDM collection
    AthInfer::OutputDataMap outputData;
    for (const char* name : {"MEASUREMENTS", "CLUSTERS", "TRACKS", "TRACK_STATES"}) {
        outputData[name] = std::make_pair(
            std::vector<int64_t>{-1}, std::vector<uint8_t>{});
    }
 
    auto start_inference = std::chrono::high_resolution_clock::now();
    ATH_CHECK(m_TracccTritonTool->inference(inputData, outputData));
    auto end_inference = std::chrono::high_resolution_clock::now();
    std::chrono::duration<double, std::milli> inference_time = end_inference - start_inference;
    ATH_MSG_INFO("Traccc Triton inference time: " << inference_time.count() << " ms");
 
    auto take = [&outputData](const char* name) {
        return std::move(std::get<std::vector<uint8_t>>(outputData[name].second));
    };
    result.measurements = take("MEASUREMENTS");
    result.clusters = take("CLUSTERS");
    result.tracks = take("TRACKS");
    result.trackStates = take("TRACK_STATES");
 
    ATH_MSG_DEBUG("Received " << result.measurements.size() << " + "
                  << result.clusters.size() << " + " << result.tracks.size()
                  << " + " << result.trackStates.size()
                  << " bytes of measurements, clusters, tracks and track states");
 
    return StatusCode::SUCCESS;
}

initialize()

StatusCode TracccTritonTool::initialize ( )

overridevirtual

Definition at line 17 of file TracccTritonTool.cxx.

                                        {
 
    ATH_CHECK(m_TracccTritonTool.retrieve());
 
    return StatusCode::SUCCESS;
}

Member Data Documentation

m_TracccTritonTool

ToolHandle<AthInfer::IAthInferenceTool> TracccTritonTool::m_TracccTritonTool

private

Initial value:

{
        this, "TritonTool", "AthInfer::TritonTool"}

Definition at line 42 of file TracccTritonTool.h.

                                                            {
        this, "TritonTool", "AthInfer::TritonTool"};

---

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

• TracccTritonTool.h
• TracccTritonTool.cxx