1
2 // Licensed under the Apache License, Version 2.0 (the "License");
3 // you may not use this file except in compliance with the License.
4 // You may obtain a copy of the License at
5 //
6 // http://www.apache.org/licenses/LICENSE-2.0
7 //
8 // Unless required by applicable law or agreed to in writing, software
9 // distributed under the License is distributed on an "AS IS" BASIS,
10 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11 // See the License for the specific language governing permissions and
12 // limitations under the License.
13 //
14 // Copyright 2005-2010 Google, Inc.
15 // Author: jpr@google.com (Jake Ratkiewicz)
16
17 #ifndef FST_SCRIPT_PUSH_H_
18 #define FST_SCRIPT_PUSH_H_
19
20 #include <fst/script/arg-packs.h>
21 #include <fst/script/fst-class.h>
22 #include <fst/push.h>
23
24 namespace fst {
25 namespace script {
26
27 // 1
28 typedef args::Package<MutableFstClass*, ReweightType, float, bool> PushArgs1;
29
30 template<class Arc>
Push(PushArgs1 * args)31 void Push(PushArgs1 *args) {
32 MutableFst<Arc> *ofst = args->arg1->GetMutableFst<Arc>();
33
34 if (args->arg2 == REWEIGHT_TO_FINAL) {
35 fst::Push(ofst, REWEIGHT_TO_FINAL, args->arg3, args->arg4);
36 } else {
37 fst::Push(ofst, REWEIGHT_TO_INITIAL, args->arg3, args->arg4);
38 }
39 }
40
41 // 2
42 typedef args::Package<const FstClass &, MutableFstClass *, uint32,
43 ReweightType, float> PushArgs2;
44
45 template<class Arc>
Push(PushArgs2 * args)46 void Push(PushArgs2 *args) {
47 const Fst<Arc> &ifst = *(args->arg1.GetFst<Arc>());
48 MutableFst<Arc> *ofst = args->arg2->GetMutableFst<Arc>();
49
50 if (args->arg4 == REWEIGHT_TO_FINAL) {
51 fst::Push<Arc, REWEIGHT_TO_FINAL>(ifst, ofst, args->arg3, args->arg5);
52 } else {
53 fst::Push<Arc, REWEIGHT_TO_INITIAL>(ifst, ofst, args->arg3, args->arg5);
54 }
55 }
56
57 // 1
58 void Push(MutableFstClass *ofst, ReweightType type, float delta = kDelta,
59 bool remove_total_weight = false);
60
61 // 2
62 void Push(const FstClass &ifst, MutableFstClass *ofst, uint32 flags,
63 ReweightType dir, float delta);
64
65 } // namespace script
66 } // namespace fst
67
68
69
70 #endif // FST_SCRIPT_PUSH_H_
71