Line data    Source code

       1             : /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
       2             : /* vim: set sw=2 ts=8 et ft=cpp : */
       3             : /* This Source Code Form is subject to the terms of the Mozilla Public
       4             :  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
       5             :  * You can obtain one at http://mozilla.org/MPL/2.0/. */
       6             : 
       7             : #include "VideoFrameUtils.h"
       8             : #include "webrtc/video_frame.h"
       9             : #include "mozilla/ShmemPool.h"
      10             : 
      11             : namespace mozilla {
      12             : 
      13             : size_t
      14           0 : VideoFrameUtils::TotalRequiredBufferSize(
      15             :                   const webrtc::VideoFrame& aVideoFrame)
      16             : {
      17           0 :   auto height = aVideoFrame.video_frame_buffer()->height();
      18           0 :   return height * aVideoFrame.video_frame_buffer()->StrideY() +
      19           0 :     ((height+1)/2) * aVideoFrame.video_frame_buffer()->StrideU() +
      20           0 :     ((height+1)/2) * aVideoFrame.video_frame_buffer()->StrideV();
      21             : }
      22             : 
      23           0 : void VideoFrameUtils::InitFrameBufferProperties(
      24             :                   const webrtc::VideoFrame& aVideoFrame,
      25             :                   camera::VideoFrameProperties& aDestProps)
      26             : {
      27             :   // The VideoFrameBuffer image data stored in the accompanying buffer
      28             :   // the buffer is at least this size of larger.
      29           0 :   aDestProps.bufferSize() = TotalRequiredBufferSize(aVideoFrame);
      30             : 
      31           0 :   aDestProps.timeStamp() = aVideoFrame.timestamp();
      32           0 :   aDestProps.ntpTimeMs() = aVideoFrame.ntp_time_ms();
      33           0 :   aDestProps.renderTimeMs() = aVideoFrame.render_time_ms();
      34             : 
      35           0 :   aDestProps.rotation() = aVideoFrame.rotation();
      36             : 
      37           0 :   auto height = aVideoFrame.video_frame_buffer()->height();
      38           0 :   aDestProps.yAllocatedSize() = height * aVideoFrame.video_frame_buffer()->StrideY();
      39           0 :   aDestProps.uAllocatedSize() = ((height+1)/2) * aVideoFrame.video_frame_buffer()->StrideU();
      40           0 :   aDestProps.vAllocatedSize() = ((height+1)/2) * aVideoFrame.video_frame_buffer()->StrideV();
      41             : 
      42           0 :   aDestProps.width() = aVideoFrame.video_frame_buffer()->width();
      43           0 :   aDestProps.height() = height;
      44             : 
      45           0 :   aDestProps.yStride() = aVideoFrame.video_frame_buffer()->StrideY();
      46           0 :   aDestProps.uStride() = aVideoFrame.video_frame_buffer()->StrideU();
      47           0 :   aDestProps.vStride() = aVideoFrame.video_frame_buffer()->StrideV();
      48           0 : }
      49             : 
      50           0 : void VideoFrameUtils::CopyVideoFrameBuffers(uint8_t* aDestBuffer,
      51             :                        const size_t aDestBufferSize,
      52             :                        const webrtc::VideoFrame& aFrame)
      53             : {
      54           0 :   size_t aggregateSize = TotalRequiredBufferSize(aFrame);
      55             : 
      56           0 :   MOZ_ASSERT(aDestBufferSize >= aggregateSize);
      57             : 
      58             :   // If planes are ordered YUV and contiguous then do a single copy
      59           0 :   if ((aFrame.video_frame_buffer()->DataY() != nullptr)
      60             :       // Check that the three planes are ordered
      61           0 :       && (aFrame.video_frame_buffer()->DataY() < aFrame.video_frame_buffer()->DataU())
      62           0 :       && (aFrame.video_frame_buffer()->DataU() < aFrame.video_frame_buffer()->DataV())
      63             :       //  Check that the last plane ends at firstPlane[totalsize]
      64           0 :       && (&aFrame.video_frame_buffer()->DataY()[aggregateSize] ==
      65           0 :           &aFrame.video_frame_buffer()->DataV()[((aFrame.video_frame_buffer()->height()+1)/2) *
      66           0 :                                                 aFrame.video_frame_buffer()->StrideV()]))
      67             :   {
      68           0 :     memcpy(aDestBuffer, aFrame.video_frame_buffer()->DataY(), aggregateSize);
      69           0 :     return;
      70             :   }
      71             : 
      72             :   // Copy each plane
      73           0 :   size_t offset = 0;
      74             :   size_t size;
      75           0 :   auto height = aFrame.video_frame_buffer()->height();
      76           0 :   size = height * aFrame.video_frame_buffer()->StrideY();
      77           0 :   memcpy(&aDestBuffer[offset], aFrame.video_frame_buffer()->DataY(), size);
      78           0 :   offset += size;
      79           0 :   size = ((height+1)/2) * aFrame.video_frame_buffer()->StrideU();
      80           0 :   memcpy(&aDestBuffer[offset], aFrame.video_frame_buffer()->DataU(), size);
      81           0 :   offset += size;
      82           0 :   size = ((height+1)/2) * aFrame.video_frame_buffer()->StrideV();
      83           0 :   memcpy(&aDestBuffer[offset], aFrame.video_frame_buffer()->DataV(), size);
      84             : }
      85             : 
      86           0 : void VideoFrameUtils::CopyVideoFrameBuffers(ShmemBuffer& aDestShmem,
      87             :                         const webrtc::VideoFrame& aVideoFrame)
      88             : {
      89           0 :   CopyVideoFrameBuffers(aDestShmem.Get().get<uint8_t>(), aDestShmem.Get().Size<uint8_t>(), aVideoFrame);
      90           0 : }
      91             : 
      92           9 : }