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"
namespace indri
{
  namespace api
  {
    class QueryEnvironment;
  }
}
namespace lemur 
{
  namespace retrieval 
  {
    
    class IndriQueryModel : public lemur::api::QueryRep {
    public:
      IndriQueryModel(const lemur::api::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 lemur::api::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 lemur::api::RetrievalMethod {
    public:
      // parameters? want to get the rules in certainly...
      IndriRetMethod(const lemur::api::Index &dbIndex);

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

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

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

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


#endif /* _INDRIRETMETHOD_HPP */

---