ProtonTransport.h

Go to the documentation of this file.

#ifndef PROTONTRANSPORT
#define PROTONTRANSPORT
#include "FWCore/Framework/interface/EventSetup.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 "TLorentzVector.h"

namespace CLHEP
{
    class HepRandomEngine;
}
class ProtonTransport {
      public:
            ProtonTransport();
            virtual ~ProtonTransport();
            std::vector<LHCTransportLink>& getCorrespondenceMap() { return m_CorrespondenceMap; }
            virtual void process(const HepMC::GenEvent * ev , const edm::EventSetup & es, CLHEP::HepRandomEngine * engine)=0;
            void ApplyBeamCorrection(HepMC::GenParticle* p);
            void ApplyBeamCorrection(TLorentzVector& p);
            void addPartToHepMC( HepMC::GenEvent * );
            void clear();

      protected:
            enum class TransportMode {HECTOR,TOTEM};
            TransportMode MODE;
            int NEvent;
            bool m_verbosity;
            CLHEP::HepRandomEngine * engine;

            bool bApplyZShift;
            double fPPSRegionStart_56;
            double fPPSRegionStart_45;
            double fCrossingAngle_45;
            double fCrossingAngle_56;

            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;

            double m_sigmaSX;
            double m_sigmaSY;
            double m_sigmaSTX;
            double m_sigmaSTY;
            double m_sig_E;
            double fVtxMeanX;
            double fVtxMeanY;
            double fVtxMeanZ;
            double fBeamXatIP;
            double fBeamYatIP;
            double fBeamMomentum;
            double fBeamEnergy;

};
#endif