#include <QApplication>
#include <QMutex>
#include <QThread>
#include <QSemaphore>
#include <QMutexLocker>
#include <QReadWriteLock>
#include <iostream>

using namespace std;

const int DATA_SIZE = 100;
const int BUF_SIZE = 10;
int buf[BUF_SIZE] = {0};

QMutex mutex;// 控制執行緒間的互斥

/********freeSpace和useSpace控制執行緒間的同步********/

// 控制可被生產者填充資料的那部分緩衝區
QSemaphore freeSpace(BUF_SIZE);// 程式開始啟動時，生產者會有BUF_SIZE個空間生產資料

// 控制可被消費者讀取資料的那部分緩衝區
QSemaphore useSpace(0);// 程式開始啟動時，消費者會有0個空間取資料


class CProducer : public QThread
{
protected:
    virtual void run()
    {
        for (int i = 0 ; i < DATA_SIZE ; ++i)
        {
            freeSpace.acquire();// 當前訊號量是否有可以填充資料的緩衝區，如果資源計數為0，則沒有可以填充的緩衝區，那麼該函式會阻塞當前執行緒
            {
                QMutexLocker locker(&mutex);

                buf[i%BUF_SIZE] = qrand() % 50 + 1;
                cout << "produce data is :" << buf[i%BUF_SIZE] << endl;
            }
            useSpace.release();// 使useSpace物件的資源計數加1，此時消費者執行緒就有機會取資料了
        }
    }
};

class CConsumer : public QThread
{
protected:
    virtual void run()
    {
        for (int i = 0 ; i < DATA_SIZE ; ++i)
        {
            useSpace.acquire();// 當前訊號量是否有可以取資料的緩衝區，如果資源計數為0，則沒有可取的緩衝區，那麼該函式會阻塞當前執行緒
            {
                QMutexLocker locker(&mutex);

                int nData = buf[i%BUF_SIZE];
                cout << "consume data is: " << nData << endl;
            }
            freeSpace.release();// 使freeSpace物件的資源計數加1，此時生產者執行緒就有機會生產資料了
        }
    }
};


int main(int argc, char *argv[])
{
    CProducer producer;
    CConsumer consumer;

    producer.start();
    consumer.start();

    producer.wait();
    consumer.wait();

    return 0;
}