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.
virtual	~AFP_VertexRecoBasic ()=default
virtual StatusCode	initialize () override
	Loads parameterization.
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.
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.
void	linkToFTracksToVertex (const xAOD::AFPToFTrack toFTrack, SG::ReadHandle< xAOD::AFPToFTrackContainer > &tofTrackContainer, xAOD::AFPVertex vertex) const
	Links ToF track pair to reconstructed vertex.
void	linkProtonsToVertex (const xAOD::AFPProton proton, SG::ReadHandle< xAOD::AFPProtonContainer > &protonContainer, xAOD::AFPVertex vertex) const
	Links proton pair to reconstructed vertex.

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

AFP_VertexRecoBasic::m_vertexContainerName

Gaudi::Property< std::string > m_vertexContainerName

Definition AFP_VertexRecoBasic.h:83

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

81{this, "AFPProtonContainerKey", "AFPProtonContainer", "Name of the container with protons from which vertices are to be reconstructed"};

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

79{this, "AFPToFTrackContainerKey", "AFPToFTrackContainer", "Name of the container with ToF tracks from which vertices are to be reconstructed"};

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

75{this, "tofVtxParamDBTool", "AFP__ToFLocParamDBTool", "Tool to access DB to get the vertex ToF parameters"};

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

77{this, "TrackDistance", 0.5, "Maximum distance between ToF train edge and proton"};

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

83{this, "verticesContainerName", "AFPVertexContainer", "Name of the container in which vertices are saved"};

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