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 (at) google.com (Jake Ratkiewicz) 16 17 #ifndef FST_SCRIPT_PUSH_H_ 18 #define FST_SCRIPT_PUSH_H_ 19 20 #include <fst/script/arg-packs.h> 21 #include <fst/script/fst-class.h> 22 #include <fst/push.h> 23 24 namespace fst { 25 namespace script { 26 27 // 1 28 typedef args::Package<MutableFstClass*, ReweightType, float, bool> PushArgs1; 29 30 template<class Arc> 31 void Push(PushArgs1 *args) { 32 MutableFst<Arc> *ofst = args->arg1->GetMutableFst<Arc>(); 33 34 if (args->arg2 == REWEIGHT_TO_FINAL) { 35 fst::Push(ofst, REWEIGHT_TO_FINAL, args->arg3, args->arg4); 36 } else { 37 fst::Push(ofst, REWEIGHT_TO_INITIAL, args->arg3, args->arg4); 38 } 39 } 40 41 // 2 42 typedef args::Package<const FstClass &, MutableFstClass *, uint32, 43 ReweightType, float> PushArgs2; 44 45 template<class Arc> 46 void Push(PushArgs2 *args) { 47 const Fst<Arc> &ifst = *(args->arg1.GetFst<Arc>()); 48 MutableFst<Arc> *ofst = args->arg2->GetMutableFst<Arc>(); 49 50 if (args->arg4 == REWEIGHT_TO_FINAL) { 51 fst::Push<Arc, REWEIGHT_TO_FINAL>(ifst, ofst, args->arg3, args->arg5); 52 } else { 53 fst::Push<Arc, REWEIGHT_TO_INITIAL>(ifst, ofst, args->arg3, args->arg5); 54 } 55 } 56 57 // 1 58 void Push(MutableFstClass *ofst, ReweightType type, float delta = kDelta, 59 bool remove_total_weight = false); 60 61 // 2 62 void Push(const FstClass &ifst, MutableFstClass *ofst, uint32 flags, 63 ReweightType dir, float delta); 64 65 } // namespace script 66 } // namespace fst 67 68 69 70 #endif // FST_SCRIPT_PUSH_H_ 71
