1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "media/cast/test/utility/video_utility.h"
6
7 #include <math.h>
8 #include <cstdio>
9
10 #include "base/rand_util.h"
11 #include "third_party/libyuv/include/libyuv/compare.h"
12 #include "ui/gfx/size.h"
13
14 namespace media {
15 namespace cast {
16
I420PSNR(const scoped_refptr<media::VideoFrame> & frame1,const scoped_refptr<media::VideoFrame> & frame2)17 double I420PSNR(const scoped_refptr<media::VideoFrame>& frame1,
18 const scoped_refptr<media::VideoFrame>& frame2) {
19 if (frame1->coded_size().width() != frame2->coded_size().width() ||
20 frame1->coded_size().height() != frame2->coded_size().height())
21 return -1;
22
23 return libyuv::I420Psnr(frame1->data(VideoFrame::kYPlane),
24 frame1->stride(VideoFrame::kYPlane),
25 frame1->data(VideoFrame::kUPlane),
26 frame1->stride(VideoFrame::kUPlane),
27 frame1->data(VideoFrame::kVPlane),
28 frame1->stride(VideoFrame::kVPlane),
29 frame2->data(VideoFrame::kYPlane),
30 frame2->stride(VideoFrame::kYPlane),
31 frame2->data(VideoFrame::kUPlane),
32 frame2->stride(VideoFrame::kUPlane),
33 frame2->data(VideoFrame::kVPlane),
34 frame2->stride(VideoFrame::kVPlane),
35 frame1->coded_size().width(),
36 frame1->coded_size().height());
37 }
38
PopulateVideoFrame(VideoFrame * frame,int start_value)39 void PopulateVideoFrame(VideoFrame* frame, int start_value) {
40 int height = frame->coded_size().height();
41 int stride_y = frame->stride(VideoFrame::kYPlane);
42 int stride_u = frame->stride(VideoFrame::kUPlane);
43 int stride_v = frame->stride(VideoFrame::kVPlane);
44 int half_height = (height + 1) / 2;
45 uint8* y_plane = frame->data(VideoFrame::kYPlane);
46 uint8* u_plane = frame->data(VideoFrame::kUPlane);
47 uint8* v_plane = frame->data(VideoFrame::kVPlane);
48
49 // Set Y.
50 for (int j = 0; j < height; ++j) {
51 for (int i = 0; i < stride_y; ++i) {
52 *y_plane = static_cast<uint8>(start_value + i + j);
53 ++y_plane;
54 }
55 }
56
57 // Set U.
58 for (int j = 0; j < half_height; ++j) {
59 for (int i = 0; i < stride_u; ++i) {
60 *u_plane = static_cast<uint8>(start_value + i + j);
61 ++u_plane;
62 }
63 }
64
65 // Set V.
66 for (int j = 0; j < half_height; ++j) {
67 for (int i = 0; i < stride_v; ++i) {
68 *v_plane = static_cast<uint8>(start_value + i + j);
69 ++v_plane;
70 }
71 }
72 }
73
PopulateVideoFrameWithNoise(VideoFrame * frame)74 void PopulateVideoFrameWithNoise(VideoFrame* frame) {
75 int height = frame->coded_size().height();
76 int stride_y = frame->stride(VideoFrame::kYPlane);
77 int stride_u = frame->stride(VideoFrame::kUPlane);
78 int stride_v = frame->stride(VideoFrame::kVPlane);
79 int half_height = (height + 1) / 2;
80 uint8* y_plane = frame->data(VideoFrame::kYPlane);
81 uint8* u_plane = frame->data(VideoFrame::kUPlane);
82 uint8* v_plane = frame->data(VideoFrame::kVPlane);
83
84 base::RandBytes(y_plane, height * stride_y);
85 base::RandBytes(u_plane, half_height * stride_u);
86 base::RandBytes(v_plane, half_height * stride_v);
87 }
88
PopulateVideoFrameFromFile(VideoFrame * frame,FILE * video_file)89 bool PopulateVideoFrameFromFile(VideoFrame* frame, FILE* video_file) {
90 int width = frame->coded_size().width();
91 int height = frame->coded_size().height();
92 int half_width = (width + 1) / 2;
93 int half_height = (height + 1) / 2;
94 size_t frame_size = width * height + 2 * half_width * half_height;
95 uint8* y_plane = frame->data(VideoFrame::kYPlane);
96 uint8* u_plane = frame->data(VideoFrame::kUPlane);
97 uint8* v_plane = frame->data(VideoFrame::kVPlane);
98
99 uint8* raw_data = new uint8[frame_size];
100 size_t count = fread(raw_data, 1, frame_size, video_file);
101 if (count != frame_size)
102 return false;
103
104 memcpy(y_plane, raw_data, width * height);
105 memcpy(u_plane, raw_data + width * height, half_width * half_height);
106 memcpy(v_plane,
107 raw_data + width * height + half_width * half_height,
108 half_width * half_height);
109 delete[] raw_data;
110 return true;
111 }
112
113 } // namespace cast
114 } // namespace media
115