#include <TrackOverlayDecisionAlg.h>

Inheritance diagram for TrackOverlayDecisionAlg::TrackOverlayDecisionAlg:

Collaboration diagram for TrackOverlayDecisionAlg::TrackOverlayDecisionAlg:

Public Member Functions
	TrackOverlayDecisionAlg (const std::string &name, ISvcLocator *pSvcLocator)
	Constructor with parameters. More...

virtual	~TrackOverlayDecisionAlg ()=default
	Destructor. More...

virtual StatusCode	initialize () override final
	Athena algorithm's interface method initialize() More...

virtual StatusCode	execute (const EventContext &ctx) const override final
	Athena algorithm's interface method execute() More...

virtual StatusCode	finalize () override final
	Athena algorithm's interface method finalize() More...

virtual StatusCode	sysInitialize () override
	Override sysInitialize. More...

virtual bool	isClonable () const override
	Specify if the algorithm is clonable. More...

virtual unsigned int	cardinality () const override
	Cardinality (Maximum number of clones that can exist) special value 0 means that algorithm is reentrant. More...

virtual StatusCode	sysExecute (const EventContext &ctx) override
	Execute an algorithm. More...

virtual const DataObjIDColl &	extraOutputDeps () const override
	Return the list of extra output dependencies. More...

virtual bool	filterPassed (const EventContext &ctx) const

virtual void	setFilterPassed (bool state, const EventContext &ctx) const

ServiceHandle< StoreGateSvc > &	evtStore ()
	The standard `StoreGateSvc` (event store) Returns (kind of) a pointer to the `StoreGateSvc`. More...

const ServiceHandle< StoreGateSvc > &	evtStore () const
	The standard `StoreGateSvc` (event store) Returns (kind of) a pointer to the `StoreGateSvc`. More...

const ServiceHandle< StoreGateSvc > &	detStore () const
	The standard `StoreGateSvc/DetectorStore` Returns (kind of) a pointer to the `StoreGateSvc`. More...

virtual StatusCode	sysStart () override
	Handle START transition. More...

virtual std::vector< Gaudi::DataHandle * >	inputHandles () const override
	Return this algorithm's input handles. More...

virtual std::vector< Gaudi::DataHandle * >	outputHandles () const override
	Return this algorithm's output handles. More...

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

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleKey &hndl, const std::string &doc, const SG::VarHandleKeyType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleBase &hndl, const std::string &doc, const SG::VarHandleType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleKeyArray &hndArr, const std::string &doc, const SG::VarHandleKeyArrayType &)

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, T &property, const std::string &doc, const SG::NotHandleType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, T &property, const std::string &doc="none")
	Declare a new Gaudi property. More...

void	updateVHKA (Gaudi::Details::PropertyBase &)

MsgStream &	msg () const

MsgStream &	msg (const MSG::Level lvl) const

bool	msgLvl (const MSG::Level lvl) const

Protected Member Functions
void	renounceArray (SG::VarHandleKeyArray &handlesArray)
	remove all handles from I/O resolution More...

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. More...

Private Types
typedef ServiceHandle< StoreGateSvc >	StoreGateSvc_t

Private Member Functions
void	markSelectedByPileupSwitch (const std::vector< const xAOD::TruthParticle * > &truthParticles) const

const std::vector< const xAOD::TruthParticle * >	getTruthParticles () const

std::tuple< std::vector< int64_t >, std::vector< char * > >	GetInputNodeInfo (const std::unique_ptr< Ort::Session > &session)

std::tuple< std::vector< int64_t >, std::vector< char * > >	GetOutputNodeInfo (const std::unique_ptr< Ort::Session > &session)

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
	specialization for handling Gaudi::Property<SG::VarHandleKey> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyArrayType &)
	specialization for handling Gaudi::Property<SG::VarHandleKeyArray> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleType &)
	specialization for handling Gaudi::Property<SG::VarHandleBase> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T, V, H > &t, const SG::NotHandleType &)
	specialization for handling everything that's not a Gaudi::Property<SG::VarHandleKey> or a <SG::VarHandleKeyArray> More...

Private Attributes
ToolHandle< IAthSelectionTool >	m_truthSelectionTool {this, "TruthSelectionTool","AthTruthSelectionTool", "Truth selection tool (for efficiencies and resolutions)"}

SG::AuxElement::Decorator< bool >	m_dec_selectedByPileupSwitch {"selectedByPileupSwitch"}

bool	m_usingSpecialPileupSwitch {false}

BooleanProperty	m_useTrackSelection {this, "useTrackSelection", false, "plot only tracks accepted by selection tool"}

StringProperty	m_pileupSwitch {this, "PileupSwitch", "HardScatter", "Pileup truth strategy to use. May be \"All\", \"HardScatter\", or \"PileUp\""}

FloatProperty	m_lowProb {this,"LowProb",0.5,"Truth match prob. cutoff for efficiency (lower bound) and fake (upper bound) classification."}

SG::ReadHandleKey< xAOD::EventInfo >	m_eventInfoContainerName {this,"EventInfoContainerName", "EventInfo", ""}

SG::ReadHandleKey< xAOD::TruthPileupEventContainer >	m_truthPileUpEventName {this, "TruthPileupEvents", "TruthPileupEvents","Name of the truth pileup events container probably TruthPileupEvent(s)"}

SG::ReadHandleKey< xAOD::TruthParticleContainer >	m_truthParticleName {this, "TruthParticleContainerName", "TruthParticles", ""}

SG::ReadHandleKey< xAOD::TruthEventContainer >	m_truthEventName {this, "TruthEvents", "TruthEvents","Name of the truth events container probably either TruthEvent or TruthEvents"}

FilterReporterParams	m_filterParams {this, "TrackOverlayDecisionAlg", "Decides whether events should be reconstructed in track-overlay workflow or MC-overlay."}

Gaudi::Property< bool >	m_invertfilter {this, "InvertFilter", false, "Invert filter decision."}
	invert filter decision at the end More...

Gaudi::Property< float >	m_MLthreshold {this, "MLThreshold", 0.74201, "ML threshold for bad/good tracks decision. ML scores larger than this threshold are considered as bad tracks."}

ServiceHandle< AthOnnx::IOnnxRuntimeSvc >	m_svc {this, "ONNXRuntimeSvc", "AthOnnx::OnnxRuntimeSvc", "CaloMuonScoreTool ONNXRuntimeSvc"}

std::tuple< std::vector< int64_t >, std::vector< char * > >	m_inputInfo

std::tuple< std::vector< int64_t >, std::vector< char * > >	m_outputInfo

std::unique_ptr< Ort::Session >	m_session

DataObjIDColl	m_extendedExtraObjects
	Extra output dependency collection, extended by AthAlgorithmDHUpdate to add symlinks. More...

StoreGateSvc_t	m_evtStore
	Pointer to StoreGate (event store by default) More...

StoreGateSvc_t	m_detStore
	Pointer to StoreGate (detector store by default) More...

std::vector< SG::VarHandleKeyArray * >	m_vhka

bool	m_varHandleArraysDeclared

Detailed Description

Definition at line 46 of file TrackOverlayDecisionAlg.h.

Member Typedef Documentation

◆ StoreGateSvc_t

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

privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

◆ TrackOverlayDecisionAlg()

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

Constructor with parameters.

Definition at line 27 of file TrackOverlayDecisionAlg.cxx.

                                                                                                       :
      ::AthReentrantAlgorithm( name, pSvcLocator )
     {
     }

◆ ~TrackOverlayDecisionAlg()

virtual TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::~TrackOverlayDecisionAlg ( )

virtualdefault

Destructor.

Member Function Documentation

◆ cardinality()

unsigned int AthCommonReentrantAlgorithm< Gaudi::Algorithm >::cardinality

overridevirtualinherited

Cardinality (Maximum number of clones that can exist) special value 0 means that algorithm is reentrant.

Override this to return 0 for reentrant algorithms.

Definition at line 75 of file AthCommonReentrantAlgorithm.cxx.

 {
   return 0;
 }

◆ declareGaudiProperty() [1/4]

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

inlineprivateinherited

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

Definition at line 170 of file AthCommonDataStore.h.

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

◆ declareGaudiProperty() [2/4]

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

◆ declareGaudiProperty() [3/4]

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

inlineprivateinherited

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

Definition at line 184 of file AthCommonDataStore.h.

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

◆ declareGaudiProperty() [4/4]

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

inlineprivateinherited

specialization for handling everything that's not a Gaudi::Property<SG::VarHandleKey> or a <SG::VarHandleKeyArray>

Definition at line 199 of file AthCommonDataStore.h.

   {
     return PBASE::declareProperty(t);
   }

◆ declareProperty() [1/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareProperty	(	const std::string &	name,
		SG::VarHandleBase &	hndl,
		const std::string &	doc,
		const SG::VarHandleType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
hndl	Object holding the property value.
doc	Documentation string for the property.

This is the version for types that derive from SG::VarHandleBase. The property value object is put on the input and output lists as appropriate; then we forward to the base class.

Definition at line 245 of file AthCommonDataStore.h.

   {
     this->declare(hndl.vhKey());
     hndl.vhKey().setOwner(this);
  
     return PBASE::declareProperty(name,hndl,doc);
   }

◆ declareProperty() [2/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareProperty	(	const std::string &	name,
		SG::VarHandleKey &	hndl,
		const std::string &	doc,
		const SG::VarHandleKeyType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
hndl	Object holding the property value.
doc	Documentation string for the property.

This is the version for types that derive from SG::VarHandleKey. The property value object is put on the input and output lists as appropriate; then we forward to the base class.

Definition at line 221 of file AthCommonDataStore.h.

   {
     this->declare(hndl);
     hndl.setOwner(this);
  
     return PBASE::declareProperty(name,hndl,doc);
   }

◆ declareProperty() [3/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareProperty	(	const std::string &	name,
		SG::VarHandleKeyArray &	hndArr,
		const std::string &	doc,
		const SG::VarHandleKeyArrayType &
	)

inlineinherited

Definition at line 259 of file AthCommonDataStore.h.

   {
  
     // std::ostringstream ost;
     // ost << Algorithm::name() << " VHKA declareProp: " << name 
     //     << " size: " << hndArr.keys().size() 
     //     << " mode: " << hndArr.mode() 
     //     << "  vhka size: " << m_vhka.size()
     //     << "\n";
     // debug() << ost.str() << endmsg;
  
     hndArr.setOwner(this);
     m_vhka.push_back(&hndArr);
  
     Gaudi::Details::PropertyBase* p =  PBASE::declareProperty(name, hndArr, doc);
     if (p != 0) {
       p->declareUpdateHandler(&AthCommonDataStore<PBASE>::updateVHKA, this);
     } else {
       ATH_MSG_ERROR("unable to call declareProperty on VarHandleKeyArray " 
                     << name);
     }
  
     return p;
  
   }

◆ declareProperty() [4/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareProperty	(	const std::string &	name,
		T &	property,
		const std::string &	doc,
		const SG::NotHandleType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
property	Object holding the property value.
doc	Documentation string for the property.

This is the generic version, for types that do not derive from SG::VarHandleKey. It just forwards to the base class version of declareProperty.

Definition at line 333 of file AthCommonDataStore.h.

   {
     return PBASE::declareProperty(name, property, doc);
   }

◆ declareProperty() [5/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareProperty	(	const std::string &	name,
		T &	property,
		const std::string &	doc = `"none"`
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
property	Object holding the property value.
doc	Documentation string for the property.

This dispatches to either the generic declareProperty or the one for VarHandle/Key/KeyArray.

Definition at line 352 of file AthCommonDataStore.h.

   {
     typedef typename SG::HandleClassifier<T>::type htype;
     return declareProperty (name, property, doc, htype());
   }

◆ declareProperty() [6/6]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Gaudi::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< Gaudi::Algorithm > >::detStore ( ) const

inlineinherited

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

Definition at line 95 of file AthCommonDataStore.h.

95 { return m_detStore; }

◆ evtStore() [1/2]

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

inlineinherited

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

Definition at line 85 of file AthCommonDataStore.h.

85 { return m_evtStore; }

◆ evtStore() [2/2]

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

inlineinherited

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

Definition at line 90 of file AthCommonDataStore.h.

90 { return m_evtStore; }

◆ execute()

StatusCode TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::execute ( const EventContext & ctx ) const

finaloverridevirtual

Athena algorithm's interface method execute()

Definition at line 122 of file TrackOverlayDecisionAlg.cxx.

 {
     ATH_MSG_DEBUG ("Executing ...");
  
     std::vector<const xAOD::TruthParticle*> truthParticlesVec = TrackOverlayDecisionAlg::getTruthParticles();
  
     //Access truth info for the NN input
     float eventPxSum = 0.0;
     float eventPySum = 0.0;
     float eventPt = 0.0;
     float puEvents = 0.0;
  
     std::vector<float> pxValues, pyValues, pzValues, eValues, etaValues, phiValues, ptValues;
     float truthMultiplicity = 0.0;
     const int truthParticles = truthParticlesVec.size();
     for (int itruth = 0; itruth < truthParticles; itruth++) {
         const xAOD::TruthParticle* thisTruth = truthParticlesVec[itruth];
         const IAthSelectionTool::CutResult accept = m_truthSelectionTool->accept(thisTruth);
         if(accept){
            pxValues.push_back((thisTruth->px()*0.001-1.46988000e+03)* px_diff); //as MinMaxScaler: 1.46988000e+03 is the lowest value of px from a J7 sample; *(0.001) is used to convert unit rather than *(1/1000) to speed up.
            pyValues.push_back((thisTruth->py()*0.001-1.35142000e+03)* py_diff); //the lowest value of py: 1.35142000e+03
            pzValues.push_back((thisTruth->pz()*0.001-1.50464000e+03)* pz_diff); //the lowest value of pz: 1.50464000e+03
            ptValues.push_back((thisTruth->pt()*0.001-5.00006000e-01)* pt_diff); //the lowest value of pt: 5.00006000e-01
  
            etaValues.push_back(thisTruth->eta());
            phiValues.push_back(thisTruth->phi());
            eValues.push_back((thisTruth->e()*0.001-5.08307000e-01)*e_diff); //the lowest value of energy: 5.08307000e-01
  
            eventPxSum += thisTruth->px();
            eventPySum += thisTruth->py();
            truthMultiplicity++;
         }//accept
     }//for itruth
     SG::ReadHandle<xAOD::TruthPileupEventContainer> truthPileupEventContainer;
     SG::ReadHandle<xAOD::EventInfo> pie = SG::ReadHandle<xAOD::EventInfo>(m_eventInfoContainerName, ctx);
     if (!m_truthPileUpEventName.key().empty()) {
          truthPileupEventContainer = SG::ReadHandle<xAOD::TruthPileupEventContainer>(m_truthPileUpEventName, ctx);
     }
     puEvents = !m_truthPileUpEventName.key().empty() and truthPileupEventContainer.isValid() ?  static_cast<int>( truthPileupEventContainer->size() ) : pie.isValid() ? pie->actualInteractionsPerCrossing() : 0;
     eventPt = std::sqrt(eventPxSum*eventPxSum + eventPySum*eventPySum)*0.001;
  
     std::vector<float> puEventsVec(pxValues.size(), (puEvents-1.55000000e+01)*pu_diff); //min of puEvents= 15.5, max of puEvents=84.5
     std::vector<float> truthMultiplicityVec(pxValues.size(), (truthMultiplicity-1.80000000e+01)*multi_diff);
     std::vector<float> eventPtVec(pxValues.size(), (eventPt-3.42359395e-01)*eventPt_diff);
     std::vector<float> predictions;
  
     //Compute the distances using Eigen for Eigen's optimized operations. Initialize matirces. Observed a significant improvement on computing calculation.
     Eigen::VectorXf ptEigen = Eigen::VectorXf::Map(ptValues.data(), ptValues.size());
     Eigen::VectorXf phiEigen = Eigen::VectorXf::Map(phiValues.data(), phiValues.size());
     Eigen::VectorXf etaEigen = Eigen::VectorXf::Map(etaValues.data(), etaValues.size());
     for (std::size_t i = 0; i < truthMultiplicity; ++i) {
         float multiplicity_0p05 = 0.0, multiplicity_0p2 = 0.0;
         float sum_0p05 = 0.0, sum_0p2 = 0.0;
         float pt_0p05 = 0.0, pt_0p2 = 0.0;
         float deltaEtaI = etaEigen[i];
         float phiI = phiEigen[i];
         for (std::size_t j = 0; j < truthMultiplicity; ++j) {
             if (i == j) continue; // Skip the particle itself
             float deltaEta = deltaEtaI - etaEigen[j];
             float deltaPhi = phiI - phiEigen[j];
             if (deltaPhi > M_PI) {
                 deltaPhi -= 2.0 * M_PI;
             }
             float distances = std::sqrt(deltaEta * deltaEta + deltaPhi * deltaPhi);
             if (distances < 0.05){
                 multiplicity_0p05++;
                 sum_0p05 += distances;
                 pt_0p05 += ptEigen[j];
             }
             if (distances < 0.2){
                 multiplicity_0p2++;
                 sum_0p2 += distances;
                 pt_0p2 += ptEigen[j];
             }
         }// for j
  
     std::vector<float> featData;
     featData.push_back(pxValues[i]);
     featData.push_back(pyValues[i]);
     featData.push_back(pzValues[i]);
     featData.push_back(eValues[i]);
     featData.push_back(ptValues[i]);
     featData.push_back((multiplicity_0p2 * area0p2) * constant1);
     featData.push_back((multiplicity_0p05 * area0p05) * constant2);
     featData.push_back((sum_0p2 * area0p2) * constant3);
     featData.push_back((sum_0p05 * area0p05) * constant4);
     featData.push_back(pt_0p2 * constant5);
     featData.push_back(pt_0p05 * constant6);
  
     featData.push_back(puEventsVec[i]);
     featData.push_back(truthMultiplicityVec[i]);
     featData.push_back(eventPtVec[i]);
  
     std::vector<int64_t> input_node_dims;
     std::vector<char*> input_node_names;
     input_node_dims = std::get<0>(m_inputInfo);
     input_node_names = std::get<1>(m_inputInfo);
  
     std::vector<int64_t> output_node_dims;
     std::vector<char*> output_node_names;
     output_node_dims = std::get<0>(m_outputInfo);
     output_node_names = std::get<1>(m_outputInfo);
  
     Ort::MemoryInfo memoryInfo = Ort::MemoryInfo::CreateCpu(OrtArenaAllocator, OrtMemTypeCPU);
     input_node_dims[0]=1;
     Ort::Value input_data = Ort::Value::CreateTensor(memoryInfo, featData.data(), featData.size(), input_node_dims.data(), input_node_dims.size());
     Ort::RunOptions run_options(nullptr);
     //Run the inference
     Ort::Session& mysession ATLAS_THREAD_SAFE = *m_session;
     auto output_values = mysession.Run(run_options, input_node_names.data(), &input_data, input_node_names.size(), output_node_names.data(), output_node_names.size());
     float* predictionData = output_values[0].GetTensorMutableData<float>();
     float prediction = predictionData[0];
  
     predictions.push_back(prediction);
     }//for i
     float threshold = m_MLthreshold;
     ATH_MSG_ALWAYS("ML threshold:" << threshold);
     int badTracks = 0;
     for (float prediction : predictions) {
        if (prediction > threshold) {
           badTracks++;
         }
     }
     float rouletteScore = static_cast<float>(badTracks) / static_cast<float>(truthMultiplicity);
  
     FilterReporter filter(m_filterParams, false, ctx);
     bool pass = false;
     int decision = rouletteScore == 0;
     if (decision==0){ //if ML decision is False, it goes to the MC-overlay workflow
       pass = true;
     }
     else{
       pass = false;
     }
  
     if (m_invertfilter) {
     pass =! pass;
     }
     filter.setPassed(pass);
     ATH_MSG_ALWAYS("End TrackOverlayDecisionAlg, difference in filters: "<<(pass ? "found" : "not found")<<"="<<pass<<", invert="<<m_invertfilter);
     return StatusCode::SUCCESS;
 }

◆ extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< Gaudi::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 & AthCommonReentrantAlgorithm< Gaudi::Algorithm >::extraOutputDeps

overridevirtualinherited

Return the list of extra output dependencies.

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

Definition at line 94 of file AthCommonReentrantAlgorithm.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 BaseAlg::extraOutputDeps();
 }

◆ filterPassed()

virtual bool AthCommonReentrantAlgorithm< Gaudi::Algorithm >::filterPassed ( const EventContext & ctx ) const

inlinevirtualinherited

Definition at line 96 of file AthCommonReentrantAlgorithm.h.

                                                            {
     return execState( ctx ).filterPassed();
   }

◆ finalize()

StatusCode TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::finalize ( )

finaloverridevirtual

Athena algorithm's interface method finalize()

Definition at line 51 of file TrackOverlayDecisionAlg.cxx.

 {
   ATH_MSG_VERBOSE ( "Finalizing ..." );
   ATH_MSG_VERBOSE("-----------------------------------------------------------------");
   ATH_MSG_VERBOSE("m_filterParams.summary()" << m_filterParams.summary());
   ATH_MSG_VERBOSE("-----------------------------------------------------------------");
   ATH_MSG_INFO(m_filterParams.summary());
   ATH_MSG_VERBOSE(" =====================================================================");
  
   return StatusCode::SUCCESS;
 }

◆ GetInputNodeInfo()

std::tuple<std::vector<int64_t>, std::vector<char*> > TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::GetInputNodeInfo ( const std::unique_ptr< Ort::Session > & session )

inlineprivate

Definition at line 82 of file TrackOverlayDecisionAlg.h.

                                                                                                                         {
       std::vector<int64_t> input_node_dims;
       size_t num_input_nodes = session->GetInputCount();
       std::vector<char*> input_node_names(num_input_nodes);
       Ort::AllocatorWithDefaultOptions allocator;
       for( std::size_t i = 0; i < num_input_nodes; i++ ) {
         char* input_name = session->GetInputNameAllocated(i, allocator).release();
         input_node_names[i] = input_name;
         Ort::TypeInfo type_info = session->GetInputTypeInfo(i);
         auto tensor_info = type_info.GetTensorTypeAndShapeInfo();
         input_node_dims = tensor_info.GetShape();
         }
       return std::make_tuple(input_node_dims, input_node_names);
      }

◆ GetOutputNodeInfo()

std::tuple<std::vector<int64_t>, std::vector<char*> > TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::GetOutputNodeInfo ( const std::unique_ptr< Ort::Session > & session )

inlineprivate

Definition at line 97 of file TrackOverlayDecisionAlg.h.

                                                                                                                         {
      std::vector<int64_t> output_node_dims;
      size_t num_output_nodes = session->GetOutputCount();
      std::vector<char*> output_node_names(num_output_nodes);
      Ort::AllocatorWithDefaultOptions allocator;
      for( std::size_t i = 0; i < num_output_nodes; i++ ) {
          char* output_name = session->GetOutputNameAllocated(i, allocator).release();
          output_node_names[i] = output_name;
          Ort::TypeInfo type_info = session->GetOutputTypeInfo(i);
          auto tensor_info = type_info.GetTensorTypeAndShapeInfo();
          output_node_dims = tensor_info.GetShape();
      }
      return std::make_tuple(output_node_dims, output_node_names);
      }

◆ getTruthParticles()

const std::vector< const xAOD::TruthParticle * > TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::getTruthParticles ( ) const

private

Definition at line 63 of file TrackOverlayDecisionAlg.cxx.

                                                                                            {
    std::vector<const xAOD::TruthParticle*> tempVec {};
    if (m_pileupSwitch == "All") {
     if (m_truthParticleName.key().empty()) {
       return tempVec;
     }
     SG::ReadHandle<xAOD::TruthParticleContainer> truthParticleContainer( m_truthParticleName);
     if (not truthParticleContainer.isValid()) {
       return tempVec;
     }
     tempVec.insert(tempVec.begin(), truthParticleContainer->begin(), truthParticleContainer->end());
   } else {
    if (m_pileupSwitch == "HardScatter") {
     if (not m_truthEventName.key().empty()) {
     ATH_MSG_VERBOSE("Getting TruthEvents container.");
     SG::ReadHandle<xAOD::TruthEventContainer> truthEventContainer( m_truthEventName);
     const xAOD::TruthEvent* event = (truthEventContainer.isValid()) ? truthEventContainer->at(0) : nullptr;
     if (not event) {
         return tempVec;
       }
     const auto& links = event->truthParticleLinks();
       tempVec.reserve(event->nTruthParticles());
       for (const auto& link : links) {
         if (link.isValid()){
           tempVec.push_back(*link);
         }
        }
       }
     }else if (m_pileupSwitch == "PileUp") {
       if (not m_truthPileUpEventName.key().empty()) {
       ATH_MSG_VERBOSE("getting TruthPileupEvents container");
       // get truth particles from all pileup events
       SG::ReadHandle<xAOD::TruthPileupEventContainer> truthPileupEventContainer(m_truthPileUpEventName);
       if (truthPileupEventContainer.isValid()) {
         const unsigned int nPileup = truthPileupEventContainer->size();
         tempVec.reserve(nPileup * 200); // quick initial guess, will still save some time
         for (unsigned int i(0); i != nPileup; ++i) {
           const auto *eventPileup = truthPileupEventContainer->at(i);
           // get truth particles from each pileup event
           int ntruth = eventPileup->nTruthParticles();
           ATH_MSG_VERBOSE("Adding " << ntruth << " truth particles from TruthPileupEvents container");
           const auto& links = eventPileup->truthParticleLinks();
           for (const auto& link : links) {
             if (link.isValid()){
               tempVec.push_back(*link);
             }
           }
         }
       } else {
         ATH_MSG_ERROR("no entries in TruthPileupEvents container!");
       }
       }
     } else {
       ATH_MSG_ERROR("bad value for PileUpSwitch");
     }
    }
    return tempVec;
 }

◆ initialize()

StatusCode TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::initialize ( )

finaloverridevirtual

Athena algorithm's interface method initialize()

Definition at line 32 of file TrackOverlayDecisionAlg.cxx.

 {
   ATH_CHECK(m_filterParams.initialize(false));
   ATH_CHECK( m_eventInfoContainerName.initialize() );
   ATH_CHECK( m_truthParticleName.initialize( (m_pileupSwitch == "HardScatter" or m_pileupSwitch == "All") and not m_truthParticleName.key().empty() ) );
   ATH_CHECK(m_truthSelectionTool.retrieve(EnableTool {not m_truthParticleName.key().empty()} ));
   ATH_CHECK( m_truthEventName.initialize( (m_pileupSwitch == "HardScatter" or m_pileupSwitch == "All") and not m_truthEventName.key().empty() ) );
   ATH_CHECK( m_truthPileUpEventName.initialize( (m_pileupSwitch == "PileUp" or m_pileupSwitch == "All") and not m_truthPileUpEventName.key().empty() ) );
   ATH_CHECK(m_svc.retrieve());
   std::string this_file = __FILE__;
   const std::string model_path = PathResolverFindCalibFile("TrackOverlay/TrackOverlay_J7_model.onnx");
   Ort::SessionOptions session_options;
  
   m_session = std::make_unique<Ort::Session>(m_svc->env(), model_path.c_str(), session_options);
   m_inputInfo = TrackOverlayDecisionAlg::GetInputNodeInfo(m_session);
   m_outputInfo = TrackOverlayDecisionAlg::GetOutputNodeInfo(m_session);
   return StatusCode::SUCCESS;
 }

◆ inputHandles()

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

◆ isClonable()

bool AthCommonReentrantAlgorithm< Gaudi::Algorithm >::isClonable

overridevirtualinherited

Specify if the algorithm is clonable.

Reentrant algorithms are clonable.

Definition at line 68 of file AthCommonReentrantAlgorithm.cxx.

 {
   // Reentrant algorithms are clonable.
   return true;
 }

◆ markSelectedByPileupSwitch()

void TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::markSelectedByPileupSwitch ( const std::vector< const xAOD::TruthParticle * > & truthParticles ) const

private

◆ msg() [1/2]

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

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                 {
     return this->msgStream();
   }

◆ msg() [2/2]

MsgStream& AthCommonMsg< Gaudi::Algorithm >::msg ( const MSG::Level lvl ) const

inlineinherited

Definition at line 27 of file AthCommonMsg.h.

                                                   {
     return this->msgStream(lvl);
   }

◆ msgLvl()

bool AthCommonMsg< Gaudi::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< Gaudi::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.

◆ 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< Gaudi::Algorithm > >::renounce ( T & h )

inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

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

◆ renounceArray()

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

inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

                                                           {
     handlesArray.renounce();
   }

◆ setFilterPassed()

virtual void AthCommonReentrantAlgorithm< Gaudi::Algorithm >::setFilterPassed	(	bool	state,
		const EventContext &	ctx
	)		const

inlinevirtualinherited

Definition at line 100 of file AthCommonReentrantAlgorithm.h.

                                                                             {
     execState( ctx ).setFilterPassed( state );
   }

◆ sysExecute()

StatusCode AthCommonReentrantAlgorithm< Gaudi::Algorithm >::sysExecute ( const EventContext & ctx )

overridevirtualinherited

Execute an algorithm.

We override this in order to work around an issue with the Algorithm base class storing the event context in a member variable that can cause crashes in MT jobs.

Definition at line 85 of file AthCommonReentrantAlgorithm.cxx.

 {
   return BaseAlg::sysExecute (ctx);
 }

◆ sysInitialize()

StatusCode AthCommonReentrantAlgorithm< Gaudi::Algorithm >::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< Gaudi::Algorithm > >.

Reimplemented in InputMakerBase, and HypoBase.

Definition at line 61 of file AthCommonReentrantAlgorithm.cxx.

                                                                {
   StatusCode sc=AthCommonDataStore<AthCommonMsg<BaseAlg>>::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< Gaudi::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< Gaudi::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_dec_selectedByPileupSwitch

SG::AuxElement::Decorator<bool> TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_dec_selectedByPileupSwitch {"selectedByPileupSwitch"}

private

Definition at line 59 of file TrackOverlayDecisionAlg.h.

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_detStore

privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

◆ m_eventInfoContainerName

SG::ReadHandleKey<xAOD::EventInfo> TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_eventInfoContainerName {this,"EventInfoContainerName", "EventInfo", ""}

private

Definition at line 67 of file TrackOverlayDecisionAlg.h.

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_evtStore

privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

◆ m_extendedExtraObjects

DataObjIDColl AthCommonReentrantAlgorithm< Gaudi::Algorithm >::m_extendedExtraObjects

privateinherited

Extra output dependency collection, extended by AthAlgorithmDHUpdate to add symlinks.

Empty if no symlinks were found.

Definition at line 114 of file AthCommonReentrantAlgorithm.h.

◆ m_filterParams

FilterReporterParams TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_filterParams {this, "TrackOverlayDecisionAlg", "Decides whether events should be reconstructed in track-overlay workflow or MC-overlay."}

private

Definition at line 74 of file TrackOverlayDecisionAlg.h.

◆ m_inputInfo

std::tuple<std::vector<int64_t>, std::vector<char*> > TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_inputInfo

private

Definition at line 79 of file TrackOverlayDecisionAlg.h.

◆ m_invertfilter

Gaudi::Property<bool> TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_invertfilter {this, "InvertFilter", false, "Invert filter decision."}

private

invert filter decision at the end

Definition at line 75 of file TrackOverlayDecisionAlg.h.

◆ m_lowProb

FloatProperty TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_lowProb {this,"LowProb",0.5,"Truth match prob. cutoff for efficiency (lower bound) and fake (upper bound) classification."}

private

Definition at line 65 of file TrackOverlayDecisionAlg.h.

◆ m_MLthreshold

Gaudi::Property<float> TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_MLthreshold {this, "MLThreshold", 0.74201, "ML threshold for bad/good tracks decision. ML scores larger than this threshold are considered as bad tracks."}

private

Definition at line 76 of file TrackOverlayDecisionAlg.h.

◆ m_outputInfo

std::tuple<std::vector<int64_t>, std::vector<char*> > TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_outputInfo

private

Definition at line 80 of file TrackOverlayDecisionAlg.h.

◆ m_pileupSwitch

StringProperty TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_pileupSwitch {this, "PileupSwitch", "HardScatter", "Pileup truth strategy to use. May be \"All\", \"HardScatter\", or \"PileUp\""}

private

Definition at line 64 of file TrackOverlayDecisionAlg.h.

◆ m_session

std::unique_ptr<Ort::Session> TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_session

private

Definition at line 81 of file TrackOverlayDecisionAlg.h.

◆ m_svc

ServiceHandle<AthOnnx::IOnnxRuntimeSvc> TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_svc {this, "ONNXRuntimeSvc", "AthOnnx::OnnxRuntimeSvc", "CaloMuonScoreTool ONNXRuntimeSvc"}

private

Definition at line 78 of file TrackOverlayDecisionAlg.h.

◆ m_truthEventName

SG::ReadHandleKey<xAOD::TruthEventContainer> TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_truthEventName {this, "TruthEvents", "TruthEvents","Name of the truth events container probably either TruthEvent or TruthEvents"}

private

Definition at line 71 of file TrackOverlayDecisionAlg.h.

◆ m_truthParticleName

SG::ReadHandleKey<xAOD::TruthParticleContainer> TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_truthParticleName {this, "TruthParticleContainerName", "TruthParticles", ""}

private

Definition at line 70 of file TrackOverlayDecisionAlg.h.

◆ m_truthPileUpEventName

SG::ReadHandleKey<xAOD::TruthPileupEventContainer> TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_truthPileUpEventName {this, "TruthPileupEvents", "TruthPileupEvents","Name of the truth pileup events container probably TruthPileupEvent(s)"}

private

Definition at line 69 of file TrackOverlayDecisionAlg.h.

◆ m_truthSelectionTool

ToolHandle<IAthSelectionTool> TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_truthSelectionTool {this, "TruthSelectionTool","AthTruthSelectionTool", "Truth selection tool (for efficiencies and resolutions)"}

private

Definition at line 58 of file TrackOverlayDecisionAlg.h.

◆ m_useTrackSelection

BooleanProperty TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_useTrackSelection {this, "useTrackSelection", false, "plot only tracks accepted by selection tool"}

private

Definition at line 63 of file TrackOverlayDecisionAlg.h.

◆ m_usingSpecialPileupSwitch

bool TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_usingSpecialPileupSwitch {false}

private

Definition at line 60 of file TrackOverlayDecisionAlg.h.

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_varHandleArraysDeclared

privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vhka

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

privateinherited

Definition at line 398 of file AthCommonDataStore.h.

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

Public Member Functions

Protected Member Functions

Private Types

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ StoreGateSvc_t

Constructor & Destructor Documentation

◆ TrackOverlayDecisionAlg()

◆ ~TrackOverlayDecisionAlg()

Member Function Documentation

◆ cardinality()

◆ declareGaudiProperty() [1/4]

◆ declareGaudiProperty() [2/4]

◆ declareGaudiProperty() [3/4]

◆ declareGaudiProperty() [4/4]

◆ declareProperty() [1/6]

◆ declareProperty() [2/6]

◆ declareProperty() [3/6]

◆ declareProperty() [4/6]

◆ declareProperty() [5/6]

◆ declareProperty() [6/6]

◆ detStore()

◆ evtStore() [1/2]

◆ evtStore() [2/2]

◆ execute()

◆ extraDeps_update_handler()

◆ extraOutputDeps()

◆ filterPassed()

◆ finalize()

◆ GetInputNodeInfo()

◆ GetOutputNodeInfo()

◆ getTruthParticles()

◆ initialize()

◆ inputHandles()

◆ isClonable()

◆ markSelectedByPileupSwitch()

◆ msg() [1/2]

◆ msg() [2/2]

◆ msgLvl()

◆ outputHandles()

◆ renounce()

◆ renounceArray()

◆ setFilterPassed()

◆ sysExecute()

◆ sysInitialize()

◆ sysStart()

◆ updateVHKA()

Member Data Documentation

◆ m_dec_selectedByPileupSwitch

◆ m_detStore

◆ m_eventInfoContainerName

◆ m_evtStore

◆ m_extendedExtraObjects

◆ m_filterParams

◆ m_inputInfo

◆ m_invertfilter

◆ m_lowProb

◆ m_MLthreshold

◆ m_outputInfo

◆ m_pileupSwitch

◆ m_session

◆ m_svc

◆ m_truthEventName

◆ m_truthParticleName

◆ m_truthPileUpEventName

◆ m_truthSelectionTool

◆ m_useTrackSelection

◆ m_usingSpecialPileupSwitch

◆ m_varHandleArraysDeclared

◆ m_vhka