WebrtcImageBuffer.h (1783B)
1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ 2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */ 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 #ifndef WebrtcImageBuffer_h__ 8 #define WebrtcImageBuffer_h__ 9 10 #include "ImageContainer.h" 11 #include "common_video/include/video_frame_buffer.h" 12 #include "mozilla/RefPtr.h" 13 14 namespace mozilla { 15 namespace layers { 16 class Image; 17 } 18 19 class ImageBuffer : public webrtc::VideoFrameBuffer { 20 public: 21 explicit ImageBuffer(RefPtr<layers::Image>&& aImage) 22 : mImage(std::move(aImage)) {} 23 24 webrtc::scoped_refptr<webrtc::I420BufferInterface> ToI420() override { 25 RefPtr<layers::PlanarYCbCrImage> image = mImage->AsPlanarYCbCrImage(); 26 MOZ_ASSERT(image); 27 if (!image) { 28 // TODO. YUV420 ReadBack, Image only provides a RGB readback. 29 return nullptr; 30 } 31 const layers::PlanarYCbCrData* data = image->GetData(); 32 webrtc::scoped_refptr<webrtc::I420BufferInterface> buf = 33 webrtc::WrapI420Buffer( 34 data->mPictureRect.width, data->mPictureRect.height, 35 data->mYChannel, data->mYStride, data->mCbChannel, 36 data->mCbCrStride, data->mCrChannel, data->mCbCrStride, 37 [image] { /* keep reference alive*/ }); 38 return buf; 39 } 40 41 Type type() const override { return Type::kNative; } 42 43 int width() const override { return mImage->GetSize().width; } 44 45 int height() const override { return mImage->GetSize().height; } 46 47 RefPtr<layers::Image> GetNativeImage() const { return mImage; } 48 49 private: 50 const RefPtr<layers::Image> mImage; 51 }; 52 53 } // namespace mozilla 54 55 #endif // WebrtcImageBuffer_h__