#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>
#include "putNget.h"
#include "sem_ops.h"
#define NUM_CONS 7
#define NUM_PROD 9

int full, empty;

void *producer(void * arg1){ 
  int *i = (int *) arg1;
  /* acquire right to a slot */
  P( empty ); /* wait until some are empty */
  printf("Producer %d running\n", *i );
  put_item( (*i) * (*i) );
  V( full ); /* increment full count */
  pthread_exit(NULL);
}

void *consumer(void *arg1){
  int *i = (int *) arg1;
  int *thisitem = (int *)malloc( sizeof(int));
  /* wait until a slot is full */
  P( full );
  get_item(thisitem);
  printf("Consumer %d got %d. \n", *i, *thisitem );
  V( empty ); /* increment empty count */
  pthread_exit(thisitem);
}

int max( int i, int j ){
  if( i >= j ) return i;
  return j;
}
int main(void){
  pthread_t prodtid[NUM_PROD], constid[NUM_CONS];
  int slots, i, id[NUM_PROD]; int *returned_item_ptr;

  empty = semtran( IPC_PRIVATE );
  full = semtran( IPC_PRIVATE );
  slots = get_buffersize();
  for( i = 0; i < slots; i++ )
    V( empty ); /* set empty = BUFSIZE, full = 0 */

  for( i=0; i<max(NUM_PROD,NUM_CONS); i++)
    id[i] = i; 
  printf( "The buffer has %d slots. %d producers. %d consumers\n", 
                         slots, NUM_PROD, NUM_CONS );
  for( i=0; i<NUM_PROD; i++) {
    if (pthread_create(&prodtid[i], NULL, producer, (void *) &id[i])){
      perror("Could not create producer");
      exit(EXIT_FAILURE);
    }
  }
  for( i=0; i<NUM_CONS; i++) {
    if (pthread_create(&constid[i], NULL, consumer, (void *) &id[i])){
      perror("Could not create consumer");
      exit(EXIT_FAILURE);
    }
  }
  for( i=0; i<NUM_PROD; i++)
    pthread_join(prodtid[i], NULL);
  printf( "All producers finished.\n" );

  for( i=0; i<NUM_CONS; i++) {
    pthread_join(constid[i], (void **) &returned_item_ptr );
//    printf( "Consumer returned %d\n", *returned_item_ptr );
    }
  rm_sem( empty );
  rm_sem( full );
  exit(EXIT_SUCCESS);
  return 0;
}