Index: media/video/capture/win/video_capture_device_mf_win.cc |
=================================================================== |
--- media/video/capture/win/video_capture_device_mf_win.cc (revision 170924) |
+++ media/video/capture/win/video_capture_device_mf_win.cc (working copy) |
@@ -1,399 +0,0 @@ |
-// Copyright (c) 2012 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include "media/video/capture/win/video_capture_device_mf_win.h" |
- |
-#include <mfapi.h> |
-#include <mferror.h> |
- |
-#include "base/lazy_instance.h" |
-#include "base/memory/ref_counted.h" |
-#include "base/synchronization/waitable_event.h" |
-#include "base/sys_string_conversions.h" |
-#include "base/win/scoped_co_mem.h" |
-#include "media/video/capture/win/capability_list_win.h" |
- |
-using base::win::ScopedCoMem; |
-using base::win::ScopedComPtr; |
- |
-namespace media { |
-namespace { |
- |
-class MFInitializerSingleton { |
- public: |
- MFInitializerSingleton() { MFStartup(MF_VERSION, MFSTARTUP_LITE); } |
- ~MFInitializerSingleton() { MFShutdown(); } |
-}; |
- |
-static base::LazyInstance<MFInitializerSingleton> g_mf_initialize = |
- LAZY_INSTANCE_INITIALIZER; |
- |
-void EnsureMFInit() { |
- g_mf_initialize.Get(); |
-} |
- |
-bool PrepareVideoCaptureAttributes(IMFAttributes** attributes, int count) { |
- EnsureMFInit(); |
- |
- if (FAILED(MFCreateAttributes(attributes, count))) |
- return false; |
- |
- return SUCCEEDED((*attributes)->SetGUID(MF_DEVSOURCE_ATTRIBUTE_SOURCE_TYPE, |
- MF_DEVSOURCE_ATTRIBUTE_SOURCE_TYPE_VIDCAP_GUID)); |
-} |
- |
-bool EnumerateVideoDevices(IMFActivate*** devices, |
- UINT32* count) { |
- ScopedComPtr<IMFAttributes> attributes; |
- if (!PrepareVideoCaptureAttributes(attributes.Receive(), 1)) |
- return false; |
- |
- return SUCCEEDED(MFEnumDeviceSources(attributes, devices, count)); |
-} |
- |
-bool CreateVideoCaptureDevice(const char* sym_link, IMFMediaSource** source) { |
- ScopedComPtr<IMFAttributes> attributes; |
- if (!PrepareVideoCaptureAttributes(attributes.Receive(), 2)) |
- return false; |
- |
- attributes->SetString(MF_DEVSOURCE_ATTRIBUTE_SOURCE_TYPE_VIDCAP_SYMBOLIC_LINK, |
- base::SysUTF8ToWide(sym_link).c_str()); |
- |
- return SUCCEEDED(MFCreateDeviceSource(attributes, source)); |
-} |
- |
-bool FormatFromGuid(const GUID& guid, VideoCaptureCapability::Format* format) { |
- struct { |
- const GUID& guid; |
- const VideoCaptureCapability::Format format; |
- } static const kFormatMap[] = { |
- { MFVideoFormat_I420, VideoCaptureCapability::kI420 }, |
- { MFVideoFormat_YUY2, VideoCaptureCapability::kYUY2 }, |
- { MFVideoFormat_UYVY, VideoCaptureCapability::kUYVY }, |
- { MFVideoFormat_RGB24, VideoCaptureCapability::kRGB24 }, |
- { MFVideoFormat_ARGB32, VideoCaptureCapability::kARGB }, |
- { MFVideoFormat_MJPG, VideoCaptureCapability::kMJPEG }, |
- { MFVideoFormat_YV12, VideoCaptureCapability::kYV12 }, |
- }; |
- |
- for (int i = 0; i < arraysize(kFormatMap); ++i) { |
- if (kFormatMap[i].guid == guid) { |
- *format = kFormatMap[i].format; |
- return true; |
- } |
- } |
- |
- return false; |
-} |
- |
-bool GetFrameSize(IMFMediaType* type, int* width, int* height) { |
- UINT32 width32, height32; |
- if (FAILED(MFGetAttributeSize(type, MF_MT_FRAME_SIZE, &width32, &height32))) |
- return false; |
- *width = width32; |
- *height = height32; |
- return true; |
-} |
- |
-bool GetFrameRate(IMFMediaType* type, int* frame_rate) { |
- UINT32 numerator, denominator; |
- if (FAILED(MFGetAttributeRatio(type, MF_MT_FRAME_RATE, &numerator, |
- &denominator))) { |
- return false; |
- } |
- |
- *frame_rate = denominator ? numerator / denominator : 0; |
- |
- return true; |
-} |
- |
-bool FillCapabilitiesFromType(IMFMediaType* type, |
- VideoCaptureCapability* capability) { |
- GUID type_guid; |
- if (FAILED(type->GetGUID(MF_MT_SUBTYPE, &type_guid)) || |
- !FormatFromGuid(type_guid, &capability->color) || |
- !GetFrameSize(type, &capability->width, &capability->height) || |
- !GetFrameRate(type, &capability->frame_rate)) { |
- return false; |
- } |
- |
- capability->expected_capture_delay = 0; // Currently not used. |
- capability->interlaced = false; // Currently not used. |
- |
- return true; |
-} |
- |
-HRESULT FillCapabilities(IMFSourceReader* source, |
- CapabilityList* capabilities) { |
- DWORD stream_index = 0; |
- ScopedComPtr<IMFMediaType> type; |
- HRESULT hr; |
- while (SUCCEEDED(hr = source->GetNativeMediaType( |
- MF_SOURCE_READER_FIRST_VIDEO_STREAM, stream_index, type.Receive()))) { |
- VideoCaptureCapabilityWin capability(stream_index++); |
- if (FillCapabilitiesFromType(type, &capability)) |
- capabilities->Add(capability); |
- type.Release(); |
- } |
- |
- if (SUCCEEDED(hr) && capabilities->empty()) |
- hr = HRESULT_FROM_WIN32(ERROR_EMPTY); |
- |
- return (hr == MF_E_NO_MORE_TYPES) ? S_OK : hr; |
-} |
- |
-} // namespace |
- |
-class MFReaderCallback |
- : public base::RefCountedThreadSafe<MFReaderCallback>, |
- public IMFSourceReaderCallback { |
- public: |
- MFReaderCallback(VideoCaptureDeviceMFWin* observer) |
- : observer_(observer), wait_event_(NULL) { |
- } |
- |
- void SetSignalOnFlush(base::WaitableEvent* event) { |
- wait_event_ = event; |
- } |
- |
- STDMETHOD(QueryInterface)(REFIID riid, void** object) { |
- if (riid != IID_IUnknown && riid != IID_IMFSourceReaderCallback) |
- return E_NOINTERFACE; |
- *object = static_cast<IMFSourceReaderCallback*>(this); |
- AddRef(); |
- return S_OK; |
- } |
- |
- STDMETHOD_(ULONG, AddRef)() { |
- base::RefCountedThreadSafe<MFReaderCallback>::AddRef(); |
- return 1U; |
- } |
- |
- STDMETHOD_(ULONG, Release)() { |
- base::RefCountedThreadSafe<MFReaderCallback>::Release(); |
- return 1U; |
- } |
- |
- STDMETHOD(OnReadSample)(HRESULT status, DWORD stream_index, |
- DWORD stream_flags, LONGLONG time_stamp, IMFSample* sample) { |
- base::Time stamp(base::Time::Now()); |
- if (!sample) { |
- observer_->OnIncomingCapturedFrame(NULL, 0, stamp); |
- return S_OK; |
- } |
- |
- DWORD count = 0; |
- sample->GetBufferCount(&count); |
- |
- for (DWORD i = 0; i < count; ++i) { |
- ScopedComPtr<IMFMediaBuffer> buffer; |
- sample->GetBufferByIndex(i, buffer.Receive()); |
- if (buffer) { |
- DWORD length = 0, max_length = 0; |
- BYTE* data = NULL; |
- buffer->Lock(&data, &max_length, &length); |
- observer_->OnIncomingCapturedFrame(data, length, stamp); |
- buffer->Unlock(); |
- } |
- } |
- return S_OK; |
- } |
- |
- STDMETHOD(OnFlush)(DWORD stream_index) { |
- if (wait_event_) { |
- wait_event_->Signal(); |
- wait_event_ = NULL; |
- } |
- return S_OK; |
- } |
- |
- STDMETHOD(OnEvent)(DWORD stream_index, IMFMediaEvent* event) { |
- NOTIMPLEMENTED(); |
- return S_OK; |
- } |
- |
- private: |
- friend class base::RefCountedThreadSafe<MFReaderCallback>; |
- ~MFReaderCallback() {} |
- |
- VideoCaptureDeviceMFWin* observer_; |
- base::WaitableEvent* wait_event_; |
-}; |
- |
-// static |
-void VideoCaptureDeviceMFWin::GetDeviceNames(Names* device_names) { |
- ScopedCoMem<IMFActivate*> devices; |
- UINT32 count; |
- if (!EnumerateVideoDevices(&devices, &count)) |
- return; |
- |
- HRESULT hr; |
- for (UINT32 i = 0; i < count; ++i) { |
- UINT32 name_size, id_size; |
- ScopedCoMem<wchar_t> name, id; |
- if (SUCCEEDED(hr = devices[i]->GetAllocatedString( |
- MF_DEVSOURCE_ATTRIBUTE_FRIENDLY_NAME, &name, &name_size)) && |
- SUCCEEDED(hr = devices[i]->GetAllocatedString( |
- MF_DEVSOURCE_ATTRIBUTE_SOURCE_TYPE_VIDCAP_SYMBOLIC_LINK, &id, |
- &id_size))) { |
- std::wstring name_w(name, name_size), id_w(id, id_size); |
- Name device; |
- device.device_name = base::SysWideToUTF8(name_w); |
- device.unique_id = base::SysWideToUTF8(id_w); |
- device_names->push_back(device); |
- } else { |
- DLOG(WARNING) << "GetAllocatedString failed: " << std::hex << hr; |
- } |
- devices[i]->Release(); |
- } |
-} |
- |
-VideoCaptureDeviceMFWin::VideoCaptureDeviceMFWin(const Name& device_name) |
- : name_(device_name), observer_(NULL), capture_(0) { |
- DetachFromThread(); |
-} |
- |
-VideoCaptureDeviceMFWin::~VideoCaptureDeviceMFWin() { |
- DCHECK(CalledOnValidThread()); |
-} |
- |
-bool VideoCaptureDeviceMFWin::Init() { |
- DCHECK(CalledOnValidThread()); |
- DCHECK(!reader_); |
- |
- ScopedComPtr<IMFMediaSource> source; |
- if (!CreateVideoCaptureDevice(name_.unique_id.c_str(), source.Receive())) |
- return false; |
- |
- ScopedComPtr<IMFAttributes> attributes; |
- MFCreateAttributes(attributes.Receive(), 1); |
- DCHECK(attributes); |
- |
- callback_ = new MFReaderCallback(this); |
- attributes->SetUnknown(MF_SOURCE_READER_ASYNC_CALLBACK, callback_.get()); |
- |
- return SUCCEEDED(MFCreateSourceReaderFromMediaSource(source, attributes, |
- reader_.Receive())); |
-} |
- |
-void VideoCaptureDeviceMFWin::Allocate( |
- int width, |
- int height, |
- int frame_rate, |
- VideoCaptureDevice::EventHandler* observer) { |
- DCHECK(CalledOnValidThread()); |
- |
- base::AutoLock lock(lock_); |
- |
- if (observer_) { |
- DCHECK_EQ(observer, observer_); |
- return; |
- } |
- |
- observer_ = observer; |
- DCHECK_EQ(capture_, false); |
- |
- CapabilityList capabilities; |
- HRESULT hr = S_OK; |
- if (!reader_ || FAILED(hr = FillCapabilities(reader_, &capabilities))) { |
- OnError(hr); |
- return; |
- } |
- |
- const VideoCaptureCapabilityWin& found_capability = |
- capabilities.GetBestMatchedCapability(width, height, frame_rate); |
- |
- ScopedComPtr<IMFMediaType> type; |
- if (FAILED(hr = reader_->GetNativeMediaType( |
- MF_SOURCE_READER_FIRST_VIDEO_STREAM, found_capability.stream_index, |
- type.Receive())) || |
- FAILED(hr = reader_->SetCurrentMediaType( |
- MF_SOURCE_READER_FIRST_VIDEO_STREAM, NULL, type))) { |
- OnError(hr); |
- return; |
- } |
- |
- observer_->OnFrameInfo(found_capability); |
-} |
- |
-void VideoCaptureDeviceMFWin::Start() { |
- DCHECK(CalledOnValidThread()); |
- |
- base::AutoLock lock(lock_); |
- if (!capture_) { |
- capture_ = true; |
- HRESULT hr; |
- if (FAILED(hr = reader_->ReadSample(MF_SOURCE_READER_FIRST_VIDEO_STREAM, 0, |
- NULL, NULL, NULL, NULL))) { |
- OnError(hr); |
- capture_ = false; |
- } |
- } |
-} |
- |
-void VideoCaptureDeviceMFWin::Stop() { |
- DCHECK(CalledOnValidThread()); |
- base::WaitableEvent flushed(false, false); |
- bool wait = false; |
- { |
- base::AutoLock lock(lock_); |
- if (capture_) { |
- capture_ = false; |
- callback_->SetSignalOnFlush(&flushed); |
- HRESULT hr = reader_->Flush(MF_SOURCE_READER_ALL_STREAMS); |
- wait = SUCCEEDED(hr); |
- if (!wait) { |
- callback_->SetSignalOnFlush(NULL); |
- OnError(hr); |
- } |
- } |
- } |
- |
- if (wait) |
- flushed.Wait(); |
-} |
- |
-void VideoCaptureDeviceMFWin::DeAllocate() { |
- DCHECK(CalledOnValidThread()); |
- |
- Stop(); |
- |
- base::AutoLock lock(lock_); |
- observer_ = NULL; |
-} |
- |
-const VideoCaptureDevice::Name& VideoCaptureDeviceMFWin::device_name() { |
- DCHECK(CalledOnValidThread()); |
- return name_; |
-} |
- |
-void VideoCaptureDeviceMFWin::OnIncomingCapturedFrame( |
- const uint8* data, |
- int length, |
- const base::Time& time_stamp) { |
- base::AutoLock lock(lock_); |
- if (data && observer_) |
- observer_->OnIncomingCapturedFrame(data, length, time_stamp); |
- |
- if (capture_) { |
- HRESULT hr = reader_->ReadSample(MF_SOURCE_READER_FIRST_VIDEO_STREAM, 0, |
- NULL, NULL, NULL, NULL); |
- if (FAILED(hr)) { |
- // If running the *VideoCap* unit tests on repeat, this can sometimes |
- // fail with HRESULT_FROM_WINHRESULT_FROM_WIN32(ERROR_INVALID_FUNCTION). |
- // It's not clear to me why this is, but it is possible that it has |
- // something to do with this bug: |
- // http://support.microsoft.com/kb/979567 |
- OnError(hr); |
- } |
- } |
-} |
- |
-void VideoCaptureDeviceMFWin::OnError(HRESULT hr) { |
- DLOG(ERROR) << "VideoCaptureDeviceMFWin: " << std::hex << hr; |
- if (observer_) |
- observer_->OnError(); |
-} |
- |
-} // namespace media |