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#include <fst/script/fst-class.h>
18#include <fst/script/script-impl.h>
19#include <fst/script/replace.h>
20
21namespace fst {
22namespace script {
23
24void Replace(const vector<pair<int64, const FstClass *> > &tuples,
25             MutableFstClass *ofst, const int64 &root,
26             bool epsilon_on_replace) {
27  for (unsigned i = 0; i < tuples.size() - 1; ++i) {
28    if (!ArcTypesMatch(*tuples[i].second, *tuples[i+1].second, "Replace")) {
29      return;
30    }
31  }
32
33  if (!ArcTypesMatch(*tuples[0].second, *ofst, "Replace")) return;
34
35  ReplaceArgs args(tuples, ofst, root, epsilon_on_replace);
36
37  Apply<Operation<ReplaceArgs> >("Replace", ofst->ArcType(), &args);
38}
39
40REGISTER_FST_OPERATION(Replace, StdArc, ReplaceArgs);
41REGISTER_FST_OPERATION(Replace, LogArc, ReplaceArgs);
42REGISTER_FST_OPERATION(Replace, Log64Arc, ReplaceArgs);
43
44}  // namespace script
45}  // namespace fst
46