#pragma once
#pragma include_alias( "dxtrans.h", "qedit.h" )
#define __IDxtCompositor_INTERFACE_DEFINED__
#define __IDxtAlphaSetter_INTERFACE_DEFINED__
#define __IDxtJpeg_INTERFACE_DEFINED__
#define __IDxtKey_INTERFACE_DEFINED__
#include "qedit.h"

//---------------ṹ-----------------
#include <vector>
using namespace std;
struct GrabDataInfo
{
	BYTE *pData;
	int nDataSize;
	double dSampleTime;

	GrabDataInfo()
	{
		pData = NULL;
		nDataSize = 0;
		dSampleTime = 0.0;
	};

	GrabDataInfo(const GrabDataInfo &other)
	{
		*this = other;
	};

	GrabDataInfo& operator = (const GrabDataInfo& other)
	{
		pData = other.pData;
		nDataSize = other.nDataSize;
		dSampleTime = other.dSampleTime;
		return *this;
	};
};
typedef vector<GrabDataInfo> ASGrabDataInfoArray;
//---------------ṹ-----------------

class CSampleGrabberCB :
	public ISampleGrabberCB
{
public:
	CSampleGrabberCB(void);
	virtual ~CSampleGrabberCB(void);

public:
	long lWidth;
	long lHeight;
	BOOL m_bBeginEncode;
	BOOL m_bFirst;
	BOOL m_bEndEncode;
	//CString m_sSavePath;
	ULONG m_ulBitRate;

	//FILE* m_fp_dst;
	//x264_param_t* m_pParam;
	//x264_picture_t* m_pPic_in;
	//x264_picture_t* m_pPic_out;
	//x264_t* m_pHandle;
	int m_nFrameIndex;
	ASGrabDataInfoArray m_arrGrabDataArr;
	//CMutex  m_mxMsgLog;
	HANDLE	      m_hMainExitEvent; 

public:
	ULONG STDMETHODCALLTYPE AddRef();
	ULONG STDMETHODCALLTYPE Release();
	HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, void ** ppv);
	HRESULT STDMETHODCALLTYPE SampleCB(double SampleTime, IMediaSample *pSample);
	HRESULT STDMETHODCALLTYPE BufferCB(double SampleTime, BYTE *pBuffer, long BufferLen);
	BOOL RGB2YUV(LPBYTE RgbBuf, UINT nWidth, UINT nHeight, LPBYTE yuvBuf, unsigned long *len);
	void VideoDeal();

};