tris/recoll
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

recollq: specify max snippets list length with option -p

Browse Source
This commit is contained in:
Jean-Francois Dockes 2020-08-31 11:54:44 +02:00
parent 5ea2f7cc64
commit 1795320873
1 changed files with 62 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/query/recollq.cpp
View File

@ -58,13 +58,14 @@ bool dump_contents(RclConfig *rclconfig, Rcl::Doc& idoc)
return true;
}
string make_abstract(Rcl::Doc& doc, Rcl::Query& query, bool asSnippets)
string make_abstract(Rcl::Doc& doc, Rcl::Query& query, bool asSnippets,
int snipcount)
{
string abstract;
if (asSnippets) {
std::vector<Rcl::Snippet> snippets;
std::ostringstream str;
if (query.makeDocAbstract(doc, snippets, -1, -1, true)) {
if (query.makeDocAbstract(doc, snippets, snipcount, -1, true)) {
for (const auto snippet : snippets) {
str << snippet.page << " : " << snippet.snippet << endl;
}
@ -78,7 +79,8 @@ string make_abstract(Rcl::Doc& doc, Rcl::Query& query, bool asSnippets)
}
void output_fields(vector<string> fields, Rcl::Doc& doc,
Rcl::Query& query, Rcl::Db&, bool printnames, bool asSnippets)
Rcl::Query& query, Rcl::Db&, bool printnames,
bool asSnippets, int snipcnt)
{
if (fields.empty()) {
map<string,string>::const_iterator it;
@ -90,7 +92,7 @@ void output_fields(vector<string> fields, Rcl::Doc& doc,
it != fields.end(); it++) {
string out;
if (!it->compare("abstract")) {
base64_encode(make_abstract(doc, query, asSnippets), out);
base64_encode(make_abstract(doc, query, asSnippets, snipcnt), out);
} else if (!it->compare("xdocid")) {
char cdocid[30];
sprintf(cdocid, "%lu", (unsigned long)doc.xdocid);
@ -212,6 +214,8 @@ int recollq(RclConfig **cfp, int argc, char **argv)
int firstres = 0;
int maxcount = 2000;
int snipcnt = -1;
thisprog = argv[0];
argc--; argv++;
@ -265,7 +269,18 @@ int recollq(RclConfig **cfp, int argc, char **argv)
argc--; goto b1;
case 'o': op_flags |= OPT_o; break;
case 'P': op_flags |= OPT_P; break;
case 'p': op_flags |= OPT_p; break;
case 'p':
{
op_flags |= OPT_p;
if (argc < 2)
Usage();
const char *cp = *(++argv);
char *cpe;
snipcnt = strtol(cp, &cpe, 10);
if (*cpe != 0)
Usage();
argc--; goto b1;
}
case 'q': op_flags |= OPT_q; break;
case 'Q': op_flags |= OPT_Q; break;
case 'S': op_flags |= OPT_S; if (argc < 2) Usage();
@ -424,7 +439,7 @@ endopts:
if (op_flags & OPT_F) {
output_fields(fields, doc, query, rcldb,
op_flags & OPT_N, op_flags & OPT_p);
op_flags & OPT_N, op_flags & OPT_p, snipcnt);
continue;
}
@ -458,7 +473,7 @@ endopts:
}
if (op_flags & OPT_A) {
bool asSnippets = (op_flags & OPT_p) != 0;
string abstract = make_abstract(doc, query, asSnippets);
string abstract = make_abstract(doc, query, asSnippets, snipcnt);
string marker = asSnippets ? "SNIPPETS" : "ABSTRACT";
if (!abstract.empty()) {
cout << marker << endl;