LQGen/EditLHE.py at main · CMSROMA/LQGen · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
import os
import sys
import argparse

#couplings=["utau"]
#masses=["600","900","1200","1500","1800","2100","2400","2700","3000"]
#lambdas=["0p2","0p5","1p0","1p5","2p0"]

couplings=["umu"]
masses=["3000"]
lambdas=["0p1","1p0"]

for j in range(0,len(masses)):
  for k in range(0,len(lambdas)):
      for l in range(1,11):
         input_lhe="/afs/cern.ch/work/s/santanas/Workspace/CMS/generateLQ_NLO/LQGen/LeptonInducedLQ_"+couplings[0]+"_M"+masses[j]+"_Lambda"+lambdas[k]+"/split/LeptonInducedLQ_"+couplings[0]+"_M"+masses[j]+"_Lambda"+lambdas[k]+"__"+str(l)+".lhe"
         output_lhe="/afs/cern.ch/work/s/santanas/Workspace/CMS/generateLQ_NLO/LQGen/LeptonInducedLQ_"+couplings[0]+"_M"+masses[j]+"_Lambda"+lambdas[k]+"/split/LeptonInducedLQ_mod_"+couplings[0]+"_M"+masses[j]+"_Lambda"+lambdas[k]+"__"+str(l)+".lhe"
         print (input_lhe,output_lhe)

         fin = open(input_lhe, "rt")
         #output file to write the result to
         fout = open(output_lhe, "wt")
         #for each line in the input file
         i=-1

         for line in fin:
                 #read replace the string and write to output file
                 if '#------------------------------------------------' in line:
                    fout.write(line.replace('#------------------------------------------------', '#'))
                 elif '<weights>' in line:
                    fout.write(line.replace('<weights>', '<rwgt>'))
                    i=0
                 elif '</weights>' in line:
                    fout.write(line.replace('</weights>', '</rwgt>'))
                    i=-1
                 elif i>=0:
                    i=i+1
                    if i==1 : fout.write("<wgt id='101'> " + line.rstrip('\n') + " </wgt> \n")
                    elif i==2 : fout.write("<wgt id='102'> " + line.rstrip('\n') + " </wgt> \n")
                    elif i==3 : fout.write("<wgt id='103'> " + line.rstrip('\n') + " </wgt> \n")
                    elif i==4 : fout.write("<wgt id='104'> " + line.rstrip('\n') + " </wgt> \n")
                    elif i==5 : fout.write("<wgt id='105'> " + line.rstrip('\n') + " </wgt> \n")
                    elif i==6 : fout.write("<wgt id='106'> " + line.rstrip('\n') + " </wgt> \n")
                    elif i==7 : fout.write("<wgt id='107'> " + line.rstrip('\n') + " </wgt> \n")
                    else : fout.write("<wgt id='"+str(i-8)+"'> " + line.rstrip('\n') + " </wgt> \n")
                 else:
                    fout.write(line)
         #close input and output files
         fin.close()
         fout.close()