#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <libspe2.h>
#include <pthread.h>

extern spe_program_handle_t SPU;

#define SPU_THREADS 8

void *ppu_pthread_function(void *arg){
   spe_context_ptr_t ctx;
   unsigned int entry = SPE_DEFAULT_ENTRY;
   ctx = *((spe_context_ptr_t *)arg);
   if (spe_context_run(ctx, &entry, 0, NULL, NULL, NULL) < 0){
      perror ("Failed running context");
      exit (1);
   }
   pthread_exit(NULL);
}

int main(){
   int i;
   spe_context_ptr_t ctxs[SPU_THREADS];
   pthread_t threads[SPU_THREADS];

   /* Create several SPE-threads to execute 'SPU'. */
   for(i=0; i<SPU_THREADS; i++){
      /* Create context */
      if ((ctxs[i] = spe_context_create (0, NULL)) == NULL){
         perror ("Failed creating context");
         exit (1);
      }
      /* Load program into context */
      if (spe_program_load (ctxs[i], &SPU)){
         perror ("Failed loading program");
         exit (1);
      }
      /* Create thread for each SPE context */
      if (pthread_create (&threads[i], NULL, &ppu_pthread_function, &ctxs[i])){
         perror ("Failed creating thread");
         exit (1);
      }
   }

   /* Wait for SPU-thread to complete execution. */
   for (i=0; i<SPU_THREADS; i++){
      if (pthread_join (threads[i], NULL)){
         perror("Failed pthread_join");
         exit (1);
      }
      /* Destroy context */
      if (spe_context_destroy (ctxs[i]) != 0){
         perror("Failed destroying context");
         exit (1);
      }
   }

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