TrackOverlayDecisionAlg::TrackOverlayDecisionAlg Class Reference

#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.
virtual	~TrackOverlayDecisionAlg ()=default
	Destructor.
virtual StatusCode	initialize () override final
	Athena algorithm's interface method initialize().
virtual StatusCode	execute (const EventContext &ctx) const override final
	Athena algorithm's interface method execute().
virtual StatusCode	finalize () override final
	Athena algorithm's interface method finalize().
virtual StatusCode	sysInitialize () override
	Override sysInitialize.
virtual bool	isClonable () const override
	Specify if the algorithm is clonable.
virtual unsigned int	cardinality () const override
	Cardinality (Maximum number of clones that can exist) special value 0 means that algorithm is reentrant.
virtual StatusCode	sysExecute (const EventContext &ctx) override
	Execute an algorithm.
virtual const DataObjIDColl &	extraOutputDeps () const override
	Return the list of extra output dependencies.
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.
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

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

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
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.
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 55 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 19 of file TrackOverlayDecisionAlg.cxx.

                                                                                                      :
     ::AthReentrantAlgorithm( name, pSvcLocator )
    {
    }

~TrackOverlayDecisionAlg()

virtual TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::~TrackOverlayDecisionAlg ( )

virtualdefault

Destructor.

Member Function Documentation

cardinality()

unsigned int AthCommonReentrantAlgorithm< Gaudi::Algorithm >::cardinality ( ) const

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()

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());
 
  }

declareProperty()

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.

{ return m_detStore; }

evtStore()

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.

{ return m_evtStore; }

execute()

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

finaloverridevirtual

Athena algorithm's interface method execute().

Definition at line 114 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 matrices. 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;
            }
        else 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++;
        }
    }
    if (truthMultiplicity == 0){
      ATH_MSG_ERROR("truthMultiplicity is zero!");
      return StatusCode::FAILURE;
    }
    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 ( ) const

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 43 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 91 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 106 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 55 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 24 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 ( ) const

overridevirtualinherited

Specify if the algorithm is clonable.

Reentrant algorithms are clonable.

Reimplemented in HiveAlgBase, InDet::GNNSeedingTrackMaker, InDet::SCT_Clusterization, InDet::SiSPGNNTrackMaker, InDet::SiSPSeededTrackFinder, InDet::SiTrackerSpacePointFinder, ITkPixelCablingAlg, ITkStripCablingAlg, RoIBResultToxAOD, SCT_ByteStreamErrorsTestAlg, SCT_CablingCondAlgFromCoraCool, SCT_CablingCondAlgFromText, SCT_ConditionsParameterTestAlg, SCT_ConditionsSummaryTestAlg, SCT_ConfigurationConditionsTestAlg, SCT_FlaggedConditionTestAlg, SCT_LinkMaskingTestAlg, SCT_MajorityConditionsTestAlg, SCT_ModuleVetoTestAlg, SCT_MonitorConditionsTestAlg, SCT_PrepDataToxAOD, SCT_RawDataToxAOD, SCT_ReadCalibChipDataTestAlg, SCT_ReadCalibDataTestAlg, SCT_RODVetoTestAlg, SCT_SensorsTestAlg, SCT_SiliconConditionsTestAlg, SCT_StripVetoTestAlg, SCT_TdaqEnabledTestAlg, SCT_TestCablingAlg, SCTEventFlagWriter, SCTRawDataProvider, SCTSiLorentzAngleTestAlg, SCTSiPropertiesTestAlg, and Simulation::BeamEffectsAlg.

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()

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

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                {
    return this->msgStream();
  }

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 HypoBase, and InputMakerBase.

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 68 of file TrackOverlayDecisionAlg.h.

{"selectedByPileupSwitch"};

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 76 of file TrackOverlayDecisionAlg.h.

{this,"EventInfoContainerName", "EventInfo", ""};

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 83 of file TrackOverlayDecisionAlg.h.

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

m_inputInfo

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

private

Definition at line 88 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 84 of file TrackOverlayDecisionAlg.h.

{this, "InvertFilter", false, "Invert filter decision."}; 

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 74 of file TrackOverlayDecisionAlg.h.

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

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 85 of file TrackOverlayDecisionAlg.h.

{this, "MLThreshold", 0.74201, "ML threshold for bad/good tracks decision. ML scores larger than this threshold are considered as bad tracks."};

m_outputInfo

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

private

Definition at line 89 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 73 of file TrackOverlayDecisionAlg.h.

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

m_session

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

private

Definition at line 90 of file TrackOverlayDecisionAlg.h.

m_svc

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

private

Definition at line 87 of file TrackOverlayDecisionAlg.h.

{this, "ONNXRuntimeSvc", "AthOnnx::OnnxRuntimeSvc", "CaloMuonScoreTool ONNXRuntimeSvc"};

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 80 of file TrackOverlayDecisionAlg.h.

{this, "TruthEvents", "TruthEvents","Name of the truth events container probably either TruthEvent or TruthEvents"};

m_truthParticleName

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

private

Definition at line 79 of file TrackOverlayDecisionAlg.h.

{this, "TruthParticleContainerName",  "TruthParticles", ""};

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 78 of file TrackOverlayDecisionAlg.h.

{this, "TruthPileupEvents", "TruthPileupEvents","Name of the truth pileup events container probably TruthPileupEvent(s)"};

m_truthSelectionTool

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

private

Definition at line 67 of file TrackOverlayDecisionAlg.h.

{this, "TruthSelectionTool","AthTruthSelectionTool", "Truth selection tool (for efficiencies and resolutions)"};

m_useTrackSelection

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

private

Definition at line 72 of file TrackOverlayDecisionAlg.h.

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

m_usingSpecialPileupSwitch

bool TrackOverlayDecisionAlg::TrackOverlayDecisionAlg::m_usingSpecialPileupSwitch {false}

private

Definition at line 69 of file TrackOverlayDecisionAlg.h.

{false};

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:

• TrackOverlayDecisionAlg.h
• TrackOverlayDecisionAlg.cxx