/ani/mrses

#include <stdio.h>

#include <string.h>

#include <assert.h>

#include <libspe2.h>

#include "msg.h"

#include "mrses_spu.h"

int mrses_spu_run(HWThread thr __attribute__ ((unused)), spe_context_ptr_t hwctx __attribute__ ((unused)), int block __attribute__ ((unused)), MRSESContext mrses __attribute__ ((unused))) {

	// Have no sence to implement

    assert(0);

    return 0;

}

int mrses_spu_iterate(HWThread thr, spe_context_ptr_t hwctx, int block_group, MRSESContext mrses) {

    int err;

    int i;

    unsigned int *hist;

    MRSESIntType *index; 

    MRSESIntType *ires = mrses->ires;

    int iterate_size = mrses->iterate_size;

    int properties = mrses->properties;

    int width = mrses->width;

    int palloc = mrses->palloc;

    int block = block_group * iterate_size;

    int block_end = block + iterate_size;

    SPUParameters param __attribute__ ((aligned (16)));

    spe_stop_info_t stop_info;

	// We need this variable renewed before each execution of spe_context_run!!!

    unsigned int spe_entry = SPE_DEFAULT_ENTRY;

    //printf("running spu thread: %p\n", hwctx);

    if (thr->data) hist = (unsigned int*)(thr->data);

    else {

	posix_memalign((void*)&hist, HW_ALIGN, calc_alloc(properties * sizeof(uint32_t), HW_ALIGN));

	memset(hist, 0, calc_alloc(properties * sizeof(uint32_t), HW_ALIGN));

	if (hist) thr->data = hist;

	else return 1;

    }

    param.mrses = mrses;

    param.block_group = block_group;

    err = spe_context_run(hwctx, &spe_entry,  0, &param, NULL, &stop_info);

    if (err < 0) {

    	reportError ("Failed to run SPE program, error: %i", err);

	return 1;

    }

    if (stop_info.stop_reason !=  SPE_EXIT) {

    	reportError ("SPE program terminated with non-exit stop reason: %i", stop_info.stop_reason);

	return 1;

    }	

    err = stop_info.result.spe_exit_code;

    if (err) {

    	reportError ("SPE program terminated with error code: %i", err);

	return 1;

    }

    for (; block < block_end; ++block) { 

	index = mrses->index + block * palloc;

	if (ires) memcpy(ires + block * width, index, width * sizeof(MRSESIntType));

	for (i = 0; i < width; i++) {

	    hist[index[i]]++;

	}

    }

/*

    {

	int i;

	int iterate_size = mrses->iterate_size;

	MRSESIntType *index = mrses->index;

	mrses->result = malloc(iterate_size * (block_group + 1) * sizeof(MRSESDataType));

	mrses->index = malloc(mrses->width * iterate_size * (block_group + 1) * sizeof(MRSESIntType));

	for (i = 0; i < iterate_size; i++) {

	    memcpy(mrses->index + (iterate_size * block_group + i) * mrses->width, index + (iterate_size * block_group + i) * palloc, mrses->width * sizeof(MRSESIntType));

	}

	for (i = 0; i < iterate_size * mrses->width; i++) {

	    ++mrses->index[iterate_size * block_group * mrses->width + i];

	}

	float res, mahal, corcor;

	MRSESDistance dist = mrses->dist;

	void * data = thr->data;

	thr->data = NULL;

	for (i = 0; i < iterate_size; i++) {

	    mrses->dist = dist;

	    mrses_ppu_run(thr, hwctx, block_group * iterate_size + i, mrses);

	    res = mrses->result[block_group * iterate_size + i];

	    mrses->dist = MAHALANOBIS;

	    mrses_ppu_run(thr, hwctx, block_group * iterate_size + i, mrses);

	    mahal = mrses->result[block_group * iterate_size + i];

	    mrses->dist = CORCOR;

	    mrses_ppu_run(thr, hwctx, block_group * iterate_size + i, mrses);

	    corcor = mrses->result[block_group * iterate_size + i];

	    printf("PPU result, block %i: %e (mahal: %e, corcor: %e)\n", i, res, mahal, corcor);

	    int j;

	    for (j = 0; j < width; j++) {

		    printf("%i ", mrses->index[(iterate_size * block_group + i)*mrses->width + j]);

	    }

	    printf("\n");

	}

	free(thr->data);

	thr->data = data;

	free(mrses->index); mrses->index = index;

	free(mrses->result); mrses->result = NULL;

	mrses->dist = dist;

    }

*/

    return 0;

}