Prompt::DecoratePromptLeptonRNN Class Reference

#include <DecoratePromptLeptonRNN.h>

Inheritance diagram for Prompt::DecoratePromptLeptonRNN:

Public Types
using	decoratorFloatH_t = SG::WriteDecorHandle<xAOD::IParticleContainer, float>

Public Member Functions
	DecoratePromptLeptonRNN (const std::string &name, ISvcLocator *pSvcLocator)
virtual StatusCode	initialize () override
virtual StatusCode	execute () override
virtual StatusCode	finalize () override
const xAOD::TrackParticle *	findMuonTrack (const xAOD::Muon *muon)
const xAOD::Jet *	findClosestTrackJet (const xAOD::TrackParticle *particle, const xAOD::JetContainer &trackJets)
bool	compDummy (const xAOD::IParticle &particle, std::vector< decoratorFloatH_t > &decors) const
bool	prepTrackObject (Prompt::VarHolder &p, const xAOD::TrackParticle &track, const xAOD::TrackParticle &lepton, const xAOD::Jet &trackJet, const xAOD::Vertex &priVtx, const xAOD::EventInfo event)
bool	compScore (const xAOD::IParticle &particle, const std::vector< Prompt::VarHolder > &tracks, std::vector< decoratorFloatH_t > &decors)
bool	passTrack (Prompt::VarHolder &p)
StatusCode	makeHist (TH1 *&h, const std::string &key, int nbin, double xmin, double xmax)
virtual StatusCode	sysInitialize () override
	Override sysInitialize.
virtual const DataObjIDColl &	extraOutputDeps () const override
	Return the list of extra output dependencies.
ServiceHandle< StoreGateSvc > &	evtStore ()
	The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.
const ServiceHandle< StoreGateSvc > &	detStore () const
	The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.
virtual StatusCode	sysStart () override
	Handle START transition.
virtual std::vector< Gaudi::DataHandle * >	inputHandles () const override
	Return this algorithm's input handles.
virtual std::vector< Gaudi::DataHandle * >	outputHandles () const override
	Return this algorithm's output handles.
Gaudi::Details::PropertyBase &	declareProperty (Gaudi::Property< T, V, H > &t)
void	updateVHKA (Gaudi::Details::PropertyBase &)
MsgStream &	msg () const
bool	msgLvl (const MSG::Level lvl) const

Public Attributes
Gaudi::Property< std::string >	m_outputStream {this, "outputStream", "", "Path of the ROOT output directory of the histograms for RNN debug"}
Gaudi::Property< std::string >	m_decorationPrefixRNN {this, "decorationPrefixRNN", "", "Prefix of the name for the decorator of RNN to the lepton"}
Gaudi::Property< double >	m_minTrackpT {this, "minTrackpT", 500.0, "pT cut config for the input tracks to RNN"}
Gaudi::Property< double >	m_maxTrackEta {this, "maxTrackEta", 2.5, "abs(eta) cut config for the input tracks to RNN"}
Gaudi::Property< double >	m_maxTrackZ0Sin {this, "maxTrackZ0Sin", 1.0, "Z0sin cut config for the input tracks to RNN"}
Gaudi::Property< double >	m_minTrackLeptonDR {this, "minTrackLeptonDR", 1.0e-6, "Delta R between lepton and track cut config for the input tracks to RNN"}
Gaudi::Property< double >	m_maxTrackLeptonDR {this, "maxTrackLeptonDR", 0.4, "Delta R between lepton and track cut config for the input tracks to RNN"}
Gaudi::Property< double >	m_maxLepTrackJetDR {this, "maxLepTrackJetDR", 0.4, "Maximum distance between lepton and track jet for track jet matching"}
Gaudi::Property< double >	m_maxTrackSharedSiHits {this, "maxTrackSharedSiHits", 1.5, "track shared si hits cut config for the input tracks to RNN"}
Gaudi::Property< unsigned >	m_minTrackSiHits {this, "minTrackSiHits", 6.5, "track silicon detector hits cut config for the input tracks to RNN"}
Gaudi::Property< unsigned >	m_maxTrackSiHoles {this, "maxTrackSiHoles", 2.5, "track holes cut config for the input tracks to RNN"}
Gaudi::Property< unsigned >	m_maxTrackPixHoles {this, "maxTrackPixHoles", 1.5, "track pixel holes cut config for the input tracks to RNN"}
Gaudi::Property< bool >	m_debug {this, "debug", false, "debug statement"}
Gaudi::Property< bool >	m_printTime {this, "printTime", false, "print running time, for debug"}
ToolHandle< IRNNTool >	m_toolRNN
ServiceHandle< ITHistSvc >	m_histSvc
SG::ReadHandleKey< xAOD::IParticleContainer >	m_inputContainerLeptonKey
SG::ReadHandleKey< xAOD::TrackParticleContainer >	m_inputContainerTrackKey
SG::ReadHandleKey< xAOD::JetContainer >	m_inputContainerTrackJetKey
SG::ReadHandleKey< xAOD::VertexContainer >	m_inputContainerPrimaryVerticesKey
SG::ReadHandleKey< xAOD::EventInfo >	m_eventHandleKey
TStopwatch	m_timerEvent
int	m_countEvent
std::map< std::string, TH1 * >	m_hists
std::unique_ptr< SG::AuxElement::ConstAccessor< unsigned char > >	m_accessQuality
SG::WriteDecorHandleKeyArray< xAOD::IParticleContainer >	m_decorHandleKeys { this, "DecorHandleKeys", {} }
std::unordered_map< std::string, size_t >	m_decorNameMap

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
Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
	specialization for handling Gaudi::Property<SG::VarHandleKey>

Private Attributes
DataObjIDColl	m_extendedExtraObjects
StoreGateSvc_t	m_evtStore
	Pointer to StoreGate (event store by default)
StoreGateSvc_t	m_detStore
	Pointer to StoreGate (detector store by default)
std::vector< SG::VarHandleKeyArray * >	m_vhka
bool	m_varHandleArraysDeclared

Detailed Description

Definition at line 50 of file DecoratePromptLeptonRNN.h.

Member Typedef Documentation

decoratorFloatH_t

using Prompt::DecoratePromptLeptonRNN::decoratorFloatH_t = SG::WriteDecorHandle<xAOD::IParticleContainer, float>

Definition at line 66 of file DecoratePromptLeptonRNN.h.

StoreGateSvc_t

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

privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

DecoratePromptLeptonRNN()

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

Definition at line 21 of file DecoratePromptLeptonRNN.cxx.

                                                                                                     :
  AthAlgorithm(name, pSvcLocator),
  m_histSvc   ("THistSvc/THistSvc", name),
  m_countEvent(0)
{}

Member Function Documentation

compDummy()

bool Prompt::DecoratePromptLeptonRNN::compDummy	(	const xAOD::IParticle &	particle,
		std::vector< decoratorFloatH_t > &	decors ) const

Definition at line 436 of file DecoratePromptLeptonRNN.cxx.

{
  //
  // Fill dummy values for RNN outputs
  //
  for (auto& d : decors) {
    d(particle) = -1.0;
  }
 
  return true;
}

compScore()

bool Prompt::DecoratePromptLeptonRNN::compScore	(	const xAOD::IParticle &	particle,
		const std::vector< Prompt::VarHolder > &	tracks,
		std::vector< decoratorFloatH_t > &	decors )

Definition at line 375 of file DecoratePromptLeptonRNN.cxx.

{
  //
  // Call the RNN tool to get the RNN prediction for the leptons and decorate the lepton with those RNN scores.
  //
  ATH_MSG_DEBUG("compScore - number of tracks: " << tracks.size());
 
  for(const Prompt::VarHolder &o: tracks) {
    ATH_MSG_DEBUG("compScore - track: LepTrackDR = " << o.getVar(Def::LepTrackDR)
      << ", TrackJetDR = " << o.getVar(Def::TrackJetDR)
      << ", D0Sig = " << o.getVar(Def::D0Sig)
      << ", Z0Sin = " << o.getVar(Def::Z0Sin)
      << ", NumberOfPIXHits = " << o.getVar(Def::NumberOfPIXHits)
      << ", NumberOfSCTHits = " << o.getVar(Def::NumberOfSCTHits)
      << ", PtFrac = " << o.getVar(Def::TrackPtOverTrackJetPt) );
  }
 
  const std::map<std::string, double> results = m_toolRNN->computeRNNOutput(tracks);
 
  for(const std::pair<const std::string, double>& v: results) {
    //
    // Decorate muon
    //
 
    ATH_MSG_DEBUG("DecoratePromptLeptonRNN compScore - " << v.first << " = " << v.second );
 
    auto dit = m_decorNameMap.find (v.first);
 
    if(dit != m_decorNameMap.end()) {
      decors.at(dit->second)(particle) = v.second;
    }
    else {
      ATH_MSG_WARNING("CompScore - unknown output label=\"" << v.first << "\"");
    }
 
    if(m_debug) {
      std::map<std::string, TH1*>::iterator hit = m_hists.find(v.first);
 
      if(hit == m_hists.end()) {
        TH1* h = 0;
 
        StatusCode hist_status = makeHist(h, v.first, 100, 0.0, 1.0);
        if (hist_status != StatusCode::SUCCESS){
          ATH_MSG_WARNING("DecoratePromptLeptonRNN compScore - failed to make hist");
        }
 
        hit = m_hists.insert(std::map<std::string, TH1*>::value_type(v.first, h)).first;
      }
 
      if(hit->second) {
        hit->second->Fill(v.second);
      }
    }
  }
 
  return true;
}

declareGaudiProperty()

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

inlineprivateinherited

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

Definition at line 156 of file AthCommonDataStore.h.

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

declareProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< Algorithm > >::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< Algorithm > >::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; }

evtStore()

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

inlineinherited

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

Definition at line 85 of file AthCommonDataStore.h.

{ return m_evtStore; }

execute()

StatusCode Prompt::DecoratePromptLeptonRNN::execute ( )

overridevirtual

Definition at line 72 of file DecoratePromptLeptonRNN.cxx.

{
  //
  // Process current event
  //
  TimerScopeHelper timer(m_timerEvent);
 
  const EventContext& ctx = Gaudi::Hive::currentContext();
 
  ATH_MSG_DEBUG("execute() - begin...");
 
  m_countEvent++;
 
  //
  // Retrieve object containers and event info
  //
  SG::ReadHandle<xAOD::IParticleContainer> h_leptons (m_inputContainerLeptonKey, ctx);
  SG::ReadHandle<xAOD::TrackParticleContainer> h_tracks (m_inputContainerTrackKey, ctx);
  SG::ReadHandle<xAOD::JetContainer> h_trackJets (m_inputContainerTrackJetKey, ctx);
  SG::ReadHandle<xAOD::VertexContainer> h_vertices  (m_inputContainerPrimaryVerticesKey, ctx);
 
  SG::ReadHandle<xAOD::EventInfo> event_handle (m_eventHandleKey, ctx);
 
  ATH_MSG_DEBUG("Size of LeptonContainer: " << h_leptons->size());
 
  std::vector<decoratorFloatH_t> decors;
  decors.reserve (m_decorHandleKeys.size());
  for (const SG::WriteDecorHandleKey<xAOD::IParticleContainer>& k : m_decorHandleKeys) {
    decors.emplace_back (k, ctx);
  }
 
  //
  // Find default Primary Vertex
  //
  const xAOD::Vertex *primaryVertex = nullptr;
 
  for(const xAOD::Vertex *vertex: *h_vertices) {
    if(vertex->vertexType() == xAOD::VxType::PriVtx) {
      primaryVertex = vertex;
      break;
    }
  }
 
  //
  // Collect tracks
  //
  for(const xAOD::IParticle *lepton: *h_leptons) {
    //
    // Select lepton track
    //
    const xAOD::TrackParticle *trackLep = nullptr;
    const xAOD::Electron      *elec     = dynamic_cast<const xAOD::Electron*>(lepton);
    const xAOD::Muon          *muon     = dynamic_cast<const xAOD::Muon    *>(lepton);
 
    if(elec) {
      const xAOD::TrackParticle *bestmatchedGSFElTrack = elec->trackParticle(0);
      if(bestmatchedGSFElTrack) {
        trackLep = xAOD::EgammaHelpers::getOriginalTrackParticleFromGSF(bestmatchedGSFElTrack);
      }
    }
    else if (muon) {
      trackLep = findMuonTrack(muon);
    }
    else {
      ATH_MSG_WARNING("execute - failed to find electron or muon: should never happen!");
    }
 
    //
    // Find closest track jet
    //
    const xAOD::Jet *trackJet = findClosestTrackJet(trackLep, *h_trackJets);
 
    if(!trackLep || !trackJet) {
      compDummy(*lepton, decors);
      continue;
    }
 
    //
    // Select tracks within cone around lepton track.
    //
    std::vector<Prompt::VarHolder > select_tracks;
 
    Prompt::VarHolder lepton_obj;
 
    if(!prepTrackObject(lepton_obj, *trackLep, *trackLep, *trackJet, *primaryVertex, *event_handle)) {
      continue;
    }
 
    //
    // Add lepton track as one of the cone tracks.
    //
    select_tracks.push_back(lepton_obj);
 
    for(const xAOD::TrackParticle *track: *h_tracks) {
 
      if(!track) {
        ATH_MSG_WARNING("Prompt::DecoratePromptLeptonRNN::execute - skip null track pointer - should never happen");
        continue;
      }
 
      Prompt::VarHolder track_obj;
 
      if(!prepTrackObject(track_obj, *track, *trackLep, *trackJet, *primaryVertex, *event_handle)) {
  continue;
      }
 
      if(passTrack(track_obj)) {
  select_tracks.push_back(track_obj);
 
  ATH_MSG_DEBUG("Prompt::DecoratePromptLeptonRNN::execute - passed track pT= " << track->pt());
      }
    }
 
    //
    // Sort tracks by DR distance to lepton
    //
    std::sort(select_tracks.begin(), select_tracks.end(), Prompt::SortObjectByVar(Def::LepTrackDR, msg()));
 
    //
    // Compute RNN
    //
    compScore(*lepton, select_tracks, decors);
 
    ATH_MSG_DEBUG("DecoratePromptLeptonRNN::CompScore - " << std::endl
                    << "lepton pT= " << lepton->pt()
        << ", number of tracks: " << select_tracks.size());
  }
 
  return StatusCode::SUCCESS;
}

extraDeps_update_handler()

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

protectedinherited

Add StoreName to extra input/output deps as needed.

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

extraOutputDeps()

const DataObjIDColl & AthAlgorithm::extraOutputDeps ( ) const

overridevirtualinherited

Return the list of extra output dependencies.

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

Definition at line 50 of file AthAlgorithm.cxx.

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

finalize()

StatusCode Prompt::DecoratePromptLeptonRNN::finalize ( )

overridevirtual

Definition at line 204 of file DecoratePromptLeptonRNN.cxx.

{
  //
  // Finalize output
  //
  if(m_printTime) {
    //
    // Print full time stopwatch
    //
    ATH_MSG_INFO("DecoratePromptLeptonRNN - total time:   " << PrintResetStopWatch(m_timerEvent));
    ATH_MSG_INFO("DecoratePromptLeptonRNN - processed "     << m_countEvent << " events.");
  }
 
  return StatusCode::SUCCESS;
}

findClosestTrackJet()

const xAOD::Jet * Prompt::DecoratePromptLeptonRNN::findClosestTrackJet	(	const xAOD::TrackParticle *	particle,
		const xAOD::JetContainer &	trackJets )

Definition at line 240 of file DecoratePromptLeptonRNN.cxx.

{
  //
  // Find track jet closest to IParticle and within fixed cone
  //
  if(!particle) {
    return 0;
  }
 
  const xAOD::Jet *trackJet = 0;
  double currTrackJetDR = -1.0;
 
  //
  // Select track jet within cone around muon
  //
  for(const xAOD::Jet *jet: trackJets) {
    const double dr = particle->p4().DeltaR(jet->p4());
 
    if(currTrackJetDR < 0.0 || dr < currTrackJetDR) {
      trackJet       = jet;
      currTrackJetDR = dr;
    }
  }
 
  if(trackJet && currTrackJetDR < m_maxLepTrackJetDR) {
    return trackJet;
  }
 
  return 0;
}

findMuonTrack()

const xAOD::TrackParticle * Prompt::DecoratePromptLeptonRNN::findMuonTrack

(

const xAOD::Muon *

muon

)

Definition at line 221 of file DecoratePromptLeptonRNN.cxx.

{
  //
  // Process muon - return true if all information present for RNN
  //
  if(muon->muonType() != xAOD::Muon::Combined || !muon->inDetTrackParticleLink().isValid()) {
    return 0;
  }
 
  const unsigned char quality = (*m_accessQuality)(*muon);
 
  ATH_MSG_DEBUG("muon pT=" << muon->pt() << " quality=" << int(quality) << " medium=" << int(xAOD::Muon::Medium));
 
  const xAOD::TrackParticle *trackLep = *(muon->inDetTrackParticleLink());
 
  return trackLep;
}

initialize()

StatusCode Prompt::DecoratePromptLeptonRNN::initialize ( )

overridevirtual

Definition at line 28 of file DecoratePromptLeptonRNN.cxx.

{
  ATH_MSG_DEBUG("Initializing DecoratePromptLeptonRNN...");
 
  // Initialize read/write handles
  ATH_CHECK(m_inputContainerLeptonKey.initialize());
  ATH_CHECK(m_inputContainerTrackKey.initialize());
  ATH_CHECK(m_inputContainerTrackJetKey.initialize());
  ATH_CHECK(m_inputContainerPrimaryVerticesKey.initialize());
 
  ATH_CHECK(m_eventHandleKey.initialize());
 
  //
  // Initialize tools and services
  //
  ATH_CHECK( m_histSvc.retrieve() );
 
  m_decorHandleKeys.clear();
  for(const std::string &label: m_toolRNN->getOutputLabels()) {
    const std::string key = m_decorationPrefixRNN + label;
 
    ATH_MSG_DEBUG("Add output RNN label: \"" << key << "\"");
 
    m_decorNameMap.try_emplace (label, m_decorHandleKeys.size());
    m_decorHandleKeys.push_back (SG::makeContDecorKey (m_inputContainerLeptonKey, key));
  }
  ATH_CHECK(m_decorHandleKeys.initialize());
 
  ATH_MSG_DEBUG("inputContainerMuon=\"" << m_inputContainerLeptonKey << "\"");
 
  //
  // Instantiate Muon quality accessors
  //
  m_accessQuality = std::make_unique<SG::AuxElement::ConstAccessor<unsigned char> >("quality");
 
  m_timerEvent.Reset();
 
  ATH_MSG_DEBUG("DecoratePromptLeptonRNN initialized successfully.");
 
  return StatusCode::SUCCESS;
 
}

inputHandles()

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

overridevirtualinherited

Return this algorithm's input handles.

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

makeHist()

StatusCode Prompt::DecoratePromptLeptonRNN::makeHist	(	TH1 *&	h,
		const std::string &	key,
		int	nbin,
		double	xmin,
		double	xmax )

Definition at line 450 of file DecoratePromptLeptonRNN.cxx.

{
  //
  // Initiliase histogram pointer. If configured to run in validation mode, then create and register histogram
  //
  h = 0;
 
  if(m_outputStream.empty() || key.empty()) {
    return StatusCode::SUCCESS;
  }
 
  const std::string hname    = name() + "_" + key;
  const std::string hist_key = "/"+m_outputStream+"/"+hname;
 
  h = new TH1D(hname.c_str(), hname.c_str(), nbin, xmin, xmax);
  h->SetDirectory(0);
 
  return m_histSvc->regHist(hist_key, h);
}

msg()

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

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                {
    return this->msgStream();
  }

msgLvl()

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

inlineinherited

Definition at line 30 of file AthCommonMsg.h.

                                               {
    return this->msgLevel(lvl);
  }

outputHandles()

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

overridevirtualinherited

Return this algorithm's output handles.

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

passTrack()

bool Prompt::DecoratePromptLeptonRNN::passTrack

(

Prompt::VarHolder &

p

)

Definition at line 348 of file DecoratePromptLeptonRNN.cxx.

{
  //
  // Select cone tracks
  //
  if(p.getVar(Def::LepTrackDR)           < m_minTrackLeptonDR)     return false;
  if(p.getVar(Def::LepTrackDR)           > m_maxTrackLeptonDR)     return false;
 
  //
  // Kinematic track selection
  //
  if(p.getVar(Def::Pt)                   < m_minTrackpT)           return false;
  if(p.getVar(Def::AbsEta)               > m_maxTrackEta)          return false;
  if(std::fabs(p.getVar(Def::Z0Sin))     > m_maxTrackZ0Sin)        return false;
 
  //
  // Hit quality track selection
  //
  if(p.getVar(Def::NumberOfSiHits)       < m_minTrackSiHits)       return false;
  if(p.getVar(Def::NumberOfSharedSiHits) > m_maxTrackSharedSiHits) return false;
  if(p.getVar(Def::NumberOfSiHoles)      > m_maxTrackSiHoles)      return false;
  if(p.getVar(Def::NumberOfPixelHoles)   > m_maxTrackPixHoles)      return false;
 
  return true;
}

prepTrackObject()

bool Prompt::DecoratePromptLeptonRNN::prepTrackObject	(	Prompt::VarHolder &	p,
		const xAOD::TrackParticle &	track,
		const xAOD::TrackParticle &	lepton,
		const xAOD::Jet &	trackJet,
		const xAOD::Vertex &	priVtx,
		const xAOD::EventInfo	event )

Definition at line 273 of file DecoratePromptLeptonRNN.cxx.

{
  //
  // Add xAOD::TrackParticle variables to the track object VarHolder
  //
  uint8_t numberOfPixelHits       = 0;
  uint8_t numberOfSCTHits         = 0;
  uint8_t numberOfPixelHoles      = 0;
  uint8_t numberOfSCTHoles        = 0;
  uint8_t numberOfPixelSharedHits = 0;
  uint8_t numberOfSCTSharedHits   = 0;
 
  if(!(track.summaryValue(numberOfPixelHits,       xAOD::numberOfPixelHits)))       return false;
  if(!(track.summaryValue(numberOfSCTHits,         xAOD::numberOfSCTHits)))         return false;
  if(!(track.summaryValue(numberOfPixelHoles,      xAOD::numberOfPixelHoles)))      return false;
  if(!(track.summaryValue(numberOfSCTHoles,        xAOD::numberOfSCTHoles)))        return false;
  if(!(track.summaryValue(numberOfPixelSharedHits, xAOD::numberOfPixelSharedHits))) return false;
  if(!(track.summaryValue(numberOfSCTSharedHits,   xAOD::numberOfSCTSharedHits)))   return false;
 
  const uint8_t NSiHits   = numberOfPixelHits  + numberOfSCTHits;
  const uint8_t NSiHoles  = numberOfPixelHoles + numberOfSCTHoles;
  const float   NSiShHits = float(numberOfPixelSharedHits) + float(numberOfSCTSharedHits)/2.0;
 
  p.addVar(Def::Pt,                    track.pt());
  p.addVar(Def::AbsEta,                std::fabs(track.eta()));
  p.addVar(Def::NumberOfPIXHits,       numberOfPixelHits);
  p.addVar(Def::NumberOfSCTHits,       numberOfSCTHits);
  p.addVar(Def::NumberOfSiHits,        NSiHits);
  p.addVar(Def::NumberOfSharedSiHits,  NSiShHits);
  p.addVar(Def::NumberOfSiHoles,       NSiHoles);
  p.addVar(Def::NumberOfPixelHoles,    numberOfPixelHoles);
 
  //
  // Add lepton - jet variables to VarHolder
  //
  double PtFrac = -99.;
 
  if(track.pt() > 0.0 && trackJet.pt() > 0.0) {
    PtFrac = track.pt() / trackJet.pt();
  }
 
  p.addVar(Def::TrackPtOverTrackJetPt, PtFrac);
  p.addVar(Def::TrackJetDR,            track.p4().DeltaR(trackJet.p4()));
  p.addVar(Def::LepTrackDR,            track.p4().DeltaR(lepton.p4()));
 
  //
  // Add Impact Parameters
  //
  double d0_significance = -99.;
  double Z0Sin           = 0.0;
 
  if(track.definingParametersCovMatrixVec().size() > 0 && track.definingParametersCovMatrixVec().at(0) > 0.0) {
    d0_significance = xAOD::TrackingHelpers::d0significance(&track,
                  event.beamPosSigmaX(),
                  event.beamPosSigmaY(),
                  event.beamPosSigmaXY());
  }
 
  const double deltaZ0  = track.z0() + track.vz() - priVtx.z();
  Z0Sin                 = deltaZ0*std::sin(track.theta());
 
  p.addVar(Def::Z0Sin,  Z0Sin);
  p.addVar(Def::D0Sig,  d0_significance);
 
  return true;
}

renounce()

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

inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

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

renounceArray()

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

inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

                                                          {
    handlesArray.renounce();
  }

sysInitialize()

StatusCode AthAlgorithm::sysInitialize ( )

overridevirtualinherited

Override sysInitialize.

Override sysInitialize from the base class.

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

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

Reimplemented from AthCommonDataStore< AthCommonMsg< Algorithm > >.

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

Definition at line 66 of file AthAlgorithm.cxx.

                                       {
  StatusCode sc=AthCommonDataStore<AthCommonMsg<Algorithm>>::sysInitialize();
 
  if (sc.isFailure()) {
    return sc;
  }
  ServiceHandle<ICondSvc> cs("CondSvc",name());
  for (auto h : outputHandles()) {
    if (h->isCondition() && h->mode() == Gaudi::DataHandle::Writer) {
      // do this inside the loop so we don't create the CondSvc until needed
      if ( cs.retrieve().isFailure() ) {
        ATH_MSG_WARNING("no CondSvc found: won't autoreg WriteCondHandles");
        return StatusCode::SUCCESS;
      }
      if (cs->regHandle(this,*h).isFailure()) {
        sc = StatusCode::FAILURE;
        ATH_MSG_ERROR("unable to register WriteCondHandle " << h->fullKey()
                      << " with CondSvc");
      }
    }
  }
  return sc;
}

sysStart()

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

overridevirtualinherited

Handle START transition.

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

updateVHKA()

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

inlineinherited

Definition at line 308 of file AthCommonDataStore.h.

                                               {
    // 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_accessQuality

std::unique_ptr<SG::AuxElement::ConstAccessor<unsigned char> > Prompt::DecoratePromptLeptonRNN::m_accessQuality

Definition at line 142 of file DecoratePromptLeptonRNN.h.

m_countEvent

int Prompt::DecoratePromptLeptonRNN::m_countEvent

Definition at line 138 of file DecoratePromptLeptonRNN.h.

m_debug

Gaudi::Property<bool> Prompt::DecoratePromptLeptonRNN::m_debug {this, "debug", false, "debug statement"}

Definition at line 107 of file DecoratePromptLeptonRNN.h.

{this, "debug", false, "debug statement"};

m_decorationPrefixRNN

Gaudi::Property<std::string> Prompt::DecoratePromptLeptonRNN::m_decorationPrefixRNN {this, "decorationPrefixRNN", "", "Prefix of the name for the decorator of RNN to the lepton"}

Definition at line 92 of file DecoratePromptLeptonRNN.h.

{this, "decorationPrefixRNN", "", "Prefix of the name for the decorator of RNN to the lepton"};

m_decorHandleKeys

SG::WriteDecorHandleKeyArray<xAOD::IParticleContainer> Prompt::DecoratePromptLeptonRNN::m_decorHandleKeys { this, "DecorHandleKeys", {} }

Definition at line 144 of file DecoratePromptLeptonRNN.h.

{ this, "DecorHandleKeys", {} };

m_decorNameMap

std::unordered_map<std::string, size_t> Prompt::DecoratePromptLeptonRNN::m_decorNameMap

Definition at line 146 of file DecoratePromptLeptonRNN.h.

m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Algorithm > >::m_detStore

privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

m_eventHandleKey

SG::ReadHandleKey<xAOD::EventInfo> Prompt::DecoratePromptLeptonRNN::m_eventHandleKey

Initial value:

{
      this, "EventHandleKey", "EventInfo"
    }

Definition at line 132 of file DecoratePromptLeptonRNN.h.

                                                    {
      this, "EventHandleKey", "EventInfo"
    };

m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Algorithm > >::m_evtStore

privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

m_extendedExtraObjects

DataObjIDColl AthAlgorithm::m_extendedExtraObjects

privateinherited

Definition at line 79 of file AthAlgorithm.h.

m_hists

std::map<std::string, TH1*> Prompt::DecoratePromptLeptonRNN::m_hists

Definition at line 140 of file DecoratePromptLeptonRNN.h.

m_histSvc

ServiceHandle<ITHistSvc> Prompt::DecoratePromptLeptonRNN::m_histSvc

Definition at line 115 of file DecoratePromptLeptonRNN.h.

m_inputContainerLeptonKey

SG::ReadHandleKey<xAOD::IParticleContainer> Prompt::DecoratePromptLeptonRNN::m_inputContainerLeptonKey

Initial value:

{
      this, "inputContainerLepton", "",
      "Container's name of the lepton that you want to decorate"
    }

Definition at line 118 of file DecoratePromptLeptonRNN.h.

                                                                      {
      this, "inputContainerLepton", "",
      "Container's name of the lepton that you want to decorate"
    };

m_inputContainerPrimaryVerticesKey

SG::ReadHandleKey<xAOD::VertexContainer> Prompt::DecoratePromptLeptonRNN::m_inputContainerPrimaryVerticesKey

Initial value:

{
      this, "inputContainerPrimaryVertices", "", "Primary vertex container name"
    }

Definition at line 128 of file DecoratePromptLeptonRNN.h.

                                                                            {
      this, "inputContainerPrimaryVertices", "", "Primary vertex container name"
    };

m_inputContainerTrackJetKey

SG::ReadHandleKey<xAOD::JetContainer> Prompt::DecoratePromptLeptonRNN::m_inputContainerTrackJetKey

Initial value:

{
      this, "inputContainerTrackJet", "", "Track Jet container name"
    }

Definition at line 125 of file DecoratePromptLeptonRNN.h.

                                                                  {
      this, "inputContainerTrackJet", "", "Track Jet container name"
    };

m_inputContainerTrackKey

SG::ReadHandleKey<xAOD::TrackParticleContainer> Prompt::DecoratePromptLeptonRNN::m_inputContainerTrackKey

Initial value:

{
      this, "inputContainerTrack", "", "Track container name"
    }

Definition at line 122 of file DecoratePromptLeptonRNN.h.

                                                                         {
      this, "inputContainerTrack", "", "Track container name"
    };

m_maxLepTrackJetDR

Gaudi::Property<double> Prompt::DecoratePromptLeptonRNN::m_maxLepTrackJetDR {this, "maxLepTrackJetDR", 0.4, "Maximum distance between lepton and track jet for track jet matching"}

Definition at line 100 of file DecoratePromptLeptonRNN.h.

{this, "maxLepTrackJetDR", 0.4, "Maximum distance between lepton and track jet for track jet matching"};

m_maxTrackEta

Gaudi::Property<double> Prompt::DecoratePromptLeptonRNN::m_maxTrackEta {this, "maxTrackEta", 2.5, "abs(eta) cut config for the input tracks to RNN"}

Definition at line 95 of file DecoratePromptLeptonRNN.h.

{this, "maxTrackEta", 2.5, "abs(eta) cut config for the input tracks to RNN"};

m_maxTrackLeptonDR

Gaudi::Property<double> Prompt::DecoratePromptLeptonRNN::m_maxTrackLeptonDR {this, "maxTrackLeptonDR", 0.4, "Delta R between lepton and track cut config for the input tracks to RNN"}

Definition at line 99 of file DecoratePromptLeptonRNN.h.

{this, "maxTrackLeptonDR", 0.4, "Delta R between lepton and track cut config for the input tracks to RNN"};

m_maxTrackPixHoles

Gaudi::Property<unsigned> Prompt::DecoratePromptLeptonRNN::m_maxTrackPixHoles {this, "maxTrackPixHoles", 1.5, "track pixel holes cut config for the input tracks to RNN"}

Definition at line 105 of file DecoratePromptLeptonRNN.h.

{this, "maxTrackPixHoles", 1.5, "track pixel holes cut config for the input tracks to RNN"};

m_maxTrackSharedSiHits

Gaudi::Property<double> Prompt::DecoratePromptLeptonRNN::m_maxTrackSharedSiHits {this, "maxTrackSharedSiHits", 1.5, "track shared si hits cut config for the input tracks to RNN"}

Definition at line 102 of file DecoratePromptLeptonRNN.h.

{this, "maxTrackSharedSiHits", 1.5, "track shared si hits cut config for the input tracks to RNN"};

m_maxTrackSiHoles

Gaudi::Property<unsigned> Prompt::DecoratePromptLeptonRNN::m_maxTrackSiHoles {this, "maxTrackSiHoles", 2.5, "track holes cut config for the input tracks to RNN"}

Definition at line 104 of file DecoratePromptLeptonRNN.h.

{this, "maxTrackSiHoles", 2.5, "track holes cut config for the input tracks to RNN"};

m_maxTrackZ0Sin

Gaudi::Property<double> Prompt::DecoratePromptLeptonRNN::m_maxTrackZ0Sin {this, "maxTrackZ0Sin", 1.0, "Z0sin cut config for the input tracks to RNN"}

Definition at line 96 of file DecoratePromptLeptonRNN.h.

{this, "maxTrackZ0Sin", 1.0, "Z0sin cut config for the input tracks to RNN"};

m_minTrackLeptonDR

Gaudi::Property<double> Prompt::DecoratePromptLeptonRNN::m_minTrackLeptonDR {this, "minTrackLeptonDR", 1.0e-6, "Delta R between lepton and track cut config for the input tracks to RNN"}

Definition at line 98 of file DecoratePromptLeptonRNN.h.

{this, "minTrackLeptonDR", 1.0e-6, "Delta R between lepton and track cut config for the input tracks to RNN"};

m_minTrackpT

Gaudi::Property<double> Prompt::DecoratePromptLeptonRNN::m_minTrackpT {this, "minTrackpT", 500.0, "pT cut config for the input tracks to RNN"}

Definition at line 94 of file DecoratePromptLeptonRNN.h.

{this, "minTrackpT", 500.0, "pT cut config for the input tracks to RNN"};

m_minTrackSiHits

Gaudi::Property<unsigned> Prompt::DecoratePromptLeptonRNN::m_minTrackSiHits {this, "minTrackSiHits", 6.5, "track silicon detector hits cut config for the input tracks to RNN"}

Definition at line 103 of file DecoratePromptLeptonRNN.h.

{this, "minTrackSiHits", 6.5, "track silicon detector hits cut config for the input tracks to RNN"};

m_outputStream

Gaudi::Property<std::string> Prompt::DecoratePromptLeptonRNN::m_outputStream {this, "outputStream", "", "Path of the ROOT output directory of the histograms for RNN debug"}

Definition at line 91 of file DecoratePromptLeptonRNN.h.

{this, "outputStream", "", "Path of the ROOT output directory of the histograms for RNN debug"};

m_printTime

Gaudi::Property<bool> Prompt::DecoratePromptLeptonRNN::m_printTime {this, "printTime", false, "print running time, for debug"}

Definition at line 108 of file DecoratePromptLeptonRNN.h.

{this, "printTime", false, "print running time, for debug"};

m_timerEvent

TStopwatch Prompt::DecoratePromptLeptonRNN::m_timerEvent

Definition at line 137 of file DecoratePromptLeptonRNN.h.

m_toolRNN

ToolHandle<IRNNTool> Prompt::DecoratePromptLeptonRNN::m_toolRNN

Initial value:

{
      this, "toolRNN", "defaultToolRNN", "Dedicated tool for RNN prediction"
    }

Definition at line 111 of file DecoratePromptLeptonRNN.h.

                                   {
      this, "toolRNN", "defaultToolRNN", "Dedicated tool for RNN prediction"
    };

m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< Algorithm > >::m_varHandleArraysDeclared

privateinherited

Definition at line 399 of file AthCommonDataStore.h.

m_vhka

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

privateinherited

Definition at line 398 of file AthCommonDataStore.h.

---

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

• DecoratePromptLeptonRNN.h
• DecoratePromptLeptonRNN.cxx