media: Create Mojo StructTraits for VideoFrame

media/mojo/interfaces/video_frame_struct_traits.cc

Index: media/mojo/interfaces/video_frame_struct_traits.cc
diff --git a/media/mojo/interfaces/video_frame_struct_traits.cc b/media/mojo/interfaces/video_frame_struct_traits.cc
new file mode 100644
index 0000000000000000000000000000000000000000..089ac28bcaab9555882de6173d6a618d3b1731ce
--- /dev/null
+++ b/media/mojo/interfaces/video_frame_struct_traits.cc
@@ -0,0 +1,175 @@
+// Copyright 2017 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/mojo/interfaces/video_frame_struct_traits.h"
+
+#include <utility>
+#include <vector>
+
+#include "base/logging.h"
+#include "media/mojo/common/mojo_shared_buffer_video_frame.h"
+
+namespace mojo {
+
+namespace {
+
+struct VideoFrameStructTraitsContext {
+  // Tracked separately because |data| can be null.
+  bool has_data = false;
+  media::mojom::VideoFrameDataPtr data;
+};
+
+media::mojom::VideoFrameDataPtr MakeVideoFrameData(
+    const scoped_refptr<media::VideoFrame>& input) {
+  // EOS frames contain no data.
+  if (input->metadata()->IsTrue(media::VideoFrameMetadata::END_OF_STREAM))
+    return nullptr;
+
+  if (input->storage_type() == media::VideoFrame::STORAGE_MOJO_SHARED_BUFFER) {
+    media::MojoSharedBufferVideoFrame* mojo_frame =
+        static_cast<media::MojoSharedBufferVideoFrame*>(input.get());
+
+    mojo::ScopedSharedBufferHandle dup = mojo_frame->Handle().Clone();
+    DCHECK(dup.is_valid());
+
+    return media::mojom::VideoFrameData::NewSharedBufferData(
+        media::mojom::SharedBufferVideoFrameData::New(
+            std::move(dup), mojo_frame->MappedSize(),
+            mojo_frame->stride(media::VideoFrame::kYPlane),
+            mojo_frame->stride(media::VideoFrame::kUPlane),
+            mojo_frame->stride(media::VideoFrame::kVPlane),
+            mojo_frame->PlaneOffset(media::VideoFrame::kYPlane),
+            mojo_frame->PlaneOffset(media::VideoFrame::kUPlane),
+            mojo_frame->PlaneOffset(media::VideoFrame::kVPlane)));
+  }
+
+  if (input->HasTextures()) {
+    std::vector<gpu::MailboxHolder> mailbox_holder(
+        media::VideoFrame::kMaxPlanes);
+    size_t num_planes = media::VideoFrame::NumPlanes(input->format());
+    for (size_t i = 0; i < num_planes; i++)
+      mailbox_holder[i] = input->mailbox_holder(i);
+    return media::mojom::VideoFrameData::NewMailboxData(
+        media::mojom::MailboxVideoFrameData::New(std::move(mailbox_holder)));
+  }
+
+  NOTREACHED() << "Unsupported VideoFrame conversion";
+  return nullptr;
+}
+
+}  // namespace
+
+// static
+void* StructTraits<media::mojom::VideoFrameDataView,
+                   scoped_refptr<media::VideoFrame>>::
+    SetUpContext(const scoped_refptr<media::VideoFrame>& input) {
+  return new VideoFrameStructTraitsContext();
+}
+
+// static
+void StructTraits<media::mojom::VideoFrameDataView,
+                  scoped_refptr<media::VideoFrame>>::
+    TearDownContext(const scoped_refptr<media::VideoFrame>& input,
+                    void* context) {
+  delete static_cast<VideoFrameStructTraitsContext*>(context);
+}
+
+// static
+media::mojom::VideoFrameDataPtr&
+StructTraits<media::mojom::VideoFrameDataView,
+             scoped_refptr<media::VideoFrame>>::
+    data(const scoped_refptr<media::VideoFrame>& input, void* context) {
+  VideoFrameStructTraitsContext* ctx =
+      static_cast<VideoFrameStructTraitsContext*>(context);
+
+  if (!ctx->has_data) {

[dcheng, 2017/06/08 20:43:15]

Rather than using has_data, just call MakeVideoFrameData in SetUpContext as
well.

+    ctx->data = MakeVideoFrameData(input);
+    ctx->has_data = true;
+  }
+
+  return ctx->data;
+}
+
+// static
+bool StructTraits<media::mojom::VideoFrameDataView,
+                  scoped_refptr<media::VideoFrame>>::
+    Read(media::mojom::VideoFrameDataView input,
+         scoped_refptr<media::VideoFrame>* output) {
+  // View of the |data| member of the input media::mojom::VideoFrame.
+  media::mojom::VideoFrameDataDataView data;
+  input.GetDataDataView(&data);
+
+  DCHECK_EQ(input.end_of_stream(), data.is_null());
+  if (input.end_of_stream()) {
+    *output = media::VideoFrame::CreateEOSFrame();
+    return !!*output;
+  }
+
+  media::VideoPixelFormat format;
+  if (!input.ReadFormat(&format))
+    return false;
+
+  gfx::Size coded_size;
+  if (!input.ReadCodedSize(&coded_size))
+    return false;
+
+  gfx::Rect visible_rect;
+  if (!input.ReadVisibleRect(&visible_rect))
+    return false;
+
+  // Coded size must contain the visible rect.
+  if (visible_rect.right() > coded_size.width() ||
+      visible_rect.bottom() > coded_size.height()) {
+    return false;
+  }
+
+  gfx::Size natural_size;
+  if (!input.ReadNaturalSize(&natural_size))
+    return false;
+
+  base::TimeDelta timestamp;
+  if (!input.ReadTimestamp(&timestamp))
+    return false;
+
+  if (data.is_shared_buffer_data()) {
+    media::mojom::SharedBufferVideoFrameDataDataView shared_buffer_data;
+    data.GetSharedBufferDataDataView(&shared_buffer_data);
+
+    // TODO(sandersd): Conversion from uint64_t to size_t could cause
+    // corruption. Platform-dependent types should be removed from the
+    // implementation (limiting to 32-bit offsets is fine).
+    *output = media::MojoSharedBufferVideoFrame::Create(
+        format, coded_size, visible_rect, natural_size,
+        shared_buffer_data.TakeFrameData(),
+        shared_buffer_data.frame_data_size(), shared_buffer_data.y_offset(),
+        shared_buffer_data.u_offset(), shared_buffer_data.v_offset(),
+        shared_buffer_data.y_stride(), shared_buffer_data.u_stride(),
+        shared_buffer_data.v_stride(), timestamp);
+    return !!*output;
+  }
+
+  if (data.is_mailbox_data()) {
+    media::mojom::MailboxVideoFrameDataDataView mailbox_data;
+    data.GetMailboxDataDataView(&mailbox_data);
+
+    std::vector<gpu::MailboxHolder> mailbox_holder;
+    if (!mailbox_data.ReadMailboxHolder(&mailbox_holder))
+      return false;
+
+    gpu::MailboxHolder mailbox_holder_array[media::VideoFrame::kMaxPlanes];
+    for (size_t i = 0; i < media::VideoFrame::kMaxPlanes; i++)
+      mailbox_holder_array[i] = mailbox_holder[i];
+
+    *output = media::VideoFrame::WrapNativeTextures(
+        format, mailbox_holder_array, media::VideoFrame::ReleaseMailboxCB(),
+        coded_size, visible_rect, natural_size, timestamp);
+    return !!*output;
+  }
+
+  // TODO(sandersd): Switch on the union tag to avoid this ugliness?
+  NOTREACHED();
+  return false;
+}
+
+}  // namespace mojo