提交

kongdeqiang

6 天以前 28ccfbfc51068a663a80764e14074df5edf2b5ba

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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
// See www.openfst.org for extensive documentation on this weighted
// finite-state transducer library.
//
// Function objects to restrict which arcs are traversed in an FST.
 
#ifndef FST_ARCFILTER_H_
#define FST_ARCFILTER_H_
 
 
#include <fst/fst.h>
#include <fst/util.h>
 
 
namespace fst {
 
// True for all arcs.
template <class Arc>
class AnyArcFilter {
 public:
  bool operator()(const Arc &arc) const { return true; }
};
 
// True for (input/output) epsilon arcs.
template <class Arc>
class EpsilonArcFilter {
 public:
  bool operator()(const Arc &arc) const {
    return arc.ilabel == 0 && arc.olabel == 0;
  }
};
 
// True for input epsilon arcs.
template <class Arc>
class InputEpsilonArcFilter {
 public:
  bool operator()(const Arc &arc) const { return arc.ilabel == 0; }
};
 
// True for output epsilon arcs.
template <class Arc>
class OutputEpsilonArcFilter {
 public:
  bool operator()(const Arc &arc) const { return arc.olabel == 0; }
};
 
// True if specified label matches (doesn't match) when keep_match is
// true (false).
template <class Arc>
class LabelArcFilter {
 public:
  using Label = typename Arc::Label;
 
  explicit LabelArcFilter(Label label, bool match_input = true,
                          bool keep_match = true)
      : label_(label), match_input_(match_input), keep_match_(keep_match) {}
 
  bool operator()(const Arc &arc) const {
    const bool match = (match_input_ ? arc.ilabel : arc.olabel) == label_;
    return keep_match_ ? match : !match;
  }
 
 private:
  const Label label_;
  const bool match_input_;
  const bool keep_match_;
};
 
// True if specified labels match (don't match) when keep_match is true (false).
template <class Arc>
class MultiLabelArcFilter {
 public:
  using Label = typename Arc::Label;
 
  explicit MultiLabelArcFilter(bool match_input = true, bool keep_match = true)
      : match_input_(match_input), keep_match_(keep_match) {}
 
  bool operator()(const Arc &arc) const {
    const Label label = match_input_ ? arc.ilabel : arc.olabel;
    const bool match = labels_.Find(label) != labels_.End();
    return keep_match_ ? match : !match;
  }
 
  void AddLabel(Label label) { labels_.Insert(label); }
 
 private:
  CompactSet<Label, kNoLabel> labels_;
  const bool match_input_;
  const bool keep_match_;
};
 
}  // namespace fst
 
#endif  // FST_ARCFILTER_H_