Blame view
src/fstbin/fstdeterminizestar.cc
5.08 KB
8dcb6dfcb 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 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 |
// fstbin/fstdeterminizestar.cc // Copyright 2009-2011 Microsoft Corporation // See ../../COPYING for clarification regarding multiple authors // // 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. #include "base/kaldi-common.h" #include "util/common-utils.h" #include "fst/fstlib.h" #include "fstext/determinize-star.h" #include "fstext/fstext-utils.h" #include "fstext/kaldi-fst-io.h" #if !defined(_MSC_VER) && !defined(__APPLE__) #include <signal.h> // Comment this line and the call to signal below if // it causes compilation problems. It is only to enable a debugging procedure // when determinization does not terminate. We are disabling this code if // compiling on Windows because signal.h is not available there, and on // MacOS due to a problem with <signal.h> in the initial release of Sierra. #endif /* some test examples: ( echo "0 0 0 0"; echo "0 0" ) | fstcompile | fstdeterminizestar | fstprint ( echo "0 0 1 0"; echo "0 0" ) | fstcompile | fstdeterminizestar | fstprint ( echo "0 0 1 0"; echo "0 1 1 0"; echo "0 0" ) | fstcompile | fstdeterminizestar | fstprint # this last one fails [correctly]: ( echo "0 0 0 1"; echo "0 0" ) | fstcompile | fstdeterminizestar | fstprint cd ~/tmpdir while true; do fstrand > 1.fst fstpredeterminize out.lst 1.fst | fstdeterminizestar | fstrmsymbols out.lst > 2.fst fstequivalent --random=true 1.fst 2.fst || echo "Test failed" echo -n "." done Test of debugging [with non-determinizable input]: ( echo " 0 0 1 0 1.0"; echo "0 1 1 0"; echo "1 1 1 0 0"; echo "0 2 2 0"; echo "2"; echo "1" ) | fstcompile | fstdeterminizestar kill -SIGUSR1 [the process-id of fstdeterminizestar] # prints out a bunch of debugging output showing the mess it got itself into. */ bool debug_location = false; void signal_handler(int) { debug_location = true; } int main(int argc, char *argv[]) { try { using namespace kaldi; using namespace fst; using kaldi::int32; const char *usage = "Removes epsilons and determinizes in one step " " " "Usage: fstdeterminizestar [in.fst [out.fst] ] " " " "See also: fstdeterminizelog, lattice-determinize "; float delta = kDelta; int max_states = -1; bool use_log = false; ParseOptions po(usage); po.Register("use-log", &use_log, "Determinize in log semiring."); po.Register("delta", &delta, "Delta value used to determine equivalence of weights."); po.Register("max-states", &max_states, "Maximum number of states in determinized FST before it will abort."); po.Read(argc, argv); if (po.NumArgs() > 2) { po.PrintUsage(); exit(1); } std::string fst_in_str = po.GetOptArg(1), fst_out_str = po.GetOptArg(2); // This enables us to get traceback info from determinization that is // not seeming to terminate. #if !defined(_MSC_VER) && !defined(__APPLE__) signal(SIGUSR1, signal_handler); #endif if (ClassifyRspecifier(fst_in_str, NULL, NULL) == kNoRspecifier) { // Normal case: just files. VectorFst<StdArc> *fst = ReadFstKaldi(fst_in_str); ArcSort(fst, ILabelCompare<StdArc>()); // improves speed. if (use_log) { DeterminizeStarInLog(fst, delta, &debug_location, max_states); } else { VectorFst<StdArc> det_fst; DeterminizeStar(*fst, &det_fst, delta, &debug_location, max_states); *fst = det_fst; // will do shallow copy and then det_fst goes // out of scope anyway. } WriteFstKaldi(*fst, fst_out_str); delete fst; } else { // Dealing with archives. SequentialTableReader<VectorFstHolder> fst_reader(fst_in_str); TableWriter<VectorFstHolder> fst_writer(fst_out_str); for (; !fst_reader.Done(); fst_reader.Next()) { std::string key = fst_reader.Key(); VectorFst<StdArc> fst(fst_reader.Value()); fst_reader.FreeCurrent(); ArcSort(&fst, ILabelCompare<StdArc>()); // improves speed. try { if (use_log) { DeterminizeStarInLog(&fst, delta, &debug_location, max_states); } else { VectorFst<StdArc> det_fst; DeterminizeStar(fst, &det_fst, delta, &debug_location, max_states); fst = det_fst; // will do shallow copy and then det_fst goes out // of scope anyway. } fst_writer.Write(key, fst); } catch (const std::runtime_error e) { KALDI_WARN << "Error during determinization for key " << key; } } } return 0; } catch(const std::exception &e) { std::cerr << e.what(); return -1; } } |