1f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// fstunion.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// Creates the union of two FSTs. 21f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// 22f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 23f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson#include <string> 24f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 25f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson#include <fst/script/union.h> 26f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson#include <iostream> 27f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson#include <fstream> 28f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 29f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodsonint main(int argc, char **argv) { 30f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson using fst::script::FstClass; 31f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson using fst::script::MutableFstClass; 32f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson using fst::script::Union; 33f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 34f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson string usage = "Creates the union of two FSTs.\n\n Usage: "; 35f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson usage += argv[0]; 36f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson usage += " in1.fst in2.fst [out.fst]\n"; 37f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 38f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson std::set_new_handler(FailedNewHandler); 39f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson SetFlags(usage.c_str(), &argc, &argv, true); 40f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson if (argc < 3 || argc > 4) { 41f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson ShowUsage(); 42f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson return 1; 43f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson } 44f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 45f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson string in1_name = strcmp(argv[1], "-") != 0 ? argv[1] : ""; 46f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson string in2_name = strcmp(argv[2], "-") != 0 ? argv[2] : ""; 47f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson string out_name = argc > 3 ? argv[3] : ""; 48f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 49f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson if (in1_name == "" && in2_name == "") { 50f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson LOG(ERROR) << argv[0] 51f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson << ": Can't use standard i/o for both inputs."; 52f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson return 1; 53f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson } 54f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 55f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson MutableFstClass *fst1 = MutableFstClass::Read(in1_name, true); 56f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson if (!fst1) return 1; 57f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 58f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson FstClass *fst2 = FstClass::Read(in2_name); 59f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson if (!fst2) { 60f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson return 1; 61f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson } 62f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 63f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson Union(fst1, *fst2); 64f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson fst1->Write(out_name); 65f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 66f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson return 0; 67f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson} 68