---

StructQueryRetMethod.hpp

Go to the documentation of this file.

/*==========================================================================
 * Copyright (c) 2002 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.cs.cmu.edu/~lemur/license.html
 *
 *==========================================================================
*/
/*
  Author: fff
 */


#ifndef _STRUCTQUERYRETMETHOD_HPP
#define _STRUCTQUERYRETMETHOD_HPP

#include "RetrievalMethod.hpp"
#include "StructQueryRep.hpp"
#include "DocumentRep.hpp"
#include "ScoreFunction.hpp"
#include "ScoreAccumulator.hpp"
#include "FreqVector.hpp"


//--------------------------------------------------------------------------
//      Abstract Interface for A Retrieval Method/Model for Structured Query
//--------------------------------------------------------------------------

class StructQueryRetMethod : public RetrievalMethod {
public:
  StructQueryRetMethod(InvFPIndex &ind,  ScoreAccumulator & accumulator) : 
    RetrievalMethod(ind), scAcc(accumulator), index(ind){}
  virtual ~StructQueryRetMethod() {}
  virtual StructQueryRep *computeStructQueryRep(StructQuery &qry)=0;

  virtual QueryRep *computeQueryRep(Query &qry); 
  virtual double scoreDocVector(StructQueryRep &qRep, int docID, FreqVector &docVector);
  virtual double scoreDoc(QueryRep &qry, int docID);
  virtual void scoreCollection(QueryRep &qry, IndexedRealVector &results);

  virtual DocumentRep *computeDocRep(int docID) =0;
  virtual ScoreFunction *scoreFunc() = 0;
  virtual void updateQuery(QueryRep &qryRep, DocIDSet &relDocs) {
    updateStructQuery(*((StructQueryRep *)(&qryRep)), relDocs);
  }

  virtual void updateStructQuery(StructQueryRep &qryRep, DocIDSet &relDocs)=0;

  virtual void scoreInvertedIndex(QueryRep &qryRep, IndexedRealVector &scores, bool scoreAll=false);

protected:
  ScoreAccumulator &scAcc;
  InvFPIndex &index;
};

class StructQueryScoreFunc : public ScoreFunction {
public:
  virtual double evalQuery(QueryNode *qn, DocumentRep *dRep) {
    return qn->eval(dRep);
  }
  virtual double adjustedScore(double w, QueryNode *qn) {
    return w;
  }  
};

//=============== inlines ========================================

inline QueryRep *StructQueryRetMethod::computeQueryRep(Query &qry) { 
  StructQuery *q = static_cast<StructQuery *>(&qry);
  return (computeStructQueryRep(*q));
}

#endif /* _STRUCTQUERYRETMETHOD_HPP */

---