concordia_search_result.hpp

#ifndef CONCORDIA_SEARCH_RESULT_HDR
#define CONCORDIA_SEARCH_RESULT_HDR

#include "concordia/common/config.hpp"
#include "concordia/matched_pattern_fragment.hpp"
#include "concordia/tokenized_sentence.hpp"

#include <boost/shared_ptr.hpp>
#include <vector>
#include <string>

class ConcordiaSearchResult {
public:
    explicit ConcordiaSearchResult(TokenizedSentence tokenizedPattern);

    virtual ~ConcordiaSearchResult();

    void addFragment(const MatchedPatternFragment & fragment);

    void sortFragments();

    void computeBestOverlay();

    TokenizedSentence getTokenizedPattern() const {
        return _tokenizedPattern;
    }

    std::vector<MatchedPatternFragment> getFragments() const {
        return _matchedPatternFragments;
    }

    std::vector<MatchedPatternFragment> getBestOverlay() const {
        return _bestOverlay;
    }

    double getBestOverlayScore() const {
        return _bestOverlayScore;
    }

    friend std::ostream & operator << (std::ostream & o,
                                       const ConcordiaSearchResult & result) {
        o << "Best overlay {" << std::endl;
        BOOST_FOREACH(MatchedPatternFragment fragment,
                      result.getBestOverlay()) {
            o << fragment << std::endl;
        }
        o << "}" << std::endl;
        o << "All fragments {" << std::endl;
        BOOST_FOREACH(MatchedPatternFragment fragment,
                      result.getFragments()) {
            o << fragment << std::endl;
        }
        o << "}";
        return o;
    }


private:
    void _checkPossibleOverlays(
                std::vector<MatchedPatternFragment> currentOverlay,
                SUFFIX_MARKER_TYPE lastAddedPos,
                SUFFIX_MARKER_TYPE patternSize);

    TokenizedSentence _tokenizedPattern;

    std::vector<MatchedPatternFragment> _matchedPatternFragments;

    std::vector<MatchedPatternFragment> _bestOverlay;

    double _bestOverlayScore;
};

#endif