#include "windows.h"
#include <conio.h>  //屏幕操作函數的頭文件，即調用DOS控制臺I/O子程序的各個函數
#include<stdlib.h>     //說明一些常用的子程序：轉換子程序、搜索排序程序等
#include<fstream.h>   //c++輸入輸出流iostream的派生類，用于磁盤文件I/O
#include<string.h>   //說明一些字符串和內存操作函數
#include<stdio.h> //定義UNIX中標準和擴展的類型和宏，定義標準I/O預定義流，說明
//I/O流子程序
#define READER 'R'              //讀者
#define WRITER 'W'              //寫者
#define INTE PER SEC  100      //每秒時鐘中斷數目
#define MAX THREAD NUM   64      //最大線程數目
#define MAX FILE NUM     32      //最大數據文件數目
#define MAX STR LEN      32      //字符串長度

int   readcount=0;               //讀者數目
int   writecunt=0;               //寫者數目

CRITICAL SECTION  RP Write;      //臨界區
CRITICAL SECTION  cs Write;      //臨界區
CRITICAL SECTION  cs Read;       //臨界區
struct ThreadInfo
{
int     serial;                //線程序號
char    entity;               //線程類別
double  delay;                //線程延遲
double  persist;              //線程讀寫操作的持續時間
};
/////////////////////////////////////////////////////////////////////////////////////讀者優先-讀者線程
//p:讀者線程信息

void RP ReaderThread(void* p)
{
//互斥變量
HANDLE h Mutex;
h Mutex=OpenMutex(MUTEX ALL ACCESS,FALSE,"mutex for readcount");
DWORD wait for mutex;     //等待互斥變量的所有權
DWORD m delay;            //延遲時間
DWORD m persist;          //讀文件持續時間
int m serial;             //線程序號
//從參數中獲得信息
m serial =((ThreadInfo*)(p))->serial;
m delay =(DWORD)(((ThreadInfo*)(p))->delay*INTE PER SEC);
Sleep(m delay);               //延遲等待
printf("Reader thread %d sents the reading require.\n", m serial);
//等待互斥信號，保證對readcount的訪問、修改互斥
wait for mutex =WaitForSingleObject(h Mutex , -1);
if(readcount ==0) 
  EnterCriticalSection(&RP Write);
readcount++;
ReleaseMutex(h Mutex);       //釋放互斥信號
//讀文件
printf("Reader thread %d begins to read file,\n",m serial);
Sleep(m persist);            //延遲一下，滿足讀文件的持續時間
printf("Reader thread %d finished reading file.\n",m serial);
wait for mutex =WaitForSingleObject(h Mutex,-1);
readcount--;
if (readcount==0)
LeaveCriticalSection(&RP Write);
ReleaseMutex(h Mutex);
}

/////////////////////////////////////////////////////////////////////////////////////讀者優先---寫者線程
//p: 寫者線程信息
void RP WriterThread(void* p)
{
DWORD m delay;
DWORD m persist;
int m serial;
m serial=((ThreadInfo*)(p))->serial;
m delay=(DWORD)(((ThreadInfo*)(p))->delay*INTE PER SEC);
m persist=(DWORD)(((ThreadInfo*)(p))->persist*INTE PER SEC);
Sleep(m delay);
printf("writer thread %d sents the writing require.\n",m serial);
EnterCriticalSection(&RP Write);
//寫文件
printf("writer thread %d begins to write to the file .\n",m serial);
Sleep(m persist);
printf("writer thread %d finishing writing to the file.\n",m serial);
LeaveCriticalSection(&RP Write);
}


/////////////////////////////////////////////////////////////////////////////////////讀者優先處理函數
//file:文件名
void ReaderPriority(char* file)
{
DWORD n thread=0;                //線程數目
DWORD thread ID;                 //線程ID
DWORD wait for all;              //等待所有線程結束
HANDLE  h Mutex;
h Mutex=CreateMutex(NULL,FALSE,"mutex for readcount");
//線程對象數組
HANDLE h Thread[MAX THREAD NUM];
ThreadInfo thread info[MAX THREAD NUM];
readcount=0;                    //初始化readcount
InitializeCriticalSection(&RP Write);           //初始化臨界區    
ifstream inFile;
inFile.open(file);                           //打開文件
printf("reader priority:\n\n");
while(inFile)
{
//讀入每一個讀者、寫者信息
inFile>>thread info[n thread].serial;
inFile>>thread info[n thread].entity;
inFile>>thread info[n thread].delay;
inFile>>thread info[n thread++].persist;
inFile.get();
}
for (int i=0;i<(int)(n thread);i++)
{
   if(thread info[i].entity==READER||thread info[i].entity=='R')
   {   //創建讀者線程,&thread ID是輸出參數接受線程ID
    h Thread[i]=CreateThread(NULL,0,(LPTHREAD START ROUTINE)(RP ReaderThread),&thread info[i],0,&thread ID);
   }
    else{
    h Thread[i]=CreateThread(NULL,0,(LPTHREAD START ROUTINE)(RP WriterThread),&thread info[i],0,&thread ID);
   }
   
}
//等待所有線程結束
wait for all=WaitForMultipleObjects(n thread,h Thread,TRUE,-1);
printf("All reader and writer have finished operating ,\n");
}

//主函數
int main(int argc,char*argv[])
{
char ch;
while(true)
{
printf("    1:Reader priority\n");
printf("    2:Writer priority\n");
printf("    3:Exit to windows\n");
do{
   ch=(char) getch();
}while(ch!='1'&&ch!='2'&&ch!='3');
system("cls");
if(ch=='3')
return 0;
else if(ch=='1')
  ReaderPriority("thread.dat");
printf("\npress any key to continue:");
getch();
system("cls");
return 0;
}
}