This is the ALFA Beam Transport package. More...

#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.
	~ALFA_BeamTransport ()
	Default Destructor.
StatusCode	initialize ()
	standard Athena-Algorithm method
StatusCode	execute (const EventContext &ctx)
	standard Athena-Algorithm method
StatusCode	finalize ()
	standard Athena-Algorithm method
int	SelectParticles (HepMC::GenEvent *evt)
	Selects particles for beam transported Sets event status code of outgoing particles from generator to status != 1.
int	DoBeamTracking (int evt_number)
	Function which calls BeamTrack class to calcualte Position at RPs.
int	TransportSelectedParticle (HepMC::GenEvent &evt, int evt_number)
	This function is event selection, tracking and HepMC save ine one function.
virtual StatusCode	sysInitialize () override
	Override sysInitialize.
virtual bool	isClonable () const override
	Specify if the algorithm is clonable.
virtual StatusCode	sysExecute (const EventContext &ctx) override
	Execute an algorithm.
virtual const DataObjIDColl &	extraOutputDeps () const override
	Return the list of extra output dependencies.
virtual bool	filterPassed (const EventContext &ctx) const
	Get filter decision:
virtual void	setFilterPassed (bool state, const EventContext &ctx) const
	Set filter decision:
ServiceHandle< StoreGateSvc > &	evtStore ()
	The standard `StoreGateSvc` (event store) Returns (kind of) a pointer to the `StoreGateSvc`.
const ServiceHandle< StoreGateSvc > &	detStore () const
	The standard `StoreGateSvc/DetectorStore` Returns (kind of) a pointer to the `StoreGateSvc`.
virtual StatusCode	sysStart () override
	Handle START transition.
virtual std::vector< Gaudi::DataHandle * >	inputHandles () const override
	Return this algorithm's input handles.
virtual std::vector< Gaudi::DataHandle * >	outputHandles () const override
	Return this algorithm's output handles.
Gaudi::Details::PropertyBase &	declareProperty (Gaudi::Property< T, V, H > &t)
void	updateVHKA (Gaudi::Details::PropertyBase &)
MsgStream &	msg () const
bool	msgLvl (const MSG::Level lvl) const

Protected Member Functions
virtual bool	isReEntrant () const override final
	Legacy algorithms are not thread-safe.
void	renounceArray (SG::VarHandleKeyArray &handlesArray)
	remove all handles from I/O resolution
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.

Private Types
typedef ServiceHandle< StoreGateSvc >	StoreGateSvc_t

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

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
	Extra output dependency collection, extended by AthAlgorithmDHUpdate to add symlinks.
StoreGateSvc_t	m_evtStore
	Pointer to StoreGate (event store by default).
StoreGateSvc_t	m_detStore
	Pointer to StoreGate (detector store by default).
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 45 of file ALFA_BeamTransport.h.

Member Typedef Documentation

◆ StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< Gaudi::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 35 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 60 of file ALFA_BeamTransport.cxx.

60{}

Member Function Documentation

◆ declareGaudiProperty()

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

◆ declareProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareProperty ( Gaudi::Property< T, V, H > & 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< Gaudi::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; }

AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_detStore

StoreGateSvc_t m_detStore

Definition AthCommonDataStore.h:393

◆ DoBeamTracking()

int ALFA_BeamTransport::DoBeamTracking ( int evt_number )

Function which calls BeamTrack class to calcualte Position at RPs.

Definition at line 174 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()

ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< Gaudi::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; }

AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_evtStore

StoreGateSvc_t m_evtStore

Definition AthCommonDataStore.h:390

◆ execute()

StatusCode ALFA_BeamTransport::execute ( const EventContext & ctx )

virtual

standard Athena-Algorithm method

Implements AthAlgorithm.

Definition at line 108 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, ctx);
     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, ctx);
     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);
       MC::MeVToGeV(&evt);  //Only scales momenta and masses
       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< Gaudi::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 & AthCommonAlgorithm< Gaudi::Algorithm >::extraOutputDeps ( ) const

overridevirtualinherited

Return the list of extra output dependencies.

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

Definition at line 89 of file AthCommonAlgorithm.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 BaseAlg::extraOutputDeps();
}

◆ filterPassed()

virtual bool AthCommonAlgorithm< Gaudi::Algorithm >::filterPassed ( const EventContext & ctx ) const

inlinevirtualinherited

Get filter decision:

Definition at line 93 of file AthCommonAlgorithm.h.

                                                           {
    return execState( ctx ).filterPassed();
  }

◆ finalize()

StatusCode ALFA_BeamTransport::finalize ( )

standard Athena-Algorithm method

Definition at line 95 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;
}

◆ initialize()

StatusCode ALFA_BeamTransport::initialize ( )

standard Athena-Algorithm method

Definition at line 63 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< Gaudi::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.

◆ isClonable()

virtual bool AthCommonAlgorithm< Gaudi::Algorithm >::isClonable ( ) const

inlineoverridevirtualinherited

Specify if the algorithm is clonable.

Only relevant for non-reentrant algorithms. Actual number of clones needs to be set via the "Cardinality" property.

Reimplemented in AFP_DigiTop, AlgB, AlgT, BCM_Digitization, CscDigitBuilder, CscDigitToCscRDO, G4AtlasAlg, G4RunAlg, HGTD_Digitization, HiveAlgBase, InDet::GNNSeedingTrackMaker, InDet::SCT_Clusterization, InDet::SiSPGNNTrackMaker, InDet::SiSPSeededTrackFinder, InDet::SiTrackerSpacePointFinder, ISF::SimKernelMT, ITk::StripDigitization, ITkPixelCablingAlg, ITkStripCablingAlg, LArHitEMapMaker, LArTTL1Maker, LUCID_DigiTop, LVL1::L1TopoSimulation, MergeCalibHits, MergeGenericMuonSimHitColl, MergeHijingPars, MergeMcEventCollection, MergeTrackRecordCollection, MergeTruthJets, MergeTruthParticles, MuonDigitizer, PileUpMTAlg, PixelDigitization, RoIBResultToxAOD, SCT_ByteStreamErrorsTestAlg, SCT_CablingCondAlgFromCoraCool, SCT_CablingCondAlgFromText, SCT_ConditionsParameterTestAlg, SCT_ConditionsSummaryTestAlg, SCT_ConfigurationConditionsTestAlg, SCT_Digitization, SCT_FlaggedConditionTestAlg, SCT_LinkMaskingTestAlg, SCT_MajorityConditionsTestAlg, SCT_ModuleVetoTestAlg, SCT_MonitorConditionsTestAlg, SCT_PrepDataToxAOD, SCT_RawDataToxAOD, SCT_ReadCalibChipDataTestAlg, SCT_ReadCalibDataTestAlg, SCT_RODVetoTestAlg, SCT_SensorsTestAlg, SCT_SiliconConditionsTestAlg, SCT_StripVetoTestAlg, SCT_TdaqEnabledTestAlg, SCT_TestCablingAlg, SCTEventFlagWriter, SCTRawDataProvider, SCTSiLorentzAngleTestAlg, SCTSiPropertiesTestAlg, SGInputLoader, Simulation::BeamEffectsAlg, TileHitVecToCnt, TileMuonFitter, TilePulseForTileMuonReceiver, TileRawChannelMaker, TRTDigitization, and ZDC_DigiTop.

Definition at line 68 of file AthCommonAlgorithm.h.

                                           {
    return true;
  }

◆ isReEntrant()

virtual bool AthAlgorithm::isReEntrant ( ) const

inlinefinaloverrideprotectedvirtualinherited

Legacy algorithms are not thread-safe.

Definition at line 47 of file AthAlgorithm.h.

47{ return false; }

◆ msg()

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

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                {
    return this->msgStream();
  }

◆ msgLvl()

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

inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

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

◆ renounceArray()

void AthCommonDataStore< AthCommonMsg< Gaudi::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.

◆ setFilterPassed()

virtual void AthCommonAlgorithm< Gaudi::Algorithm >::setFilterPassed	(	bool	state,
		const EventContext &	ctx ) const

inlinevirtualinherited

Set filter decision:

Reimplemented in AthFilterAlgorithm.

Definition at line 99 of file AthCommonAlgorithm.h.

                                                                            {
    execState( ctx ).setFilterPassed( state );
  }

◆ sysExecute()

StatusCode AthCommonAlgorithm< Gaudi::Algorithm >::sysExecute ( const EventContext & ctx )

overridevirtualinherited

Execute an algorithm.

We override this in order to work around an issue with the Algorithm base class storing the event context in a member variable that can cause crashes in MT jobs.

Reimplemented in AthAnalysisAlgorithm.

Definition at line 80 of file AthCommonAlgorithm.cxx.

{
  return BaseAlg::sysExecute (ctx);
}

◆ sysInitialize()

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

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

Definition at line 60 of file AthCommonAlgorithm.cxx.

                                                      {
  StatusCode sc=AthCommonDataStore<AthCommonMsg<BaseAlg>>::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< Gaudi::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 218 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(std::move(Particle));
           evt.add_vertex(std::move(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(std::move(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(std::move(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(std::move(ParticleRP3));
          if (m_PosRP3.x() != -99) { // add vertex to event record if the // particle in the second beam was not lost
            evt.add_vertex(std::move(VertexRP3));
          }
     }
 
    MC::GeVToMeV(&evt);  //Only scales momenta and masses
    return true;
}

◆ updateVHKA()

void AthCommonDataStore< AthCommonMsg< Gaudi::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 91 of file ALFA_BeamTransport.h.

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Gaudi::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 99 of file ALFA_BeamTransport.h.

◆ m_EnergyRP3

double ALFA_BeamTransport::m_EnergyRP3 = 0.0

private

Definition at line 100 of file ALFA_BeamTransport.h.

◆ m_EtaCut

double ALFA_BeamTransport::m_EtaCut

private

Definition at line 112 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 79 of file ALFA_BeamTransport.h.

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

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_evtStore

privateinherited

Pointer to StoreGate (event store by default).

Definition at line 390 of file AthCommonDataStore.h.

◆ m_extendedExtraObjects

DataObjIDColl AthCommonAlgorithm< Gaudi::Algorithm >::m_extendedExtraObjects

privateinherited

Extra output dependency collection, extended by AthAlgorithmDHUpdate to add symlinks.

Empty if no symlinks were found.

Definition at line 108 of file AthCommonAlgorithm.h.

◆ m_FileBeam1

std::ofstream ALFA_BeamTransport::m_FileBeam1

private

Definition at line 116 of file ALFA_BeamTransport.h.

◆ m_FileBeam2

std::ofstream ALFA_BeamTransport::m_FileBeam2

private

Definition at line 117 of file ALFA_BeamTransport.h.

◆ m_FPConfig

FPConfig ALFA_BeamTransport::m_FPConfig

private

Definition at line 86 of file ALFA_BeamTransport.h.

◆ m_FPOutputBeam1

std::string ALFA_BeamTransport::m_FPOutputBeam1

private

Definition at line 102 of file ALFA_BeamTransport.h.

◆ m_FPOutputBeam2

std::string ALFA_BeamTransport::m_FPOutputBeam2

private

Definition at line 103 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 72 of file ALFA_BeamTransport.h.

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

◆ m_MomRP1

FPTracker::Point ALFA_BeamTransport::m_MomRP1

private

Definition at line 96 of file ALFA_BeamTransport.h.

◆ m_MomRP3

FPTracker::Point ALFA_BeamTransport::m_MomRP3

private

Definition at line 97 of file ALFA_BeamTransport.h.

◆ m_Particle1

FPTracker::Particle ALFA_BeamTransport::m_Particle1

private

Definition at line 89 of file ALFA_BeamTransport.h.

◆ m_Particle2

FPTracker::Particle ALFA_BeamTransport::m_Particle2

private

Definition at line 90 of file ALFA_BeamTransport.h.

◆ m_pcount

int ALFA_BeamTransport::m_pcount = 0

private

Definition at line 108 of file ALFA_BeamTransport.h.

◆ m_pint

int ALFA_BeamTransport::m_pint = 0

private

Definition at line 110 of file ALFA_BeamTransport.h.

◆ m_PosRP1

FPTracker::Point ALFA_BeamTransport::m_PosRP1

private

Definition at line 93 of file ALFA_BeamTransport.h.

◆ m_PosRP3

FPTracker::Point ALFA_BeamTransport::m_PosRP3

private

Definition at line 94 of file ALFA_BeamTransport.h.

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_varHandleArraysDeclared

privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vhka

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

privateinherited

Definition at line 398 of file AthCommonDataStore.h.

◆ m_WriteDebugOutput

bool ALFA_BeamTransport::m_WriteDebugOutput

private

Definition at line 119 of file ALFA_BeamTransport.h.

◆ m_XiCut

double ALFA_BeamTransport::m_XiCut

private

Definition at line 113 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()

◆ declareProperty()

◆ detStore()

◆ DoBeamTracking()

◆ evtStore()

◆ execute()

◆ extraDeps_update_handler()

◆ extraOutputDeps()

◆ filterPassed()

◆ finalize()

◆ initialize()

◆ inputHandles()

◆ isClonable()

◆ isReEntrant()

◆ msg()

◆ msgLvl()

◆ outputHandles()

◆ renounce()

◆ renounceArray()

◆ SelectParticles()

◆ setFilterPassed()

◆ sysExecute()

◆ 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