Stemmer.cpp
上传用户:sanxfzhen
上传日期:2014-12-28
资源大小:2324k
文件大小:10k
源码类别:

多国语言处理

开发平台:

Visual C++

  1. // Stemmer.cpp: implementation of the CStemmer class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  4. #include "stdafx.h"
  5. #include "Stemmer.h"
  6. #ifdef _DEBUG
  7. #undef THIS_FILE
  8. static char THIS_FILE[]=__FILE__;
  9. #define new DEBUG_NEW
  10. #endif
  11. CStemmer theStemmer;
  12. //////////////////////////////////////////////////////////////////////
  13. // Construction/Destruction
  14. //////////////////////////////////////////////////////////////////////
  15. CStemmer::CStemmer()
  16. {
  17. }
  18. CStemmer::~CStemmer()
  19. {
  20. }
  21. /* The main part of the stemming algorithm starts here. b is a buffer
  22.    holding a word to be stemmed. The letters are in b[k0], b[k0+1] ...
  23.    ending at b[k]. In fact k0 = 0 in this demo program. k is readjusted
  24.    downwards as the stemming progresses. Zero termination is not in fact
  25.    used in the algorithm.
  26.    Note that only lower case sequences are stemmed. Forcing to lower case
  27.    should be done before stem(...) is called.
  28. */
  29. /* cons(i) is TRUE <=> b[i] is a consonant. */
  30. int CStemmer::cons(int i)
  31. {  switch (b[i])
  32.    {  case 'a': case 'e': case 'i': case 'o': case 'u': return FALSE;
  33.       case 'y': return (i==k0) ? TRUE : !cons(i-1);
  34.       default: return TRUE;
  35.    }
  36. }
  37. /* m() measures the number of consonant sequences between k0 and j. if c is
  38.    a consonant sequence and v a vowel sequence, and <..> indicates arbitrary
  39.    presence,
  40.       <c><v>       gives 0
  41.       <c>vc<v>     gives 1
  42.       <c>vcvc<v>   gives 2
  43.       <c>vcvcvc<v> gives 3
  44.       ....
  45. */
  46. int CStemmer::m()
  47. {  int n = 0;
  48.    int i = k0;
  49.    while(TRUE)
  50.    {  if (i > j) return n;
  51.       if (! cons(i)) break; i++;
  52.    }
  53.    i++;
  54.    while(TRUE)
  55.    {  while(TRUE)
  56.       {  if (i > j) return n;
  57.             if (cons(i)) break;
  58.             i++;
  59.       }
  60.       i++;
  61.       n++;
  62.       while(TRUE)
  63.       {  if (i > j) return n;
  64.          if (! cons(i)) break;
  65.          i++;
  66.       }
  67.       i++;
  68.    }
  69. }
  70. /* vowelinstem() is TRUE <=> k0,...j contains a vowel */
  71. int CStemmer::vowelinstem()
  72. {  int i; for (i = k0; i <= j; i++) if (! cons(i)) return TRUE;
  73.    return FALSE;
  74. }
  75. /* doublec(j) is TRUE <=> j,(j-1) contain a double consonant. */
  76. int CStemmer::doublec(int j)
  77. {  if (j < k0+1) return FALSE;
  78.    if (b[j] != b[j-1]) return FALSE;
  79.    return cons(j);
  80. }
  81. /* cvc(i) is TRUE <=> i-2,i-1,i has the form consonant - vowel - consonant
  82.    and also if the second c is not w,x or y. this is used when trying to
  83.    restore an e at the end of a short word. e.g.
  84.       cav(e), lov(e), hop(e), crim(e), but
  85.       snow, box, tray.
  86. */
  87. int CStemmer::cvc(int i)
  88. {  if (i < k0+2 || !cons(i) || cons(i-1) || !cons(i-2)) return FALSE;
  89.    {  int ch = b[i];
  90.       if (ch == 'w' || ch == 'x' || ch == 'y') return FALSE;
  91.    }
  92.    return TRUE;
  93. }
  94. /* ends(s) is TRUE <=> k0,...k ends with the string s. */
  95. int CStemmer::ends(char * s)
  96. {  int length = s[0];
  97.    if (s[length] != b[k]) return FALSE; /* tiny speed-up */
  98.    if (length > k-k0+1) return FALSE;
  99.    if (memcmp(b+k-length+1,s+1,length) != 0) return FALSE;
  100.    j = k-length;
  101.    return TRUE;
  102. }
  103. /* setto(s) sets (j+1),...k to the characters in the string s, readjusting
  104.    k. */
  105. void CStemmer::setto(char * s)
  106. {  int length = s[0];
  107.    memmove(b+j+1,s+1,length);
  108.    k = j+length;
  109. }
  110. /* r(s) is used further down. */
  111. void CStemmer::r(char * s) { if (m() > 0) setto(s); }
  112. /* step1ab() gets rid of plurals and -ed or -ing. e.g.
  113.        caresses  ->  caress
  114.        ponies    ->  poni
  115.        ties      ->  ti
  116.        caress    ->  caress
  117.        cats      ->  cat
  118.        feed      ->  feed
  119.        agreed    ->  agree
  120.        disabled  ->  disable
  121.        matting   ->  mat
  122.        mating    ->  mate
  123.        meeting   ->  meet
  124.        milling   ->  mill
  125.        messing   ->  mess
  126.        meetings  ->  meet
  127. */
  128. void CStemmer::step1ab()
  129. {  if (b[k] == 's')
  130.    {  if (ends("4" "sses")) k -= 2; else
  131.       if (ends("3" "ies")) setto("1" "i"); else
  132.       if (b[k-1] != 's') k--;
  133.    }
  134.    if (ends("3" "eed")) { if (m() > 0) k--; } else
  135.    if ((ends("2" "ed") || ends("3" "ing")) && vowelinstem())
  136.    {  k = j;
  137.       if (ends("2" "at")) setto("3" "ate"); else
  138.       if (ends("2" "bl")) setto("3" "ble"); else
  139.       if (ends("2" "iz")) setto("3" "ize"); else
  140.       if (doublec(k))
  141.       {  k--;
  142.          {  int ch = b[k];
  143.             if (ch == 'l' || ch == 's' || ch == 'z') k++;
  144.          }
  145.       }
  146.       else if (m() == 1 && cvc(k)) setto("1" "e");
  147.    }
  148. }
  149. /* step1c() turns terminal y to i when there is another vowel in the stem. */
  150. void CStemmer::step1c() { if (ends("1" "y") && vowelinstem()) b[k] = 'i'; }
  151. /* step2() maps double suffices to single ones. so -ization ( = -ize plus
  152.    -ation) maps to -ize etc. note that the string before the suffix must give
  153.    m() > 0. */
  154. void CStemmer::step2() { switch (b[k-1])
  155. {
  156.     case 'a': if (ends("7" "ational")) { r("3" "ate"); break; }
  157.               if (ends("6" "tional")) { r("4" "tion"); break; }
  158.               break;
  159.     case 'c': if (ends("4" "enci")) { r("4" "ence"); break; }
  160.               if (ends("4" "anci")) { r("4" "ance"); break; }
  161.               break;
  162.     case 'e': if (ends("4" "izer")) { r("3" "ize"); break; }
  163.               break;
  164.     case 'l': if (ends("3" "bli")) { r("3" "ble"); break; } /*-DEPARTURE-*/
  165.  /* To match the published algorithm, replace this line with
  166.     case 'l': if (ends("4" "abli")) { r("4" "able"); break; } */
  167.               if (ends("4" "alli")) { r("2" "al"); break; }
  168.               if (ends("5" "entli")) { r("3" "ent"); break; }
  169.               if (ends("3" "eli")) { r("1" "e"); break; }
  170.               if (ends("5" "ousli")) { r("3" "ous"); break; }
  171.               break;
  172.     case 'o': if (ends("7" "ization")) { r("3" "ize"); break; }
  173.               if (ends("5" "ation")) { r("3" "ate"); break; }
  174.               if (ends("4" "ator")) { r("3" "ate"); break; }
  175.               break;
  176.     case 's': if (ends("5" "alism")) { r("2" "al"); break; }
  177.               if (ends("7" "iveness")) { r("3" "ive"); break; }
  178.               if (ends("7" "fulness")) { r("3" "ful"); break; }
  179.               if (ends("7" "ousness")) { r("3" "ous"); break; }
  180.               break;
  181.     case 't': if (ends("5" "aliti")) { r("2" "al"); break; }
  182.               if (ends("5" "iviti")) { r("3" "ive"); break; }
  183.               if (ends("6" "biliti")) { r("3" "ble"); break; }
  184.               break;
  185.     case 'g': if (ends("4" "logi")) { r("3" "log"); break; } /*-DEPARTURE-*/
  186.  /* To match the published algorithm, delete this line */
  187. } }
  188. /* step3() deals with -ic-, -full, -ness etc. similar strategy to step2. */
  189. void CStemmer::step3() { switch (b[k])
  190. {
  191.     case 'e': if (ends("5" "icate")) { r("2" "ic"); break; }
  192.               if (ends("5" "ative")) { r("0" ""); break; }
  193.               if (ends("5" "alize")) { r("2" "al"); break; }
  194.               break;
  195.     case 'i': if (ends("5" "iciti")) { r("2" "ic"); break; }
  196.               break;
  197.     case 'l': if (ends("4" "ical")) { r("2" "ic"); break; }
  198.               if (ends("3" "ful")) { r("0" ""); break; }
  199.               break;
  200.     case 's': if (ends("4" "ness")) { r("0" ""); break; }
  201.               break;
  202. } }
  203. /* step4() takes off -ant, -ence etc., in context <c>vcvc<v>. */
  204. void CStemmer::step4()
  205. {  switch (b[k-1])
  206.     {  case 'a': if (ends("2" "al")) break; return;
  207.        case 'c': if (ends("4" "ance")) break;
  208.                  if (ends("4" "ence")) break; return;
  209.        case 'e': if (ends("2" "er")) break; return;
  210.        case 'i': if (ends("2" "ic")) break; return;
  211.        case 'l': if (ends("4" "able")) break;
  212.                  if (ends("4" "ible")) break; return;
  213.        case 'n': if (ends("3" "ant")) break;
  214.                  if (ends("5" "ement")) break;
  215.                  if (ends("4" "ment")) break;
  216.                  if (ends("3" "ent")) break; return;
  217.        case 'o': if (ends("3" "ion") && (b[j] == 's' || b[j] == 't')) break;
  218.                  if (ends("2" "ou")) break; return;
  219.                  /* takes care of -ous */
  220.        case 's': if (ends("3" "ism")) break; return;
  221.        case 't': if (ends("3" "ate")) break;
  222.                  if (ends("3" "iti")) break; return;
  223.        case 'u': if (ends("3" "ous")) break; return;
  224.        case 'v': if (ends("3" "ive")) break; return;
  225.        case 'z': if (ends("3" "ize")) break; return;
  226.        default: return;
  227.     }
  228.     if (m() > 1) k = j;
  229. }
  230. /* step5() removes a final -e if m() > 1, and changes -ll to -l if
  231.    m() > 1. */
  232. void CStemmer::step5()
  233. {  j = k;
  234.    if (b[k] == 'e')
  235.    {  int a = m();
  236.       if (a > 1 || a == 1 && !cvc(k-1)) k--;
  237.    }
  238.    if (b[k] == 'l' && doublec(k) && m() > 1) k--;
  239. }
  240. /* In stem(p,i,j), p is a char pointer, and the string to be stemmed is from
  241.    p[i] to p[j] inclusive. Typically i is zero and j is the offset to the last
  242.    character of a string, (p[j+1] == ''). The stemmer adjusts the
  243.    characters p[i] ... p[j] and returns the new end-point of the string, k.
  244.    Stemming never increases word length, so i <= k <= j. To turn the stemmer
  245.    into a module, declare 'stem' as extern, and delete the remainder of this
  246.    file.
  247. */
  248. int CStemmer::stem(char * p, int i, int j)
  249. {  b = p; k = j; k0 = i; /* copy the parameters into statics */
  250.    if (k <= k0+1) return k; /*-DEPARTURE-*/
  251.    /* With this line, strings of length 1 or 2 don't go through the
  252.       stemming process, although no mention is made of this in the
  253.       published algorithm. Remove the line to match the published
  254.       algorithm. */
  255.    step1ab(); step1c(); step2(); step3(); step4(); step5();
  256.    return k;
  257. }
  258. void CStemmer::stem(char *p)
  259. {
  260. stem(p,0,strlen(p)-1);
  261. }