postag_dll.cpp

#include "postag_dll.h"

#include "postagger.h"
#include "settings.h"

#include "logging.hpp"
#include "codecs.hpp"
#include "sbcdbc.hpp"

#include <iostream>

class PostaggerWrapper : public ltp::postagger::Postagger {
public:
    PostaggerWrapper() {}
    ~PostaggerWrapper() {}

    bool load(const char * model_file) {
        std::ifstream mfs(model_file, std::ifstream::binary);

        if (!mfs) {
            return false;
        }

        model = new ltp::postagger::Model;
        if (!model->load(mfs)) {
            delete model;
            return false;
        }

        return true;
    }

    int postag(const std::vector<std::string> & words,
            std::vector<std::string> & tags) {
        ltp::postagger::Instance * inst = new ltp::postagger::Instance;
        ltp::postagger::Decoder deco(model->num_labels());

        for (int i = 0; i < words.size(); ++ i) {
            inst->forms.push_back(ltp::strutils::chartypes::sbc2dbc_x(words[i]));
        }

        ltp::postagger::Postagger::extract_features(inst);
        ltp::postagger::Postagger::calculate_scores(inst, true);
        deco.decode(inst);

        ltp::postagger::Postagger::build_labels(inst, tags);

        delete inst;
        return tags.size();
    }
};

void * postagger_create_postagger(const char * path) {
    PostaggerWrapper * wrapper = new PostaggerWrapper();

    if (!wrapper->load(path)) {
        return 0;
    }

    return reinterpret_cast<void *>(wrapper);
}

int postagger_release_postagger(void * postagger) {
    if (!postagger) {
        return -1;
    }
    delete reinterpret_cast<PostaggerWrapper *>(postagger);
    return 0;
}

int postagger_postag(void * postagger,
        const std::vector<std::string> & words,
        std::vector<std::string> & tags) {
    // std::cout << "input str = " << str << std::endl;
    PostaggerWrapper * wrapper = 0;
    wrapper = reinterpret_cast<PostaggerWrapper *>(postagger);
    return wrapper->postag(words, tags);
}