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