#include <ALFA_BeamTransport.h>

Inheritance diagram for ALFA_BeamTransport:

Collaboration diagram for ALFA_BeamTransport:

Public Member Functions
	ALFA_BeamTransport (const std::string &name, ISvcLocator *pSvcLocator)
	Standard Athena-Algorithm Constructor. More...

	~ALFA_BeamTransport ()
	Default Destructor. More...

StatusCode	initialize ()
	standard Athena-Algorithm method More...

StatusCode	execute ()
	standard Athena-Algorithm method More...

StatusCode	finalize ()
	standard Athena-Algorithm method More...

void	MeVToGeV (HepMC::GenEvent &evt)
	convert unit MeV to GeV for energy and momenta More...

void	GeVToMeV (HepMC::GenEvent &evt)
	convert GeV to MeV for HepMC event record More...

int	SelectParticles (HepMC::GenEvent *evt)
	Selects particles for beam transported Sets event status code of outgoing particles from generator to status != 1. More...

int	DoBeamTracking (int evt_number)
	Function which calls BeamTrack class to calcualte Position at RPs. More...

int	TransportSelectedParticle (HepMC::GenEvent &evt, int evt_number)
	This function is event selection, tracking and HepMC save ine one function. More...

virtual StatusCode	sysInitialize () override
	Override sysInitialize. More...

virtual const DataObjIDColl &	extraOutputDeps () const override
	Return the list of extra output dependencies. More...

ServiceHandle< StoreGateSvc > &	evtStore ()
	The standard `StoreGateSvc` (event store) Returns (kind of) a pointer to the `StoreGateSvc`. More...

const ServiceHandle< StoreGateSvc > &	evtStore () const
	The standard `StoreGateSvc` (event store) Returns (kind of) a pointer to the `StoreGateSvc`. More...

const ServiceHandle< StoreGateSvc > &	detStore () const
	The standard `StoreGateSvc/DetectorStore` Returns (kind of) a pointer to the `StoreGateSvc`. More...

virtual StatusCode	sysStart () override
	Handle START transition. More...

virtual std::vector< Gaudi::DataHandle * >	inputHandles () const override
	Return this algorithm's input handles. More...

virtual std::vector< Gaudi::DataHandle * >	outputHandles () const override
	Return this algorithm's output handles. More...

Gaudi::Details::PropertyBase &	declareProperty (Gaudi::Property< T > &t)

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleKey &hndl, const std::string &doc, const SG::VarHandleKeyType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleBase &hndl, const std::string &doc, const SG::VarHandleType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleKeyArray &hndArr, const std::string &doc, const SG::VarHandleKeyArrayType &)

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, T &property, const std::string &doc, const SG::NotHandleType &)
	Declare a new Gaudi property. More...

Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, T &property, const std::string &doc="none")
	Declare a new Gaudi property. More...

void	updateVHKA (Gaudi::Details::PropertyBase &)

MsgStream &	msg () const

MsgStream &	msg (const MSG::Level lvl) const

bool	msgLvl (const MSG::Level lvl) const

Protected Member Functions
void	renounceArray (SG::VarHandleKeyArray &handlesArray)
	remove all handles from I/O resolution More...

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

Private Types
typedef ServiceHandle< StoreGateSvc >	StoreGateSvc_t

Private Member Functions
Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleKeyType &)
	specialization for handling Gaudi::Property<SG::VarHandleKey> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleKeyArrayType &)
	specialization for handling Gaudi::Property<SG::VarHandleKeyArray> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleType &)
	specialization for handling Gaudi::Property<SG::VarHandleBase> More...

Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &t, const SG::NotHandleType &)
	specialization for handling everything that's not a Gaudi::Property<SG::VarHandleKey> or a <SG::VarHandleKeyArray> More...

Private Attributes
SG::ReadHandleKey< McEventCollection >	m_MCKey

SG::ReadHandleKey< xAOD::EventInfo >	m_eventInfoKey

FPConfig	m_FPConfig

FPTracker::Particle	m_Particle1

FPTracker::Particle	m_Particle2

ALFA_BeamTrack	m_BeamTracker

FPTracker::Point	m_PosRP1

FPTracker::Point	m_PosRP3

FPTracker::Point	m_MomRP1

FPTracker::Point	m_MomRP3

double	m_EnergyRP1 = 0.0

double	m_EnergyRP3 = 0.0

std::string	m_FPOutputBeam1

std::string	m_FPOutputBeam2

int	m_pcount = 0

int	m_pint = 0

double	m_EtaCut

double	m_XiCut

std::ofstream	m_FileBeam1

std::ofstream	m_FileBeam2

bool	m_WriteDebugOutput

DataObjIDColl	m_extendedExtraObjects

StoreGateSvc_t	m_evtStore
	Pointer to StoreGate (event store by default) More...

StoreGateSvc_t	m_detStore
	Pointer to StoreGate (detector store by default) More...

std::vector< SG::VarHandleKeyArray * >	m_vhka

bool	m_varHandleArraysDeclared

Detailed Description

This is the ALFA Beam Transport package. It is an interface which uses FPTracker for beamtransportation. Getting from the HepMC eventrecord the particle data at the IP it calculates the particle porition at the first RP.

Author: Daniel Pelikan danie.nosp@m.l.pe.nosp@m.likan.nosp@m.@cer.nosp@m.n.ch

Definition at line 49 of file ALFA_BeamTransport.h.

Member Typedef Documentation

◆ StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< Algorithm > >::StoreGateSvc_t

privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

◆ ALFA_BeamTransport()

ALFA_BeamTransport::ALFA_BeamTransport	(	const std::string &	name,
		ISvcLocator *	pSvcLocator
	)

Standard Athena-Algorithm Constructor.

Definition at line 36 of file ALFA_BeamTransport.cxx.

   :
   AthAlgorithm(name,pSvcLocator)
 {
   declareProperty("ConfDir", m_FPConfig.ConfDir="./config");
   declareProperty("UseALFA", m_FPConfig.UseALFA=true);
   declareProperty("Debug", m_WriteDebugOutput=false);
   declareProperty("IP",m_FPConfig.IP = 1);
   declareProperty("UseAper",m_FPConfig.useaper = true);
   declareProperty("apermb",m_FPConfig.apermb = 0.);
   declareProperty("xcol1",m_FPConfig.xcol1 = 999.);//999e-4 but we need open collimators
   declareProperty("xcol2",m_FPConfig.xcol2 = 999.);//999e-4
   declareProperty("BeamEnergy",m_FPConfig.pbeam0 = 7000.);//beam energy
   declareProperty("RPDistance",m_FPConfig.RPDistance = 237.398);
   declareProperty("absZMagMax",m_FPConfig.absZMagMax = 500.);//dont read magnets after this value
   
   //Cuts for production
   declareProperty("EtaCut",m_EtaCut = 8.);
   declareProperty("XiCut",m_XiCut = 0.2); 
   
   declareProperty("FPOutputBeam1", m_FPOutputBeam1="Beam1");
   declareProperty("FPOutputBeam2", m_FPOutputBeam2="Beam2");
  
 }

◆ ~ALFA_BeamTransport()

ALFA_BeamTransport::~ALFA_BeamTransport ( )

Default Destructor.

Definition at line 61 of file ALFA_BeamTransport.cxx.

61 {}

Member Function Documentation

◆ declareGaudiProperty() [1/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Algorithm > >::declareGaudiProperty	(	Gaudi::Property< T > &	hndl,
		const SG::VarHandleKeyArrayType &
	)

inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleKeyArray>

Definition at line 170 of file AthCommonDataStore.h.

   {
     return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
                                                        hndl.value(), 
                                                        hndl.documentation());
  
   }

◆ declareGaudiProperty() [2/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Algorithm > >::declareGaudiProperty	(	Gaudi::Property< T > &	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());
  
   }

◆ declareGaudiProperty() [3/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Algorithm > >::declareGaudiProperty	(	Gaudi::Property< T > &	hndl,
		const SG::VarHandleType &
	)

inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleBase>

Definition at line 184 of file AthCommonDataStore.h.

   {
     return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
                                                        hndl.value(), 
                                                        hndl.documentation());
   }

◆ declareGaudiProperty() [4/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Algorithm > >::declareGaudiProperty	(	Gaudi::Property< T > &	t,
		const SG::NotHandleType &
	)

inlineprivateinherited

specialization for handling everything that's not a Gaudi::Property<SG::VarHandleKey> or a <SG::VarHandleKeyArray>

Definition at line 199 of file AthCommonDataStore.h.

   {
     return PBASE::declareProperty(t);
   }

◆ declareProperty() [1/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty	(	const std::string &	name,
		SG::VarHandleBase &	hndl,
		const std::string &	doc,
		const SG::VarHandleType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
hndl	Object holding the property value.
doc	Documentation string for the property.

This is the version for types that derive from SG::VarHandleBase. The property value object is put on the input and output lists as appropriate; then we forward to the base class.

Definition at line 245 of file AthCommonDataStore.h.

   {
     this->declare(hndl.vhKey());
     hndl.vhKey().setOwner(this);
  
     return PBASE::declareProperty(name,hndl,doc);
   }

◆ declareProperty() [2/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty	(	const std::string &	name,
		SG::VarHandleKey &	hndl,
		const std::string &	doc,
		const SG::VarHandleKeyType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
hndl	Object holding the property value.
doc	Documentation string for the property.

This is the version for types that derive from SG::VarHandleKey. The property value object is put on the input and output lists as appropriate; then we forward to the base class.

Definition at line 221 of file AthCommonDataStore.h.

   {
     this->declare(hndl);
     hndl.setOwner(this);
  
     return PBASE::declareProperty(name,hndl,doc);
   }

◆ declareProperty() [3/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty	(	const std::string &	name,
		SG::VarHandleKeyArray &	hndArr,
		const std::string &	doc,
		const SG::VarHandleKeyArrayType &
	)

inlineinherited

Definition at line 259 of file AthCommonDataStore.h.

   {
  
     // std::ostringstream ost;
     // ost << Algorithm::name() << " VHKA declareProp: " << name 
     //     << " size: " << hndArr.keys().size() 
     //     << " mode: " << hndArr.mode() 
     //     << "  vhka size: " << m_vhka.size()
     //     << "\n";
     // debug() << ost.str() << endmsg;
  
     hndArr.setOwner(this);
     m_vhka.push_back(&hndArr);
  
     Gaudi::Details::PropertyBase* p =  PBASE::declareProperty(name, hndArr, doc);
     if (p != 0) {
       p->declareUpdateHandler(&AthCommonDataStore<PBASE>::updateVHKA, this);
     } else {
       ATH_MSG_ERROR("unable to call declareProperty on VarHandleKeyArray " 
                     << name);
     }
  
     return p;
  
   }

◆ declareProperty() [4/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty	(	const std::string &	name,
		T &	property,
		const std::string &	doc,
		const SG::NotHandleType &
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
property	Object holding the property value.
doc	Documentation string for the property.

This is the generic version, for types that do not derive from SG::VarHandleKey. It just forwards to the base class version of declareProperty.

Definition at line 333 of file AthCommonDataStore.h.

   {
     return PBASE::declareProperty(name, property, doc);
   }

◆ declareProperty() [5/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty	(	const std::string &	name,
		T &	property,
		const std::string &	doc = `"none"`
	)

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
property	Object holding the property value.
doc	Documentation string for the property.

This dispatches to either the generic declareProperty or the one for VarHandle/Key/KeyArray.

Definition at line 352 of file AthCommonDataStore.h.

   {
     typedef typename SG::HandleClassifier<T>::type htype;
     return declareProperty (name, property, doc, htype());
   }

◆ declareProperty() [6/6]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty ( Gaudi::Property< T > & 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< Algorithm > >::detStore ( ) const

inlineinherited

The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 95 of file AthCommonDataStore.h.

95 { return m_detStore; }

◆ DoBeamTracking()

int ALFA_BeamTransport::DoBeamTracking ( int evt_number )

Function which calls BeamTrack class to calcualte Position at RPs.

Definition at line 206 of file ALFA_BeamTransport.cxx.

 {
      //do particle tracking for beam 1 
      
      //tracking funktion
      m_BeamTracker.ALFA_BeamTrack::CalculatePosRP(m_Particle1);//calculates position and momentum at RP1
      //Position at RP
      m_PosRP1=m_BeamTracker.ALFA_BeamTrack::PosRP();
      //Momentum at RP
      m_MomRP1=m_BeamTracker.ALFA_BeamTrack::MomRP();
      
      //do particle tracking for beam 2 
  
      //tracking funktion
      m_BeamTracker.ALFA_BeamTrack::CalculatePosRP(m_Particle2);//gives position and momentum at RP3
      //Position at RP
      m_PosRP3=m_BeamTracker.ALFA_BeamTrack::PosRP();
      //Momentum at RP
      m_MomRP3=m_BeamTracker.ALFA_BeamTrack::MomRP();
  
      //Write Output
      if(m_WriteDebugOutput)
      {
        m_FileBeam1 << evt_number << "\t" << std::setprecision(17)
                    << m_PosRP1.x() << std::setprecision(17) << "\t"
                    << m_PosRP1.y() << "\t" << std::setprecision(17)
                    << m_PosRP1.z() << "\t" << std::setprecision(17)
                    << m_MomRP1.x() << std::setprecision(17) << "\t"
                    << m_MomRP1.y() << "\t" << std::setprecision(17)
                    << m_MomRP1.z() << "\t" << std::setprecision(17)
                    << m_EnergyRP1 << std::endl;
        m_FileBeam2 << evt_number << "\t" << std::setprecision(17)
                    << m_PosRP3.x() << std::setprecision(17) << "\t"
                    << m_PosRP3.y() << "\t" << std::setprecision(17)
                    << m_PosRP3.z() << "\t" << std::setprecision(17)
                    << m_MomRP3.x() << std::setprecision(17) << "\t"
                    << m_MomRP3.y() << "\t" << std::setprecision(17)
                    << m_MomRP3.z() << "\t" << std::setprecision(17)
                    << m_EnergyRP3 << std::endl;
      }
  
      return true;
 }

◆ evtStore() [1/2]

ServiceHandle<StoreGateSvc>& AthCommonDataStore< AthCommonMsg< Algorithm > >::evtStore ( )

inlineinherited

The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 85 of file AthCommonDataStore.h.

85 { return m_evtStore; }

◆ evtStore() [2/2]

const ServiceHandle<StoreGateSvc>& AthCommonDataStore< AthCommonMsg< Algorithm > >::evtStore ( ) const

inlineinherited

The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 90 of file AthCommonDataStore.h.

90 { return m_evtStore; }

◆ execute()

StatusCode ALFA_BeamTransport::execute ( )

standard Athena-Algorithm method

Definition at line 109 of file ALFA_BeamTransport.cxx.

 {
        // Code entered here will be executed once per event
      ATH_MSG_DEBUG ("Executing " << name() << "...");
        //-----------------------------------------------------------------------------------------
      int run_number;
      uint64_t evt_number;
  
      //Set particles counter to zero!
      //counter for particles marked as outgoing in event record
      m_pcount=0;
      //counter for particles marked as incomming in event record
      m_pint=0;
  
      //Load event info
      SG::ReadHandle<xAOD::EventInfo> eventInfo (m_eventInfoKey,getContext());
      if(!eventInfo.isValid()) {
        ATH_MSG_ERROR("Could not retrieve EventInfo");
        return StatusCode::FAILURE;
      }
      else {
        run_number=eventInfo->runNumber();
        evt_number=eventInfo->eventNumber();
  
        ATH_MSG_DEBUG("run: " << run_number << " event: " << evt_number);
      }
  
      SG::ReadHandle<McEventCollection> mcColl(m_MCKey, getContext());
      if (!mcColl.isValid()) {
        ATH_MSG_WARNING("Could not retrieve McEventCollection: " << m_MCKey);
        return StatusCode::SUCCESS;
      }
      // Loop over all events in McEventCollectio
      ATH_MSG_INFO("successful load of HepMC info");
      for (const HepMC::GenEvent* itr  : *mcColl){
        HepMC::GenEvent evt = (*itr);
        // convert unit MeV to GeV for energy and momenta
        MeVToGeV(evt);
  
        HepMC::Print::line(std::cout, evt);
  
        // Select final state particle from event generator
        // set event status !=1 (final state)
        // fill member variables with particle data
        //       ALFA_BeamTransport::SelectParticles(evt);
  
        // There should be a check if the first particle is really the
        // particle for beam 1!!!!
  
        // run Funktion which does the beamtracking
        //       ALFA_BeamTransport::DoBeamTracking(evt_number);
  
  
        TransportSelectedParticle(evt, evt_number);
  
        // Print new data collection on screen
        HepMC::Print::line(std::cout, evt);
      }
  
      ATH_MSG_INFO("after running the process function");
  
      //-----------------------------------------------------------------------------------------
  
      return StatusCode::SUCCESS;
 }

◆ extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< 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 & AthAlgorithm::extraOutputDeps ( ) const

overridevirtualinherited

Return the list of extra output dependencies.

This list is extended to include symlinks implied by inheritance relations.

Definition at line 50 of file AthAlgorithm.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 Algorithm::extraOutputDeps();
 }

◆ finalize()

StatusCode ALFA_BeamTransport::finalize ( )

standard Athena-Algorithm method

Definition at line 96 of file ALFA_BeamTransport.cxx.

 {
   // close outputfiles for FPTracker
   if (m_WriteDebugOutput) {
     m_FileBeam1.close();
     m_FileBeam2.close();
   }
   // Code entered here will be executed once at the end of the program run.
   return StatusCode::SUCCESS;
 }

◆ GeVToMeV()

void ALFA_BeamTransport::GeVToMeV ( HepMC::GenEvent & evt )

convert GeV to MeV for HepMC event record

Definition at line 192 of file ALFA_BeamTransport.cxx.

 {
   for (const HepMC::GenParticlePtr& p : evt) {
     const HepMC::FourVector fv(p->momentum().px() * 1000.,
                                p->momentum().py() * 1000.,
                                p->momentum().pz() * 1000.,
                                p->momentum().e() * 1000.);
  
     p->set_momentum(fv);
   }
 }

◆ initialize()

StatusCode ALFA_BeamTransport::initialize ( )

standard Athena-Algorithm method

Definition at line 64 of file ALFA_BeamTransport.cxx.

 {
   ATH_MSG_INFO("Initializing " << name() << "...");
   ATH_CHECK(m_eventInfoKey.initialize());
   ATH_CHECK(m_MCKey.initialize());
   // open files for beamtracking output
   // string in char
   char* cBeam1Name = new char[m_FPOutputBeam1.length()];
   strcpy(cBeam1Name, m_FPOutputBeam1.c_str());
   char* cBeam2Name = new char[m_FPOutputBeam2.length()];
   strcpy(cBeam2Name, m_FPOutputBeam2.c_str());
  
   if (m_WriteDebugOutput) {
     m_FileBeam1.open(cBeam1Name);
     m_FileBeam2.open(cBeam2Name);
   }
  
   //delete char arrays
   delete[] cBeam1Name;
   delete[] cBeam2Name;  
   
   //Initialisation of FPTracker
   m_BeamTracker.ALFA_BeamTrack::initialize(m_FPConfig);
  
   //-----------------------------------------------------------------------------------------
   ATH_MSG_INFO( name() << " initialize()"  );
   ATH_MSG_INFO( "initialize() successful in " << name()  );
   return StatusCode::SUCCESS;
 }

◆ inputHandles()

virtual std::vector<Gaudi::DataHandle*> AthCommonDataStore< AthCommonMsg< 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.

◆ MeVToGeV()

void ALFA_BeamTransport::MeVToGeV ( HepMC::GenEvent & evt )

convert unit MeV to GeV for energy and momenta

Definition at line 178 of file ALFA_BeamTransport.cxx.

 {
   for (const HepMC::GenParticlePtr& p:  evt) {
     const HepMC::FourVector fv(p->momentum().px() / 1000.,
                                p->momentum().py() / 1000.,
                                p->momentum().pz() / 1000.,
                                p->momentum().e() / 1000.);
  
     p->set_momentum(fv);
   }
 }

◆ msg() [1/2]

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

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                 {
     return this->msgStream();
   }

◆ msg() [2/2]

MsgStream& AthCommonMsg< Algorithm >::msg ( const MSG::Level lvl ) const

inlineinherited

Definition at line 27 of file AthCommonMsg.h.

                                                   {
     return this->msgStream(lvl);
   }

◆ msgLvl()

bool AthCommonMsg< 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< 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< Algorithm > >::renounce ( T & h )

inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

   {
     h.renounce();
     PBASE::renounce (h);
   }

◆ renounceArray()

void AthCommonDataStore< AthCommonMsg< Algorithm > >::renounceArray ( SG::VarHandleKeyArray & handlesArray )

inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

                                                           {
     handlesArray.renounce();
   }

◆ SelectParticles()

int ALFA_BeamTransport::SelectParticles ( HepMC::GenEvent * evt )

Selects particles for beam transported Sets event status code of outgoing particles from generator to status != 1.

◆ sysInitialize()

StatusCode AthAlgorithm::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< Algorithm > >.

Reimplemented in AthAnalysisAlgorithm, AthFilterAlgorithm, PyAthena::Alg, and AthHistogramAlgorithm.

Definition at line 66 of file AthAlgorithm.cxx.

                                        {
   StatusCode sc=AthCommonDataStore<AthCommonMsg<Algorithm>>::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< 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.

◆ TransportSelectedParticle()

int ALFA_BeamTransport::TransportSelectedParticle	(	HepMC::GenEvent &	evt,
		int	evt_number
	)

This function is event selection, tracking and HepMC save ine one function.

Definition at line 250 of file ALFA_BeamTransport.cxx.

                                                                                    {
      HepMC::GenParticlePtr p1{nullptr};
      HepMC::GenParticlePtr p2{nullptr};
  
      std::vector<FPTracker::Point> PosAtRP1;
      std::vector<FPTracker::Point> PosAtRP3;
      std::vector<FPTracker::Point> MomAtPR1;
      std::vector<FPTracker::Point> MomAtPR3;
      std::vector<double> EnergyRP1;
      std::vector<double> EnergyRP3;
      
      
      double mom=0.;
      double eta=0.;
      double theta=0.;
  
      // First we have to select the final state particles from the MC
      for (const HepMC::GenParticlePtr& p : evt) {
  
        // Simple Eta Pt cut to remove particles from BeamTransportation which
        // have no chance to reach RP plane
        mom = std::sqrt(std::pow(p->momentum().px(), 2) +
                        std::pow(p->momentum().py(), 2) +
                        std::pow(p->momentum().pz(), 2));
        theta = std::acos(std::abs(p->momentum().pz()) / mom);
        eta = -std::log(std::tan(theta / 2));
  
        if (MC::isStable(p) && (!p->end_vertex())) {
  
          int pid = p->pdg_id();
          if (eta > m_EtaCut && 1 - std::abs(mom / m_FPConfig.pbeam0) < m_XiCut) {
  
            // save a copy of the particles which passed the cut
            HepMC::FourVector Position = p->production_vertex()->position();
            HepMC::FourVector Momentum = p->momentum();
            HepMC::GenVertexPtr Vertex = HepMC::newGenVertexPtr(Position); // copy of the vertex
            HepMC::GenParticlePtr Particle = HepMC::newGenParticlePtr(Momentum, pid, 1);
  
            Vertex->add_particle_out(Particle);
            evt.add_vertex(Vertex);
  
            // select direction of particle
            if (p->momentum().pz() > 0. && pid == 2212) { // Beam1 TODO Tracking only works for protons!!!!
              p1 = p;
              // now we want to track the final particle if it's a protons
              // Positions are given in mm FPTracker needs them in meter
              m_Particle1 = FPTracker::Particle(
                p1->production_vertex()->position().x() / 1000.,
                p1->production_vertex()->position().y() / 1000.,
                p1->production_vertex()->position().z() / 1000.,
                p1->momentum().px(),
                p1->momentum().py(),
                p1->momentum().pz());
  
              // do particle tracking for beam 1
  
              // tracking funktion
              m_BeamTracker.ALFA_BeamTrack::CalculatePosRP(m_Particle1); // calculates position and momentum at RP1
              // Position at RP
              m_PosRP1 = m_BeamTracker.ALFA_BeamTrack::PosRP();
              PosAtRP1.push_back(m_PosRP1);
              // Momentum at RP
              m_MomRP1 = m_BeamTracker.ALFA_BeamTrack::MomRP();
              MomAtPR1.push_back(m_MomRP1);
              // no Energy change between IP and RP
              m_EnergyRP1 = p1->momentum().e();
              EnergyRP1.push_back(m_EnergyRP1);
              if (m_WriteDebugOutput) {
                m_FileBeam1 << evt_number << "\t" << std::setprecision(17)
                            << m_PosRP1.x() << std::setprecision(17) << "\t"
                            << m_PosRP1.y() << "\t" << std::setprecision(17)
                            << m_PosRP1.z() << "\t" << std::setprecision(17)
                            << m_MomRP1.x() << std::setprecision(17) << "\t"
                            << m_MomRP1.y() << "\t" << std::setprecision(17)
                            << m_MomRP1.z() << "\t" << std::setprecision(17)
                            << m_EnergyRP1 << std::endl;
              }
  
            } else if (p->momentum().pz() < 0. && pid == 2212) { // beam 2
              p2 = p;
  
              m_Particle2 = FPTracker::Particle(
                p2->production_vertex()->position().x() / 1000.,
                p2->production_vertex()->position().y() / 1000.,
                p2->production_vertex()->position().z() / 1000.,
                p2->momentum().px(),
                p2->momentum().py(),
                p2->momentum().pz());
              // tracking funktion
              m_BeamTracker.ALFA_BeamTrack::CalculatePosRP(m_Particle2); // gives position and momentum at RP3
              // Position at RP
              m_PosRP3 = m_BeamTracker.ALFA_BeamTrack::PosRP();
              PosAtRP3.push_back(m_PosRP3);
              // Momentum at RP
              m_MomRP3 = m_BeamTracker.ALFA_BeamTrack::MomRP();
              MomAtPR3.push_back(m_MomRP3);
  
              m_EnergyRP3 = p2->momentum().e();
              EnergyRP3.push_back(m_EnergyRP3);
              // Write Output
              if (m_WriteDebugOutput) {
                m_FileBeam2 << evt_number << "\t" << std::setprecision(17)
                            << m_PosRP3.x() << std::setprecision(17) << "\t"
                            << m_PosRP3.y() << "\t" << std::setprecision(17)
                            << m_PosRP3.z() << "\t" << std::setprecision(17)
                            << m_MomRP3.x() << std::setprecision(17) << "\t"
                            << m_MomRP3.y() << "\t" << std::setprecision(17)
                            << m_MomRP3.z() << "\t" << std::setprecision(17)
                            << m_EnergyRP3 << std::endl;
              }
  
            } else {
              ATH_MSG_ERROR("Strange: Particle rests at IP");
            }
  
            if (pid == 2212) { // Find the protons
              m_pcount++;
              if (m_pcount > 2) {
                ATH_MSG_ERROR("Strange: More than two protons in this event!");
              }
            }
          }
        }
      }
  
      //from here the transported particles are added to HepMC
      
      ATH_MSG_INFO("Add transproted particle into HepMC event record Beam 1");
      //Add Data for HepMC Collection
      
      for(int i=0;i<(int)PosAtRP1.size();i++){//Beam1
         //The factor 1000 comes from the fact that HepMC saves length in mm
      HepMC::FourVector PositionVectorRP1 = HepMC::FourVector(PosAtRP1.at(i).x()*1000.,PosAtRP1.at(i).y()*1000.,PosAtRP1.at(i).z()*1000.,0.*1000.);
      HepMC::FourVector MomentumVectorRP1 = HepMC::FourVector(MomAtPR1.at(i).x(),MomAtPR1.at(i).y(),MomAtPR1.at(i).z(),EnergyRP1.at(i));
      
        HepMC::GenVertexPtr VertexRP1 = HepMC::newGenVertexPtr(PositionVectorRP1);
        HepMC::GenParticlePtr ParticleRP1 = HepMC::newGenParticlePtr(MomentumVectorRP1,2212,1); //save the transported particle with status code 1 (added 120124) preview was 201
        //Add particle to vertex
        VertexRP1->add_particle_out(ParticleRP1);
        //add new vertex to HepMC event record
        if(m_PosRP1.x()!=-99){ // add vertex to event record if the particle in the first beam was not lost
         evt.add_vertex(VertexRP1);
        }
      }
      ATH_MSG_INFO ("Add transproted particle into HepMC event record Beam 2" );
      for(int i=0;i<(int)PosAtRP3.size();i++){
       HepMC::FourVector PositionVectorRP3 = HepMC::FourVector(PosAtRP3.at(i).x()*1000.,PosAtRP3.at(i).y()*1000.,PosAtRP3.at(i).z()*1000.,0.*1000.);
       HepMC::FourVector MomentumVectorRP3 = HepMC::FourVector(MomAtPR3.at(i).x(),MomAtPR3.at(i).y(),MomAtPR3.at(i).z(),EnergyRP3.at(i));
       HepMC::GenVertexPtr VertexRP3 = HepMC::newGenVertexPtr(PositionVectorRP3);
       HepMC::GenParticlePtr ParticleRP3 = HepMC::newGenParticlePtr(MomentumVectorRP3,2212,1);//save the transported particle with status code 1 (added 120124) preview was 201   
       VertexRP3->add_particle_out(ParticleRP3);
           if (m_PosRP3.x() != -99) { // add vertex to event record if the // particle in the second beam was not lost
             evt.add_vertex(VertexRP3);
           }
      }
     
     //convert HepMC data back to HepMC standart ( momentum and energy in MeV)
     ALFA_BeamTransport::GeVToMeV(evt);
     return true;
 }

◆ updateVHKA()

void AthCommonDataStore< AthCommonMsg< 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_BeamTracker

ALFA_BeamTrack ALFA_BeamTransport::m_BeamTracker

private

Definition at line 99 of file ALFA_BeamTransport.h.

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Algorithm > >::m_detStore

privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

◆ m_EnergyRP1

double ALFA_BeamTransport::m_EnergyRP1 = 0.0

private

Definition at line 107 of file ALFA_BeamTransport.h.

◆ m_EnergyRP3

double ALFA_BeamTransport::m_EnergyRP3 = 0.0

private

Definition at line 108 of file ALFA_BeamTransport.h.

◆ m_EtaCut

double ALFA_BeamTransport::m_EtaCut

private

Definition at line 120 of file ALFA_BeamTransport.h.

◆ m_eventInfoKey

SG::ReadHandleKey<xAOD::EventInfo> ALFA_BeamTransport::m_eventInfoKey

private

Initial value:

{ this,
                                                     "EvtInfo",
                                                     "EventInfo",
                                                     "EventInfo name" }

Definition at line 87 of file ALFA_BeamTransport.h.

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Algorithm > >::m_evtStore

privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

◆ m_extendedExtraObjects

DataObjIDColl AthAlgorithm::m_extendedExtraObjects

privateinherited

Definition at line 79 of file AthAlgorithm.h.

◆ m_FileBeam1

std::ofstream ALFA_BeamTransport::m_FileBeam1

private

Definition at line 124 of file ALFA_BeamTransport.h.

◆ m_FileBeam2

std::ofstream ALFA_BeamTransport::m_FileBeam2

private

Definition at line 125 of file ALFA_BeamTransport.h.

◆ m_FPConfig

FPConfig ALFA_BeamTransport::m_FPConfig

private

Definition at line 94 of file ALFA_BeamTransport.h.

◆ m_FPOutputBeam1

std::string ALFA_BeamTransport::m_FPOutputBeam1

private

Definition at line 110 of file ALFA_BeamTransport.h.

◆ m_FPOutputBeam2

std::string ALFA_BeamTransport::m_FPOutputBeam2

private

Definition at line 111 of file ALFA_BeamTransport.h.

◆ m_MCKey

SG::ReadHandleKey<McEventCollection> ALFA_BeamTransport::m_MCKey

private

Initial value:

{
    this,
    "McEventCollectionName",
    "GEN_EVENT",
    "MC Event Collection name"
  }

Definition at line 80 of file ALFA_BeamTransport.h.

◆ m_MomRP1

FPTracker::Point ALFA_BeamTransport::m_MomRP1

private

Definition at line 104 of file ALFA_BeamTransport.h.

◆ m_MomRP3

FPTracker::Point ALFA_BeamTransport::m_MomRP3

private

Definition at line 105 of file ALFA_BeamTransport.h.

◆ m_Particle1

FPTracker::Particle ALFA_BeamTransport::m_Particle1

private

Definition at line 97 of file ALFA_BeamTransport.h.

◆ m_Particle2

FPTracker::Particle ALFA_BeamTransport::m_Particle2

private

Definition at line 98 of file ALFA_BeamTransport.h.

◆ m_pcount

int ALFA_BeamTransport::m_pcount = 0

private

Definition at line 116 of file ALFA_BeamTransport.h.

◆ m_pint

int ALFA_BeamTransport::m_pint = 0

private

Definition at line 118 of file ALFA_BeamTransport.h.

◆ m_PosRP1

FPTracker::Point ALFA_BeamTransport::m_PosRP1

private

Definition at line 101 of file ALFA_BeamTransport.h.

◆ m_PosRP3

FPTracker::Point ALFA_BeamTransport::m_PosRP3

private

Definition at line 102 of file ALFA_BeamTransport.h.

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< Algorithm > >::m_varHandleArraysDeclared

privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vhka

std::vector<SG::VarHandleKeyArray*> AthCommonDataStore< AthCommonMsg< Algorithm > >::m_vhka

privateinherited

Definition at line 398 of file AthCommonDataStore.h.

◆ m_WriteDebugOutput

bool ALFA_BeamTransport::m_WriteDebugOutput

private

Definition at line 127 of file ALFA_BeamTransport.h.

◆ m_XiCut

double ALFA_BeamTransport::m_XiCut

private

Definition at line 121 of file ALFA_BeamTransport.h.

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

Public Member Functions

Protected Member Functions

Private Types

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ StoreGateSvc_t

Constructor & Destructor Documentation

◆ ALFA_BeamTransport()

◆ ~ALFA_BeamTransport()

Member Function Documentation

◆ declareGaudiProperty() [1/4]

◆ declareGaudiProperty() [2/4]

◆ declareGaudiProperty() [3/4]

◆ declareGaudiProperty() [4/4]

◆ declareProperty() [1/6]

◆ declareProperty() [2/6]

◆ declareProperty() [3/6]

◆ declareProperty() [4/6]

◆ declareProperty() [5/6]

◆ declareProperty() [6/6]

◆ detStore()

◆ DoBeamTracking()

◆ evtStore() [1/2]

◆ evtStore() [2/2]

◆ execute()

◆ extraDeps_update_handler()

◆ extraOutputDeps()

◆ finalize()

◆ GeVToMeV()

◆ initialize()

◆ inputHandles()

◆ MeVToGeV()

◆ msg() [1/2]

◆ msg() [2/2]

◆ msgLvl()

◆ outputHandles()

◆ renounce()

◆ renounceArray()

◆ SelectParticles()

◆ sysInitialize()

◆ sysStart()

◆ TransportSelectedParticle()

◆ updateVHKA()

Member Data Documentation

◆ m_BeamTracker

◆ m_detStore

◆ m_EnergyRP1

◆ m_EnergyRP3

◆ m_EtaCut

◆ m_eventInfoKey

◆ m_evtStore

◆ m_extendedExtraObjects

◆ m_FileBeam1

◆ m_FileBeam2

◆ m_FPConfig

◆ m_FPOutputBeam1

◆ m_FPOutputBeam2

◆ m_MCKey

◆ m_MomRP1

◆ m_MomRP3

◆ m_Particle1

◆ m_Particle2

◆ m_pcount

◆ m_pint

◆ m_PosRP1

◆ m_PosRP3

◆ m_varHandleArraysDeclared

◆ m_vhka

◆ m_WriteDebugOutput

◆ m_XiCut