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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
#include <algorithm>
#include <vector>
#include <fstream>
#include <iostream>
#include <catch.hpp>
#include "libffmpegthumbnailer/videothumbnailer.h"
#include "libffmpegthumbnailer/imagetypes.h"
#include "libffmpegthumbnailer/histogram.h"
#include "libffmpegthumbnailer/histogramutils.h"
namespace ffmpegthumbnailer
{
TEST_CASE("C++ API Usage")
{
Histogram<int> histogram;
VideoFrame frame;
frame.width = 10;
frame.height = 10;
frame.lineSize = 30;
frame.frameData.resize(300);
VideoThumbnailer videoThumbnailer;
videoThumbnailer.setLogCallback([] (ThumbnailerLogLevel, const std::string& msg) {
std::cout << msg << std::endl;
});
SECTION("CreateThumb")
{
std::string input = std::string(TEST_DATADIR) + "/test_sample.flv";
std::vector<uint8_t> buffer;
videoThumbnailer.generateThumbnail(input, Png, buffer);
CHECK_FALSE(buffer.empty());
}
SECTION("CreateThumbStringSize")
{
std::string input = std::string(TEST_DATADIR) + "/test_sample.flv";
std::vector<uint8_t> buffer;
videoThumbnailer.setThumbnailSize("w=128:h=-1");
auto info = videoThumbnailer.generateThumbnail(input, Png, buffer);
CHECK_FALSE(buffer.empty());
CHECK(128 == info.width);
CHECK(info.height > 0);
}
SECTION("CreateThumbRawData")
{
std::string input = std::string(TEST_DATADIR) + "/test_sample.flv";
std::vector<uint8_t> buffer;
videoThumbnailer.generateThumbnail(input, Rgb, buffer);
CHECK_FALSE(buffer.empty());
}
SECTION("SpecifyInvalidSize")
{
CHECK_THROWS_AS(videoThumbnailer.setThumbnailSize("w="), std::invalid_argument);
CHECK_THROWS_AS(videoThumbnailer.setThumbnailSize("h="), std::invalid_argument);
CHECK_THROWS_AS(videoThumbnailer.setThumbnailSize("w=?"), std::invalid_argument);
CHECK_THROWS_AS(videoThumbnailer.setThumbnailSize("h=?"), std::invalid_argument);
}
SECTION("CreateThumbSpecifyWidth")
{
std::string input = std::string(TEST_DATADIR) + "/test_sample.flv";
std::vector<uint8_t> buffer;
videoThumbnailer.setThumbnailSize("w=64");
videoThumbnailer.generateThumbnail(input, Png, buffer);
CHECK_FALSE(buffer.empty());
}
SECTION("CreateThumbSpecifyHeight")
{
std::string input = std::string(TEST_DATADIR) + "/test_sample.flv";
std::vector<uint8_t> buffer;
videoThumbnailer.setThumbnailSize("h=64");
videoThumbnailer.generateThumbnail(input, Png, buffer);
CHECK_FALSE(buffer.empty());
}
SECTION("CreateThumbNonAscii")
{
std::string input = std::string(TEST_DATADIR) + "/test_Кругом_шумел.flv";
std::vector<uint8_t> buffer;
videoThumbnailer.generateThumbnail(input, Png, buffer);
CHECK_FALSE(buffer.empty());
}
SECTION("CreateThumbInvalidFile")
{
std::vector<uint8_t> buffer;
CHECK_THROWS_AS(videoThumbnailer.generateThumbnail("invalidfile.mpg", Png, buffer), std::logic_error);
}
SECTION("CreateHistogramBlackFrame")
{
std::fill(frame.frameData.begin(), frame.frameData.end(), 0);
utils::generateHistogram(frame, histogram);
CHECK(100 == histogram.r[0]);
CHECK(100 == histogram.g[0]);
CHECK(100 == histogram.b[0]);
for (int i = 1; i < 255; ++i)
{
CHECK(0 == histogram.r[i]);
CHECK(0 == histogram.g[i]);
CHECK(0 == histogram.b[i]);
}
}
SECTION("CreateHistogramWhiteFrame")
{
std::fill(frame.frameData.begin(), frame.frameData.end(), 255);
utils::generateHistogram(frame, histogram);
CHECK(100 == histogram.r[255]);
CHECK(100 == histogram.g[255]);
CHECK(100 == histogram.b[255]);
for (int i = 0; i < 254; ++i)
{
CHECK(0 == histogram.r[i]);
CHECK(0 == histogram.g[i]);
CHECK(0 == histogram.b[i]);
}
}
}
}
|
