HerwigToHepMCProduction/HepMC_scripts/Process_xs.py

0001 #! /bin/python
0002
0003 import sys
0004 import os
0005
0006 filetag="jet30"
0007 path="."
0008 if len(sys.argv) > 1:
0009     filetag=sys.argv[1]
0010     if len(sys.argv) > 2:
0011         path=sys.argv[2]
0012 avg_val = 0
0013 n = 0
0014 for fname in os.listdir(path):
0015     if fname.find(filetag) != -1:
0016         with open(str(path)+"/"+str(fname)) as f:
0017             mtpx = 1.
0018             val = 0.
0019             for line in f:
0020                 if line.find("Generated") != -1:
0021                     mtpx = line.split("=")[1:][0]
0022                 if line.find("XS/N") != -1:
0023                     val= line.split("=")[1:][0]
0024                     val= val.split("\n")[:1][0]
0025                     if float(val) <= 0:
0026                         continue
0027                     val=float(val)*float(mtpx)
0028                     n += 1
0029                     avg_val += val
0030             if n >= 1000:
0031                 break
0032 avg_val = avg_val /float(n)
0033 print("Average cross section for " +str(filetag)+" is "+str(avg_val)+" nb")
0034