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: riley@google.com (Jake Ratkiewicz)
16
17#ifndef FST_SCRIPT_DISAMBIGUATE_H_
18#define FST_SCRIPT_DISAMBIGUATE_H_
19
20#include <fst/disambiguate.h>
21#include <fst/script/arg-packs.h>
22#include <fst/script/fst-class.h>
23#include <fst/script/weight-class.h>
24
25namespace fst {
26namespace script {
27
28struct DisambiguateOptions {
29  float delta;
30  WeightClass weight_threshold;
31  int64 state_threshold;
32  int64 subsequential_label;
33
34  explicit DisambiguateOptions(float d = fst::kDelta,
35                              WeightClass w =
36                                fst::script::WeightClass::Zero(),
37                              int64 n = fst::kNoStateId, int64 l = 0)
38      : delta(d), weight_threshold(w), state_threshold(n),
39        subsequential_label(l) {}
40};
41
42typedef args::Package<const FstClass&, MutableFstClass*,
43                      const DisambiguateOptions &> DisambiguateArgs;
44
45template<class Arc>
46void Disambiguate(DisambiguateArgs *args) {
47  const Fst<Arc> &ifst = *(args->arg1.GetFst<Arc>());
48  MutableFst<Arc> *ofst = args->arg2->GetMutableFst<Arc>();
49  const DisambiguateOptions &opts = args->arg3;
50
51  fst::DisambiguateOptions<Arc> detargs;
52  detargs.delta = opts.delta;
53  detargs.weight_threshold =
54      *(opts.weight_threshold.GetWeight<typename Arc::Weight>());
55  detargs.state_threshold = opts.state_threshold;
56  detargs.subsequential_label = opts.subsequential_label;
57
58  Disambiguate(ifst, ofst, detargs);
59}
60
61void Disambiguate(const FstClass &ifst, MutableFstClass *ofst,
62                 const DisambiguateOptions &opts =
63                   fst::script::DisambiguateOptions());
64
65}  // namespace script
66}  // namespace fst
67
68#endif  // FST_SCRIPT_DISAMBIGUATE_H_
69