1 /*
2  *  Copyright (c) 2011 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 
11 #ifndef WEBRTC_TEST_TESTSUPPORT_FRAME_READER_H_
12 #define WEBRTC_TEST_TESTSUPPORT_FRAME_READER_H_
13 
14 #include <cstdio>
15 #include <string>
16 
17 #include "typedefs.h"
18 
19 namespace webrtc {
20 namespace test {
21 
22 // Handles reading of frames from video files.
23 class FrameReader {
24  public:
~FrameReader()25   virtual ~FrameReader() {}
26 
27   // Initializes the frame reader, i.e. opens the input file.
28   // This must be called before reading of frames has started.
29   // Returns false if an error has occurred, in addition to printing to stderr.
30   virtual bool Init() = 0;
31 
32   // Reads a frame into the supplied buffer, which must contain enough space
33   // for the frame size.
34   // Returns true if there are more frames to read, false if we've already
35   // read the last frame (in the previous call).
36   virtual bool ReadFrame(WebRtc_UWord8* source_buffer) = 0;
37 
38   // Closes the input file if open. Essentially makes this class impossible
39   // to use anymore. Will also be invoked by the destructor.
40   virtual void Close() = 0;
41 
42   // Frame length in bytes of a single frame image.
43   virtual int FrameLength() = 0;
44   // Total number of frames in the input video source.
45   virtual int NumberOfFrames() = 0;
46 };
47 
48 class FrameReaderImpl : public FrameReader {
49  public:
50   // Creates a file handler. The input file is assumed to exist and be readable.
51   // Parameters:
52   //   input_filename          The file to read from.
53   //   frame_length_in_bytes   The size of each frame.
54   //                           For YUV this is 3 * width * height / 2
55   FrameReaderImpl(std::string input_filename, int frame_length_in_bytes);
56   virtual ~FrameReaderImpl();
57   bool Init();
58   bool ReadFrame(WebRtc_UWord8* source_buffer);
59   void Close();
FrameLength()60   int FrameLength() { return frame_length_in_bytes_; }
NumberOfFrames()61   int NumberOfFrames() { return number_of_frames_; }
62 
63  private:
64   std::string input_filename_;
65   int frame_length_in_bytes_;
66   int number_of_frames_;
67   FILE* input_file_;
68 };
69 
70 }  // namespace test
71 }  // namespace webrtc
72 
73 #endif  // WEBRTC_TEST_TESTSUPPORT_FRAME_READER_H_
74