#include	<sys/types.h>
#include	<sys/ipc.h>
#include	<sys/sem.h>
#include	<sys/shm.h>
#include	<stdio.h>
#include	<unistd.h>

#define	DEFAULT_N		(15+3)	/*沒有參數時，缺省為15個顧客和3個理發師*/

/*宏定義wait和signal*/
#define SIGNAL(A,B)			V.sem_num = (A+B);if(semop(semid,&V,1)==-1) { perror("signal fail");exit(1);}
					
#define WAIT(A,B)			P.sem_num = (A+B);if(semop(semid,&P,1)==-1) { perror("wait fail");exit(1);}
				

union	semun
{
	int		val;/*for SETVAL*/
	struct semid_ds *buf;/*for IPC_STAT and IPC_SET*/
	ushort		*array;/*for GETALL and SETALL*/
};

enum
{
	MAX_CAPACITY=0,
	SOFA,
	BARBER_CHAIR,
	CASHIER=5,
	MUTEX,
	CUST_READY=7,
	LEAVE_B_CHAIR=10,
	PAYMENT=13,
	RECEIPT=16,
	FINISHED=19,
	queueLOCK=22
};

static struct	sembuf P = {0,-1,SEM_UNDO},V = {0,1,SEM_UNDO};
			/******wait和signal操作******/
static ushort	start_val[23] = {20,4,1,1,1,1,3,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,1};
			/******23個信號量的初值******/
static int		semid,shmid;/*信號量ID和共享內存ID*/

struct QueueType
{
	int	value[4];
	int out;
	int	in;
};


void	enqueue(struct QueueType* q,int val )/*入隊列*/
{
	if((q->in+1)%4==q->out)
	{
		printf("queue is full\n");
		return;
	}
	else
	{
		q->value[q->in] = val;
		q->in = (q->in+1)%4;
		printf("Barber %d(his pid is %d) is ready ,he wants to send No.%d to a customer\n",val,getpid(),val);
	}
			
}
void	dequeue(struct QueueType* q,int *val,int i)/*出隊列*/
{
	if(q->out==q->in)
	{
		printf("queue is empty\n");
		return;
	}
	else{
		*val = q->value[q->out];
		q->out = (q->out+1)%4;
		printf("Customer %d has got the No.%d\n",i,*val);
	}
}
void	enter_shop(int j)
{
	printf("Customer %d enter the barber shop.\n",j);
}

void	sit_on_sofa(int j)
{
	printf("Customer %d sit on the sofa.\n",j);
}

void	get_up_from_sofa(int j)
{
	printf("Customer %d get up from sofa.\n",j);
}

void	sit_in_barber_chair(int j)
{
	printf("Customer %d sit in the barber chair.\n",j);
}

void	leave_barber_chair(int j)
{
	printf("Customer %d leave the barber chair.\n",j);
}

void	pay(int  j)
{
	printf("Customer %d pay the money.\n",j);
}

void	exit_shop(int j)
{
	printf("Customer %d exit shop.\n",j);
}

void	accept_pay(int bnum)
{
	printf("Barber %d accept the payment .\n",bnum);
	printf("Barber %d has given the receipt .\n",bnum);
}

void	cut_hair(int bnum)
{
	sleep(0.5);
	printf("Barber %d is cutting hair.\n",bnum);
	printf("Barber %d has finished cutting hair.\n",bnum);
}

void	customer(struct QueueType *queue,int i)
{
	int chairnum;/*顧客取得的椅子號碼*/
	WAIT(MAX_CAPACITY,0);/*顧客等待進門*/
	enter_shop(i);/*進門*/
	WAIT(SOFA,0);/*等待沙發*/
	sit_on_sofa(i);/*顧客在沙發上等待*/
	WAIT(MUTEX,0);/***有空閑的理發師？***/
	WAIT(queueLOCK,0);/*確保顧客從等待隊列中取號碼不被打斷*/
		//WAIT(MUTEX,0);/***有空閑的理發師？***/
		dequeue(queue,&chairnum,i);/*顧客從隊列中取得椅子號碼*/
	SIGNAL(queueLOCK,0);/*確保顧客從等待隊列中取號碼不被打斷*/
	WAIT(BARBER_CHAIR,chairnum);/*顧客等待自己號碼的理發椅*/
	get_up_from_sofa(i);/*顧客離開沙發*/
	SIGNAL(SOFA,0);/*釋放沙發資源*/
	sit_in_barber_chair(i);/*顧客坐上理發椅*/
	SIGNAL(CUST_READY,chairnum);/*顧客告之已在自己號碼的理發椅上做好準備*/
	WAIT(FINISHED,chairnum);/*顧客等待理發完成*/
	leave_barber_chair(i);/*顧客離開理發椅*/
	SIGNAL(LEAVE_B_CHAIR,chairnum);/*釋放理發椅資源*/
	//WAIT(CASHIER,0);/*等待收銀員*/
	pay(i);/*付帳*/
	SIGNAL(PAYMENT,chairnum);/*顧客索要自己的收據*/
	WAIT(RECEIPT,chairnum);/*顧客等待自己應得的收據*/
	exit_shop(i);/*顧客離開理發店*/
	SIGNAL(MAX_CAPACITY,0);/*釋放理發店資源*/
}

void	barber(int num,struct QueueType* queue)/*理發師同時兼職收銀員*/
{
	WAIT(queueLOCK,0);/*確保理發師發放顧客號碼時不被打斷*/
		enqueue(queue,num);
		SIGNAL(MUTEX,0);/***已有空閑的理發師***/
	SIGNAL(queueLOCK,0);/*確保理發師發放顧客號碼時不被打斷*/
	
	while(1) {
		WAIT(CUST_READY,num);/*理發師等待，直到有一個顧客坐上自己的理發椅*/
		cut_hair(num);/*理發過程*/
		SIGNAL(FINISHED,num);/*告之自己的顧客已完成理發*/
		WAIT(LEAVE_B_CHAIR,num);/*理發師等待自己的顧客從理發椅中站起*/
		SIGNAL(BARBER_CHAIR,num);/*理發師在它的理發椅空時發信號*/
		WAIT(PAYMENT,num);/*理發師接著等待收銀*/
		accept_pay(num);/*提供收據*/
		SIGNAL(RECEIPT,num);/*告之自己的顧客已發出收據*/
		//SIGNAL(CASHIER,0);/*確保收銀過程不被打斷*/
		WAIT(queueLOCK,0);/*確保理發師取顧客號碼時不被打斷*/
			enqueue(queue,num);
		SIGNAL(MUTEX,0);/*已有空閑的理發師*/	
		SIGNAL(queueLOCK,0);/*確保理發師取顧客號碼時不被打斷*/
	
	}
}

void create_barber(struct QueueType *queue,char* shmptr,pid_t pid)
{
int i;
for( i=0; i<3; i++ )
	{
		if( (pid = fork())<0 )
			perror("fork error");
		else if( pid == 0 )/*子進程*/
				{
				if( (shmptr = (char*) shmat(shmid,(char*)0,0) ) ==(char*)-1) 
					{
					perror("get mermory shmat in child");
					exit(4);
					}
				queue = (struct QueueType *)shmptr;
				barber(i+1,queue);
				shmdt(shmptr);/*刪除物理鏈接*/
				sleep(3);
				_exit(0);
				}
	}
}

void create_customer1(struct QueueType *queue,char* shmptr,pid_t pid,int max)
{
int i;
for( i=0; i<max-3; i++ )
	{
		if( (pid = fork())<0 )
			perror("fork error");
		else if( pid == 0 )/*子進程*/
				{
				if( (shmptr = (char*) shmat(shmid,(char*)0,0) ) ==(char*)-1) 
					{
					perror("get mermory shmat in child");
					exit(4);
					}
				queue = (struct QueueType *)shmptr;
				customer(queue,i+1);
				shmdt(shmptr);/*刪除物理鏈接*/
				sleep(3);
				_exit(0);
				}
	}
}

void create_customer2(struct QueueType *queue,char* shmptr,pid_t pid,int max,int av)
{
int i,upbound;
upbound=2*av+1;
srand( (unsigned int)time(NULL) );
for( i=0; i<max-3; i++ )
	{
		if( (pid = fork())<0 )
			perror("fork error");
		else if( pid == 0 )/*子進程*/
				{
				sleep( rand() % upbound );
				if( (shmptr = (char*) shmat(shmid,(char*)0,0) ) ==(char*)-1) 
					{
					perror("get mermory shmat in child");
					exit(4);
					}
				queue = (struct QueueType *)shmptr;
				customer(queue,i+1);
				shmdt(shmptr);/*刪除物理鏈接*/
				sleep(3);
				_exit(0);
				}
	}
}

int main( int argc, char *argv[] )
{
	pid_t		pid;
	union semun	arg;
	int		order;
	int		max;
	int		i=0;
	key_t		key;
	char 		*shmptr;
	static struct QueueType q;
	struct QueueType 	*queue;

	q.out = q.in = 0;
	
	if(argc>2)
	{
		printf("usage: barbershop [number]\n");
		exit(1);
	}
		
	max = (argc==1)?DEFAULT_N:(atoi(argv[1])+3);
	
	if( (semid = semget( IPC_PRIVATE,23,IPC_CREAT|IPC_EXCL|0600))!=-1)
		/*創建23個信號量，返回信號量ID*/
	{
		arg.array = start_val;
		if( semctl( semid, 0, SETALL, arg )==-1 ) 
		/*按arg.array指向的數組中的值設置該集合中所有信號量的值*/
		{
			perror("semctl error");
			exit(1);
		}
	}
	else if( ( semid = semget( IPC_PRIVATE, 23, 0 ))==-1 ) /*返回信號量ID*/
	{
		perror("semget error");
		exit(2);
	}
	
	if( (shmid = shmget( IPC_PRIVATE, sizeof(q), IPC_CREAT|0600)) != -1) 
	/*創建共享存儲區，返回一個共享存儲標識符*/
	{
		if( (shmptr = (char*)shmat(shmid,0,0)) == (char*)-1 )/*連接到物理地址*/
		{
			perror("shmptr_parent_attach");
			exit(1);
		}
		memcpy(shmptr,(char*)&q,sizeof(q));/*復制存儲區*/
	}
	else
	{
		perror("shmid_parent_creation");
		exit(2);
	}


	printf("\n***** Welcome to XMUCSD Barber Shop, cheap but best in the world! *****\n\n");
	create_barber(queue,shmptr,pid);
	//create_customer1(queue,shmptr,pid,max);
	create_customer2(queue,shmptr,pid,max,2);
	sleep(5);
	shmctl(shmid,IPC_RMID,0);/*從系統中刪去共享存儲段*/

	while(wait(0) != -1 && i++ < max-3);
	if( semctl(semid,0,IPC_RMID,0)==-1 ) /*刪除信號量集合*/
	{
		perror("semctl delete error");
		exit(4);
	}
	return 0;
}