/* author: Guochun Shi <gshi@ncsa.uiuc.edu>
 */

#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <libspe.h>
#include "spu/speinfo.h"
#include <sched.h>

extern spe_program_handle_t simple_spu;

#define SPU_THREADS 	8

int main(int argc, char **argv)
{
    speid_t spe_ids[SPU_THREADS];
    spe_gid_t gid;    
    int i,  status = 0;
    struct spe_info si[SPU_THREADS] __attribute__ ((aligned (128)));
    struct spe_event e =  {
	.events = SPE_EVENT_MAILBOX,
    };
    int ret; 
    unsigned long msgdata;

    (void)argc;
    (void)argv;

    /*
    // spe_get_event does not work with default group,  bug?
    if (spe_group_defaults(SCHED_OTHER, 1, 1) != 0){	
	fprintf(stderr, "Chaning group's default behavor failed\n");	
	perror("ERROR:");	
	exit(1);
	
    }
    */

    gid = spe_create_group(SCHED_RR, 1, 1);
    if (gid <= 0){
	perror("Creating new group failed\n");
	return 1;	
    }
    

    for(i=0; i<SPU_THREADS; i++){
	spe_ids[i] = spe_create_thread(gid, &simple_spu, &si[0]  , NULL, -1, SPE_MAP_PS);
	if (spe_ids[i] == 0) {
	    fprintf(stderr, "Failed spu_create_thread(rc=%p, errno=%d)\n", 
		    spe_ids[i], errno);
	    exit(1);
	}
	
    }
    
    /* fill the spe_info struct*/
    for (i = 0; i < SPU_THREADS; i++){

	si[i].rank = i;
	si[i].speid = (unsigned long)spe_ids[i];
	
	if ((si[i].sig1 = spe_get_ps_area(spe_ids[i], SPE_SIG_NOTIFY_1_AREA)) == NULL){
	    printf("ERROR: spe_get_ps_area failed for sig1(%d)\n", errno);
	    return -1;
	}
	
	
	if ( (si[i].ls = spe_get_ls(spe_ids[i])) == NULL){
	    fprintf(stderr, "ERROR: get ls return NULL");
	    return -1;
	}	
    }
    
    for (i = 0; i < SPU_THREADS; i++){
	if (spe_write_in_mbox(spe_ids[i], SPU_THREADS) < 0
	    || spe_write_in_mbox(spe_ids[i], si[i].rank) < 0){
	    fprintf(stderr, "ERROR: writing messages to spe failed\n");
	    exit(1);
	}
    }
    
    
    for ( i = 0 ; i < SPU_THREADS; i++){
	e.gid = gid;	
	ret = spe_get_event(&e, 1, -1);
	if (!ret ){
	    fprintf(stderr, "Got zero event\n");
	    exit(1);
	}
	
	if (e.revents !=  SPE_EVENT_MAILBOX){
	    fprintf(stderr, "spe_get_event:  unexpected event%d\n",
		    e.revents);
	    exit(1);
	}
	
	msgdata = e.data;	
	printf("speid=0x%lx, msgdata=%lu\n", (unsigned long)e.speid, msgdata);
    }
    
    /* Wait for SPU-thread to complete execution.
     */
    for (i=0; i<SPU_THREADS; i++) {
	(void)spe_wait(spe_ids[i], &status, 0);
    }

    printf("\nThe program has successfully executed.\n");

    return (0);
}