Changeset View
Changeset View
Standalone View
Standalone View
scripts/printParticleDecays.py
- This file was added.
| Property | Old Value | New Value |
|---|---|---|
| File Mode | null | 100755 |
| #!/usr/bin/env python | |||||
| import os,sys | |||||
| import argparse | |||||
| parser = argparse.ArgumentParser( | |||||
| description='Guide mark calculation') | |||||
| parser.add_argument('-decfile', default='DECAY.DEC') | |||||
| parser.add_argument('-particle', default='') | |||||
| parser.add_argument('-daughter1', default=[], nargs='*') | |||||
| parser.add_argument('-daughter2', default=[], nargs='*') | |||||
| args = parser.parse_args() | |||||
| print(args) | |||||
| decFiles = open( args.decfile ) | |||||
| inDecay = False | |||||
| bfSum = 0.0 | |||||
| nDecays = 0 | |||||
| for line in decFiles: | |||||
| if (inDecay == False) and (line.find('Decay') != -1): | |||||
| ss = line.split(' ') | |||||
| if ( ss[len(ss)-1].strip() == args.particle ): | |||||
| inDecay = True | |||||
| print(line) | |||||
| continue | |||||
| if (inDecay == True) and (line.find('Enddecay') != -1): | |||||
| inDecay = False | |||||
| break | |||||
| if inDecay == True: | |||||
| if line[0] == '#' or len(line)<2: # This is comment, skip | |||||
| continue | |||||
| # Now check whether we have given daughters | |||||
| wanted = False | |||||
| if len(args.daughter1)!=0: | |||||
| for ii in args.daughter1: | |||||
| if ii in line: # one of the daughers is in | |||||
| if len(args.daughter2)!=0: | |||||
| for jj in args.daughter2: | |||||
| if jj in line: | |||||
| wanted = True | |||||
| else: | |||||
| wanted = True | |||||
| if wanted: | |||||
| print(line.strip()) | |||||
| decFiles.close() | |||||