#include "FIGradients.h"

#include "vislearning/baselib/FastFilter.h"
#include "vislearning/image/GenericImageTools.h"

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

void FIGradients::buildEOHMap ( CachedExample *ce,
				int subsamplex, int subsampley,
				int numBins, bool usesigned )
{
    int xsize;
    int ysize;
    ce->getImageSize ( xsize, ysize );
    int xsize_s = xsize / subsamplex;
    int ysize_s = ysize / subsampley;

    NICE::MultiChannelImageT<double> & eohimg = ce->getDChannel ( CachedExample::D_EOH );
    eohimg.reInit ( xsize_s, ysize_s, numBins, true );
    
    double *gradient = new double[xsize*ysize];
    int *dir = new int[xsize*ysize];

    if ( ce->colorInformationAvailable() ) {
	NICE::MultiChannelImageT<int> & colorimg = ce->getIChannel ( CachedExample::I_COLOR );
	const int *r = colorimg.data[0];
	const int *g = colorimg.data[1];
	const int *b = colorimg.data[2];
	FastFilter::calcColorGradient ( r,g,b,xsize,ysize,
			    gradient, dir, numBins, usesigned );
    } else {
	NICE::MultiChannelImageT<int> & grayvalues = ce->getIChannel ( CachedExample::I_GRAYVALUES );
	const int *gr = grayvalues.data[0];
	FastFilter::calcGradient ( gr, xsize, ysize, gradient, dir, numBins, usesigned );
    }

    eohimg.setAll ( 0 );

    long korig = 0;
    for ( int y = 0 ; y < ysize ; y++ )
	for ( int x = 0 ; x < xsize ; x++,korig++ )
	{
	    int xs = x / subsamplex;
	    int ys = y / subsampley;

	    if ( xs >= xsize_s ) xs = xsize_s-1;
	    if ( xs < 0 ) xs = 0;
	    if ( ys >= ysize_s ) ys = ysize_s-1;
	    if ( ys < 0 ) ys = 0;
	    int k = xs + ys*xsize_s;
	    int val = dir[korig];
	    double strength = gradient[korig];

	    assert ( val < eohimg.numChannels );
	    eohimg.data[val][k] += strength;

	    if ( !finite(eohimg.data[val][k]) ) {
		fprintf (stderr, "EOH Image failed: %f\n", eohimg.data[val][k]);
		exit(-1);
	    }
	}

    delete [] gradient;
    delete [] dir;

    NICE::MultiChannelImageT<double> & eohintimg = ce->getDChannel ( CachedExample::D_INTEGRALEOH );
    eohintimg.reInit ( xsize_s, ysize_s, numBins, true );
    for ( uint i = 0 ; i < (uint)numBins ; i++ )
	GenericImageTools::calcIntegralImage ( eohintimg.data[i], eohimg.data[i], xsize_s, ysize_s );

}