Tool for vertex reconstruction by basic algorithm. More...

#include <AFP_VertexRecoBasic.h>

Inheritance diagram for AFP_VertexRecoBasic:

Collaboration diagram for AFP_VertexRecoBasic:

Public Member Functions
	AFP_VertexRecoBasic (const std::string &type, const std::string &name, const IInterface *parent)
	Default constructor. More...

virtual	~AFP_VertexRecoBasic ()=default

virtual StatusCode	initialize () override
	Loads parameterization. More...

const std::string &	outputContainerName () const override

StatusCode	doVertexReco (std::unique_ptr< xAOD::AFPVertexContainer > &outputContainer, const EventContext &ctx) const override

Private Member Functions
StatusCode	configInfo () const

xAOD::AFPVertex *	reco (const double distA, const double distC, const xAOD::AFPToFTrack toFTrackA, const xAOD::AFPToFTrack toFTrackC, const xAOD::AFPProton protonA, const xAOD::AFPProton protonC, const AFP::ToFVtxParamData &TVP_A, const AFP::ToFVtxParamData &TVP_C, std::unique_ptr< xAOD::AFPVertexContainer > &outputContainer) const
	Reconstructs single vertex from pair of ToFTracks and a pair of protons. More...

xAOD::AFPVertex *	createVertex (const double position, const double distA, const double distC, std::unique_ptr< xAOD::AFPVertexContainer > &outputContainer) const
	Creates and sets up a vertex. More...

void	linkToFTracksToVertex (const xAOD::AFPToFTrack toFTrack, SG::ReadHandle< xAOD::AFPToFTrackContainer > &tofTrackContainer, xAOD::AFPVertex vertex) const
	Links ToF track pair to reconstructed vertex. More...

void	linkProtonsToVertex (const xAOD::AFPProton proton, SG::ReadHandle< xAOD::AFPProtonContainer > &protonContainer, xAOD::AFPVertex vertex) const
	Links proton pair to reconstructed vertex. More...

Private Attributes
ToolHandle< AFP::IToFVtxParamDBTool >	m_tofVtxParamDBTool {this, "tofVtxParamDBTool", "AFP__ToFLocParamDBTool", "Tool to access DB to get the vertex ToF parameters"}
	@ brief Tool for accessing DB to get the vertex ToF parameters More...

Gaudi::Property< double >	m_trackDistance {this, "TrackDistance", 0.5, "Maximum distance between ToF train edge and proton"}

SG::ReadHandleKey< xAOD::AFPToFTrackContainer >	m_tofTrackContainerKey {this, "AFPToFTrackContainerKey", "AFPToFTrackContainer", "Name of the container with ToF tracks from which vertices are to be reconstructed"}

SG::ReadHandleKey< xAOD::AFPProtonContainer >	m_protonContainerKey {this, "AFPProtonContainerKey", "AFPProtonContainer", "Name of the container with protons from which vertices are to be reconstructed"}

Gaudi::Property< std::string >	m_vertexContainerName {this, "verticesContainerName", "AFPVertexContainer", "Name of the container in which vertices are saved"}

Detailed Description

Tool for vertex reconstruction by basic algorithm.

Definition at line 39 of file AFP_VertexRecoBasic.h.

Constructor & Destructor Documentation

◆ AFP_VertexRecoBasic()

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

Default constructor.

Definition at line 9 of file AFP_VertexRecoBasic.cxx.

  : base_class(type, name, parent)
 {
   ATH_MSG_DEBUG("in AFP_VertexRecoBasic constructor");
 }

◆ ~AFP_VertexRecoBasic()

virtual AFP_VertexRecoBasic::~AFP_VertexRecoBasic ( )

virtualdefault

Member Function Documentation

◆ configInfo()

StatusCode AFP_VertexRecoBasic::configInfo ( ) const

private

Definition at line 16 of file AFP_VertexRecoBasic.cxx.

 {
   ATH_MSG_INFO("----- AFP_VertexRecoBasic -----");
  
   ATH_MSG_INFO("\tCuts:\n");
   ATH_MSG_INFO("\t\ttrackDistance [mm]: " << m_trackDistance );
   
   CHECK( m_tofVtxParamDBTool.retrieve() );
   
   return StatusCode::SUCCESS;
 }

◆ createVertex()

xAOD::AFPVertex * AFP_VertexRecoBasic::createVertex	(	const double	position,
		const double	distA,
		const double	distC,
		std::unique_ptr< xAOD::AFPVertexContainer > &	outputContainer
	)		const

private

Creates and sets up a vertex.

Definition at line 158 of file AFP_VertexRecoBasic.cxx.

 {
  
   auto * vertex = outputContainer->push_back(std::make_unique<xAOD::AFPVertex>());
   
   // Set vertex properties
  
   vertex->setPosition(position);
   vertex->setDistA(distA);
   vertex->setDistC(distC);
   vertex->setAlgID(0);
   
   ATH_MSG_DEBUG("Reconstructed  AFP vertex (position): " << vertex->position() <<", distA "<<vertex->distA()<<", distC "<<vertex->distC());
  
   return vertex;
 }

◆ doVertexReco()

StatusCode AFP_VertexRecoBasic::doVertexReco	(	std::unique_ptr< xAOD::AFPVertexContainer > &	outputContainer,
		const EventContext &	ctx
	)		const

override

Definition at line 38 of file AFP_VertexRecoBasic.cxx.

 {
   
   SG::ReadHandle<xAOD::AFPToFTrackContainer> tofTrackContainer( m_tofTrackContainerKey, ctx );
   if(!tofTrackContainer.isValid())
   {
     // this is allowed, there might be no AFP data in the input
     return StatusCode::SUCCESS;
   }
   
  
   SG::ReadHandle<xAOD::AFPProtonContainer> protonContainer( m_protonContainerKey, ctx );
   if(!protonContainer.isValid())
   {
     // this is allowed, there might be no AFP data in the input
     return StatusCode::SUCCESS;
   }
  
  
   // Select ToF tracks on side A
   std::vector<const xAOD::AFPToFTrack*> tofTrackSideAContainer;
   std::copy_if(tofTrackContainer->begin(), tofTrackContainer->end(), std::back_inserter(tofTrackSideAContainer),
       [](auto track) { return track->stationID() == 0; });
   if(tofTrackSideAContainer.empty()) return StatusCode::SUCCESS;
   
   // Select ToF tracks on side C      
   std::vector<const xAOD::AFPToFTrack*> tofTrackSideCContainer;
   std::copy_if(tofTrackContainer->begin(), tofTrackContainer->end(), std::back_inserter(tofTrackSideCContainer),
       [](auto track) { return track->stationID() == 3; });
   if(tofTrackSideCContainer.empty()) return StatusCode::SUCCESS;
  
   // Select protons on side A      
   std::vector<const xAOD::AFPProton*> protonSideAContainer;
   std::copy_if(protonContainer->begin(), protonContainer->end(), std::back_inserter(protonSideAContainer),
       [](auto proton) { return proton->side() == 0; });
   if(protonSideAContainer.empty()) return StatusCode::SUCCESS;
   
   // Select protons on side C
   std::vector<const xAOD::AFPProton*> protonSideCContainer;
   std::copy_if(protonContainer->begin(), protonContainer->end(), std::back_inserter(protonSideCContainer),
       [](auto proton) { return proton->side() == 1; });
   if(protonSideCContainer.empty()) return StatusCode::SUCCESS;
   
   ATH_MSG_DEBUG("tofTrackSideAContainer size: " << tofTrackSideAContainer.size());
   ATH_MSG_DEBUG("tofTrackSideCContainer size: " << tofTrackSideCContainer.size());
   ATH_MSG_DEBUG("protonSideAContainer size: " << protonSideAContainer.size());
   ATH_MSG_DEBUG("protonSideCContainer size: " << protonSideCContainer.size());
   
   
   nlohmann::json dataTVP=m_tofVtxParamDBTool->parametersData(ctx);
   const AFP::ToFVtxParamData TVP_A=m_tofVtxParamDBTool->parameters(dataTVP, 0);
   const AFP::ToFVtxParamData TVP_C=m_tofVtxParamDBTool->parameters(dataTVP, 3);
   
   // Loop over four containers
   for (const xAOD::AFPToFTrack* tofTrackSideA : tofTrackSideAContainer) {
  
     for (const xAOD::AFPProton* protonSideA : protonSideAContainer) {
       // Apply cuts
  
       double protonXPositionFar = protonSideA->track(0)->xLocal();
       if(protonSideA->nTracks()==2 && protonSideA->track(1)->stationID()==0) 
                                    protonXPositionFar = protonSideA->track(1)->xLocal(); 
       
       double dx = std::min(std::abs(protonXPositionFar-TVP_A.trainEdge(tofTrackSideA->trainID())),std::abs(protonXPositionFar-TVP_A.trainEdge(tofTrackSideA->trainID()+1))); 
       double distA = dx;
       if( protonXPositionFar > TVP_A.trainEdge(tofTrackSideA->trainID()) && protonXPositionFar < TVP_A.trainEdge(tofTrackSideA->trainID()+1) ) distA = -dx;
  
       if (distA > m_trackDistance) {
         ATH_MSG_DEBUG(
             "Tracks too far away from each other (xProton, xLeftPositionSideA; xRightPositionSideA; distance) [mm]: "
             << protonXPositionFar << ", " << TVP_A.trainEdge(tofTrackSideA->trainID()) << "; "
             << TVP_A.trainEdge(tofTrackSideA->trainID()+1)  << ", " << distA  << "; " << distA);
  
         continue;
       }
  
  
  
      for (const xAOD::AFPToFTrack* tofTrackSideC : tofTrackSideCContainer) {
  
        for (const xAOD::AFPProton* protonSideC : protonSideCContainer) {
          // Apply cuts
  
          double protonXPositionFar = protonSideC->track(0)->xLocal();
          if(protonSideC->nTracks()==2 && protonSideC->track(1)->stationID()==3) 
                                    protonXPositionFar = protonSideC->track(1)->xLocal();
          double dx = std::min(std::abs(protonXPositionFar-TVP_C.trainEdge(tofTrackSideC->trainID())),std::abs(protonXPositionFar-TVP_C.trainEdge(tofTrackSideC->trainID()+1)));
          double distC = dx;
          if( protonXPositionFar > TVP_C.trainEdge(tofTrackSideC->trainID()) && protonXPositionFar < TVP_C.trainEdge(tofTrackSideC->trainID()+1) ) distC = -dx;
  
          if (distC > m_trackDistance) {
             ATH_MSG_DEBUG(
               "Tracks too far away from each other (xProton, xLeftPositionSideC; xRightPositionSideC; distance) [mm]: "
               << protonXPositionFar << ", " << TVP_C.trainEdge(tofTrackSideC->trainID()) << "; "
               << TVP_C.trainEdge(tofTrackSideC->trainID()+1)  << ", " << distC  << "; " << distC);
  
            continue;
          }
  
            // Reconstruct vertex and add it to the container
            xAOD::AFPVertex * vertex = reco(distA, distC, tofTrackSideA, tofTrackSideC, protonSideA, protonSideC, TVP_A, TVP_C, outputContainer);
  
            if (!vertex)
               continue;
  
  
            // Create link to tracks
            linkToFTracksToVertex(tofTrackSideA, tofTrackContainer, vertex);
            linkToFTracksToVertex(tofTrackSideC, tofTrackContainer, vertex);
            linkProtonsToVertex(protonSideA, protonContainer,vertex);
            linkProtonsToVertex(protonSideC, protonContainer,vertex);
         }
       }
     }
   }
   
   return StatusCode::SUCCESS;
 }

◆ initialize()

StatusCode AFP_VertexRecoBasic::initialize ( )

overridevirtual

Loads parameterization.

Definition at line 29 of file AFP_VertexRecoBasic.cxx.

 {
  
   CHECK( m_tofTrackContainerKey.initialize() );
   CHECK( m_protonContainerKey.initialize() );
  
   return StatusCode::SUCCESS;
 }

◆ linkProtonsToVertex()

void AFP_VertexRecoBasic::linkProtonsToVertex	(	const xAOD::AFPProton *	proton,
		SG::ReadHandle< xAOD::AFPProtonContainer > &	protonContainer,
		xAOD::AFPVertex *	vertex
	)		const

private

Links proton pair to reconstructed vertex.

Definition at line 185 of file AFP_VertexRecoBasic.cxx.

                                                                                                                      {
  
   ElementLink<xAOD::AFPProtonContainer> protonLink;
  
   protonLink.toContainedElement(*protonContainer, proton);
   vertex->addProton(protonLink);
 }

◆ linkToFTracksToVertex()

void AFP_VertexRecoBasic::linkToFTracksToVertex	(	const xAOD::AFPToFTrack *	toFTrack,
		SG::ReadHandle< xAOD::AFPToFTrackContainer > &	tofTrackContainer,
		xAOD::AFPVertex *	vertex
	)		const

private

Links ToF track pair to reconstructed vertex.

Definition at line 176 of file AFP_VertexRecoBasic.cxx.

                                                                                                                              {
  
   ElementLink<xAOD::AFPToFTrackContainer> tofTrackLink;
  
   tofTrackLink.toContainedElement(*tofTrackContainer, tofTrack);
   vertex->addToFTrack(tofTrackLink);
 }

◆ outputContainerName()

const std::string& AFP_VertexRecoBasic::outputContainerName ( ) const

inlineoverride

Definition at line 51 of file AFP_VertexRecoBasic.h.

51 {return m_vertexContainerName;}

◆ reco()

xAOD::AFPVertex * AFP_VertexRecoBasic::reco	(	const double	distA,
		const double	distC,
		const xAOD::AFPToFTrack *	toFTrackA,
		const xAOD::AFPToFTrack *	toFTrackC,
		const xAOD::AFPProton *	protonA,
		const xAOD::AFPProton *	protonC,
		const AFP::ToFVtxParamData &	TVP_A,
		const AFP::ToFVtxParamData &	TVP_C,
		std::unique_ptr< xAOD::AFPVertexContainer > &	outputContainer
	)		const

private

Reconstructs single vertex from pair of ToFTracks and a pair of protons.

Sets up measurement and calculates distances and times
Reconstucts vertex using basic algorithm
Calculates initial matching quality
Adds vertex to outputContainer and sets it's properties

Returns: Poiner to reconstucted AFPVertex

Definition at line 197 of file AFP_VertexRecoBasic.cxx.

 {
   int trainID = tofTrackSideA->trainID();
   double protonYPositionFar = protonSideA->track(0)->yLocal();
   if(protonSideA->nTracks()==2 && protonSideA->track(1)->stationID()==0)
     protonYPositionFar = protonSideA->track(1)->yLocal();
   double timeA = tofTrackSideA->trainTime() - (TVP_A.timeOffset(trainID)+TVP_A.timeSlope(trainID)*protonYPositionFar);
  
   trainID = tofTrackSideC->trainID();
   protonYPositionFar = protonSideC->track(0)->yLocal();
   if(protonSideC->nTracks()==2 && protonSideC->track(1)->stationID()==3)
     protonYPositionFar = protonSideC->track(1)->yLocal();
  
   double timeC = tofTrackSideC->trainTime() - (TVP_C.timeOffset(trainID)+TVP_C.timeSlope(trainID)*protonYPositionFar);
  
   double position = ( (timeC*1000.-TVP_C.timeGlobalOffset()) - (timeA*1000.-TVP_A.timeGlobalOffset()))/2.*0.299792458 ; 
  
   return createVertex(position, distA, distC, outputContainer);
 }

Member Data Documentation

◆ m_protonContainerKey

SG::ReadHandleKey<xAOD::AFPProtonContainer> AFP_VertexRecoBasic::m_protonContainerKey {this, "AFPProtonContainerKey", "AFPProtonContainer", "Name of the container with protons from which vertices are to be reconstructed"}

private

Definition at line 81 of file AFP_VertexRecoBasic.h.

◆ m_tofTrackContainerKey

SG::ReadHandleKey<xAOD::AFPToFTrackContainer> AFP_VertexRecoBasic::m_tofTrackContainerKey {this, "AFPToFTrackContainerKey", "AFPToFTrackContainer", "Name of the container with ToF tracks from which vertices are to be reconstructed"}

private

Definition at line 79 of file AFP_VertexRecoBasic.h.

◆ m_tofVtxParamDBTool

ToolHandle<AFP::IToFVtxParamDBTool> AFP_VertexRecoBasic::m_tofVtxParamDBTool {this, "tofVtxParamDBTool", "AFP__ToFLocParamDBTool", "Tool to access DB to get the vertex ToF parameters"}

private

@ brief Tool for accessing DB to get the vertex ToF parameters

Definition at line 75 of file AFP_VertexRecoBasic.h.

◆ m_trackDistance

Gaudi::Property<double> AFP_VertexRecoBasic::m_trackDistance {this, "TrackDistance", 0.5, "Maximum distance between ToF train edge and proton"}

private

Definition at line 77 of file AFP_VertexRecoBasic.h.

◆ m_vertexContainerName

Gaudi::Property<std::string> AFP_VertexRecoBasic::m_vertexContainerName {this, "verticesContainerName", "AFPVertexContainer", "Name of the container in which vertices are saved"}

private

Definition at line 83 of file AFP_VertexRecoBasic.h.

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

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Constructor & Destructor Documentation

◆ AFP_VertexRecoBasic()

◆ ~AFP_VertexRecoBasic()

Member Function Documentation

◆ configInfo()

◆ createVertex()

◆ doVertexReco()

◆ initialize()

◆ linkProtonsToVertex()

◆ linkToFTracksToVertex()

◆ outputContainerName()

◆ reco()

Member Data Documentation

◆ m_protonContainerKey

◆ m_tofTrackContainerKey

◆ m_tofVtxParamDBTool

◆ m_trackDistance

◆ m_vertexContainerName