concordia-library/concordia/hash_generator.cpp

#include "concordia/hash_generator.hpp"
#include <boost/filesystem.hpp>
#include <boost/archive/binary_oarchive.hpp>
#include <boost/archive/binary_iarchive.hpp>
#include <boost/algorithm/string.hpp>
#include <fstream>

HashGenerator::HashGenerator(const string & wordMapFilename)
                                         throw(ConcordiaException) :
    _wordMapFilename(wordMapFilename),
    _wordMap(boost::shared_ptr<WordMap>(new WordMap)) {
    if (boost::filesystem::exists(_wordMapFilename)) {
        ifstream ifs(_wordMapFilename.c_str(), std::ios::binary);
        boost::archive::binary_iarchive ia(ifs);
        boost::shared_ptr<WordMap> restoredWordMap(new WordMap);
        ia >> *_wordMap;
    }
}

HashGenerator::~HashGenerator() {
}

vector<int> HashGenerator::generateHash(const string & sentence) {
    vector<int> result;
    vector<string> tokenTexts;
    boost::split(tokenTexts, sentence, boost::is_any_of(" "));

    for (vector<string>::iterator it = tokenTexts.begin();
                                it != tokenTexts.end(); ++it) {
        string token = *it;
        int code = _wordMap->getWordCode(token);
        result.push_back(code);
    }

    return result;
}

void HashGenerator::serializeWordMap() {
    ofstream ofs(_wordMapFilename.c_str(), std::ios::binary);
    boost::archive::binary_oarchive oa(ofs);
    oa << *_wordMap;
}