1 /* 2 * Copyright (c) 2012 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_WRITER_H_ 12 #define WEBRTC_TEST_TESTSUPPORT_FRAME_WRITER_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 writing of video files. 24 class FrameWriter { 25 public: ~FrameWriter()26 virtual ~FrameWriter() {} 27 28 // Initializes the file handler, i.e. opens the input and output files etc. 29 // This must be called before reading or writing frames has started. 30 // Returns false if an error has occurred, in addition to printing to stderr. 31 virtual bool Init() = 0; 32 33 // Writes a frame of the configured frame length to the output file. 34 // Returns true if the write was successful, false otherwise. 35 virtual bool WriteFrame(uint8_t* frame_buffer) = 0; 36 37 // Closes the output file if open. Essentially makes this class impossible 38 // to use anymore. Will also be invoked by the destructor. 39 virtual void Close() = 0; 40 41 // Frame length in bytes of a single frame image. 42 virtual size_t FrameLength() = 0; 43 }; 44 45 class FrameWriterImpl : public FrameWriter { 46 public: 47 // Creates a file handler. The input file is assumed to exist and be readable 48 // and the output file must be writable. 49 // Parameters: 50 // output_filename The file to write. Will be overwritten if already 51 // existing. 52 // frame_length_in_bytes The size of each frame. 53 // For YUV: 3*width*height/2 54 FrameWriterImpl(std::string output_filename, size_t frame_length_in_bytes); 55 ~FrameWriterImpl() override; 56 bool Init() override; 57 bool WriteFrame(uint8_t* frame_buffer) override; 58 void Close() override; 59 size_t FrameLength() override; 60 61 private: 62 std::string output_filename_; 63 size_t frame_length_in_bytes_; 64 FILE* output_file_; 65 }; 66 67 } // namespace test 68 } // namespace webrtc 69 70 #endif // WEBRTC_TEST_TESTSUPPORT_FRAME_WRITER_H_ 71