#include "tagging.h"
#include <ruby/encoding.h>
#include <PosTagger.hpp>
#include <MixSegment.hpp>

static rb_encoding* u8_enc;

struct Tagging{
    cppjieba::MixSegment *p;
};

static void tagger_free(void *p){
    delete reinterpret_cast<Tagging *>(p)->p;
    delete reinterpret_cast<Tagging *>(p);
}

static VALUE alloc(VALUE klass)
{
    Tagging * tagging = new Tagging();
    return Data_Wrap_Struct(klass, NULL, tagger_free, tagging);
}

static void init(VALUE self,
                 VALUE jieba_dict_rbs,
                 VALUE hmm_dict_rbs,
                 VALUE user_dict_rbs)
{
    Tagging *tagging;
    Data_Get_Struct(self, Tagging, tagging);

    Check_Type(jieba_dict_rbs, T_STRING);
    Check_Type(hmm_dict_rbs, T_STRING);
    Check_Type(user_dict_rbs, T_STRING);

    std::string jieba_dict = StringValueCStr(jieba_dict_rbs);
    std::string hmm_dict = StringValueCStr(hmm_dict_rbs);
    std::string user_dict = StringValueCStr(user_dict_rbs);

    tagging->p = new cppjieba::MixSegment(jieba_dict, hmm_dict, user_dict);
}

static VALUE tag(VALUE self, VALUE text_rbs)
{
    Check_Type(text_rbs, T_STRING);
    std::string text = StringValueCStr(text_rbs);

    Tagging *tagging;
    Data_Get_Struct(self, Tagging, tagging);

    std::vector<std::pair<std::string, std::string>> pairs;
    tagging->p->Tag(text, pairs);

    volatile VALUE arr = rb_ary_new();
    for (std::vector<std::pair<std::string, std::string>>::const_iterator j = pairs.begin(); j != pairs.end(); j++)
    {
        VALUE pair = rb_hash_new();
        rb_hash_aset(pair, rb_enc_str_new(std::get<0>(*j).c_str(), std::get<0>(*j).length(), u8_enc), rb_enc_str_new(std::get<1>(*j).c_str(), std::get<1>(*j).length(), u8_enc));
        rb_ary_push(arr, pair);

    }
    return arr;
}

#define DEF(k,n,f,c) rb_define_method(k,n,RUBY_METHOD_FUNC(f),c)

extern "C" {
    void Init_tagging()
    {
        VALUE cTagging = rb_define_class_under(mJieba, "Tagging", rb_cObject);
        u8_enc = rb_utf8_encoding();
        rb_define_alloc_func(cTagging, alloc);
        DEF(cTagging, "_init",init,3);
        DEF(cTagging, "tag",tag,1);
    }

}