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 |