---

RawScorerNodeExtractor.hpp

Go to the documentation of this file.

/*==========================================================================
 * Copyright (c) 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
 *
 *==========================================================================
*/


//
// RawScorerNodeExtractor.hpp
//
// 5 March 2004 -- tds
//

#ifndef INDRI_RAWSCORERNODEEXTRACTOR_HPP
#define INDRI_RAWSCORERNODEEXTRACTOR_HPP

#include "indri/Walker.hpp"
#include "indri/QuerySpec.hpp"

class RawScorerNodeExtractor : public indri::lang::Walker {
private:
  std::vector<indri::lang::RawScorerNode*> _scorerNodes;
  std::vector<indri::lang::Node*> _reachableNodes;
  int _insideCount;

public:
  void defaultBefore( indri::lang::Node* n ) {
    if( _insideCount ) {
      _reachableNodes.push_back(n);
    }
  }

  void before( indri::lang::RawScorerNode* node ) {
    _insideCount++;
    _scorerNodes.push_back(node);
    _reachableNodes.push_back(node);
  }

  void after( indri::lang::RawScorerNode* node ) {
    _insideCount--;
  }

  std::vector<indri::lang::RawScorerNode*>& getScorerNodes() {
    return _scorerNodes;
  }

  std::vector<indri::lang::Node*>& getReachableNodes() {
    return _reachableNodes;
  }
};

#endif // INDRI_RAWSCORERNODEEXTRACTOR_HPP

---