OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "media/formats/webm/tracks_builder.h" | 5 #include "media/formats/webm/tracks_builder.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "media/formats/webm/webm_constants.h" | 8 #include "media/formats/webm/webm_constants.h" |
9 | 9 |
10 namespace media { | 10 namespace media { |
11 | 11 |
12 // Returns size of an integer, formatted using Matroska serialization. | 12 // Returns size of an integer, formatted using Matroska serialization. |
13 static int GetUIntMkvSize(uint64 value) { | 13 static int GetUIntMkvSize(uint64_t value) { |
14 if (value < 0x07FULL) | 14 if (value < 0x07FULL) |
15 return 1; | 15 return 1; |
16 if (value < 0x03FFFULL) | 16 if (value < 0x03FFFULL) |
17 return 2; | 17 return 2; |
18 if (value < 0x01FFFFFULL) | 18 if (value < 0x01FFFFFULL) |
19 return 3; | 19 return 3; |
20 if (value < 0x0FFFFFFFULL) | 20 if (value < 0x0FFFFFFFULL) |
21 return 4; | 21 return 4; |
22 if (value < 0x07FFFFFFFFULL) | 22 if (value < 0x07FFFFFFFFULL) |
23 return 5; | 23 return 5; |
24 if (value < 0x03FFFFFFFFFFULL) | 24 if (value < 0x03FFFFFFFFFFULL) |
25 return 6; | 25 return 6; |
26 if (value < 0x01FFFFFFFFFFFFULL) | 26 if (value < 0x01FFFFFFFFFFFFULL) |
27 return 7; | 27 return 7; |
28 return 8; | 28 return 8; |
29 } | 29 } |
30 | 30 |
31 // Returns the minimium size required to serialize an integer value. | 31 // Returns the minimium size required to serialize an integer value. |
32 static int GetUIntSize(uint64 value) { | 32 static int GetUIntSize(uint64_t value) { |
33 if (value < 0x0100ULL) | 33 if (value < 0x0100ULL) |
34 return 1; | 34 return 1; |
35 if (value < 0x010000ULL) | 35 if (value < 0x010000ULL) |
36 return 2; | 36 return 2; |
37 if (value < 0x01000000ULL) | 37 if (value < 0x01000000ULL) |
38 return 3; | 38 return 3; |
39 if (value < 0x0100000000ULL) | 39 if (value < 0x0100000000ULL) |
40 return 4; | 40 return 4; |
41 if (value < 0x010000000000ULL) | 41 if (value < 0x010000000000ULL) |
42 return 5; | 42 return 5; |
43 if (value < 0x01000000000000ULL) | 43 if (value < 0x01000000000000ULL) |
44 return 6; | 44 return 6; |
45 if (value < 0x0100000000000000ULL) | 45 if (value < 0x0100000000000000ULL) |
46 return 7; | 46 return 7; |
47 return 8; | 47 return 8; |
48 } | 48 } |
49 | 49 |
50 static int MasterElementSize(int element_id, int payload_size) { | 50 static int MasterElementSize(int element_id, int payload_size) { |
51 return GetUIntSize(element_id) + GetUIntMkvSize(payload_size) + payload_size; | 51 return GetUIntSize(element_id) + GetUIntMkvSize(payload_size) + payload_size; |
52 } | 52 } |
53 | 53 |
54 static int UIntElementSize(int element_id, uint64 value) { | 54 static int UIntElementSize(int element_id, uint64_t value) { |
55 return GetUIntSize(element_id) + 1 + GetUIntSize(value); | 55 return GetUIntSize(element_id) + 1 + GetUIntSize(value); |
56 } | 56 } |
57 | 57 |
58 static int DoubleElementSize(int element_id) { | 58 static int DoubleElementSize(int element_id) { |
59 return GetUIntSize(element_id) + 1 + 8; | 59 return GetUIntSize(element_id) + 1 + 8; |
60 } | 60 } |
61 | 61 |
62 static int StringElementSize(int element_id, const std::string& value) { | 62 static int StringElementSize(int element_id, const std::string& value) { |
63 return GetUIntSize(element_id) + | 63 return GetUIntSize(element_id) + |
64 GetUIntMkvSize(value.length()) + | 64 GetUIntMkvSize(value.length()) + |
65 value.length(); | 65 value.length(); |
66 } | 66 } |
67 | 67 |
68 static void SerializeInt(uint8** buf_ptr, int* buf_size_ptr, | 68 static void SerializeInt(uint8_t** buf_ptr, |
69 int64 value, int size) { | 69 int* buf_size_ptr, |
70 uint8*& buf = *buf_ptr; | 70 int64_t value, |
| 71 int size) { |
| 72 uint8_t*& buf = *buf_ptr; |
71 int& buf_size = *buf_size_ptr; | 73 int& buf_size = *buf_size_ptr; |
72 | 74 |
73 for (int idx = 1; idx <= size; ++idx) { | 75 for (int idx = 1; idx <= size; ++idx) { |
74 *buf++ = static_cast<uint8>(value >> ((size - idx) * 8)); | 76 *buf++ = static_cast<uint8_t>(value >> ((size - idx) * 8)); |
75 --buf_size; | 77 --buf_size; |
76 } | 78 } |
77 } | 79 } |
78 | 80 |
79 static void SerializeDouble(uint8** buf_ptr, int* buf_size_ptr, | 81 static void SerializeDouble(uint8_t** buf_ptr, |
| 82 int* buf_size_ptr, |
80 double value) { | 83 double value) { |
81 // Use a union to convert |value| to native endian integer bit pattern. | 84 // Use a union to convert |value| to native endian integer bit pattern. |
82 union { | 85 union { |
83 double src; | 86 double src; |
84 int64 dst; | 87 int64_t dst; |
85 } tmp; | 88 } tmp; |
86 tmp.src = value; | 89 tmp.src = value; |
87 | 90 |
88 // Write the bytes from native endian |tmp.dst| to big-endian form in |buf|. | 91 // Write the bytes from native endian |tmp.dst| to big-endian form in |buf|. |
89 SerializeInt(buf_ptr, buf_size_ptr, tmp.dst, 8); | 92 SerializeInt(buf_ptr, buf_size_ptr, tmp.dst, 8); |
90 } | 93 } |
91 | 94 |
92 static void WriteElementId(uint8** buf, int* buf_size, int element_id) { | 95 static void WriteElementId(uint8_t** buf, int* buf_size, int element_id) { |
93 SerializeInt(buf, buf_size, element_id, GetUIntSize(element_id)); | 96 SerializeInt(buf, buf_size, element_id, GetUIntSize(element_id)); |
94 } | 97 } |
95 | 98 |
96 static void WriteUInt(uint8** buf, int* buf_size, uint64 value) { | 99 static void WriteUInt(uint8_t** buf, int* buf_size, uint64_t value) { |
97 const int size = GetUIntMkvSize(value); | 100 const int size = GetUIntMkvSize(value); |
98 value |= (1ULL << (size * 7)); // Matroska formatting | 101 value |= (1ULL << (size * 7)); // Matroska formatting |
99 SerializeInt(buf, buf_size, value, size); | 102 SerializeInt(buf, buf_size, value, size); |
100 } | 103 } |
101 | 104 |
102 static void WriteMasterElement(uint8** buf, int* buf_size, | 105 static void WriteMasterElement(uint8_t** buf, |
103 int element_id, int payload_size) { | 106 int* buf_size, |
| 107 int element_id, |
| 108 int payload_size) { |
104 WriteElementId(buf, buf_size, element_id); | 109 WriteElementId(buf, buf_size, element_id); |
105 WriteUInt(buf, buf_size, payload_size); | 110 WriteUInt(buf, buf_size, payload_size); |
106 } | 111 } |
107 | 112 |
108 static void WriteUIntElement(uint8** buf, | 113 static void WriteUIntElement(uint8_t** buf, |
109 int* buf_size, | 114 int* buf_size, |
110 int element_id, | 115 int element_id, |
111 uint64 value) { | 116 uint64_t value) { |
112 WriteElementId(buf, buf_size, element_id); | 117 WriteElementId(buf, buf_size, element_id); |
113 | 118 |
114 const int size = GetUIntSize(value); | 119 const int size = GetUIntSize(value); |
115 WriteUInt(buf, buf_size, size); | 120 WriteUInt(buf, buf_size, size); |
116 | 121 |
117 SerializeInt(buf, buf_size, value, size); | 122 SerializeInt(buf, buf_size, value, size); |
118 } | 123 } |
119 | 124 |
120 static void WriteDoubleElement(uint8** buf, int* buf_size, | 125 static void WriteDoubleElement(uint8_t** buf, |
121 int element_id, double value) { | 126 int* buf_size, |
| 127 int element_id, |
| 128 double value) { |
122 WriteElementId(buf, buf_size, element_id); | 129 WriteElementId(buf, buf_size, element_id); |
123 WriteUInt(buf, buf_size, 8); | 130 WriteUInt(buf, buf_size, 8); |
124 SerializeDouble(buf, buf_size, value); | 131 SerializeDouble(buf, buf_size, value); |
125 } | 132 } |
126 | 133 |
127 static void WriteStringElement(uint8** buf_ptr, int* buf_size_ptr, | 134 static void WriteStringElement(uint8_t** buf_ptr, |
128 int element_id, const std::string& value) { | 135 int* buf_size_ptr, |
129 uint8*& buf = *buf_ptr; | 136 int element_id, |
| 137 const std::string& value) { |
| 138 uint8_t*& buf = *buf_ptr; |
130 int& buf_size = *buf_size_ptr; | 139 int& buf_size = *buf_size_ptr; |
131 | 140 |
132 WriteElementId(&buf, &buf_size, element_id); | 141 WriteElementId(&buf, &buf_size, element_id); |
133 | 142 |
134 const uint64 size = value.length(); | 143 const uint64_t size = value.length(); |
135 WriteUInt(&buf, &buf_size, size); | 144 WriteUInt(&buf, &buf_size, size); |
136 | 145 |
137 memcpy(buf, value.data(), size); | 146 memcpy(buf, value.data(), size); |
138 buf += size; | 147 buf += size; |
139 buf_size -= size; | 148 buf_size -= size; |
140 } | 149 } |
141 | 150 |
142 TracksBuilder::TracksBuilder(bool allow_invalid_values) | 151 TracksBuilder::TracksBuilder(bool allow_invalid_values) |
143 : allow_invalid_values_(allow_invalid_values) {} | 152 : allow_invalid_values_(allow_invalid_values) {} |
144 TracksBuilder::TracksBuilder() | 153 TracksBuilder::TracksBuilder() |
145 : allow_invalid_values_(false) {} | 154 : allow_invalid_values_(false) {} |
146 TracksBuilder::~TracksBuilder() {} | 155 TracksBuilder::~TracksBuilder() {} |
147 | 156 |
148 void TracksBuilder::AddVideoTrack(int track_num, | 157 void TracksBuilder::AddVideoTrack(int track_num, |
149 uint64 track_uid, | 158 uint64_t track_uid, |
150 const std::string& codec_id, | 159 const std::string& codec_id, |
151 const std::string& name, | 160 const std::string& name, |
152 const std::string& language, | 161 const std::string& language, |
153 int default_duration, | 162 int default_duration, |
154 int video_pixel_width, | 163 int video_pixel_width, |
155 int video_pixel_height) { | 164 int video_pixel_height) { |
156 AddTrackInternal(track_num, kWebMTrackTypeVideo, track_uid, codec_id, name, | 165 AddTrackInternal(track_num, kWebMTrackTypeVideo, track_uid, codec_id, name, |
157 language, default_duration, video_pixel_width, | 166 language, default_duration, video_pixel_width, |
158 video_pixel_height, -1, -1); | 167 video_pixel_height, -1, -1); |
159 } | 168 } |
160 | 169 |
161 void TracksBuilder::AddAudioTrack(int track_num, | 170 void TracksBuilder::AddAudioTrack(int track_num, |
162 uint64 track_uid, | 171 uint64_t track_uid, |
163 const std::string& codec_id, | 172 const std::string& codec_id, |
164 const std::string& name, | 173 const std::string& name, |
165 const std::string& language, | 174 const std::string& language, |
166 int default_duration, | 175 int default_duration, |
167 int audio_channels, | 176 int audio_channels, |
168 double audio_sampling_frequency) { | 177 double audio_sampling_frequency) { |
169 AddTrackInternal(track_num, kWebMTrackTypeAudio, track_uid, codec_id, name, | 178 AddTrackInternal(track_num, kWebMTrackTypeAudio, track_uid, codec_id, name, |
170 language, default_duration, -1, -1, audio_channels, | 179 language, default_duration, -1, -1, audio_channels, |
171 audio_sampling_frequency); | 180 audio_sampling_frequency); |
172 } | 181 } |
173 | 182 |
174 void TracksBuilder::AddTextTrack(int track_num, | 183 void TracksBuilder::AddTextTrack(int track_num, |
175 uint64 track_uid, | 184 uint64_t track_uid, |
176 const std::string& codec_id, | 185 const std::string& codec_id, |
177 const std::string& name, | 186 const std::string& name, |
178 const std::string& language) { | 187 const std::string& language) { |
179 AddTrackInternal(track_num, kWebMTrackTypeSubtitlesOrCaptions, track_uid, | 188 AddTrackInternal(track_num, kWebMTrackTypeSubtitlesOrCaptions, track_uid, |
180 codec_id, name, language, -1, -1, -1, -1, -1); | 189 codec_id, name, language, -1, -1, -1, -1, -1); |
181 } | 190 } |
182 | 191 |
183 std::vector<uint8> TracksBuilder::Finish() { | 192 std::vector<uint8_t> TracksBuilder::Finish() { |
184 // Allocate the storage | 193 // Allocate the storage |
185 std::vector<uint8> buffer; | 194 std::vector<uint8_t> buffer; |
186 buffer.resize(GetTracksSize()); | 195 buffer.resize(GetTracksSize()); |
187 | 196 |
188 // Populate the storage with a tracks header | 197 // Populate the storage with a tracks header |
189 WriteTracks(&buffer[0], buffer.size()); | 198 WriteTracks(&buffer[0], buffer.size()); |
190 | 199 |
191 return buffer; | 200 return buffer; |
192 } | 201 } |
193 | 202 |
194 void TracksBuilder::AddTrackInternal(int track_num, | 203 void TracksBuilder::AddTrackInternal(int track_num, |
195 int track_type, | 204 int track_type, |
196 uint64 track_uid, | 205 uint64_t track_uid, |
197 const std::string& codec_id, | 206 const std::string& codec_id, |
198 const std::string& name, | 207 const std::string& name, |
199 const std::string& language, | 208 const std::string& language, |
200 int default_duration, | 209 int default_duration, |
201 int video_pixel_width, | 210 int video_pixel_width, |
202 int video_pixel_height, | 211 int video_pixel_height, |
203 int audio_channels, | 212 int audio_channels, |
204 double audio_sampling_frequency) { | 213 double audio_sampling_frequency) { |
205 tracks_.push_back(Track(track_num, track_type, track_uid, codec_id, name, | 214 tracks_.push_back(Track(track_num, track_type, track_uid, codec_id, name, |
206 language, default_duration, video_pixel_width, | 215 language, default_duration, video_pixel_width, |
207 video_pixel_height, audio_channels, | 216 video_pixel_height, audio_channels, |
208 audio_sampling_frequency, allow_invalid_values_)); | 217 audio_sampling_frequency, allow_invalid_values_)); |
209 } | 218 } |
210 | 219 |
211 int TracksBuilder::GetTracksSize() const { | 220 int TracksBuilder::GetTracksSize() const { |
212 return MasterElementSize(kWebMIdTracks, GetTracksPayloadSize()); | 221 return MasterElementSize(kWebMIdTracks, GetTracksPayloadSize()); |
213 } | 222 } |
214 | 223 |
215 int TracksBuilder::GetTracksPayloadSize() const { | 224 int TracksBuilder::GetTracksPayloadSize() const { |
216 int payload_size = 0; | 225 int payload_size = 0; |
217 | 226 |
218 for (TrackList::const_iterator itr = tracks_.begin(); | 227 for (TrackList::const_iterator itr = tracks_.begin(); |
219 itr != tracks_.end(); ++itr) { | 228 itr != tracks_.end(); ++itr) { |
220 payload_size += itr->GetSize(); | 229 payload_size += itr->GetSize(); |
221 } | 230 } |
222 | 231 |
223 return payload_size; | 232 return payload_size; |
224 } | 233 } |
225 | 234 |
226 void TracksBuilder::WriteTracks(uint8* buf, int buf_size) const { | 235 void TracksBuilder::WriteTracks(uint8_t* buf, int buf_size) const { |
227 WriteMasterElement(&buf, &buf_size, kWebMIdTracks, GetTracksPayloadSize()); | 236 WriteMasterElement(&buf, &buf_size, kWebMIdTracks, GetTracksPayloadSize()); |
228 | 237 |
229 for (TrackList::const_iterator itr = tracks_.begin(); | 238 for (TrackList::const_iterator itr = tracks_.begin(); |
230 itr != tracks_.end(); ++itr) { | 239 itr != tracks_.end(); ++itr) { |
231 itr->Write(&buf, &buf_size); | 240 itr->Write(&buf, &buf_size); |
232 } | 241 } |
233 } | 242 } |
234 | 243 |
235 TracksBuilder::Track::Track(int track_num, | 244 TracksBuilder::Track::Track(int track_num, |
236 int track_type, | 245 int track_type, |
237 uint64 track_uid, | 246 uint64_t track_uid, |
238 const std::string& codec_id, | 247 const std::string& codec_id, |
239 const std::string& name, | 248 const std::string& name, |
240 const std::string& language, | 249 const std::string& language, |
241 int default_duration, | 250 int default_duration, |
242 int video_pixel_width, | 251 int video_pixel_width, |
243 int video_pixel_height, | 252 int video_pixel_height, |
244 int audio_channels, | 253 int audio_channels, |
245 double audio_sampling_frequency, | 254 double audio_sampling_frequency, |
246 bool allow_invalid_values) | 255 bool allow_invalid_values) |
247 : track_num_(track_num), | 256 : track_num_(track_num), |
(...skipping 86 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
334 size += MasterElementSize(kWebMIdVideo, GetVideoPayloadSize()); | 343 size += MasterElementSize(kWebMIdVideo, GetVideoPayloadSize()); |
335 } | 344 } |
336 | 345 |
337 if (GetAudioPayloadSize() > 0) { | 346 if (GetAudioPayloadSize() > 0) { |
338 size += MasterElementSize(kWebMIdAudio, GetAudioPayloadSize()); | 347 size += MasterElementSize(kWebMIdAudio, GetAudioPayloadSize()); |
339 } | 348 } |
340 | 349 |
341 return size; | 350 return size; |
342 } | 351 } |
343 | 352 |
344 void TracksBuilder::Track::Write(uint8** buf, int* buf_size) const { | 353 void TracksBuilder::Track::Write(uint8_t** buf, int* buf_size) const { |
345 WriteMasterElement(buf, buf_size, kWebMIdTrackEntry, GetPayloadSize()); | 354 WriteMasterElement(buf, buf_size, kWebMIdTrackEntry, GetPayloadSize()); |
346 | 355 |
347 WriteUIntElement(buf, buf_size, kWebMIdTrackNumber, track_num_); | 356 WriteUIntElement(buf, buf_size, kWebMIdTrackNumber, track_num_); |
348 WriteUIntElement(buf, buf_size, kWebMIdTrackType, track_type_); | 357 WriteUIntElement(buf, buf_size, kWebMIdTrackType, track_type_); |
349 WriteUIntElement(buf, buf_size, kWebMIdTrackUID, track_uid_); | 358 WriteUIntElement(buf, buf_size, kWebMIdTrackUID, track_uid_); |
350 | 359 |
351 if (default_duration_ >= 0) | 360 if (default_duration_ >= 0) |
352 WriteUIntElement(buf, buf_size, kWebMIdDefaultDuration, default_duration_); | 361 WriteUIntElement(buf, buf_size, kWebMIdDefaultDuration, default_duration_); |
353 | 362 |
354 if (!codec_id_.empty()) | 363 if (!codec_id_.empty()) |
(...skipping 22 matching lines...) Expand all Loading... |
377 WriteUIntElement(buf, buf_size, kWebMIdChannels, audio_channels_); | 386 WriteUIntElement(buf, buf_size, kWebMIdChannels, audio_channels_); |
378 | 387 |
379 if (audio_sampling_frequency_ >= 0) { | 388 if (audio_sampling_frequency_ >= 0) { |
380 WriteDoubleElement(buf, buf_size, kWebMIdSamplingFrequency, | 389 WriteDoubleElement(buf, buf_size, kWebMIdSamplingFrequency, |
381 audio_sampling_frequency_); | 390 audio_sampling_frequency_); |
382 } | 391 } |
383 } | 392 } |
384 } | 393 } |
385 | 394 |
386 } // namespace media | 395 } // namespace media |
OLD | NEW |