| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2016 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 |
| 11 #include "webrtc/modules/video_coding/utility/ivf_file_writer.h" | 11 #include "webrtc/modules/video_coding/utility/ivf_file_writer.h" |
| 12 | 12 |
| 13 #include "webrtc/base/checks.h" | 13 #include "webrtc/base/checks.h" |
| 14 #include "webrtc/base/logging.h" | 14 #include "webrtc/base/logging.h" |
| 15 #include "webrtc/modules/rtp_rtcp/source/byte_io.h" | 15 #include "webrtc/modules/rtp_rtcp/source/byte_io.h" |
| 16 | 16 |
| 17 namespace webrtc { | 17 namespace webrtc { |
| 18 | 18 |
| 19 IvfFileWriter::IvfFileWriter(const std::string& file_name, | 19 IvfFileWriter::IvfFileWriter(const std::string& file_name, |
| 20 std::unique_ptr<FileWrapper> file, | 20 std::unique_ptr<FileWrapper> file, |
| 21 VideoCodecType codec_type) | 21 RtpVideoCodecTypes codec_type) |
| 22 : codec_type_(codec_type), | 22 : codec_type_(codec_type), |
| 23 num_frames_(0), | 23 num_frames_(0), |
| 24 width_(0), | 24 width_(0), |
| 25 height_(0), | 25 height_(0), |
| 26 last_timestamp_(-1), | 26 last_timestamp_(-1), |
| 27 using_capture_timestamps_(false), | 27 using_capture_timestamps_(false), |
| 28 file_name_(file_name), | 28 file_name_(file_name), |
| 29 file_(std::move(file)) {} | 29 file_(std::move(file)) {} |
| 30 | 30 |
| 31 IvfFileWriter::~IvfFileWriter() { | 31 IvfFileWriter::~IvfFileWriter() { |
| 32 Close(); | 32 Close(); |
| 33 } | 33 } |
| 34 | 34 |
| 35 const size_t kIvfHeaderSize = 32; | 35 const size_t kIvfHeaderSize = 32; |
| 36 | 36 |
| 37 std::unique_ptr<IvfFileWriter> IvfFileWriter::Open(const std::string& file_name, | 37 std::unique_ptr<IvfFileWriter> IvfFileWriter::Open( |
| 38 VideoCodecType codec_type) { | 38 const std::string& file_name, |
| 39 RtpVideoCodecTypes codec_type) { |
| 39 std::unique_ptr<IvfFileWriter> file_writer; | 40 std::unique_ptr<IvfFileWriter> file_writer; |
| 40 std::unique_ptr<FileWrapper> file(FileWrapper::Create()); | 41 std::unique_ptr<FileWrapper> file(FileWrapper::Create()); |
| 41 if (file->OpenFile(file_name.c_str(), false) != 0) | 42 if (file->OpenFile(file_name.c_str(), false) != 0) |
| 42 return file_writer; | 43 return file_writer; |
| 43 | 44 |
| 44 file_writer.reset(new IvfFileWriter( | 45 file_writer.reset(new IvfFileWriter( |
| 45 file_name, std::unique_ptr<FileWrapper>(std::move(file)), codec_type)); | 46 file_name, std::unique_ptr<FileWrapper>(std::move(file)), codec_type)); |
| 46 if (!file_writer->WriteHeader()) | 47 if (!file_writer->WriteHeader()) |
| 47 file_writer.reset(); | 48 file_writer.reset(); |
| 48 | 49 |
| 49 return file_writer; | 50 return file_writer; |
| 50 } | 51 } |
| 51 | 52 |
| 52 bool IvfFileWriter::WriteHeader() { | 53 bool IvfFileWriter::WriteHeader() { |
| 53 if (file_->Rewind() != 0) { | 54 if (file_->Rewind() != 0) { |
| 54 LOG(LS_WARNING) << "Unable to rewind output file " << file_name_; | 55 LOG(LS_WARNING) << "Unable to rewind output file " << file_name_; |
| 55 return false; | 56 return false; |
| 56 } | 57 } |
| 57 | 58 |
| 58 uint8_t ivf_header[kIvfHeaderSize] = {0}; | 59 uint8_t ivf_header[kIvfHeaderSize] = {0}; |
| 59 ivf_header[0] = 'D'; | 60 ivf_header[0] = 'D'; |
| 60 ivf_header[1] = 'K'; | 61 ivf_header[1] = 'K'; |
| 61 ivf_header[2] = 'I'; | 62 ivf_header[2] = 'I'; |
| 62 ivf_header[3] = 'F'; | 63 ivf_header[3] = 'F'; |
| 63 ByteWriter<uint16_t>::WriteLittleEndian(&ivf_header[4], 0); // Version. | 64 ByteWriter<uint16_t>::WriteLittleEndian(&ivf_header[4], 0); // Version. |
| 64 ByteWriter<uint16_t>::WriteLittleEndian(&ivf_header[6], 32); // Header size. | 65 ByteWriter<uint16_t>::WriteLittleEndian(&ivf_header[6], 32); // Header size. |
| 65 | 66 |
| 66 switch (codec_type_) { | 67 switch (codec_type_) { |
| 67 case kVideoCodecVP8: | 68 case kRtpVideoVp8: |
| 68 ivf_header[8] = 'V'; | 69 ivf_header[8] = 'V'; |
| 69 ivf_header[9] = 'P'; | 70 ivf_header[9] = 'P'; |
| 70 ivf_header[10] = '8'; | 71 ivf_header[10] = '8'; |
| 71 ivf_header[11] = '0'; | 72 ivf_header[11] = '0'; |
| 72 break; | 73 break; |
| 73 case kVideoCodecVP9: | 74 case kRtpVideoVp9: |
| 74 ivf_header[8] = 'V'; | 75 ivf_header[8] = 'V'; |
| 75 ivf_header[9] = 'P'; | 76 ivf_header[9] = 'P'; |
| 76 ivf_header[10] = '9'; | 77 ivf_header[10] = '9'; |
| 77 ivf_header[11] = '0'; | 78 ivf_header[11] = '0'; |
| 78 break; | 79 break; |
| 79 case kVideoCodecH264: | 80 case kRtpVideoH264: |
| 80 ivf_header[8] = 'H'; | 81 ivf_header[8] = 'H'; |
| 81 ivf_header[9] = '2'; | 82 ivf_header[9] = '2'; |
| 82 ivf_header[10] = '6'; | 83 ivf_header[10] = '6'; |
| 83 ivf_header[11] = '4'; | 84 ivf_header[11] = '4'; |
| 84 break; | 85 break; |
| 85 default: | 86 default: |
| 86 LOG(LS_ERROR) << "Unknown CODEC type: " << codec_type_; | 87 LOG(LS_ERROR) << "Unknown CODEC type: " << codec_type_; |
| 87 return false; | 88 return false; |
| 88 } | 89 } |
| 89 | 90 |
| (...skipping 97 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 187 if (remove(file_name_.c_str()) != 0) | 188 if (remove(file_name_.c_str()) != 0) |
| 188 LOG(LS_WARNING) << "Failed to remove empty IVF file " << file_name_; | 189 LOG(LS_WARNING) << "Failed to remove empty IVF file " << file_name_; |
| 189 | 190 |
| 190 return true; | 191 return true; |
| 191 } | 192 } |
| 192 | 193 |
| 193 return WriteHeader() && (file_->CloseFile() == 0); | 194 return WriteHeader() && (file_->CloseFile() == 0); |
| 194 } | 195 } |
| 195 | 196 |
| 196 } // namespace webrtc | 197 } // namespace webrtc |
| OLD | NEW |