1 files changed, 17 insertions, 11 deletions
diff --git a/word-aligner/support/generate_per_sentence_grammars.pl b/word-aligner/support/generate_per_sentence_grammars.pl
index 80243419..8779ac9c 100755
--- a/word-aligner/support/generate_per_sentence_grammars.pl
+++ b/word-aligner/support/generate_per_sentence_grammars.pl
@@ -2,7 +2,7 @@
 use strict;
 use utf8;
 
-die "Usage: $0 f.voc corpus.f-e grammar.f-e.gz\n" unless scalar @ARGV == 3;
+die "Usage: $0 f.voc corpus.f-e grammar.f-e.gz [OUT]filtered.f-e.gz [OUT]per_sentence_grammar.f-e [OUT]train.f-e.sgml\n" unless scalar @ARGV == 6;
 
 my $MAX_INMEM = 2500;
 
@@ -10,7 +10,14 @@ open FV,"<$ARGV[0]" or die "Can't read $ARGV[0]: $!";
 open C,"<$ARGV[1]" or die "Can't read $ARGV[1]: $!";
 open G,"gunzip -c $ARGV[2]|" or die "Can't read $ARGV[2]: $!";
 
+open FILT,"|gzip -c > $ARGV[3]" or die "Can't write $ARGV[3]: $!";
+open PSG,">$ARGV[4]" or die "Can't write $ARGV[4]: $!";
+open OTRAIN,">$ARGV[5]" or die "Can't write $ARGV[5]: $!";
+
+binmode FILT, ":utf8";
+binmode PSG, ":utf8";
 binmode STDOUT, ":utf8";
+binmode STDERR, ":utf8";
 binmode FV, ":utf8";
 binmode C, ":utf8";
 binmode G, ":utf8";
@@ -35,7 +42,7 @@ while(<G>) {
   chomp;
   my ($f, $e, $feats) = split / \|\|\| /;
   if ($most_freq{$f}) {
-    print "$_\n";
+    print FILT "$_\n";
     $memrc++;
   } else {
     $loadrc++;
@@ -47,20 +54,19 @@ while(<G>) {
     push @$r, "$e ||| $feats";
   }
 }
+close FILT;
 close G;
 print STDERR "  mem rc: $memrc\n";
 print STDERR " load rc: $loadrc\n";
 
 my $id = 0;
-open O, ">ps.grammar" or die;
-binmode(O,":utf8");
 while(<C>) {
   chomp;
   my ($f,$e) = split / \|\|\| /;
   my @fwords = split /\s+/, $f;
   my $tot = 0;
   my %used;
-  my $fpos = tell(O);
+  my $fpos = tell(PSG);
   for my $f (@fwords) {
     next if $most_freq{$f};
     next if $used{$f};
@@ -69,15 +75,15 @@ while(<C>) {
     my $num = scalar @$r;
     $tot += $num;
     for my $rule (@$r) {
-      print O "$f ||| $rule\n";
+      print PSG "$f ||| $rule\n";
     }
     $used{$f} = 1;
   }
-  print O "###EOS###\n";
-  print STDERR "<seg id=\"$id\" grammar=\"\@$fpos\"> $_ </seg>\n";
-  #print STDERR "id=$id POS=$fpos\n";
+  print PSG "###EOS###\n";
+  print OTRAIN "<seg id=\"$id\" psg=\"\@$fpos\"> $_ </seg>\n";
   $id++;
-  last if $id == 10;
 }
+close PSG;
+close OTRAIN;
+print STDERR "Done.\n";
 
-close O;