1 files changed, 108 insertions, 0 deletions
diff --git a/dom/media/VideoSegment.cpp b/dom/media/VideoSegment.cpp
new file mode 100644
index 0000000000..c34a5fd553
--- /dev/null
+++ b/dom/media/VideoSegment.cpp
@@ -0,0 +1,108 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this file,
+ * You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "VideoSegment.h"
+
+#include "gfx2DGlue.h"
+#include "ImageContainer.h"
+#include "VideoUtils.h"
+#include "mozilla/UniquePtr.h"
+
+namespace mozilla {
+
+using namespace layers;
+
+VideoFrame::VideoFrame(already_AddRefed<Image> aImage,
+                       const gfx::IntSize& aIntrinsicSize)
+    : mImage(aImage),
+      mIntrinsicSize(aIntrinsicSize),
+      mForceBlack(false),
+      mPrincipalHandle(PRINCIPAL_HANDLE_NONE) {}
+
+VideoFrame::VideoFrame()
+    : mIntrinsicSize(0, 0),
+      mForceBlack(false),
+      mPrincipalHandle(PRINCIPAL_HANDLE_NONE) {}
+
+VideoFrame::~VideoFrame() = default;
+
+void VideoFrame::SetNull() {
+  mImage = nullptr;
+  mIntrinsicSize = gfx::IntSize(0, 0);
+  mPrincipalHandle = PRINCIPAL_HANDLE_NONE;
+}
+
+void VideoFrame::TakeFrom(VideoFrame* aFrame) {
+  mImage = std::move(aFrame->mImage);
+  mIntrinsicSize = aFrame->mIntrinsicSize;
+  mForceBlack = aFrame->GetForceBlack();
+  mPrincipalHandle = aFrame->mPrincipalHandle;
+}
+
+/* static */
+already_AddRefed<Image> VideoFrame::CreateBlackImage(
+    const gfx::IntSize& aSize) {
+  RefPtr<ImageContainer> container =
+      MakeAndAddRef<ImageContainer>(ImageContainer::ASYNCHRONOUS);
+  RefPtr<PlanarYCbCrImage> image = container->CreatePlanarYCbCrImage();
+  if (!image) {
+    return nullptr;
+  }
+
+  gfx::IntSize cbcrSize((aSize.width + 1) / 2, (aSize.height + 1) / 2);
+  int yLen = aSize.width * aSize.height;
+  int cbcrLen = cbcrSize.width * cbcrSize.height;
+
+  // Generate a black image.
+  auto frame = MakeUnique<uint8_t[]>(yLen + 2 * cbcrLen);
+  // Fill Y plane.
+  memset(frame.get(), 0x10, yLen);
+  // Fill Cb/Cr planes.
+  memset(frame.get() + yLen, 0x80, 2 * cbcrLen);
+
+  layers::PlanarYCbCrData data;
+  data.mYChannel = frame.get();
+  data.mYStride = aSize.width;
+  data.mCbCrStride = cbcrSize.width;
+  data.mCbChannel = frame.get() + yLen;
+  data.mCrChannel = data.mCbChannel + cbcrLen;
+  data.mPictureRect = gfx::IntRect(0, 0, aSize.width, aSize.height);
+  data.mStereoMode = StereoMode::MONO;
+  data.mYUVColorSpace = gfx::YUVColorSpace::BT601;
+  // This could be made FULL once bug 1568745 is complete. A black pixel being
+  // 0x00, 0x80, 0x80
+  data.mColorRange = gfx::ColorRange::LIMITED;
+  data.mChromaSubsampling = gfx::ChromaSubsampling::HALF_WIDTH_AND_HEIGHT;
+
+  // Copies data, so we can free data.
+  if (!image->CopyData(data)) {
+    return nullptr;
+  }
+
+  return image.forget();
+}
+
+void VideoSegment::AppendFrame(already_AddRefed<Image>&& aImage,
+                               const IntSize& aIntrinsicSize,
+                               const PrincipalHandle& aPrincipalHandle,
+                               bool aForceBlack, TimeStamp aTimeStamp) {
+  VideoChunk* chunk = AppendChunk(0);
+  chunk->mTimeStamp = aTimeStamp;
+  VideoFrame frame(std::move(aImage), aIntrinsicSize);
+  MOZ_ASSERT_IF(!IsNull(), !aTimeStamp.IsNull());
+  frame.SetForceBlack(aForceBlack);
+  frame.SetPrincipalHandle(aPrincipalHandle);
+  chunk->mFrame.TakeFrom(&frame);
+}
+
+VideoSegment::VideoSegment()
+    : MediaSegmentBase<VideoSegment, VideoChunk>(VIDEO) {}
+
+VideoSegment::VideoSegment(VideoSegment&& aSegment)
+    : MediaSegmentBase<VideoSegment, VideoChunk>(std::move(aSegment)) {}
+
+VideoSegment::~VideoSegment() = default;
+
+}  // namespace mozilla