Optical function method for PPS proton transport #29857
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,3 @@ | ||
| import FWCore.ParameterSet.Config as cms | ||
|
|
||
| ctpps_2021 = cms.Modifier() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,13 +1,18 @@ | ||
| <use name="FWCore/Framework"/> | ||
| <use name="FWCore/ParameterSet"/> | ||
| <use name="CommonTools/Utils"/> | ||
| <use name="SimDataFormats/GeneratorProducts"/> | ||
| <use name="SimGeneral/HepPDTRecord"/> | ||
| <use name="Utilities/PPS"/> | ||
| <use name="SimTransport/TotemRPProtonTransportParametrization"/> | ||
| <use name="CondFormats/PPSObjects"/> | ||
| <use name="CondFormats/RunInfo"/> | ||
| <use name="CondFormats/DataRecord"/> | ||
| <use name="hector"/> | ||
| <use name="hepmc"/> | ||
| <use name="clhep"/> | ||
| <use name="rootcore"/> | ||
| <export> | ||
| <lib name="1"/> | ||
| </export> | ||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,78 @@ | ||
| #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(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 fCrossingAngle_45; | ||
| double fCrossingAngle_56; | ||
|
|
||
| double beamMomentum_; | ||
| double beamEnergy_; | ||
| double etaCut_; | ||
| double momentumCut_; | ||
| double fVtxMeanX; | ||
| double fVtxMeanY; | ||
| double fVtxMeanZ; | ||
|
|
||
| double m_sigmaSTX; | ||
| double m_sigmaSTY; | ||
| double m_sigmaSX; | ||
| double m_sigmaSY; | ||
| double m_sig_E; | ||
| double fBeamXatIP; | ||
| double fBeamYatIP; | ||
| }; | ||
| #endif |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,47 @@ | ||
| #ifndef OPTICALFUNCTION_TRANSPORT | ||
| #define OPTICALFUNCTION_TRANSPORT | ||
| #include "CondFormats/DataRecord/interface/CTPPSBeamParametersRcd.h" | ||
| #include "CondFormats/DataRecord/interface/CTPPSInterpolatedOpticsRcd.h" | ||
| #include "CondFormats/DataRecord/interface/LHCInfoRcd.h" | ||
| #include "CondFormats/PPSObjects/interface/CTPPSBeamParameters.h" | ||
| #include "CondFormats/PPSObjects/interface/LHCInterpolatedOpticalFunctionsSetCollection.h" | ||
| #include "CondFormats/RunInfo/interface/LHCInfo.h" | ||
| #include "FWCore/Framework/interface/ESHandle.h" | ||
| #include "FWCore/Framework/interface/Event.h" | ||
| #include "FWCore/Framework/interface/EventSetup.h" | ||
| #include "FWCore/MessageLogger/interface/MessageLogger.h" | ||
| #include "FWCore/ParameterSet/interface/ParameterSet.h" | ||
| #include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h" | ||
| #include "SimTransport/PPSProtonTransport/interface/BaseProtonTransport.h" | ||
|
|
||
| #include <array> | ||
| #include <unordered_map> | ||
|
|
||
| #include <cmath> | ||
|
|
||
| class OpticalFunctionsTransport : public BaseProtonTransport { | ||
| public: | ||
| OpticalFunctionsTransport(const edm::ParameterSet& ps); | ||
| ~OpticalFunctionsTransport() override{}; | ||
|
|
||
| void process(const HepMC::GenEvent* ev, const edm::EventSetup& es, CLHEP::HepRandomEngine* engine) override; | ||
|
|
||
| private: | ||
| bool transportProton(const HepMC::GenParticle*); | ||
|
|
||
| std::string lhcInfoLabel_; | ||
| std::string opticsLabel_; | ||
|
|
||
| edm::ESHandle<LHCInfo> lhcInfo_; | ||
| edm::ESHandle<CTPPSBeamParameters> beamParameters_; | ||
| edm::ESHandle<LHCInterpolatedOpticalFunctionsSetCollection> opticalFunctions_; | ||
|
|
||
| bool useEmpiricalApertures_; | ||
| double empiricalAperture45_xi0_int_, empiricalAperture45_xi0_slp_, empiricalAperture45_a_int_, | ||
| empiricalAperture45_a_slp_; | ||
| double empiricalAperture56_xi0_int_, empiricalAperture56_xi0_slp_, empiricalAperture56_a_int_, | ||
| empiricalAperture56_a_slp_; | ||
|
|
||
| bool produceHitsRelativeToBeam_; | ||
| }; | ||
| #endif |
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
please delete the commented lines of code.