Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(6)

Side by Side Diff: webrtc/test/testsupport/frame_writer.h

Issue 2700493006: Add optional visualization file writers to VideoProcessor tests. (Closed)
Patch Set: kjellander comments 2. Created 3 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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
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 class FrameWriterImpl : public FrameWriter { 45 // Writes raw I420 frames in sequence.
46 class YuvFrameWriterImpl : public FrameWriter {
46 public: 47 public:
47 // Creates a file handler. The input file is assumed to exist and be readable 48 // Creates a file handler. The input file is assumed to exist and be readable
48 // and the output file must be writable. 49 // and the output file must be writable.
49 // Parameters: 50 // Parameters:
50 // output_filename The file to write. Will be overwritten if already 51 // output_filename The file to write. Will be overwritten if already
51 // existing. 52 // existing.
52 // frame_length_in_bytes The size of each frame. 53 // width, height Size of each frame to read.
53 // For YUV: 3*width*height/2 54 YuvFrameWriterImpl(std::string output_filename, int width, int height);
54 FrameWriterImpl(std::string output_filename, size_t frame_length_in_bytes); 55 ~YuvFrameWriterImpl() override;
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:
62 const std::string output_filename_;
63 size_t frame_length_in_bytes_;
64 const int width_;
65 const int height_;
66 FILE* output_file_;
67 };
68
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
61 private: 81 private:
62 std::string output_filename_; 82 const int frame_rate_;
63 size_t frame_length_in_bytes_;
64 FILE* output_file_;
65 }; 83 };
66 84
67 } // namespace test 85 } // namespace test
68 } // namespace webrtc 86 } // namespace webrtc
69 87
70 #endif // WEBRTC_TEST_TESTSUPPORT_FRAME_WRITER_H_ 88 #endif // WEBRTC_TEST_TESTSUPPORT_FRAME_WRITER_H_
OLDNEW
« no previous file with comments | « webrtc/test/testsupport/frame_reader_unittest.cc ('k') | webrtc/test/testsupport/frame_writer.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698