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_RANDEQUIVALENT_H_
18#define FST_SCRIPT_RANDEQUIVALENT_H_
19
20#include <fst/script/arg-packs.h>
21#include <fst/script/fst-class.h>
22#include <fst/script/randgen.h>  // for RandArcSelection
23#include <fst/randequivalent.h>
24
25namespace fst {
26namespace script {
27
28// 1
29typedef args::Package<const FstClass&, const FstClass&,
30                      int32, float, int, int> RandEquivalentInnerArgs1;
31typedef args::WithReturnValue<bool,
32                              RandEquivalentInnerArgs1> RandEquivalentArgs1;
33
34template<class Arc>
35void RandEquivalent(RandEquivalentArgs1 *args) {
36  const Fst<Arc> &fst1 = *(args->args.arg1.GetFst<Arc>());
37  const Fst<Arc> &fst2 = *(args->args.arg2.GetFst<Arc>());
38
39  args->retval = RandEquivalent(fst1, fst2, args->args.arg3, args->args.arg4,
40                                args->args.arg5, args->args.arg6);
41}
42
43// 2
44typedef args::Package<const FstClass &, const FstClass &, int32,
45                      ssize_t, float,
46                      const RandGenOptions<RandArcSelection> &>
47  RandEquivalentInnerArgs2;
48
49typedef args::WithReturnValue<bool,
50                              RandEquivalentInnerArgs2> RandEquivalentArgs2;
51
52template<class Arc>
53void RandEquivalent(RandEquivalentArgs2 *args) {
54  const Fst<Arc> &fst1 = *(args->args.arg1.GetFst<Arc>());
55  const Fst<Arc> &fst2 = *(args->args.arg2.GetFst<Arc>());
56  const RandGenOptions<RandArcSelection> &opts = args->args.arg6;
57  int32 seed = args->args.arg3;
58
59  if (opts.arc_selector == UNIFORM_ARC_SELECTOR) {
60    UniformArcSelector<Arc> arc_selector(seed);
61    RandGenOptions< UniformArcSelector<Arc> >
62        ropts(arc_selector, opts.max_length, opts.npath);
63
64    args->retval = RandEquivalent(fst1, fst2, args->args.arg4,
65                                  args->args.arg5, ropts);
66  } else if (opts.arc_selector == FAST_LOG_PROB_ARC_SELECTOR) {
67    FastLogProbArcSelector<Arc> arc_selector(seed);
68    RandGenOptions< FastLogProbArcSelector<Arc> >
69        ropts(arc_selector, opts.max_length, opts.npath);
70
71    args->retval = RandEquivalent(fst1, fst2, args->args.arg4,
72                                  args->args.arg5, ropts);
73  } else {
74    LogProbArcSelector<Arc> arc_selector(seed);
75    RandGenOptions< LogProbArcSelector<Arc> >
76        ropts(arc_selector, opts.max_length, opts.npath);
77    args->retval = RandEquivalent(fst1, fst2, args->args.arg4,
78                                  args->args.arg5, ropts);
79  }
80}
81
82
83// 1
84bool RandEquivalent(const FstClass &fst1,
85                    const FstClass &fst2,
86                    int32 seed = time(0),
87                    ssize_t num_paths = 1,
88                    float delta = fst::kDelta,
89                    int path_length = INT_MAX);
90
91// 2
92bool RandEquivalent(const FstClass &fst1,
93                    const FstClass &fst2,
94                    int32 seed,
95                    ssize_t num_paths,
96                    float delta,
97                    const fst::RandGenOptions<
98                      fst::script::RandArcSelection> &opts);
99
100}  // namespace script
101}  // namespace fst
102
103
104
105#endif  // FST_SCRIPT_RANDEQUIVALENT_H_
106