Generator Class Reference

#include <Generator.h>

Public Member Functions
virtual const double	eventWeight () const
	Generator (const edm::ParameterSet &p)
virtual const HepMC::GenEvent *	genEvent () const
virtual const math::XYZTLorentzVector *	genVertex () const
void	HepMC2G4 (const HepMC::GenEvent *g, G4Event *e)
void	nonCentralEvent2G4 (const HepMC::GenEvent *g, G4Event *e)
void	setGenEvent (const HepMC::GenEvent *inpevt)
virtual	~Generator ()

Private Member Functions
bool	isExotic (int pdgcode) const
bool	isExoticNonDetectable (int pdgcode) const
bool	IsInTheFilterList (int pdgcode) const
void	particleAssignDaughters (G4PrimaryParticle *p, HepMC::GenParticle *hp, double length)
bool	particlePassesPrimaryCuts (const G4ThreeVector &p) const
void	setGenId (G4PrimaryParticle *p, int id) const

Private Attributes
HepMC::GenEvent *	evt_
bool	fEtaCuts
bool	fFiductialCuts
LumiMonitorFilter *	fLumiFilter
bool	fPCuts
bool	fPDGFilter
bool	fPhiCuts
bool	fPtransCut
double	maxZCentralCMS
std::vector< int >	pdgFilter
bool	pdgFilterSel
double	theDecLenCut
double	theDecRCut2
double	theEtaCutForHector
double	theMaxEtaCut
double	theMaxPCut
double	theMaxPhiCut
double	theMinEtaCut
double	theMinPCut
double	theMinPhiCut
double	theMinPtCut2
int	verbose
math::XYZTLorentzVector *	vtx_
double	weight_
double	Z_hector
double	Z_lmax
double	Z_lmin

Detailed Description

Definition at line 19 of file Generator.h.

Constructor & Destructor Documentation

Generator()

Generator::Generator

(

const edm::ParameterSet &

p

)

Definition at line 23 of file Generator.cc.

References LumiMonitorFilter::Describe(), JetChargeProducer_cfi::exp, fEtaCuts, fFiductialCuts, fLumiFilter, fPCuts, fPDGFilter, fPhiCuts, fPtransCut, cuy::ii, BXlumiParameters_cfi::lumi, visualization-live-secondInstance_cfg::m, maxZCentralCMS, AlCaHLTBitMon_ParallelJobs::p, pdgFilter, pdgFilterSel, contentValuesCheck::ss, theDecLenCut, theDecRCut2, theEtaCutForHector, theMaxEtaCut, theMaxPCut, theMaxPhiCut, theMinEtaCut, theMinPCut, theMinPhiCut, theMinPtCut2, Z_hector, Z_lmax, and Z_lmin.

    : fPCuts(p.getParameter<bool>("ApplyPCuts")),
      fPtransCut(p.getParameter<bool>("ApplyPtransCut")),
      fEtaCuts(p.getParameter<bool>("ApplyEtaCuts")),
      fPhiCuts(p.getParameter<bool>("ApplyPhiCuts")),
      theMinPhiCut(p.getParameter<double>("MinPhiCut")),  // in radians (CMS standard)
      theMaxPhiCut(p.getParameter<double>("MaxPhiCut")),
      theMinEtaCut(p.getParameter<double>("MinEtaCut")),
      theMaxEtaCut(p.getParameter<double>("MaxEtaCut")),
      theMinPCut(p.getParameter<double>("MinPCut")),  // in GeV (CMS standard)
      theMaxPCut(p.getParameter<double>("MaxPCut")),
      theEtaCutForHector(p.getParameter<double>("EtaCutForHector")),
      verbose(p.getUntrackedParameter<int>("Verbosity", 0)),
      fLumiFilter(nullptr),
      evt_(nullptr),
      vtx_(nullptr),
      weight_(0),
      Z_lmin(0),
      Z_lmax(0),
      Z_hector(0),
      pdgFilterSel(false),
      fPDGFilter(false) {
  bool lumi = p.getParameter<bool>("ApplyLumiMonitorCuts");
  if (lumi) {
    fLumiFilter = new LumiMonitorFilter();
  }

  double theRDecLenCut = p.getParameter<double>("RDecLenCut") * CLHEP::cm;
  theDecRCut2 = theRDecLenCut * theRDecLenCut;

  theMinPtCut2 = theMinPCut * theMinPCut;

  double theDecLenCut = p.getParameter<double>("LDecLenCut") * CLHEP::cm;

  maxZCentralCMS = p.getParameter<double>("MaxZCentralCMS") * CLHEP::m;

  fFiductialCuts = (fPCuts || fPtransCut || fEtaCuts || fPhiCuts);

  pdgFilter.resize(0);
  if (p.exists("PDGselection")) {
    pdgFilterSel = (p.getParameter<edm::ParameterSet>("PDGselection")).getParameter<bool>("PDGfilterSel");
    pdgFilter = (p.getParameter<edm::ParameterSet>("PDGselection")).getParameter<std::vector<int>>("PDGfilter");
    if (!pdgFilter.empty()) {
      fPDGFilter = true;
      std::stringstream ss;
      ss << "SimG4Core/Generator: ";
      if (pdgFilterSel) {
        ss << " Selecting only PDG ID = ";
      } else {
        ss << " Filtering out PDG ID = ";
      }
      for (unsigned int ii = 0; ii < pdgFilter.size(); ++ii) {
        ss << pdgFilter[ii] << "  ";
      }
      edm::LogVerbatim("SimG4CoreGenerator") << ss.str();
    }
  }

  if (fEtaCuts) {
    Z_lmax = theRDecLenCut * ((1 - exp(-2 * theMaxEtaCut)) / (2 * exp(-theMaxEtaCut)));
    Z_lmin = theRDecLenCut * ((1 - exp(-2 * theMinEtaCut)) / (2 * exp(-theMinEtaCut)));
  }

  Z_hector = theRDecLenCut * ((1 - exp(-2 * theEtaCutForHector)) / (2 * exp(-theEtaCutForHector)));

  edm::LogVerbatim("SimG4CoreGenerator") << "SimG4Core/Generator: Rdecaycut= " << theRDecLenCut / CLHEP::cm
                                         << " cm;  Zdecaycut= " << theDecLenCut / CLHEP::cm
                                         << "Z_min= " << Z_lmin / CLHEP::cm << " cm; Z_max= " << Z_lmax / CLHEP::cm
                                         << " cm;\n"
                                         << "                     MaxZCentralCMS = " << maxZCentralCMS / CLHEP::m
                                         << " m;"
                                         << " Z_hector = " << Z_hector / CLHEP::cm << " cm\n"
                                         << "                     ApplyCuts: " << fFiductialCuts
                                         << "  PCuts: " << fPCuts << "  PtransCut: " << fPtransCut
                                         << "  EtaCut: " << fEtaCuts << "  PhiCut: " << fPhiCuts
                                         << "  LumiMonitorCut: " << lumi;
  if (fFiductialCuts) {
    edm::LogVerbatim("SimG4CoreGenerator")
        << "SimG4Core/Generator: Pmin(GeV)= " << theMinPCut << "; Pmax(GeV)= " << theMaxPCut
        << "; EtaMin= " << theMinEtaCut << "; EtaMax= " << theMaxEtaCut << "; PhiMin(rad)= " << theMinPhiCut
        << "; PhiMax(rad)= " << theMaxPhiCut;
  }
  if (lumi) {
    fLumiFilter->Describe();
  }
}

~Generator()

Generator::~Generator ( )

virtual

Definition at line 110 of file Generator.cc.

References fLumiFilter.

{ delete fLumiFilter; }

Member Function Documentation

eventWeight()

virtual const double Generator::eventWeight ( ) const

inlinevirtual

Definition at line 32 of file Generator.h.

References weight_.

Referenced by RunManagerMTWorker::produce().

{ return weight_; }

genEvent()

virtual const HepMC::GenEvent* Generator::genEvent ( ) const

inlinevirtual

Definition at line 30 of file Generator.h.

References evt_.

Referenced by RunManagerMTWorker::produce().

{ return evt_; }

genVertex()

virtual const math::XYZTLorentzVector* Generator::genVertex ( ) const

inlinevirtual

Definition at line 31 of file Generator.h.

References vtx_.

Referenced by RunManagerMTWorker::produce().

{ return vtx_; }

HepMC2G4()

void Generator::HepMC2G4	(	const HepMC::GenEvent *	g,
		G4Event *	e
	)

Definition at line 112 of file Generator.cc.

References funct::abs(), ALCARECOTkAlJpsiMuMu_cff::charge, class-composition::children, fEtaCuts, fFiductialCuts, fLumiFilter, fPCuts, fPDGFilter, fPhiCuts, fPtransCut, GenParticle::GenParticle, isExotic(), isExoticNonDetectable(), LumiMonitorFilter::isGoodForLumiMonitor(), IsInTheFilterList(), LogDebug, visualization-live-secondInstance_cfg::m, maxZCentralCMS, AlCaHLTBitMon_ParallelJobs::p, particleAssignDaughters(), dileptonTrigSettings_cff::pdg, pdgFilterSel, phi, multPhiCorr_741_25nsDY_cfi::px, multPhiCorr_741_25nsDY_cfi::py, setGenId(), mathSSE::sqrt(), contentValuesCheck::ss, mps_update::status, RandomServiceHelper::t1, theDecRCut2, theMaxPCut, theMaxPhiCut, theMinPtCut2, verbose, vtx_, weight_, testProducerWithPsetDescEmpty_cfi::x1, testProducerWithPsetDescEmpty_cfi::x2, protons_cff::xi, geometryCSVtoXML::xx, testProducerWithPsetDescEmpty_cfi::y1, testProducerWithPsetDescEmpty_cfi::y2, geometryCSVtoXML::yy, testProducerWithPsetDescEmpty_cfi::z2, Z_hector, Z_lmax, and Z_lmin.

Referenced by RunManagerMTWorker::generateEvent().

                                                                      {
  HepMC::GenEvent *evt = new HepMC::GenEvent(*evt_orig);

  if (*(evt->vertices_begin()) == nullptr) {
    std::stringstream ss;
    ss << "SimG4Core/Generator: in event " << g4evt->GetEventID() << " Corrupted Event - GenEvent with no vertex \n";
    throw SimG4Exception(ss.str());
  }

  if (!evt->weights().empty()) {
    weight_ = evt->weights()[0];
    for (unsigned int iw = 1; iw < evt->weights().size(); ++iw) {
      // terminate if the vector of weights contains a zero-weight
      if (evt->weights()[iw] <= 0)
        break;
      weight_ *= evt->weights()[iw];
    }
  }

  if (vtx_ != nullptr) {
    delete vtx_;
  }
  vtx_ = new math::XYZTLorentzVector((*(evt->vertices_begin()))->position().x(),
                                     (*(evt->vertices_begin()))->position().y(),
                                     (*(evt->vertices_begin()))->position().z(),
                                     (*(evt->vertices_begin()))->position().t());

  edm::LogVerbatim("SimG4CoreGenerator") << "Generator: primary Vertex = (" << vtx_->x() << ", " << vtx_->y() << ", "
                                         << vtx_->z() << ")";

  unsigned int ng4vtx = 0;
  unsigned int ng4par = 0;

  for (HepMC::GenEvent::vertex_const_iterator vitr = evt->vertices_begin(); vitr != evt->vertices_end(); ++vitr) {
    // loop for vertex, is it a real vertex?
    // Set qvtx to true for any particles that should be propagated by GEANT,
    // i.e., status 1 particles or status 2 particles that decay outside the
    // beampipe.
    G4bool qvtx = false;
    HepMC::GenVertex::particle_iterator pitr;
    for (pitr = (*vitr)->particles_begin(HepMC::children); pitr != (*vitr)->particles_end(HepMC::children); ++pitr) {
      // For purposes of this function, the status is defined as follows:
      // 1:  particles are not decayed by generator
      // 2:  particles are decayed by generator but need to be propagated by GEANT
      // 3:  particles are decayed by generator and do not need to be propagated by GEANT
      int status = (*pitr)->status();
      int pdg = (*pitr)->pdg_id();
      if (status > 3 && isExotic(pdg) && (!(isExoticNonDetectable(pdg)))) {
        // In Pythia 8, there are many status codes besides 1, 2, 3.
        // By setting the status to 2 for exotic particles, they will be
        // checked: if its decay vertex is outside the beampipe, it will be
        // propagated by GEANT. Some Standard Model particles, e.g., K0, cannot
        // be propagated by GEANT, so do not change their status code.
        status = 2;
      }
      if (status == 2 && abs(pdg) == 9900015) {
        status = 3;
      }

      // Particles which are not decayed by generator
      if (status == 1) {
        // filter out unwanted particles and vertices
        if (fPDGFilter && !pdgFilterSel && IsInTheFilterList(pdg)) {
          continue;
        }

        qvtx = true;
        if (verbose > 2)
          LogDebug("SimG4CoreGenerator") << "GenVertex barcode = " << (*vitr)->barcode()
                                         << " selected for GenParticle barcode = " << (*pitr)->barcode();
        break;
      }
      // The selection is made considering if the partcile with status = 2
      // have the end_vertex with a radius greater than the radius of beampipe
      // cylinder (no requirement on the Z of the vertex is applyed).
      else if (status == 2) {
        if ((*pitr)->end_vertex() != nullptr) {
          double xx = (*pitr)->end_vertex()->position().x();
          double yy = (*pitr)->end_vertex()->position().y();
          double r_dd = xx * xx + yy * yy;
          if (r_dd > theDecRCut2) {
            qvtx = true;
            if (verbose > 2)
              LogDebug("SimG4CoreGenerator")
                  << "GenVertex barcode = " << (*vitr)->barcode()
                  << " selected for GenParticle barcode = " << (*pitr)->barcode() << " radius = " << std::sqrt(r_dd);
            break;
          }
        } else {
          // particles with status 2 without end_vertex are
          // equivalent to stable
          qvtx = true;
          break;
        }
      }
    }

    // if this vertex is inside fiductial volume inside the beam pipe
    // and has no long-lived secondary the vertex is not saved
    if (!qvtx) {
      continue;
    }

    double x1 = (*vitr)->position().x() * CLHEP::mm;
    double y1 = (*vitr)->position().y() * CLHEP::mm;
    double z1 = (*vitr)->position().z() * CLHEP::mm;
    double t1 = (*vitr)->position().t() * CLHEP::mm / CLHEP::c_light;

    G4PrimaryVertex *g4vtx = new G4PrimaryVertex(x1, y1, z1, t1);

    for (pitr = (*vitr)->particles_begin(HepMC::children); pitr != (*vitr)->particles_end(HepMC::children); ++pitr) {
      int status = (*pitr)->status();
      int pdg = (*pitr)->pdg_id();
      bool hasDecayVertex = (nullptr != (*pitr)->end_vertex());

      // Filter on allowed particle species if required
      if (fPDGFilter) {
        bool isInTheList = IsInTheFilterList(pdg);
        if ((!pdgFilterSel && isInTheList) || (pdgFilterSel && !isInTheList)) {
          if (0 < verbose)
            edm::LogVerbatim("SimG4CoreGenerator")
                << " Skiped GenParticle barcode= " << (*pitr)->barcode() << " PDGid= " << pdg << " status= " << status
                << " isExotic: " << isExotic(pdg) << " isExoticNotDet: " << isExoticNonDetectable(pdg)
                << " isInTheList: " << isInTheList << " hasDecayVertex: " << hasDecayVertex;
          continue;
        }
      }

      if (0 < verbose) {
        edm::LogVerbatim("SimG4CoreGenerator")
            << "Generator: pdg= " << pdg << " status= " << status << " hasPreDefinedDecay: " << hasDecayVertex
            << "\n           isExotic: " << isExotic(pdg) << " isExoticNotDet: " << isExoticNonDetectable(pdg)
            << " isInTheList: " << IsInTheFilterList(pdg) << "\n"
            << " MaxZCentralCMS = " << maxZCentralCMS / CLHEP::m << " m;  (x,y,z,t): (" << x1 << "," << y1 << "," << z1
            << "," << t1 << ")";
      }
      if (status > 3 && isExotic(pdg) && (!(isExoticNonDetectable(pdg)))) {
        status = hasDecayVertex ? 2 : 1;
      }
      if (status == 2 && abs(pdg) == 9900015) {
        status = 3;
      }

      // this particle has predefined decay but has no vertex
      if (2 == status && !hasDecayVertex) {
        edm::LogWarning("SimG4CoreGenerator: in event ")
            << g4evt->GetEventID() << " a particle "
            << " pdgid= " << pdg << " has status=2 but has no decay vertex, so will be fully tracked by Geant4";
        status = 1;
      }

      double x2 = x1;
      double y2 = y1;
      double z2 = z1;
      double decay_length = 0.0;
      if (2 == status) {
        x2 = (*pitr)->end_vertex()->position().x();
        y2 = (*pitr)->end_vertex()->position().y();
        z2 = (*pitr)->end_vertex()->position().z();
        decay_length = std::sqrt((x2 - x1) * (x2 - x1) + (y2 - y1) * (y2 - y1) + (z2 - z1) * (z2 - z1));
      }

      bool toBeAdded = !fFiductialCuts;

      double px = (*pitr)->momentum().px();
      double py = (*pitr)->momentum().py();
      double pz = (*pitr)->momentum().pz();
      double ptot = std::sqrt(px * px + py * py + pz * pz);
      math::XYZTLorentzVector p(px, py, pz, (*pitr)->momentum().e());

      double ximpact = x1;
      double yimpact = y1;
      double zimpact = z1;

      // protection against numerical problems for extremely low momenta
      // compute impact point at transition to Hector
      const double minTan = 1.e-20;
      if (std::abs(z1) < Z_hector && std::abs(pz) >= minTan * ptot) {
        zimpact = (pz > 0.0) ? Z_hector : -Z_hector;
        double del = (zimpact - z1) / pz;
        ximpact += del * px;
        yimpact += del * py;
      }
      double rimpact2 = ximpact * ximpact + yimpact * yimpact;

      if (verbose > 2)
        LogDebug("SimG4CoreGenerator") << "Processing GenParticle barcode= " << (*pitr)->barcode() << " pdg= " << pdg
                                       << " status= " << (*pitr)->status() << " st= " << status
                                       << " rimpact(cm)= " << std::sqrt(rimpact2) / cm
                                       << " zimpact(cm)= " << zimpact / cm << " ptot(GeV)= " << ptot
                                       << " pz(GeV)= " << pz;

      // Particles of status 1 trasnported along the beam pipe
      // HECTOR transport of protons are done in corresponding PPS producer
      if (1 == status && std::abs(zimpact) >= Z_hector && rimpact2 <= theDecRCut2) {
        // very forward n, nbar, gamma are allowed
        toBeAdded = (2112 == std::abs(pdg) || 22 == pdg);
        if (verbose > 1) {
          edm::LogVerbatim("SimG4CoreGenerator")
              << "GenParticle barcode = " << (*pitr)->barcode() << " very forward; to be added: " << toBeAdded;
        }
      } else {
        // Standard case: particles not decayed by the generator and not forward
        if (1 == status && (std::abs(zimpact) < Z_hector || rimpact2 > theDecRCut2)) {
          // Ptot cut for all particles
          if (fPCuts && (ptot < theMinPCut || ptot > theMaxPCut)) {
            continue;
          }
          // phi cut is applied mainly for particle gun
          if (fPhiCuts) {
            double phi = p.phi();
            if (phi < theMinPhiCut || phi > theMaxPhiCut) {
              continue;
            }
          }
          // eta cut is applied if position of the decay
          // is within vacuum chamber and limited in Z
          if (fEtaCuts) {
            // eta cut
            double xi = x1;
            double yi = y1;
            double zi = z1;

            // can be propagated along Z
            if (std::abs(pz) >= minTan * ptot) {
              if ((zi >= Z_lmax) & (pz < 0.0)) {
                zi = Z_lmax;
              } else if ((zi <= Z_lmin) & (pz > 0.0)) {
                zi = Z_lmin;
              } else {
                if (pz > 0) {
                  zi = Z_lmax;
                } else {
                  zi = Z_lmin;
                }
              }
              double del = (zi - z1) / pz;
              xi += del * px;
              yi += del * py;
            }
            // check eta cut
            if ((zi >= Z_lmin) & (zi <= Z_lmax) & (xi * xi + yi * yi < theDecRCut2)) {
              continue;
            }
          }
          if (fLumiFilter && !fLumiFilter->isGoodForLumiMonitor(*pitr)) {
            continue;
          }
          toBeAdded = true;
          if (verbose > 1)
            edm::LogVerbatim("SimG4CoreGenerator")
                << "GenParticle barcode = " << (*pitr)->barcode() << " passed case 1";

          // Decay chain outside the fiducial cylinder defined by theRDecLenCut
          // are used for Geant4 tracking with predefined decay channel
          // In the case of decay in vacuum particle is not tracked by Geant4
        } else if (2 == status && x2 * x2 + y2 * y2 >= theDecRCut2 && std::abs(z2) < Z_hector) {
          toBeAdded = true;
          if (verbose > 1)
            edm::LogVerbatim("SimG4CoreGenerator") << "GenParticle barcode = " << (*pitr)->barcode() << " passed case 2"
                                                   << " decay_length(cm)= " << decay_length / CLHEP::cm;
        }
      }
      if (toBeAdded) {
        G4PrimaryParticle *g4prim = new G4PrimaryParticle(pdg, px * GeV, py * GeV, pz * GeV);

        if (g4prim->GetG4code() != nullptr) {
          g4prim->SetMass(g4prim->GetG4code()->GetPDGMass());
          double charge = g4prim->GetG4code()->GetPDGCharge();

          // apply Pt cut
          if (fPtransCut && 1 == status && 0.0 != charge && px * px + py * py < theMinPtCut2) {
            delete g4prim;
            continue;
          }
          g4prim->SetCharge(charge);
        }

        // V.I. do not use SetWeight but the same code
        // value of the code compute inside TrackWithHistory
        // g4prim->SetWeight( 10000*(*vpitr)->barcode() ) ;
        setGenId(g4prim, (*pitr)->barcode());

        if (2 == status) {
          particleAssignDaughters(g4prim, (HepMC::GenParticle *)*pitr, decay_length);
        }
        if (verbose > 1)
          g4prim->Print();

        ++ng4par;
        g4vtx->SetPrimary(g4prim);
        edm::LogVerbatim("SimG4CoreGenerator") << "   " << ng4par << ". new Geant4 particle pdg= " << pdg
                                               << " Ptot(GeV/c)= " << ptot << " Pt= " << std::sqrt(px * px + py * py)
                                               << " status= " << status << "; dir= " << g4prim->GetMomentumDirection();
      }
    }

    if (verbose > 1)
      g4vtx->Print();
    g4evt->AddPrimaryVertex(g4vtx);
    ++ng4vtx;
  }

  // Add a protection for completely empty events (produced by LHCTransport):
  // add a dummy vertex with no particle attached to it
  if (ng4vtx == 0) {
    G4PrimaryVertex *g4vtx = new G4PrimaryVertex(0.0, 0.0, 0.0, 0.0);
    if (verbose > 1)
      g4vtx->Print();

    g4evt->AddPrimaryVertex(g4vtx);
  }

  edm::LogVerbatim("SimG4CoreGenerator") << "The list of Geant4 primaries includes " << ng4par << " particles in "
                                         << ng4vtx << " vertex";
  delete evt;
}

isExotic()

bool Generator::isExotic ( int  pdgcode ) const

private

Definition at line 527 of file Generator.cc.

References funct::abs(), and EgammaValidation_cff::pdgid.

Referenced by HepMC2G4().

                                          {
  int pdgid = std::abs(pdgcode);
  return ((pdgid >= 1000000 && pdgid < 4000000 && pdgid != 3000022) ||  // SUSY, R-hadron, and technicolor particles
          pdgid == 17 ||                                                // 4th generation lepton
          pdgid == 34 ||                                                // W-prime
          pdgid == 37);                                                 // charged Higgs
}

isExoticNonDetectable()

bool Generator::isExoticNonDetectable ( int  pdgcode ) const

private

Definition at line 535 of file Generator.cc.

References funct::abs(), ALCARECOTkAlJpsiMuMu_cff::charge, LHEGenericFilter_cfi::ParticleID, and EgammaValidation_cff::pdgid.

Referenced by HepMC2G4().

                                                       {
  int pdgid = std::abs(pdgcode);
  HepPDT::ParticleID pid(pdgcode);
  int charge = pid.threeCharge();
  return ((charge == 0) && (pdgid >= 1000000 && pdgid < 1000040));  // SUSY
}

IsInTheFilterList()

bool Generator::IsInTheFilterList ( int  pdgcode ) const

private

Definition at line 542 of file Generator.cc.

References funct::abs(), pdgFilter, and EgammaValidation_cff::pdgid.

Referenced by HepMC2G4().

                                                   {
  int pdgid = std::abs(pdgcode);
  for (auto &pdg : pdgFilter) {
    if (std::abs(pdg) == pdgid) {
      return true;
    }
  }
  return false;
}

nonCentralEvent2G4()

void Generator::nonCentralEvent2G4	(	const HepMC::GenEvent *	g,
		G4Event *	e
	)

Definition at line 552 of file Generator.cc.

References GenParticle::GenParticle, runTauDisplay::gp, mps_fire::i, ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it, setGenId(), findQualityFiles::v, and verbose.

Referenced by RunManagerMTWorker::generateEvent().

                                                                           {
  int i = g4evt->GetNumberOfPrimaryVertex();
  for (HepMC::GenEvent::particle_const_iterator it = evt->particles_begin(); it != evt->particles_end(); ++it) {
    ++i;
    HepMC::GenParticle *gp = (*it);

    // storing only particle with status == 1
    if (gp->status() != 1)
      continue;

    int pdg = gp->pdg_id();
    G4PrimaryParticle *g4p = new G4PrimaryParticle(
        pdg, gp->momentum().px() * CLHEP::GeV, gp->momentum().py() * CLHEP::GeV, gp->momentum().pz() * CLHEP::GeV);
    if (g4p->GetG4code() != nullptr) {
      g4p->SetMass(g4p->GetG4code()->GetPDGMass());
      g4p->SetCharge(g4p->GetG4code()->GetPDGCharge());
    }
    setGenId(g4p, i);
    G4PrimaryVertex *v = new G4PrimaryVertex(gp->production_vertex()->position().x() * CLHEP::mm,
                                             gp->production_vertex()->position().y() * CLHEP::mm,
                                             gp->production_vertex()->position().z() * CLHEP::mm,
                                             gp->production_vertex()->position().t() * CLHEP::mm / CLHEP::c_light);
    v->SetPrimary(g4p);
    g4evt->AddPrimaryVertex(v);
    if (verbose > 0)
      v->Print();
  }  // end loop on HepMC particles
}

particleAssignDaughters()

void Generator::particleAssignDaughters ( G4PrimaryParticle *  p, HepMC::GenParticle *  hp, double  length  )

private

Definition at line 430 of file Generator.cc.

References funct::abs(), class-composition::children, createTree::dd, LogDebug, AlCaHLTBitMon_ParallelJobs::p, setGenId(), mathSSE::sqrt(), mps_update::status, verbose, testProducerWithPsetDescEmpty_cfi::x1, testProducerWithPsetDescEmpty_cfi::x2, testProducerWithPsetDescEmpty_cfi::y1, testProducerWithPsetDescEmpty_cfi::y2, and testProducerWithPsetDescEmpty_cfi::z2.

Referenced by HepMC2G4().

                                                                                                        {
  if (verbose > 1) {
    LogDebug("SimG4CoreGenerator") << "Special case of long decay length \n"
                                   << "Assign daughters with to mother with decaylength=" << decaylength / cm << " cm";
  }
  math::XYZTLorentzVector p(vp->momentum().px(), vp->momentum().py(), vp->momentum().pz(), vp->momentum().e());

  // defined preassigned decay time
  double proper_time = decaylength / (p.Beta() * p.Gamma() * c_light);
  g4p->SetProperTime(proper_time);

  if (verbose > 2) {
    LogDebug("SimG4CoreGenerator") << " px= " << p.px() << " py= " << p.py() << " pz= " << p.pz() << " e= " << p.e()
                                   << " beta= " << p.Beta() << " gamma= " << p.Gamma()
                                   << " Proper time= " << proper_time / ns << " ns";
  }

  // the particle will decay after the same length if it
  // has not interacted before
  double x1 = vp->end_vertex()->position().x();
  double y1 = vp->end_vertex()->position().y();
  double z1 = vp->end_vertex()->position().z();

  for (HepMC::GenVertex::particle_iterator vpdec = vp->end_vertex()->particles_begin(HepMC::children);
       vpdec != vp->end_vertex()->particles_end(HepMC::children);
       ++vpdec) {
    // transform decay products such that in the rest frame of mother
    math::XYZTLorentzVector pdec(
        (*vpdec)->momentum().px(), (*vpdec)->momentum().py(), (*vpdec)->momentum().pz(), (*vpdec)->momentum().e());

    // children should only be taken into account once
    G4PrimaryParticle *g4daught =
        new G4PrimaryParticle((*vpdec)->pdg_id(), pdec.x() * CLHEP::GeV, pdec.y() * CLHEP::GeV, pdec.z() * CLHEP::GeV);

    if (g4daught->GetG4code() != nullptr) {
      g4daught->SetMass(g4daught->GetG4code()->GetPDGMass());
      g4daught->SetCharge(g4daught->GetG4code()->GetPDGCharge());
    }

    // V.I. do not use SetWeight but the same code
    // value of the code compute inside TrackWithHistory
    setGenId(g4daught, (*vpdec)->barcode());

    int status = (*vpdec)->status();
    if (verbose > 1)
      LogDebug("SimG4CoreGenerator::::particleAssignDaughters")
          << "Assigning a " << (*vpdec)->pdg_id() << " as daughter of a " << vp->pdg_id() << " status=" << status;

    if ((status == 2 || (status == 23 && std::abs(vp->pdg_id()) == 1000015) || (status > 50 && status < 100)) &&
        (*vpdec)->end_vertex() != nullptr) {
      double x2 = (*vpdec)->end_vertex()->position().x();
      double y2 = (*vpdec)->end_vertex()->position().y();
      double z2 = (*vpdec)->end_vertex()->position().z();
      double dd = std::sqrt((x1 - x2) * (x1 - x2) + (y1 - y2) * (y1 - y2) + (z1 - z2) * (z1 - z2));
      particleAssignDaughters(g4daught, *vpdec, dd);
    }
    (*vpdec)->set_status(1000 + status);
    g4p->SetDaughter(g4daught);

    if (verbose > 1)
      g4daught->Print();
  }
}

particlePassesPrimaryCuts()

bool Generator::particlePassesPrimaryCuts ( const G4ThreeVector &  p ) const

private

Definition at line 495 of file Generator.cc.

References MillePedeFileConverter_cfg::e, PVValHelper::eta, fEtaCuts, RemoveAddSevLevel::flag, fPCuts, fPhiCuts, LogDebug, AlCaHLTBitMon_ParallelJobs::p, phi, theMaxEtaCut, theMaxPCut, theMaxPhiCut, and verbose.

                                                                      {
  bool flag = true;
  double ptot = p.mag();
  if (fPCuts && (ptot < theMinPCut * CLHEP::GeV || ptot > theMaxPCut * CLHEP::GeV)) {
    flag = false;
  }
  if (fEtaCuts && flag) {
    double pz = p.z();
    if (ptot < pz + 1.e-10) {
      flag = false;

    } else {
      double eta = 0.5 * G4Log((ptot + pz) / (ptot - pz));
      if (eta < theMinEtaCut || eta > theMaxEtaCut) {
        flag = false;
      }
    }
  }
  if (fPhiCuts && flag) {
    double phi = p.phi();
    if (phi < theMinPhiCut || phi > theMaxPhiCut) {
      flag = false;
    }
  }

  if (verbose > 2)
    LogDebug("SimG4CoreGenerator") << "Generator ptot(GeV)= " << ptot / GeV << " eta= " << p.eta()
                                   << "  phi= " << p.phi() << " Flag= " << flag;

  return flag;
}

setGenEvent()

void Generator::setGenEvent ( const HepMC::GenEvent *  inpevt )

inline

Definition at line 24 of file Generator.h.

References evt_.

Referenced by RunManagerMTWorker::generateEvent().

                                                {
    evt_ = (HepMC::GenEvent *)inpevt;
    return;
  }

setGenId()

void Generator::setGenId ( G4PrimaryParticle *  p, int  id  ) const

inlineprivate

Definition at line 40 of file Generator.h.

References AlCaHLTBitMon_ParallelJobs::p.

Referenced by HepMC2G4(), nonCentralEvent2G4(), and particleAssignDaughters().

{ p->SetUserInformation(new GenParticleInfo(id)); }

Member Data Documentation

evt_

HepMC::GenEvent* Generator::evt_

private

Definition at line 61 of file Generator.h.

Referenced by genEvent(), and setGenEvent().

fEtaCuts

bool Generator::fEtaCuts

private

Definition at line 45 of file Generator.h.

Referenced by Generator(), HepMC2G4(), and particlePassesPrimaryCuts().

fFiductialCuts

bool Generator::fFiductialCuts

private

Definition at line 47 of file Generator.h.

Referenced by Generator(), and HepMC2G4().

fLumiFilter

LumiMonitorFilter* Generator::fLumiFilter

private

Definition at line 60 of file Generator.h.

Referenced by Generator(), HepMC2G4(), and ~Generator().

fPCuts

bool Generator::fPCuts

private

Definition at line 43 of file Generator.h.

Referenced by Generator(), HepMC2G4(), and particlePassesPrimaryCuts().

fPDGFilter

bool Generator::fPDGFilter

private

Definition at line 67 of file Generator.h.

Referenced by Generator(), and HepMC2G4().

fPhiCuts

bool Generator::fPhiCuts

private

Definition at line 46 of file Generator.h.

Referenced by Generator(), HepMC2G4(), and particlePassesPrimaryCuts().

fPtransCut

bool Generator::fPtransCut

private

Definition at line 44 of file Generator.h.

Referenced by Generator(), and HepMC2G4().

maxZCentralCMS

double Generator::maxZCentralCMS

private

Definition at line 58 of file Generator.h.

Referenced by Generator(), and HepMC2G4().

pdgFilter

std::vector<int> Generator::pdgFilter

private

Definition at line 65 of file Generator.h.

Referenced by Generator(), and IsInTheFilterList().

pdgFilterSel

bool Generator::pdgFilterSel

private

Definition at line 66 of file Generator.h.

Referenced by Generator(), and HepMC2G4().

theDecLenCut

double Generator::theDecLenCut

private

Definition at line 57 of file Generator.h.

Referenced by Generator().

theDecRCut2

double Generator::theDecRCut2

private

Definition at line 55 of file Generator.h.

Referenced by Generator(), and HepMC2G4().

theEtaCutForHector

double Generator::theEtaCutForHector

private

Definition at line 56 of file Generator.h.

Referenced by Generator().

theMaxEtaCut

double Generator::theMaxEtaCut

private

Definition at line 51 of file Generator.h.

Referenced by Generator(), and particlePassesPrimaryCuts().

theMaxPCut

double Generator::theMaxPCut

private

Definition at line 54 of file Generator.h.

Referenced by Generator(), HepMC2G4(), and particlePassesPrimaryCuts().

theMaxPhiCut

double Generator::theMaxPhiCut

private

Definition at line 49 of file Generator.h.

Referenced by Generator(), HepMC2G4(), and particlePassesPrimaryCuts().

theMinEtaCut

double Generator::theMinEtaCut

private

Definition at line 50 of file Generator.h.

Referenced by Generator().

theMinPCut

double Generator::theMinPCut

private

Definition at line 52 of file Generator.h.

Referenced by Generator().

theMinPhiCut

double Generator::theMinPhiCut

private

Definition at line 48 of file Generator.h.

Referenced by Generator().

theMinPtCut2

double Generator::theMinPtCut2

private

Definition at line 53 of file Generator.h.

Referenced by Generator(), and HepMC2G4().

verbose

int Generator::verbose

private

Definition at line 59 of file Generator.h.

Referenced by HepMC2G4(), nonCentralEvent2G4(), particleAssignDaughters(), and particlePassesPrimaryCuts().

vtx_

math::XYZTLorentzVector* Generator::vtx_

private

Definition at line 62 of file Generator.h.

Referenced by genVertex(), and HepMC2G4().

weight_

double Generator::weight_

private

Definition at line 63 of file Generator.h.

Referenced by eventWeight(), and HepMC2G4().

Z_hector

double Generator::Z_hector

private

Definition at line 64 of file Generator.h.

Referenced by Generator(), and HepMC2G4().

Z_lmax

double Generator::Z_lmax

private

Definition at line 64 of file Generator.h.

Referenced by Generator(), and HepMC2G4().

Z_lmin

double Generator::Z_lmin

private

Definition at line 64 of file Generator.h.

Referenced by Generator(), and HepMC2G4().