SOURCE FILE: diamond.cpp



/*  Diamond Formation Program   */

#include <iostream>
#include <sched.h>
#include <time.h>
#include <pthread.h>
#include "sem.h"

#define THREAD_COUNT 9

using namespace std;

extern sim_semaphore create_sim_sem(int) ;
extern void wait_sem (sim_semaphore) ;
extern void signal_sem (sim_semaphore) ;

      /* For technical reasons, we use the pthread mutex below,
         instead of a semaphore, to lock the screen for output.  Don't
         change this.  */

pthread_mutex_t stdoutLock ;

     /* Here (between the lines) declare whatever shared 
        variables you need for 
        synchronization - such as semaphores, flags, counters, 
        and so forth. Variables declared here will be visible 
        to (shared by) all the threads in the task. */
   /* HHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHH */ 


   /* HHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHH */ 

     /* These are global variable to represent threads created
        dynamically. */
pthread_t thr[THREAD_COUNT] ; 

    /* This is included to facilitate adding random delays in the code
       -- as a debugging aid. */
extern long random(void);  

   /* This can be changed to 1, but the resulting diagnostic output
      will probably seem excessive. */
int checking = 0 ;

     /* A data type - a struct (class) with an int field to represent
        a thread ID. */
struct threadIdType
{
  int id ;
};

/* ################################################## */
/*                         init                       */
/* ################################################## */
void init() 
{ 
      /* This code initializes special mutex lock for screen output.
         Just leave this alone. */
  if ( 0!=pthread_mutex_init(&stdoutLock, NULL) )
  {  cout << "MUTEX INITIALIZATION FAILURE!" << endl ;
     exit(-1) ;}

    /* Between the lines below, insert the code you want to 
       initialize the shared variables you are using for 
       synchronization - like semaphores, flags,
       counters, and so forth. */
   /* HHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHH */

   /* HHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHH */

       /* This initializes a random number generator */ 
  srandom(time((time_t *) 0));
}

/* ################################################## */
/*                     rand_delay                     */
/* ################################################## */
void rand_delay(int max) 
{
  int m_delay, j ;
  m_delay = (int) random()%max ;
  for (j=0; j<m_delay; j++) sched_yield();  
}

/* ################################################## */
/*                childMessage                        */
/* ################################################## */
void childMessage (int ID)
{
   pthread_mutex_lock(&stdoutLock) ;
   cout << "Child # " << ID 
        << " is now in position.\n" ;
   pthread_mutex_unlock(&stdoutLock) ;
}

/* ################################################## */
/*                        child                       */
/* ################################################## */

void * child(void * idPtr) 
{
       /* This is just a change of data type for convenience.  Now
          'me' is the number of the child.  Children have numbers from
          0 to 8. */
   int me = ((threadIdType *) (idPtr))->id, temp ;

       /* Delay code inserted here to magnify the chances that child
          threads will delay each other.  The student
          completing the program should think of more ways to insert
          random delays that are likely to 'stress test' the program. */

   rand_delay(100) ;

       /* You need to put some of the synchronization code 
          between the lines, before the statement further down,
          where the child process declares it has
          put itself into position. */

   /* HHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHH */
 
   /* HHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHH */

   childMessage(me) ;  // get into position
  
      /* Depending on what amount of synchronization code you placed
         earlier in this function, you may need some more between
         these lines, before the child exits (and 'dies'). */

   /* HHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHH */ 

   /* HHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHH */ 

   pthread_exit ((void *)0) ;
}

/* ################################################## */
/*                       mother                       */
/* ################################################## */

/* The mother spawns child threads and then waits for
   them all to finish.  The mother's waiting has to be
   implemented as part of the overall protocol - using
   the sim_semaphore data type. */

void mother() 
{ 
   int i; 

       /* This is a pointer to a struct (class) that contains an int
          field - it is a convenient data type to use as the parameter
          to the child function.  */
   threadIdType * idPtr ; 

        /* Create child threads numbered 1 through 8.  The mother
           will play the role of position #9 */

   pthread_mutex_lock(&stdoutLock) ;
   cout << "Mother Begins Spawning Threads.\n"   << endl ;
   pthread_mutex_unlock(&stdoutLock) ;

   for (i = 0; i < THREAD_COUNT ; i++) 
   {
      idPtr = new threadIdType ; /* allocate memory for struct */
      idPtr->id = i ;  /* records current index as the child's ID */

         /* The call below is what actually creates the child thread
            and passes a pointer to the struct 'idPtr' as the
            parameter to the child function. */

      if ( 0!=pthread_create(&thr[i], NULL, child, (void *) idPtr) )
      {  pthread_mutex_lock(&stdoutLock) ; /* 'error out' code here */
         cout << "THREAD CREATION FAILURE!" << endl ;
         pthread_mutex_unlock(&stdoutLock) ;
         exit(-1) ; }

         /* The call to pthread_detach() below marks a child 
            thread as 'detached' so that the system will not 
            expect the parent to 'join' (wait for) it. */     

      if (0!=pthread_detach(thr[i]))
      {  pthread_mutex_lock(&stdoutLock) ;/* 'error out' code here */
         cout << "THREAD DETACHMENT FAILURE!" << endl ;
         pthread_mutex_unlock(&stdoutLock) ;
         exit(-1) ;}
   }

   pthread_mutex_lock(&stdoutLock) ;
   cout << "Mother Finishes Spawning Threads.\n"   << endl ;
   pthread_mutex_unlock(&stdoutLock) ;

      /* Some protocol code is needed here to make the mother behave 
         correctly - she is not supposed to return to main and
         exit until all the children have taken their positions. */

      /* HHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHH */ 

      /* HHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHHH */ 

   pthread_mutex_lock(&stdoutLock) ;
   cout << "\nAll children are now in position.\n" ;
   cout << "Mother exiting ... \n\n" ;
   pthread_mutex_unlock(&stdoutLock) ;
   return ;
}

/* ################################################## */
/*                         main                       */
/* ################################################## */

int main() 
{ 

   cout << "\nWelcome to Diamond Heights!\n\n" ;

     /* This calls the function that performs initializations. */
  init(); 

        /* Execute the mother() function */ 
  mother();

  return 0 ;
}