Base class for all proton reconstruction tools. More...

#include <AFP_ProtonRecoBase.h>

Inheritance diagram for AFP_ProtonRecoBase:

Collaboration diagram for AFP_ProtonRecoBase:

Classes
struct	Measurement
	Local class for storing tracks positions. More...

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

const std::string &	outputContainerName () const override

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

Protected Types
using	Momentum = std::array< double, 3 >
	3-momentum of reconstructed proton More...

Protected Member Functions
xAOD::AFPProton *	createProton (const Momentum &momentum, const Measurement &my_measAFP, const int algID, std::unique_ptr< xAOD::AFPProtonContainer > &outputContainer) const
	Creates and sets up a proton. More...

void	linkTracksToProton (const xAOD::AFPTrack track, SG::ReadHandle< xAOD::AFPTrackContainer > &trackContainer, xAOD::AFPProton proton) const
	Links track pair to reconstructed proton. More...

virtual xAOD::AFPProton *	reco (const xAOD::AFPTrack , const xAOD::AFPTrack , std::unique_ptr< xAOD::AFPProtonContainer > &) const

virtual xAOD::AFPProton *	reco (const xAOD::AFPTrack *, std::unique_ptr< xAOD::AFPProtonContainer > &) const

virtual double	chi2 (double, double, double, const Measurement &) const

Protected Attributes
Gaudi::Property< std::vector< double > >	m_detectorPositions {this, "detectorPositions", {}, "absolute values of detector positions for each station on one side"}

Gaudi::Property< int >	m_side {this, "side", 0, "side id, A=0, C=1"}

Gaudi::Property< double >	m_trackDistance {this, "trackDistance", 2.0, "Maximum distance between tracks in the near and the far station on xy-plane"}

Gaudi::Property< bool >	m_allowSingleStationReco {this, "allowSingleStationReco", false, "Switch for allowing proton reconstruction using only far station"}

SG::ReadHandleKey< xAOD::AFPTrackContainer >	m_trackContainerKey {this, "AFPTrackContainerKey", "AFPTrackContainer", "Name of the container with tracks of hits from which protons are to be reconstructed"}

Gaudi::Property< std::string >	m_protonsContainerName {this, "protonsContainerName", "AFPProtonContainer", "Name of the container in which protons are saved"}

double	m_detectorPositionNear = 0.0
	Default position of AFP near station. More...

double	m_detectorPositionFar = 0.0
	Default position of AFP far station. More...

const std::vector< double >	m_vertexIP = {0, 0, 0}
	Vertex position. More...

Static Protected Attributes
static constexpr double	m_xSigma = 10e-6
	x-Sigma value More...

static constexpr double	m_ySigma = 30e-6
	y-Sigma value More...

Detailed Description

Base class for all proton reconstruction tools.

Definition at line 34 of file AFP_ProtonRecoBase.h.

Member Typedef Documentation

◆ Momentum

using AFP_ProtonRecoBase::Momentum = std::array<double, 3>

protected

3-momentum of reconstructed proton

Definition at line 83 of file AFP_ProtonRecoBase.h.

Constructor & Destructor Documentation

◆ AFP_ProtonRecoBase()

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

Default constructor.

Definition at line 9 of file AFP_ProtonRecoBase.cxx.

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

Member Function Documentation

◆ chi2()

virtual double AFP_ProtonRecoBase::chi2	(	double	,
		double	,
		double	,
		const Measurement &
	)		const

inlineprotectedvirtual

Reimplemented in AFP_ProtonRecoAnalytical.

Definition at line 120 of file AFP_ProtonRecoBase.h.

     {
       ATH_MSG_ERROR("this is chi2(energy, sx, sy, sz, my_measAFP) in AFP_ProtonRecoBase, shall not be used, returning 0");
       return 0.;
     }

◆ createProton()

xAOD::AFPProton * AFP_ProtonRecoBase::createProton	(	const Momentum &	momentum,
		const Measurement &	my_measAFP,
		const int	algID,
		std::unique_ptr< xAOD::AFPProtonContainer > &	outputContainer
	)		const

protected

Creates and sets up a proton.

Definition at line 94 of file AFP_ProtonRecoBase.cxx.

 {
  
   // Return nullptr if any of momentum components is not a number
   if ( std::any_of(begin(momentum), end(momentum), [](auto& el) { return !std::isfinite(el); }) )
     return nullptr;
  
   const auto [px, py, pz] = momentum;
  
   auto * proton = outputContainer->push_back(std::make_unique<xAOD::AFPProton>());
   
   // Set proton properties
   constexpr double protonMass = 0.938; // in GeV
   
   proton->setPxPyPzE(px, py, pz, sqrt(px*px + py*py + pz*pz + protonMass*protonMass));
   proton->setChi2(chi2(px, py, pz, my_measAFP));
   proton->setSide(m_side);
   proton->setMethodID(algID);
   
   ATH_MSG_DEBUG("Reconstructed proton (px, py, pz): " << proton->px() << ", " << proton->py() << ", " << proton->pz()<<", chi2 "<<proton->chi2()<<", side "<<proton->side());
  
   return proton;
 }

◆ doProtonReco()

StatusCode AFP_ProtonRecoBase::doProtonReco	(	std::unique_ptr< xAOD::AFPProtonContainer > &	outputContainer,
		const EventContext &	ctx
	)		const

override

Definition at line 16 of file AFP_ProtonRecoBase.cxx.

 {
   
   SG::ReadHandle<xAOD::AFPTrackContainer> trackContainer( m_trackContainerKey, ctx );
   if(!trackContainer.isValid())
   {
     // this is allowed, there might be no AFP data in the input
     return StatusCode::SUCCESS;
   }
   
   const double trackDistanceRadiusSq = m_trackDistance*m_trackDistance;
  
   // Select tracks in near station
   std::vector<const xAOD::AFPTrack*> trackNearContainer;
   const int nearId = m_side + 1;
   std::copy_if(trackContainer->begin(), trackContainer->end(), std::back_inserter(trackNearContainer),
       [&nearId](auto track) { return track->stationID() == nearId; });
  
   // Select tracks in far station
   std::vector<const xAOD::AFPTrack*> trackFarContainer;
   const int farId = 3 * m_side;
   std::copy_if(trackContainer->begin(), trackContainer->end(), std::back_inserter(trackFarContainer),
       [&farId](auto track) { return track->stationID() == farId; });
  
   ATH_MSG_DEBUG("trackNearContainer size: " << trackNearContainer.size()<<", side "<<m_side);
   ATH_MSG_DEBUG("trackFarContainer  size: " << trackFarContainer.size()<<", side "<<m_side);
  
   // Loop over both containers
   for (const xAOD::AFPTrack* trackFar : trackFarContainer) {
     bool foundMatchingTrack = false;
  
     for (const xAOD::AFPTrack* trackNear : trackNearContainer) {
       // Apply cuts
       const double dx = trackFar->xLocal() - trackNear->xLocal();
       const double dy = trackFar->yLocal() - trackNear->yLocal();
       const double r2 = dx*dx + dy*dy;
  
       if (r2 > trackDistanceRadiusSq) {
         ATH_MSG_DEBUG(
             "Tracks too far away from each other (xNear, yNear; xFar, yFar; distance) [mm]: "
             << trackNear->xLocal() << ", " << trackNear->yLocal() << "; "
             << trackFar->xLocal()  << ", " << trackFar->yLocal()  << "; " << r2);
  
         continue;
       }
  
       // Reconstruct proton and add it to the container
       xAOD::AFPProton * proton = reco(trackNear, trackFar, outputContainer);
  
       if (!proton)
         continue;
  
       foundMatchingTrack = true;
  
       // Create link to tracks
       linkTracksToProton(trackNear, trackContainer, proton);
       linkTracksToProton(trackFar, trackContainer, proton);
     }
  
     // Reconstuct proton using only FAR station if
     // no matching track on NEAR station was found
     if (m_allowSingleStationReco and !foundMatchingTrack) {
       // Apply cuts
       // none
  
       xAOD::AFPProton * proton = reco(trackFar, outputContainer);
  
       if (!proton)
         continue;
  
       linkTracksToProton(trackFar, trackContainer, proton);
     }
   }
   
   return StatusCode::SUCCESS;
 }

◆ linkTracksToProton()

void AFP_ProtonRecoBase::linkTracksToProton	(	const xAOD::AFPTrack *	track,
		SG::ReadHandle< xAOD::AFPTrackContainer > &	trackContainer,
		xAOD::AFPProton *	proton
	)		const

protected

Links track pair to reconstructed proton.

Definition at line 119 of file AFP_ProtonRecoBase.cxx.

                                                                                                                  {
  
   ElementLink<xAOD::AFPTrackContainer> trackLink;
  
   trackLink.toContainedElement(*trackContainer, track);
   proton->addAFPTrackLink(trackLink);
 }

◆ outputContainerName()

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

inlineoverride

Definition at line 41 of file AFP_ProtonRecoBase.h.

41 {return m_protonsContainerName;}

◆ reco() [1/2]

virtual xAOD::AFPProton* AFP_ProtonRecoBase::reco	(	const xAOD::AFPTrack *	,
		const xAOD::AFPTrack *	,
		std::unique_ptr< xAOD::AFPProtonContainer > &
	)		const

inlineprotectedvirtual

Reimplemented in AFP_ProtonRecoAnalytical.

Definition at line 108 of file AFP_ProtonRecoBase.h.

     {
       ATH_MSG_ERROR("this is reco(trkNear, trkFar, outputContainer) in AFP_ProtonRecoBase, shall not be used, returning 0");
       return 0;
     }

◆ reco() [2/2]

virtual xAOD::AFPProton* AFP_ProtonRecoBase::reco	(	const xAOD::AFPTrack *	,
		std::unique_ptr< xAOD::AFPProtonContainer > &
	)		const

inlineprotectedvirtual

Reimplemented in AFP_ProtonRecoAnalytical.

Definition at line 113 of file AFP_ProtonRecoBase.h.

     {
       ATH_MSG_ERROR("this is reco(trkFar, outputContainer) in AFP_ProtonRecoBase, shall not be used, returning 0");
       return 0;
     }

Member Data Documentation

◆ m_allowSingleStationReco

Gaudi::Property<bool> AFP_ProtonRecoBase::m_allowSingleStationReco {this, "allowSingleStationReco", false, "Switch for allowing proton reconstruction using only far station"}

protected

Definition at line 76 of file AFP_ProtonRecoBase.h.

◆ m_detectorPositionFar

double AFP_ProtonRecoBase::m_detectorPositionFar = 0.0

protected

Default position of AFP far station.

Definition at line 95 of file AFP_ProtonRecoBase.h.

◆ m_detectorPositionNear

double AFP_ProtonRecoBase::m_detectorPositionNear = 0.0

protected

Default position of AFP near station.

Definition at line 92 of file AFP_ProtonRecoBase.h.

◆ m_detectorPositions

Gaudi::Property<std::vector<double> > AFP_ProtonRecoBase::m_detectorPositions {this, "detectorPositions", {}, "absolute values of detector positions for each station on one side"}

protected

Definition at line 70 of file AFP_ProtonRecoBase.h.

◆ m_protonsContainerName

Gaudi::Property<std::string> AFP_ProtonRecoBase::m_protonsContainerName {this, "protonsContainerName", "AFPProtonContainer", "Name of the container in which protons are saved"}

protected

Definition at line 80 of file AFP_ProtonRecoBase.h.

◆ m_side

Gaudi::Property<int> AFP_ProtonRecoBase::m_side {this, "side", 0, "side id, A=0, C=1"}

protected

Definition at line 72 of file AFP_ProtonRecoBase.h.

◆ m_trackContainerKey

SG::ReadHandleKey<xAOD::AFPTrackContainer> AFP_ProtonRecoBase::m_trackContainerKey {this, "AFPTrackContainerKey", "AFPTrackContainer", "Name of the container with tracks of hits from which protons are to be reconstructed"}

protected

Definition at line 78 of file AFP_ProtonRecoBase.h.

◆ m_trackDistance

Gaudi::Property<double> AFP_ProtonRecoBase::m_trackDistance {this, "trackDistance", 2.0, "Maximum distance between tracks in the near and the far station on xy-plane"}

protected

Definition at line 74 of file AFP_ProtonRecoBase.h.

◆ m_vertexIP

const std::vector<double> AFP_ProtonRecoBase::m_vertexIP = {0, 0, 0}

protected

Vertex position.

Definition at line 105 of file AFP_ProtonRecoBase.h.

◆ m_xSigma

constexpr double AFP_ProtonRecoBase::m_xSigma = 10e-6

staticconstexprprotected

x-Sigma value

Definition at line 98 of file AFP_ProtonRecoBase.h.

◆ m_ySigma

constexpr double AFP_ProtonRecoBase::m_ySigma = 30e-6

staticconstexprprotected

y-Sigma value

Definition at line 101 of file AFP_ProtonRecoBase.h.

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

Classes

Public Member Functions

Protected Types

Protected Member Functions

Protected Attributes

Static Protected Attributes

Detailed Description

Member Typedef Documentation

◆ Momentum

Constructor & Destructor Documentation

◆ AFP_ProtonRecoBase()

Member Function Documentation

◆ chi2()

◆ createProton()

◆ doProtonReco()

◆ linkTracksToProton()

◆ outputContainerName()

◆ reco() [1/2]

◆ reco() [2/2]

Member Data Documentation

◆ m_allowSingleStationReco

◆ m_detectorPositionFar

◆ m_detectorPositionNear

◆ m_detectorPositions

◆ m_protonsContainerName

◆ m_side

◆ m_trackContainerKey

◆ m_trackDistance

◆ m_vertexIP

◆ m_xSigma

◆ m_ySigma