DocScore.hpp

Go to the documentation of this file.

/*==========================================================================
 * Copyright (c) 2001 Carnegie Mellon University.  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 _DOCSCORE_HPP
#define _DOCSCORE_HPP

#include "common_headers.hpp"
#include <algorithm>
#include <string>
namespace lemur
{
  namespace distrib 
  {
    struct DocScore {
      double val;
      string id;
    };

    class DocScoreVector : public vector<DocScore> {
    public:
  
      DocScoreVector() : vector<DocScore>() {}
      DocScoreVector(int size) : vector<DocScore>(size) {}
      virtual ~DocScoreVector();

      virtual void PushValue(const string &docid, double value);

      virtual void Sort(bool descending = true);
    private:


      class ScoreAscending {
      public:
        bool operator()(const DocScore & a, const DocScore & b) {
          return a.val < b.val;  // based on the real value
        }
      };

      class ScoreDescending { 
      public: 
        bool operator()(const DocScore & a, const DocScore & b) {
          return a.val > b.val;  // based on the real value
        }
      };

      static ScoreAscending ascendOrder;
      static ScoreDescending descendOrder;

    };
 
  }
}


#endif

---