1f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 2f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// Licensed under the Apache License, Version 2.0 (the "License"); 3f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// you may not use this file except in compliance with the License. 4f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// You may obtain a copy of the License at 5f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// 6f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// http://www.apache.org/licenses/LICENSE-2.0 7f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// 8f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// Unless required by applicable law or agreed to in writing, software 9f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// distributed under the License is distributed on an "AS IS" BASIS, 10f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 11f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// See the License for the specific language governing permissions and 12f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// limitations under the License. 13f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// 14f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// Copyright 2005-2010 Google, Inc. 15f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// Author: jpr@google.com (Jake Ratkiewicz) 16f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 17f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson#ifndef FST_SCRIPT_PUSH_H_ 18f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson#define FST_SCRIPT_PUSH_H_ 19f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 20f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson#include <fst/script/arg-packs.h> 21f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson#include <fst/script/fst-class.h> 22f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson#include <fst/push.h> 23f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 24f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodsonnamespace fst { 25f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodsonnamespace script { 26f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 27f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// 1 28f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodsontypedef args::Package<MutableFstClass*, ReweightType, float, bool> PushArgs1; 29f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 30f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodsontemplate<class Arc> 31f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodsonvoid Push(PushArgs1 *args) { 32f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson MutableFst<Arc> *ofst = args->arg1->GetMutableFst<Arc>(); 33f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 34f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson if (args->arg2 == REWEIGHT_TO_FINAL) { 35f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson fst::Push(ofst, REWEIGHT_TO_FINAL, args->arg3, args->arg4); 36f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson } else { 37f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson fst::Push(ofst, REWEIGHT_TO_INITIAL, args->arg3, args->arg4); 38f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson } 39f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson} 40f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 41f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// 2 42f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodsontypedef args::Package<const FstClass &, MutableFstClass *, uint32, 43f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson ReweightType, float> PushArgs2; 44f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 45f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodsontemplate<class Arc> 46f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodsonvoid Push(PushArgs2 *args) { 47f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson const Fst<Arc> &ifst = *(args->arg1.GetFst<Arc>()); 48f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson MutableFst<Arc> *ofst = args->arg2->GetMutableFst<Arc>(); 49f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 50f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson if (args->arg4 == REWEIGHT_TO_FINAL) { 51f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson fst::Push<Arc, REWEIGHT_TO_FINAL>(ifst, ofst, args->arg3, args->arg5); 52f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson } else { 53f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson fst::Push<Arc, REWEIGHT_TO_INITIAL>(ifst, ofst, args->arg3, args->arg5); 54f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson } 55f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson} 56f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 57f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// 1 58f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodsonvoid Push(MutableFstClass *ofst, ReweightType type, float delta = kDelta, 59f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson bool remove_total_weight = false); 60f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 61f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson// 2 62f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodsonvoid Push(const FstClass &ifst, MutableFstClass *ofst, uint32 flags, 63f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson ReweightType dir, float delta); 64f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 65f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson} // namespace script 66f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson} // namespace fst 67f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 68f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 69f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson 70f4c12fce1ee58e670f9c3fce46c40296ba9ee8a2Ian Hodson#endif // FST_SCRIPT_PUSH_H_ 71