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

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

extern spe_program_handle_t simple_spu;

#define SPU_THREADS 	2

int main(int argc, char **argv)
{
    speid_t spe_ids[SPU_THREADS];
    int i,  status = 0;
    struct spe_info si[SPU_THREADS] __attribute__ ((aligned (128)));
    
    (void)argc;
    (void)argv;

    
    for(i=0; i<SPU_THREADS; i++){
      spe_ids[i] = spe_create_thread(0, &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].control_area = spe_get_ps_area(spe_ids[i], SPE_CONTROL_AREA)) == NULL){
	    printf("ERROR: spe_get_ps_area failed for control area(%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);
	}
    }
    
    /* 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);
}