UAIFile.h

Go to the documentation of this file.

/*
 * 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) 2014 Abinash Panda
 */

#ifndef __UAIFILE_H__
#define __UAIFILE_H__

#include <shogun/io/File.h>

#include <shogun/io/LineReader.h>
#include <shogun/io/Parser.h>
#include <shogun/lib/DelimiterTokenizer.h>
#include <shogun/base/Parameter.h>

namespace shogun
{

class CUAIFile : public CFile
{
public:
    CUAIFile();

    CUAIFile(FILE* f, const char* name=NULL);

#ifdef HAVE_FDOPEN

    CUAIFile(int fd, const char* mode, const char* name=NULL);
#endif

    CUAIFile(const char* fname, char rw='r', const char* name=NULL);

    virtual ~CUAIFile();

    virtual void parse();

    virtual void set_net_type(const char* net_type);

    virtual void set_num_vars(int32_t num_vars);

    virtual void set_num_factors(int32_t num_vars);

    virtual void set_vars_card(SGVector<int32_t> vars_card);

    virtual void set_factors_scope(int32_t num_factors,
                                   const SGVector<int32_t>* factors_scope);

    virtual void set_factors_table(int32_t num_factors,
                                   const SGVector<float64_t>* factors_table);

    virtual void get_preamble(SGVector<char>& net_type,
                              int32_t& num_vars,
                              SGVector<int32_t>& vars_card,
                              int32_t& num_factors,
                              SGVector<int32_t>*& factors_scope);

    virtual void get_factors_table(SGVector<float64_t>*& factors_table);

#ifndef SWIG // SWIG should skip this

    virtual void get_vector(int8_t*& vector, int32_t& len);
    virtual void get_vector(uint8_t*& vector, int32_t& len);
    virtual void get_vector(char*& vector, int32_t& len);
    virtual void get_vector(int32_t*& vector, int32_t& len);
    virtual void get_vector(uint32_t*& vector, int32_t& len);
    virtual void get_vector(float64_t*& vector, int32_t& len);
    virtual void get_vector(float32_t*& vector, int32_t& len);
    virtual void get_vector(floatmax_t*& vector, int32_t& len);
    virtual void get_vector(int16_t*& vector, int32_t& len);
    virtual void get_vector(uint16_t*& vector, int32_t& len);
    virtual void get_vector(int64_t*& vector, int32_t& len);
    virtual void get_vector(uint64_t*& vector, int32_t& len);

    /*virtual void get_vector(void*& vector, int32_t& len, DataType& dtype);*/

    virtual void set_vector(const int8_t* vector, int32_t len);
    virtual void set_vector(const uint8_t* vector, int32_t len);
    virtual void set_vector(const char* vector, int32_t len);
    virtual void set_vector(const int32_t* vector, int32_t len);
    virtual void set_vector(const uint32_t* vector, int32_t len);
    virtual void set_vector(const float32_t* vector, int32_t len);
    virtual void set_vector(const float64_t* vector, int32_t len);
    virtual void set_vector(const floatmax_t* vector, int32_t len);
    virtual void set_vector(const int16_t* vector, int32_t len);
    virtual void set_vector(const uint16_t* vector, int32_t len);
    virtual void set_vector(const int64_t* vector, int32_t len);
    virtual void set_vector(const uint64_t* vector, int32_t len);

#endif // #ifndef SWIG // SWIG should skip this

    virtual const char* get_name() const { return "UAIFile"; }

private:
    void init();

    void init_with_defaults();

protected:
    CLineReader* m_line_reader;

    CParser* m_parser;

    CDelimiterTokenizer* m_line_tokenizer;

    CDelimiterTokenizer* m_tokenizer;

    char m_delimiter;

    int32_t m_num_vars;

    int32_t m_num_factors;

    SGVector<char> m_net_type;

    SGVector<int32_t> m_vars_card;

    SGVector<int32_t>* m_factors_scope;

    SGVector<float64_t>* m_factors_table;
};

}

#endif