118 lines
2.9 KiB
C++
118 lines
2.9 KiB
C++
//////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// capture.h: Manages video capture.
|
|
//
|
|
// THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF
|
|
// ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO
|
|
// THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A
|
|
// PARTICULAR PURPOSE.
|
|
//
|
|
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
//
|
|
//////////////////////////////////////////////////////////////////////////
|
|
|
|
#pragma once
|
|
|
|
const UINT WM_APP_PREVIEW_ERROR = WM_APP + 1; // wparam = HRESULT
|
|
|
|
class DeviceList
|
|
{
|
|
UINT32 m_cDevices;
|
|
IMFActivate **m_ppDevices;
|
|
|
|
public:
|
|
DeviceList() : m_ppDevices(NULL), m_cDevices(0)
|
|
{
|
|
|
|
}
|
|
~DeviceList()
|
|
{
|
|
Clear();
|
|
}
|
|
|
|
UINT32 Count() const { return m_cDevices; }
|
|
|
|
void Clear();
|
|
HRESULT EnumerateDevices();
|
|
HRESULT GetDevice(UINT32 index, IMFActivate **ppActivate);
|
|
HRESULT GetDeviceName(UINT32 index, WCHAR **ppszName);
|
|
};
|
|
|
|
|
|
struct EncodingParameters
|
|
{
|
|
GUID subtype;
|
|
UINT32 bitrate;
|
|
};
|
|
|
|
class CCapture : public IMFSourceReaderCallback
|
|
{
|
|
public:
|
|
static HRESULT CreateInstance(
|
|
HWND hwnd,
|
|
CCapture **ppPlayer
|
|
);
|
|
|
|
// IUnknown methods
|
|
STDMETHODIMP QueryInterface(REFIID iid, void** ppv);
|
|
STDMETHODIMP_(ULONG) AddRef();
|
|
STDMETHODIMP_(ULONG) Release();
|
|
|
|
// IMFSourceReaderCallback methods
|
|
STDMETHODIMP OnReadSample(
|
|
HRESULT hrStatus,
|
|
DWORD dwStreamIndex,
|
|
DWORD dwStreamFlags,
|
|
LONGLONG llTimestamp,
|
|
IMFSample *pSample
|
|
);
|
|
|
|
STDMETHODIMP OnEvent(DWORD, IMFMediaEvent *)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP OnFlush(DWORD)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT StartCapture(IMFActivate *pActivate, const WCHAR *pwszFileName, const EncodingParameters& param);
|
|
HRESULT EndCaptureSession();
|
|
BOOL IsCapturing();
|
|
HRESULT CheckDeviceLost(DEV_BROADCAST_HDR *pHdr, BOOL *pbDeviceLost);
|
|
|
|
protected:
|
|
|
|
enum State
|
|
{
|
|
State_NotReady = 0,
|
|
State_Ready,
|
|
State_Capturing,
|
|
};
|
|
|
|
// Constructor is private. Use static CreateInstance method to instantiate.
|
|
CCapture(HWND hwnd);
|
|
|
|
// Destructor is private. Caller should call Release.
|
|
virtual ~CCapture();
|
|
|
|
void NotifyError(HRESULT hr) { PostMessage(m_hwndEvent, WM_APP_PREVIEW_ERROR, (WPARAM)hr, 0L); }
|
|
|
|
HRESULT OpenMediaSource(IMFMediaSource *pSource);
|
|
HRESULT ConfigureCapture(const EncodingParameters& param);
|
|
HRESULT EndCaptureInternal();
|
|
|
|
long m_nRefCount; // Reference count.
|
|
CRITICAL_SECTION m_critsec;
|
|
|
|
HWND m_hwndEvent; // Application window to receive events.
|
|
|
|
IMFSourceReader *m_pReader;
|
|
IMFSinkWriter *m_pWriter;
|
|
|
|
BOOL m_bFirstSample;
|
|
LONGLONG m_llBaseTime;
|
|
|
|
WCHAR *m_pwszSymbolicLink;
|
|
}; |