| |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| // |
| // Copyright 2005-2010 Google, Inc. |
| // Author: jpr@google.com (Jake Ratkiewicz) |
| |
| #ifndef FST_SCRIPT_RANDEQUIVALENT_H_ |
| #define FST_SCRIPT_RANDEQUIVALENT_H_ |
| |
| #include <fst/script/arg-packs.h> |
| #include <fst/script/fst-class.h> |
| #include <fst/script/randgen.h> // for RandArcSelection |
| #include <fst/randequivalent.h> |
| |
| namespace fst { |
| namespace script { |
| |
| // 1 |
| typedef args::Package<const FstClass&, const FstClass&, |
| int32, float, int, int> RandEquivalentInnerArgs1; |
| typedef args::WithReturnValue<bool, |
| RandEquivalentInnerArgs1> RandEquivalentArgs1; |
| |
| template<class Arc> |
| void RandEquivalent(RandEquivalentArgs1 *args) { |
| const Fst<Arc> &fst1 = *(args->args.arg1.GetFst<Arc>()); |
| const Fst<Arc> &fst2 = *(args->args.arg2.GetFst<Arc>()); |
| |
| args->retval = RandEquivalent(fst1, fst2, args->args.arg3, args->args.arg4, |
| args->args.arg5, args->args.arg6); |
| } |
| |
| // 2 |
| typedef args::Package<const FstClass &, const FstClass &, int32, |
| ssize_t, float, |
| const RandGenOptions<RandArcSelection> &> |
| RandEquivalentInnerArgs2; |
| |
| typedef args::WithReturnValue<bool, |
| RandEquivalentInnerArgs2> RandEquivalentArgs2; |
| |
| template<class Arc> |
| void RandEquivalent(RandEquivalentArgs2 *args) { |
| const Fst<Arc> &fst1 = *(args->args.arg1.GetFst<Arc>()); |
| const Fst<Arc> &fst2 = *(args->args.arg2.GetFst<Arc>()); |
| const RandGenOptions<RandArcSelection> &opts = args->args.arg6; |
| int32 seed = args->args.arg3; |
| |
| if (opts.arc_selector == UNIFORM_ARC_SELECTOR) { |
| UniformArcSelector<Arc> arc_selector(seed); |
| RandGenOptions< UniformArcSelector<Arc> > |
| ropts(arc_selector, opts.max_length, opts.npath); |
| |
| args->retval = RandEquivalent(fst1, fst2, args->args.arg4, |
| args->args.arg5, ropts); |
| } else if (opts.arc_selector == FAST_LOG_PROB_ARC_SELECTOR) { |
| FastLogProbArcSelector<Arc> arc_selector(seed); |
| RandGenOptions< FastLogProbArcSelector<Arc> > |
| ropts(arc_selector, opts.max_length, opts.npath); |
| |
| args->retval = RandEquivalent(fst1, fst2, args->args.arg4, |
| args->args.arg5, ropts); |
| } else { |
| LogProbArcSelector<Arc> arc_selector(seed); |
| RandGenOptions< LogProbArcSelector<Arc> > |
| ropts(arc_selector, opts.max_length, opts.npath); |
| args->retval = RandEquivalent(fst1, fst2, args->args.arg4, |
| args->args.arg5, ropts); |
| } |
| } |
| |
| |
| // 1 |
| bool RandEquivalent(const FstClass &fst1, |
| const FstClass &fst2, |
| int32 seed = time(0), |
| ssize_t num_paths = 1, |
| float delta = fst::kDelta, |
| int path_length = INT_MAX); |
| |
| // 2 |
| bool RandEquivalent(const FstClass &fst1, |
| const FstClass &fst2, |
| int32 seed, |
| ssize_t num_paths, |
| float delta, |
| const fst::RandGenOptions< |
| fst::script::RandArcSelection> &opts); |
| |
| } // namespace script |
| } // namespace fst |
| |
| |
| |
| #endif // FST_SCRIPT_RANDEQUIVALENT_H_ |