StreamingFileFromDenseFeatures.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_FILEFROMDENSE_H__
#define __STREAMING_FILEFROMDENSE_H__

#include <shogun/io/streaming/StreamingFileFromFeatures.h>
#include <shogun/features/DenseFeatures.h>

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

    CStreamingFileFromDenseFeatures(CDenseFeatures<T>* feat,
            float64_t* lab=NULL);

    virtual ~CStreamingFileFromDenseFeatures();

    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;
    }

    virtual const char* get_name() const
    {
        return "StreamingFileFromDenseFeatures";

    }

private:
    void init();

protected:

    CDenseFeatures<T>* features;

    int32_t vector_num;

};

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

template<class T>
CStreamingFileFromDenseFeatures<T>::CStreamingFileFromDenseFeatures(
        CDenseFeatures<T>* feat, float64_t* lab) :
        CStreamingFileFromFeatures()
{
    init();

    REQUIRE(feat,"%s::CStreamingFileFromDenseFeatures() features required!\n",
            get_name());
    features=feat;
    SG_REF(feat);

    labels=lab;

}

template<class T>
CStreamingFileFromDenseFeatures<T>::~CStreamingFileFromDenseFeatures()
{
    SG_UNREF(features);
}

template<class T>
void CStreamingFileFromDenseFeatures<T>::init()
{
    vector_num=0;
    features=NULL;

    SG_WARNING("%s::init(): register params!\n", get_name());
}

/* Functions to return the vector from the DenseFeatures object
 * If the class is of type T, specialize this function to work for
 * vectors of that type. */
template<class T>
void CStreamingFileFromDenseFeatures<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 DenseFeatures object with label */
template<class T>
void CStreamingFileFromDenseFeatures<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_FILEFROMDENSE_H__