#include <test/EmbeddingLHEProducer/plugins/EmbeddingLHEProducer.cc>

Inheritance diagram for EmbeddingLHEProducer:

Public Member Functions
	EmbeddingLHEProducer (const edm::ParameterSet &)

	~EmbeddingLHEProducer () override

Public Member Functions inherited from edm::one::EDProducer< edm::BeginRunProducer, edm::EndRunProducer >
	EDProducer ()=default

	EDProducer (const EDProducer &)=delete

SerialTaskQueue *	globalLuminosityBlocksQueue () final

SerialTaskQueue *	globalRunsQueue () final

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginProcessBlocks () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndProcessBlocks () const final

bool	hasAbilityToProduceInEndRuns () const final

const EDProducer &	operator= (const EDProducer &)=delete

bool	wantsGlobalLuminosityBlocks () const noexcept final

bool	wantsGlobalRuns () const noexcept final

bool	wantsInputProcessBlocks () const noexcept final

bool	wantsProcessBlocks () const noexcept final

Public Member Functions inherited from edm::one::EDProducerBase
	EDProducerBase ()

ModuleDescription const &	moduleDescription () const

bool	wantsStreamLuminosityBlocks () const noexcept

bool	wantsStreamRuns () const noexcept

	~EDProducerBase () override

Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

std::vector< edm::ProductResolverIndex > const &	putTokenIndexToProductResolverIndex () const

std::vector< bool > const &	recordProvenanceList () const

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

void	resolvePutIndicies (BranchType iBranchType, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)

TypeLabelList const &	typeLabelList () const
	used by the fwk to register the list of products of this module More...

	~ProducerBase () noexcept(false) override

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ESResolverIndex const *	esGetTokenIndices (edm::Transition iTrans) const

std::vector< ESResolverIndex > const &	esGetTokenIndicesVector (edm::Transition iTrans) const

std::vector< ESRecordIndex > const &	esGetTokenRecordIndicesVector (edm::Transition iTrans) const

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::array< std::vector< ModuleDescription const > , NumBranchTypes > &modulesAll, std::vector< ModuleProcessName > &modulesInPreviousProcesses, ProductRegistry const &preg, std::map< std::string, ModuleDescription const *> const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

void	selectInputProcessBlocks (ProductRegistry const &productRegistry, ProcessBlockHelperBase const &processBlockHelperBase)

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

void	updateLookup (eventsetup::ESRecordsToProductResolverIndices const &)

virtual	~EDConsumerBase () noexcept(false)

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)

Static Public Member Functions inherited from edm::one::EDProducerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Private Member Functions
void	assign_4vector (TLorentzVector &Lepton, const pat::Muon *muon, std::string FSRmode)

void	beginJob () override

void	beginRunProduce (edm::Run &run, edm::EventSetup const &es) override

void	endJob () override

void	endRunProduce (edm::Run &, edm::EventSetup const &) override

void	fill_lhe_from_mumu (TLorentzVector &positiveLepton, TLorentzVector &negativeLepton, lhef::HEPEUP &outlhe, CLHEP::HepRandomEngine *engine)

void	fill_lhe_with_particle (lhef::HEPEUP &outlhe, TLorentzVector &particle, int pdgid, double spin, double ctau)

const reco::Candidate *	find_original_muon (const reco::Candidate *muon)

LHERunInfoProduct::Header	give_slha ()

void	InitialRecoCorrection (TLorentzVector &positiveLepton, TLorentzVector &negativeLepton)

void	mirror (TLorentzVector &positiveLepton, TLorentzVector &negativeLepton)

void	produce (edm::Event &, const edm::EventSetup &) override

void	rotate180 (TLorentzVector &positiveLepton, TLorentzVector &negativeLepton)

void	transform_mumu_to_tautau (TLorentzVector &positiveLepton, TLorentzVector &negativeLepton)

Private Attributes
const int	embeddingParticles [3] {11, 13, 15}

std::ofstream	file

bool	InitialRecoCorrection_

bool	mirror_

edm::EDGetTokenT< edm::View< pat::Muon > >	muonsCollection_

int	particleToEmbed_

bool	rotate180_

std::string	studyFSRmode_

edm::EDGetTokenT< reco::VertexCollection >	vertexCollection_

bool	write_lheout

Static Private Attributes
static constexpr double	elMass_ = 0.00051

static constexpr double	muonMass_ = 0.1057

static constexpr double	tauMass_ = 1.77682

Additional Inherited Members
Public Types inherited from edm::one::EDProducerBase
typedef EDProducerBase	ModuleType

Public Types inherited from edm::ProducerBase
template<typename T >
using	BranchAliasSetterT = ProductRegistryHelper::BranchAliasSetterT< T >

using	ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const , const char , edm::ProductResolverIndex > >

typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Protected Member Functions inherited from edm::ProducerBase
template<Transition Tr = Transition::Event>
auto	produces (std::string instanceName) noexcept
	declare what type of product will make and with which optional label More...

template<Transition B>
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)

template<BranchType B>
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)

BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)

template<typename ProductType , Transition B>
BranchAliasSetterT< ProductType >	produces (std::string instanceName)

template<class ProductType >
BranchAliasSetterT< ProductType >	produces ()

template<typename ProductType , BranchType B>
BranchAliasSetterT< ProductType >	produces (std::string instanceName)

template<typename ProductType , BranchType B>
BranchAliasSetterT< ProductType >	produces ()

template<class ProductType >
BranchAliasSetterT< ProductType >	produces (std::string instanceName)

template<typename ProductType , Transition B>
BranchAliasSetterT< ProductType >	produces ()

template<Transition Tr = Transition::Event>
auto	produces () noexcept

ProducesCollector	producesCollector ()

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

template<BranchType B = InEvent>
EDConsumerBaseAdaptor< B >	consumes (edm::InputTag tag) noexcept

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

template<Transition Tr = Transition::Event>
constexpr auto	esConsumes ()

template<Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag tag)

template<Transition Tr = Transition::Event>
ESGetTokenGeneric	esConsumes (eventsetup::EventSetupRecordKey const &iRecord, eventsetup::DataKey const &iKey)
	Used with EventSetupRecord::doGet. More...

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

void	resetItemsToGetFrom (BranchType iType)

Detailed Description

Description: [one line class summary]

Implementation: [Notes on implementation]

Definition at line 63 of file EmbeddingLHEProducer.cc.

Constructor & Destructor Documentation

◆ EmbeddingLHEProducer()

EmbeddingLHEProducer::EmbeddingLHEProducer ( const edm::ParameterSet & iConfig )

explicit

Definition at line 114 of file EmbeddingLHEProducer.cc.

References embeddingParticles, Exception, file, spr::find(), edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), InitialRecoCorrection_, edm::Service< T >::isAvailable(), mirror_, muonsCollection_, MillePedeFileConverter_cfg::out, particleToEmbed_, rotate180_, AlCaHLTBitMon_QueryRunRegistry::string, studyFSRmode_, pileupReCalc_HLTpaths::trunc, vertexCollection_, and write_lheout.

                                                                          {
   // register your products
   produces<LHEEventProduct>();
   produces<LHERunInfoProduct, edm::Transition::BeginRun>();
   produces<math::XYZTLorentzVectorD>("vertexPosition");
 
   muonsCollection_ = consumes<edm::View<pat::Muon>>(iConfig.getParameter<edm::InputTag>("src"));
   vertexCollection_ = consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("vertices"));
   particleToEmbed_ = iConfig.getParameter<int>("particleToEmbed");
   mirror_ = iConfig.getParameter<bool>("mirror");
   InitialRecoCorrection_ = iConfig.getParameter<bool>("InitialRecoCorrection");
   rotate180_ = iConfig.getParameter<bool>("rotate180");
   studyFSRmode_ = iConfig.getUntrackedParameter<std::string>("studyFSRmode", "");
 
   write_lheout = false;
   std::string lhe_ouputfile = iConfig.getUntrackedParameter<std::string>("lhe_outputfilename", "");
   if (!lhe_ouputfile.empty()) {
     write_lheout = true;
     file.open(lhe_ouputfile, std::fstream::out | std::fstream::trunc);
   }
 
   // check if particle can be embedded
   if (std::find(std::begin(embeddingParticles), std::end(embeddingParticles), particleToEmbed_) ==
       std::end(embeddingParticles)) {
     throw cms::Exception("Configuration") << "The given particle to embed is not in the list of allowed particles.";
   }
 
   edm::Service<edm::RandomNumberGenerator> rng;
   if (!rng.isAvailable()) {
     throw cms::Exception("Configuration") << "The EmbeddingLHEProducer requires the RandomNumberGeneratorService\n"
                                              "which is not present in the configuration file. \n"
                                              "You must add the service\n"
                                              "in the configuration file or remove the modules that require it.";
   }
 }

◆ ~EmbeddingLHEProducer()

EmbeddingLHEProducer::~EmbeddingLHEProducer ( )

override

Definition at line 150 of file EmbeddingLHEProducer.cc.

150 {}

Member Function Documentation

◆ assign_4vector()

void EmbeddingLHEProducer::assign_4vector	(	TLorentzVector &	Lepton,
		const pat::Muon *	muon,
		std::string	FSRmode
	)

private

Definition at line 352 of file EmbeddingLHEProducer.cc.

References find_original_muon(), reco::Candidate::p4(), and reco::LeafCandidate::p4().

Referenced by produce().

                                                                                                         {
   if ("afterFSR" == FSRmode && muon->genParticle() != nullptr) {
     const reco::GenParticle *afterFSRMuon = muon->genParticle();
     Lepton.SetPxPyPzE(
         afterFSRMuon->p4().px(), afterFSRMuon->p4().py(), afterFSRMuon->p4().pz(), afterFSRMuon->p4().e());
   } else if ("beforeFSR" == FSRmode && muon->genParticle() != nullptr) {
     const reco::Candidate *beforeFSRMuon = find_original_muon(muon->genParticle());
     Lepton.SetPxPyPzE(
         beforeFSRMuon->p4().px(), beforeFSRMuon->p4().py(), beforeFSRMuon->p4().pz(), beforeFSRMuon->p4().e());
   } else {
     Lepton.SetPxPyPzE(muon->p4().px(), muon->p4().py(), muon->p4().pz(), muon->p4().e());
   }
   return;
 }

◆ beginJob()

void EmbeddingLHEProducer::beginJob ( void )

overrideprivatevirtual

Reimplemented from edm::one::EDProducerBase.

Definition at line 212 of file EmbeddingLHEProducer.cc.

212 {}

◆ beginRunProduce()

void EmbeddingLHEProducer::beginRunProduce	(	edm::Run &	run,
		edm::EventSetup const &	es
	)

overrideprivate

Definition at line 219 of file EmbeddingLHEProducer.cc.

References filterCSVwithJSON::copy, file, give_slha(), lhef::HEPRUP::IDWTUP, lhe2HepMCConverter_cff::LHERunInfoProduct, lhef::HEPRUP::LPRUP, eostools::move(), lhef::HEPRUP::resize(), writedatasetfile::run, submitPVValidationJobs::runInfo, write_lheout, lhef::HEPRUP::XERRUP, lhef::HEPRUP::XMAXUP, and lhef::HEPRUP::XSECUP.

                                                                              {
   // fill HEPRUP common block and store in edm::Run
   lhef::HEPRUP heprup;
 
   // set number of processes: 1 for Z to tau tau
   heprup.resize(1);
 
   // master switch for event weight iterpretation (same as in officially produced DYJets LHE files)
   heprup.IDWTUP = 3;
 
   // Information for first process (Z to tau tau), for now only placeholder:
   heprup.XSECUP[0] = 1.;
   heprup.XERRUP[0] = 0;
   heprup.XMAXUP[0] = 1;
   heprup.LPRUP[0] = 1;
 
   std::unique_ptr<LHERunInfoProduct> runInfo(new LHERunInfoProduct(heprup));
   runInfo->addHeader(give_slha());
 
   if (write_lheout)
     std::copy(runInfo->begin(), runInfo->end(), std::ostream_iterator<std::string>(file));
   run.put(std::move(runInfo));
 }

◆ endJob()

void EmbeddingLHEProducer::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::one::EDProducerBase.

Definition at line 215 of file EmbeddingLHEProducer.cc.

215 {}

◆ endRunProduce()

void EmbeddingLHEProducer::endRunProduce	(	edm::Run &	run,
		edm::EventSetup const &	es
	)

overrideprivate

Definition at line 243 of file EmbeddingLHEProducer.cc.

References LHERunInfoProduct::endOfFile(), file, and write_lheout.

                                                                              {
   if (write_lheout) {
     file << LHERunInfoProduct::endOfFile();
     file.close();
   }
 }

◆ fill_lhe_from_mumu()

void EmbeddingLHEProducer::fill_lhe_from_mumu	(	TLorentzVector &	positiveLepton,
		TLorentzVector &	negativeLepton,
		lhef::HEPEUP &	outlhe,
		CLHEP::HepRandomEngine *	engine
	)

private

Definition at line 250 of file EmbeddingLHEProducer.cc.

References ALPAKA_ACCELERATOR_NAMESPACE::brokenline::constexpr(), fill_lhe_with_particle(), particleToEmbed_, and beamSpotPI::Z.

Referenced by produce().

                                                                             {
   TLorentzVector Z = positiveLepton + negativeLepton;
   int leptonPDGID = particleToEmbed_;
 
   static constexpr double tau_ctau0 = 8.71100e-02;  // mm (for Pythia)
   double tau_ctau_p = tau_ctau0 * CLHEP::RandExponential::shoot(engine);
   double tau_ctau_n = tau_ctau0 * CLHEP::RandExponential::shoot(engine);
 
   fill_lhe_with_particle(outlhe, Z, 23, 9.0, 0);
   fill_lhe_with_particle(outlhe, positiveLepton, -leptonPDGID, 1.0, tau_ctau_p);
   fill_lhe_with_particle(outlhe, negativeLepton, leptonPDGID, -1.0, tau_ctau_n);
 
   return;
 }

◆ fill_lhe_with_particle()

void EmbeddingLHEProducer::fill_lhe_with_particle	(	lhef::HEPEUP &	outlhe,
		TLorentzVector &	particle,
		int	pdgid,
		double	spin,
		double	ctau
	)

private

Definition at line 268 of file EmbeddingLHEProducer.cc.

References funct::abs(), embeddingParticles, spr::find(), lhef::HEPEUP::ICOLUP, lhef::HEPEUP::IDUP, lhef::HEPEUP::ISTUP, lhef::HEPEUP::MOTHUP, lhef::HEPEUP::NUP, EgammaValidation_cff::pdgid, lhef::HEPEUP::PUP, lhef::HEPEUP::resize(), lhef::HEPEUP::SPINUP, and lhef::HEPEUP::VTIMUP.

Referenced by fill_lhe_from_mumu().

                                                                                        {
   // Pay attention to different index conventions:
   // 'particleindex' follows usual C++ index conventions starting at 0 for a list.
   // 'motherindex' follows the LHE index conventions: 0 is for 'not defined', so the listing starts at 1.
   // That means: LHE index 1 == C++ index 0.
   int particleindex = outlhe.NUP;
   outlhe.resize(outlhe.NUP + 1);
 
   outlhe.PUP[particleindex][0] = particle.Px();
   outlhe.PUP[particleindex][1] = particle.Py();
   outlhe.PUP[particleindex][2] = particle.Pz();
   outlhe.PUP[particleindex][3] = particle.E();
   outlhe.PUP[particleindex][4] = particle.M();
   outlhe.IDUP[particleindex] = pdgid;
   outlhe.SPINUP[particleindex] = spin;
   outlhe.VTIMUP[particleindex] = ctau;
 
   outlhe.ICOLUP[particleindex].first = 0;
   outlhe.ICOLUP[particleindex].second = 0;
 
   if (std::abs(pdgid) == 23) {
     outlhe.MOTHUP[particleindex].first = 0;  // No Mother
     outlhe.MOTHUP[particleindex].second = 0;
     outlhe.ISTUP[particleindex] = 2;  // status
   }
 
   if (std::find(std::begin(embeddingParticles), std::end(embeddingParticles), std::abs(pdgid)) !=
       std::end(embeddingParticles)) {
     outlhe.MOTHUP[particleindex].first = 1;   // Mother is the Z (first partile)
     outlhe.MOTHUP[particleindex].second = 1;  // Mother is the Z (first partile)
 
     outlhe.ISTUP[particleindex] = 1;  // status
   }
 
   return;
 }

◆ fillDescriptions()

void EmbeddingLHEProducer::fillDescriptions ( edm::ConfigurationDescriptions & descriptions )

static

Definition at line 562 of file EmbeddingLHEProducer.cc.

References edm::ConfigurationDescriptions::addDefault(), and submitPVResolutionJobs::desc.

                                                                                       {
   // The following says we do not know what parameters are allowed so do no validation
   //  Please change this to state exactly what you do use, even if it is no parameters
   edm::ParameterSetDescription desc;
   desc.setUnknown();
   descriptions.addDefault(desc);
 }

◆ find_original_muon()

const reco::Candidate * EmbeddingLHEProducer::find_original_muon ( const reco::Candidate * muon )

private

Definition at line 367 of file EmbeddingLHEProducer.cc.

References HLT_2024v14_cff::muon.

Referenced by assign_4vector().

                                                                                        {
   if (muon->mother(0) == nullptr)
     return muon;
   if (muon->pdgId() == muon->mother(0)->pdgId())
     return find_original_muon(muon->mother(0));
   else
     return muon;
 }

◆ give_slha()

LHERunInfoProduct::Header EmbeddingLHEProducer::give_slha ( )

private

Definition at line 466 of file EmbeddingLHEProducer.cc.

References LHERunInfoProduct::Header::addLine().

Referenced by beginRunProduce().

                                                         {
   LHERunInfoProduct::Header slhah("slha");
 
   slhah.addLine("######################################################################\n");
   slhah.addLine("## PARAM_CARD AUTOMATICALY GENERATED BY MG5 FOLLOWING UFO MODEL   ####\n");
   slhah.addLine("######################################################################\n");
   slhah.addLine("##                                                                  ##\n");
   slhah.addLine("##  Width set on Auto will be computed following the information    ##\n");
   slhah.addLine("##        present in the decay.py files of the model.               ##\n");
   slhah.addLine("##        See  arXiv:1402.1178 for more details.                    ##\n");
   slhah.addLine("##                                                                  ##\n");
   slhah.addLine("######################################################################\n");
   slhah.addLine("\n");
   slhah.addLine("###################################\n");
   slhah.addLine("## INFORMATION FOR MASS\n");
   slhah.addLine("###################################\n");
   slhah.addLine("Block mass \n");
   slhah.addLine("    6 1.730000e+02 # MT \n");
   slhah.addLine("   15 1.777000e+00 # MTA \n");
   slhah.addLine("   23 9.118800e+01 # MZ \n");
   slhah.addLine("   25 1.250000e+02 # MH \n");
   slhah.addLine("## Dependent parameters, given by model restrictions.\n");
   slhah.addLine("## Those values should be edited following the \n");
   slhah.addLine("## analytical expression. MG5 ignores those values \n");
   slhah.addLine("## but they are important for interfacing the output of MG5\n");
   slhah.addLine("## to external program such as Pythia.\n");
   slhah.addLine("  1 0.000000 # d : 0.0 \n");
   slhah.addLine("  2 0.000000 # u : 0.0 \n");
   slhah.addLine("  3 0.000000 # s : 0.0 \n");
   slhah.addLine("  4 0.000000 # c : 0.0 \n");
   slhah.addLine("  5 0.000000 # b : 0.0 \n");
   slhah.addLine("  11 0.000000 # e- : 0.0 \n");
   slhah.addLine("  12 0.000000 # ve : 0.0 \n");
   slhah.addLine("  13 0.000000 # mu- : 0.0 \n");
   slhah.addLine("  14 0.000000 # vm : 0.0 \n");
   slhah.addLine("  16 0.000000 # vt : 0.0 \n");
   slhah.addLine("  21 0.000000 # g : 0.0 \n");
   slhah.addLine("  22 0.000000 # a : 0.0 \n");
   slhah.addLine(
       "  24 80.419002 # w+ : cmath.sqrt(MZ__exp__2/2. + cmath.sqrt(MZ__exp__4/4. - "
       "(aEW*cmath.pi*MZ__exp__2)/(Gf*sqrt__2))) \n");
   slhah.addLine("\n");
   slhah.addLine("###################################\n");
   slhah.addLine("## INFORMATION FOR SMINPUTS\n");
   slhah.addLine("###################################\n");
   slhah.addLine("Block sminputs \n");
   slhah.addLine("    1 1.325070e+02 # aEWM1 \n");
   slhah.addLine("    2 1.166390e-05 # Gf \n");
   slhah.addLine("    3 1.180000e-01 # aS \n");
   slhah.addLine("\n");
   slhah.addLine("###################################\n");
   slhah.addLine("## INFORMATION FOR WOLFENSTEIN\n");
   slhah.addLine("###################################\n");
   slhah.addLine("Block wolfenstein \n");
   slhah.addLine("    1 2.253000e-01 # lamWS \n");
   slhah.addLine("    2 8.080000e-01 # AWS \n");
   slhah.addLine("    3 1.320000e-01 # rhoWS \n");
   slhah.addLine("    4 3.410000e-01 # etaWS \n");
   slhah.addLine("\n");
   slhah.addLine("###################################\n");
   slhah.addLine("## INFORMATION FOR YUKAWA\n");
   slhah.addLine("###################################\n");
   slhah.addLine("Block yukawa \n");
   slhah.addLine("    6 1.730000e+02 # ymt \n");
   slhah.addLine("   15 1.777000e+00 # ymtau \n");
   slhah.addLine("\n");
   slhah.addLine("###################################\n");
   slhah.addLine("## INFORMATION FOR DECAY\n");
   slhah.addLine("###################################\n");
   slhah.addLine("DECAY   6 1.491500e+00 # WT \n");
   slhah.addLine("DECAY  15 2.270000e-12 # WTau \n");
   slhah.addLine("DECAY  23 2.441404e+00 # WZ \n");
   slhah.addLine("DECAY  24 2.047600e+00 # WW \n");
   slhah.addLine("DECAY  25 6.382339e-03 # WH \n");
   slhah.addLine("## Dependent parameters, given by model restrictions.\n");
   slhah.addLine("## Those values should be edited following the \n");
   slhah.addLine("## analytical expression. MG5 ignores those values \n");
   slhah.addLine("## but they are important for interfacing the output of MG5\n");
   slhah.addLine("## to external program such as Pythia.\n");
   slhah.addLine("DECAY  1 0.000000 # d : 0.0 \n");
   slhah.addLine("DECAY  2 0.000000 # u : 0.0 \n");
   slhah.addLine("DECAY  3 0.000000 # s : 0.0 \n");
   slhah.addLine("DECAY  4 0.000000 # c : 0.0 \n");
   slhah.addLine("DECAY  5 0.000000 # b : 0.0 \n");
   slhah.addLine("DECAY  11 0.000000 # e- : 0.0 \n");
   slhah.addLine("DECAY  12 0.000000 # ve : 0.0 \n");
   slhah.addLine("DECAY  13 0.000000 # mu- : 0.0 \n");
   slhah.addLine("DECAY  14 0.000000 # vm : 0.0 \n");
   slhah.addLine("DECAY  16 0.000000 # vt : 0.0 \n");
   slhah.addLine("DECAY  21 0.000000 # g : 0.0 \n");
   slhah.addLine("DECAY  22 0.000000 # a : 0.0\n");
 
   return slhah;
 }

◆ InitialRecoCorrection()

void EmbeddingLHEProducer::InitialRecoCorrection	(	TLorentzVector &	positiveLepton,
		TLorentzVector &	negativeLepton
	)

private

Definition at line 406 of file EmbeddingLHEProducer.cc.

References ALPAKA_ACCELERATOR_NAMESPACE::brokenline::constexpr(), JetChargeProducer_cfi::exp, InitialRecoCorrection_, muonMass_, funct::pow(), mathSSE::sqrt(), and beamSpotPI::Z.

Referenced by produce().

                                                                                                                {
   if (!InitialRecoCorrection_)
     return;
   edm::LogInfo("TauEmbedding") << "Applying initial reconstruction correction";
   TLorentzVector Z = positiveLepton + negativeLepton;
   edm::LogInfo("TauEmbedding") << "MuMinus before. Pt: " << negativeLepton.Pt() << " Mass: " << negativeLepton.M();
   float diLeptonMass = (positiveLepton + negativeLepton).M();
   if (diLeptonMass > 60. && diLeptonMass < 122.) {
     static constexpr float zmass = 91.1876;
     float correction_deviation =
         5.;  // to ensure only a correction that drops corresponding to a Gaussian with mean zmass and std. dev. 5 GeV
     double EmbeddingCorrection =
         1.138;  // value derived by function fitting to fold embedded mass spectrum back to original selection when using mu -> mu embedding
     EmbeddingCorrection /=
         (EmbeddingCorrection -
          (EmbeddingCorrection - 1.) * exp(-pow((diLeptonMass - zmass), 2.) / (2. * pow(correction_deviation, 2.))));
     EmbeddingCorrection = ((diLeptonMass + (EmbeddingCorrection - 1.) * zmass) / (diLeptonMass * EmbeddingCorrection));
     double correctedpositiveLeptonEnergy =
         std::sqrt((pow(muonMass_, 2) / pow(EmbeddingCorrection, 2)) + pow(positiveLepton.Px(), 2) +
                   pow(positiveLepton.Py(), 2) + pow(positiveLepton.Pz(), 2));
     double correctednegativeLeptonEnergy =
         std::sqrt((pow(muonMass_, 2) / pow(EmbeddingCorrection, 2)) + pow(negativeLepton.Px(), 2) +
                   pow(negativeLepton.Py(), 2) + pow(negativeLepton.Pz(), 2));
     positiveLepton.SetE(correctedpositiveLeptonEnergy);
     negativeLepton.SetE(correctednegativeLeptonEnergy);
     positiveLepton *= EmbeddingCorrection;
     negativeLepton *= EmbeddingCorrection;
     edm::LogInfo("TauEmbedding") << "MuMinus after. Pt: " << negativeLepton.Pt() << " Mass: " << negativeLepton.M();
   }
   return;
 }

◆ mirror()

void EmbeddingLHEProducer::mirror	(	TLorentzVector &	positiveLepton,
		TLorentzVector &	negativeLepton
	)

private

Definition at line 438 of file EmbeddingLHEProducer.cc.

References EcalCondDBWriter_cfi::beam, mirror_, and beamSpotPI::Z.

Referenced by produce().

                                                                                                 {
   if (!mirror_)
     return;
   edm::LogInfo("TauEmbedding") << "Applying mirroring";
   TLorentzVector Z = positiveLepton + negativeLepton;
 
   edm::LogInfo("TauEmbedding") << "MuMinus before. Pt: " << negativeLepton.Pt() << " Eta: " << negativeLepton.Eta()
                                << " Phi: " << negativeLepton.Phi() << " Mass: " << negativeLepton.M();
 
   TVector3 Z3 = Z.Vect();
   TVector3 positiveLepton3 = positiveLepton.Vect();
   TVector3 negativeLepton3 = negativeLepton.Vect();
 
   TVector3 beam(0., 0., 1.);
   TVector3 perpToZandBeam = Z3.Cross(beam).Unit();
 
   positiveLepton3 -= 2 * positiveLepton3.Dot(perpToZandBeam) * perpToZandBeam;
   negativeLepton3 -= 2 * negativeLepton3.Dot(perpToZandBeam) * perpToZandBeam;
 
   positiveLepton.SetVect(positiveLepton3);
   negativeLepton.SetVect(negativeLepton3);
 
   edm::LogInfo("TauEmbedding") << "MuMinus after. Pt: " << negativeLepton.Pt() << " Eta: " << negativeLepton.Eta()
                                << " Phi: " << negativeLepton.Phi() << " Mass: " << negativeLepton.M();
 
   return;
 }

◆ produce()

void EmbeddingLHEProducer::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivatevirtual

Implements edm::one::EDProducerBase.

Definition at line 157 of file EmbeddingLHEProducer.cc.

References lhef::HEPEUP::AQCDUP, lhef::HEPEUP::AQEDUP, assign_4vector(), edm::View< T >::begin(), filterCSVwithJSON::copy, edm::View< T >::end(), file, fill_lhe_from_mumu(), edm::RandomNumberGenerator::getEngine(), lhef::HEPEUP::IDPRUP, iEvent, InitialRecoCorrection(), lhe2HepMCConverter_cff::LHEEventProduct, mirror(), eostools::move(), HLT_2024v14_cff::muon, muonsCollection_, rotate180(), lhef::HEPEUP::SCALUP, studyFSRmode_, transform_mumu_to_tautau(), vertexCollection_, write_lheout, and lhef::HEPEUP::XWGTUP.

                                                                                 {
   using namespace edm;
 
   edm::Service<edm::RandomNumberGenerator> rng;
   CLHEP::HepRandomEngine *engine = &rng->getEngine(iEvent.streamID());
 
   edm::Handle<edm::View<pat::Muon>> muonHandle;
   iEvent.getByToken(muonsCollection_, muonHandle);
   edm::View<pat::Muon> coll_muons = *muonHandle;
 
   Handle<std::vector<reco::Vertex>> coll_vertices;
   iEvent.getByToken(vertexCollection_, coll_vertices);
 
   TLorentzVector positiveLepton, negativeLepton;
   bool mu_plus_found = false;
   bool mu_minus_found = false;
   lhef::HEPEUP hepeup;
   hepeup.IDPRUP = 0;
   hepeup.XWGTUP = 1;
   hepeup.SCALUP = -1;
   hepeup.AQEDUP = -1;
   hepeup.AQCDUP = -1;
   // Assuming Pt-Order
   for (edm::View<pat::Muon>::const_iterator muon = coll_muons.begin(); muon != coll_muons.end(); ++muon) {
     if (muon->charge() == 1 && !mu_plus_found) {
       assign_4vector(positiveLepton, &(*muon), studyFSRmode_);
       mu_plus_found = true;
     } else if (muon->charge() == -1 && !mu_minus_found) {
       assign_4vector(negativeLepton, &(*muon), studyFSRmode_);
       mu_minus_found = true;
     } else if (mu_minus_found && mu_plus_found)
       break;
   }
   InitialRecoCorrection(
       positiveLepton,
       negativeLepton);  // corrects Z mass peak to take into account smearing happening due to first muon reconstruction in the selection step
   mirror(positiveLepton, negativeLepton);                    // if no mirror, function does nothing.
   rotate180(positiveLepton, negativeLepton);                 // if no rotate180, function does nothing
   transform_mumu_to_tautau(positiveLepton, negativeLepton);  // if MuonEmbedding, function does nothing.
   fill_lhe_from_mumu(positiveLepton, negativeLepton, hepeup, engine);
 
   double originalXWGTUP_ = 1.;
   std::unique_ptr<LHEEventProduct> product(new LHEEventProduct(hepeup, originalXWGTUP_));
 
   if (write_lheout)
     std::copy(product->begin(), product->end(), std::ostream_iterator<std::string>(file));
 
   iEvent.put(std::move(product));
   // Saving vertex position
   std::unique_ptr<math::XYZTLorentzVectorD> vertex_position(
       new math::XYZTLorentzVectorD(coll_vertices->at(0).x(), coll_vertices->at(0).y(), coll_vertices->at(0).z(), 0.0));
   iEvent.put(std::move(vertex_position), "vertexPosition");
 }

◆ rotate180()

void EmbeddingLHEProducer::rotate180	(	TLorentzVector &	positiveLepton,
		TLorentzVector &	negativeLepton
	)

private

Definition at line 376 of file EmbeddingLHEProducer.cc.

References rotate180_, and beamSpotPI::Z.

Referenced by produce().

                                                                                                    {
   if (!rotate180_)
     return;
   edm::LogInfo("TauEmbedding") << "Applying 180<C2><B0> rotation";
   // By construction, the 3-momenta of mu-, mu+ and Z are in one plane.
   // That means, one vector for perpendicular projection can be used for both leptons.
   TLorentzVector Z = positiveLepton + negativeLepton;
 
   edm::LogInfo("TauEmbedding") << "MuMinus before. Pt: " << negativeLepton.Pt() << " Eta: " << negativeLepton.Eta()
                                << " Phi: " << negativeLepton.Phi() << " Mass: " << negativeLepton.M();
 
   TVector3 Z3 = Z.Vect();
   TVector3 positiveLepton3 = positiveLepton.Vect();
   TVector3 negativeLepton3 = negativeLepton.Vect();
 
   TVector3 p3_perp = positiveLepton3 - positiveLepton3.Dot(Z3) / Z3.Dot(Z3) * Z3;
   p3_perp = p3_perp.Unit();
 
   positiveLepton3 -= 2 * positiveLepton3.Dot(p3_perp) * p3_perp;
   negativeLepton3 -= 2 * negativeLepton3.Dot(p3_perp) * p3_perp;
 
   positiveLepton.SetVect(positiveLepton3);
   negativeLepton.SetVect(negativeLepton3);
 
   edm::LogInfo("TauEmbedding") << "MuMinus after. Pt: " << negativeLepton.Pt() << " Eta: " << negativeLepton.Eta()
                                << " Phi: " << negativeLepton.Phi() << " Mass: " << negativeLepton.M();
 
   return;
 }

◆ transform_mumu_to_tautau()

void EmbeddingLHEProducer::transform_mumu_to_tautau	(	TLorentzVector &	positiveLepton,
		TLorentzVector &	negativeLepton
	)

private

Definition at line 306 of file EmbeddingLHEProducer.cc.

References elMass_, particleToEmbed_, isotrackNtupler::scale, mathSSE::sqrt(), tauMass_, and beamSpotPI::Z.

Referenced by produce().

                                                                                                                   {
   // No corrections applied for muon embedding
   double lep_mass;
   if (particleToEmbed_ == 11) {
     lep_mass = elMass_;
   } else if (particleToEmbed_ == 15) {
     lep_mass = tauMass_;
   } else {
     return;
   }
 
   TLorentzVector Z = positiveLepton + negativeLepton;
 
   TVector3 boost_from_Z_to_LAB = Z.BoostVector();
   TVector3 boost_from_LAB_to_Z = -Z.BoostVector();
   // Boosting the two leptons to Z restframe, then both are back to back. This means, same 3-momentum squared
   positiveLepton.Boost(boost_from_LAB_to_Z);
   negativeLepton.Boost(boost_from_LAB_to_Z);
 
   // Energy of tau = 0.5*Z-mass
   double lep_mass_squared = lep_mass * lep_mass;
   double lep_energy_squared = 0.25 * Z.M2();
   double lep_3momentum_squared = lep_energy_squared - lep_mass_squared;
   if (lep_3momentum_squared < 0) {
     edm::LogWarning("TauEmbedding") << "3-Momentum squared is negative";
     return;
   }
 
   // Computing scale, applying it on the 3-momenta and building new 4 momenta of the taus
   double scale = std::sqrt(lep_3momentum_squared / positiveLepton.Vect().Mag2());
   positiveLepton.SetPxPyPzE(scale * positiveLepton.Px(),
                             scale * positiveLepton.Py(),
                             scale * positiveLepton.Pz(),
                             std::sqrt(lep_energy_squared));
   negativeLepton.SetPxPyPzE(scale * negativeLepton.Px(),
                             scale * negativeLepton.Py(),
                             scale * negativeLepton.Pz(),
                             std::sqrt(lep_energy_squared));
 
   // Boosting the new taus back to LAB frame
   positiveLepton.Boost(boost_from_Z_to_LAB);
   negativeLepton.Boost(boost_from_Z_to_LAB);
 
   return;
 }

Member Data Documentation

◆ elMass_

constexpr double EmbeddingLHEProducer::elMass_ = 0.00051

staticprivate

Definition at line 99 of file EmbeddingLHEProducer.cc.

Referenced by transform_mumu_to_tautau().

◆ embeddingParticles

const int EmbeddingLHEProducer::embeddingParticles[3] {11, 13, 15}

private

Definition at line 100 of file EmbeddingLHEProducer.cc.

Referenced by EmbeddingLHEProducer(), and fill_lhe_with_particle().

◆ file

std::ofstream EmbeddingLHEProducer::file

private

Definition at line 102 of file EmbeddingLHEProducer.cc.

Referenced by beginRunProduce(), EmbeddingLHEProducer(), endRunProduce(), and produce().

◆ InitialRecoCorrection_

bool EmbeddingLHEProducer::InitialRecoCorrection_

private

Definition at line 96 of file EmbeddingLHEProducer.cc.

Referenced by EmbeddingLHEProducer(), and InitialRecoCorrection().

◆ mirror_

bool EmbeddingLHEProducer::mirror_

private

Definition at line 96 of file EmbeddingLHEProducer.cc.

Referenced by EmbeddingLHEProducer(), and mirror().

◆ muonMass_

constexpr double EmbeddingLHEProducer::muonMass_ = 0.1057

staticprivate

Definition at line 98 of file EmbeddingLHEProducer.cc.

Referenced by InitialRecoCorrection().

◆ muonsCollection_

edm::EDGetTokenT<edm::View<pat::Muon> > EmbeddingLHEProducer::muonsCollection_

private

Definition at line 93 of file EmbeddingLHEProducer.cc.

Referenced by EmbeddingLHEProducer(), and produce().

◆ particleToEmbed_

int EmbeddingLHEProducer::particleToEmbed_

private

Definition at line 95 of file EmbeddingLHEProducer.cc.

Referenced by EmbeddingLHEProducer(), fill_lhe_from_mumu(), and transform_mumu_to_tautau().

◆ rotate180_

bool EmbeddingLHEProducer::rotate180_

private

Definition at line 96 of file EmbeddingLHEProducer.cc.

Referenced by EmbeddingLHEProducer(), and rotate180().

◆ studyFSRmode_

std::string EmbeddingLHEProducer::studyFSRmode_

private

Definition at line 108 of file EmbeddingLHEProducer.cc.

Referenced by EmbeddingLHEProducer(), and produce().

◆ tauMass_

constexpr double EmbeddingLHEProducer::tauMass_ = 1.77682

staticprivate

Definition at line 97 of file EmbeddingLHEProducer.cc.

Referenced by transform_mumu_to_tautau().

◆ vertexCollection_

edm::EDGetTokenT<reco::VertexCollection> EmbeddingLHEProducer::vertexCollection_

private

Definition at line 94 of file EmbeddingLHEProducer.cc.

Referenced by EmbeddingLHEProducer(), and produce().

◆ write_lheout

bool EmbeddingLHEProducer::write_lheout

private

Definition at line 103 of file EmbeddingLHEProducer.cc.

Referenced by beginRunProduce(), EmbeddingLHEProducer(), endRunProduce(), and produce().

Public Member Functions

Static Public Member Functions

Private Member Functions

Private Attributes

Static Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ EmbeddingLHEProducer()

◆ ~EmbeddingLHEProducer()

Member Function Documentation

◆ assign_4vector()

◆ beginJob()

◆ beginRunProduce()

◆ endJob()

◆ endRunProduce()

◆ fill_lhe_from_mumu()

◆ fill_lhe_with_particle()

◆ fillDescriptions()

◆ find_original_muon()

◆ give_slha()

◆ InitialRecoCorrection()

◆ mirror()

◆ produce()

◆ rotate180()

◆ transform_mumu_to_tautau()

Member Data Documentation

◆ elMass_

◆ embeddingParticles

◆ file

◆ InitialRecoCorrection_

◆ mirror_

◆ muonMass_

◆ muonsCollection_

◆ particleToEmbed_

◆ rotate180_

◆ studyFSRmode_

◆ tauMass_

◆ vertexCollection_

◆ write_lheout