Switch to side-by-side view

--- a/src/rcldb/searchdata.h
+++ b/src/rcldb/searchdata.h
@@ -1,16 +1,38 @@
+/*
+ *   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 _SEARCHDATA_H_INCLUDED_
 #define _SEARCHDATA_H_INCLUDED_
-/* @(#$Id: searchdata.h,v 1.5 2006-11-15 14:57:53 dockes Exp $  (C) 2004 J.F.Dockes */
-
+/* @(#$Id: searchdata.h,v 1.6 2006-11-17 10:06:34 dockes Exp $  (C) 2004 J.F.Dockes */
+
+/** 
+ * Structures to hold data coming almost directly from the gui
+ * search fields and handle its translation to Xapian queries. 
+ * This is not generic code, it reflects the choices made for the user 
+ * interface, and it also knows some specific of recoll's usage of Xapian 
+ * (ie: term prefixes)
+ */
 #include <string>
-#include <list>
+#include <vector>
 
 #include "rcldb.h"
 
 #ifndef NO_NAMESPACES
-using std::list;
+using std::vector;
 using std::string;
-
 namespace Rcl {
 #endif // NO_NAMESPACES
 
@@ -24,87 +46,147 @@
 class SearchDataClause;
 
 /** 
- * Holder for a list of search clauses. Some of the clauses can be comples
- * subqueries.
+ * Holder for a list of search clauses. Some of the clauses may be be reference
+ * to other subqueries in the future. For now, they just reflect user entry in 
+ * a query field: type, some text and possibly a distance. Each clause may
+ * hold several queries in the Xapian sense, for exemple several terms
+ * and phrases as would result from ["this is a phrase" term1 term2]
  */
 class SearchData {
- public:
-    SClType                  m_tp; // Only SCLT_AND or SCLT_OR here
-    list<SearchDataClause *> m_query;
-    list<string>             m_filetypes; // Restrict to filetypes if set.
-    string                   m_topdir; // Restrict to subtree.
-    // Printable expanded version of the complete query, obtained from Xapian
-    // valid after setQuery() call
+public:
+    SearchData(SClType tp) : m_tp(tp) {}
+    ~SearchData() {erase();}
+
+    /** Make pristine */
+    void erase();
+
+    /** Is there anything but a file name search in here ? */
+    bool fileNameOnly();
+
+    /** Translate to Xapian query. rcldb knows about the void*  */
+    bool toNativeQuery(Rcl::Db &db, void *, const string& stemlang);
+
+    /** We become the owner of cl and will delete it */
+    bool addClause(SearchDataClause *cl);
+
+    /** Retrieve error description */
+    string getReason() {return m_reason;}
+
+    /** Get terms and phrase/near groups. Used in the GUI for highlighting 
+     * The groups and gslks vectors are parallel and hold the phrases/near
+     * string groups and their associated slacks (distance in excess of group
+     * size)
+     */
+    bool getTerms(vector<string>& terms, 
+		  vector<vector<string> >& groups, vector<int>& gslks) const;
+    /** 
+     * Get/set the description field which is retrieved from xapian after
+     * initializing the query. It is stored here for usage in the GUI.
+     */
+    string getDescription() {return m_description;}
+    void setDescription(const string& d) {m_description = d;}
+    string getTopdir() {return m_topdir;}
+    void setTopdir(const string& t) {m_topdir = t;}
+    void addFiletype(const string& ft) {m_filetypes.push_back(ft);}
+private:
+    SClType                    m_tp; // Only SCLT_AND or SCLT_OR here
+    vector<SearchDataClause *> m_query;
+    vector<string>             m_filetypes; // Restrict to filetypes if set.
+    string                     m_topdir; // Restrict to subtree.
+    // Printable expanded version of the complete query, retrieved/set
+    // from rcldb after the Xapian::setQuery() call
     string m_description; 
     string m_reason;
 
-    SearchData(SClType tp) : m_tp(tp) {}
-    ~SearchData() {erase();}
-
-    /** Make pristine */
-    void erase();
-
-    /** Is there anything but a file name search in here ? */
-    bool fileNameOnly();
-
-    /** Translate to Xapian query. rcldb knows about the void*  */
-    bool toNativeQuery(Rcl::Db &db, void *, const string& stemlang);
-
-    /** We become the owner of cl and will delete it */
-    bool addClause(SearchDataClause *cl);
-
-    string getReason() {return m_reason;}
-
- private:
     /* Copyconst and assignment private and forbidden */
     SearchData(const SearchData &) {}
     SearchData& operator=(const SearchData&) {return *this;};
 };
 
 class SearchDataClause {
- public:
-    SClType m_tp;
-
+public:
     SearchDataClause(SClType tp) : m_tp(tp) {}
     virtual ~SearchDataClause() {}
+
     virtual bool toNativeQuery(Rcl::Db &db, void *, const string&) = 0;
-    virtual bool isFileName() {return m_tp == SCLT_FILENAME ? true : false;}
-    string getReason() {return m_reason;}
- protected:
+
+    virtual bool isFileName() const {return m_tp==SCLT_FILENAME ? true: false;}
+
+    virtual string getReason() const {return m_reason;}
+
+    virtual bool getTerms(vector<string>&, vector<vector<string> >&,
+			  vector<int>&) const
+    {return true;}
+    virtual SClType getTp() {return m_tp;}
+
+    friend class SearchData;
+
+protected:
     string m_reason;
-};
-
+    SClType m_tp;
+};
+    
+/**
+ * "Simple" data clause with user-entered query text. This can include 
+ * multiple phrases and words, but no specified distance.
+ */
 class SearchDataClauseSimple : public SearchDataClause {
 public:
     SearchDataClauseSimple(SClType tp, string txt)
-	: SearchDataClause(tp), m_text(txt) {}
+	: SearchDataClause(tp), m_text(txt), m_slack(0) {}
     virtual ~SearchDataClauseSimple() {}
-    virtual bool toNativeQuery(Rcl::Db &db, void *, const string& stemlang);
+
+    virtual bool toNativeQuery(Rcl::Db &db, void *, const string& stemlang);
+
+    virtual bool getTerms(vector<string>& terms, 
+			  vector<vector<string> >& groups,
+			  vector<int>& gslks) const
+    {
+	terms.insert(terms.end(), m_terms.begin(), m_terms.end());
+	groups.insert(groups.end(), m_groups.begin(), m_groups.end());
+	gslks.insert(gslks.end(), m_groups.size(), m_slack);
+	return true;
+    }
+
 protected:
     string  m_text;
-};
-
+    // Single terms and phrases resulting from breaking up m_text;
+    // valid after toNativeQuery() call
+    vector<string>          m_terms;
+    vector<vector<string> > m_groups; 
+    // Declare m_slack here. Always 0, but allows getTerms to work for
+    // SearchDataClauseDist
+    int m_slack;
+};
+
+/** Filename search. */
 class SearchDataClauseFilename : public SearchDataClauseSimple {
- public:
+public:
     SearchDataClauseFilename(string txt)
 	: SearchDataClauseSimple(SCLT_FILENAME, txt) {}
     virtual ~SearchDataClauseFilename() {}
     virtual bool toNativeQuery(Rcl::Db &db, void *, const string& stemlang);
 };
 
+/** 
+ * A clause coming from a NEAR or PHRASE entry field. There is only one 
+ * string group, and a specified distance, which applies to it.
+ */
 class SearchDataClauseDist : public SearchDataClauseSimple {
 public:
     SearchDataClauseDist(SClType tp, string txt, int slack) 
-	: SearchDataClauseSimple(tp, txt), m_slack(slack) {}
+	: SearchDataClauseSimple(tp, txt) {m_slack = slack;}
     virtual ~SearchDataClauseDist() {}
-    virtual bool toNativeQuery(Rcl::Db &db, void *, const string& stemlang);
-
-protected:
-    int     m_slack;
-};
-
+
+    virtual bool toNativeQuery(Rcl::Db &db, void *, const string& stemlang);
+
+    // m_slack is declared in SearchDataClauseSimple
+};
+
+#ifdef NOTNOW
+/** Future pointer to subquery ? */
 class SearchDataClauseSub : public SearchDataClause {
- public:
+public:
     SearchDataClauseSub(SClType tp, SClType stp) 
 	: SearchDataClause(tp), m_sub(stp) {}
     virtual ~SearchDataClauseSub() {}
@@ -113,6 +195,7 @@
 protected:
     SearchData m_sub;
 };
+#endif // NOTNOW
 
 } // Namespace Rcl
 #endif /* _SEARCHDATA_H_INCLUDED_ */