Merge pull request #2565 from mkirsano/fix_residualdecay71

Generators updates -- Fix ResidualDecay port from 62X
cms-sw · Feb 23, 2014 · a9a6f3b · a9a6f3b
2 parents ab1084c + ea29541
commit a9a6f3b
Show file tree

Hide file tree

Showing 3 changed files with 91 additions and 82 deletions.
diff --git a/GeneratorInterface/Pythia8Interface/interface/Py8InterfaceBase.h b/GeneratorInterface/Pythia8Interface/interface/Py8InterfaceBase.h
@@ -25,7 +25,6 @@ namespace gen {
 
          virtual bool generatePartonsAndHadronize() = 0;
          bool decay() { return true; } // NOT used - let's call it "design imperfection"
-         virtual bool residualDecay();
          bool readSettings( int ); // common func
          virtual bool initializeForInternalPartons() = 0;
          bool declareStableParticles( const std::vector<int>& ); // common func

diff --git a/GeneratorInterface/Pythia8Interface/plugins/Pythia8Hadronizer.cc b/GeneratorInterface/Pythia8Interface/plugins/Pythia8Hadronizer.cc
@@ -63,6 +63,9 @@ class Pythia8Hadronizer : public BaseHadronizer, public Py8InterfaceBase {
 
     bool generatePartonsAndHadronize() override;
     bool hadronize();
+
+    virtual bool residualDecay();
+
     void finalizeEvent() override;
 
     void statistics() override;
@@ -308,6 +311,12 @@ bool Pythia8Hadronizer::initializeForInternalPartons()
            fMasterGen->particleData.listAll();
   }
 
+  // init decayer
+  fDecayer->readString("ProcessLevel:all = off"); // trick
+  fDecayer->readString("Standalone:allowResDec=on");
+  // pythia->readString("ProcessLevel::resonanceDecays=on");
+  fDecayer->init();
+
   return true;
 }
 
@@ -376,6 +385,12 @@ bool Pythia8Hadronizer::initializeForExternalPartons()
            fMasterGen->particleData.listAll();
   }
 
+  // init decayer
+  fDecayer->readString("ProcessLevel:all = off"); // trick
+  fDecayer->readString("Standalone:allowResDec=on");
+  // pythia->readString("ProcessLevel::resonanceDecays=on");
+  fDecayer->init();
+
   return true;
 }
 
@@ -444,6 +459,82 @@ bool Pythia8Hadronizer::hadronize()
 }
 
 
+bool Pythia8Hadronizer::residualDecay()
+{
+
+  Event* pythiaEvent = &(fMasterGen->event);
+
+  int NPartsBeforeDecays = pythiaEvent->size();
+  int NPartsAfterDecays = event().get()->particles_size();
+  int NewBarcode = NPartsAfterDecays;
+
+  for ( int ipart=NPartsAfterDecays; ipart>NPartsBeforeDecays; ipart-- )
+  {
+
+    HepMC::GenParticle* part = event().get()->barcode_to_particle( ipart );
+
+    if ( part->status() == 1 )
+    {
+      fDecayer->event.reset();
+      Particle py8part(  part->pdg_id(), 93, 0, 0, 0, 0, 0, 0,
+                         part->momentum().x(),
+                         part->momentum().y(),
+                         part->momentum().z(),
+                         part->momentum().t(),
+                         part->generated_mass() );
+      HepMC::GenVertex* ProdVtx = part->production_vertex();
+      py8part.vProd( ProdVtx->position().x(), ProdVtx->position().y(),
+                     ProdVtx->position().z(), ProdVtx->position().t() );
+      py8part.tau( (fDecayer->particleData).tau0( part->pdg_id() ) );
+      fDecayer->event.append( py8part );
+      int nentries = fDecayer->event.size();
+      if ( !fDecayer->event[nentries-1].mayDecay() ) continue;
+      fDecayer->next();
+      int nentries1 = fDecayer->event.size();
+      // fDecayer->event.list(std::cout);
+      if ( nentries1 <= nentries ) continue; //same number of particles, no decays...
+
+      part->set_status(2);
+
+      Particle& py8daughter = fDecayer->event[nentries]; // the 1st daughter
+      HepMC::GenVertex* DecVtx = new HepMC::GenVertex( HepMC::FourVector(py8daughter.xProd(),
+                                                       py8daughter.yProd(),
+                                                       py8daughter.zProd(),
+                                                       py8daughter.tProd()) );
+
+      DecVtx->add_particle_in( part ); // this will cleanup end_vertex if exists, replace with the new one
+                                       // I presume (vtx) barcode will be given automatically
+
+      HepMC::FourVector pmom( py8daughter.px(), py8daughter.py(), py8daughter.pz(), py8daughter.e() );
+
+      HepMC::GenParticle* daughter =
+                        new HepMC::GenParticle( pmom, py8daughter.id(), 1 );
+
+      NewBarcode++;
+      daughter->suggest_barcode( NewBarcode );
+      DecVtx->add_particle_out( daughter );
+
+      for ( int ipart1=nentries+1; ipart1<nentries1; ipart1++ )
+      {
+        py8daughter = fDecayer->event[ipart1];
+        HepMC::FourVector pmomN( py8daughter.px(), py8daughter.py(), py8daughter.pz(), py8daughter.e() );
+        HepMC::GenParticle* daughterN =
+                        new HepMC::GenParticle( pmomN, py8daughter.id(), 1 );
+        NewBarcode++;
+        daughterN->suggest_barcode( NewBarcode );
+        DecVtx->add_particle_out( daughterN );
+      }
+
+      event().get()->add_vertex( DecVtx );
+      // fCurrentEventState->add_vertex( DecVtx );
+
+    }
+ }
+ return true;
+
+}
+
+
 void Pythia8Hadronizer::finalizeEvent()
 {
   bool lhe = lheEvent() != 0;

diff --git a/GeneratorInterface/Pythia8Interface/src/Py8InterfaceBase.cc b/GeneratorInterface/Pythia8Interface/src/Py8InterfaceBase.cc
@@ -82,87 +82,6 @@ bool Py8InterfaceBase:: declareSpecialSettings( const std::vector<std::string>&
    return true;
 }
 
-bool Py8InterfaceBase::residualDecay() 
-{
-
-/*
-  Event* pythiaEvent = &(fMasterPtr->event);
-  
-  assert(fCurrentEventState);
-  
-  int NPartsBeforeDecays = pythiaEvent->size();
-  // int NPartsAfterDecays = event().get()->particles_size();
-  int NPartsAfterDecays = fCurrentEventState->particles_size();
-  int NewBarcode = NPartsAfterDecays;
-   
-  for ( int ipart=NPartsAfterDecays; ipart>NPartsBeforeDecays; ipart-- )
-  {
-
-    // HepMC::GenParticle* part = event().get()->barcode_to_particle( ipart );
-    HepMC::GenParticle* part = fCurrentEventState->barcode_to_particle( ipart );
-
-    if ( part->status() == 1 )
-    {
-      fDecayerPtr->event.reset();
-      Particle py8part(  part->pdg_id(), 93, 0, 0, 0, 0, 0, 0,
-                         part->momentum().x(),
-                         part->momentum().y(),
-                         part->momentum().z(),
-                         part->momentum().t(),
-                         part->generated_mass() );
-      HepMC::GenVertex* ProdVtx = part->production_vertex();
-      py8part.vProd( ProdVtx->position().x(), ProdVtx->position().y(), 
-                     ProdVtx->position().z(), ProdVtx->position().t() );
-      py8part.tau( (fDecayerPtr->particleData).tau0( part->pdg_id() ) );
-      fDecayerPtr->event.append( py8part );
-      int nentries = fDecayerPtr->event.size();
-      if ( !fDecayerPtr->event[nentries-1].mayDecay() ) continue;
-      fDecayerPtr->next();
-      int nentries1 = fDecayerPtr->event.size();
-      // --> fDecayerPtr->event.list(std::cout);
-      if ( nentries1 <= nentries ) continue; //same number of particles, no decays...
-	    
-      part->set_status(2);
-	    
-      Particle& py8daughter = fDecayerPtr->event[nentries]; // the 1st daughter
-      HepMC::GenVertex* DecVtx = new HepMC::GenVertex( HepMC::FourVector(py8daughter.xProd(),
-                                                       py8daughter.yProd(),
-                                                       py8daughter.zProd(),
-                                                       py8daughter.tProd()) );
-
-      DecVtx->add_particle_in( part ); // this will cleanup end_vertex if exists, replace with the new one
-                                       // I presume (vtx) barcode will be given automatically
-	    
-      HepMC::FourVector pmom( py8daughter.px(), py8daughter.py(), py8daughter.pz(), py8daughter.e() );
-	    
-      HepMC::GenParticle* daughter =
-                        new HepMC::GenParticle( pmom, py8daughter.id(), 1 );
-	    
-      NewBarcode++;
-      daughter->suggest_barcode( NewBarcode );
-      DecVtx->add_particle_out( daughter );
-	    	    
-      for ( ipart=nentries+1; ipart<nentries1; ipart++ )
-      {
-        py8daughter = fDecayerPtr->event[ipart];
-        HepMC::FourVector pmomN( py8daughter.px(), py8daughter.py(), py8daughter.pz(), py8daughter.e() );	    
-        HepMC::GenParticle* daughterN =
-                        new HepMC::GenParticle( pmomN, py8daughter.id(), 1 );
-        NewBarcode++;
-        daughterN->suggest_barcode( NewBarcode );
-        DecVtx->add_particle_out( daughterN );
-      }
-	    
-      // event().get()->add_vertex( DecVtx );
-      fCurrentEventState->add_vertex( DecVtx );
-
-    }
- } 
-*/   
- return true;
-
-}
-
 
 void Py8InterfaceBase::statistics()
 {