compile.h
3.48 KB
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_