Add fsrmode for codes that support it

Author: apricePhy

python/Generators/Pythia.py

@@ -47,6 +47,10 @@ def write_run(self):
         else:
             self.add_option("PartonLevel:ISR", "off")
             self.add_option("PDF:lepton", "off")
+        if self.procinfo.get("fsrmode"):
+            self.add_option("PartonLevel:FSR", "on")
+        else:
+            self.add_option("PartonLevel:FSR", "off")
 
         self.add_option("Main:numberOfEvents", self.procinfo.get("events"))
         self.run += "\n"

python/Generators/Sherpa.py

@@ -44,6 +44,10 @@ def write_run(self):
             self.add_run_option("PDF_LIBRARY", "PDFESherpa")
         else:
             self.add_run_option("PDF_LIBRARY", "None")
+        if self.procinfo.get("fsrmode"):
+            self.add_run_option("YFS_MODE", "FULL")
+        else:
+            self.add_run_option("YFS_MODE", "None")
         self.add_run_option("EVENTS", self.procinfo.get("events"))
         self.run += "\n\n"
         for p in self.procinfo.get_data_particles():
@@ -61,7 +65,7 @@ def write_run(self):
             self.add_run_option("EVENT_OUTPUT", eoutname)
 
         elif self.procinfo.get("output_format") == "hepmc3":
-            eoutname = "HepMC3_GenEvent[{0}.hepmc3g]".format(self.GeneratorDatacardBase)
+            eoutname = "HepMC3_GenEvent[{0}.hepmc3]".format(self.GeneratorDatacardBase)
             self.add_run_option("EVENT_OUTPUT", eoutname)
         self.run += self.procDB.get_run_out()
         self.add_run_option("EVENT_GENERATION_MODE", self.procinfo.eventmode)

python/Process.py

@@ -82,7 +82,10 @@ def get_final_pdg_list(self):
         return list(self._finfo)
 
     def get(self, name):
-        return getattr(self, name)
+        try:
+            return getattr(self, name)
+        except:
+            return None
 
     def get_args(self):
         return self._required_args
@@ -141,6 +144,7 @@ def __init__(self, settings):
         self.settings = settings
         self.model = settings.get_model()
         self.events = settings.get_event_number()
+        print(self.events)
         self.output_format = settings.get_output_format()
         self.PythiaTune = settings.get_PythiaTune()
         self.ElectronPolarisation = settings.get_ElectronPolarisation()