Support reading FST from stream in text normalizer (#57)

k2-fsa · Nov 5, 2023 · ed4259c · ed4259c
1 parent 1270c62
commit ed4259c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/kaldifst/csrc/text-normalizer.cc b/kaldifst/csrc/text-normalizer.cc
@@ -90,6 +90,15 @@ TextNormalizer::TextNormalizer(const std::string &rule) {
  CastOrConvertToConstFst(fst::ReadFstKaldiGeneric(rule)));
 }
 
+TextNormalizer::TextNormalizer(std::istream &is) {
+ fst::StdVectorFst *fst = new fst::StdVectorFst;
+ bool binary = true;
+ ReadFstKaldi(is, binary, fst);
+
+ // fst is released inside CastOrConvertToConstFst()
+ rule_ = std::unique_ptr<fst::StdConstFst>(CastOrConvertToConstFst(fst));
+}
+
 TextNormalizer::TextNormalizer(std::unique_ptr<fst::StdConstFst> rule)
  : rule_(std::move(rule)) {}
 

diff --git a/kaldifst/csrc/text-normalizer.h b/kaldifst/csrc/text-normalizer.h
@@ -17,6 +17,7 @@ class TextNormalizer {
  public:
  // Path to rule.fst
  explicit TextNormalizer(const std::string &rule);
+ explicit TextNormalizer(std::istream &is);
 
  explicit TextNormalizer(std::unique_ptr<fst::StdConstFst> rule);