| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| (...skipping 24 matching lines...) Expand all Loading... |
| 35 virtual bool WriteFrame(uint8_t* frame_buffer) = 0; | 35 virtual bool WriteFrame(uint8_t* frame_buffer) = 0; |
| 36 | 36 |
| 37 // Closes the output file if open. Essentially makes this class impossible | 37 // Closes the output file if open. Essentially makes this class impossible |
| 38 // to use anymore. Will also be invoked by the destructor. | 38 // to use anymore. Will also be invoked by the destructor. |
| 39 virtual void Close() = 0; | 39 virtual void Close() = 0; |
| 40 | 40 |
| 41 // Frame length in bytes of a single frame image. | 41 // Frame length in bytes of a single frame image. |
| 42 virtual size_t FrameLength() = 0; | 42 virtual size_t FrameLength() = 0; |
| 43 }; | 43 }; |
| 44 | 44 |
| 45 // Writes raw I420 frames in sequence. | 45 class FrameWriterImpl : public FrameWriter { |
| 46 class YuvFrameWriterImpl : public FrameWriter { | |
| 47 public: | 46 public: |
| 48 // Creates a file handler. The input file is assumed to exist and be readable | 47 // Creates a file handler. The input file is assumed to exist and be readable |
| 49 // and the output file must be writable. | 48 // and the output file must be writable. |
| 50 // Parameters: | 49 // Parameters: |
| 51 // output_filename The file to write. Will be overwritten if already | 50 // output_filename The file to write. Will be overwritten if already |
| 52 // existing. | 51 // existing. |
| 53 // width, height Size of each frame to read. | 52 // frame_length_in_bytes The size of each frame. |
| 54 YuvFrameWriterImpl(std::string output_filename, int width, int height); | 53 // For YUV: 3*width*height/2 |
| 55 ~YuvFrameWriterImpl() override; | 54 FrameWriterImpl(std::string output_filename, size_t frame_length_in_bytes); |
| 55 ~FrameWriterImpl() override; |
| 56 bool Init() override; | 56 bool Init() override; |
| 57 bool WriteFrame(uint8_t* frame_buffer) override; | 57 bool WriteFrame(uint8_t* frame_buffer) override; |
| 58 void Close() override; | 58 void Close() override; |
| 59 size_t FrameLength() override; | 59 size_t FrameLength() override; |
| 60 | 60 |
| 61 protected: | 61 private: |
| 62 const std::string output_filename_; | 62 std::string output_filename_; |
| 63 size_t frame_length_in_bytes_; | 63 size_t frame_length_in_bytes_; |
| 64 const int width_; | |
| 65 const int height_; | |
| 66 FILE* output_file_; | 64 FILE* output_file_; |
| 67 }; | 65 }; |
| 68 | 66 |
| 69 // Writes raw I420 frames in sequence, but with Y4M file and frame headers for | |
| 70 // more convenient playback in external media players. | |
| 71 class Y4mFrameWriterImpl : public YuvFrameWriterImpl { | |
| 72 public: | |
| 73 Y4mFrameWriterImpl(std::string output_filename, | |
| 74 int width, | |
| 75 int height, | |
| 76 int frame_rate); | |
| 77 ~Y4mFrameWriterImpl() override; | |
| 78 bool Init() override; | |
| 79 bool WriteFrame(uint8_t* frame_buffer) override; | |
| 80 | |
| 81 private: | |
| 82 const int frame_rate_; | |
| 83 }; | |
| 84 | |
| 85 } // namespace test | 67 } // namespace test |
| 86 } // namespace webrtc | 68 } // namespace webrtc |
| 87 | 69 |
| 88 #endif // WEBRTC_TEST_TESTSUPPORT_FRAME_WRITER_H_ | 70 #endif // WEBRTC_TEST_TESTSUPPORT_FRAME_WRITER_H_ |
| OLD | NEW |