You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

57 lines
1.5 KiB

  1. // Copyright 2005-2024 Google LLC
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the 'License');
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an 'AS IS' BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. //
  15. // See www.openfst.org for extensive documentation on this weighted
  16. // finite-state transducer library.
  17. #ifndef FST_SCRIPT_ARCSORT_H_
  18. #define FST_SCRIPT_ARCSORT_H_
  19. #include <cstdint>
  20. #include <utility>
  21. #include <fst/arcsort.h>
  22. #include <fst/mutable-fst.h>
  23. #include <fst/script/fst-class.h>
  24. namespace fst {
  25. namespace script {
  26. enum class ArcSortType : uint8_t { ILABEL, OLABEL };
  27. using FstArcSortArgs = std::pair<MutableFstClass *, ArcSortType>;
  28. template <class Arc>
  29. void ArcSort(FstArcSortArgs *args) {
  30. MutableFst<Arc> *fst = std::get<0>(*args)->GetMutableFst<Arc>();
  31. switch (std::get<1>(*args)) {
  32. case ArcSortType::ILABEL: {
  33. const ILabelCompare<Arc> icomp;
  34. ArcSort(fst, icomp);
  35. return;
  36. }
  37. case ArcSortType::OLABEL: {
  38. const OLabelCompare<Arc> ocomp;
  39. ArcSort(fst, ocomp);
  40. return;
  41. }
  42. }
  43. }
  44. void ArcSort(MutableFstClass *ofst, ArcSortType);
  45. } // namespace script
  46. } // namespace fst
  47. #endif // FST_SCRIPT_ARCSORT_H_