map.h
4.54 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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
// See www.openfst.org for extensive documentation on this weighted
// finite-state transducer library.
#ifndef FST_SCRIPT_MAP_H_
#define FST_SCRIPT_MAP_H_
#include <memory>
#include <tuple>
#include <fst/arc-map.h>
#include <fst/state-map.h>
#include <fst/script/arg-packs.h>
#include <fst/script/fst-class.h>
#include <fst/script/weight-class.h>
namespace fst {
namespace script {
template <class M>
Fst<typename M::ToArc> *ArcMap(const Fst<typename M::FromArc> &fst,
const M &mapper) {
using ToArc = typename M::ToArc;
auto *ofst = new VectorFst<ToArc>;
ArcMap(fst, ofst, mapper);
return ofst;
}
template <class M>
Fst<typename M::ToArc> *StateMap(const Fst<typename M::FromArc> &fst,
const M &mapper) {
using ToArc = typename M::ToArc;
auto *ofst = new VectorFst<ToArc>;
StateMap(fst, ofst, mapper);
return ofst;
}
enum MapType {
ARC_SUM_MAPPER,
ARC_UNIQUE_MAPPER,
IDENTITY_MAPPER,
INPUT_EPSILON_MAPPER,
INVERT_MAPPER,
OUTPUT_EPSILON_MAPPER,
PLUS_MAPPER,
POWER_MAPPER,
QUANTIZE_MAPPER,
RMWEIGHT_MAPPER,
SUPERFINAL_MAPPER,
TIMES_MAPPER,
TO_LOG_MAPPER,
TO_LOG64_MAPPER,
TO_STD_MAPPER
};
using MapInnerArgs =
std::tuple<const FstClass &, MapType, float, double, const WeightClass &>;
using MapArgs = WithReturnValue<FstClass *, MapInnerArgs>;
template <class Arc>
void Map(MapArgs *args) {
using Weight = typename Arc::Weight;
const Fst<Arc> &ifst = *(std::get<0>(args->args).GetFst<Arc>());
const auto map_type = std::get<1>(args->args);
switch (map_type) {
case ARC_SUM_MAPPER: {
std::unique_ptr<Fst<Arc>> ofst(StateMap(ifst, ArcSumMapper<Arc>(ifst)));
args->retval = new FstClass(*ofst);
return;
}
case ARC_UNIQUE_MAPPER: {
std::unique_ptr<Fst<Arc>> ofst(
StateMap(ifst, ArcUniqueMapper<Arc>(ifst)));
args->retval = new FstClass(*ofst);
return;
}
case IDENTITY_MAPPER: {
std::unique_ptr<Fst<Arc>> ofst(ArcMap(ifst, IdentityArcMapper<Arc>()));
args->retval = new FstClass(*ofst);
return;
}
case INPUT_EPSILON_MAPPER: {
std::unique_ptr<Fst<Arc>> ofst(ArcMap(ifst, InputEpsilonMapper<Arc>()));
args->retval = new FstClass(*ofst);
return;
}
case INVERT_MAPPER: {
std::unique_ptr<Fst<Arc>> ofst(ArcMap(ifst, InvertWeightMapper<Arc>()));
args->retval = new FstClass(*ofst);
return;
}
case OUTPUT_EPSILON_MAPPER: {
std::unique_ptr<Fst<Arc>> ofst(ArcMap(ifst, OutputEpsilonMapper<Arc>()));
args->retval = new FstClass(*ofst);
return;
}
case PLUS_MAPPER: {
const auto weight = *(std::get<4>(args->args).GetWeight<Weight>());
std::unique_ptr<Fst<Arc>> ofst(ArcMap(ifst, PlusMapper<Arc>(weight)));
args->retval = new FstClass(*ofst);
return;
}
case POWER_MAPPER: {
const auto power = std::get<3>(args->args);
std::unique_ptr<Fst<Arc>> ofst(ArcMap(ifst, PowerMapper<Arc>(power)));
args->retval = new FstClass(*ofst);
return;
}
case QUANTIZE_MAPPER: {
const auto delta = std::get<2>(args->args);
std::unique_ptr<Fst<Arc>> ofst(ArcMap(ifst, QuantizeMapper<Arc>(delta)));
args->retval = new FstClass(*ofst);
return;
}
case RMWEIGHT_MAPPER: {
std::unique_ptr<Fst<Arc>> ofst(ArcMap(ifst, RmWeightMapper<Arc>()));
args->retval = new FstClass(*ofst);
return;
}
case SUPERFINAL_MAPPER: {
std::unique_ptr<Fst<Arc>> ofst(ArcMap(ifst, SuperFinalMapper<Arc>()));
args->retval = new FstClass(*ofst);
return;
}
case TIMES_MAPPER: {
const auto weight = *(std::get<4>(args->args).GetWeight<Weight>());
std::unique_ptr<Fst<Arc>> ofst(ArcMap(ifst, TimesMapper<Arc>(weight)));
args->retval = new FstClass(*ofst);
return;
}
case TO_LOG_MAPPER: {
std::unique_ptr<Fst<LogArc>> ofst(
ArcMap(ifst, WeightConvertMapper<Arc, LogArc>()));
args->retval = new FstClass(*ofst);
return;
}
case TO_LOG64_MAPPER: {
std::unique_ptr<Fst<Log64Arc>> ofst(
ArcMap(ifst, WeightConvertMapper<Arc, Log64Arc>()));
args->retval = new FstClass(*ofst);
return;
}
case TO_STD_MAPPER: {
std::unique_ptr<Fst<StdArc>> ofst(
ArcMap(ifst, WeightConvertMapper<Arc, StdArc>()));
args->retval = new FstClass(*ofst);
return;
}
}
}
FstClass *Map(const FstClass &ifst, MapType map_type, float delta, double power,
const WeightClass &weight);
} // namespace script
} // namespace fst
#endif // FST_SCRIPT_MAP_H_