1 /*
2  *  Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
3  *
4  *  Use of this source code is governed by a BSD-style license
5  *  that can be found in the LICENSE file in the root of the source
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 #ifndef WEBRTC_COMMON_VIDEO_TEST_FRAME_GENERATOR_H_
11 #define WEBRTC_COMMON_VIDEO_TEST_FRAME_GENERATOR_H_
12 
13 #include <string>
14 #include <vector>
15 
16 #include "webrtc/typedefs.h"
17 #include "webrtc/video_frame.h"
18 
19 namespace webrtc {
20 class Clock;
21 namespace test {
22 
23 class FrameGenerator {
24  public:
FrameGenerator()25   FrameGenerator() {}
~FrameGenerator()26   virtual ~FrameGenerator() {}
27 
28   // Returns video frame that remains valid until next call.
29   virtual VideoFrame* NextFrame() = 0;
30 
31   // Creates a test frame generator that creates fully saturated frames with
32   // varying U, V values over time.
33   static FrameGenerator* CreateChromaGenerator(size_t width, size_t height);
34 
35   // Creates a frame generator that repeatedly plays a set of yuv files.
36   // The frame_repeat_count determines how many times each frame is shown,
37   // with 1 = show each frame once, etc.
38   static FrameGenerator* CreateFromYuvFile(std::vector<std::string> files,
39                                            size_t width,
40                                            size_t height,
41                                            int frame_repeat_count);
42 
43   // Creates a frame generator which takes a set of yuv files (wrapping a
44   // frame generator created by CreateFromYuvFile() above), but outputs frames
45   // that have been cropped to specified resolution: source_width/source_height
46   // is the size of the source images, target_width/target_height is the size of
47   // the cropped output. For each source image read, the cropped viewport will
48   // be scrolled top to bottom/left to right for scroll_tim_ms milliseconds.
49   // After that the image will stay in place for pause_time_ms milliseconds,
50   // and then this will be repeated with the next file from the input set.
51   static FrameGenerator* CreateScrollingInputFromYuvFiles(
52       Clock* clock,
53       std::vector<std::string> filenames,
54       size_t source_width,
55       size_t source_height,
56       size_t target_width,
57       size_t target_height,
58       int64_t scroll_time_ms,
59       int64_t pause_time_ms);
60 };
61 }  // namespace test
62 }  // namespace webrtc
63 
64 #endif  // WEBRTC_COMMON_VIDEO_TEST_FRAME_GENERATOR_H_
65