Changes between Version 23 and Version 24 of pSPSS

Timestamp:

Oct 17, 2023, 8:46:19 PM (2 years ago)

Author:

Jan Hajer

Comment:

--

pSPSS

@@ -108 +108 @@
 
 {{{
-def do_add_time_of_flight(self, line):
-    print("Running patched do_add_time_of_flight ")
-
-    args = self.split_arg(line)
-    # check the validity of the arguments and reformat args
-    self. check_add_time_of_flight(args)
-
-    event_path, threshold = args
-    # gunzip the file
-    if event_path.endswith('.gz'):
-        need_zip = True
-        misc.gunzip(event_path)
-        event_path = event_path[:-3]
-    else:
-        need_zip = False
-
-    import random
-    try:
-        import madgraph.various.lhe_parser as lhe_parser
-    except:
-        import internal.lhe_parser as lhe_parser
-
-    logger.info('Add time of flight information on file %s' % event_path)
-    lhe = lhe_parser.EventFile(event_path)
-    output = open('%s_2vertex.lhe' % event_path, 'w')
-    # write the banner to the output file
-    output.write(lhe.banner)
-
-    # get the associate param_card
-    begin_param = lhe.banner.find('<slha >')
-    end_param = lhe.banner.find('</slha >')
-    param_card = lhe.banner[begin_param + 6: end_param]. split('\n')
-    param_card = param_card_mod.ParamCard(param_card)
-
-    cst = 6.58211915e-25  # hbar in GeV s
-    c = 299792458000  # speed of light in mm/s
-    sm_lepton_list = [11, 12, 13, 14, 15, 16]
-
-    pspss_n_list = [8000011, 8000012]
-    mass_splitting = param_card.get_value('PSPSS ', 2)
-    damping = param_card.get_value('PSPSS ', 6)
-
-    # Loop over all events
-    for event in lhe:
-        leptonnumber = 0
-        for particle in event:
-            if particle.status == 1:
-                if particle.pid in sm_lepton_list:
-                    leptonnumber += 1
-                elif -particle.pid in sm_lepton_list:
-                    leptonnumber -= 1
-
-    write_event = True
-    for particle in event:
-        id = particle.pid
-        width = param_card['decay '].get((abs(id),)).value
-
-        if width:
-            if id in pspss_n_list:
-                tau0 = random.expovariate(width / cst)
-
-                if 0.5 * (1 + math.exp(-damping) * math.cos(mass_splitting * tau0 / cst)) >= random.random():
-                    write_event = (leptonnumber == 0)
-                else:
-                    write_event = (leptonnumber != 0)
-
-            else:
-                tau0 = random.expovariate(width / cst)
-
-            vtim = c * tau0
-            if vtim > threshold:
-                particle.vtim = vtim
-
-    # write this modify event
-    if write_event:
-        output.write(str(event))
-
-
-    output.write('</LesHouchesEvents >\n')
-    output.close()
-    files.mv('%s_2vertex.lhe' % event_path, event_path)
-
-    ifneed_zip: misc.gzip(event_path)
-
+    def do_add_time_of_flight(self, line):
+        print("Running patched do_add_time_of_flight")
+
+        args = self.split_arg(line)
+        #check the validity of the arguments and reformat args
+        self.check_add_time_of_flight(args)
+
+        event_path, threshold = args
+        #gunzip the file
+        if event_path.endswith('.gz'):
+            need_zip = True
+            misc.gunzip(event_path)
+            event_path = event_path[:-3]
+        else:
+            need_zip = False
+
+        import random
+        try:
+            import madgraph.various.lhe_parser as lhe_parser
+        except:
+            import internal.lhe_parser as lhe_parser
+
+        logger.info('Add time of flight information on file %s' % event_path)
+        lhe = lhe_parser.EventFile(event_path)
+        output = open('%s_2vertex.lhe' % event_path, 'w')
+        #write the banner to the output file
+        output.write(lhe.banner)
+
+        # get the associate param_card
+        begin_param = lhe.banner.find('<slha>')
+        end_param = lhe.banner.find('</slha>')
+        param_card = lhe.banner[begin_param+6:end_param].split('\n')
+        param_card = param_card_mod.ParamCard(param_card)
+
+        cst = 6.58211915e-25 # hbar in GeV s
+        c = 299792458000 # speed of light in mm/s
+        sm_lepton_list = [11, 12, 13, 14, 15, 16]
+
+        pspss_n_list = [8000011, 8000012]
+        mass_splitting = param_card.get_value('PSPSS', 2)
+        damping = param_card.get_value('PSPSS', 6)
+
+        # Loop over all events
+        for event in lhe:
+            leptonnumber = 0
+            for particle in event:
+                if particle.status == 1:
+                    if particle.pid in sm_lepton_list:
+                        leptonnumber += 1
+                    elif -particle.pid in sm_lepton_list:
+                        leptonnumber -= 1
+
+            write_event = True
+            for particle in event:
+                id = particle.pid
+                width = param_card['decay'].get((abs(id),)).value
+
+                if width:
+                    if id in pspss_n_list:
+                        tau0 = random.expovariate(width / cst)
+
+                        if 0.5 * (1 + math.exp(-damping) * math.cos(mass_splitting * tau0 / cst)) >= random.random():
+                            write_event = (leptonnumber == 0)
+                        else:
+                            write_event = (leptonnumber != 0)
+
+                    else:
+                        tau0 = random.expovariate(width / cst)
+
+                    vtim = c * tau0
+                    if vtim > threshold:
+                        particle.vtim = vtim
+
+            # write this modify event
+            if write_event:
+                output.write(str(event))
+
+        output.write('</LesHouchesEvents>\n')
+        output.close()
+        files.mv('%s_2vertex.lhe' % event_path, event_path)
+
+        if need_zip:
+            misc.gzip(event_path)
 }}}