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.

64 lines
1.9 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_UNION_H_
  18. #define FST_SCRIPT_UNION_H_
  19. #include <tuple>
  20. #include <utility>
  21. #include <vector>
  22. #include <fst/fst.h>
  23. #include <fst/mutable-fst.h>
  24. #include <fst/union.h>
  25. #include <fst/script/fst-class.h>
  26. namespace fst {
  27. namespace script {
  28. using FstUnionArgs1 = std::pair<MutableFstClass *, const FstClass &>;
  29. template <class Arc>
  30. void Union(FstUnionArgs1 *args) {
  31. MutableFst<Arc> *fst1 = std::get<0>(*args)->GetMutableFst<Arc>();
  32. const Fst<Arc> &fst2 = *std::get<1>(*args).GetFst<Arc>();
  33. Union(fst1, fst2);
  34. }
  35. using FstUnionArgs2 =
  36. std::tuple<MutableFstClass *, const std::vector<const FstClass *> &>;
  37. template <class Arc>
  38. void Union(FstUnionArgs2 *args) {
  39. MutableFst<Arc> *fst1 = std::get<0>(*args)->GetMutableFst<Arc>();
  40. const auto &untyped_fsts2 = std::get<1>(*args);
  41. std::vector<const Fst<Arc> *> typed_fsts2;
  42. typed_fsts2.reserve(untyped_fsts2.size());
  43. for (const auto &untyped_fst2 : untyped_fsts2) {
  44. typed_fsts2.emplace_back(untyped_fst2->GetFst<Arc>());
  45. }
  46. Union(fst1, typed_fsts2);
  47. }
  48. void Union(MutableFstClass *fst1, const FstClass &fst2);
  49. void Union(MutableFstClass *fst1, const std::vector<const FstClass *> &fsts2);
  50. } // namespace script
  51. } // namespace fst
  52. #endif // FST_SCRIPT_UNION_H_