summaryrefslogtreecommitdiff
path: root/extractor/phrase_location.cc
blob: 13140cac086f54a2aa91197fa7fff6e4d9fe50e5 (plain)
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
#include "phrase_location.h"

namespace extractor {

PhraseLocation::PhraseLocation(int sa_low, int sa_high) :
    sa_low(sa_low), sa_high(sa_high), num_subpatterns(0) {}

PhraseLocation::PhraseLocation(const vector<int>& matchings,
                               int num_subpatterns) :
    sa_low(0), sa_high(0),
    matchings(make_shared<vector<int>>(matchings)),
    num_subpatterns(num_subpatterns) {}

bool PhraseLocation::IsEmpty() const {
  return GetSize() == 0;
}

int PhraseLocation::GetSize() const {
  if (num_subpatterns > 0) {
    return matchings->size();
  } else {
    return sa_high - sa_low;
  }
}

bool operator==(const PhraseLocation& a, const PhraseLocation& b) {
  if (a.sa_low != b.sa_low || a.sa_high != b.sa_high ||
      a.num_subpatterns != b.num_subpatterns) {
    return false;
  }

  if (a.matchings == NULL && b.matchings == NULL) {
    return true;
  }

  if (a.matchings == NULL || b.matchings == NULL) {
    return false;
  }

  return *a.matchings == *b.matchings;
}

} // namespace extractor