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_ARCSORT_H_ 18 #define FST_SCRIPT_ARCSORT_H_ 19 20 #include <fst/arcsort.h> 21 #include <fst/script/arg-packs.h> 22 #include <fst/script/fst-class.h> 23 24 namespace fst { 25 namespace script { 26 27 enum ArcSortType { ILABEL_COMPARE, OLABEL_COMPARE }; 28 29 typedef args::Package<MutableFstClass*, const ArcSortType> ArcSortArgs; 30 31 template<class Arc> 32 void ArcSort(ArcSortArgs *args) { 33 MutableFst<Arc> *fst = args->arg1->GetMutableFst<Arc>(); 34 35 if (args->arg2 == ILABEL_COMPARE) { 36 ILabelCompare<Arc> icomp; 37 ArcSort(fst, icomp); 38 } else { // OLABEL_COMPARE 39 OLabelCompare<Arc> ocomp; 40 ArcSort(fst, ocomp); 41 } 42 } 43 44 void ArcSort(MutableFstClass *ofst, ArcSortType sort_type); 45 46 } // namespace script 47 } // namespace fst 48 49 #endif // FST_SCRIPT_ARCSORT_H_ 50