96 lines
2.5 KiB
C
96 lines
2.5 KiB
C
|
#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();
|
||
|
|
|
||
|
|
|
||
|
|
};
|