CrabKissingVertexPositioner.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef BEAMEFFECTS_CRABKISSINGVERTEXPOSITIONER_H
#define BEAMEFFECTS_CRABKISSINGVERTEXPOSITIONER_H 1
 
// STL includes
#include <string>
#include <utility>
#include <vector>
#include <map>
 
// FrameWork includes
#include "GaudiKernel/ServiceHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "AthenaKernel/IAthRNGSvc.h"
#include "CxxUtils/checker_macros.h"
#include "HepMC_Interfaces/ILorentzVectorGenerator.h"
 
// InDetBeamSpotService
#include "BeamSpotConditionsData/BeamSpotData.h"
 
// Forward declarations
namespace ATHRNG {
  class RNGWrapper;
}
namespace CLHEP {
  class HepRandomEngine;
}
 
namespace Simulation
{

  class CrabKissingVertexPositioner : public extends<AthAlgTool, ILorentzVectorGenerator> {
 
    public:

      CrabKissingVertexPositioner( const std::string& t, const std::string& n, const IInterface* p );

     virtual ~CrabKissingVertexPositioner() = default;

      StatusCode  initialize() override final;
      StatusCode  finalize() override final;

      CLHEP::HepLorentzVector  *generate(const EventContext& ctx) const override final;
 
    private:
 
      inline double heaviside(double val) const {return (val >= 0.0) ? 1.0 : 0.0;};
      double getDisplacement(double bunchSize, double angle1, double angle2,
                             CLHEP::HepRandomEngine* rng) const;
      double beamspotFunction(double displacement, double angle1, double angle2) const;
 
      SG::ReadCondHandleKey<InDet::BeamSpotData> m_beamSpotKey { this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot" };
      ServiceHandle<IAthRNGSvc>   m_rndGenSvc{this, "RandomSvc", "AthRNGSvc"};
      ATHRNG::RNGWrapper*             m_randomEngine ATLAS_THREAD_SAFE{};  
      Gaudi::Property<std::string>     m_randomEngineName{this, "RandomStream", "VERTEX", "Name of the random number stream"};
      Gaudi::Property<std::string> m_bunchShapeProp{this, "BunchShape", "GAUSS", "GAUSS or FLAT"};
      void BunchShapeHandler(Gaudi::Details::PropertyBase&);
      enum BunchShape{GAUSS,FLAT,NSHAPES};
      BunchShape m_bunchShape{BunchShape::GAUSS}; // GAUSS or FLAT
      Gaudi::Property<double> m_bunchLength{this, "BunchLength", 75., "75.0 mm"}; 
      Gaudi::Property<double> m_betaStar{this, "BetaStar", 150., "beta* in the parallel (kissing) plane, we assume betax=betay, units: mm"};
      Gaudi::Property<double> m_epsilon{this, "Epsilon", 2.5e-3, "Normalized emittance, unit: mm"};
      Gaudi::Property<double> m_alphaPar{this, "AlfasParallel", 0., "Kissing angle (Radians)"}; // Kissing angle
      Gaudi::Property<double> m_alphaX{this, "AlfaX", 295e-6};
      Gaudi::Property<double> m_thetaX{this, "ThetaX", 295e-6};
      // ----------------------------------------------------------------------------------------------------------------
  };
 
}
 
#endif //> !BEAMEFFECTS_CRABKISSINGVERTEXPOSITIONER_H