#include <cstdio>
#include <string>
#include <cstring>
#include <deque>
#include <pthread.h>

#include "librtsp.h"


using namespace std;

template<typename T>
class MyQueue {
public:
    MyQueue():mtx(PTHREAD_MUTEX_INITIALIZER), cond(PTHREAD_COND_INITIALIZER){
        t.tv_sec = 0;
        t.tv_nsec = 20000000;
    }

    ~MyQueue() {

    }

public:
    void push(T v) {
        pthread_mutex_lock(&mtx);
        q.push_back(v);
        pthread_cond_signal(&cond);
        pthread_mutex_unlock(&mtx);
    }

    //向队列的前面插入元素
    void push_front_one(T v) {
        pthread_mutex_lock(&mtx);
        q.push_front(v);
        pthread_cond_signal(&cond);
        pthread_mutex_unlock(&mtx);
    }

    T pop() {
        pthread_mutex_lock(&mtx);
        while (q.empty()) {
            pthread_cond_wait(&cond, &mtx);
            pthread_cond_timedwait(&cond, &mtx, &t);
        }
        T value = q.front();
        q.pop_front();
        pthread_mutex_unlock(&mtx);
        return value;
    }

    T popNotWait() {
        pthread_mutex_lock(&mtx);
        T value = q.front();
        q.pop_front();
        pthread_mutex_unlock(&mtx);
        return value;

    }

    int count_queue() {
        return q.size();
    }

    T clearAll(){
		pthread_mutex_lock(&mtx);
		while (!q.empty()) q.pop_front();
		pthread_mutex_unlock(&mtx);
    }

private:
    deque<T> q;
    pthread_mutex_t mtx;
    pthread_cond_t cond;
    timespec t;
};

typedef struct _buffInfo {
	unsigned char *buff;
	int buffLen;
} frameBuffInfo;

MyQueue<frameBuffInfo *> m_rtpQueue;
long Handle;

bool pushInfo(unsigned char *data, int datalen) {

	frameBuffInfo *info = new frameBuffInfo();
	info->buff = new unsigned char[datalen];
	info->buffLen = datalen;
	memcpy(info->buff, data, datalen);

	//printf(" m_rtpQueue.push befores ");
	m_rtpQueue.push(info);
	//printf(" m_rtpQueue.push after ");

	return true;
}

int readData(void *opaque, unsigned char *buf, int bufsize) {

//	GB28181API *_this = (GB28181API *)opaque;
	int len = bufsize;
	int diff = 0;
	do {

	//printf(" m_rtpQueue.pop before ");
	//从缓存中获取buffinfo
	frameBuffInfo *buffinfo = m_rtpQueue.pop();
//        DBG(" m_rtpQueue.pop after ");
	diff = len - buffinfo->buffLen;

	//帧长大于bufsize
	if (diff < 0) {
		printf("/帧长大于bufsize:%d\n", diff);
		memcpy(buf + bufsize - len, buffinfo->buff, len);

		frameBuffInfo *info = new frameBuffInfo();
		info->buffLen = buffinfo->buffLen - len;
		info->buff = new unsigned char[buffinfo->buffLen - len]{};
		memcpy(info->buff, buffinfo->buff + len, buffinfo->buffLen - len);

		m_rtpQueue.push_front_one(info);
	} else if (diff == 0) {
		printf("/帧长等于bufsize:%d\n", diff);
		memcpy(buf + bufsize - len, buffinfo->buff, buffinfo->buffLen);
	} else if (diff > 0) {
		printf("/帧长小于bufsize:%d\n", diff);
		memcpy(buf + bufsize - len, buffinfo->buff, buffinfo->buffLen);
		len = len - buffinfo->buffLen;   //还需要填充的大小
		memset(buf + bufsize - len, 0, len);
		//不等待填充，直接进行解码
		diff = 0;
	}
	delete[] buffinfo->buff;
	delete buffinfo;
	} while (diff > 0);

	return bufsize;
}

void streamCallBack(int datatype, int frametype, unsigned char *data, unsigned int datalen, long userdata)
{
	//GB28181API *_this = (GB28181API *)userdata;
	printf("userdata:%ld,datatype:%d, frametype:%d,  datalen:%d\n", userdata, datatype, frametype, datalen);

//	//debug===============
//	static int count = 0;
//	static FILE *fp_write = NULL;
//	if(count < 100) {
//		count++;
//
//		if (!fp_write) {
//			fp_write = fopen("stream_callback.mp4", "wb+");
//		}
//
//		fwrite(data, sizeof(char), datalen, fp_write);
//	}
//	if(count >= 100){
//		if (!fp_write) {
//			fclose(fp_write);
//		}
//	}
//	//debug===============

	static bool startFlag = false;
	if(frametype == GB_VIDEO_FRAME_I){
		startFlag = true;
	}
	if((data != NULL) && (startFlag == true)){
		pushInfo(data, datalen);
	}

}

long addCamera(string &rtsp){
	printf("RTSPSTREAM_Open\n");
	long userdata = 1001;
	Handle = RTSPSTREAM_Open(rtsp.c_str(), streamCallBack, userdata);
	return Handle;
}

void deleteCamera(void){
	m_rtpQueue.clearAll();
	RTSPSTREAM_Close(Handle);
	Handle = 0;
}