NICE_VisLearning/cbaselib/Example.cpp

#include "Example.h"

#ifdef NICE_USELIB_OPENMP
#include <omp.h>
#endif

using namespace OBJREC;
using namespace std;
using namespace NICE;

Example::Example()
{
    weight = 0.0;
    height = 0;
    depth = 0;
    width = 0;
    x = y = z = 0;
    vec = NULL;
    svec = NULL;
    ce = NULL;
    position = 0;
}

Example::~Example ()
{
}

Example::Example ( CachedExample *_ce )
{
    if ( _ce->getNumImages() > 1 )
    {
        _ce->getImageSize3 ( width, height, depth );
    }
    else
    {
        _ce->getImageSize( width, height );
        depth = 1;
    }

    if ( width % 2 == 0 ) width--;

    if ( height % 2 == 0 ) height--;

    if ( depth % 2 == 0 ) depth--;

    x = width/2;
    y = height/2;
    z = depth/2;

    ce = _ce;
    vec = NULL;
    svec = NULL;
    weight = 1.0;
    position = 0;
}

Example::Example ( CachedExample *_ce,
                   int _x,
                   int _y,
                   double _weight )
{
    ce = _ce;
    x  = _x;
    y  = _y;
    z  = 0;
    width = 0;
    height = 0;
    depth = 0;
    vec = NULL;
    svec = NULL;
    weight = _weight;
    position = 0;
}

Example::Example ( CachedExample *_ce,
                   int _x,
                   int _y,
                   int _z,
                   double _weight )
{
    ce = _ce;
    x  = _x;
    y  = _y;
    z  = _z;
    width = 0;
    height = 0;
    depth = 0;
    vec = NULL;
    svec = NULL;
    weight = _weight;
    position = 0;
}

Example::Example ( CachedExample *_ce,
                   int _x,
                   int _y,
                   int _width,
                   int _height,
                   double _weight )
{
    ce = _ce;
    x  = _x;
    y  = _y;
    z  = 0;
    width = _width;
    height = _height;
    depth = 1;
    assert ( (width > 0) && (height > 0) );

    vec = NULL;
    svec = NULL;
    weight = _weight;
    position = 0;
}

Example::Example ( CachedExample *_ce,
                   int _x,
                   int _y,
                   int _z,
                   int _width,
                   int _height,
                   int _depth,
                   double _weight )
{
    ce = _ce;
    x  = _x;
    y  = _y;
    z  = _z;
    width = _width;
    height = _height;
    depth = _depth;
    assert ( (width > 0) && (height > 0) && assert(depth > 0) );

    vec = NULL;
    svec = NULL;
    weight = _weight;
    position = 0;
}

Example::Example ( NICE::Vector *_vec,
                   double _weight )
{
    x = y = z = 0;
    width = height = depth = 0;
    ce = NULL;
    vec = _vec;
    svec = NULL;
    weight = _weight;
    position = 0;
}

void Example::clean ()
{
    if ( ce != NULL )
    {
        delete ce;
        ce =  NULL;
    }

    if ( vec != NULL )
    {
        delete vec;
        vec = NULL;
    }

    if ( svec != NULL )
    {
        delete svec;
        svec = NULL;
    }
}


Example::Example ( const Example &ex)
{
	copy(ex);
}

void Example::copy ( const Example &ex)
{
    vec = ex.vec;
    svec = ex.svec;
    weight = ex.weight;
    position = ex.position;
    width = ex.width;
    height = ex.height;
    depth = ex.depth;
    x = ex.x;
    y = ex.y;
    z = ex.z;
    ce = ex.ce;
    scale = ex.scale;
}

void Example::restore (istream & is, int format)
{
    if ( format == 1 )
    {
        is >> weight;
        is >> x;
        is >> y;
        is >> z;
        is >> width;
        is >> height;
        is >> depth;
        is >> position;
    }
    else
    {
        is >> weight;
        is >> x;
        is >> y;
        is >> width;
        is >> height;
        is >> position;

        depth = 1;
        z = 0;
    }

    int tmp;
    is >> tmp;

    if(tmp == 1)
    {
        svec = new SparseVector();
        svec->restore(is);
    }
    else
	svec = NULL;

    is >> tmp;
    if(tmp >= 0 )
    {
        vec = new Vector(tmp);
        for(int i = 0; i < tmp; i++)
        {
            is >> vec[i];
        }
    }
    else
        vec = NULL;
}

void Example::store (ostream & os, int format) const
{
    if ( format == 1 )
        os << weight << " " <<  x << " " << y << " " << z << " " << width << " " << height << " " << depth << " " << position << endl;
    else
        os << weight << " " <<  x << " " << y << " " << width << " " << height << " " << position << endl;

    if( svec == NULL )
        os << 0 << endl;
    else
    {
        os << 1 << endl;
        svec->store(os);
    }

    if(vec == NULL)
        os << -1 << endl;
    else
    {
        os << vec->size() << endl;
        for(int i = 0; i < (int)vec->size(); i++)
        {
            os << vec[i] << " ";
        }
    }
}

void Examples::clean ()
{
    for ( iterator i = begin(); i != end(); i++ )
    {
        Example & example = i->second;
        example.clean();
    }
    clear();
}

bool Examples::wrapExamplesAroundFeatureMatrix(const Matrix &p_MatFeaturesColumWiseSamples, const Vector &p_VecLabels, Examples &p_Examples)
{
    size_t t_iNumSamples  = p_MatFeaturesColumWiseSamples.cols();
    size_t t_iNumFeatures = p_MatFeaturesColumWiseSamples.rows();

    if(p_VecLabels.size() != t_iNumSamples) // for every columnwise sample there need to be a label
        return false;

    p_Examples.reserve( t_iNumSamples );

    const double *pDataPtr = p_MatFeaturesColumWiseSamples.getDataPointer();
    for (size_t i = 0; i < t_iNumSamples; i++, pDataPtr+= t_iNumFeatures )
    {
        NICE::Vector *t_pVecTrainData = new NICE::Vector( pDataPtr , t_iNumFeatures);
        double t_fWeight = 1.0f;
        OBJREC::Example t_Example(t_pVecTrainData, t_fWeight);

        p_Examples.push_back( std::pair<int, OBJREC::Example>( (int)p_VecLabels[i], t_Example ) );
    }

    return true;
}

bool Examples::wrapExamplesAroundFeatureMatrix(const Matrix &p_MatFeaturesColumWiseSamples, const VectorT<int> &p_VecLabels, Examples &p_Examples)
{
    size_t t_iNumSamples  = p_MatFeaturesColumWiseSamples.cols();
    size_t t_iNumFeatures = p_MatFeaturesColumWiseSamples.rows();

    if(p_VecLabels.size() != t_iNumSamples) // for every columnwise sample there need to be a label
        return false;
    p_Examples.resize( t_iNumSamples );
    const double *pDataPtr = p_MatFeaturesColumWiseSamples.getDataPointer();

#ifdef NICE_USELIB_OPENMP
#pragma omp parallel for default(none) shared(p_VecLabels, p_Examples, t_iNumFeatures, t_iNumSamples, pDataPtr)
#endif
    for (size_t i = 0; i < t_iNumSamples; i++)
    {
	const double *pDataIteration = pDataPtr + (i * t_iNumFeatures);
        NICE::Vector *t_pVecTrainData = new NICE::Vector( (double*) pDataIteration , t_iNumFeatures, VectorBase::external);
        // OBJREC::Example t_Example(t_pVecTrainData, t_fWeight);

	p_Examples[i] = std::pair<int, OBJREC::Example>( p_VecLabels[i], OBJREC::Example(t_pVecTrainData, 1.0) ) ;
    }

    return true;
}