712338eed2
VideoFrameBuffer is currently hard coded to be either I420 or Native. This CL makes VideoFrameBuffer more generic by moving the I420 specific functions into their own class, and adds an enum tag that represents the format and storage type of the buffer. Each buffer type is then represented as a subclass. See webrtc/api/video/video_frame_buffer.h for more info. This CL also adds support for representing I444 in VideoFrameBuffer using the new interface. Possible future buffer type candidates are RGB and NV12. BUG=webrtc:7632 TBR=stefan@webrtc.org Review-Url: https://codereview.webrtc.org/2847383002 Cr-Commit-Position: refs/heads/master@{#18098}
104 lines
3.4 KiB
C++
104 lines
3.4 KiB
C++
/*
|
|
* Copyright (c) 2015 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#ifndef WEBRTC_API_VIDEO_I420_BUFFER_H_
|
|
#define WEBRTC_API_VIDEO_I420_BUFFER_H_
|
|
|
|
#include <memory>
|
|
|
|
#include "webrtc/api/video/video_rotation.h"
|
|
#include "webrtc/api/video/video_frame_buffer.h"
|
|
#include "webrtc/system_wrappers/include/aligned_malloc.h"
|
|
|
|
namespace webrtc {
|
|
|
|
// Plain I420 buffer in standard memory.
|
|
class I420Buffer : public PlanarYuvBuffer {
|
|
public:
|
|
static rtc::scoped_refptr<I420Buffer> Create(int width, int height);
|
|
static rtc::scoped_refptr<I420Buffer> Create(int width,
|
|
int height,
|
|
int stride_y,
|
|
int stride_u,
|
|
int stride_v);
|
|
|
|
// Create a new buffer and copy the pixel data.
|
|
static rtc::scoped_refptr<I420Buffer> Copy(const VideoFrameBuffer& buffer);
|
|
|
|
static rtc::scoped_refptr<I420Buffer> Copy(
|
|
int width, int height,
|
|
const uint8_t* data_y, int stride_y,
|
|
const uint8_t* data_u, int stride_u,
|
|
const uint8_t* data_v, int stride_v);
|
|
|
|
// Returns a rotated copy of |src|.
|
|
static rtc::scoped_refptr<I420Buffer> Rotate(const VideoFrameBuffer& src,
|
|
VideoRotation rotation);
|
|
|
|
// Sets the buffer to all black.
|
|
static void SetBlack(I420Buffer* buffer);
|
|
|
|
// Sets all three planes to all zeros. Used to work around for
|
|
// quirks in memory checkers
|
|
// (https://bugs.chromium.org/p/libyuv/issues/detail?id=377) and
|
|
// ffmpeg (http://crbug.com/390941).
|
|
// TODO(nisse): Deprecated. Should be deleted if/when those issues
|
|
// are resolved in a better way. Or in the mean time, use SetBlack.
|
|
void InitializeData();
|
|
|
|
Type type() const override;
|
|
int width() const override;
|
|
int height() const override;
|
|
const uint8_t* DataY() const override;
|
|
const uint8_t* DataU() const override;
|
|
const uint8_t* DataV() const override;
|
|
|
|
int StrideY() const override;
|
|
int StrideU() const override;
|
|
int StrideV() const override;
|
|
|
|
uint8_t* MutableDataY();
|
|
uint8_t* MutableDataU();
|
|
uint8_t* MutableDataV();
|
|
|
|
// Scale the cropped area of |src| to the size of |this| buffer, and
|
|
// write the result into |this|.
|
|
void CropAndScaleFrom(const VideoFrameBuffer& src,
|
|
int offset_x,
|
|
int offset_y,
|
|
int crop_width,
|
|
int crop_height);
|
|
|
|
// The common case of a center crop, when needed to adjust the
|
|
// aspect ratio without distorting the image.
|
|
void CropAndScaleFrom(const VideoFrameBuffer& src);
|
|
|
|
// Scale all of |src| to the size of |this| buffer, with no cropping.
|
|
void ScaleFrom(const VideoFrameBuffer& src);
|
|
|
|
protected:
|
|
I420Buffer(int width, int height);
|
|
I420Buffer(int width, int height, int stride_y, int stride_u, int stride_v);
|
|
|
|
~I420Buffer() override;
|
|
|
|
private:
|
|
const int width_;
|
|
const int height_;
|
|
const int stride_y_;
|
|
const int stride_u_;
|
|
const int stride_v_;
|
|
const std::unique_ptr<uint8_t, AlignedFreeDeleter> data_;
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // WEBRTC_API_VIDEO_I420_BUFFER_H_
|