126 lines
2.8 KiB
C++
126 lines
2.8 KiB
C++
/* Copyright (C) 2007 J.F.Dockes
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the
|
|
* Free Software Foundation, Inc.,
|
|
* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
*/
|
|
|
|
#ifndef TEST_STOPLIST
|
|
#include "debuglog.h"
|
|
#include "readfile.h"
|
|
#include "unacpp.h"
|
|
#include "textsplit.h"
|
|
#include "stoplist.h"
|
|
|
|
#ifndef NO_NAMESPACES
|
|
namespace Rcl
|
|
{
|
|
#endif
|
|
|
|
class TextSplitSW : public TextSplit {
|
|
public:
|
|
set<string>& stops;
|
|
TextSplitSW(Flags flags, set<string>& stps)
|
|
: TextSplit(flags), stops(stps)
|
|
{}
|
|
virtual bool takeword(const string& term, int, int, int)
|
|
{
|
|
string dterm;
|
|
unacmaybefold(term, dterm, "UTF-8", true);
|
|
stops.insert(dterm);
|
|
return true;
|
|
}
|
|
};
|
|
|
|
bool StopList::setFile(const string &filename)
|
|
{
|
|
m_hasStops = false;
|
|
m_stops.clear();
|
|
string stoptext, reason;
|
|
if (!file_to_string(filename, stoptext, &reason)) {
|
|
LOGDEB(("StopList::StopList: file_to_string(%s) failed: %s\n",
|
|
filename.c_str(), reason.c_str()));
|
|
return false;
|
|
}
|
|
TextSplitSW ts(TextSplit::TXTS_ONLYSPANS, m_stops);
|
|
ts.text_to_words(stoptext);
|
|
m_hasStops = !m_stops.empty();
|
|
return true;
|
|
}
|
|
|
|
bool StopList::isStop(const string &term) const
|
|
{
|
|
return m_hasStops ? m_stops.find(term) != m_stops.end() : false;
|
|
}
|
|
|
|
|
|
#ifndef NO_NAMESPACES
|
|
}
|
|
#endif
|
|
|
|
#else // TEST_STOPLIST
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
|
|
#include <string>
|
|
#include <iostream>
|
|
|
|
#include "stoplist.h"
|
|
|
|
using namespace std;
|
|
using namespace Rcl;
|
|
|
|
static char *thisprog;
|
|
|
|
static char usage [] =
|
|
"trstoplist stopstermsfile\n\n"
|
|
;
|
|
static void
|
|
Usage(void)
|
|
{
|
|
fprintf(stderr, "%s: usage:\n%s", thisprog, usage);
|
|
exit(1);
|
|
}
|
|
|
|
const string tstwords[] = {
|
|
"the", "is", "xweird"
|
|
};
|
|
const int tstsz = sizeof(tstwords) / sizeof(string);
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int count = 10;
|
|
|
|
thisprog = argv[0];
|
|
argc--; argv++;
|
|
|
|
if (argc != 1)
|
|
Usage();
|
|
string filename = argv[0]; argc--;
|
|
|
|
StopList sl(filename);
|
|
|
|
for (int i = 0; i < tstsz; i++) {
|
|
const string &tst = tstwords[i];
|
|
cout << "[" << tst << "] " <<
|
|
(sl.isStop(tst) ? "in stop list" : "not in stop list") << endl;
|
|
}
|
|
exit(0);
|
|
}
|
|
|
|
#endif // TEST_STOPLIST
|