StreamingVwFile.cpp

浏览该文件的文档.

/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * Written (W) 2011 Shashwat Lal Das
 * Copyright (C) 2011 Berlin Institute of Technology and Max-Planck-Society
 */

#include <shogun/io/streaming/StreamingVwFile.h>

using namespace shogun;

CStreamingVwFile::CStreamingVwFile()
    : CStreamingFile()
{
    init();
}

CStreamingVwFile::CStreamingVwFile(const char* fname, char rw)
    : CStreamingFile(fname, rw)
{
    init();
}

CStreamingVwFile::~CStreamingVwFile()
{
    SG_UNREF(env);
    SG_UNREF(parser);
}

void CStreamingVwFile::set_parser_type(E_VW_PARSER_TYPE type)
{
    switch (type)
    {
    case T_VW:
        parse_example = &CVwParser::read_features;
        parser_type = T_VW;
        return;
    case T_SVMLIGHT:
        parse_example = &CVwParser::read_svmlight_features;
        parser_type = T_SVMLIGHT;
        return;
    case T_DENSE:
        parse_example = &CVwParser::read_dense_features;
        parser_type = T_DENSE;
        return;
    }

    SG_SERROR("Unrecognized parser type!\n")
}

void CStreamingVwFile::get_vector(VwExample* &ex, int32_t &len)
{
    len = (parser->*parse_example)(buf, ex);
    if (len == 0)
        len = -1;   // indicates failure
}

void CStreamingVwFile::get_vector_and_label(VwExample* &ex, int32_t &len, float64_t &label)
{
    len = (parser->*parse_example)(buf, ex);
    if (len == 0)
        len = -1;   // indicates failure
}

void CStreamingVwFile::init()
{
    parser = new CVwParser();
    env = parser->get_env();

    set_parser_type(T_VW);
    write_to_cache = false;
    SG_REF(env);
}