Merge pull request #8 from alisw/rdev-leif-patch

Patch from Leif to fix Angantyr when Beams:allowVertexSpread
alisw · Apr 6, 2021 · 3abbf0e · 3abbf0e
2 parents 1984832 + cbc800e
commit 3abbf0e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/include/Pythia8/Pythia.h b/include/Pythia8/Pythia.h
@@ -184,6 +184,9 @@ class Pythia {
   // collisions.
   HeavyIonsPtr getHeavyIonsPtr() { return heavyIonsPtr;}
 
+  // Possibility to access the pointer to the BeamShape object.
+  BeamShapePtr getBeamShapePtr() { return beamShapePtr; }
+
   // Possibility to get the pointer to the parton-shower model.
   ShowerModelPtr getShowerModelPtr() { return showerModelPtr; }
 

diff --git a/src/HeavyIons.cc b/src/HeavyIons.cc
@@ -7,6 +7,7 @@
 // heavy ion classes classes, and some related global functions.
 
 #include "Pythia8/HeavyIons.h"
+#include "Pythia8/BeamShape.h"
 #include "Pythia8/Pythia.h"
 #include <cassert>
 
@@ -408,6 +409,7 @@ bool Angantyr::init() {
   for ( int i = MBIAS; i < ALL; ++i ) {
     pythia[i] = new Pythia(*settingsPtr, *particleDataPtr, false);
     pythia[i]->settings.mode("HeavyIon:mode", 1);
+    pythia[i]->settings.flag("Beams:allowVertexSpread", false);
   }
 
   sigtot.init();
@@ -1649,6 +1651,12 @@ bool Angantyr::next() {
       }
     }
 
+    if ( settingsPtr->flag("Beams:allowVertexSpread") ) {
+      pythia[HADRON]->getBeamShapePtr()->pick();
+      Vec4 vertex = pythia[HADRON]->getBeamShapePtr()->vertex();
+      for ( Particle & p : pythia[HADRON]->event ) p.vProdAdd( vertex);
+    }
+
     hiInfo.accept();
 
     updateInfo();