blob: 4ff64be7dc53b826c55779b71265ca0711a2fe80 (
plain)
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
|
/*
* Copyright (c) 2023 The WebRTC project authors. All Rights Reserved.
*
* Use of this source code is governed by a BSD-style license
* that can be found in the LICENSE file in the root of the source
* tree. An additional intellectual property rights grant can be found
* in the file PATENTS. All contributing project authors may
* be found in the AUTHORS file in the root of the source tree.
*/
#ifndef TEST_TESTSUPPORT_Y4M_FRAME_GENERATOR_H_
#define TEST_TESTSUPPORT_Y4M_FRAME_GENERATOR_H_
#include <cstddef>
#include <memory>
#include <string>
#include "absl/strings/string_view.h"
#include "absl/types/optional.h"
#include "api/test/frame_generator_interface.h"
#include "rtc_base/checks.h"
#include "test/testsupport/frame_reader.h"
namespace webrtc {
namespace test {
// Generates frames from a Y4M file. The behaviour when reaching EOF is
// configurable via RepeatMode.
class Y4mFrameGenerator : public FrameGeneratorInterface {
public:
enum class RepeatMode {
// Generate frames from the input file, but it stops generating new frames
// once EOF is reached.
kSingle,
// Generate frames from the input file, when EOF is reached it starts from
// the beginning.
kLoop,
// Generate frames from the input file, when EOF is reached it plays frames
// backwards from the end to the beginning of the file (and vice versa,
// literally doing Ping/Pong between the beginning and the end of the file).
kPingPong,
};
Y4mFrameGenerator(absl::string_view filename, RepeatMode repeat_mode);
~Y4mFrameGenerator() override = default;
VideoFrameData NextFrame() override;
void ChangeResolution(size_t width, size_t height) override;
Resolution GetResolution() const override;
absl::optional<int> fps() const override { return fps_; }
private:
YuvFrameReaderImpl::RepeatMode ToYuvFrameReaderRepeatMode(
RepeatMode repeat_mode) const;
std::unique_ptr<webrtc::test::FrameReader> frame_reader_ = nullptr;
std::string filename_;
size_t width_;
size_t height_;
int fps_;
const RepeatMode repeat_mode_;
};
} // namespace test
} // namespace webrtc
#endif // TEST_TESTSUPPORT_Y4M_FRAME_GENERATOR_H_
|
