Index: media/gpu/mojo/service/gpu_jpeg_decode_accelerator.cc |
diff --git a/media/gpu/mojo/service/gpu_jpeg_decode_accelerator.cc b/media/gpu/mojo/service/gpu_jpeg_decode_accelerator.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..f848ab97af0f2cf605b304c1431974d504ef615e |
--- /dev/null |
+++ b/media/gpu/mojo/service/gpu_jpeg_decode_accelerator.cc |
@@ -0,0 +1,289 @@ |
+// Copyright 2015 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/gpu/mojo/service/gpu_jpeg_decode_accelerator.h" |
+ |
+#include <stdint.h> |
+ |
+#include <memory> |
+#include <utility> |
+ |
+#include "base/command_line.h" |
+#include "base/containers/hash_tables.h" |
+#include "base/logging.h" |
+#include "base/memory/ptr_util.h" |
+#include "base/memory/shared_memory.h" |
+#include "base/single_thread_task_runner.h" |
+#include "base/threading/thread_task_runner_handle.h" |
+#include "base/trace_event/trace_event.h" |
+#include "build/build_config.h" |
+#include "media/base/media_switches.h" |
+#include "media/filters/jpeg_parser.h" |
+#include "media/gpu/fake_jpeg_decode_accelerator.h" |
+#include "mojo/public/cpp/bindings/strong_binding.h" |
+#include "mojo/public/cpp/system/platform_handle.h" |
+#include "ui/gfx/geometry/size.h" |
+ |
+#if defined(OS_CHROMEOS) |
+#if defined(ARCH_CPU_X86_FAMILY) |
+#include "media/gpu/vaapi_jpeg_decode_accelerator.h" |
+#endif |
+#if defined(USE_V4L2_CODEC) && defined(ARCH_CPU_ARM_FAMILY) |
+#include "media/gpu/v4l2_device.h" |
+#include "media/gpu/v4l2_jpeg_decode_accelerator.h" |
+#endif |
+ |
+#endif |
+ |
+namespace { |
+ |
+std::unique_ptr<media::JpegDecodeAccelerator> CreateV4L2JDA( |
+ scoped_refptr<base::SingleThreadTaskRunner> io_task_runner) { |
+ std::unique_ptr<media::JpegDecodeAccelerator> decoder; |
+#if defined(OS_CHROMEOS) && defined(USE_V4L2_CODEC) && \ |
+ defined(ARCH_CPU_ARM_FAMILY) |
+ scoped_refptr<media::V4L2Device> device = media::V4L2Device::Create(); |
+ if (device) |
+ decoder.reset(new media::V4L2JpegDecodeAccelerator( |
+ device, std::move(io_task_runner))); |
+#endif |
+ return decoder; |
+} |
+ |
+std::unique_ptr<media::JpegDecodeAccelerator> CreateVaapiJDA( |
+ scoped_refptr<base::SingleThreadTaskRunner> io_task_runner) { |
+ std::unique_ptr<media::JpegDecodeAccelerator> decoder; |
+#if defined(OS_CHROMEOS) && defined(ARCH_CPU_X86_FAMILY) |
+ decoder.reset( |
+ new media::VaapiJpegDecodeAccelerator(std::move(io_task_runner))); |
+#endif |
+ return decoder; |
+} |
+ |
+std::unique_ptr<media::JpegDecodeAccelerator> CreateFakeJDA( |
+ scoped_refptr<base::SingleThreadTaskRunner> io_task_runner) { |
+ return base::MakeUnique<media::FakeJpegDecodeAccelerator>( |
+ std::move(io_task_runner)); |
+} |
+ |
+void DecodeFinished(std::unique_ptr<base::SharedMemory> shm) { |
+ // Do nothing. Because VideoFrame is backed by |shm|, the purpose of this |
+ // function is to just keep reference of |shm| to make sure it lives until |
+ // decode finishes. |
+} |
+ |
+bool VerifyDecodeParams(const gfx::Size& coded_size, |
+ mojo::ScopedSharedBufferHandle* output_handle, |
+ uint32_t output_buffer_size) { |
+ const int kJpegMaxDimension = UINT16_MAX; |
+ if (coded_size.IsEmpty() || coded_size.width() > kJpegMaxDimension || |
+ coded_size.height() > kJpegMaxDimension) { |
+ LOG(ERROR) << "invalid coded_size " << coded_size.ToString(); |
+ return false; |
+ } |
+ |
+ if (!output_handle->is_valid()) { |
+ LOG(ERROR) << "invalid output_handle"; |
+ return false; |
+ } |
+ |
+ if (output_buffer_size < |
+ media::VideoFrame::AllocationSize(media::PIXEL_FORMAT_I420, coded_size)) { |
+ LOG(ERROR) << "output_buffer_size is too small: " << output_buffer_size; |
+ return false; |
+ } |
+ |
+ return true; |
+} |
+ |
+} // namespace |
+ |
+namespace media { |
+ |
+// static |
+bool GpuJpegDecodeAcceleratorFactoryProvider:: |
+ IsAcceleratedJpegDecodeSupported() { |
+ auto accelerator_factory_functions = GetAcceleratorFactories(); |
+ for (const auto& create_jda_function : accelerator_factory_functions) { |
+ std::unique_ptr<JpegDecodeAccelerator> accelerator = |
+ create_jda_function.Run(base::ThreadTaskRunnerHandle::Get()); |
+ if (accelerator && accelerator->IsSupported()) |
+ return true; |
+ } |
+ return false; |
+} |
+ |
+// static |
+std::vector<GpuJpegDecodeAcceleratorFactoryProvider::CreateAcceleratorCB> |
+GpuJpegDecodeAcceleratorFactoryProvider::GetAcceleratorFactories() { |
+ // This list is ordered by priority of use. |
+ std::vector<CreateAcceleratorCB> result; |
+ if (base::CommandLine::ForCurrentProcess()->HasSwitch( |
+ switches::kUseFakeJpegDecodeAccelerator)) { |
+ result.push_back(base::Bind(&CreateFakeJDA)); |
+ } else { |
+ result.push_back(base::Bind(&CreateV4L2JDA)); |
+ result.push_back(base::Bind(&CreateVaapiJDA)); |
+ } |
+ return result; |
+} |
+ |
+// static |
+void GpuJpegDecodeAccelerator::Create( |
+ scoped_refptr<base::SingleThreadTaskRunner> io_task_runner, |
+ const service_manager::BindSourceInfo& source_info, |
+ mojom::GpuJpegDecodeAcceleratorRequest request) { |
+ auto* jpeg_decoder = new GpuJpegDecodeAccelerator(std::move(io_task_runner)); |
+ mojo::MakeStrongBinding(base::WrapUnique(jpeg_decoder), std::move(request)); |
+} |
+ |
+GpuJpegDecodeAccelerator::GpuJpegDecodeAccelerator( |
+ scoped_refptr<base::SingleThreadTaskRunner> io_task_runner) |
+ : accelerator_factory_functions_( |
+ GpuJpegDecodeAcceleratorFactoryProvider::GetAcceleratorFactories()), |
+ child_task_runner_(base::ThreadTaskRunnerHandle::Get()), |
+ io_task_runner_(std::move(io_task_runner)) {} |
+ |
+GpuJpegDecodeAccelerator::~GpuJpegDecodeAccelerator() { |
+ DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
+} |
+ |
+void GpuJpegDecodeAccelerator::NotifyDecodeStatus(int32_t bitstream_buffer_id, |
+ mojom::Error error) { |
+ if (io_task_runner_->BelongsToCurrentThread()) { |
+ child_task_runner_->PostTask( |
+ FROM_HERE, |
+ base::Bind(&GpuJpegDecodeAccelerator::NotifyDecodeStatus, |
+ base::Unretained(this), bitstream_buffer_id, error)); |
+ return; |
+ } |
+ |
+ DCHECK(decode_cb_); |
+ std::move(decode_cb_).Run(bitstream_buffer_id, error); |
+} |
+ |
+void GpuJpegDecodeAccelerator::VideoFrameReady(int32_t bitstream_buffer_id) { |
+ DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
+ NotifyDecodeStatus(bitstream_buffer_id, mojom::Error::NO_ERRORS); |
+} |
+ |
+void GpuJpegDecodeAccelerator::NotifyError(int32_t bitstream_buffer_id, |
+ JpegDecodeAccelerator::Error error) { |
+ DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
+ NotifyDecodeStatus(bitstream_buffer_id, static_cast<mojom::Error>(error)); |
+} |
+ |
+void GpuJpegDecodeAccelerator::Initialize(InitializeCallback callback) { |
+ DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
+ |
+ // When adding non-chromeos platforms, VideoCaptureGpuJpegDecoder::Initialize |
+ // needs to be updated. |
+ |
+ std::unique_ptr<JpegDecodeAccelerator> accelerator; |
+ for (const auto& create_jda_function : accelerator_factory_functions_) { |
+ std::unique_ptr<JpegDecodeAccelerator> tmp_accelerator = |
+ create_jda_function.Run(io_task_runner_); |
+ if (tmp_accelerator && tmp_accelerator->Initialize(this)) { |
+ accelerator = std::move(tmp_accelerator); |
+ break; |
+ } |
+ } |
+ |
+ if (!accelerator) { |
+ DLOG(ERROR) << "JPEG accelerator Initialize failed"; |
+ std::move(callback).Run(false); |
+ return; |
+ } |
+ |
+ accelerator_ = std::move(accelerator); |
+ std::move(callback).Run(true); |
+} |
+ |
+void GpuJpegDecodeAccelerator::Decode( |
+ mojom::BitstreamBufferPtr input_buffer, |
+ const gfx::Size& coded_size, |
+ mojo::ScopedSharedBufferHandle output_handle, |
+ uint32_t output_buffer_size, |
+ DecodeCallback callback) { |
+ DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); |
+ decode_cb_ = std::move(callback); |
+ io_task_runner_->PostTask( |
+ FROM_HERE, |
+ base::Bind(&GpuJpegDecodeAccelerator::DecodeOnIOThread, |
+ base::Unretained(this), base::Passed(&input_buffer), |
+ coded_size, base::Passed(&output_handle), output_buffer_size)); |
+} |
+ |
+void GpuJpegDecodeAccelerator::DecodeOnIOThread( |
+ mojom::BitstreamBufferPtr input_buffer, |
+ const gfx::Size& coded_size, |
+ mojo::ScopedSharedBufferHandle output_handle, |
+ uint32_t output_buffer_size) { |
+ DCHECK(io_task_runner_->BelongsToCurrentThread()); |
+ TRACE_EVENT0("jpeg", "GpuJpegDecodeAccelerator::DecodeOnIOThread"); |
+ |
+ if (!VerifyDecodeParams(coded_size, &output_handle, output_buffer_size)) { |
+ NotifyDecodeStatus(input_buffer->id, mojom::Error::INVALID_ARGUMENT); |
+ return; |
+ } |
+ |
+ base::SharedMemoryHandle memory_handle; |
+ size_t memory_size = 0; |
+ bool read_only_flag = false; |
+ |
+ MojoResult result = mojo::UnwrapSharedMemoryHandle( |
+ std::move(output_handle), &memory_handle, &memory_size, &read_only_flag); |
+ DCHECK_EQ(MOJO_RESULT_OK, result); |
+ |
+ std::unique_ptr<base::SharedMemory> output_shm( |
+ new base::SharedMemory(memory_handle, false)); |
+ if (!output_shm->Map(output_buffer_size)) { |
+ LOG(ERROR) << "Could not map output shared memory for input buffer id " |
+ << input_buffer->id; |
+ NotifyDecodeStatus(input_buffer->id, mojom::Error::PLATFORM_FAILURE); |
+ return; |
+ } |
+ |
+ uint8_t* shm_memory = static_cast<uint8_t*>(output_shm->memory()); |
+ scoped_refptr<VideoFrame> frame = VideoFrame::WrapExternalSharedMemory( |
+ PIXEL_FORMAT_I420, // format |
+ coded_size, // coded_size |
+ gfx::Rect(coded_size), // visible_rect |
+ coded_size, // natural_size |
+ shm_memory, // data |
+ output_buffer_size, // data_size |
+ memory_handle, // handle |
+ 0, // data_offset |
+ base::TimeDelta()); // timestamp |
+ if (!frame.get()) { |
+ LOG(ERROR) << "Could not create VideoFrame for input buffer id " |
+ << input_buffer->id; |
+ NotifyDecodeStatus(input_buffer->id, mojom::Error::PLATFORM_FAILURE); |
+ return; |
+ } |
+ frame->AddDestructionObserver( |
+ base::Bind(DecodeFinished, base::Passed(&output_shm))); |
+ |
+ result = mojo::UnwrapSharedMemoryHandle( |
+ std::move(input_buffer->memory_handle), &memory_handle, &memory_size, |
+ &read_only_flag); |
+ DCHECK_EQ(MOJO_RESULT_OK, result); |
+ |
+ BitstreamBuffer bitstream_buffer(input_buffer->id, memory_handle, |
+ input_buffer->size, input_buffer->offset, |
+ input_buffer->timestamp); |
+ |
+ bitstream_buffer.SetDecryptConfig(DecryptConfig( |
+ input_buffer->key_id, input_buffer->iv, input_buffer->subsamples)); |
+ |
+ DCHECK(accelerator_); |
+ accelerator_->Decode(bitstream_buffer, frame); |
+} |
+ |
+void GpuJpegDecodeAccelerator::Uninitialize() { |
+ // TODO(c.padhi): see http://crbug.com/699255. |
+ NOTIMPLEMENTED(); |
+} |
+ |
+} // namespace media |