| /* |
| * Copyright (c) 2012 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. |
| */ |
| |
| /* |
| * WebRTC's wrapper to libyuv. |
| */ |
| |
| #ifndef WEBRTC_COMMON_VIDEO_LIBYUV_INCLUDE_WEBRTC_LIBYUV_H_ |
| #define WEBRTC_COMMON_VIDEO_LIBYUV_INCLUDE_WEBRTC_LIBYUV_H_ |
| |
| #include <stdio.h> |
| |
| #include "webrtc/common_types.h" // RawVideoTypes. |
| #include "webrtc/common_video/rotation.h" |
| #include "webrtc/typedefs.h" |
| #include "webrtc/video_frame.h" |
| |
| namespace webrtc { |
| |
| // Supported video types. |
| enum VideoType { |
| kUnknown, |
| kI420, |
| kIYUV, |
| kRGB24, |
| kABGR, |
| kARGB, |
| kARGB4444, |
| kRGB565, |
| kARGB1555, |
| kYUY2, |
| kYV12, |
| kUYVY, |
| kMJPG, |
| kNV21, |
| kNV12, |
| kBGRA, |
| }; |
| |
| // This is the max PSNR value our algorithms can return. |
| const double kPerfectPSNR = 48.0f; |
| |
| // Conversion between the RawVideoType and the LibYuv videoType. |
| // TODO(wu): Consolidate types into one type throughout WebRtc. |
| VideoType RawVideoTypeToCommonVideoVideoType(RawVideoType type); |
| |
| // Calculate the required buffer size. |
| // Input: |
| // - type :The type of the designated video frame. |
| // - width :frame width in pixels. |
| // - height :frame height in pixels. |
| // Return value: :The required size in bytes to accommodate the specified |
| // video frame. |
| size_t CalcBufferSize(VideoType type, int width, int height); |
| |
| // TODO(mikhal): Add unit test for these two functions and determine location. |
| // Print VideoFrame to file |
| // Input: |
| // - frame : Reference to video frame. |
| // - file : pointer to file object. It is assumed that the file is |
| // already open for writing. |
| // Return value: 0 if OK, < 0 otherwise. |
| int PrintVideoFrame(const VideoFrame& frame, FILE* file); |
| |
| // Extract buffer from VideoFrame or VideoFrameBuffer (consecutive |
| // planes, no stride) |
| // Input: |
| // - frame : Reference to video frame. |
| // - size : pointer to the size of the allocated buffer. If size is |
| // insufficient, an error will be returned. |
| // - buffer : Pointer to buffer |
| // Return value: length of buffer if OK, < 0 otherwise. |
| int ExtractBuffer(const rtc::scoped_refptr<VideoFrameBuffer>& input_frame, |
| size_t size, |
| uint8_t* buffer); |
| int ExtractBuffer(const VideoFrame& input_frame, size_t size, uint8_t* buffer); |
| // Convert To I420 |
| // Input: |
| // - src_video_type : Type of input video. |
| // - src_frame : Pointer to a source frame. |
| // - crop_x/crop_y : Starting positions for cropping (0 for no crop). |
| // - src_width : src width in pixels. |
| // - src_height : src height in pixels. |
| // - sample_size : Required only for the parsing of MJPG (set to 0 else). |
| // - rotate : Rotation mode of output image. |
| // Output: |
| // - dst_frame : Reference to a destination frame. |
| // Return value: 0 if OK, < 0 otherwise. |
| |
| int ConvertToI420(VideoType src_video_type, |
| const uint8_t* src_frame, |
| int crop_x, |
| int crop_y, |
| int src_width, |
| int src_height, |
| size_t sample_size, |
| VideoRotation rotation, |
| VideoFrame* dst_frame); |
| |
| // Convert From I420 |
| // Input: |
| // - src_frame : Reference to a source frame. |
| // - dst_video_type : Type of output video. |
| // - dst_sample_size : Required only for the parsing of MJPG. |
| // - dst_frame : Pointer to a destination frame. |
| // Return value: 0 if OK, < 0 otherwise. |
| // It is assumed that source and destination have equal height. |
| int ConvertFromI420(const VideoFrame& src_frame, |
| VideoType dst_video_type, |
| int dst_sample_size, |
| uint8_t* dst_frame); |
| |
| // Compute PSNR for an I420 frame (all planes). |
| // Returns the PSNR in decibel, to a maximum of kInfinitePSNR. |
| double I420PSNR(const VideoFrame* ref_frame, const VideoFrame* test_frame); |
| // Compute SSIM for an I420 frame (all planes). |
| double I420SSIM(const VideoFrame* ref_frame, const VideoFrame* test_frame); |
| |
| } // namespace webrtc |
| |
| #endif // WEBRTC_COMMON_VIDEO_LIBYUV_INCLUDE_WEBRTC_LIBYUV_H_ |