---

IndriRetMethod.hpp

Go to the documentation of this file.

/*==========================================================================
 * Copyright (c) 2003-2004 University of Massachusetts.  All Rights Reserved.
 *
 * Use of the Lemur Toolkit for Language Modeling and Information Retrieval
 * is subject to the terms of the software license set forth in the LICENSE
 * file included with this software, and also available at
 * http://www.lemurproject.org/license.html
 *
 *==========================================================================
*/
#ifndef _INDRIRETMETHOD_HPP
#define _INDRIRETMETHOD_HPP

#include "LemurIndriIndex.hpp"
#include "TextQueryRep.hpp"
#include "RetrievalMethod.hpp"

class IndriQueryModel : public QueryRep {
public:
  IndriQueryModel(const TermQuery &qry) {
    // collect terms (with operators) into a single string.
    // could also collect real terms for matchinfo use here...

    qry.startTermIteration();
    while (qry.hasMore()) {
      const Term *t = qry.nextTerm();
      realQuery += t->spelling();
      realQuery += " ";
    }
  }
  
  virtual ~IndriQueryModel() {
  }

  const string &getQuery() const {return realQuery;}
  // for update method.
  void updateQuery(const string &newQuery) const { realQuery = newQuery;}

private:
  mutable string realQuery;
};

class IndriRetMethod : public RetrievalMethod {
public:
  // parameters? want to get the rules in certainly...
  IndriRetMethod(const Index &dbIndex);

  virtual ~IndriRetMethod();
  virtual void setParams(Parameters *parms);
  virtual void setStopwords(const string& stopfile) ;
  virtual void scoreCollection(const QueryRep &qry, 
                               IndexedRealVector &results);
  virtual void scoreCollection(const string &qry, 
                               IndexedRealVector &results);
  virtual QueryRep *computeQueryRep(const Query &qry) {
    if (const TermQuery *q = dynamic_cast<const TermQuery *>(&qry))
      return (new IndriQueryModel(*q));
    else LEMUR_THROW(LEMUR_RUNTIME_ERROR, 
                     "IndriRetMethod expects query of type TermQuery");
  } 

  virtual void scoreDocSet(const QueryRep &qry, const DocIDSet &docSet, 
                           IndexedRealVector &results);

  virtual double scoreDoc(const QueryRep &qry, DOCID_T docID);

  virtual void updateQuery(QueryRep &qryRep, const DocIDSet &relDocs);
private:
  class QueryEnvironment *env;
  Parameters *params;
};

#endif /* _INDRIRETMETHOD_HPP */

---