OLD | NEW |
---|---|
(Empty) | |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "base/logging.h" | |
6 #include "base/sys_byteorder.h" | |
7 #include "media/base/bit_reader.h" | |
8 #include "media/filters/ivf_parser.h" | |
9 | |
10 namespace media { | |
11 | |
12 void IvfFileHeader::ByteSwap() { | |
13 version = base::ByteSwapToLE16(version); | |
14 header_size = base::ByteSwapToLE16(header_size); | |
15 fourcc = base::ByteSwapToLE32(fourcc); | |
16 width = base::ByteSwapToLE16(width); | |
17 height = base::ByteSwapToLE16(height); | |
18 timebase_denum = base::ByteSwapToLE32(timebase_denum); | |
19 timebase_num = base::ByteSwapToLE32(timebase_num); | |
20 num_frames = base::ByteSwapToLE32(num_frames); | |
21 unused = base::ByteSwapToLE32(unused); | |
22 } | |
23 | |
24 void IvfFrameHeader::ByteSwap() { | |
25 frame_size = base::ByteSwapToLE32(frame_size); | |
26 timestamp = base::ByteSwapToLE64(timestamp); | |
27 } | |
28 | |
29 IvfParser::IvfParser() : ptr_(nullptr), end_(nullptr) {} | |
30 | |
31 bool IvfParser::Initialize(const uint8_t* stream, | |
32 size_t size, | |
33 IvfFileHeader* file_header) { | |
34 DCHECK(stream); | |
35 DCHECK(file_header); | |
36 ptr_ = stream; | |
37 end_ = stream + size; | |
38 | |
39 if (size < sizeof(IvfFileHeader)) { | |
40 DLOG(ERROR) << "EOF before file header"; | |
41 return false; | |
42 } | |
43 | |
44 memcpy(file_header, ptr_, sizeof(IvfFileHeader)); | |
45 file_header->ByteSwap(); | |
46 | |
47 if (memcmp(file_header->signature, "DKIF", 4) != 0) { | |
henryhsu
2015/08/05 10:45:14
s/"DKIF"/media::kIvfHeaderSignature/
kcwu
2015/08/05 11:05:15
Done.
| |
48 DLOG(ERROR) << "IVF signature mismatch"; | |
49 return false; | |
50 } | |
51 DLOG_IF(WARNING, file_header->version != 0) | |
52 << "IVF version unknown: " << file_header->version | |
53 << ", the parser may not be able to parse correctly"; | |
54 if (file_header->header_size != sizeof(IvfFileHeader)) { | |
55 DLOG(ERROR) << "IVF file header size mismatch"; | |
56 return false; | |
57 } | |
58 | |
59 ptr_ += sizeof(IvfFileHeader); | |
60 | |
61 return true; | |
62 } | |
63 | |
64 bool IvfParser::ParseNextFrame(IvfFrameHeader* frame_header, | |
65 const uint8_t** payload) { | |
66 DCHECK(ptr_); | |
67 DCHECK(payload); | |
68 | |
69 if (end_ < ptr_ + sizeof(IvfFrameHeader)) { | |
70 DLOG_IF(ERROR, ptr_ != end_) << "Incomplete frame header"; | |
71 return false; | |
72 } | |
73 | |
74 memcpy(frame_header, ptr_, sizeof(IvfFrameHeader)); | |
75 frame_header->ByteSwap(); | |
76 ptr_ += sizeof(IvfFrameHeader); | |
77 | |
78 if (end_ < ptr_ + frame_header->frame_size) { | |
79 DLOG(ERROR) << "Not enough frame data"; | |
80 return false; | |
81 } | |
82 | |
83 *payload = ptr_; | |
84 ptr_ += frame_header->frame_size; | |
85 | |
86 return true; | |
87 } | |
88 | |
89 } // namespace media | |
OLD | NEW |