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
24namespace fst {
25namespace script {
26
27// 1
28typedef args::Package<MutableFstClass*, ReweightType, float, bool> PushArgs1;
29
30template<class Arc>
31void 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
42typedef args::Package<const FstClass &, MutableFstClass *, uint32,
43                      ReweightType, float> PushArgs2;
44
45template<class Arc>
46void 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
58void Push(MutableFstClass *ofst, ReweightType type, float delta = kDelta,
59          bool remove_total_weight = false);
60
61// 2
62void 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