BaseProtonTransport.h

Go to the documentation of this file.

#ifndef BASEPROTONTRANSPORT
#define BASEPROTONTRANSPORT
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "HepMC/GenEvent.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "Utilities/PPS/interface/PPSUnitConversion.h"
#include "SimDataFormats/Forward/interface/LHCTransportLink.h"
#include <vector>
#include <map>
#include <string>
#include "TLorentzVector.h"
 
namespace CLHEP {
  class HepRandomEngine;
}
class BaseProtonTransport {
public:
  BaseProtonTransport(const edm::ParameterSet& iConfig);
  virtual ~BaseProtonTransport() { this->clear(); };
 
  std::vector<LHCTransportLink>& getCorrespondenceMap() { return m_CorrespondenceMap; }
  virtual void process(const HepMC::GenEvent* ev, const edm::EventSetup& es, CLHEP::HepRandomEngine* engine) = 0;
 
  void clear();
 
  void addPartToHepMC(const HepMC::GenEvent*, HepMC::GenEvent*);
 
  void ApplyBeamCorrection(HepMC::GenParticle* p);
  void ApplyBeamCorrection(TLorentzVector& p);
 
  void setBeamEnergy(double e) {
    beamEnergy_ = e;
    beamMomentum_ = sqrt(beamEnergy_ * beamEnergy_ - ProtonMassSQ);
  }
 
  double beamEnergy() { return beamEnergy_; };
  double beamMomentum() { return beamMomentum_; };
 
protected:
  enum class TransportMode { HECTOR, TOTEM, OPTICALFUNCTIONS };
  TransportMode MODE;
 
  int NEvent;
  CLHEP::HepRandomEngine* engine_;
  std::vector<LHCTransportLink> m_CorrespondenceMap;
  std::map<unsigned int, TLorentzVector> m_beamPart;
  std::map<unsigned int, double> m_xAtTrPoint;
  std::map<unsigned int, double> m_yAtTrPoint;
 
  bool verbosity_;
  bool bApplyZShift;
 
  std::string beam1Filename_;
  std::string beam2Filename_;
 
  double fPPSRegionStart_45;
  double fPPSRegionStart_56;
  double fCrossingAngleX_45;
  double fCrossingAngleX_56;
  double fCrossingAngleY_45;
  double fCrossingAngleY_56;
 
  double beamMomentum_;
  double beamEnergy_;
  double etaCut_;
  double momentumCut_;
 
  double m_sigmaSTX;
  double m_sigmaSTY;
  double m_sigmaSX;
  double m_sigmaSY;
  double m_sig_E;
  double fBeamXatIP;
  double fBeamYatIP;
};
#endif