#include <error.h>
#include <errno.h>
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/ipc.h>
#include <sys/sem.h>
#include <sys/wait.h>
#include "dormirUnPeu.h"

// Nombre de trains dans le circuit
#define NB_TRAINS 3

// Nombre de tours de montagne russe que fait chaque train
#define NB_TOURS 5

// Cle utilisee pour les semaphores
#define CLE 42

int zoneE;
int zoneH;
int zoneA;

void P(int semId){
  struct sembuf op;
  op.sem_num =  0;
  op.sem_op  = -1;
  op.sem_flg =  0;
  if (semop(semId, &op, 1) == -1){
    perror("semop");
    exit(EXIT_FAILURE);
  }
}

void V(int semId){
  struct sembuf op = {0, 1, 0};
  if (semop(semId, &op, 1) == -1){
    perror("semop");
    exit(EXIT_FAILURE);
  }
}

void codeTrain(int numTrain){
  int nbTours;

  printf("Train %d : Je suis dans la zone zoneG\n", numTrain);
  dormirUnPeu();
  
  P(zoneE);

  for (nbTours=0 ; nbTours<NB_TOURS ; nbTours++){

    printf("Train %d : Je suis dans la zone zoneE\n", numTrain);
    dormirUnPeu();
    P(zoneH);
    V(zoneE);    

    printf("Train %d : Je suis dans la zone zoneH\n", numTrain);
    dormirUnPeu();
    P(zoneA);
    V(zoneH);

    printf("Train %d : Je suis dans la zone zoneA\n", numTrain);
    dormirUnPeu();
    P(zoneE);
    V(zoneA);
  }

  V(zoneE);
  printf("Train %d : J'ai fini de tourner\n", numTrain);
}

int main() {
  int numEnfant;
  int rc;

  /* Creation des sémaphores */
  zoneE = semget(CLE, 1, IPC_CREAT|0666);
  if (zoneE < 0)
    error_at_line(EXIT_FAILURE, errno, __FILE__, __LINE__, "semget");
  zoneH = semget(CLE+1, 1, IPC_CREAT|0666);
  if (zoneH < 0)
    error_at_line(EXIT_FAILURE, errno, __FILE__, __LINE__, "semget");
  zoneA = semget(CLE+2, 1, IPC_CREAT|0666);
  if (zoneA < 0)
    error_at_line(EXIT_FAILURE, errno, __FILE__, __LINE__, "semget");
  /* Intialisation des semaphores */
  if (semctl(zoneE, 0, SETVAL, NB_TRAINS - 1) < 0)
    error_at_line(EXIT_FAILURE, errno, __FILE__, __LINE__, "semctl");
  if (semctl(zoneH, 0, SETVAL, 1) < 0)
    error_at_line(EXIT_FAILURE, errno, __FILE__, __LINE__, "semctl");
  if (semctl(zoneA, 0, SETVAL, 1) < 0)
    error_at_line(EXIT_FAILURE, errno, __FILE__, __LINE__, "semctl");

  for (numEnfant=0; numEnfant<NB_TRAINS; numEnfant++){
    rc = fork();
    switch (rc) {
    case -1:
      error_at_line(EXIT_FAILURE, errno, __FILE__, __LINE__, "fork");
    case 0:
      codeTrain(numEnfant);
      exit(EXIT_SUCCESS);
    default:
      break;
    }
  }

  //On attend que tous les enfants aient fini de s'exécuter
  for (numEnfant=0; numEnfant<NB_TRAINS; numEnfant++){
    rc = wait(NULL);
    if (rc < 0)
      error_at_line(EXIT_FAILURE, rc, __FILE__, __LINE__, "wait");
  }

  // On detruit tous les objets de synchronisation
  if (semctl(zoneE, 0, IPC_RMID) < 0)
    error_at_line(EXIT_FAILURE, errno, __FILE__, __LINE__, "semctl");
  if (semctl(zoneH, 0, IPC_RMID) < 0)
    error_at_line(EXIT_FAILURE, errno, __FILE__, __LINE__, "semctl");
  if (semctl(zoneA, 0, IPC_RMID) < 0)
    error_at_line(EXIT_FAILURE, errno, __FILE__, __LINE__, "semctl");

  printf("Parent termine\n");

  exit(EXIT_SUCCESS);
}