Blame view
tools/openfst-1.6.7/include/fst/script/compile.h
3.48 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 |
// See www.openfst.org for extensive documentation on this weighted
// finite-state transducer library.
#ifndef FST_SCRIPT_COMPILE_H_
#define FST_SCRIPT_COMPILE_H_
#include <istream>
#include <memory>
#include <fst/script/arg-packs.h>
#include <fst/script/compile-impl.h>
#include <fst/script/fst-class.h>
namespace fst {
namespace script {
// This operation exists in two forms. 1 is a void operation which writes the
// compiled machine to disk; 2 returns an FstClass. I/O should normally be done
// using the binary format for efficiency, so users are STRONGLY ENCOURAGED to
// use 1 or to construct FSTs using the C++ FST mutation operations.
// Note: it is safe to pass these strings as references because
// this struct is only used to pass them deeper in the call graph.
// Be sure you understand why this is so before using this struct
// for anything else!
struct CompileFstInnerArgs {
std::istream &istrm;
const string &source;
const string &fst_type;
const fst::SymbolTable *isyms;
const fst::SymbolTable *osyms;
const fst::SymbolTable *ssyms;
const bool accep;
const bool ikeep;
const bool okeep;
const bool nkeep;
const bool allow_negative_labels;
CompileFstInnerArgs(std::istream &istrm, const string &source,
const string &fst_type, const fst::SymbolTable *isyms,
const fst::SymbolTable *osyms,
const fst::SymbolTable *ssyms, bool accep, bool ikeep,
bool okeep, bool nkeep,
bool allow_negative_labels = false)
: istrm(istrm),
source(source),
fst_type(fst_type),
isyms(isyms),
osyms(osyms),
ssyms(ssyms),
accep(accep),
ikeep(ikeep),
okeep(okeep),
nkeep(nkeep),
allow_negative_labels(allow_negative_labels) {}
};
using CompileFstArgs = WithReturnValue<FstClass *, CompileFstInnerArgs>;
template <class Arc>
void CompileFstInternal(CompileFstArgs *args) {
using fst::Convert;
using fst::Fst;
using fst::FstCompiler;
FstCompiler<Arc> fstcompiler(
args->args.istrm, args->args.source, args->args.isyms, args->args.osyms,
args->args.ssyms, args->args.accep, args->args.ikeep, args->args.okeep,
args->args.nkeep, args->args.allow_negative_labels);
const Fst<Arc> *fst = &fstcompiler.Fst();
std::unique_ptr<const Fst<Arc>> owned_fst;
if (args->args.fst_type != "vector") {
owned_fst.reset(Convert<Arc>(*fst, args->args.fst_type));
if (!owned_fst) {
FSTERROR() << "Failed to convert FST to desired type: "
<< args->args.fst_type;
}
fst = owned_fst.get();
}
args->retval = fst ? new FstClass(*fst) : nullptr;
}
void CompileFst(std::istream &istrm, const string &source, const string &dest,
const string &fst_type, const string &arc_type,
const SymbolTable *isyms, const SymbolTable *osyms,
const SymbolTable *ssyms, bool accep, bool ikeep, bool okeep,
bool nkeep, bool allow_negative_labels);
FstClass *CompileFstInternal(std::istream &istrm, const string &source,
const string &fst_type, const string &arc_type,
const SymbolTable *isyms, const SymbolTable *osyms,
const SymbolTable *ssyms, bool accep, bool ikeep,
bool okeep, bool nkeep,
bool allow_negative_labels);
} // namespace script
} // namespace fst
#endif // FST_SCRIPT_COMPILE_H_
|