mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-12-29 06:52:34 +03:00
ef028446f3
This is not pleasant to read (and much, much less pleasant to write!) but sort of necessary in an open project. Right now it's quite hard to figure out what is licensed how, which doesn't matter much to most people but can suddenly become very important when people want to know what they're being allowed to do. I kept the notices as short as I could. As far as I could see, everything without a clear license notice is LGPL v2.1 or later.
229 lines
6.9 KiB
Python
229 lines
6.9 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# This file is part of moses. Its use is licensed under the GNU Lesser General
|
|
# Public License version 2.1 or, at your option, any later version.
|
|
|
|
"""
|
|
Python utilities for moses
|
|
|
|
This package mostly wraps standard Moses utilities into pipes.
|
|
|
|
Written by Ulrich Germann
|
|
|
|
This package borrows from scripts written by Christian Buck
|
|
|
|
The package assumes that there is a complete moses installation
|
|
(including scripts) under one root directory,
|
|
e.g., via ::
|
|
bjam --with-xmlrpc-c=... [...] --install-scripts --prefix=${HOME}/moses
|
|
By default, this root directory is "${HOME}/moses".
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import time
|
|
import xmlrpclib
|
|
from subprocess import (
|
|
PIPE,
|
|
Popen,
|
|
)
|
|
|
|
|
|
moses_root = os.environ.get('MOSES_ROOT', os.environ.get('HOME') + "/moses")
|
|
|
|
|
|
class ProcessWrapper:
|
|
|
|
def __init__(self, cmd=[]):
|
|
self.process = None
|
|
self.cmd = cmd
|
|
|
|
def start(self, stdin=PIPE, stdout=PIPE):
|
|
if self.process:
|
|
raise Exception("Process is already running")
|
|
self.process = Popen(self.cmd, stdin=stdin, stdout=stdout)
|
|
|
|
def __del__(self):
|
|
if self.process:
|
|
self.process.terminate()
|
|
|
|
|
|
class LineProcessor(ProcessWrapper):
|
|
|
|
def __call__(self, input):
|
|
if not self.process:
|
|
self.start()
|
|
self.process.stdin.write("%s\n" % input.strip())
|
|
self.process.stdin.flush()
|
|
return self.process.stdout.readline().strip()
|
|
|
|
|
|
class SentenceSplitter(ProcessWrapper):
|
|
"""Wrapper for standard Moses sentence splitter."""
|
|
|
|
def __init__(self, lang):
|
|
ssplit_cmd = moses_root + "/scripts/ems/support/split-sentences.perl"
|
|
self.cmd = [ssplit_cmd, "-b", "-q", "-l", lang]
|
|
self.process = None
|
|
|
|
def __call__(self, input):
|
|
if not self.process:
|
|
self.start()
|
|
self.process.stdin.write(input.strip() + "\n<P>\n")
|
|
self.process.stdin.flush()
|
|
x = self.process.stdout.readline().strip()
|
|
ret = []
|
|
while x != '<P>' and x != '':
|
|
ret.append(x)
|
|
x = self.process.stdout.readline().strip()
|
|
return ret
|
|
|
|
|
|
class Pretokenizer(LineProcessor):
|
|
"""Pretokenizer wrapper.
|
|
|
|
The pretokenizer fixes known issues with the input.
|
|
"""
|
|
def __init__(self, lang):
|
|
pretok_cmd = moses_root + "/scripts/tokenizer/pre-tokenizer.perl"
|
|
self.cmd = [pretok_cmd, "-b", "-q", "-l", lang]
|
|
self.process = None
|
|
|
|
|
|
class Tokenizer(LineProcessor):
|
|
"""Tokenizer wrapper.
|
|
|
|
The pretokenizer fixes known issues with the input.
|
|
"""
|
|
def __init__(self, lang, args=["-a", "-no-escape"]):
|
|
tok_cmd = moses_root + "/scripts/tokenizer/tokenizer.perl"
|
|
self.cmd = [tok_cmd, "-b", "-q", "-l", lang] + args
|
|
self.process = None
|
|
|
|
|
|
class Truecaser(LineProcessor):
|
|
"""Truecaser wrapper."""
|
|
def __init__(self, model):
|
|
truecase_cmd = moses_root + "/scripts/recaser/truecase.perl"
|
|
self.cmd = [truecase_cmd, "-b", "--model", model]
|
|
self.process = None
|
|
|
|
|
|
class LineProcessorPipeline:
|
|
"""Line processor: one line in, one line out."""
|
|
def __init__(self, parts=[]):
|
|
self.chain = [LineProcessor(p.cmd) for p in parts]
|
|
|
|
def start(self):
|
|
if len(self.chain) == 0:
|
|
return
|
|
if self.chain[0].process:
|
|
return
|
|
self.chain[0].start()
|
|
for i in xrange(1, len(self.chain)):
|
|
self.chain[i].start(stdin=self.chain[i - 1].process.stdout)
|
|
|
|
def __call__(self, input):
|
|
if len(self.chain) == 0:
|
|
return input
|
|
self.start()
|
|
self.chain[0].process.stdin.write("%s\n" % input.strip())
|
|
self.chain[0].process.stdin.flush()
|
|
return self.chain[0].process.stdout.readline().strip()
|
|
|
|
|
|
def find_free_port(p):
|
|
"""Find a free port, starting at /p/.
|
|
|
|
:return: The free port, or False if none found.
|
|
"""
|
|
ret = p
|
|
while ret - p < 20:
|
|
devnull = open(os.devnull, "w")
|
|
n = Popen(["netstat", "-tnp"], stdout=PIPE, stderr=devnull)
|
|
if n.communicate()[0].find(":%d " % ret) < 0:
|
|
return p
|
|
ret += 1
|
|
return False
|
|
|
|
|
|
class MosesServer(ProcessWrapper):
|
|
|
|
def __init__(self, args=[]):
|
|
self.process = None
|
|
mserver_cmd = moses_root + "/bin/mosesserver"
|
|
self.cmd = [mserver_cmd] + args
|
|
self.url = None
|
|
self.proxy = None
|
|
|
|
def start(self, config=None, args=[], port=7447, debug=False):
|
|
self.cmd.extend(args)
|
|
if config:
|
|
if "-f" in args:
|
|
raise Exception("Config file specified twice")
|
|
else:
|
|
self.cmd.extend(["-f", config])
|
|
self.port = port # find_free_port(port)
|
|
if not self.port:
|
|
raise Exception("Cannot find free port for moses server!")
|
|
self.cmd.extend(["--server-port", "%d" % self.port])
|
|
if debug:
|
|
print >>sys.stderr, self.cmd
|
|
# self.stderr = open("mserver.%d.stderr"%self.port,'w')
|
|
# self.stdout = open("mserver.%d.stdout"%self.port,'w')
|
|
# self.process = Popen(
|
|
# self.cmd, stderr=self.stderr, stdout=self.stdout)
|
|
self.process = Popen(self.cmd)
|
|
else:
|
|
devnull = open(os.devnull, "w")
|
|
self.process = Popen(self.cmd, stderr=devnull, stdout=devnull)
|
|
|
|
if self.process.poll():
|
|
raise Exception("FATAL ERROR: Could not launch moses server!")
|
|
if debug:
|
|
print >>sys.stderr, "MOSES port is %d." % self.port
|
|
print >>sys.stderr, "Moses poll status is", self.process.poll()
|
|
|
|
self.url = "http://localhost:%d/RPC2" % self.port
|
|
self.connect(self.url)
|
|
|
|
return True
|
|
|
|
def connect(self, url):
|
|
if url[:4] != "http":
|
|
url = "http://%s" % url
|
|
if url[-5:] != "/RPC2":
|
|
url += "/RPC2"
|
|
self.url = url
|
|
self.proxy = xmlrpclib.ServerProxy(self.url)
|
|
|
|
def translate(self, input):
|
|
attempts = 0
|
|
while attempts < 100:
|
|
try:
|
|
if type(input) is unicode:
|
|
# If the server does not expect unicode, provide a
|
|
# properly encoded string!
|
|
param = {'text': input.strip().encode('utf8')}
|
|
return self.proxy.translate(param)['text'].decode('utf8')
|
|
|
|
elif type(input) is str:
|
|
param = {'text': input.strip()}
|
|
return self.proxy.translate(param)['text']
|
|
|
|
elif type(input) is list:
|
|
return [self.translate(x) for x in input]
|
|
|
|
elif type(input) is dict:
|
|
return self.proxy.translate(input)
|
|
|
|
else:
|
|
raise Exception("Can't handle input of this type!")
|
|
|
|
except:
|
|
attempts += 1
|
|
print >>sys.stderr, "WAITING", attempts
|
|
time.sleep(1)
|
|
raise Exception("Translation request failed")
|