xapian-core/queryparser/termgenerator.cc

   1 /** @file termgenerator.cc
   2  * @brief TermGenerator class implementation
   3  */
   4 /* Copyright (C) 2007,2012 Olly Betts
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  19  */
  20
  21 #include <config.h>
  22
  23 #include <xapian/termgenerator.h>
  24 #include <xapian/types.h>
  25 #include <xapian/unicode.h>
  26
  27 #include "termgenerator_internal.h"
  28
  29 #include "str.h"
  30
  31 using namespace std;
  32 using namespace Xapian;
  33
  34 TermGenerator::TermGenerator(const TermGenerator &) = default;
  35
  36 TermGenerator &
  37 TermGenerator::operator=(const TermGenerator &) = default;
  38
  39 TermGenerator::TermGenerator(TermGenerator &&) = default;
  40
  41 TermGenerator &
  42 TermGenerator::operator=(TermGenerator &&) = default;
  43
  44 TermGenerator::TermGenerator() : internal(new TermGenerator::Internal) { }
  45
  46 TermGenerator::~TermGenerator() { }
  47
  48 void
  49 TermGenerator::set_stemmer(const Xapian::Stem & stemmer)
  50 {
  51     internal->stemmer = stemmer;
  52 }
  53
  54 void
  55 TermGenerator::set_stopper(const Xapian::Stopper * stopper)
  56 {
  57     internal->stopper = stopper;
  58 }
  59
  60 void
  61 TermGenerator::set_document(const Xapian::Document & doc)
  62 {
  63     internal->doc = doc;
  64     internal->cur_pos = 0;
  65 }
  66
  67 const Xapian::Document &
  68 TermGenerator::get_document() const
  69 {
  70     return internal->doc;
  71 }
  72
  73 void
  74 TermGenerator::set_database(const Xapian::WritableDatabase &db)
  75 {
  76     internal->db = db;
  77 }
  78
  79 TermGenerator::flags
  80 TermGenerator::set_flags(flags toggle, flags mask)
  81 {
  82     TermGenerator::flags old_flags = internal->flags;
  83     internal->flags = flags((old_flags & mask) ^ toggle);
  84     return old_flags;
  85 }
  86
  87 void
  88 TermGenerator::set_stemming_strategy(stem_strategy strategy)
  89 {
  90     internal->strategy = strategy;
  91 }
  92
  93 void
  94 TermGenerator::set_stopper_strategy(stop_strategy strategy)
  95 {
  96     internal->stop_mode = strategy;
  97 }
  98
  99 void
 100 TermGenerator::set_max_word_length(unsigned max_word_length)
 101 {
 102     internal->max_word_length = max_word_length;
 103 }
 104
 105 void
 106 TermGenerator::index_text(const Xapian::Utf8Iterator & itor,
 107                           Xapian::termcount weight,
 108                           const string & prefix)
 109 {
 110     internal->index_text(itor, weight, prefix, true);
 111 }
 112
 113 void
 114 TermGenerator::index_text_without_positions(const Xapian::Utf8Iterator & itor,
 115                                             Xapian::termcount weight,
 116                                             const string & prefix)
 117 {
 118     internal->index_text(itor, weight, prefix, false);
 119 }
 120
 121 void
 122 TermGenerator::increase_termpos(Xapian::termpos delta)
 123 {
 124     internal->cur_pos += delta;
 125 }
 126
 127 Xapian::termpos
 128 TermGenerator::get_termpos() const
 129 {
 130     return internal->cur_pos;
 131 }
 132
 133 void
 134 TermGenerator::set_termpos(Xapian::termpos termpos)
 135 {
 136     internal->cur_pos = termpos;
 137 }
 138
 139 string
 140 TermGenerator::get_description() const
 141 {
 142     string s("Xapian::TermGenerator(stem=");
 143     s += internal->stemmer.get_description();
 144     if (internal->stopper.get()) {
 145         s += ", stopper set";
 146     }
 147     s += ", doc=";
 148     s += internal->doc.get_description();
 149     s += ", termpos=";
 150     s += str(internal->cur_pos);
 151     s += ")";
 152     return s;
 153 }