Home | History | Annotate | Download | only in script
      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