GeneratorInterface/Pythia6Interface/plugins/Pythia6PartonPtGun.cc - master - githubmirror/cmssw

GeneratorInterface/Pythia6Interface/plugins/Pythia6PartonPtGun.cc
98 строк3 KB
Christopher Jones
Remved unnecessary include
14 окт 2021, 00:26
14 окт 2021, 00:26b4ee2fd

#include <iostream>

#include "Pythia6PartonPtGun.h"

#include "FWCore/Utilities/interface/Exception.h"

#include "FWCore/Framework/interface/EventSetup.h"

#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"

#include "FWCore/Framework/interface/MakerMacros.h"

using namespace edm;
using namespace gen;

Pythia6PartonPtGun::Pythia6PartonPtGun(const ParameterSet& pset) : Pythia6PartonGun(pset) {
  // ParameterSet defpset ;
  ParameterSet pgun_params = pset.getParameter<ParameterSet>("PGunParameters");  //, defpset ) ;
  fMinEta = pgun_params.getParameter<double>("MinEta");                          // ,-2.2);
  fMaxEta = pgun_params.getParameter<double>("MaxEta");                          // , 2.2);
  fMinPt = pgun_params.getParameter<double>("MinPt");                            // ,  20.);
  fMaxPt = pgun_params.getParameter<double>("MaxPt");                            // , 420.);
}

Pythia6PartonPtGun::~Pythia6PartonPtGun() {}

void Pythia6PartonPtGun::generateEvent(CLHEP::HepRandomEngine*) {
  Pythia6Service::InstanceWrapper guard(fPy6Service);  // grab Py6 instance

  // now actualy, start cooking up the event gun
  //

  // 1st, primary vertex
  //
  HepMC::GenVertex* Vtx = new HepMC::GenVertex(HepMC::FourVector(0., 0., 0.));

  // here re-create fEvt (memory)
  //
  fEvt = new HepMC::GenEvent();

  int ip = 1;

  int py6PID = HepPID::translatePDTtoPythia(fPartonID);
  int dum = 0;
  double pt = 0, mom = 0, ee = 0, the = 0, eta = 0;
  double mass = pymass_(py6PID);

  // fill p(ip,5) (in PYJETS) with mass value right now,
  // because the (hardcoded) mstu(10)=1 will make py1ent
  // pick the mass from there
  pyjets.p[4][ip - 1] = mass;

  double phi = (fMaxPhi - fMinPhi) * pyr_(&dum) + fMinPhi;

  eta = (fMaxEta - fMinEta) * pyr_(&dum) + fMinEta;

  the = 2. * atan(exp(-eta));

  pt = (fMaxPt - fMinPt) * pyr_(&dum) + fMinPt;

  mom = pt / sin(the);
  ee = sqrt(mom * mom + mass * mass);

  py1ent_(ip, py6PID, ee, the, phi);

  double px = pyjets.p[0][ip - 1];  // pt*cos(phi) ;
  double py = pyjets.p[1][ip - 1];  // pt*sin(phi) ;
  double pz = pyjets.p[2][ip - 1];  // mom*cos(the) ;

  HepMC::FourVector p(px, py, pz, ee);
  HepMC::GenParticle* Part = new HepMC::GenParticle(p, fPartonID, 1);
  Part->suggest_barcode(ip);
  Vtx->add_particle_out(Part);

  // now add anti-quark
  ip = ip + 1;
  HepMC::GenParticle* APart = addAntiParticle(ip, fPartonID, ee, eta, phi);
  if (APart) {
    Vtx->add_particle_out(APart);
  } else {
    // otherwise it should throw !
  }

  // this should probably be configurable...
  //
  double qmax = 2. * ee;

  joinPartons(qmax);

  fEvt->add_vertex(Vtx);

  // run pythia
  pyexec_();

  return;
}

DEFINE_FWK_MODULE(Pythia6PartonPtGun);