blob: 68f4a029bbc2f954bb0f475e5a66747755290db5 [file] [log] [blame]
// 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 "content/browser/renderer_host/media/video_capture_controller.h"
#include <set>
#include "base/bind.h"
#include "base/debug/trace_event.h"
#include "base/stl_util.h"
#include "content/browser/renderer_host/media/media_stream_manager.h"
#include "content/browser/renderer_host/media/video_capture_manager.h"
#include "content/public/browser/browser_thread.h"
#include "media/base/video_frame.h"
#include "media/base/video_util.h"
#include "media/base/yuv_convert.h"
#if !defined(AVOID_LIBYUV_FOR_ANDROID_WEBVIEW)
#include "third_party/libyuv/include/libyuv.h"
#endif
using media::VideoCaptureFormat;
namespace content {
namespace {
// The number of buffers that VideoCaptureBufferPool should allocate.
const int kNoOfBuffers = 3;
class PoolBuffer : public media::VideoCaptureDevice::Client::Buffer {
public:
PoolBuffer(const scoped_refptr<VideoCaptureBufferPool>& pool,
int buffer_id,
void* data,
size_t size)
: Buffer(buffer_id, data, size), pool_(pool) {
DCHECK(pool_);
}
private:
virtual ~PoolBuffer() { pool_->RelinquishProducerReservation(id()); }
const scoped_refptr<VideoCaptureBufferPool> pool_;
};
} // anonymous namespace
struct VideoCaptureController::ControllerClient {
ControllerClient(const VideoCaptureControllerID& id,
VideoCaptureControllerEventHandler* handler,
base::ProcessHandle render_process,
media::VideoCaptureSessionId session_id,
const media::VideoCaptureParams& params)
: controller_id(id),
event_handler(handler),
render_process_handle(render_process),
session_id(session_id),
parameters(params),
session_closed(false) {}
~ControllerClient() {}
// ID used for identifying this object.
const VideoCaptureControllerID controller_id;
VideoCaptureControllerEventHandler* const event_handler;
// Handle to the render process that will receive the capture buffers.
const base::ProcessHandle render_process_handle;
const media::VideoCaptureSessionId session_id;
const media::VideoCaptureParams parameters;
// Buffers that are currently known to this client.
std::set<int> known_buffers;
// Buffers currently held by this client.
std::set<int> active_buffers;
// State of capture session, controlled by VideoCaptureManager directly. This
// transitions to true as soon as StopSession() occurs, at which point the
// client is sent an OnEnded() event. However, because the client retains a
// VideoCaptureController* pointer, its ControllerClient entry lives on until
// it unregisters itself via RemoveClient(), which may happen asynchronously.
//
// TODO(nick): If we changed the semantics of VideoCaptureHost so that
// OnEnded() events were processed synchronously (with the RemoveClient() done
// implicitly), we could avoid tracking this state here in the Controller, and
// simplify the code in both places.
bool session_closed;
};
// Receives events from the VideoCaptureDevice and posts them to a
// VideoCaptureController on the IO thread. An instance of this class may safely
// outlive its target VideoCaptureController.
//
// Methods of this class may be called from any thread, and in practice will
// often be called on some auxiliary thread depending on the platform and the
// device type; including, for example, the DirectShow thread on Windows, the
// v4l2_thread on Linux, and the UI thread for tab capture.
class VideoCaptureController::VideoCaptureDeviceClient
: public media::VideoCaptureDevice::Client {
public:
explicit VideoCaptureDeviceClient(
const base::WeakPtr<VideoCaptureController>& controller,
const scoped_refptr<VideoCaptureBufferPool>& buffer_pool);
virtual ~VideoCaptureDeviceClient();
// VideoCaptureDevice::Client implementation.
virtual scoped_refptr<Buffer> ReserveOutputBuffer(
media::VideoFrame::Format format,
const gfx::Size& size) OVERRIDE;
virtual void OnIncomingCapturedFrame(const uint8* data,
int length,
base::Time timestamp,
int rotation,
const VideoCaptureFormat& frame_format)
OVERRIDE;
virtual void OnIncomingCapturedBuffer(const scoped_refptr<Buffer>& buffer,
media::VideoFrame::Format format,
const gfx::Size& dimensions,
base::Time timestamp,
int frame_rate) OVERRIDE;
virtual void OnError() OVERRIDE;
private:
scoped_refptr<Buffer> DoReserveOutputBuffer(media::VideoFrame::Format format,
const gfx::Size& dimensions);
// The controller to which we post events.
const base::WeakPtr<VideoCaptureController> controller_;
// The pool of shared-memory buffers used for capturing.
const scoped_refptr<VideoCaptureBufferPool> buffer_pool_;
};
VideoCaptureController::VideoCaptureController()
: buffer_pool_(new VideoCaptureBufferPool(kNoOfBuffers)),
state_(VIDEO_CAPTURE_STATE_STARTED),
weak_ptr_factory_(this) {
}
VideoCaptureController::VideoCaptureDeviceClient::VideoCaptureDeviceClient(
const base::WeakPtr<VideoCaptureController>& controller,
const scoped_refptr<VideoCaptureBufferPool>& buffer_pool)
: controller_(controller), buffer_pool_(buffer_pool) {}
VideoCaptureController::VideoCaptureDeviceClient::~VideoCaptureDeviceClient() {}
base::WeakPtr<VideoCaptureController> VideoCaptureController::GetWeakPtr() {
return weak_ptr_factory_.GetWeakPtr();
}
scoped_ptr<media::VideoCaptureDevice::Client>
VideoCaptureController::NewDeviceClient() {
scoped_ptr<media::VideoCaptureDevice::Client> result(
new VideoCaptureDeviceClient(this->GetWeakPtr(), buffer_pool_));
return result.Pass();
}
void VideoCaptureController::AddClient(
const VideoCaptureControllerID& id,
VideoCaptureControllerEventHandler* event_handler,
base::ProcessHandle render_process,
media::VideoCaptureSessionId session_id,
const media::VideoCaptureParams& params) {
DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
DVLOG(1) << "VideoCaptureController::AddClient, id " << id.device_id
<< ", " << params.requested_format.frame_size.ToString()
<< ", " << params.requested_format.frame_rate
<< ", " << session_id
<< ")";
// If this is the first client added to the controller, cache the parameters.
if (!controller_clients_.size())
video_capture_format_ = params.requested_format;
// Signal error in case device is already in error state.
if (state_ == VIDEO_CAPTURE_STATE_ERROR) {
event_handler->OnError(id);
return;
}
// Do nothing if this client has called AddClient before.
if (FindClient(id, event_handler, controller_clients_))
return;
ControllerClient* client = new ControllerClient(
id, event_handler, render_process, session_id, params);
// If we already have gotten frame_info from the device, repeat it to the new
// client.
if (state_ == VIDEO_CAPTURE_STATE_STARTED) {
controller_clients_.push_back(client);
return;
}
}
int VideoCaptureController::RemoveClient(
const VideoCaptureControllerID& id,
VideoCaptureControllerEventHandler* event_handler) {
DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
DVLOG(1) << "VideoCaptureController::RemoveClient, id " << id.device_id;
ControllerClient* client = FindClient(id, event_handler, controller_clients_);
if (!client)
return kInvalidMediaCaptureSessionId;
// Take back all buffers held by the |client|.
for (std::set<int>::iterator buffer_it = client->active_buffers.begin();
buffer_it != client->active_buffers.end();
++buffer_it) {
int buffer_id = *buffer_it;
buffer_pool_->RelinquishConsumerHold(buffer_id, 1);
}
client->active_buffers.clear();
int session_id = client->session_id;
controller_clients_.remove(client);
delete client;
return session_id;
}
void VideoCaptureController::StopSession(int session_id) {
DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
DVLOG(1) << "VideoCaptureController::StopSession, id " << session_id;
ControllerClient* client = FindClient(session_id, controller_clients_);
if (client) {
client->session_closed = true;
client->event_handler->OnEnded(client->controller_id);
}
}
void VideoCaptureController::ReturnBuffer(
const VideoCaptureControllerID& id,
VideoCaptureControllerEventHandler* event_handler,
int buffer_id) {
DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
ControllerClient* client = FindClient(id, event_handler, controller_clients_);
// If this buffer is not held by this client, or this client doesn't exist
// in controller, do nothing.
if (!client || !client->active_buffers.erase(buffer_id)) {
NOTREACHED();
return;
}
buffer_pool_->RelinquishConsumerHold(buffer_id, 1);
}
const media::VideoCaptureFormat&
VideoCaptureController::GetVideoCaptureFormat() const {
DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
return video_capture_format_;
}
scoped_refptr<media::VideoCaptureDevice::Client::Buffer>
VideoCaptureController::VideoCaptureDeviceClient::ReserveOutputBuffer(
media::VideoFrame::Format format,
const gfx::Size& size) {
return DoReserveOutputBuffer(format, size);
}
void VideoCaptureController::VideoCaptureDeviceClient::OnIncomingCapturedFrame(
const uint8* data,
int length,
base::Time timestamp,
int rotation,
const VideoCaptureFormat& frame_format) {
TRACE_EVENT0("video", "VideoCaptureController::OnIncomingCapturedFrame");
if (!frame_format.IsValid())
return;
// Chopped pixels in width/height in case video capture device has odd
// numbers for width/height.
#if !defined(AVOID_LIBYUV_FOR_ANDROID_WEBVIEW)
int chopped_width = 0;
int chopped_height = 0;
#endif
int new_unrotated_width = frame_format.frame_size.width();
int new_unrotated_height = frame_format.frame_size.height();
if (new_unrotated_width & 1) {
--new_unrotated_width;
#if !defined(AVOID_LIBYUV_FOR_ANDROID_WEBVIEW)
chopped_width = 1;
#endif
}
if (new_unrotated_height & 1) {
--new_unrotated_height;
#if !defined(AVOID_LIBYUV_FOR_ANDROID_WEBVIEW)
chopped_height = 1;
#endif
}
int destination_width = new_unrotated_width;
int destination_height = new_unrotated_height;
if (rotation == 90 || rotation == 270) {
destination_width = new_unrotated_height;
destination_height = new_unrotated_width;
}
const gfx::Size dimensions(destination_width, destination_height);
scoped_refptr<Buffer> buffer =
DoReserveOutputBuffer(media::VideoFrame::I420, dimensions);
if (!buffer)
return;
#if !defined(AVOID_LIBYUV_FOR_ANDROID_WEBVIEW)
uint8* yplane = reinterpret_cast<uint8*>(buffer->data());
uint8* uplane =
yplane +
media::VideoFrame::PlaneAllocationSize(
media::VideoFrame::I420, media::VideoFrame::kYPlane, dimensions);
uint8* vplane =
uplane +
media::VideoFrame::PlaneAllocationSize(
media::VideoFrame::I420, media::VideoFrame::kUPlane, dimensions);
int yplane_stride = dimensions.width();
int uv_plane_stride = yplane_stride / 2;
int crop_x = 0;
int crop_y = 0;
libyuv::FourCC origin_colorspace = libyuv::FOURCC_ANY;
libyuv::RotationMode rotation_mode = libyuv::kRotate0;
if (rotation == 90)
rotation_mode = libyuv::kRotate90;
else if (rotation == 180)
rotation_mode = libyuv::kRotate180;
else if (rotation == 270)
rotation_mode = libyuv::kRotate270;
switch (frame_format.pixel_format) {
case media::PIXEL_FORMAT_UNKNOWN: // Color format not set.
break;
case media::PIXEL_FORMAT_I420:
DCHECK(!chopped_width && !chopped_height);
origin_colorspace = libyuv::FOURCC_I420;
break;
case media::PIXEL_FORMAT_YV12:
DCHECK(!chopped_width && !chopped_height);
origin_colorspace = libyuv::FOURCC_YV12;
break;
case media::PIXEL_FORMAT_NV21:
DCHECK(!chopped_width && !chopped_height);
origin_colorspace = libyuv::FOURCC_NV21;
break;
case media::PIXEL_FORMAT_YUY2:
DCHECK(!chopped_width && !chopped_height);
origin_colorspace = libyuv::FOURCC_YUY2;
break;
case media::PIXEL_FORMAT_UYVY:
DCHECK(!chopped_width && !chopped_height);
origin_colorspace = libyuv::FOURCC_UYVY;
break;
case media::PIXEL_FORMAT_RGB24:
origin_colorspace = libyuv::FOURCC_RAW;
break;
case media::PIXEL_FORMAT_ARGB:
origin_colorspace = libyuv::FOURCC_ARGB;
break;
case media::PIXEL_FORMAT_MJPEG:
origin_colorspace = libyuv::FOURCC_MJPG;
break;
default:
NOTREACHED();
}
int need_convert_rgb24_on_win = false;
#if defined(OS_WIN)
// TODO(wjia): Use libyuv::ConvertToI420 since support for image inversion
// (vertical flipping) has been added. Use negative src_height as indicator.
if (frame_format.pixel_format == media::PIXEL_FORMAT_RGB24) {
// Rotation is not supported in kRGB24 and OS_WIN case.
DCHECK(!rotation);
need_convert_rgb24_on_win = true;
}
#endif
if (need_convert_rgb24_on_win) {
int rgb_stride = -3 * (new_unrotated_width + chopped_width);
const uint8* rgb_src =
data + 3 * (new_unrotated_width + chopped_width) *
(new_unrotated_height - 1 + chopped_height);
media::ConvertRGB24ToYUV(rgb_src,
yplane,
uplane,
vplane,
new_unrotated_width,
new_unrotated_height,
rgb_stride,
yplane_stride,
uv_plane_stride);
} else {
libyuv::ConvertToI420(data,
length,
yplane,
yplane_stride,
uplane,
uv_plane_stride,
vplane,
uv_plane_stride,
crop_x,
crop_y,
new_unrotated_width + chopped_width,
new_unrotated_height,
new_unrotated_width,
new_unrotated_height,
rotation_mode,
origin_colorspace);
}
#else
// Libyuv is not linked in for Android WebView builds, but video capture is
// not used in those builds either. Whenever libyuv is added in that build,
// address all these #ifdef parts, see http://crbug.com/299611 .
NOTREACHED();
#endif // if !defined(AVOID_LIBYUV_FOR_ANDROID_WEBVIEW)
BrowserThread::PostTask(
BrowserThread::IO,
FROM_HERE,
base::Bind(
&VideoCaptureController::DoIncomingCapturedI420BufferOnIOThread,
controller_,
buffer,
dimensions,
frame_format.frame_rate,
timestamp));
}
void VideoCaptureController::VideoCaptureDeviceClient::OnIncomingCapturedBuffer(
const scoped_refptr<Buffer>& buffer,
media::VideoFrame::Format format,
const gfx::Size& dimensions,
base::Time timestamp,
int frame_rate) {
// The capture pipeline expects I420 for now.
DCHECK_EQ(format, media::VideoFrame::I420)
<< "Non-I420 output buffer returned";
BrowserThread::PostTask(
BrowserThread::IO,
FROM_HERE,
base::Bind(
&VideoCaptureController::DoIncomingCapturedI420BufferOnIOThread,
controller_,
buffer,
dimensions,
frame_rate,
timestamp));
}
void VideoCaptureController::VideoCaptureDeviceClient::OnError() {
BrowserThread::PostTask(BrowserThread::IO,
FROM_HERE,
base::Bind(&VideoCaptureController::DoErrorOnIOThread, controller_));
}
scoped_refptr<media::VideoCaptureDevice::Client::Buffer>
VideoCaptureController::VideoCaptureDeviceClient::DoReserveOutputBuffer(
media::VideoFrame::Format format,
const gfx::Size& dimensions) {
// The capture pipeline expects I420 for now.
DCHECK_EQ(format, media::VideoFrame::I420)
<< "Non-I420 output buffer requested";
int buffer_id_to_drop = VideoCaptureBufferPool::kInvalidId;
const size_t frame_bytes =
media::VideoFrame::AllocationSize(format, dimensions);
int buffer_id =
buffer_pool_->ReserveForProducer(frame_bytes, &buffer_id_to_drop);
if (buffer_id == VideoCaptureBufferPool::kInvalidId)
return NULL;
void* data;
size_t size;
buffer_pool_->GetBufferInfo(buffer_id, &data, &size);
scoped_refptr<media::VideoCaptureDevice::Client::Buffer> output_buffer(
new PoolBuffer(buffer_pool_, buffer_id, data, size));
if (buffer_id_to_drop != VideoCaptureBufferPool::kInvalidId) {
BrowserThread::PostTask(BrowserThread::IO,
FROM_HERE,
base::Bind(&VideoCaptureController::DoBufferDestroyedOnIOThread,
controller_, buffer_id_to_drop));
}
return output_buffer;
}
VideoCaptureController::~VideoCaptureController() {
STLDeleteContainerPointers(controller_clients_.begin(),
controller_clients_.end());
}
void VideoCaptureController::DoIncomingCapturedI420BufferOnIOThread(
scoped_refptr<media::VideoCaptureDevice::Client::Buffer> buffer,
const gfx::Size& dimensions,
int frame_rate,
base::Time timestamp) {
DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
DCHECK_NE(buffer->id(), VideoCaptureBufferPool::kInvalidId);
VideoCaptureFormat frame_format(
dimensions, frame_rate, media::PIXEL_FORMAT_I420);
int count = 0;
if (state_ == VIDEO_CAPTURE_STATE_STARTED) {
for (ControllerClients::iterator client_it = controller_clients_.begin();
client_it != controller_clients_.end(); ++client_it) {
ControllerClient* client = *client_it;
if (client->session_closed)
continue;
bool is_new_buffer = client->known_buffers.insert(buffer->id()).second;
if (is_new_buffer) {
// On the first use of a buffer on a client, share the memory handle.
size_t memory_size = 0;
base::SharedMemoryHandle remote_handle = buffer_pool_->ShareToProcess(
buffer->id(), client->render_process_handle, &memory_size);
client->event_handler->OnBufferCreated(
client->controller_id, remote_handle, memory_size, buffer->id());
}
client->event_handler->OnBufferReady(
client->controller_id, buffer->id(), timestamp, frame_format);
bool inserted = client->active_buffers.insert(buffer->id()).second;
DCHECK(inserted) << "Unexpected duplicate buffer: " << buffer->id();
count++;
}
}
buffer_pool_->HoldForConsumers(buffer->id(), count);
}
void VideoCaptureController::DoErrorOnIOThread() {
DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
state_ = VIDEO_CAPTURE_STATE_ERROR;
for (ControllerClients::iterator client_it = controller_clients_.begin();
client_it != controller_clients_.end(); ++client_it) {
ControllerClient* client = *client_it;
if (client->session_closed)
continue;
client->event_handler->OnError(client->controller_id);
}
}
void VideoCaptureController::DoBufferDestroyedOnIOThread(
int buffer_id_to_drop) {
DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
for (ControllerClients::iterator client_it = controller_clients_.begin();
client_it != controller_clients_.end(); ++client_it) {
ControllerClient* client = *client_it;
if (client->session_closed)
continue;
if (client->known_buffers.erase(buffer_id_to_drop)) {
client->event_handler->OnBufferDestroyed(client->controller_id,
buffer_id_to_drop);
}
}
}
VideoCaptureController::ControllerClient*
VideoCaptureController::FindClient(
const VideoCaptureControllerID& id,
VideoCaptureControllerEventHandler* handler,
const ControllerClients& clients) {
for (ControllerClients::const_iterator client_it = clients.begin();
client_it != clients.end(); ++client_it) {
if ((*client_it)->controller_id == id &&
(*client_it)->event_handler == handler) {
return *client_it;
}
}
return NULL;
}
VideoCaptureController::ControllerClient*
VideoCaptureController::FindClient(
int session_id,
const ControllerClients& clients) {
for (ControllerClients::const_iterator client_it = clients.begin();
client_it != clients.end(); ++client_it) {
if ((*client_it)->session_id == session_id) {
return *client_it;
}
}
return NULL;
}
int VideoCaptureController::GetClientCount() {
DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
return controller_clients_.size();
}
} // namespace content