OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
(...skipping 29 matching lines...) Expand all Loading... |
40 | 40 |
41 src_width_ = src_width; | 41 src_width_ = src_width; |
42 src_height_ = src_height; | 42 src_height_ = src_height; |
43 dst_width_ = dst_width; | 43 dst_width_ = dst_width; |
44 dst_height_ = dst_height; | 44 dst_height_ = dst_height; |
45 method_ = method; | 45 method_ = method; |
46 set_ = true; | 46 set_ = true; |
47 return 0; | 47 return 0; |
48 } | 48 } |
49 | 49 |
50 // TODO(nisse): Should work with VideoFrameBuffer instead. | |
51 int Scaler::Scale(const VideoFrame& src_frame, VideoFrame* dst_frame) { | 50 int Scaler::Scale(const VideoFrame& src_frame, VideoFrame* dst_frame) { |
52 assert(dst_frame); | 51 assert(dst_frame); |
53 if (src_frame.IsZeroSize()) | 52 if (src_frame.IsZeroSize()) |
54 return -1; | 53 return -1; |
55 if (!set_) | 54 if (!set_) |
56 return -2; | 55 return -2; |
57 | 56 |
58 // Making sure that destination frame is of sufficient size. | 57 // Making sure that destination frame is of sufficient size. |
59 dst_frame->set_video_frame_buffer( | 58 dst_frame->set_video_frame_buffer( |
60 buffer_pool_.CreateBuffer(dst_width_, dst_height_)); | 59 buffer_pool_.CreateBuffer(dst_width_, dst_height_)); |
61 | 60 |
62 // We want to preserve aspect ratio instead of stretching the frame. | 61 // We want to preserve aspect ratio instead of stretching the frame. |
63 // Therefore, we need to crop the source frame. Calculate the largest center | 62 // Therefore, we need to crop the source frame. Calculate the largest center |
64 // aligned region of the source frame that can be used. | 63 // aligned region of the source frame that can be used. |
65 const int cropped_src_width = | 64 const int cropped_src_width = |
66 std::min(src_width_, dst_width_ * src_height_ / dst_height_); | 65 std::min(src_width_, dst_width_ * src_height_ / dst_height_); |
67 const int cropped_src_height = | 66 const int cropped_src_height = |
68 std::min(src_height_, dst_height_ * src_width_ / dst_width_); | 67 std::min(src_height_, dst_height_ * src_width_ / dst_width_); |
69 // Make sure the offsets are even to avoid rounding errors for the U/V planes. | 68 // Make sure the offsets are even to avoid rounding errors for the U/V planes. |
70 const int src_offset_x = ((src_width_ - cropped_src_width) / 2) & ~1; | 69 const int src_offset_x = ((src_width_ - cropped_src_width) / 2) & ~1; |
71 const int src_offset_y = ((src_height_ - cropped_src_height) / 2) & ~1; | 70 const int src_offset_y = ((src_height_ - cropped_src_height) / 2) & ~1; |
72 | 71 |
73 const uint8_t* y_ptr = | 72 const uint8_t* y_ptr = src_frame.buffer(kYPlane) + |
74 src_frame.video_frame_buffer()->DataY() + | 73 src_offset_y * src_frame.stride(kYPlane) + |
75 src_offset_y * src_frame.video_frame_buffer()->StrideY() + | 74 src_offset_x; |
76 src_offset_x; | 75 const uint8_t* u_ptr = src_frame.buffer(kUPlane) + |
77 const uint8_t* u_ptr = | 76 src_offset_y / 2 * src_frame.stride(kUPlane) + |
78 src_frame.video_frame_buffer()->DataU() + | 77 src_offset_x / 2; |
79 src_offset_y / 2 * src_frame.video_frame_buffer()->StrideU() + | 78 const uint8_t* v_ptr = src_frame.buffer(kVPlane) + |
80 src_offset_x / 2; | 79 src_offset_y / 2 * src_frame.stride(kVPlane) + |
81 const uint8_t* v_ptr = | 80 src_offset_x / 2; |
82 src_frame.video_frame_buffer()->DataV() + | |
83 src_offset_y / 2 * src_frame.video_frame_buffer()->StrideV() + | |
84 src_offset_x / 2; | |
85 | 81 |
86 return libyuv::I420Scale( | 82 return libyuv::I420Scale(y_ptr, |
87 y_ptr, | 83 src_frame.stride(kYPlane), |
88 src_frame.video_frame_buffer()->StrideY(), | 84 u_ptr, |
89 u_ptr, | 85 src_frame.stride(kUPlane), |
90 src_frame.video_frame_buffer()->StrideU(), | 86 v_ptr, |
91 v_ptr, | 87 src_frame.stride(kVPlane), |
92 src_frame.video_frame_buffer()->StrideV(), | 88 cropped_src_width, cropped_src_height, |
93 cropped_src_width, cropped_src_height, | 89 dst_frame->buffer(kYPlane), |
94 dst_frame->video_frame_buffer()->MutableDataY(), | 90 dst_frame->stride(kYPlane), |
95 dst_frame->video_frame_buffer()->StrideY(), | 91 dst_frame->buffer(kUPlane), |
96 dst_frame->video_frame_buffer()->MutableDataU(), | 92 dst_frame->stride(kUPlane), |
97 dst_frame->video_frame_buffer()->StrideU(), | 93 dst_frame->buffer(kVPlane), |
98 dst_frame->video_frame_buffer()->MutableDataV(), | 94 dst_frame->stride(kVPlane), |
99 dst_frame->video_frame_buffer()->StrideV(), | 95 dst_width_, dst_height_, |
100 dst_width_, dst_height_, | 96 libyuv::FilterMode(method_)); |
101 libyuv::FilterMode(method_)); | |
102 } | 97 } |
103 | 98 |
104 bool Scaler::SupportedVideoType(VideoType src_video_type, | 99 bool Scaler::SupportedVideoType(VideoType src_video_type, |
105 VideoType dst_video_type) { | 100 VideoType dst_video_type) { |
106 if (src_video_type != dst_video_type) | 101 if (src_video_type != dst_video_type) |
107 return false; | 102 return false; |
108 | 103 |
109 if ((src_video_type == kI420) || (src_video_type == kIYUV) || | 104 if ((src_video_type == kI420) || (src_video_type == kIYUV) || |
110 (src_video_type == kYV12)) | 105 (src_video_type == kYV12)) |
111 return true; | 106 return true; |
112 | 107 |
113 return false; | 108 return false; |
114 } | 109 } |
115 | 110 |
116 } // namespace webrtc | 111 } // namespace webrtc |
OLD | NEW |