Blame view

Scripts/utils/sym2int_s3.pl 2.62 KB
ec85f8892   bigot benjamin   first commit
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
  #!/usr/bin/perl
  # Copyright 2010-2012 Microsoft Corporation  Johns Hopkins University (Author: Daniel Povey)
  
  # Licensed under the Apache License, Version 2.0 (the "License");
  # you may not use this file except in compliance with the License.
  # You may obtain a copy of the License at
  #
  #  http://www.apache.org/licenses/LICENSE-2.0
  #
  # THIS CODE IS PROVIDED *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  # KIND, EITHER EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION ANY IMPLIED
  # WARRANTIES OR CONDITIONS OF TITLE, FITNESS FOR A PARTICULAR PURPOSE,
  # MERCHANTABLITY OR NON-INFRINGEMENT.
  # See the Apache 2 License for the specific language governing permissions and
  # limitations under the License.
  
  
  $ignore_oov = 0;
  $ignore_first_field = 0;
  for($x = 0; $x < 3; $x++) {
      # Note: it will just print OOVS unmodified if you specify --ignore-oov.
      # Else will complain and put nothing out.
      if($ARGV[0] eq "--ignore-oov") { $ignore_oov = 1; shift @ARGV; } 
      if($ARGV[0] eq "--ignore-first-field") { $ignore_first_field = 1; shift @ARGV; }
      if($ARGV[0] eq "--map-oov") { shift @ARGV; $map_oov = shift @ARGV; }
  }
  
  $symtab = shift @ARGV;
  if(!defined $symtab) {
      die "Usage: sym2int.pl symtab [input transcriptions] > output transcriptions
  ";
  }
  
  
  open(F, "<$symtab") || die "Error opening symbol table file $symtab";
  while(<F>) {
      @A = split(" ", $_);
      @A == 2 || die "bad line in symbol table file: $_";
      $sym2int{$A[0]} = $A[1] + 0;
  }
  
  
  $num_warning = 0;
  $max_warning = 20;
  $error = 0;
  while(<>) {
      @A = split(" ", $_);
      if(@A == 0) {
          die "Empty line in transcriptions input.";
      }
      if($ignore_first_field) {
          $key = shift @A;
          print $key . " ";
      }
      @B = ();
      foreach $a (@A) {
          $i = $sym2int{$a};
          if(!defined ($i)) {
              if (defined $map_oov) {
                  if (!defined $sym2int{$map_oov}) {
                      die "sym2int.pl: invalid map-oov option $map_oov (symbol not defined in $symtab)";
                  }
                  if ($num_warning++ < $max_warning) {
                      print STDERR "sym2int.pl: replacing $a with $map_oov
  ";
                      if ($num_warning == $max_warning) {
                          print STDERR "sym2int.pl: not warning for OOVs any more times
  ";
                      }
                  }
                  $i = $sym2int{$map_oov};
              } elsif($ignore_oov) {
                  $i = $a; # just print them out unmodified..
              } else {
                  die "sym2int.pl: undefined symbol $a
  ";
              }
          }
          push @B, $i;
      }
      print join(" ", @B);
      print "
  ";
  }
  
  if($error) { exit(1); }
  else { exit(0); }