00001 /*========================================================================== 00002 * Copyright (c) 2004 University of Massachusetts. All Rights Reserved. 00003 * 00004 * Use of the Lemur Toolkit for Language Modeling and Information Retrieval 00005 * is subject to the terms of the software license set forth in the LICENSE 00006 * file included with this software, and also available at 00007 * http://www.lemurproject.org/license.html 00008 * 00009 *========================================================================== 00010 */ 00011 00012 // 00013 // TermFrequencyBeliefNode 00014 // 00015 // 25 August 2004 -- tds 00016 // 00017 00018 #ifndef INDRI_TERMFREQUENCYBELIEFNODE_HPP 00019 #define INDRI_TERMFREQUENCYBELIEFNODE_HPP 00020 00021 #include "indri/Annotator.hpp" 00022 #include "indri/ScoredExtentResult.hpp" 00023 #include <string> 00024 #include "indri/TermScoreFunction.hpp" 00025 #include "indri/ListBeliefNode.hpp" 00026 #include "indri/DocListIterator.hpp" 00027 namespace indri 00028 { 00029 namespace infnet 00030 { 00031 00032 class TermFrequencyBeliefNode : public BeliefNode { 00033 private: 00034 class InferenceNetwork& _network; 00035 indri::query::TermScoreFunction& _function; 00036 indri::utility::greedy_vector<indri::api::ScoredExtentResult> _extents; 00037 indri::utility::greedy_vector<bool> _matches; 00038 indri::index::DocListIterator* _list; 00039 double _maximumBackgroundScore; 00040 double _maximumScore; 00041 std::string _name; 00042 int _listID; 00043 00044 indri::utility::greedy_vector<indri::index::DocListIterator::TopDocument> _emptyTopdocs; 00045 00046 public: 00047 TermFrequencyBeliefNode( const std::string& name, 00048 class InferenceNetwork& network, 00049 int listID, 00050 indri::query::TermScoreFunction& scoreFunction ); 00051 00052 ~TermFrequencyBeliefNode(); 00053 00054 const indri::utility::greedy_vector<indri::index::DocListIterator::TopDocument>& topdocs() const; 00055 lemur::api::DOCID_T nextCandidateDocument(); 00056 void indexChanged( indri::index::Index& index ); 00057 double maximumBackgroundScore(); 00058 double maximumScore(); 00059 const indri::utility::greedy_vector<indri::api::ScoredExtentResult>& score( lemur::api::DOCID_T documentID, indri::index::Extent &extent, int documentLength ); 00060 void annotate( class Annotator& annotator, lemur::api::DOCID_T documentID, indri::index::Extent &extent ); 00061 bool hasMatch( lemur::api::DOCID_T documentID ); 00062 const indri::utility::greedy_vector<bool>& hasMatch( lemur::api::DOCID_T documentID, const indri::utility::greedy_vector<indri::index::Extent>& extents ); 00063 const std::string& getName() const; 00064 }; 00065 } 00066 } 00067 00068 #endif // INDRI_TERMFREQUENCYBELIEFNODE_HPP 00069 00070