Blame view
tools/openfst-1.6.7/src/bin/fstreplace-main.cc
2.72 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 |
// See www.openfst.org for extensive documentation on this weighted
// finite-state transducer library.
//
// Performs the dynamic replacement of arcs in one FST with another FST,
// allowing for the definition of FSTs analogous to RTNs.
#include <cstring>
#include <string>
#include <vector>
#include <fst/flags.h>
#include <fst/script/getters.h>
#include <fst/script/replace.h>
DECLARE_string(call_arc_labeling);
DECLARE_string(return_arc_labeling);
DECLARE_int64(return_label);
DECLARE_bool(epsilon_on_replace);
void Cleanup(std::vector<fst::script::LabelFstClassPair> *pairs) {
for (const auto &pair : *pairs) {
delete pair.second;
}
pairs->clear();
}
int fstreplace_main(int argc, char **argv) {
namespace s = fst::script;
using fst::script::FstClass;
using fst::script::VectorFstClass;
using fst::ReplaceLabelType;
string usage = "Recursively replaces FST arcs with other FST(s).
"
" Usage: ";
usage += argv[0];
usage += " root.fst rootlabel [rule1.fst label1 ...] [out.fst]
";
std::set_new_handler(FailedNewHandler);
SET_FLAGS(usage.c_str(), &argc, &argv, true);
if (argc < 4) {
ShowUsage();
return 1;
}
const string in_name = argv[1];
const string out_name = argc % 2 == 0 ? argv[argc - 1] : "";
auto *ifst = FstClass::Read(in_name);
if (!ifst) return 1;
std::vector<s::LabelFstClassPair> pairs;
// Note that if the root label is beyond the range of the underlying FST's
// labels, truncation will occur.
const auto root = atoll(argv[2]);
pairs.emplace_back(root, ifst);
for (auto i = 3; i < argc - 1; i += 2) {
ifst = FstClass::Read(argv[i]);
if (!ifst) {
Cleanup(&pairs);
return 1;
}
// Note that if the root label is beyond the range of the underlying FST's
// labels, truncation will occur.
const auto label = atoll(argv[i + 1]);
pairs.emplace_back(label, ifst);
}
ReplaceLabelType call_label_type;
if (!s::GetReplaceLabelType(FLAGS_call_arc_labeling, FLAGS_epsilon_on_replace,
&call_label_type)) {
LOG(ERROR) << argv[0] << ": Unknown or unsupported call arc replace "
<< "label type: " << FLAGS_call_arc_labeling;
}
ReplaceLabelType return_label_type;
if (!s::GetReplaceLabelType(FLAGS_return_arc_labeling,
FLAGS_epsilon_on_replace, &return_label_type)) {
LOG(ERROR) << argv[0] << ": Unknown or unsupported return arc replace "
<< "label type: " << FLAGS_return_arc_labeling;
}
s::ReplaceOptions opts(root, call_label_type, return_label_type,
FLAGS_return_label);
VectorFstClass ofst(ifst->ArcType());
s::Replace(pairs, &ofst, opts);
Cleanup(&pairs);
return !ofst.Write(out_name);
}
|