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;
  bool useBeamPositionFromLHCInfo_;
  bool produceHitsRelativeToBeam_;

  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;
};
#endif