mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-09-11 11:25:40 +03:00
Python implementation of parallel scoring.
Re-implementation of score-parallel.perl. Not a drop-in replacement; the command line is similar but different and uses the standard Python command-line parser. Written without much knowledge of the original script, so documentation in particular may seem nonsensical to experts. If you see something wrong, please help!
This commit is contained in:
parent
94cd1f7433
commit
710915c088
775
scripts/generic/score_parallel.py
Executable file
775
scripts/generic/score_parallel.py
Executable file
@ -0,0 +1,775 @@
|
||||
#! /usr/bin/env python
|
||||
#
|
||||
# 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.
|
||||
#
|
||||
# Script contributed by Precision Translation Tools.
|
||||
|
||||
"""Run Moses `score` jobs in parallel.
|
||||
|
||||
This script is a replacement for `score-parallel.perl`. The two are similar,
|
||||
but there are differences in usage. In addition, this script can be called
|
||||
directly from Python code without the need to run it as a separate process.
|
||||
"""
|
||||
|
||||
from __future__ import (
|
||||
absolute_import,
|
||||
print_function,
|
||||
unicode_literals,
|
||||
)
|
||||
|
||||
__metaclass__ = type
|
||||
|
||||
from argparse import ArgumentParser
|
||||
from contextlib import contextmanager
|
||||
from datetime import datetime
|
||||
import errno
|
||||
import gzip
|
||||
from multiprocessing import Pool
|
||||
import os
|
||||
import os.path
|
||||
import pipes
|
||||
from shutil import rmtree
|
||||
from subprocess import check_call
|
||||
import sys
|
||||
import tempfile
|
||||
|
||||
|
||||
def get_unicode_type():
|
||||
"""Return the Unicode string type appropriate to this Python version."""
|
||||
if sys.version_info.major <= 2:
|
||||
# Unicode string type. In Python 2 this is the "unicode" type,
|
||||
# while "str" is a binary string type.
|
||||
return unicode
|
||||
else:
|
||||
# Unicode string type. In Python 3 this is the default "str" type.
|
||||
# The binary string type is now called "bytes".
|
||||
return str
|
||||
|
||||
|
||||
UNICODE_TYPE = get_unicode_type()
|
||||
|
||||
|
||||
class CommandLineError(Exception):
|
||||
"""Invalid command line."""
|
||||
|
||||
|
||||
class ProgramFailure(Exception):
|
||||
"""Failure, not a bug, which is reported neatly to the user."""
|
||||
|
||||
|
||||
def parse_args():
|
||||
"""Parse command line arguments, return as `Namespace`."""
|
||||
parser = ArgumentParser(description=__doc__)
|
||||
parser.add_argument(
|
||||
'--extract-file', '-e', metavar='PATH', required=True,
|
||||
help=(
|
||||
"Path to input file: extract file (e.g. 'extract.sorted.gz' or "
|
||||
"'extract.inv.sorted.gz'). Required."))
|
||||
parser.add_argument(
|
||||
'--lex-file', '-l', metavar='PATH', required=True,
|
||||
help=(
|
||||
"Path to input file: lex file (e.g. 'lex.f2e' or 'lex.e2f'). "
|
||||
"Required."))
|
||||
parser.add_argument(
|
||||
'--output', '-o', metavar='PATH', required=True,
|
||||
help=(
|
||||
"Write phrase table to file PATH (e.g. 'phrase-table.half.f2e' "
|
||||
"or 'phrase-table.half.e2f'). Required."))
|
||||
parser.add_argument(
|
||||
'--inverse', '-i', action='store_true',
|
||||
help="Inverse scoring. Defaults to direct scoring.")
|
||||
parser.add_argument(
|
||||
'--labels-file', '-L', metavar='PATH',
|
||||
help="Also write source labels to file PATH.")
|
||||
parser.add_argument(
|
||||
'--parts-of-speech', '-p', metavar='PATH',
|
||||
help="Also write parts-of-speech file to PATH.")
|
||||
parser.add_argument(
|
||||
'--flexibility-score', '-F', metavar='PATH',
|
||||
help="Path to the 'flexibility_score.py' script. Defaults to none.")
|
||||
parser.add_argument(
|
||||
'--hierarchical', '-H', action='store_true',
|
||||
help="Process hierarchical rules.")
|
||||
parser.add_argument(
|
||||
'--args', '-a', metavar='ARGUMENTS',
|
||||
help="Additional arguments for `score` and `flexibility_score`.")
|
||||
parser.add_argument(
|
||||
'--sort', '-s', action='store_true',
|
||||
help="Sort output file.")
|
||||
parser.add_argument(
|
||||
'--jobs', '-j', metavar='N', type=int, default=1,
|
||||
help="Run up to N jobs in parallel. Defaults to %(default)s.")
|
||||
parser.add_argument(
|
||||
'--score-exe', '-x', metavar='PROGRAM',
|
||||
help="Name of, or path to, the 'score' executable.")
|
||||
parser.add_argument(
|
||||
'--sort-command', '-S', metavar='COMMAND-LINE',
|
||||
help=(
|
||||
"Command line for sorting text files to standard output. "
|
||||
"Must support operation as a pipe, as well as input files named "
|
||||
"as command-line arguments."))
|
||||
parser.add_argument(
|
||||
'--gzip-command', '-z', metavar='PROGRAM',
|
||||
help="Path to a gzip or pigz executable.")
|
||||
parser.add_argument(
|
||||
'--verbose', '-v', action='store_true',
|
||||
help="Print what's going on.")
|
||||
parser.add_argument(
|
||||
'--debug', '-d', action='store_true',
|
||||
help="Don't delete temporary directories when done.")
|
||||
return parser.parse_args()
|
||||
|
||||
|
||||
def normalize_path(optional_path=None):
|
||||
"""Return a cleaned-up version of a given filesystem path, or None.
|
||||
|
||||
Converts the path to the operating system's native conventions, and
|
||||
removes redundancies like `.`.
|
||||
|
||||
The return value will be `None`, an absolute path, or a relative path,
|
||||
same as the argument. But it will have redundant path separators,
|
||||
unnecessary detours through parent directories, and use of the current
|
||||
directory "." removed.
|
||||
"""
|
||||
if optional_path is None:
|
||||
return None
|
||||
else:
|
||||
path = os.path.normpath(optional_path)
|
||||
path = path.replace('/', os.path.sep)
|
||||
path = path.replace('\\', os.path.sep)
|
||||
return path
|
||||
|
||||
|
||||
def quote(path):
|
||||
"""Quote and escape a filename for use in a shell command.
|
||||
|
||||
The Windows implementation is very limited and will break on anything
|
||||
more advanced than a space.
|
||||
"""
|
||||
if os.name == 'posix':
|
||||
return pipes.quote(path)
|
||||
else:
|
||||
# TODO: Improve escaping for Windows.
|
||||
return '"%s"' % path
|
||||
|
||||
|
||||
def sanitize_args(args):
|
||||
"""Check `args` for sanity, clean up, and set nontrivial defaults."""
|
||||
if args.jobs < 1:
|
||||
raise CommandLineError("Number of parallel jobs must be 1 or more.")
|
||||
if args.sort_command is None:
|
||||
args.sort_command = find_first_executable(
|
||||
['neandersort', 'gsort', 'sort'])
|
||||
if args.sort_command is None:
|
||||
raise CommandLineError(
|
||||
"No 'sort' command is available. "
|
||||
"Choose one using the --sort-command option.")
|
||||
if args.gzip_command is None:
|
||||
args.gzip_command = find_first_executable(['pigz', 'gzip'])
|
||||
if args.gzip_command is None:
|
||||
raise CommandLineError(
|
||||
"No 'gzip' or 'pigz' command is available. "
|
||||
"Choose one using the --gzip-command option.")
|
||||
if args.score_exe is None:
|
||||
# Look for "score" executable. It may be in the current project
|
||||
# directory somewhere, or in the PATH.
|
||||
moses_dir = os.path.dirname(os.path.dirname(
|
||||
os.path.abspath(__file__)))
|
||||
args.score_exe = find_first_executable(
|
||||
['score'],
|
||||
[
|
||||
moses_dir,
|
||||
os.path.join(moses_dir, 'phrase-extract'),
|
||||
os.path.join(moses_dir, 'binaries'),
|
||||
])
|
||||
args.extract_file = normalize_path(args.extract_file)
|
||||
args.lex_file = normalize_path(args.lex_file)
|
||||
args.output = normalize_path(args.output)
|
||||
args.labels_file = normalize_path(args.labels_file)
|
||||
args.parts_of_speech = normalize_path(args.parts_of_speech)
|
||||
args.flexibility_score = normalize_path(args.flexibility_score)
|
||||
args.score_exe = normalize_path(args.score_exe)
|
||||
|
||||
|
||||
def add_exe_suffix(program):
|
||||
"""Return the full filename for an executable.
|
||||
|
||||
On Windows, this adds a `.exe` suffix to the name. On other
|
||||
systems, it returns the original name unchanged.
|
||||
"""
|
||||
if os.name == 'nt':
|
||||
# Windows.
|
||||
return program + '.exe'
|
||||
else:
|
||||
# Assume POSIX or similar.
|
||||
return program
|
||||
|
||||
|
||||
def find_executable(exe, extra_path=None):
|
||||
"""Return full path to an executable of the given name, or `None`.
|
||||
|
||||
If the given name is a qualified path to an executable, it will be returned
|
||||
unchanged. A qualified path where no executable is found results in a
|
||||
`CommandLineError`.
|
||||
"""
|
||||
if extra_path is None:
|
||||
extra_path = []
|
||||
|
||||
if os.path.sep in exe:
|
||||
# The executable name includes a path. Only one place it can be.
|
||||
if not os.path.isfile(exe) or not os.access(exe, os.X_OK):
|
||||
raise CommandLineError("Not an executable: '%s'." % exe)
|
||||
return exe
|
||||
|
||||
for path in extra_path + os.getenv('PATH').split(os.pathsep):
|
||||
full_path = os.path.join(path, exe)
|
||||
if os.access(full_path, os.X_OK):
|
||||
return full_path
|
||||
return None
|
||||
|
||||
|
||||
def find_first_executable(candidates, extra_path=None):
|
||||
"""Find the first available of the given candidate programs.
|
||||
|
||||
:raise ProgramFailure: If none of `candidates` was found.
|
||||
"""
|
||||
for program in candidates:
|
||||
executable = find_executable(add_exe_suffix(program), extra_path)
|
||||
if executable is not None:
|
||||
return executable
|
||||
raise ProgramFailure(
|
||||
"Could not find any of these executables in path: %s."
|
||||
% ', '.join(candidates))
|
||||
|
||||
|
||||
def execute_shell(command, verbose=False):
|
||||
"""Run `command` string through the shell.
|
||||
|
||||
Inherits environment, but sets `LC_ALL` to `C` for predictable results,
|
||||
especially from sort commands.
|
||||
|
||||
This uses a full-featured shell, including pipes, substitution, etc. So
|
||||
remember to quote/escape arguments where appropriate!
|
||||
"""
|
||||
assert isinstance(command, UNICODE_TYPE), (
|
||||
"Wrong argument for execute_shell.")
|
||||
if verbose:
|
||||
print("Executing: %s" % command)
|
||||
env = os.environ.copy()
|
||||
if os.name == 'posix':
|
||||
env['LC_ALL'] = 'C'
|
||||
check_call(command, shell=True, env=env)
|
||||
|
||||
|
||||
@contextmanager
|
||||
def tempdir(keep=False):
|
||||
"""Context manager: temporary directory."""
|
||||
directory = tempfile.mkdtemp()
|
||||
yield directory
|
||||
if not keep:
|
||||
rmtree(directory)
|
||||
|
||||
|
||||
def make_dirs(path):
|
||||
"""Equivalent to `mkdir -p -- path`."""
|
||||
try:
|
||||
os.makedirs(path)
|
||||
except OSError as error:
|
||||
if error.errno != errno.EEXIST:
|
||||
raise
|
||||
|
||||
|
||||
def open_file(path, mode='r'):
|
||||
"""Open a file, which may be gzip-compressed."""
|
||||
if path.endswith('.gz'):
|
||||
return gzip.open(path, mode)
|
||||
else:
|
||||
return open(path, mode)
|
||||
|
||||
|
||||
def count_lines(filename):
|
||||
"""Count the number of lines in `filename` (may be gzip-compressed)."""
|
||||
count = 0
|
||||
with open_file(filename) as stream:
|
||||
for _ in stream:
|
||||
count += 1
|
||||
return count
|
||||
|
||||
|
||||
def set_temp_dir():
|
||||
"""Set temporary directory to `$MOSES_TEMP_DIR`, if set.
|
||||
|
||||
Create the directory if necessary.
|
||||
"""
|
||||
temp_dir = os.getenv('MOSES_TEMP_DIR')
|
||||
if temp_dir is not None:
|
||||
make_dirs(temp_dir)
|
||||
tempfile.tempdir = temp_dir
|
||||
|
||||
|
||||
def strip_newline(line):
|
||||
"""Remove trailing carriage return and/or line feed, if present."""
|
||||
if line.endswith('\n'):
|
||||
line = line[:-1]
|
||||
if line.endswith('\r'):
|
||||
line = line[:-1]
|
||||
return line
|
||||
|
||||
|
||||
def open_chunk_file(split_dir, chunk_number):
|
||||
"""Open a file to write one chunk of the extract file."""
|
||||
return open_file(
|
||||
os.path.join(split_dir, 'extract.%d.gz' % chunk_number), 'w')
|
||||
|
||||
|
||||
def name_context_chunk_file(split_dir, chunk_number):
|
||||
"""Compose file name for one chunk of the extract context file."""
|
||||
return os.path.join(
|
||||
split_dir, 'extract.context.%d.gz' % chunk_number)
|
||||
|
||||
|
||||
def extract_source_phrase(line):
|
||||
"""Extract the source phrase from an extract-file line."""
|
||||
return line.split(b'|||', 1)[0]
|
||||
|
||||
|
||||
def cut_context_file(last_source_phrase, chunk_file, last_line,
|
||||
context_stream):
|
||||
"""Write one chunk of extract context file into its own file.
|
||||
|
||||
:param last_source_phrase: Last source phrase that should be in the
|
||||
chunk. Stop processing after this source phrase.
|
||||
:param chunk_file: Path to the extract context file for this chunk.
|
||||
:param last_line: Previously read line that may still need writing.
|
||||
:param context_stream: Extract context file, opened for reading.
|
||||
:return: Last line read from `context_stream`. This line will still
|
||||
need processing.
|
||||
"""
|
||||
# TODO: Use open_file.
|
||||
with gzip.open(chunk_file, 'w') as chunk:
|
||||
if last_line is not None:
|
||||
chunk.write('%s\n' % last_line)
|
||||
|
||||
# Are we processing our last source phrase yet?
|
||||
on_last_source_phrase = False
|
||||
|
||||
# Write all lines in context file until we meet last source phrase
|
||||
# in extract file.
|
||||
for line in context_stream:
|
||||
# Reading from a gzip file returns lines *including the newline*.
|
||||
# Either way, we want to ignore carriage returns as well.
|
||||
line = strip_newline(line)
|
||||
source_phrase = extract_source_phrase(line)
|
||||
if on_last_source_phrase and source_phrase != last_source_phrase:
|
||||
# First new source phrase after our last one. We're done.
|
||||
return line
|
||||
else:
|
||||
# Still adding lines to our chunk.
|
||||
chunk.write('%s\n' % line)
|
||||
if source_phrase == last_source_phrase:
|
||||
# We're on our last source phrase now.
|
||||
on_last_source_phrase = True
|
||||
|
||||
|
||||
def split_extract_files(split_dir, extract_file, extract_context_file=None,
|
||||
jobs=1):
|
||||
"""Split extract file into chunks, so we can process them in parallel.
|
||||
|
||||
:param split_dir: A temporary directory where this function can write
|
||||
temporary files. The caller must ensure that this directory will be
|
||||
cleaned up after it's done with the files.
|
||||
:return: An iterable of tuples. Each tuple hols a partial extract file,
|
||||
and the corresponding context file. The files may be in `split_dir`,
|
||||
or there may just be the original extract file.
|
||||
"""
|
||||
if jobs == 1:
|
||||
# No splitting needed. Read the original file(s).
|
||||
return [(extract_file, extract_context_file)]
|
||||
|
||||
# Otherwise: split files.
|
||||
files = []
|
||||
num_lines = count_lines(extract_file)
|
||||
chunk_size = (num_lines + jobs - 1) / jobs
|
||||
assert isinstance(chunk_size, int)
|
||||
|
||||
line_count = 0
|
||||
chunk_number = 0
|
||||
prev_source_phrase = None
|
||||
last_line_context = None
|
||||
extract_stream = open_file(extract_file)
|
||||
chunk_file = open_chunk_file(split_dir, chunk_number)
|
||||
if extract_context_file is None:
|
||||
chunk_context_file = None
|
||||
if extract_context_file is not None:
|
||||
context_stream = open_file(extract_context_file)
|
||||
|
||||
for line in extract_stream:
|
||||
line_count += 1
|
||||
line = line.decode('utf-8')
|
||||
line = strip_newline(line)
|
||||
if line_count >= chunk_size:
|
||||
# At or over chunk size. Cut off at next source phrase change.
|
||||
source_phrase = extract_source_phrase(line)
|
||||
if prev_source_phrase is None:
|
||||
# Start looking for a different source phrase.
|
||||
prev_source_phrase = source_phrase
|
||||
elif source_phrase == prev_source_phrase:
|
||||
# Can't cut yet. Still working on the same source phrase.
|
||||
pass
|
||||
else:
|
||||
# Hit first new source phrase after chunk limit. Cut new
|
||||
# file(s).
|
||||
chunk_file.close()
|
||||
if extract_context_file is not None:
|
||||
chunk_context_file = name_context_chunk_file(
|
||||
split_dir, chunk_number)
|
||||
last_line_context = cut_context_file(
|
||||
prev_source_phrase, chunk_context_file,
|
||||
last_line_context, context_stream)
|
||||
files.append((chunk_file.name, chunk_context_file))
|
||||
|
||||
# Start on new chunk.
|
||||
prev_source_phrase = None
|
||||
line_count = 0
|
||||
chunk_number += 1
|
||||
chunk_file = open_chunk_file(split_dir, chunk_number)
|
||||
chunk_file.write(('%s\n' % line).encode('utf-8'))
|
||||
|
||||
chunk_file.close()
|
||||
if extract_context_file is not None:
|
||||
chunk_context_file = name_context_chunk_file(split_dir, chunk_number)
|
||||
last_line_context = cut_context_file(
|
||||
prev_source_phrase, chunk_number, last_line_context,
|
||||
context_stream)
|
||||
files.append((chunk_file.name, chunk_context_file))
|
||||
return files
|
||||
|
||||
|
||||
def compose_score_command(extract_file, context_file, half_file,
|
||||
flex_half_file, args):
|
||||
"""Compose command line text to run one instance of `score`.
|
||||
|
||||
:param extract_file: One chunk of extract file.
|
||||
:param context_file: If doing flexibility scoring, one chunk of
|
||||
extract context file. Otherwise, None.
|
||||
:param half_file: ???
|
||||
:param flex_half_file: ???
|
||||
:param args: Arguments namespace.
|
||||
"""
|
||||
command = [
|
||||
args.score_exe,
|
||||
extract_file,
|
||||
args.lex_file,
|
||||
half_file,
|
||||
]
|
||||
if args.args not in (None, ''):
|
||||
command.append(args.args)
|
||||
other_args = build_score_args(args)
|
||||
if other_args != '':
|
||||
command.append(other_args)
|
||||
if context_file is not None:
|
||||
command += [
|
||||
'&&',
|
||||
find_first_executable(['bzcat']),
|
||||
'|',
|
||||
quote(args.flexibility_score),
|
||||
quote(context_file),
|
||||
]
|
||||
if args.inverse:
|
||||
command.append('--Inverse')
|
||||
if args.hierarchical:
|
||||
command.append('--Hierarchical')
|
||||
command += [
|
||||
'|',
|
||||
quote(args.gzip_command),
|
||||
'-c',
|
||||
'>%s' % quote(flex_half_file),
|
||||
]
|
||||
return ' '.join(command)
|
||||
|
||||
|
||||
def score_parallel(split_dir, file_pairs, args):
|
||||
"""Run the `score` command in parallel.
|
||||
|
||||
:param split_dir: Temporary directory where we can create split files.
|
||||
:param file_pairs: Sequence of tuples for the input files, one tuple
|
||||
per chunk of the work. Each tuple consists of a partial extract
|
||||
file, and optionally a partial extract context file.
|
||||
:param args: Arguments namespace.
|
||||
:return: A list of tuples. Each tuple contains two file paths. The first
|
||||
is for a partial half-phrase-table file. The second is for the
|
||||
corresponding partial flex file, if a context file is given; or
|
||||
`None` otherwise.
|
||||
"""
|
||||
partial_files = []
|
||||
# Pool of worker processes for executing the partial "score" invocations
|
||||
# concurrently.
|
||||
pool = Pool(args.jobs)
|
||||
try:
|
||||
for chunk_num, file_pair in enumerate(file_pairs):
|
||||
half_file = os.path.join(
|
||||
split_dir, 'phrase-table.half.%06d.gz' % chunk_num)
|
||||
extract_file, context_file = file_pair
|
||||
if context_file is None:
|
||||
flex_half_file = None
|
||||
else:
|
||||
flex_half_file = os.path.join(
|
||||
split_dir, 'phrase-table.half.%06d.flex.gz' % chunk_num)
|
||||
# Pickling of arguments for the pool is awkward on Windows, so
|
||||
# keep them simple. Compose the command line in the parent
|
||||
# process, then hand them to worker processes which execute them.
|
||||
command_line = compose_score_command(
|
||||
extract_file, context_file, half_file, flex_half_file, args)
|
||||
pool.apply_async(
|
||||
execute_shell, (command_line, ), {'verbose': args.verbose})
|
||||
partial_files.append((half_file, flex_half_file))
|
||||
pool.close()
|
||||
except BaseException:
|
||||
pool.terminate()
|
||||
raise
|
||||
finally:
|
||||
pool.join()
|
||||
return partial_files
|
||||
|
||||
|
||||
def merge_and_sort(files, output, sort_command=None, gzip_exe=None,
|
||||
verbose=False):
|
||||
"""Merge partial files.
|
||||
|
||||
:param files: List of partial half-phrase-table files.
|
||||
:param output: Path for resulting combined phrase-table file.
|
||||
"""
|
||||
# TODO: The Perl code mentioned "sort" and "flexibility_score" here.
|
||||
# What do we do with those?
|
||||
|
||||
# Sort whether we're asked to or not, as a way of combining the input
|
||||
# files.
|
||||
if sort_command == 'neandersort':
|
||||
# Neandersort transparently decompresses input and compresses output.
|
||||
check_call([
|
||||
'neandersort',
|
||||
'-o', output,
|
||||
] + files)
|
||||
else:
|
||||
command = (
|
||||
"%(gzip)s -c -d %(files)s | "
|
||||
"%(sort)s | "
|
||||
"%(gzip)s -c >>%(output)s"
|
||||
% {
|
||||
'gzip': quote(gzip_exe),
|
||||
'sort': sort_command,
|
||||
'files': ' '.join(map(quote, files)),
|
||||
'output': quote(output),
|
||||
})
|
||||
execute_shell(command, verbose=verbose)
|
||||
|
||||
|
||||
def build_score_args(args):
|
||||
"""Compose command line for the `score` program."""
|
||||
command_line = []
|
||||
if args.labels_file:
|
||||
command_line += [
|
||||
'--SourceLabels',
|
||||
'--SourceLabelCountsLHS',
|
||||
'--SourceLabelSet',
|
||||
]
|
||||
if args.parts_of_speech:
|
||||
command_line.append('--PartsOfSpeech')
|
||||
if args.inverse:
|
||||
command_line.append('--Inverse')
|
||||
if args.args is not None:
|
||||
command_line.append(args.args)
|
||||
return ' '.join(command_line)
|
||||
|
||||
|
||||
def list_existing(paths):
|
||||
"""Return, in the same order, those of the given files which exist."""
|
||||
return filter(os.path.exists, paths)
|
||||
|
||||
|
||||
def compose_coc_path_for(path):
|
||||
"""Compose COC-file path for the given file."""
|
||||
return '%s.coc' % path
|
||||
|
||||
|
||||
def read_cocs(path):
|
||||
"""Read COC file at `path`, return contents as tuple of ints."""
|
||||
with open(path) as lines:
|
||||
return tuple(
|
||||
int(line.rstrip('\r\n'))
|
||||
for line in lines
|
||||
)
|
||||
|
||||
|
||||
def add_cocs(original, additional):
|
||||
"""Add two tuples of COCs. Extend as needed."""
|
||||
assert not (original is None and additional is None), "No COCs to add!"
|
||||
if original is None:
|
||||
return additional
|
||||
elif additional is None:
|
||||
return original
|
||||
else:
|
||||
common = tuple(lhs + rhs for lhs, rhs in zip(original, additional))
|
||||
return (
|
||||
common +
|
||||
tuple(original[len(common):]) +
|
||||
tuple(additional[len(common):]))
|
||||
|
||||
|
||||
def merge_coc(files, output):
|
||||
"""Merge COC files for the given partial files.
|
||||
|
||||
Each COC file is a series of integers, one per line. This reads them, and
|
||||
adds them up line-wise into one file of the same format: the sum of the
|
||||
numbers the respective files have at line 1, the sum of the numbers the
|
||||
respective files have at line 2, and so on.
|
||||
"""
|
||||
assert len(files) > 0, "No partial files - no work to do."
|
||||
extract_files = [extract_file for extract_file, _ in files]
|
||||
if not os.path.exists(compose_coc_path_for(extract_files[0])):
|
||||
# Nothing to merge.
|
||||
return
|
||||
totals = None
|
||||
# TODO: Shouldn't we just fail if any of these files is missing?
|
||||
for coc_path in list_existing(map(compose_coc_path_for, extract_files)):
|
||||
totals = add_cocs(totals, read_cocs(coc_path))
|
||||
|
||||
# Write to output file.
|
||||
with open(output, 'w') as output_stream:
|
||||
for entry in totals:
|
||||
output_stream.write('%d\n' % entry)
|
||||
|
||||
|
||||
def suffix_line_numbers(infile, outfile):
|
||||
"""Rewrite `infile` to `outfile`; suffix line number to each line.
|
||||
|
||||
The line number is zero-based, and separated from the rest of the line
|
||||
by a single space.
|
||||
"""
|
||||
temp_file = '%s.numbering' % outfile
|
||||
with open(infile, 'r') as instream, open(outfile, 'w') as outstream:
|
||||
line_no = 0
|
||||
for line in instream:
|
||||
outstream.write(line)
|
||||
outstream.write(' %d\n' % line_no)
|
||||
line_no += 1
|
||||
os.rename(temp_file, outfile)
|
||||
|
||||
|
||||
def compose_source_labels_path_for(path):
|
||||
"""Return source labels file path for given file."""
|
||||
return '%s.syntaxLabels.src' % path
|
||||
|
||||
|
||||
def merge_numbered_files(inputs, output, header_lines, sort_command,
|
||||
verbose=False):
|
||||
"""Sort and merge files `inputs`, add header and line numbers.
|
||||
|
||||
:param inputs: Iterable of input files.
|
||||
:param output: Output file.
|
||||
:header_lines: Iterable of header lines.
|
||||
:sort_command: Command line for sorting input files.
|
||||
"""
|
||||
sort_temp = '%s.sorting' % output
|
||||
with open(sort_temp, 'w') as stream:
|
||||
for line in header_lines:
|
||||
stream.write(line)
|
||||
stream.write('\n')
|
||||
execute_shell(
|
||||
"%s %s >>%s" % (
|
||||
sort_command,
|
||||
' '.join(map(quote, inputs)),
|
||||
quote(sort_temp)),
|
||||
verbose=verbose)
|
||||
suffix_line_numbers(sort_temp, output)
|
||||
|
||||
|
||||
def merge_source_labels(files, output, sort_command, verbose=False):
|
||||
"""Merge source labels files."""
|
||||
# TODO: Shouldn't we just fail if any of these files is missing?
|
||||
labels_files = list_existing(map(compose_source_labels_path_for, files))
|
||||
header = [
|
||||
'GlueTop',
|
||||
'GlueX',
|
||||
'SSTART',
|
||||
'SEND',
|
||||
]
|
||||
merge_numbered_files(
|
||||
labels_files, output, header, sort_command, verbose=verbose)
|
||||
|
||||
|
||||
def compose_parts_of_speech_path_for(path):
|
||||
"""Return parts-of-speech file path for given file."""
|
||||
return '%s.partsOfSpeech' % path
|
||||
|
||||
|
||||
def merge_parts_of_speech(files, output, sort_command, verbose=False):
|
||||
"""Merge parts-of-speech files into output."""
|
||||
# TODO: Shouldn't we just fail if any of these files is missing?
|
||||
parts_files = list_existing(map(compose_parts_of_speech_path_for, files))
|
||||
header = [
|
||||
'SSTART',
|
||||
'SEND',
|
||||
]
|
||||
merge_numbered_files(
|
||||
parts_files, output, header, sort_command, verbose=verbose)
|
||||
|
||||
|
||||
def main():
|
||||
"""Command-line entry point. Marshals and forwards to `score_parallel`."""
|
||||
args = parse_args()
|
||||
sanitize_args(args)
|
||||
set_temp_dir()
|
||||
|
||||
if args.flexibility_score is None:
|
||||
extract_context_file = None
|
||||
else:
|
||||
extract_context_file = args.extract_file.replace(
|
||||
'extract.', 'extract.context.')
|
||||
|
||||
if args.verbose:
|
||||
print("Started %s." % datetime.now())
|
||||
print("Using '%s' for gzip." % args.gzip_command)
|
||||
|
||||
with tempdir(args.debug) as split_dir:
|
||||
extract_files = split_extract_files(
|
||||
split_dir, args.extract_file,
|
||||
extract_context_file=extract_context_file, jobs=args.jobs)
|
||||
|
||||
scored_files = score_parallel(split_dir, extract_files, args)
|
||||
|
||||
if args.verbose:
|
||||
sys.stderr.write("Finished score %s.\n" % datetime.now())
|
||||
|
||||
# TODO: Pass on "sort" and "flexibility-score" arguments?
|
||||
merge_and_sort(
|
||||
[phrase_chunk for phrase_chunk, _ in scored_files], args.output,
|
||||
sort_command=args.sort_command, gzip_exe=args.gzip_command,
|
||||
verbose=args.verbose)
|
||||
merge_coc(extract_files, compose_coc_path_for(args.output))
|
||||
|
||||
if not args.inverse and args.labels_file is not None:
|
||||
if args.verbose:
|
||||
print("Merging source labels files.")
|
||||
merge_source_labels(
|
||||
extract_files, args.labels_file,
|
||||
sort_command=args.sort_command, verbose=args.verbose)
|
||||
|
||||
if not args.inverse and args.parts_of_speech is not None:
|
||||
if args.verbose:
|
||||
print("Merging parts-of-speech files.")
|
||||
merge_parts_of_speech(
|
||||
extract_files, args.parts_of_speech,
|
||||
sort_command=args.sort_command, verbose=args.verbose)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
try:
|
||||
main()
|
||||
except ProgramFailure as error:
|
||||
sys.stderr.write('%s\n' % error)
|
||||
sys.exit(1)
|
||||
except CommandLineError as error:
|
||||
sys.stderr.write("Command line error: %s\n" % error)
|
||||
sys.exit(2)
|
Loading…
Reference in New Issue
Block a user