// online2/onlinebin-util.cc // Copyright 2012 Cisco Systems (author: Matthias Paulik) // Modifications to the original contribution by Cisco Systems made by: // Vassil Panayotov // 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 "online2/onlinebin-util.h" namespace kaldi { fst::Fst *ReadDecodeGraph(std::string filename) { // read decoding network FST Input ki(filename); // use ki.Stream() instead of is. if (!ki.Stream().good()) KALDI_ERR << "Could not open decoding-graph FST " << filename; fst::FstHeader hdr; if (!hdr.Read(ki.Stream(), "")) { KALDI_ERR << "Reading FST: error reading FST header."; } if (hdr.ArcType() != fst::StdArc::Type()) { KALDI_ERR << "FST with arc type " << hdr.ArcType() << " not supported.\n"; } fst::FstReadOptions ropts("", &hdr); fst::Fst *decode_fst = NULL; if (hdr.FstType() == "vector") { decode_fst = fst::VectorFst::Read(ki.Stream(), ropts); } else if (hdr.FstType() == "const") { decode_fst = fst::ConstFst::Read(ki.Stream(), ropts); } else { KALDI_ERR << "Reading FST: unsupported FST type: " << hdr.FstType(); } if (decode_fst == NULL) { // fst code will warn. KALDI_ERR << "Error reading FST (after reading header)."; return NULL; } else { return decode_fst; } } void PrintPartialResult(const std::vector& words, const fst::SymbolTable *word_syms, bool line_break) { KALDI_ASSERT(word_syms != NULL); for (size_t i = 0; i < words.size(); i++) { std::string word = word_syms->Find(words[i]); if (word == "") KALDI_ERR << "Word-id " << words[i] <<" not in symbol table."; std::cout << word << ' '; } if (line_break) std::cout << "\n\n"; else std::cout.flush(); } } // namespace kaldi