InDet::NnPixelClusterSplitProbTool Class Reference

#include <NnPixelClusterSplitProbTool.h>

Inheritance diagram for InDet::NnPixelClusterSplitProbTool:

Collaboration diagram for InDet::NnPixelClusterSplitProbTool:

Public Member Functions
	NnPixelClusterSplitProbTool (const std::string &t, const std::string &n, const IInterface *p)
virtual	~NnPixelClusterSplitProbTool ()=default
StatusCode	initialize () override
virtual InDet::PixelClusterSplitProb	splitProbability (const InDet::PixelCluster &origCluster) const override
virtual InDet::PixelClusterSplitProb	splitProbability (const InDet::PixelCluster &origCluster, const Trk::TrackParameters &trackParameters) const override

Private Member Functions
InDet::PixelClusterSplitProb	compileSplitProbability (std::vector< double > &vectorOfProbs) const

Private Attributes
ToolHandle< NnClusterizationFactory >	m_NnClusterizationFactory { this, "NnClusterizationFactory", "InDet::NnClusterizationFactory/NnClusterizationFactory" }
SG::ReadCondHandleKey< InDet::BeamSpotData >	m_beamSpotKey { this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot" }
DoubleArrayProperty	m_priorMultiplicityContent { this, "PriorMultiplicityContent", {2793337, 82056, 19944} }
BooleanProperty	m_useBeamSpotInfo { this, "useBeamSpotInfo", true }

Detailed Description

Definition at line 34 of file NnPixelClusterSplitProbTool.h.

Constructor & Destructor Documentation

NnPixelClusterSplitProbTool()

InDet::NnPixelClusterSplitProbTool::NnPixelClusterSplitProbTool	(	const std::string &	t,
		const std::string &	n,
		const IInterface *	p
	)

Definition at line 28 of file NnPixelClusterSplitProbTool.cxx.

          :base_class(t,n,p)
  {
  }

~NnPixelClusterSplitProbTool()

virtual InDet::NnPixelClusterSplitProbTool::~NnPixelClusterSplitProbTool ( )

virtualdefault

Member Function Documentation

compileSplitProbability()

InDet::PixelClusterSplitProb InDet::NnPixelClusterSplitProbTool::compileSplitProbability ( std::vector< double > &  vectorOfProbs ) const

private

Definition at line 117 of file NnPixelClusterSplitProbTool.cxx.

  {
 
 
    double sum=0;
 
    std::vector<double>::iterator begin=vectorOfProbs.begin();
    std::vector<double>::iterator end=vectorOfProbs.end();
 
    for (std::vector<double>::iterator iter=begin;iter!=end;++iter)
    {
      sum+=*iter;
    }
 
    
    ATH_MSG_VERBOSE(" Sum of cluster probabilities is: "<<sum);
 
    std::vector<double> vectorOfSplitProbs;
 
    for (std::vector<double>::iterator iter=begin;iter!=end;++iter)
    {
      (*iter)/=sum;
    }
 
    if (m_priorMultiplicityContent.value().size()<vectorOfProbs.size())
    {
      ATH_MSG_ERROR("Prior compatibilities count " <<  m_priorMultiplicityContent.value().size() << " is too small: please correct through job properties.");
      return InDet::PixelClusterSplitProb(std::vector<double>());
    }
 
    double psum=0;
    int count=0;
    for (std::vector<double>::iterator iter=begin;iter!=end;++iter,++count)
    {
      psum+=(*iter)/m_priorMultiplicityContent.value()[count];
    }
 
    count=0;
    for (std::vector<double>::iterator iter=begin;iter!=end;++iter,++count)
    {
      (*iter)/=m_priorMultiplicityContent.value()[count];
      (*iter)/=psum;
    }
 
    double sumTest=0;
 
    for (std::vector<double>::iterator iter=begin;iter!=end;++iter)
    {
      ATH_MSG_VERBOSE("After update prob is: " << *iter);
      sumTest+=*iter;
    }
 
    ATH_MSG_VERBOSE(" Sum of cluster probabilities is: "<<sumTest);
    for (std::vector<double>::iterator iter=begin;iter!=end;++iter)
    {
      if (iter!=begin)
      {
        vectorOfSplitProbs.push_back(*iter);
      }
    }
 
    ATH_MSG_VERBOSE(" normalized P(1->2): " << vectorOfSplitProbs[0] << " P(2->3): " << vectorOfSplitProbs[1] );
 
    PixelClusterSplitProb  clusterSplitProb(vectorOfSplitProbs);
 
    ATH_MSG_VERBOSE("SplitProb: " << clusterSplitProb.splitProbability(2) << " -->3 " << clusterSplitProb.splitProbability(3) );
    
    return clusterSplitProb;
  }

initialize()

StatusCode InDet::NnPixelClusterSplitProbTool::initialize ( )

override

Definition at line 34 of file NnPixelClusterSplitProbTool.cxx.

  {
    
 
    if (m_NnClusterizationFactory.retrieve().isFailure())
    {
      ATH_MSG_ERROR(" Unable to retrieve "<< m_NnClusterizationFactory );
      return StatusCode::FAILURE;
    }
 
    ATH_CHECK(m_beamSpotKey.initialize());
 
    ATH_MSG_DEBUG(" Cluster split prob tool initialized successfully "<< m_NnClusterizationFactory );
    return StatusCode::SUCCESS;
  }

splitProbability() [1/2]

InDet::PixelClusterSplitProb InDet::NnPixelClusterSplitProbTool::splitProbability ( const InDet::PixelCluster &  origCluster ) const

overridevirtual

Definition at line 51 of file NnPixelClusterSplitProbTool.cxx.

  {
 
    Amg::Vector3D beamSpotPosition(0,0,0);
    if(m_useBeamSpotInfo){
       SG::ReadCondHandle<InDet::BeamSpotData> beamSpotHandle { m_beamSpotKey };
       beamSpotPosition = beamSpotHandle->beamPos();
    }
 
    std::vector<double> vectorOfProbs=m_NnClusterizationFactory->estimateNumberOfParticles(origCluster,beamSpotPosition);
 
    ATH_MSG_VERBOSE(" Got splitProbability, size of vector: " << vectorOfProbs.size() );
 
    if (vectorOfProbs.empty())
    {
      std::vector<double> vectorOfSplitProbs;
      vectorOfSplitProbs.push_back(-100);
      PixelClusterSplitProb  clusterSplitProb(vectorOfSplitProbs);
      ATH_MSG_VERBOSE(" Returning single split prob equal to -100 " );
      return clusterSplitProb;
    }
     
 
    ATH_MSG_VERBOSE( 
        " P(1): " << vectorOfProbs[0] << 
        " P(2): " << vectorOfProbs[1] << 
        " P(>=3): " << vectorOfProbs[2] );
 
 
    return compileSplitProbability(vectorOfProbs); 
  }

splitProbability() [2/2]

InDet::PixelClusterSplitProb InDet::NnPixelClusterSplitProbTool::splitProbability ( const InDet::PixelCluster &  origCluster, const Trk::TrackParameters &  trackParameters  ) const

overridevirtual

Definition at line 83 of file NnPixelClusterSplitProbTool.cxx.

  {
    
    Amg::Vector3D beamSpotPosition(0,0,0);
    if(m_useBeamSpotInfo){
       SG::ReadCondHandle<InDet::BeamSpotData> beamSpotHandle { m_beamSpotKey };
       beamSpotPosition = beamSpotHandle->beamPos();
    }
 
    std::vector<double> vectorOfProbs=m_NnClusterizationFactory->estimateNumberOfParticles(origCluster, trackParameters.associatedSurface(), trackParameters);
 
    ATH_MSG_VERBOSE(" Got splitProbability, size of vector: " << vectorOfProbs.size() );
 
    if (vectorOfProbs.empty())
    {
      std::vector<double> vectorOfSplitProbs;
      vectorOfSplitProbs.push_back(-100);
      PixelClusterSplitProb  clusterSplitProb(vectorOfSplitProbs);
      ATH_MSG_VERBOSE(" Returning single split prob equal to -100 " );
      return clusterSplitProb;
    }
     
 
    ATH_MSG_VERBOSE( 
        " P(1): " << vectorOfProbs[0] << 
        " P(2): " << vectorOfProbs[1] << 
        " P(>=3): " << vectorOfProbs[2] );
 
 
    return compileSplitProbability(vectorOfProbs); 
  }

Member Data Documentation

m_beamSpotKey

SG::ReadCondHandleKey<InDet::BeamSpotData> InDet::NnPixelClusterSplitProbTool::m_beamSpotKey { this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot" }

private

Definition at line 56 of file NnPixelClusterSplitProbTool.h.

m_NnClusterizationFactory

ToolHandle<NnClusterizationFactory> InDet::NnPixelClusterSplitProbTool::m_NnClusterizationFactory { this, "NnClusterizationFactory", "InDet::NnClusterizationFactory/NnClusterizationFactory" }

private

Definition at line 55 of file NnPixelClusterSplitProbTool.h.

m_priorMultiplicityContent

DoubleArrayProperty InDet::NnPixelClusterSplitProbTool::m_priorMultiplicityContent { this, "PriorMultiplicityContent", {2793337, 82056, 19944} }

private

Definition at line 58 of file NnPixelClusterSplitProbTool.h.

m_useBeamSpotInfo

BooleanProperty InDet::NnPixelClusterSplitProbTool::m_useBeamSpotInfo { this, "useBeamSpotInfo", true }

private

Definition at line 60 of file NnPixelClusterSplitProbTool.h.

---

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

• NnPixelClusterSplitProbTool.h
• NnPixelClusterSplitProbTool.cxx