1f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// fstminimize.cc
2f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson
3f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// Licensed under the Apache License, Version 2.0 (the "License");
4f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// you may not use this file except in compliance with the License.
5f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// You may obtain a copy of the License at
6f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson//
7f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson//     http://www.apache.org/licenses/LICENSE-2.0
8f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson//
9f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// Unless required by applicable law or agreed to in writing, software
10f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// distributed under the License is distributed on an "AS IS" BASIS,
11f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// See the License for the specific language governing permissions and
13f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// limitations under the License.
14f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson//
15f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// Copyright 2005-2010 Google, Inc.
16f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// Author: riley@google.com (Michael Riley)
17f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// Modified: jpr@google.com (Jake Ratkiewicz) to use FstClass
18f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson//
19f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// \file
20f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// Minimizes a deterministic FSA.
21f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson//
22f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson
23f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson#include <fst/script/minimize.h>
24f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson
25f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian HodsonDEFINE_double(delta, fst::kDelta, "Comparison/quantization delta");
26f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson
27f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson
28f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodsonint main(int argc, char **argv) {
29f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  namespace s = fst::script;
30f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  using fst::script::FstClass;
31f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  using fst::script::MutableFstClass;
32f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  using fst::script::VectorFstClass;
33f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson
34f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  string usage = "Minimizes a deterministic FST.\n\n  Usage: ";
35f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  usage += argv[0];
36f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  usage += " [in.fst [out1.fst [out2.fst]]]\n";
37f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson
38f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  std::set_new_handler(FailedNewHandler);
39f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  SetFlags(usage.c_str(), &argc, &argv, true);
40f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  if (argc > 4) {
41f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson    ShowUsage();
42f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson    return 1;
43f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  }
44f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson
45f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  string in_name = (argc > 1 && strcmp(argv[1], "-") != 0) ? argv[1] : "";
46f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  string out1_name = (argc > 2 && strcmp(argv[2], "-") != 0) ? argv[2] : "";
47f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  string out2_name = (argc > 3 && strcmp(argv[3], "-") != 0) ? argv[3] : "";
48f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson
49f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  if (out1_name.empty() && out2_name.empty() && argc > 3) {
50f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson    LOG(ERROR) << "Both outputs can't be standard out.";
51f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson    return 1;
52f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  }
53f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson
54f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  MutableFstClass *fst1 = MutableFstClass::Read(in_name, true);
55f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  if (!fst1) return 1;
56f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson
57f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  MutableFstClass *fst2 = argc > 3 ?
58f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson      new VectorFstClass(fst1->ArcType()) : 0;
59f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson
60f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  s::Minimize(fst1, fst2, FLAGS_delta);
61f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson
62f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  fst1->Write(out1_name);
63f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  if (fst2)
64f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson    fst2->Write(out2_name);
65f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson
66f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson  return 0;
67f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson}
68