StreamingFileFromSimpleFeatures.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) 2011 Shashwat Lal Das
 * Copyright (C) 2011 Berlin Institute of Technology and Max-Planck-Society
 */
#ifndef __STREAMING_FILEFROMSIMPLE_H__
#define __STREAMING_FILEFROMSIMPLE_H__

#include <shogun/io/StreamingFileFromFeatures.h>
#include <shogun/features/SimpleFeatures.h>

namespace shogun
{
template <class T> class CStreamingFileFromSimpleFeatures: public CStreamingFileFromFeatures
{
public:
    CStreamingFileFromSimpleFeatures();

    CStreamingFileFromSimpleFeatures(CSimpleFeatures<T>* feat);

    CStreamingFileFromSimpleFeatures(CSimpleFeatures<T>* feat, float64_t* lab);

    virtual ~CStreamingFileFromSimpleFeatures();

    virtual void get_vector(T* &vec, int32_t &len);

    virtual void get_vector_and_label(T* &vec, int32_t &len, float64_t &label);

    void reset_stream()
    {
        vector_num = 0;
    }

    inline virtual const char* get_name() const
    {
        return "StreamingFileFromSimpleFeatures";

    }

private:
    void init();

protected:

    CSimpleFeatures<T>* features;

    int32_t vector_num;

};

template <class T>
CStreamingFileFromSimpleFeatures<T>::CStreamingFileFromSimpleFeatures()
    : CStreamingFileFromFeatures()
{
    init();
}

template <class T>
CStreamingFileFromSimpleFeatures<T>::CStreamingFileFromSimpleFeatures(CSimpleFeatures<T>* feat)
    : CStreamingFileFromFeatures()
{
    ASSERT(feat);
    features=feat;

    init();
}

template <class T>
CStreamingFileFromSimpleFeatures<T>::CStreamingFileFromSimpleFeatures(CSimpleFeatures<T>* feat, float64_t* lab)
    : CStreamingFileFromFeatures()
{
    ASSERT(feat);
    ASSERT(lab);
    features=feat;
    labels=lab;

    init();
}

template <class T>
CStreamingFileFromSimpleFeatures<T>::~CStreamingFileFromSimpleFeatures()
{
}

template <class T>
void CStreamingFileFromSimpleFeatures<T>::init()
{
    vector_num=0;
}

/* Functions to return the vector from the SimpleFeatures object
 * If the class is of type T, specialize this function to work for
 * vectors of that type. */
template <class T>
void CStreamingFileFromSimpleFeatures<T>::get_vector(T*& vector, int32_t& num_feat)
{
    if (vector_num >= features->get_num_vectors())
    {
        vector=NULL;
        num_feat=-1;
        return;
    }

    SGVector<T> sg_vector=
        features->get_feature_vector(vector_num);

    vector = sg_vector.vector;
    num_feat = sg_vector.vlen;;
    vector_num++;

}

/* Functions to return the vector from the SimpleFeatures object with label */
template <class T>
void CStreamingFileFromSimpleFeatures<T>::get_vector_and_label
(T*& vector, int32_t& num_feat, float64_t& label)
{
    if (vector_num >= features->get_num_vectors())
    {
        vector=NULL;
        num_feat=-1;
        return;
    }

    SGVector<T> sg_vector
        =features->get_feature_vector(vector_num);

    vector = sg_vector.vector;
    num_feat = sg_vector.vlen;
    label = labels[vector_num];

    vector_num++;
}

}
#endif //__STREAMING_FILEFROMSIMPLE_H__