moses-smt/mosesdecoder
1
1
Fork 0
mirror of https://github.com/moses-smt/mosesdecoder.git synced 2024-12-26 05:14:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

tighten up extract-parallel on osx. Can now use gsplit and bsd split

Browse Source
This commit is contained in:
Hieu Hoang 2015-06-26 11:37:35 +04:00
parent b83803203e
commit ca54852641
1 changed files with 31 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
scripts/generic/extract-parallel.perl
View File

@ -1,4 +1,4 @@
#!/usr/bin/env perl
#!/usr/bin/env perl
#
# 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.
@ -15,8 +15,7 @@ sub systemCheck($);
sub NumStr($);
sub DigitStr($);
sub CharStr($);
my $is_osx = ($^O eq "darwin");
sub GetSplitVersion($);
my $alph = "abcdefghijklmnopqrstuvwxyz";
my @alph = (split(//,$alph));
@ -42,7 +41,7 @@ my $baselineExtract;
my $glueFile;
my $phraseOrientation = 0;
my $phraseOrientationPriorsFile;
my $splitCmdOption="-d";
my $splitCmdOption = "";
my $GZIP_EXEC;
if(`which pigz`) {
@ -53,6 +52,15 @@ else {
}
print STDERR "using $GZIP_EXEC \n";
my $isBSDSplit = GetSplitVersion($splitCmd);
print STDERR "isBSDSplit=$isBSDSplit \n";
if ($isBSDSplit == 0) {
$splitCmdOption .= "-d";
}
my $gzOut = 0;
for (my $i = 8; $i < $#ARGV + 1; ++$i)
{
$makeTTable = 0 if $ARGV[$i] eq "--NoTTable";
@ -73,11 +81,15 @@ for (my $i = 8; $i < $#ARGV + 1; ++$i)
$phraseOrientationPriorsFile = $ARGV[++$i];
next;
}
$splitCmdOption="",next if $ARGV[$i] eq "--NoNumericSuffix";
if ($ARGV[$i] eq '--GZOutput') {
$gzOut = 1;
}
$otherExtractArgs .= $ARGV[$i] ." ";
}
die("Need to specify --GZOutput for parallel extract") if ($gzOut == 0);
my $cmd;
my $TMPDIR=dirname($extract) ."/tmp.$$";
$cmd = "mkdir -p $TMPDIR; ls -l $TMPDIR";
@ -272,7 +284,7 @@ if ($phraseOrientation && defined($phraseOrientationPriorsFile)) {
# delete temporary files
$cmd = "rm -rf $TMPDIR \n";
`$cmd`;
systemCheck($cmd);
print STDERR "Finished ".localtime() ."\n";
@ -352,10 +364,22 @@ sub CharStr($)
sub NumStr($)
{
my $i = shift;
if ($is_osx){
if ($isBSDSplit){
return CharStr($i);
}else{
return DigitStr($i);
}
}
sub GetSplitVersion($)
{
my $splitCmd = shift;
my $retVal = system("$splitCmd -h");
if ($retVal != 0) {
return 1;
}
else {
return 0;
}
}