History

Add ComputeFirstPassStats()

angiebird committed 7 years ago

Commit 7ee697a1 parent 9f33d7510 files · +241 −23Browse files

Showing the changes to test (4 of 10 files). Show all

Unified · Split

4 files
  1. test/simple_encode_test.cc +36 −0
  2. test/test-data.mk +1 −0
  3. test/test-data.sha1 +1 −0
  4. test/test.mk +1 −0
test/simple_encode_test.cc+36−0
@@ -0,0 +1,36 @@
1#include <memory>
2#include <vector>
3#include "third_party/googletest/src/include/gtest/gtest.h"
4#include "vp9/simple_encode.h"
5
6namespace {
7
8TEST(SimpleEncode, ComputeFirstPassStats) {
9 int w = 352;
10 int h = 288;
11 int frame_rate_num = 30;
12 int frame_rate_den = 1;
13 int target_bitrate = 200;
14 int num_frames = 17;
15 // TODO(angiebird): Figure out how to upload test video to our codebase
16 FILE *file = fopen("bus_352x288_420_f20_b8.yuv", "r");
17 SimpleEncode simple_encode(w, h, frame_rate_num, frame_rate_den,
18 target_bitrate, num_frames, file);
19 simple_encode.ComputeFirstPassStats();
20 std::vector<std::vector<double>> frame_stats =
21 simple_encode.ObserveFirstPassStats();
22 EXPECT_EQ(frame_stats.size(), static_cast<size_t>(num_frames));
23 size_t data_num = frame_stats[0].size();
24 // Read ObserveFirstPassStats before changing FIRSTPASS_STATS.
25 EXPECT_EQ(data_num, static_cast<size_t>(25));
26 for (size_t i = 0; i < frame_stats.size(); ++i) {
27 EXPECT_EQ(frame_stats[i].size(), data_num);
28 // FIRSTPASS_STATS's first element is frame
29 EXPECT_EQ(frame_stats[i][0], i);
30 // FIRSTPASS_STATS's last element is count, and the count is 1 for single
31 // frame stats
32 EXPECT_EQ(frame_stats[i][data_num - 1], 1);
33 }
34}
35
36} // namespace
test/test-data.mk+1−0
@@ -26,6 +26,7 @@ LIBVPX_TEST_DATA-$(CONFIG_VP9_ENCODER) += noisy_clip_640_360.y4m · 25 unmodified lines
26LIBVPX_TEST_DATA-$(CONFIG_VP9_ENCODER) += rush_hour_444.y4m26LIBVPX_TEST_DATA-$(CONFIG_VP9_ENCODER) += rush_hour_444.y4m
27LIBVPX_TEST_DATA-$(CONFIG_VP9_ENCODER) += screendata.y4m27LIBVPX_TEST_DATA-$(CONFIG_VP9_ENCODER) += screendata.y4m
28LIBVPX_TEST_DATA-$(CONFIG_VP9_ENCODER) += niklas_640_480_30.yuv28LIBVPX_TEST_DATA-$(CONFIG_VP9_ENCODER) += niklas_640_480_30.yuv
29LIBVPX_TEST_DATA-$(CONFIG_RATE_CTRL) += bus_352x288_420_f20_b8.yuv
29 30
30# Test vectors31# Test vectors
31LIBVPX_TEST_DATA-$(CONFIG_VP8_DECODER) += vp80-00-comprehensive-001.ivf32LIBVPX_TEST_DATA-$(CONFIG_VP8_DECODER) += vp80-00-comprehensive-001.ivf
test/test-data.sha1+1−0
@@ -1,3 +1,4 @@
13eaf216d9fc8b4b9bb8c3956311f49a85974806c *bus_352x288_420_f20_b8.yuv
1d5dfb0151c9051f8c85999255645d7a23916d3c0 *hantro_collage_w352h288.yuv2d5dfb0151c9051f8c85999255645d7a23916d3c0 *hantro_collage_w352h288.yuv
2b87815bf86020c592ccc7a846ba2e28ec8043902 *hantro_odd.yuv3b87815bf86020c592ccc7a846ba2e28ec8043902 *hantro_odd.yuv
376024eb753cdac6a5e5703aaea189d35c3c30ac7 *invalid-vp90-2-00-quantizer-00.webm.ivf.s5861_r01-05_b6-.v2.ivf476024eb753cdac6a5e5703aaea189d35c3c30ac7 *invalid-vp90-2-00-quantizer-00.webm.ivf.s5861_r01-05_b6-.v2.ivf
test/test.mk+1−0
@@ -58,6 +58,7 @@ LIBVPX_TEST_SRCS-$(CONFIG_VP9_ENCODER) += svc_test.cc · 57 unmodified lines
58LIBVPX_TEST_SRCS-$(CONFIG_VP9_ENCODER) += svc_test.h58LIBVPX_TEST_SRCS-$(CONFIG_VP9_ENCODER) += svc_test.h
59LIBVPX_TEST_SRCS-$(CONFIG_VP9_ENCODER) += svc_end_to_end_test.cc59LIBVPX_TEST_SRCS-$(CONFIG_VP9_ENCODER) += svc_end_to_end_test.cc
60LIBVPX_TEST_SRCS-$(CONFIG_VP9_ENCODER) += timestamp_test.cc60LIBVPX_TEST_SRCS-$(CONFIG_VP9_ENCODER) += timestamp_test.cc
61LIBVPX_TEST_SRCS-$(CONFIG_RATE_CTRL) += simple_encode_test.cc
61 62
62LIBVPX_TEST_SRCS-yes += decode_test_driver.cc63LIBVPX_TEST_SRCS-yes += decode_test_driver.cc
63LIBVPX_TEST_SRCS-yes += decode_test_driver.h64LIBVPX_TEST_SRCS-yes += decode_test_driver.h

Keyboard shortcuts

General

Show keyboard shortcuts
?
Focus the current search
/

Issues and pull requests

Open filters
f

Issues, pull requests and Inbox

Select next item
j↓
Select previous item
k↑
Open selected item
Enter
Dismiss filters or clear selection
Esc

Repository code

Go to file
t