labs-edaf30/lab2/word.cc

#include "word.h"
#include "dictionary.h"
#include <algorithm>
#include <string>
#include <vector>

using std::string;
using std::vector;

Word::Word(const string &w, const vector<string> &t) : word(w), trigrams(t) {
    std::sort(trigrams.begin(), trigrams.end());
}

std::vector<std::string> get_trigrams(const std::string &text) {
    std::vector<std::string> trigrams;
    if (text.size() < 3) {
        return trigrams; // Return an empty vector if the input is too short
    }

    for (size_t i = 0; i <= text.size() - 3; ++i) {
        trigrams.push_back(
            text.substr(i, 3)); // Extract a substring of length 3
    }

    return trigrams;
}

Word::Word(const std::string &w) : word(w) {
    this->trigrams = ::get_trigrams(w);
    std::sort(trigrams.begin(), trigrams.end());
}

string Word::get_word() const { return word; }

vector<std::string> Word::get_trigrams() const { return trigrams; }

unsigned int Word::get_matches(const vector<string> &t) const {
    unsigned int matches = 0;

    for (const auto &trigram : t) {
        if (std::binary_search(trigrams.begin(), trigrams.end(), trigram)) {
            ++matches;
        }
    }
    return matches;
}

std::ostream &operator<<(std::ostream &out, const Word &w) {
    auto space = string(" ");
    out << w.word;
    out << space;
    out << w.trigrams.size();
    for (const auto &tri : w.trigrams) {
        out << space << tri;
    }
    return out;
}

bool operator==(const Word &lhs, const Word &rhs) {
    return lhs.word == rhs.word &&
           std::equal(lhs.trigrams.begin(), lhs.trigrams.end(),
                      rhs.trigrams.begin());
}