Blame view

tools/openfst-1.6.7/include/fst/script/print.h 2.54 KB
8dcb6dfcb   Yannick Estève   first commit
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
  // See www.openfst.org for extensive documentation on this weighted
  // finite-state transducer library.
  
  #ifndef FST_SCRIPT_PRINT_H_
  #define FST_SCRIPT_PRINT_H_
  
  #include <ostream>
  
  #include <fst/flags.h>
  #include <fst/script/fst-class.h>
  #include <fst/script/print-impl.h>
  
  DECLARE_string(fst_field_separator);
  
  namespace fst {
  namespace script {
  
  // Note: it is safe to pass these strings as references because
  // this struct is only used to pass them deeper in the call graph.
  // Be sure you understand why this is so before using this struct
  // for anything else!
  struct FstPrinterArgs {
    const FstClass &fst;
    const SymbolTable *isyms;
    const SymbolTable *osyms;
    const SymbolTable *ssyms;
    const bool accept;
    const bool show_weight_one;
    std::ostream *ostrm;
    const string &dest;
    const string &sep;  // NOLINT
    const string &missing_symbol;
  
    FstPrinterArgs(const FstClass &fst, const SymbolTable *isyms,
                   const SymbolTable *osyms, const SymbolTable *ssyms,
                   bool accept, bool show_weight_one, std::ostream *ostrm,
                   const string &dest, const string &sep,
                   const string &missing_sym = "")
        : fst(fst),
          isyms(isyms),
          osyms(osyms),
          ssyms(ssyms),
          accept(accept),
          show_weight_one(show_weight_one),
          ostrm(ostrm),
          dest(dest),
          sep(sep),
          missing_symbol(missing_sym) {}
  };
  
  template <class Arc>
  void PrintFst(FstPrinterArgs *args) {
    const Fst<Arc> &fst = *(args->fst.GetFst<Arc>());
    FstPrinter<Arc> fstprinter(fst, args->isyms, args->osyms, args->ssyms,
                               args->accept, args->show_weight_one, args->sep,
                               args->missing_symbol);
    fstprinter.Print(args->ostrm, args->dest);
  }
  
  void PrintFst(const FstClass &fst, std::ostream &ostrm, const string &dest,
                const SymbolTable *isyms, const SymbolTable *osyms,
                const SymbolTable *ssyms, bool accept, bool show_weight_one,
                const string &missing_sym = "");
  
  // The same, but with more sensible defaults.
  template <class Arc>
  void PrintFst(const Fst<Arc> &fst, std::ostream &ostrm, const string &dest = "",
                const SymbolTable *isyms = nullptr,
                const SymbolTable *osyms = nullptr,
                const SymbolTable *ssyms = nullptr) {
    const string sep = FLAGS_fst_field_separator.substr(0, 1);
    FstPrinter<Arc> fstprinter(fst, isyms, osyms, ssyms, true, true, sep);
    fstprinter.Print(&ostrm, dest);
  }
  
  }  // namespace script
  }  // namespace fst
  
  #endif  // FST_SCRIPT_PRINT_H_