aboutsummaryrefslogtreecommitdiff
path: root/src/bin/fstencode.cc
diff options
context:
space:
mode:
authorIan Hodson <idh@google.com>2012-05-30 21:27:06 +0100
committerIan Hodson <idh@google.com>2012-05-30 22:47:36 +0100
commitf4c12fce1ee58e670f9c3fce46c40296ba9ee8a2 (patch)
treeb131ed907f9b2d5af09c0983b651e9e69bc6aab9 /src/bin/fstencode.cc
parenta92766f0a6ba4fac46cd6fd3856ef20c3b204f0d (diff)
downloadopenfst-f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2.tar.gz
Moved from GoogleTTS Change-Id: I6bc6bdadaa53bd0f810b88443339f6d899502cc8
Diffstat (limited to 'src/bin/fstencode.cc')
-rw-r--r--src/bin/fstencode.cc68
1 files changed, 68 insertions, 0 deletions
diff --git a/src/bin/fstencode.cc b/src/bin/fstencode.cc
new file mode 100644
index 0000000..95dd49c
--- /dev/null
+++ b/src/bin/fstencode.cc
@@ -0,0 +1,68 @@
+// fstencode.cc
+
+// 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
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
+// Copyright 2005-2010 Google, Inc.
+// Author: riley@google.com (Michael Riley)
+// Modified: jpr@google.com (Jake Ratkiewicz) to use FstClass
+//
+// \file
+// Encode transducer labels and/or weights.
+//
+
+#include <fst/script/encode.h>
+#include <fst/script/decode.h>
+
+/// EncodeMain specific flag definitions
+DEFINE_bool(encode_labels, false, "Encode output labels");
+DEFINE_bool(encode_weights, false, "Encode weights");
+DEFINE_bool(encode_reuse, false, "Re-use existing codex");
+DEFINE_bool(decode, false, "Decode labels and/or weights");
+
+int main(int argc, char **argv) {
+ namespace s = fst::script;
+ using fst::script::FstClass;
+ using fst::script::MutableFstClass;
+
+ string usage = "Encodes transducer labels and/or weights.\n\n Usage: ";
+ usage += argv[0];
+ usage += " in.fst codex [out.fst]\n";
+
+ std::set_new_handler(FailedNewHandler);
+ SetFlags(usage.c_str(), &argc, &argv, true);
+ if (argc < 3 || argc > 4) {
+ ShowUsage();
+ return 1;
+ }
+
+ string in_name = (strcmp(argv[1], "-") != 0) ? argv[1] : "";
+ string codex_name = argv[2];
+ string out_name = argc > 3 ? argv[3] : "";
+
+ MutableFstClass *fst = MutableFstClass::Read(in_name, true);
+ if (!fst) return 1;
+
+ if (FLAGS_decode == false) {
+ uint32 flags = 0;
+ flags |= FLAGS_encode_labels ? fst::kEncodeLabels : 0;
+ flags |= FLAGS_encode_weights ? fst::kEncodeWeights : 0;
+ s::Encode(fst, flags, FLAGS_encode_reuse, codex_name);
+ fst->Write(out_name);
+ } else {
+ s::Decode(fst, codex_name);
+ fst->Write(out_name);
+ }
+
+ delete fst;
+ return 0;
+}