cws_cmdline.cpp

/*
 * Single-threaded segmentor test program. The user input a line
 * of Chinese sentence an the program will output its segment
 * result.
 *
 *  @dependency package: tinythread - a portable c++ wrapper for
 *                       multi-thread library.
 *  @author:             LIU, Yijia
 *  @data:               2013-09-24
 *
 * This program is special designed for UNIX user, for get time
 * is not compilable under MSVC
 */
#include <iostream>
#include <ctime>
#include <string>
#include <sys/time.h>
#include <sys/types.h>
#include "segment_dll.h"

double get_time(void) {
    struct timeval tv;
    gettimeofday(&tv, NULL);
    return tv.tv_sec + (tv.tv_usec / 1000000.0);
}

int main(int argc, char * argv[]) {
    if (argc < 2) {
        std::cerr << "cws [model path] [lexicon_file]" << std::endl;
        return 1;
    }

    void * engine = 0;
    if (argc == 2) {
        engine = segmentor_create_segmentor(argv[1]);
    } else if (argc == 3) {
        engine = segmentor_create_segmentor(argv[1], argv[2]);
    }

    if (!engine) {
        return -1;
    }
    std::vector<std::string> words;
    std::string sentence;

    std::cerr << "TRACE: Model is loaded" << std::endl;
    double tm = get_time();

    while (std::getline(std::cin, sentence, '\n')) {
        words.clear();
        if (sentence.size() == 0) { continue; }
        int len = segmentor_segment(engine, sentence, words);
        for (int i = 0; i < len; ++ i) {
            std::cout << words[i];
            if (i+1 == len) std::cout <<std::endl;
            else std::cout<< "|";
        }
    }

    segmentor_release_segmentor(engine);

    tm = get_time() - tm;
    std::cerr << "TRACE: consume "
        << tm 
        << " seconds."
        << std::endl;

    return 0;
}