DiTauOnnxDiscriminantTool Class Reference

#include <DiTauOnnxDiscriminantTool.h>

Inheritance diagram for DiTauOnnxDiscriminantTool:

Collaboration diagram for DiTauOnnxDiscriminantTool:

Classes
struct	DitauTrackingInfo
struct	InferenceOutput
struct	OnnxInputs
struct	SubjetTrackingInfo

Public Member Functions
	DiTauOnnxDiscriminantTool (const std::string &type, const std::string &name, const IInterface *parent)
virtual	~DiTauOnnxDiscriminantTool ()
virtual StatusCode	initialize () override
	Tool initializer.
virtual StatusCode	finalize () override
	Finalizer.
virtual StatusCode	execute (DiTauCandidateData *data, const EventContext &ctx) const override
	Execute - called for each Ditau candidate.
virtual StatusCode	executeObj (xAOD::DiTauJet &xDiTau, const EventContext &ctx) const override
	Execute - called for each Ditau jet.
float	GetDiTauObjOnnxScore (const xAOD::DiTauJet &ditau) const
virtual StatusCode	eventInitialize (DiTauCandidateData *data)
	Event initializer - called at the beginning of each event.
template<class T>
bool	retrieveTool (T &tool)
	Convenience functions to handle storegate objects.
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	sysInitialize () override
	Perform system initialization for an algorithm.
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

Static Public Member Functions
static const InterfaceID &	interfaceID ()
	InterfaceID implementation needed for ToolHandle.

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< StoreGateSvc >	StoreGateSvc_t

Private Member Functions
int	n_subjets (const xAOD::DiTauJet &xDiTau) const
float	ditau_pt (const xAOD::DiTauJet &xDiTau) const
float	f_core (const xAOD::DiTauJet &xDiTau, int iSubjet) const
float	f_subjet (const xAOD::DiTauJet &xDiTau, int iSubjet) const
float	f_subjets (const xAOD::DiTauJet &xDiTau) const
float	R_max (const xAOD::DiTauJet &xDiTau, const DitauTrackingInfo &ditauInfo, int iSubjet) const
int	n_track (const xAOD::DiTauJet &xDiTau) const
float	R_isotrack (const xAOD::DiTauJet &xDiTau, const DitauTrackingInfo &ditauInfo) const
float	R_tracks (const xAOD::DiTauJet &xDiTau, const DitauTrackingInfo &ditauInfo, int iSubjet) const
float	mass_core (const xAOD::DiTauJet &xDiTau, const DitauTrackingInfo &ditauInfo, int iSubjet) const
float	mass_tracks (const xAOD::DiTauJet &xDiTau, const DitauTrackingInfo &ditauInfo, int iSubjet) const
float	d0_leadtrack (const xAOD::DiTauJet &xDiTau, const DitauTrackingInfo &ditauInfo, int iSubjet) const
float	f_isotracks (const xAOD::DiTauJet &xDiTau, const DitauTrackingInfo &ditauInfo) const
StatusCode	getTrackingInfo (const xAOD::DiTauJet &xDiTau, DitauTrackingInfo &trackingInfo) const
Ort::Value	create_tensor (std::vector< float > &data, const std::vector< int64_t > &shape) const
InferenceOutput	run_inference (OnnxInputs &inputs) const
std::vector< float >	flatten (const std::vector< std::vector< float > > &vec_2d) const
std::vector< float >	extract_points (const std::vector< std::vector< float > > &track_features) const
std::vector< float >	create_mask (const std::vector< std::vector< float > > &track_features) const
Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
	specialization for handling Gaudi::Property<SG::VarHandleKey>

Private Attributes
float	m_dDefault = -1234
Gaudi::Property< std::string >	m_onnxModelPath {this, "onnxModelPath", "TrigTauRec/00-11-02/dev/boosted_ditau_omni_model.onnx"}
Gaudi::Property< size_t >	m_maxTracks {this, "maxTracks", 10}
std::unique_ptr< Ort::Env >	m_ort_env
std::unique_ptr< Ort::Session >	m_ort_session
const std::vector< std::string >	m_input_node_names = {"input_features", "input_points", "input_mask", "input_jet", "input_time"}
const std::vector< std::string >	m_output_node_names = {"output_1", "output_2"}
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 29 of file DiTauRec/DiTauOnnxDiscriminantTool.h.

Member Typedef Documentation

StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< AlgTool > >::StoreGateSvc_t

privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

DiTauOnnxDiscriminantTool()

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

Definition at line 20 of file src/DiTauOnnxDiscriminantTool.cxx.

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

~DiTauOnnxDiscriminantTool()

DiTauOnnxDiscriminantTool::~DiTauOnnxDiscriminantTool ( )

virtualdefault

Member Function Documentation

create_mask()

std::vector< float > DiTauOnnxDiscriminantTool::create_mask ( const std::vector< std::vector< float > > & track_features ) const

private

Definition at line 97 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                                                              {
  std::vector<float> mask;
  mask.reserve(track_features.size());
  std::transform(track_features.begin(), track_features.end(), std::back_inserter(mask), [](const auto &track) {
        return std::abs(track[2]) > 1e-6 ? 1.0f : 0.0f;
  });
  return mask;
}

create_tensor()

Ort::Value DiTauOnnxDiscriminantTool::create_tensor ( std::vector< float > & data, const std::vector< int64_t > & shape ) const

private

Definition at line 106 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                                                              {
  Ort::MemoryInfo memory_info = Ort::MemoryInfo::CreateCpu(OrtArenaAllocator, OrtMemTypeDefault);
  return Ort::Value::CreateTensor<float>(memory_info, data.data(), data.size(),shape.data(), shape.size());
}

d0_leadtrack()

float DiTauOnnxDiscriminantTool::d0_leadtrack ( const xAOD::DiTauJet & xDiTau, const DitauTrackingInfo & ditauInfo, int iSubjet ) const

private

Definition at line 336 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                                                                        {
    SubjetTrackingInfo subjetInfo = ditauInfo.vSubjetInfo.at(iSubjet);
    if (!subjetInfo.leadTrack) {
        return m_dDefault;
    }
    return subjetInfo.leadTrack->d0();
}

declareGaudiProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< AlgTool > >::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.

  {
    return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
                                                       hndl.value(), 
                                                       hndl.documentation());
 
  }

declareProperty()

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

inlineinherited

Definition at line 145 of file AthCommonDataStore.h.

                                                                     {
    typedef typename SG::HandleClassifier<T>::type htype;
    return AthCommonDataStore<PBASE>::declareGaudiProperty(t, htype());
  }

detStore()

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

inlineinherited

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

Definition at line 95 of file AthCommonDataStore.h.

{ return m_detStore; }

ditau_pt()

float DiTauOnnxDiscriminantTool::ditau_pt ( const xAOD::DiTauJet & xDiTau ) const

private

Definition at line 243 of file src/DiTauOnnxDiscriminantTool.cxx.

{
  return xDiTau.subjetPt(0)+xDiTau.subjetPt(1);
}

eventInitialize()

StatusCode DiTauToolBase::eventInitialize ( DiTauCandidateData * data )

virtualinherited

Event initializer - called at the beginning of each event.

Definition at line 32 of file DiTauToolBase.cxx.

{
  return StatusCode::SUCCESS;
}

evtStore()

ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< AlgTool > >::evtStore ( )

inlineinherited

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

Definition at line 85 of file AthCommonDataStore.h.

{ return m_evtStore; }

execute()

StatusCode DiTauOnnxDiscriminantTool::execute ( DiTauCandidateData * data, const EventContext & ctx ) const

overridevirtual

Execute - called for each Ditau candidate.

Reimplemented from DiTauToolBase.

Definition at line 57 of file src/DiTauOnnxDiscriminantTool.cxx.

{
    static const SG::Accessor<float> omni_scoreDec("omni_score");
    xAOD::DiTauJet* xDitau = data->xAODDiTau;
    ATH_MSG_DEBUG("Inferencing omni DiTau ID score...");
    float score = GetDiTauObjOnnxScore(*xDitau);
    ATH_MSG_DEBUG("DiTau ID score: " << score);
    omni_scoreDec(*xDitau) = score;
    return StatusCode::SUCCESS;
}

executeObj()

StatusCode DiTauOnnxDiscriminantTool::executeObj ( xAOD::DiTauJet & xDiTau, const EventContext & ctx ) const

overridevirtual

Execute - called for each Ditau jet.

Reimplemented from DiTauToolBase.

Definition at line 68 of file src/DiTauOnnxDiscriminantTool.cxx.

{
    static const SG::Accessor<float> omni_scoreDec("omni_score");
    ATH_MSG_DEBUG("Inferencing omni DiTau ID score...");
    float score = GetDiTauObjOnnxScore(xDiTau);
    ATH_MSG_DEBUG("DiTau ID score: " << score);
    omni_scoreDec(xDiTau) = score;
    return StatusCode::SUCCESS;
}

extract_points()

std::vector< float > DiTauOnnxDiscriminantTool::extract_points ( const std::vector< std::vector< float > > & track_features ) const

private

Definition at line 87 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                                                                 {
  std::vector<float> points;
  points.reserve(track_features.size() * 2);
  for (const auto &track : track_features) {
    points.push_back(track[0]);  // delta_eta
    points.push_back(track[1]);  // delta_phi
  }
  return points;
}

extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::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

f_core()

float DiTauOnnxDiscriminantTool::f_core ( const xAOD::DiTauJet & xDiTau, int iSubjet ) const

private

Definition at line 248 of file src/DiTauOnnxDiscriminantTool.cxx.

{
  return xDiTau.fCore(iSubjet);
}

f_isotracks()

float DiTauOnnxDiscriminantTool::f_isotracks ( const xAOD::DiTauJet & xDiTau, const DitauTrackingInfo & ditauInfo ) const

private

Definition at line 344 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                                                                 { 
    float iso_pt = 0;
    for (const xAOD::TrackParticle* xTrack: ditauInfo.vIsoTracks) {
        iso_pt += xTrack->pt();
    }
    if( xDiTau.pt() == 0.){
        return m_dDefault;
    } else {
        return iso_pt / xDiTau.pt();
    }
}

f_subjet()

float DiTauOnnxDiscriminantTool::f_subjet ( const xAOD::DiTauJet & xDiTau, int iSubjet ) const

private

Definition at line 253 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                                       {
  return xDiTau.subjetPt(iSubjet) / xDiTau.pt();
}

f_subjets()

float DiTauOnnxDiscriminantTool::f_subjets ( const xAOD::DiTauJet & xDiTau ) const

private

Definition at line 257 of file src/DiTauOnnxDiscriminantTool.cxx.

{
  return (xDiTau.subjetPt(0) + xDiTau.subjetPt(1))/ xDiTau.pt();
}

finalize()

StatusCode DiTauOnnxDiscriminantTool::finalize ( )

overridevirtual

Finalizer.

Reimplemented from DiTauToolBase.

Definition at line 49 of file src/DiTauOnnxDiscriminantTool.cxx.

{
  ATH_MSG_INFO( "Finalizing DiTauOnnxDiscriminantTool" );
  m_ort_session.reset();
  m_ort_env.reset();
  return StatusCode::SUCCESS;
}

flatten()

std::vector< float > DiTauOnnxDiscriminantTool::flatten ( const std::vector< std::vector< float > > & vec_2d ) const

private

Definition at line 78 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                                                  {
  std::vector<float> flattened;
  flattened.reserve(vec_2d.size() * (vec_2d.empty() ? 0 : vec_2d[0].size()));
  for (const auto &inner : vec_2d) {
    flattened.insert(flattened.end(), inner.begin(), inner.end());
  }
  return flattened;
}

GetDiTauObjOnnxScore()

float DiTauOnnxDiscriminantTool::GetDiTauObjOnnxScore

(

const xAOD::DiTauJet &

ditau

)

const

Definition at line 140 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                                    {
 
    // do the calculation only for ditau with at least 2 subjets
    if(n_subjets(ditau)<2){
        return m_dDefault;
    }
 
    DitauTrackingInfo ditauTrackingInfo;
    if(!(getTrackingInfo(ditau, ditauTrackingInfo))){
        return m_dDefault;
    }   
    
    // Accessors for reading the necessary features from the xAOD::TrackParticle object
    static const SG::ConstAccessor< uint8_t > numberOfInrmstPxlLyrHitsAcc ("numberOfInnermostPixelLayerHits");
    static const SG::ConstAccessor< uint8_t > numberOfPixelHitsAcc        ("numberOfPixelHits");
    static const SG::ConstAccessor< uint8_t > numberOfSCTHitsAcc          ("numberOfSCTHits");
    static const SG::ConstAccessor<   float > z0Acc                       ("z0");
    static const SG::ConstAccessor<   float > d0Acc                       ("d0");
    // Input features for Ditau tagger ONNX model
    std::vector<float> jet_vars = {
        R_max(ditau, ditauTrackingInfo, 0), 
        R_max(ditau, ditauTrackingInfo, 1), 
        R_tracks(ditau, ditauTrackingInfo, 1),  
        R_isotrack(ditau, ditauTrackingInfo), 
        d0_leadtrack(ditau, ditauTrackingInfo, 0),
        d0_leadtrack(ditau, ditauTrackingInfo, 1),  
        f_core(ditau,0),                
        f_core(ditau,1),                
        f_subjet(ditau,1),              
        f_subjets(ditau),               
        f_isotracks(ditau, ditauTrackingInfo), 
        mass_core(ditau, ditauTrackingInfo, 0), 
        mass_core(ditau, ditauTrackingInfo, 1), 
        mass_tracks(ditau, ditauTrackingInfo, 0), 
        static_cast<float>( n_track(ditau)),
    };
    std::vector<int64_t> jet_shape = {1, static_cast<int64_t>(jet_vars.size())};
 
    const TrackParticleLinks_t &vTauTracks = ditau.trackLinks();
    std::vector<std::vector<float>> track_features(m_maxTracks, std::vector<float>(11, 0.0f));
 
    float jet_eta = ditau.eta();
    float jet_phi = ditau.phi();
    size_t num_tracks = std::min(static_cast<size_t>(m_maxTracks), vTauTracks.size());
 
    for (size_t i = 0; i < num_tracks; ++i) {
        const ElementLink<xAOD::TrackParticleContainer> &trackLink = vTauTracks[i];
        if (!trackLink.isValid()) continue;
        const xAOD::TrackParticle *xTrack = *trackLink;
        float track_eta    = xTrack->eta();
        float track_phi    = xTrack->phi();
        float delta_eta    = track_eta - jet_eta;
        float delta_phi    = std::remainder(track_phi - jet_phi, 2 * M_PI);
        float delta_R      = std::hypot(delta_eta, delta_phi);
        float track_pt     = static_cast<float>(xTrack->pt());
        float pt_log       = std::log(track_pt + 1e-8f);
        float jet_pt       = ditau_pt(ditau); //ditau_ptAcc(ditau);
        float pt_ratio     = track_pt / jet_pt;
        float pt_ratio_log = (pt_ratio <= 1.0f) ? std::log(1.0f - pt_ratio + 1e-8f) : 0.0f;
        float track_charge = xTrack->charge();
 
        track_features[i] = {
            delta_eta,
            delta_phi,
            pt_log,
            d0Acc(*xTrack),
            pt_ratio_log,
            z0Acc(*xTrack),
            delta_R,
            static_cast<float>(numberOfInrmstPxlLyrHitsAcc(*xTrack)),
            static_cast<float>(numberOfPixelHitsAcc(*xTrack)),
            static_cast<float>(numberOfSCTHitsAcc(*xTrack)),
            track_charge
        };
    }
    std::vector<int64_t> track_shape = {1, static_cast<int64_t>(m_maxTracks), 11};
 
    // Actual ONNX inference
    OnnxInputs inputs{
        flatten(track_features),
        track_shape,
        extract_points(track_features),
        {1, track_shape[1], 2},
        create_mask(track_features),
        {1, track_shape[1]},
        std::move(jet_vars),
        std::move(jet_shape),
        {0.0f},
        {1, 1}
    };
    auto output = run_inference(inputs);
    return output.output_1[1];
}

getTrackingInfo()

StatusCode DiTauOnnxDiscriminantTool::getTrackingInfo ( const xAOD::DiTauJet & xDiTau, DitauTrackingInfo & trackingInfo ) const

private

Definition at line 356 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                                                                       {
    static const SG::ConstAccessor<std::vector<ElementLink<xAOD::TrackParticleContainer>>> trackLinksAcc("trackLinks");
    static const SG::ConstAccessor<std::vector<ElementLink<xAOD::TrackParticleContainer>>> isoTrackLinksAcc("isoTrackLinks");
    static const SG::ConstAccessor<float> R_subjetAcc("R_subjet");
    static const SG::ConstAccessor<float> R_coreAcc("R_core");
    
 
    if (!trackLinksAcc.isAvailable(xDiTau) || !isoTrackLinksAcc.isAvailable(xDiTau)) {
        ATH_MSG_WARNING("Track " << (!trackLinksAcc.isAvailable(xDiTau) ? "DiTauJet.trackLinks" : "DiTauJet.isoTrackLinks") << " links not available.");
        return StatusCode::FAILURE;
    } 
 
    int nSubjets = n_subjets(xDiTau);
    float Rsubjet = R_subjetAcc(xDiTau);
    float RCore = R_coreAcc(xDiTau);
 
    trackingInfo.nSubjets = nSubjets;
    trackingInfo.vSubjetInfo.clear();
    trackingInfo.vIsoTracks.clear();
    trackingInfo.vTracks.clear();
    
    // Get the track links from the DiTauJet and store them in the tracking info 
    std::vector<ElementLink<xAOD::TrackParticleContainer>> isoTrackLinks = xDiTau.isoTrackLinks();
    for (const auto &trackLink: isoTrackLinks) {
        if (!trackLink.isValid()) {
        ATH_MSG_WARNING("Iso track link is not valid");
        continue;
        }
        const xAOD::TrackParticle* xTrack = *trackLink;
        trackingInfo.vIsoTracks.push_back(xTrack);
    } 
    std::vector<ElementLink<xAOD::TrackParticleContainer>> trackLinks = xDiTau.trackLinks();
    for (const auto &trackLink : trackLinks)  {
        if (!trackLink.isValid()) {
        ATH_MSG_WARNING("track link is not valid");
        continue;
        }
        const xAOD::TrackParticle* xTrack = *trackLink;
        trackingInfo.vTracks.push_back(xTrack);
    } 
    // store subjet p4
    for (int i=0; i<nSubjets; ++i){
        SubjetTrackingInfo subjetTrackingInfo;
        TLorentzVector subjet_p4 = TLorentzVector();
        subjet_p4.SetPtEtaPhiE( xDiTau.subjetPt(i), xDiTau.subjetEta(i), xDiTau.subjetPhi(i), xDiTau.subjetE(i));
        subjetTrackingInfo.subjet_p4 = subjet_p4;
        trackingInfo.vSubjetInfo.push_back(subjetTrackingInfo);
    }
    for (const auto track : trackingInfo.vTracks) {
        float dRMin = 999;
        int inSubjet = -1;
        for (int i=0; i<nSubjets; ++i){
            float dRTrackSubjet = trackingInfo.vSubjetInfo[i].subjet_p4.DeltaR(track->p4());
            if (dRTrackSubjet < Rsubjet && dRTrackSubjet < dRMin){
                dRMin = dRTrackSubjet;
                inSubjet = i;
            }
        }
        if (inSubjet >= 0){
            trackingInfo.vSubjetInfo[inSubjet].vTracks.push_back(track);
        }
    }
    // find leading track in subjets
    for (int i=0; i<nSubjets; ++i){
        float ptLeadTrack = 0;
        for (const auto track : trackingInfo.vSubjetInfo[i].vTracks){
            if (track->pt() > ptLeadTrack){
                ptLeadTrack = track->pt();
                trackingInfo.vSubjetInfo[i].leadTrack = track;
            }
        }
    }
    // find core track in subjets
    for (int i=0; i<nSubjets; ++i){
        for (const auto track : trackingInfo.vSubjetInfo[i].vTracks){
            auto subjetTrackingInfo = trackingInfo.vSubjetInfo[i];
            if (subjetTrackingInfo.subjet_p4.DeltaR(track->p4()) < RCore){
                trackingInfo.vSubjetInfo[i].vCoreTracks.push_back(track);
            }
        }
    }
    //find isotracks in subjets
    for (const auto track : trackingInfo.vIsoTracks){
        float RIso = 0.4;
        float dRMin = 999;
        int inSubjet = -1;
        for (int i=0; i<nSubjets; ++i){
            float dRTrackSubjet = trackingInfo.vSubjetInfo[i].subjet_p4.DeltaR(track->p4());
            if (dRTrackSubjet > Rsubjet && dRTrackSubjet < RIso && dRTrackSubjet < dRMin){
                dRMin = dRTrackSubjet;
                inSubjet = i;
            }
        }
        if (inSubjet >= 0){
            trackingInfo.vSubjetInfo[inSubjet].vIsoTracks.push_back(track);
        }
    }
    return StatusCode::SUCCESS;
}

initialize()

StatusCode DiTauOnnxDiscriminantTool::initialize ( )

overridevirtual

Tool initializer.

Reimplemented from DiTauToolBase.

Definition at line 30 of file src/DiTauOnnxDiscriminantTool.cxx.

{
  ATH_MSG_INFO( "Initializing DiTauOnnxDiscriminantTool" );
  ATH_MSG_INFO( "onnxModelPath: " << m_onnxModelPath );
 
  auto model_path = PathResolverFindCalibFile (m_onnxModelPath);
  if (model_path.empty()) {
    ATH_MSG_ERROR("Could not find model file: " << m_onnxModelPath);
    return StatusCode::FAILURE;
  }
  m_ort_env = std::make_unique<Ort::Env>(ORT_LOGGING_LEVEL_WARNING, "OnnxUtil");
  Ort::SessionOptions session_options;
  session_options.SetIntraOpNumThreads(1);
  session_options.SetGraphOptimizationLevel(GraphOptimizationLevel::ORT_ENABLE_EXTENDED);
  session_options.DisableCpuMemArena();
  m_ort_session = std::make_unique<Ort::Session>(*m_ort_env, model_path.c_str(), session_options);
  return StatusCode::SUCCESS;
}

inputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< AlgTool > >::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.

interfaceID()

const InterfaceID & DiTauToolBase::interfaceID ( )

staticinherited

InterfaceID implementation needed for ToolHandle.

Definition at line 9 of file DiTauToolBase.cxx.

                                              {
  return DiTauToolBaseID;
}

mass_core()

float DiTauOnnxDiscriminantTool::mass_core ( const xAOD::DiTauJet & xDiTau, const DitauTrackingInfo & ditauInfo, int iSubjet ) const

private

Definition at line 310 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                                                                     {
    TLorentzVector allCoreTracks_p4;
    SubjetTrackingInfo subjetInfo = ditauInfo.vSubjetInfo.at(iSubjet);
    for (const xAOD::TrackParticle* xTrack: subjetInfo.vCoreTracks) {
        allCoreTracks_p4 += xTrack->p4();
    }
    float mass = allCoreTracks_p4.M();
    if (mass < 0) {
        return m_dDefault;
    }
    return mass;
}

mass_tracks()

float DiTauOnnxDiscriminantTool::mass_tracks ( const xAOD::DiTauJet & xDiTau, const DitauTrackingInfo & ditauInfo, int iSubjet ) const

private

Definition at line 323 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                                                                       {
    TLorentzVector allTracks_p4;
    SubjetTrackingInfo subjetInfo = ditauInfo.vSubjetInfo.at(iSubjet);
    for (const xAOD::TrackParticle* xTrack: subjetInfo.vTracks) {
        allTracks_p4 += xTrack->p4();
    }
    float mass = allTracks_p4.M();
    if (mass < 0) {
        return m_dDefault;
    }
    return mass;
}

msg()

MsgStream & AthCommonMsg< AlgTool >::msg ( ) const

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                {
    return this->msgStream();
  }

msgLvl()

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

inlineinherited

Definition at line 30 of file AthCommonMsg.h.

                                               {
    return this->msgLevel(lvl);
  }

n_subjets()

int DiTauOnnxDiscriminantTool::n_subjets ( const xAOD::DiTauJet & xDiTau ) const

private

Definition at line 235 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                         {
    int nSubjet = 0;
    while (xDiTau.subjetPt(nSubjet) > 0. ){
        nSubjet++;
    }
    return nSubjet;
}

n_track()

int DiTauOnnxDiscriminantTool::n_track ( const xAOD::DiTauJet & xDiTau ) const

private

Definition at line 274 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                      {
    return xDiTau.nTracks();
}

outputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< AlgTool > >::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.

R_isotrack()

float DiTauOnnxDiscriminantTool::R_isotrack ( const xAOD::DiTauJet & xDiTau, const DitauTrackingInfo & ditauInfo ) const

private

Definition at line 278 of file src/DiTauOnnxDiscriminantTool.cxx.

{
    float R_sum = 0;
    float pt = 0;
    for (int i = 0; i < 2; i++) {
        SubjetTrackingInfo subjetInfo = ditauInfo.vSubjetInfo.at(i);
        for (const xAOD::TrackParticle* xTrack: subjetInfo.vIsoTracks) {
            R_sum += subjetInfo.subjet_p4.DeltaR(xTrack->p4()) * xTrack->pt();
            pt += xTrack->pt();
        }
    }
    if (pt == 0) {
        return m_dDefault;
    }
    return R_sum / pt;
}

R_max()

float DiTauOnnxDiscriminantTool::R_max ( const xAOD::DiTauJet & xDiTau, const DitauTrackingInfo & ditauInfo, int iSubjet ) const

private

Definition at line 262 of file src/DiTauOnnxDiscriminantTool.cxx.

{
    const SubjetTrackingInfo subjetInfo = ditauInfo.vSubjetInfo.at(iSubjet);
    float Rmax = 0;
    for (const xAOD::TrackParticle* xTrack: subjetInfo.vTracks) {
        if (subjetInfo.subjet_p4.DeltaR(xTrack->p4()) > Rmax) {
            Rmax = subjetInfo.subjet_p4.DeltaR(xTrack->p4());
        }
    }
    return Rmax;
}

R_tracks()

float DiTauOnnxDiscriminantTool::R_tracks ( const xAOD::DiTauJet & xDiTau, const DitauTrackingInfo & ditauInfo, int iSubjet ) const

private

Definition at line 295 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                                                                    { 
    float R_sum = 0;
    float pt = 0;
 
    SubjetTrackingInfo subjetInfo = ditauInfo.vSubjetInfo.at(iSubjet);
    for (const xAOD::TrackParticle* xTrack: subjetInfo.vTracks) {
        R_sum += subjetInfo.subjet_p4.DeltaR(xTrack->p4()) * xTrack->pt();
        pt += xTrack->pt();
    }
    if (pt == 0) {
        return m_dDefault;
    }
    return R_sum / pt;
}

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< AlgTool > >::renounce ( T & h )

inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

  {
    h.renounce();
    PBASE::renounce (h);
  }

renounceArray()

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

inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

                                                          {
    handlesArray.renounce();
  }

retrieveTool()

template<class T>

bool DiTauToolBase::retrieveTool ( T & tool )

inherited

Convenience functions to handle storegate objects.

Definition at line 59 of file DiTauToolBase.cxx.

                                         {
  if (tool.retrieve().isFailure()) {
    ATH_MSG_FATAL("Failed to retrieve tool " << tool);
    return false;
  } else {
    ATH_MSG_VERBOSE("Retrieved tool " << tool);
  }
  return true;
}

run_inference()

DiTauOnnxDiscriminantTool::InferenceOutput DiTauOnnxDiscriminantTool::run_inference ( OnnxInputs & inputs ) const

private

Definition at line 111 of file src/DiTauOnnxDiscriminantTool.cxx.

                                                                                                         {
  std::vector<Ort::Value> input_tensors;
  input_tensors.reserve(m_input_node_names.size());
  input_tensors.emplace_back(create_tensor(inputs.input_features, inputs.input_features_shape));
  input_tensors.emplace_back(create_tensor(inputs.input_points, inputs.input_points_shape));
  input_tensors.emplace_back(create_tensor(inputs.input_mask, inputs.input_mask_shape));
  input_tensors.emplace_back(create_tensor(inputs.input_jet, inputs.input_jet_shape));
  input_tensors.emplace_back(create_tensor(inputs.input_time, inputs.input_time_shape));
 
  std::vector<const char *> input_node_names;
  input_node_names.reserve(m_input_node_names.size());
  std::transform(m_input_node_names.begin(), m_input_node_names.end(), std::back_inserter(input_node_names), [](const std::string &name) { return name.c_str(); });
 
  std::vector<const char *> output_node_names;
  output_node_names.reserve(m_output_node_names.size());
  std::transform(m_output_node_names.begin(), m_output_node_names.end(), std::back_inserter(output_node_names), [](const std::string &name) { return name.c_str(); });
 
  auto output_tensors = m_ort_session->Run(Ort::RunOptions{nullptr}, input_node_names.data(), input_tensors.data(), input_node_names.size(), output_node_names.data(), output_node_names.size());
 
  InferenceOutput output;
  for (size_t i = 0; i < output_tensors.size(); ++i) {
    const auto &tensor = output_tensors[i];
    const size_t length = tensor.GetTensorTypeAndShapeInfo().GetElementCount();
    const float *data = tensor.GetTensorData<float>();
    (i == 0 ? output.output_1 : output.output_2) = std::vector<float>(data, data + length);
  } 
  return output;
}

sysInitialize()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysInitialize ( )

overridevirtualinherited

Perform system initialization for an algorithm.

We override this to declare all the elements of handle key arrays at the end of initialization. See comments on updateVHKA.

Reimplemented in asg::AsgMetadataTool, AthCheckedComponent< AthAlgTool >, AthCheckedComponent<::AthAlgTool >, and DerivationFramework::CfAthAlgTool.

sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::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< AlgTool > >::updateVHKA ( Gaudi::Details::PropertyBase & )

inlineinherited

Definition at line 308 of file AthCommonDataStore.h.

                                               {
    // debug() << "updateVHKA for property " << p.name() << " " << p.toString() 
    //         << "  size: " << m_vhka.size() << endmsg;
    for (auto &a : m_vhka) {
      std::vector<SG::VarHandleKey*> keys = a->keys();
      for (auto k : keys) {
        k->setOwner(this);
      }
    }
  }

Member Data Documentation

m_dDefault

float DiTauOnnxDiscriminantTool::m_dDefault = -1234

private

Definition at line 55 of file DiTauRec/DiTauOnnxDiscriminantTool.h.

m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_detStore

privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_evtStore

privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

m_input_node_names

const std::vector<std::string> DiTauOnnxDiscriminantTool::m_input_node_names = {"input_features", "input_points", "input_mask", "input_jet", "input_time"}

private

Definition at line 92 of file DiTauRec/DiTauOnnxDiscriminantTool.h.

{"input_features", "input_points", "input_mask", "input_jet", "input_time"};

m_maxTracks

Gaudi::Property<size_t> DiTauOnnxDiscriminantTool::m_maxTracks {this, "maxTracks", 10}

private

Definition at line 88 of file DiTauRec/DiTauOnnxDiscriminantTool.h.

{this, "maxTracks", 10};

m_onnxModelPath

Gaudi::Property<std::string> DiTauOnnxDiscriminantTool::m_onnxModelPath {this, "onnxModelPath", "TrigTauRec/00-11-02/dev/boosted_ditau_omni_model.onnx"}

private

Definition at line 87 of file DiTauRec/DiTauOnnxDiscriminantTool.h.

{this, "onnxModelPath", "TrigTauRec/00-11-02/dev/boosted_ditau_omni_model.onnx"};

m_ort_env

std::unique_ptr<Ort::Env> DiTauOnnxDiscriminantTool::m_ort_env

private

Definition at line 90 of file DiTauRec/DiTauOnnxDiscriminantTool.h.

m_ort_session

std::unique_ptr<Ort::Session> DiTauOnnxDiscriminantTool::m_ort_session

private

Definition at line 91 of file DiTauRec/DiTauOnnxDiscriminantTool.h.

m_output_node_names

const std::vector<std::string> DiTauOnnxDiscriminantTool::m_output_node_names = {"output_1", "output_2"}

private

Definition at line 93 of file DiTauRec/DiTauOnnxDiscriminantTool.h.

{"output_1", "output_2"};

m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< AlgTool > >::m_varHandleArraysDeclared

privateinherited

Definition at line 399 of file AthCommonDataStore.h.

m_vhka

std::vector<SG::VarHandleKeyArray*> AthCommonDataStore< AthCommonMsg< AlgTool > >::m_vhka

privateinherited

Definition at line 398 of file AthCommonDataStore.h.

---

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

• DiTauRec/DiTauOnnxDiscriminantTool.h
• src/DiTauOnnxDiscriminantTool.cxx
• ools/Root/DiTauOnnxDiscriminantTool.cxx