---

BasicIndexWithCat.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.cs.cmu.edu/~lemur/license.html
 *
 *==========================================================================
*/


#ifndef _BASICINDEXWITHCAT_HPP
#define _BASICINDEXWITHCAT_HPP

#include "IndexWithCat.hpp"


class BasicIndexWithCat : public IndexWithCat {
public:
  BasicIndexWithCat(Index &wordIndex, Index &categoryIndex, bool catIsTerm= true);

  
  ~BasicIndexWithCat() { delete [] base2cat; delete [] cat2base ;};

  bool open(const  string &name) {return false;} 



  virtual TERMID_T term (const string &word) const { return baseIndex->term(word);}

  virtual const string term (TERMID_T termID) const { return baseIndex->term(termID);}

  virtual DOCID_T document (const string &docIDStr) const { return baseIndex->document(docIDStr);}

  virtual const string document (DOCID_T docID) const { return baseIndex->document(docID);}

  virtual int category (const string &catIDStr) const;

  virtual const string category (int catID) const;




  virtual COUNT_T docCount () const { return baseIndex->docCount();}

  virtual COUNT_T termCountUnique () const { return baseIndex->termCountUnique();}

  virtual COUNT_T termCount (TERMID_T termID) const { return baseIndex->termCount(termID); }

  virtual COUNT_T termCount () const { return baseIndex->termCount(); }

  virtual float docLengthAvg() const { return baseIndex->docLengthAvg(); }

  virtual COUNT_T docCount(TERMID_T termID) const { return baseIndex->docCount(termID);}

  virtual COUNT_T docLength (DOCID_T docID) const {return baseIndex->docLength(docID); }

  virtual int catCount() const;

  virtual int docCountInCat(int catID) const;




  virtual DocInfoList *docInfoList(TERMID_T termID) const { return baseIndex->docInfoList(termID); }

  virtual TermInfoList *termInfoList(DOCID_T docID) const { return baseIndex->termInfoList(docID); }





  virtual void startCatIDIteration(int docID) const;
  virtual bool hasMoreCatID() const;

  virtual int nextCatID() const;


  virtual void startDocIDIteration(int catID) const;
  virtual bool hasMoreDocID() const;


  virtual int nextDocID() const;




private:
  Index *baseIndex; 
  Index *catIndex; 
  bool catAsTerm;

  mutable TermInfoList *tmInfoBuffer;
  mutable DocInfoList *docInfoBuffer;
  mutable TermInfo *tmInfo;
  mutable DocInfo *docInfo;

  int *base2cat;
  int *cat2base;
 
  static const string OOVSTRING;  
};

inline int BasicIndexWithCat::category (const string &catIDStr) const
{
  if (catAsTerm) {
    return (catIndex->term(catIDStr));
  } else {
    // docid as term, so cat as doc
    return (catIndex->document(catIDStr));
  }
}


inline const string BasicIndexWithCat::category (int catID) const
{
  if (catAsTerm) {
    return (catIndex->term(catID));
  } else {
    // docid as term, so cat as doc
    return (catIndex->document(catID));
  }
}



inline int BasicIndexWithCat::catCount() const
{
  if (catAsTerm) {
    return (catIndex->termCountUnique()-1); // -1 because of [OOV]
  } else {
    // docid as term, so cat as doc
    return (catIndex->docCount());
  }
}

inline int BasicIndexWithCat::docCountInCat(int catID) const
{
  if (catAsTerm) {
    return (catIndex->docCount(catID));
  } else {
    // docid as term, so cat as doc
    return (catIndex->docLength(catID));
  }
}



#endif /* _BASICINDEXWITHLABEL_HPP */

---