#pragma once
#include <vector>
#include <functional>
#include <dshow.h>
#include <windows.h>
#include "qedit.h"
#include <mutex>
#include <vector>
#include "guiddef.h"

using namespace std;


class Camera
{
public:
	enum CAP_STATUS {
		RUNNING = 1,
		STOP = 2,
		PAUSE = 3,
		FAIL = 4,
	};
	class CameraObserver {
	public:
		virtual int OnCameraData(uint8_t *dat, uint32_t size) { return 0; };
	};


        class SampleGrabberCallback : public ISampleGrabberCB
        {
        public:
                ULONG STDMETHODCALLTYPE AddRef();
                ULONG STDMETHODCALLTYPE Release();
                HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, void** ppvObject);
                HRESULT STDMETHODCALLTYPE SampleCB(double Time, IMediaSample *pSample);
                HRESULT STDMETHODCALLTYPE BufferCB(double Time, BYTE *pBuffer, long BufferLen);
                std::function<void(double, BYTE *, LONG)> mNewDataCallBack;
                mutex mMux;

                int SetObserver(CameraObserver *);
                int RemoveObserver(CameraObserver *p);
        private:
                vector<CameraObserver*> mObserver;
        };


        Camera(wstring camera);
        Camera(const Camera &) = delete;
        Camera& operator  =(const Camera&) = delete;
        ~Camera();

private:
        Camera();

	bool mInitOK;
	bool mIsVideoOpened;

	int mVideoWidth, mVideoHeight, mBitDepth;
	std::function<void(double, BYTE *, LONG)> mFrameCallBack;

	IGraphBuilder *mGraphBuilder;
	ICaptureGraphBuilder2 *mCaptureGB;
	IMediaControl *mMediaControl;
	IBaseFilter *mDevFilter;
	ISampleGrabber *mSampGrabber;
	IMediaEventEx *mMediaEvent;

	SampleGrabberCallback mSampleGrabberCB;
	HRESULT InitializeEnv();
	HRESULT BindFilter(int deviceID, IBaseFilter **pBaseFilter);
        GUID mMediaType;
        bool mDebug;
public:
	int SetObserver(CameraObserver *);
	int RemoveObserver(CameraObserver *p);
	CAP_STATUS mStatus;

        void SetDebug(bool);
        static std::vector<std::wstring> EnumAllCamera(void);
	GUID mPixFmt;
	bool Open(std::wstring &camera_name);
	bool Close(void);
	/*!
	* @param time : Starting time of the sample, in seconds.
	* @param buff : Pointer to a buffer that contains the sample data.
	* @param len  : Length of the buffer pointed to by pBuffer, in bytes.
	*/
	void SetCallBack(std::function<void(double time, BYTE *buff, LONG len)>);
	int GetHeight() { return mVideoHeight; }
	int GetWidth() { return mVideoWidth; }
	int GetBitDepth() { return mBitDepth; }
        GUID MediaType();


};