File size: 1,631 Bytes
158b61b |
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 51 52 53 54 55 56 57 58 59 60 61 62 |
#!/usr/bin/env python
# Combines the system definition from one .ini file with the weights contained
# in another. Works for the new moses.ini format with fully named feature
# functions. Writes the new .ini file to stdout
# Script by Ulrich Germann.
import re,sys,os
from optparse import OptionParser
SectionHeaderPattern = re.compile(r'^\[(.*)\]\s*$')
def read_ini(filename):
'''
Reads a moses.ini file and returns a dictionary mapping
from section names to a list of lines contained in that section.
'''
AllSections = {}
CurSection = AllSections.setdefault('',[])
for line in open(filename):
line = line.strip()
m = SectionHeaderPattern.match(line)
if m:
CurSection = AllSections.setdefault(m.group(1),[])
elif len(line):
CurSection.append(line)
pass
pass
return AllSections
parser = OptionParser()
parser.add_option("-s", "--system", dest = "system",
help = "moses.ini file defining the system")
parser.add_option("-w", "--weights", dest = "weight",
help = "moses.ini file defining the system")
opts,args = parser.parse_args()
system = read_ini(opts.system)
weight = read_ini(opts.weight)
for s in system:
if len(s) == 0 or s[0:6] == 'weight': continue
print "[%s]"%s
print "\n".join(system[s])
print
pass
if 'weight' in weight:
print '[weight]'
print "\n".join(weight['weight'])
else:
for s in weight:
if s[0:6] != 'weight': continue
print "[%s]"%s
print "\n".join(system[s])
print
pass
pass
|