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