61
|
1 /*
|
|
2 * The contents of this file are subject to the Mozilla Public
|
|
3 * License Version 1.1 (the "License"); you may not use this file
|
|
4 * except in compliance with the License. You may obtain a copy of
|
|
5 * the License at http://www.mozilla.org/MPL/
|
|
6 *
|
|
7 * Software distributed under the License is distributed on an "AS
|
|
8 * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express or
|
|
9 * implied. See the License for the specific language governing
|
|
10 * rights and limitations under the License.
|
|
11 *
|
|
12 * The Original Code is MPEG4IP.
|
|
13 *
|
|
14 * The Initial Developer of the Original Code is Cisco Systems Inc.
|
|
15 * Portions created by Cisco Systems Inc. are
|
|
16 * Copyright (C) Cisco Systems Inc. 2001. All Rights Reserved.
|
|
17 *
|
|
18 * Contributor(s):
|
|
19 * Dave Mackie dmackie@cisco.com
|
|
20 */
|
|
21
|
|
22 #include "mp4common.h"
|
|
23
|
|
24 /* rtp hint track operations */
|
|
25
|
|
26 MP4RtpHintTrack::MP4RtpHintTrack(MP4File* pFile, MP4Atom* pTrakAtom)
|
|
27 : MP4Track(pFile, pTrakAtom)
|
|
28 {
|
|
29 m_pRefTrack = NULL;
|
|
30
|
|
31 m_pRtpMapProperty = NULL;
|
|
32 m_pPayloadNumberProperty = NULL;
|
|
33 m_pMaxPacketSizeProperty = NULL;
|
|
34 m_pSnroProperty = NULL;
|
|
35 m_pTsroProperty = NULL;
|
|
36
|
|
37 m_pReadHint = NULL;
|
|
38 m_pReadHintSample = NULL;
|
|
39 m_readHintSampleSize = 0;
|
|
40
|
|
41 m_pWriteHint = NULL;
|
|
42 m_writeHintId = MP4_INVALID_SAMPLE_ID;
|
|
43 m_writePacketId = 0;
|
|
44
|
|
45 m_pTrpy = NULL;
|
|
46 m_pNump = NULL;
|
|
47 m_pTpyl = NULL;
|
|
48 m_pMaxr = NULL;
|
|
49 m_pDmed = NULL;
|
|
50 m_pDimm = NULL;
|
|
51 m_pPmax = NULL;
|
|
52 m_pDmax = NULL;
|
|
53
|
|
54 m_pMaxPdu = NULL;
|
|
55 m_pAvgPdu = NULL;
|
|
56 m_pMaxBitRate = NULL;
|
|
57 m_pAvgBitRate = NULL;
|
|
58
|
|
59 m_thisSec = 0;
|
|
60 m_bytesThisSec = 0;
|
|
61 m_bytesThisHint = 0;
|
|
62 m_bytesThisPacket = 0;
|
|
63 }
|
|
64
|
|
65 MP4RtpHintTrack::~MP4RtpHintTrack()
|
|
66 {
|
|
67 delete m_pReadHint;
|
|
68 delete m_pReadHintSample;
|
|
69 delete m_pWriteHint;
|
|
70 }
|
|
71
|
|
72 void MP4RtpHintTrack::InitRefTrack()
|
|
73 {
|
|
74 if (m_pRefTrack == NULL) {
|
|
75 MP4Integer32Property* pRefTrackIdProperty = NULL;
|
|
76 m_pTrakAtom->FindProperty(
|
|
77 "trak.tref.hint.entries[0].trackId",
|
|
78 (MP4Property**)&pRefTrackIdProperty);
|
|
79 ASSERT(pRefTrackIdProperty);
|
|
80
|
|
81 m_pRefTrack = m_pFile->GetTrack(pRefTrackIdProperty->GetValue());
|
|
82 }
|
|
83 }
|
|
84
|
|
85 void MP4RtpHintTrack::InitRtpStart()
|
|
86 {
|
|
87 struct timeval tv;
|
|
88 gettimeofday(&tv, NULL);
|
|
89 srandom((tv.tv_usec << 12) | (tv.tv_sec & 0xFFF));
|
|
90
|
|
91 ASSERT(m_pTrakAtom);
|
|
92
|
|
93 m_pTrakAtom->FindProperty(
|
|
94 "trak.udta.hnti.rtp .snro.offset",
|
|
95 (MP4Property**)&m_pSnroProperty);
|
|
96
|
|
97 if (m_pSnroProperty) {
|
|
98 m_rtpSequenceStart = m_pSnroProperty->GetValue();
|
|
99 } else {
|
|
100 m_rtpSequenceStart = random();
|
|
101 }
|
|
102
|
|
103 m_pTrakAtom->FindProperty(
|
|
104 "trak.udta.hnti.rtp .tsro.offset",
|
|
105 (MP4Property**)&m_pTsroProperty);
|
|
106
|
|
107 if (m_pTsroProperty) {
|
|
108 m_rtpTimestampStart = m_pTsroProperty->GetValue();
|
|
109 } else {
|
|
110 m_rtpTimestampStart = random();
|
|
111 }
|
|
112 }
|
|
113
|
|
114 void MP4RtpHintTrack::ReadHint(
|
|
115 MP4SampleId hintSampleId,
|
|
116 u_int16_t* pNumPackets)
|
|
117 {
|
|
118 if (m_pRefTrack == NULL) {
|
|
119 InitRefTrack();
|
|
120 InitRtpStart();
|
|
121 }
|
|
122
|
|
123 // dispose of any old hint
|
|
124 delete m_pReadHint;
|
|
125 m_pReadHint = NULL;
|
|
126 delete m_pReadHintSample;
|
|
127 m_pReadHintSample = NULL;
|
|
128 m_readHintSampleSize = 0;
|
|
129
|
|
130 // read the desired hint sample into memory
|
|
131 ReadSample(
|
|
132 hintSampleId,
|
|
133 &m_pReadHintSample,
|
|
134 &m_readHintSampleSize,
|
|
135 &m_readHintTimestamp);
|
|
136
|
|
137 m_pFile->EnableMemoryBuffer(m_pReadHintSample, m_readHintSampleSize);
|
|
138
|
|
139 m_pReadHint = new MP4RtpHint(this);
|
|
140 m_pReadHint->Read(m_pFile);
|
|
141
|
|
142 m_pFile->DisableMemoryBuffer();
|
|
143
|
|
144 if (pNumPackets) {
|
|
145 *pNumPackets = GetHintNumberOfPackets();
|
|
146 }
|
|
147 }
|
|
148
|
|
149 u_int16_t MP4RtpHintTrack::GetHintNumberOfPackets()
|
|
150 {
|
|
151 if (m_pReadHint == NULL) {
|
|
152 throw new MP4Error("no hint has been read",
|
|
153 "MP4GetRtpHintNumberOfPackets");
|
|
154 }
|
|
155 return m_pReadHint->GetNumberOfPackets();
|
|
156 }
|
|
157
|
|
158 bool MP4RtpHintTrack::GetPacketBFrame(u_int16_t packetIndex)
|
|
159 {
|
|
160 if (m_pReadHint == NULL) {
|
|
161 throw new MP4Error("no hint has been read",
|
|
162 "MP4GetRtpPacketBFrame");
|
|
163 }
|
|
164 MP4RtpPacket* pPacket =
|
|
165 m_pReadHint->GetPacket(packetIndex);
|
|
166
|
|
167 return pPacket->IsBFrame();
|
|
168 }
|
|
169
|
|
170 u_int16_t MP4RtpHintTrack::GetPacketTransmitOffset(u_int16_t packetIndex)
|
|
171 {
|
|
172 if (m_pReadHint == NULL) {
|
|
173 throw new MP4Error("no hint has been read",
|
|
174 "MP4GetRtpPacketTransmitOffset");
|
|
175 }
|
|
176
|
|
177 MP4RtpPacket* pPacket =
|
|
178 m_pReadHint->GetPacket(packetIndex);
|
|
179
|
|
180 return pPacket->GetTransmitOffset();
|
|
181 }
|
|
182
|
|
183 void MP4RtpHintTrack::ReadPacket(
|
|
184 u_int16_t packetIndex,
|
|
185 u_int8_t** ppBytes,
|
|
186 u_int32_t* pNumBytes,
|
|
187 u_int32_t ssrc,
|
|
188 bool addHeader,
|
|
189 bool addPayload)
|
|
190 {
|
|
191 if (m_pReadHint == NULL) {
|
|
192 throw new MP4Error("no hint has been read",
|
|
193 "MP4ReadRtpPacket");
|
|
194 }
|
|
195 if (!addHeader && !addPayload) {
|
|
196 throw new MP4Error("no data requested",
|
|
197 "MP4ReadRtpPacket");
|
|
198 }
|
|
199
|
|
200 MP4RtpPacket* pPacket =
|
|
201 m_pReadHint->GetPacket(packetIndex);
|
|
202
|
|
203 *pNumBytes = 0;
|
|
204 if (addHeader) {
|
|
205 *pNumBytes += 12;
|
|
206 }
|
|
207 if (addPayload) {
|
|
208 *pNumBytes += pPacket->GetDataSize();
|
|
209 }
|
|
210
|
|
211 // if needed, allocate the packet memory
|
|
212 bool buffer_malloc = false;
|
|
213
|
|
214 if (*ppBytes == NULL) {
|
|
215 *ppBytes = (u_int8_t*)MP4Malloc(*pNumBytes);
|
|
216 buffer_malloc = true;
|
|
217 }
|
|
218
|
|
219 try {
|
|
220 u_int8_t* pDest = *ppBytes;
|
|
221
|
|
222 if (addHeader) {
|
|
223 *pDest++ =
|
|
224 0x80 | (pPacket->GetPBit() << 5) | (pPacket->GetXBit() << 4);
|
|
225
|
|
226 *pDest++ =
|
|
227 (pPacket->GetMBit() << 7) | pPacket->GetPayload();
|
|
228
|
|
229 *((u_int16_t*)pDest) =
|
|
230 htons(m_rtpSequenceStart + pPacket->GetSequenceNumber());
|
|
231 pDest += 2;
|
|
232
|
|
233 *((u_int32_t*)pDest) =
|
|
234 htonl(m_rtpTimestampStart + (u_int32_t)m_readHintTimestamp);
|
|
235 pDest += 4;
|
|
236
|
|
237 *((u_int32_t*)pDest) =
|
|
238 htonl(ssrc);
|
|
239 pDest += 4;
|
|
240 }
|
|
241
|
|
242 if (addPayload) {
|
|
243 pPacket->GetData(pDest);
|
|
244 }
|
|
245 }
|
|
246 catch (MP4Error* e) {
|
|
247 if (buffer_malloc) {
|
|
248 MP4Free(*ppBytes);
|
|
249 *ppBytes = NULL;
|
|
250 }
|
|
251 throw e;
|
|
252 }
|
|
253
|
|
254 VERBOSE_READ_HINT(m_pFile->GetVerbosity(),
|
|
255 printf("ReadPacket: %u ", packetIndex);
|
|
256 MP4HexDump(*ppBytes, *pNumBytes););
|
|
257 }
|
|
258
|
|
259 MP4Timestamp MP4RtpHintTrack::GetRtpTimestampStart()
|
|
260 {
|
|
261 if (m_pRefTrack == NULL) {
|
|
262 InitRefTrack();
|
|
263 InitRtpStart();
|
|
264 }
|
|
265
|
|
266 return m_rtpTimestampStart;
|
|
267 }
|
|
268
|
|
269 void MP4RtpHintTrack::SetRtpTimestampStart(MP4Timestamp start)
|
|
270 {
|
|
271 if (!m_pTsroProperty) {
|
|
272 MP4Atom* pTsroAtom =
|
|
273 m_pFile->AddDescendantAtoms(m_pTrakAtom, "udta.hnti.rtp .tsro");
|
|
274
|
|
275 ASSERT(pTsroAtom);
|
|
276
|
|
277 pTsroAtom->FindProperty("offset",
|
|
278 (MP4Property**)&m_pTsroProperty);
|
|
279
|
|
280 ASSERT(m_pTsroProperty);
|
|
281 }
|
|
282
|
|
283 m_pTsroProperty->SetValue(start);
|
|
284 m_rtpTimestampStart = start;
|
|
285 }
|
|
286
|
|
287 void MP4RtpHintTrack::InitPayload()
|
|
288 {
|
|
289 ASSERT(m_pTrakAtom);
|
|
290
|
|
291 if (m_pRtpMapProperty == NULL) {
|
|
292 m_pTrakAtom->FindProperty(
|
|
293 "trak.udta.hinf.payt.rtpMap",
|
|
294 (MP4Property**)&m_pRtpMapProperty);
|
|
295 }
|
|
296
|
|
297 if (m_pPayloadNumberProperty == NULL) {
|
|
298 m_pTrakAtom->FindProperty(
|
|
299 "trak.udta.hinf.payt.payloadNumber",
|
|
300 (MP4Property**)&m_pPayloadNumberProperty);
|
|
301 }
|
|
302
|
|
303 if (m_pMaxPacketSizeProperty == NULL) {
|
|
304 m_pTrakAtom->FindProperty(
|
|
305 "trak.mdia.minf.stbl.stsd.rtp .maxPacketSize",
|
|
306 (MP4Property**)&m_pMaxPacketSizeProperty);
|
|
307 }
|
|
308 }
|
|
309
|
|
310 void MP4RtpHintTrack::GetPayload(
|
|
311 char** ppPayloadName,
|
|
312 u_int8_t* pPayloadNumber,
|
|
313 u_int16_t* pMaxPayloadSize,
|
|
314 char **ppEncodingParams)
|
|
315 {
|
|
316 const char* pRtpMap;
|
|
317 char* pSlash;
|
|
318 u_int32_t length;
|
|
319 InitPayload();
|
|
320
|
|
321 if (ppPayloadName || ppEncodingParams) {
|
|
322 if (ppPayloadName)
|
|
323 *ppPayloadName = NULL;
|
|
324 if (ppEncodingParams)
|
|
325 *ppEncodingParams = NULL;
|
|
326 if (m_pRtpMapProperty) {
|
|
327 pRtpMap = m_pRtpMapProperty->GetValue();
|
|
328 pSlash = strchr(pRtpMap, '/');
|
|
329
|
|
330 if (pSlash) {
|
|
331 length = pSlash - pRtpMap;
|
|
332 } else {
|
|
333 length = strlen(pRtpMap);
|
|
334 }
|
|
335
|
|
336 if (ppPayloadName) {
|
|
337 *ppPayloadName = (char*)MP4Calloc(length + 1);
|
|
338 strncpy(*ppPayloadName, pRtpMap, length);
|
|
339 }
|
|
340 if (pSlash && ppEncodingParams) {
|
|
341 pSlash++;
|
|
342 pSlash = strchr(pSlash, '/');
|
|
343 if (pSlash != NULL) {
|
|
344 pSlash++;
|
|
345 if (pSlash != '\0') {
|
|
346 length = strlen(pRtpMap) - (pSlash - pRtpMap);
|
|
347 *ppEncodingParams = (char *)MP4Calloc(length + 1);
|
|
348 strncpy(*ppEncodingParams, pSlash, length);
|
|
349 }
|
|
350 }
|
|
351 }
|
|
352 }
|
|
353 }
|
|
354
|
|
355 if (pPayloadNumber) {
|
|
356 if (m_pPayloadNumberProperty) {
|
|
357 *pPayloadNumber = m_pPayloadNumberProperty->GetValue();
|
|
358 } else {
|
|
359 *pPayloadNumber = 0;
|
|
360 }
|
|
361 }
|
|
362
|
|
363 if (pMaxPayloadSize) {
|
|
364 if (m_pMaxPacketSizeProperty) {
|
|
365 *pMaxPayloadSize = m_pMaxPacketSizeProperty->GetValue();
|
|
366 } else {
|
|
367 *pMaxPayloadSize = 0;
|
|
368 }
|
|
369 }
|
|
370 }
|
|
371
|
|
372 void MP4RtpHintTrack::SetPayload(
|
|
373 const char* payloadName,
|
|
374 u_int8_t payloadNumber,
|
|
375 u_int16_t maxPayloadSize,
|
|
376 const char *encoding_parms,
|
|
377 bool include_rtp_map,
|
|
378 bool include_mpeg4_esid)
|
|
379 {
|
|
380 InitRefTrack();
|
|
381 InitPayload();
|
|
382
|
|
383 ASSERT(m_pRtpMapProperty);
|
|
384 ASSERT(m_pPayloadNumberProperty);
|
|
385 ASSERT(m_pMaxPacketSizeProperty);
|
|
386
|
|
387 size_t len = strlen(payloadName) + 16;
|
|
388 if (encoding_parms != NULL) {
|
|
389 size_t temp = strlen(encoding_parms);
|
|
390 if (temp == 0) {
|
|
391 encoding_parms = NULL;
|
|
392 } else {
|
|
393 len += temp;
|
|
394 }
|
|
395 }
|
|
396
|
|
397 char* rtpMapBuf = (char*)MP4Malloc(len);
|
|
398 sprintf(rtpMapBuf, "%s/%u%c%s",
|
|
399 payloadName,
|
|
400 GetTimeScale(),
|
|
401 encoding_parms != NULL ? '/' : '\0',
|
|
402 encoding_parms == NULL ? "" : encoding_parms);
|
|
403 m_pRtpMapProperty->SetValue(rtpMapBuf);
|
|
404
|
|
405 m_pPayloadNumberProperty->SetValue(payloadNumber);
|
|
406
|
|
407 if (maxPayloadSize == 0) {
|
|
408 maxPayloadSize = 1460;
|
|
409 }
|
|
410 m_pMaxPacketSizeProperty->SetValue(maxPayloadSize);
|
|
411
|
|
412 // set sdp media type
|
|
413 const char* sdpMediaType;
|
|
414 if (!strcmp(m_pRefTrack->GetType(), MP4_AUDIO_TRACK_TYPE)) {
|
|
415 sdpMediaType = "audio";
|
|
416 } else if (!strcmp(m_pRefTrack->GetType(), MP4_VIDEO_TRACK_TYPE)) {
|
|
417 sdpMediaType = "video";
|
201
|
418 } else if (!strcmp(m_pRefTrack->GetType(), MP4_CNTL_TRACK_TYPE)) {
|
|
419 sdpMediaType = "control";
|
61
|
420 } else {
|
|
421 sdpMediaType = "application";
|
|
422 }
|
|
423
|
|
424 char* sdpBuf = (char*)MP4Malloc(
|
|
425 strlen(sdpMediaType) + strlen(rtpMapBuf) + 256);
|
|
426 uint32_t buflen;
|
|
427 buflen = sprintf(sdpBuf,
|
|
428 "m=%s 0 RTP/AVP %u\015\012"
|
|
429 "a=control:trackID=%u\015\012",
|
|
430 sdpMediaType, payloadNumber,
|
|
431 m_trackId);
|
|
432 if (include_rtp_map) {
|
|
433 buflen += sprintf(sdpBuf + buflen,
|
|
434 "a=rtpmap:%u %s\015\012",
|
|
435 payloadNumber, rtpMapBuf);
|
|
436 }
|
|
437 if (include_mpeg4_esid) {
|
|
438 sprintf(sdpBuf + buflen,
|
|
439 "a=mpeg4-esid:%u\015\012",
|
|
440 m_pRefTrack->GetId());
|
|
441 }
|
|
442
|
|
443 MP4StringProperty* pSdpProperty = NULL;
|
|
444 m_pTrakAtom->FindProperty("trak.udta.hnti.sdp .sdpText",
|
|
445 (MP4Property**)&pSdpProperty);
|
|
446 ASSERT(pSdpProperty);
|
|
447 pSdpProperty->SetValue(sdpBuf);
|
|
448
|
|
449 // cleanup
|
|
450 MP4Free(rtpMapBuf);
|
|
451 MP4Free(sdpBuf);
|
|
452 }
|
|
453
|
|
454 void MP4RtpHintTrack::AddHint(bool isBFrame, u_int32_t timestampOffset)
|
|
455 {
|
|
456 // on first hint, need to lookup the reference track
|
|
457 if (m_writeHintId == MP4_INVALID_SAMPLE_ID) {
|
|
458 InitRefTrack();
|
|
459 InitStats();
|
|
460 }
|
|
461
|
|
462 if (m_pWriteHint) {
|
|
463 throw new MP4Error("unwritten hint is still pending", "MP4AddRtpHint");
|
|
464 }
|
|
465
|
|
466 m_pWriteHint = new MP4RtpHint(this);
|
|
467 m_pWriteHint->SetBFrame(isBFrame);
|
|
468 m_pWriteHint->SetTimestampOffset(timestampOffset);
|
|
469
|
|
470 m_bytesThisHint = 0;
|
|
471 m_writeHintId++;
|
|
472 }
|
|
473
|
|
474 void MP4RtpHintTrack::AddPacket(bool setMbit, int32_t transmitOffset)
|
|
475 {
|
|
476 if (m_pWriteHint == NULL) {
|
|
477 throw new MP4Error("no hint pending", "MP4RtpAddPacket");
|
|
478 }
|
|
479
|
|
480 MP4RtpPacket* pPacket = m_pWriteHint->AddPacket();
|
|
481
|
|
482 ASSERT(m_pPayloadNumberProperty);
|
|
483
|
|
484 pPacket->Set(
|
|
485 m_pPayloadNumberProperty->GetValue(),
|
|
486 m_writePacketId++,
|
|
487 setMbit);
|
|
488 pPacket->SetTransmitOffset(transmitOffset);
|
|
489
|
|
490 m_bytesThisHint += 12;
|
|
491 if (m_bytesThisPacket > m_pPmax->GetValue()) {
|
|
492 m_pPmax->SetValue(m_bytesThisPacket);
|
|
493 }
|
|
494 m_bytesThisPacket = 12;
|
|
495 m_pNump->IncrementValue();
|
|
496 m_pTrpy->IncrementValue(12); // RTP packet header size
|
|
497 }
|
|
498
|
|
499 void MP4RtpHintTrack::AddImmediateData(
|
|
500 const u_int8_t* pBytes,
|
|
501 u_int32_t numBytes)
|
|
502 {
|
|
503 if (m_pWriteHint == NULL) {
|
|
504 throw new MP4Error("no hint pending", "MP4RtpAddImmediateData");
|
|
505 }
|
|
506
|
|
507 MP4RtpPacket* pPacket = m_pWriteHint->GetCurrentPacket();
|
|
508 if (pPacket == NULL) {
|
|
509 throw new MP4Error("no packet pending", "MP4RtpAddImmediateData");
|
|
510 }
|
|
511
|
|
512 if (pBytes == NULL || numBytes == 0) {
|
|
513 throw new MP4Error("no data",
|
|
514 "AddImmediateData");
|
|
515 }
|
|
516 if (numBytes > 14) {
|
|
517 throw new MP4Error("data size is larger than 14 bytes",
|
|
518 "AddImmediateData");
|
|
519 }
|
|
520
|
|
521 MP4RtpImmediateData* pData = new MP4RtpImmediateData(pPacket);
|
|
522 pData->Set(pBytes, numBytes);
|
|
523
|
|
524 pPacket->AddData(pData);
|
|
525
|
|
526 m_bytesThisHint += numBytes;
|
|
527 m_bytesThisPacket += numBytes;
|
|
528 m_pDimm->IncrementValue(numBytes);
|
|
529 m_pTpyl->IncrementValue(numBytes);
|
|
530 m_pTrpy->IncrementValue(numBytes);
|
|
531 }
|
|
532
|
|
533 void MP4RtpHintTrack::AddSampleData(
|
|
534 MP4SampleId sampleId,
|
|
535 u_int32_t dataOffset,
|
|
536 u_int32_t dataLength)
|
|
537 {
|
|
538 if (m_pWriteHint == NULL) {
|
|
539 throw new MP4Error("no hint pending", "MP4RtpAddSampleData");
|
|
540 }
|
|
541
|
|
542 MP4RtpPacket* pPacket = m_pWriteHint->GetCurrentPacket();
|
|
543 if (pPacket == NULL) {
|
|
544 throw new MP4Error("no packet pending", "MP4RtpAddSampleData");
|
|
545 }
|
|
546
|
|
547 MP4RtpSampleData* pData = new MP4RtpSampleData(pPacket);
|
|
548
|
|
549 pData->SetReferenceSample(sampleId, dataOffset, dataLength);
|
|
550
|
|
551 pPacket->AddData(pData);
|
|
552
|
|
553 m_bytesThisHint += dataLength;
|
|
554 m_bytesThisPacket += dataLength;
|
|
555 m_pDmed->IncrementValue(dataLength);
|
|
556 m_pTpyl->IncrementValue(dataLength);
|
|
557 m_pTrpy->IncrementValue(dataLength);
|
|
558 }
|
|
559
|
|
560 void MP4RtpHintTrack::AddESConfigurationPacket()
|
|
561 {
|
|
562 if (m_pWriteHint == NULL) {
|
|
563 throw new MP4Error("no hint pending",
|
|
564 "MP4RtpAddESConfigurationPacket");
|
|
565 }
|
|
566
|
|
567 u_int8_t* pConfig = NULL;
|
|
568 u_int32_t configSize = 0;
|
|
569
|
|
570 m_pFile->GetTrackESConfiguration(m_pRefTrack->GetId(),
|
|
571 &pConfig, &configSize);
|
|
572
|
|
573 if (pConfig == NULL) {
|
|
574 return;
|
|
575 }
|
|
576
|
|
577 ASSERT(m_pMaxPacketSizeProperty);
|
|
578
|
|
579 if (configSize > m_pMaxPacketSizeProperty->GetValue()) {
|
|
580 throw new MP4Error("ES configuration is too large for RTP payload",
|
|
581 "MP4RtpAddESConfigurationPacket");
|
|
582 }
|
|
583
|
|
584 AddPacket(false);
|
|
585
|
|
586 MP4RtpPacket* pPacket = m_pWriteHint->GetCurrentPacket();
|
|
587 ASSERT(pPacket);
|
|
588
|
|
589 // This is ugly!
|
|
590 // To get the ES configuration data somewhere known
|
|
591 // we create a sample data reference that points to
|
|
592 // this hint track (not the media track)
|
|
593 // and this sample of the hint track
|
|
594 // the offset into this sample is filled in during the write process
|
|
595 MP4RtpSampleData* pData = new MP4RtpSampleData(pPacket);
|
|
596
|
|
597 pData->SetEmbeddedImmediate(m_writeSampleId, pConfig, configSize);
|
|
598
|
|
599 pPacket->AddData(pData);
|
|
600
|
|
601 m_bytesThisHint += configSize;
|
|
602 m_bytesThisPacket += configSize;
|
|
603 m_pTpyl->IncrementValue(configSize);
|
|
604 m_pTrpy->IncrementValue(configSize);
|
|
605 }
|
|
606
|
|
607 void MP4RtpHintTrack::WriteHint(MP4Duration duration, bool isSyncSample)
|
|
608 {
|
|
609 if (m_pWriteHint == NULL) {
|
|
610 throw new MP4Error("no hint pending", "MP4WriteRtpHint");
|
|
611 }
|
|
612
|
|
613 u_int8_t* pBytes;
|
|
614 u_int64_t numBytes;
|
|
615
|
|
616 m_pFile->EnableMemoryBuffer();
|
|
617
|
|
618 m_pWriteHint->Write(m_pFile);
|
|
619
|
|
620 m_pFile->DisableMemoryBuffer(&pBytes, &numBytes);
|
|
621
|
|
622 WriteSample(pBytes, numBytes, duration, 0, isSyncSample);
|
|
623
|
|
624 MP4Free(pBytes);
|
|
625
|
|
626 // update statistics
|
|
627 if (m_bytesThisPacket > m_pPmax->GetValue()) {
|
|
628 m_pPmax->SetValue(m_bytesThisPacket);
|
|
629 }
|
|
630
|
|
631 if (duration > m_pDmax->GetValue()) {
|
|
632 m_pDmax->SetValue(duration);
|
|
633 }
|
|
634
|
|
635 MP4Timestamp startTime;
|
|
636
|
|
637 GetSampleTimes(m_writeHintId, &startTime, NULL);
|
|
638
|
|
639 if (startTime < m_thisSec + GetTimeScale()) {
|
|
640 m_bytesThisSec += m_bytesThisHint;
|
|
641 } else {
|
|
642 if (m_bytesThisSec > m_pMaxr->GetValue()) {
|
|
643 m_pMaxr->SetValue(m_bytesThisSec);
|
|
644 }
|
|
645 m_thisSec = startTime - (startTime % GetTimeScale());
|
|
646 m_bytesThisSec = m_bytesThisHint;
|
|
647 }
|
|
648
|
|
649 // cleanup
|
|
650 delete m_pWriteHint;
|
|
651 m_pWriteHint = NULL;
|
|
652 }
|
|
653
|
|
654 void MP4RtpHintTrack::FinishWrite()
|
|
655 {
|
|
656 if (m_writeHintId != MP4_INVALID_SAMPLE_ID) {
|
|
657 m_pMaxPdu->SetValue(m_pPmax->GetValue());
|
|
658 if (m_pNump->GetValue()) {
|
|
659 m_pAvgPdu->SetValue(m_pTrpy->GetValue() / m_pNump->GetValue());
|
|
660 }
|
|
661
|
|
662 m_pMaxBitRate->SetValue(m_pMaxr->GetValue() * 8);
|
|
663 if (GetDuration()) {
|
|
664 m_pAvgBitRate->SetValue(
|
|
665 m_pTrpy->GetValue() * 8 * GetTimeScale() / GetDuration());
|
|
666 }
|
|
667 }
|
|
668
|
|
669 MP4Track::FinishWrite();
|
|
670 }
|
|
671
|
|
672 void MP4RtpHintTrack::InitStats()
|
|
673 {
|
|
674 MP4Atom* pHinfAtom = m_pTrakAtom->FindAtom("trak.udta.hinf");
|
|
675
|
|
676 ASSERT(pHinfAtom);
|
|
677
|
|
678 pHinfAtom->FindProperty("hinf.trpy.bytes", (MP4Property**)&m_pTrpy);
|
|
679 pHinfAtom->FindProperty("hinf.nump.packets", (MP4Property**)&m_pNump);
|
|
680 pHinfAtom->FindProperty("hinf.tpyl.bytes", (MP4Property**)&m_pTpyl);
|
|
681 pHinfAtom->FindProperty("hinf.maxr.bytes", (MP4Property**)&m_pMaxr);
|
|
682 pHinfAtom->FindProperty("hinf.dmed.bytes", (MP4Property**)&m_pDmed);
|
|
683 pHinfAtom->FindProperty("hinf.dimm.bytes", (MP4Property**)&m_pDimm);
|
|
684 pHinfAtom->FindProperty("hinf.pmax.bytes", (MP4Property**)&m_pPmax);
|
|
685 pHinfAtom->FindProperty("hinf.dmax.milliSecs", (MP4Property**)&m_pDmax);
|
|
686
|
|
687 MP4Atom* pHmhdAtom = m_pTrakAtom->FindAtom("trak.mdia.minf.hmhd");
|
|
688
|
|
689 ASSERT(pHmhdAtom);
|
|
690
|
|
691 pHmhdAtom->FindProperty("hmhd.maxPduSize", (MP4Property**)&m_pMaxPdu);
|
|
692 pHmhdAtom->FindProperty("hmhd.avgPduSize", (MP4Property**)&m_pAvgPdu);
|
|
693 pHmhdAtom->FindProperty("hmhd.maxBitRate", (MP4Property**)&m_pMaxBitRate);
|
|
694 pHmhdAtom->FindProperty("hmhd.avgBitRate", (MP4Property**)&m_pAvgBitRate);
|
|
695
|
|
696 MP4Integer32Property* pMaxrPeriod = NULL;
|
|
697 pHinfAtom->FindProperty("hinf.maxr.granularity",
|
|
698 (MP4Property**)&pMaxrPeriod);
|
|
699 if (pMaxrPeriod) {
|
|
700 pMaxrPeriod->SetValue(1000); // 1 second
|
|
701 }
|
|
702 }
|
|
703
|
|
704
|
|
705 MP4RtpHint::MP4RtpHint(MP4RtpHintTrack* pTrack)
|
|
706 {
|
|
707 m_pTrack = pTrack;
|
|
708
|
|
709 AddProperty( /* 0 */
|
|
710 new MP4Integer16Property("packetCount"));
|
|
711 AddProperty( /* 1 */
|
|
712 new MP4Integer16Property("reserved"));
|
|
713 }
|
|
714
|
|
715 MP4RtpHint::~MP4RtpHint()
|
|
716 {
|
|
717 for (u_int32_t i = 0; i < m_rtpPackets.Size(); i++) {
|
|
718 delete m_rtpPackets[i];
|
|
719 }
|
|
720 }
|
|
721
|
|
722 MP4RtpPacket* MP4RtpHint::AddPacket()
|
|
723 {
|
|
724 MP4RtpPacket* pPacket = new MP4RtpPacket(this);
|
|
725 m_rtpPackets.Add(pPacket);
|
|
726
|
|
727 // packetCount property
|
|
728 ((MP4Integer16Property*)m_pProperties[0])->IncrementValue();
|
|
729
|
|
730 pPacket->SetBFrame(m_isBFrame);
|
|
731 pPacket->SetTimestampOffset(m_timestampOffset);
|
|
732
|
|
733 return pPacket;
|
|
734 }
|
|
735
|
|
736 void MP4RtpHint::Read(MP4File* pFile)
|
|
737 {
|
|
738 // call base class Read for required properties
|
|
739 MP4Container::Read(pFile);
|
|
740
|
|
741 u_int16_t numPackets =
|
|
742 ((MP4Integer16Property*)m_pProperties[0])->GetValue();
|
|
743
|
|
744 for (u_int16_t i = 0; i < numPackets; i++) {
|
|
745 MP4RtpPacket* pPacket = new MP4RtpPacket(this);
|
|
746
|
|
747 m_rtpPackets.Add(pPacket);
|
|
748
|
|
749 pPacket->Read(pFile);
|
|
750 }
|
|
751
|
|
752 VERBOSE_READ_HINT(pFile->GetVerbosity(),
|
|
753 printf("ReadHint:\n"); Dump(stdout, 10, false););
|
|
754 }
|
|
755
|
|
756 void MP4RtpHint::Write(MP4File* pFile)
|
|
757 {
|
|
758 u_int64_t hintStartPos = pFile->GetPosition();
|
|
759
|
|
760 MP4Container::Write(pFile);
|
|
761
|
|
762 u_int64_t packetStartPos = pFile->GetPosition();
|
|
763
|
|
764 u_int32_t i;
|
|
765
|
|
766 // first write out packet (and data) entries
|
|
767 for (i = 0; i < m_rtpPackets.Size(); i++) {
|
|
768 m_rtpPackets[i]->Write(pFile);
|
|
769 }
|
|
770
|
|
771 // now let packets write their extra data into the hint sample
|
|
772 for (i = 0; i < m_rtpPackets.Size(); i++) {
|
|
773 m_rtpPackets[i]->WriteEmbeddedData(pFile, hintStartPos);
|
|
774 }
|
|
775
|
|
776 u_int64_t endPos = pFile->GetPosition();
|
|
777
|
|
778 pFile->SetPosition(packetStartPos);
|
|
779
|
|
780 // finally rewrite the packet and data entries
|
|
781 // which now contain the correct offsets for the embedded data
|
|
782 for (i = 0; i < m_rtpPackets.Size(); i++) {
|
|
783 m_rtpPackets[i]->Write(pFile);
|
|
784 }
|
|
785
|
|
786 pFile->SetPosition(endPos);
|
|
787
|
|
788 VERBOSE_WRITE_HINT(pFile->GetVerbosity(),
|
|
789 printf("WriteRtpHint:\n"); Dump(stdout, 14, false));
|
|
790 }
|
|
791
|
|
792 void MP4RtpHint::Dump(FILE* pFile, u_int8_t indent, bool dumpImplicits)
|
|
793 {
|
|
794 MP4Container::Dump(pFile, indent, dumpImplicits);
|
|
795
|
|
796 for (u_int32_t i = 0; i < m_rtpPackets.Size(); i++) {
|
|
797 Indent(pFile, indent);
|
|
798 fprintf(pFile, "RtpPacket: %u\n", i);
|
|
799 m_rtpPackets[i]->Dump(pFile, indent + 1, dumpImplicits);
|
|
800 }
|
|
801 }
|
|
802
|
|
803 MP4RtpPacket::MP4RtpPacket(MP4RtpHint* pHint)
|
|
804 {
|
|
805 m_pHint = pHint;
|
|
806
|
|
807 AddProperty( /* 0 */
|
|
808 new MP4Integer32Property("relativeXmitTime"));
|
|
809 AddProperty( /* 1 */
|
|
810 new MP4BitfieldProperty("reserved1", 2));
|
|
811 AddProperty( /* 2 */
|
|
812 new MP4BitfieldProperty("Pbit", 1));
|
|
813 AddProperty( /* 3 */
|
|
814 new MP4BitfieldProperty("Xbit", 1));
|
|
815 AddProperty( /* 4 */
|
|
816 new MP4BitfieldProperty("reserved2", 4));
|
|
817 AddProperty( /* 5 */
|
|
818 new MP4BitfieldProperty("Mbit", 1));
|
|
819 AddProperty( /* 6 */
|
|
820 new MP4BitfieldProperty("payloadType", 7));
|
|
821 AddProperty( /* 7 */
|
|
822 new MP4Integer16Property("sequenceNumber"));
|
|
823 AddProperty( /* 8 */
|
|
824 new MP4BitfieldProperty("reserved3", 13));
|
|
825 AddProperty( /* 9 */
|
|
826 new MP4BitfieldProperty("extraFlag", 1));
|
|
827 AddProperty( /* 10 */
|
|
828 new MP4BitfieldProperty("bFrameFlag", 1));
|
|
829 AddProperty( /* 11 */
|
|
830 new MP4BitfieldProperty("repeatFlag", 1));
|
|
831 AddProperty( /* 12 */
|
|
832 new MP4Integer16Property("entryCount"));
|
|
833 }
|
|
834
|
|
835 MP4RtpPacket::~MP4RtpPacket()
|
|
836 {
|
|
837 for (u_int32_t i = 0; i < m_rtpData.Size(); i++) {
|
|
838 delete m_rtpData[i];
|
|
839 }
|
|
840 }
|
|
841
|
|
842 void MP4RtpPacket::AddExtraProperties()
|
|
843 {
|
|
844 AddProperty( /* 13 */
|
|
845 new MP4Integer32Property("extraInformationLength"));
|
|
846
|
|
847 // This is a bit of a hack, since the tlv entries are really defined
|
|
848 // as atoms but there is only one type defined now, rtpo, and getting
|
|
849 // our atom code hooked up here would be a major pain with little gain
|
|
850
|
|
851 AddProperty( /* 14 */
|
|
852 new MP4Integer32Property("tlvLength"));
|
|
853 AddProperty( /* 15 */
|
|
854 new MP4StringProperty("tlvType"));
|
|
855 AddProperty( /* 16 */
|
|
856 new MP4Integer32Property("timestampOffset"));
|
|
857
|
|
858 ((MP4Integer32Property*)m_pProperties[13])->SetValue(16);
|
|
859 ((MP4Integer32Property*)m_pProperties[14])->SetValue(12);
|
|
860 ((MP4StringProperty*)m_pProperties[15])->SetFixedLength(4);
|
|
861 ((MP4StringProperty*)m_pProperties[15])->SetValue("rtpo");
|
|
862 }
|
|
863
|
|
864 void MP4RtpPacket::Read(MP4File* pFile)
|
|
865 {
|
|
866 // call base class Read for required properties
|
|
867 MP4Container::Read(pFile);
|
|
868
|
|
869 // read extra info if present
|
|
870 // we only support the rtpo field!
|
|
871 if (((MP4BitfieldProperty*)m_pProperties[9])->GetValue() == 1) {
|
|
872 ReadExtra(pFile);
|
|
873 }
|
|
874
|
|
875 u_int16_t numDataEntries =
|
|
876 ((MP4Integer16Property*)m_pProperties[12])->GetValue();
|
|
877
|
|
878 // read data entries
|
|
879 for (u_int16_t i = 0; i < numDataEntries; i++) {
|
|
880 u_int8_t dataType;
|
|
881 pFile->PeekBytes(&dataType, 1);
|
|
882
|
|
883 MP4RtpData* pData;
|
|
884
|
|
885 switch (dataType) {
|
|
886 case 0:
|
|
887 pData = new MP4RtpNullData(this);
|
|
888 break;
|
|
889 case 1:
|
|
890 pData = new MP4RtpImmediateData(this);
|
|
891 break;
|
|
892 case 2:
|
|
893 pData = new MP4RtpSampleData(this);
|
|
894 break;
|
|
895 case 3:
|
|
896 pData = new MP4RtpSampleDescriptionData(this);
|
|
897 break;
|
|
898 default:
|
|
899 throw new MP4Error("unknown packet data entry type",
|
|
900 "MP4ReadHint");
|
|
901 }
|
|
902
|
|
903 m_rtpData.Add(pData);
|
|
904
|
|
905 // read data entry's properties
|
|
906 pData->Read(pFile);
|
|
907 }
|
|
908 }
|
|
909
|
|
910 void MP4RtpPacket::ReadExtra(MP4File* pFile)
|
|
911 {
|
|
912 AddExtraProperties();
|
|
913
|
|
914 int32_t extraLength = (int32_t)pFile->ReadUInt32();
|
|
915
|
|
916 if (extraLength < 4) {
|
|
917 throw new MP4Error("bad packet extra info length",
|
|
918 "MP4RtpPacket::ReadExtra");
|
|
919 }
|
|
920 extraLength -= 4;
|
|
921
|
|
922 while (extraLength > 0) {
|
|
923 u_int32_t entryLength = pFile->ReadUInt32();
|
|
924 u_int32_t entryTag = pFile->ReadUInt32();
|
|
925
|
|
926 if (entryLength < 8) {
|
|
927 throw new MP4Error("bad packet extra info entry length",
|
|
928 "MP4RtpPacket::ReadExtra");
|
|
929 }
|
|
930
|
|
931 if (entryTag == STRTOINT32("rtpo") && entryLength == 12) {
|
|
932 // read the rtp timestamp offset
|
|
933 m_pProperties[16]->Read(pFile);
|
|
934 } else {
|
|
935 // ignore it, LATER carry it along
|
|
936 pFile->SetPosition(pFile->GetPosition() + entryLength - 8);
|
|
937 }
|
|
938
|
|
939 extraLength -= entryLength;
|
|
940 }
|
|
941
|
|
942 if (extraLength < 0) {
|
|
943 throw new MP4Error("invalid packet extra info length",
|
|
944 "MP4RtpPacket::ReadExtra");
|
|
945 }
|
|
946 }
|
|
947
|
|
948 void MP4RtpPacket::Set(u_int8_t payloadNumber,
|
|
949 u_int32_t packetId, bool setMbit)
|
|
950 {
|
|
951 ((MP4BitfieldProperty*)m_pProperties[5])->SetValue(setMbit);
|
|
952 ((MP4BitfieldProperty*)m_pProperties[6])->SetValue(payloadNumber);
|
|
953 ((MP4Integer16Property*)m_pProperties[7])->SetValue(packetId);
|
|
954 }
|
|
955
|
|
956 int32_t MP4RtpPacket::GetTransmitOffset()
|
|
957 {
|
|
958 return ((MP4Integer32Property*)m_pProperties[0])->GetValue();
|
|
959 }
|
|
960
|
|
961 void MP4RtpPacket::SetTransmitOffset(int32_t transmitOffset)
|
|
962 {
|
|
963 ((MP4Integer32Property*)m_pProperties[0])->SetValue(transmitOffset);
|
|
964 }
|
|
965
|
|
966 bool MP4RtpPacket::GetPBit()
|
|
967 {
|
|
968 return ((MP4BitfieldProperty*)m_pProperties[2])->GetValue();
|
|
969 }
|
|
970
|
|
971 bool MP4RtpPacket::GetXBit()
|
|
972 {
|
|
973 return ((MP4BitfieldProperty*)m_pProperties[3])->GetValue();
|
|
974 }
|
|
975
|
|
976 bool MP4RtpPacket::GetMBit()
|
|
977 {
|
|
978 return ((MP4BitfieldProperty*)m_pProperties[5])->GetValue();
|
|
979 }
|
|
980
|
|
981 u_int8_t MP4RtpPacket::GetPayload()
|
|
982 {
|
|
983 return ((MP4BitfieldProperty*)m_pProperties[6])->GetValue();
|
|
984 }
|
|
985
|
|
986 u_int16_t MP4RtpPacket::GetSequenceNumber()
|
|
987 {
|
|
988 return ((MP4Integer16Property*)m_pProperties[7])->GetValue();
|
|
989 }
|
|
990
|
|
991 bool MP4RtpPacket::IsBFrame()
|
|
992 {
|
|
993 return ((MP4BitfieldProperty*)m_pProperties[10])->GetValue();
|
|
994 }
|
|
995
|
|
996 void MP4RtpPacket::SetBFrame(bool isBFrame)
|
|
997 {
|
|
998 ((MP4BitfieldProperty*)m_pProperties[10])->SetValue(isBFrame);
|
|
999 }
|
|
1000
|
|
1001 void MP4RtpPacket::SetTimestampOffset(u_int32_t timestampOffset)
|
|
1002 {
|
|
1003 if (timestampOffset == 0) {
|
|
1004 return;
|
|
1005 }
|
|
1006
|
|
1007 ASSERT(((MP4BitfieldProperty*)m_pProperties[9])->GetValue() == 0);
|
|
1008
|
|
1009 // set X bit
|
|
1010 ((MP4BitfieldProperty*)m_pProperties[9])->SetValue(1);
|
|
1011
|
|
1012 AddExtraProperties();
|
|
1013
|
|
1014 ((MP4Integer32Property*)m_pProperties[16])->SetValue(timestampOffset);
|
|
1015 }
|
|
1016
|
|
1017 void MP4RtpPacket::AddData(MP4RtpData* pData)
|
|
1018 {
|
|
1019 m_rtpData.Add(pData);
|
|
1020
|
|
1021 // increment entry count property
|
|
1022 ((MP4Integer16Property*)m_pProperties[12])->IncrementValue();
|
|
1023 }
|
|
1024
|
|
1025 u_int32_t MP4RtpPacket::GetDataSize()
|
|
1026 {
|
|
1027 u_int32_t totalDataSize = 0;
|
|
1028
|
|
1029 for (u_int32_t i = 0; i < m_rtpData.Size(); i++) {
|
|
1030 totalDataSize += m_rtpData[i]->GetDataSize();
|
|
1031 }
|
|
1032
|
|
1033 return totalDataSize;
|
|
1034 }
|
|
1035
|
|
1036 void MP4RtpPacket::GetData(u_int8_t* pDest)
|
|
1037 {
|
|
1038 for (u_int32_t i = 0; i < m_rtpData.Size(); i++) {
|
|
1039 m_rtpData[i]->GetData(pDest);
|
|
1040 pDest += m_rtpData[i]->GetDataSize();
|
|
1041 }
|
|
1042 }
|
|
1043
|
|
1044 void MP4RtpPacket::Write(MP4File* pFile)
|
|
1045 {
|
|
1046 MP4Container::Write(pFile);
|
|
1047
|
|
1048 for (u_int32_t i = 0; i < m_rtpData.Size(); i++) {
|
|
1049 m_rtpData[i]->Write(pFile);
|
|
1050 }
|
|
1051 }
|
|
1052
|
|
1053 void MP4RtpPacket::WriteEmbeddedData(MP4File* pFile, u_int64_t startPos)
|
|
1054 {
|
|
1055 for (u_int32_t i = 0; i < m_rtpData.Size(); i++) {
|
|
1056 m_rtpData[i]->WriteEmbeddedData(pFile, startPos);
|
|
1057 }
|
|
1058 }
|
|
1059
|
|
1060 void MP4RtpPacket::Dump(FILE* pFile, u_int8_t indent, bool dumpImplicits)
|
|
1061 {
|
|
1062 MP4Container::Dump(pFile, indent, dumpImplicits);
|
|
1063
|
|
1064 for (u_int32_t i = 0; i < m_rtpData.Size(); i++) {
|
|
1065 Indent(pFile, indent);
|
|
1066 fprintf(pFile, "RtpData: %u\n", i);
|
|
1067 m_rtpData[i]->Dump(pFile, indent + 1, dumpImplicits);
|
|
1068 }
|
|
1069 }
|
|
1070
|
|
1071 MP4RtpData::MP4RtpData(MP4RtpPacket* pPacket)
|
|
1072 {
|
|
1073 m_pPacket = pPacket;
|
|
1074
|
|
1075 AddProperty( /* 0 */
|
|
1076 new MP4Integer8Property("type"));
|
|
1077 }
|
|
1078
|
|
1079 MP4Track* MP4RtpData::FindTrackFromRefIndex(u_int8_t refIndex)
|
|
1080 {
|
|
1081 MP4Track* pTrack;
|
|
1082
|
|
1083 if (refIndex == (u_int8_t)-1) {
|
|
1084 // ourselves
|
|
1085 pTrack = GetPacket()->GetHint()->GetTrack();
|
|
1086 } else if (refIndex == 0) {
|
|
1087 // our reference track
|
|
1088 pTrack = GetPacket()->GetHint()->GetTrack()->GetRefTrack();
|
|
1089 } else {
|
|
1090 // some other track
|
|
1091 MP4RtpHintTrack* pHintTrack =
|
|
1092 GetPacket()->GetHint()->GetTrack();
|
|
1093
|
|
1094 MP4Atom* pTrakAtom = pHintTrack->GetTrakAtom();
|
|
1095 ASSERT(pTrakAtom);
|
|
1096
|
|
1097 MP4Integer32Property* pTrackIdProperty = NULL;
|
|
1098 pTrakAtom->FindProperty(
|
|
1099 "trak.tref.hint.entries",
|
|
1100 (MP4Property**)&pTrackIdProperty);
|
|
1101 ASSERT(pTrackIdProperty);
|
|
1102
|
|
1103 u_int32_t refTrackId =
|
|
1104 pTrackIdProperty->GetValue(refIndex - 1);
|
|
1105
|
|
1106 pTrack = pHintTrack->GetFile()->GetTrack(refTrackId);
|
|
1107 }
|
|
1108
|
|
1109 return pTrack;
|
|
1110 }
|
|
1111
|
|
1112 MP4RtpNullData::MP4RtpNullData(MP4RtpPacket* pPacket)
|
|
1113 : MP4RtpData(pPacket)
|
|
1114 {
|
|
1115 ((MP4Integer8Property*)m_pProperties[0])->SetValue(0);
|
|
1116
|
|
1117 AddProperty( /* 1 */
|
|
1118 new MP4BytesProperty("pad", 15));
|
|
1119
|
|
1120 ((MP4BytesProperty*)m_pProperties[1])->SetFixedSize(15);
|
|
1121 }
|
|
1122
|
|
1123 MP4RtpImmediateData::MP4RtpImmediateData(MP4RtpPacket* pPacket)
|
|
1124 : MP4RtpData(pPacket)
|
|
1125 {
|
|
1126 ((MP4Integer8Property*)m_pProperties[0])->SetValue(1);
|
|
1127
|
|
1128 AddProperty( /* 1 */
|
|
1129 new MP4Integer8Property("count"));
|
|
1130 AddProperty( /* 2 */
|
|
1131 new MP4BytesProperty("data", 14));
|
|
1132
|
|
1133 ((MP4BytesProperty*)m_pProperties[2])->SetFixedSize(14);
|
|
1134 }
|
|
1135
|
|
1136 void MP4RtpImmediateData::Set(const u_int8_t* pBytes, u_int8_t numBytes)
|
|
1137 {
|
|
1138 ((MP4Integer8Property*)m_pProperties[1])->SetValue(numBytes);
|
|
1139 ((MP4BytesProperty*)m_pProperties[2])->SetValue(pBytes, numBytes);
|
|
1140 }
|
|
1141
|
|
1142 u_int16_t MP4RtpImmediateData::GetDataSize()
|
|
1143 {
|
|
1144 return ((MP4Integer8Property*)m_pProperties[1])->GetValue();
|
|
1145 }
|
|
1146
|
|
1147 void MP4RtpImmediateData::GetData(u_int8_t* pDest)
|
|
1148 {
|
|
1149 u_int8_t* pValue;
|
|
1150 u_int32_t valueSize;
|
|
1151 ((MP4BytesProperty*)m_pProperties[2])->GetValue(&pValue, &valueSize);
|
|
1152
|
|
1153 memcpy(pDest, pValue, GetDataSize());
|
|
1154 MP4Free(pValue);
|
|
1155 }
|
|
1156
|
|
1157 MP4RtpSampleData::MP4RtpSampleData(MP4RtpPacket* pPacket)
|
|
1158 : MP4RtpData(pPacket)
|
|
1159 {
|
|
1160 ((MP4Integer8Property*)m_pProperties[0])->SetValue(2);
|
|
1161
|
|
1162 AddProperty( /* 1 */
|
|
1163 new MP4Integer8Property("trackRefIndex"));
|
|
1164 AddProperty( /* 2 */
|
|
1165 new MP4Integer16Property("length"));
|
|
1166 AddProperty( /* 3 */
|
|
1167 new MP4Integer32Property("sampleNumber"));
|
|
1168 AddProperty( /* 4 */
|
|
1169 new MP4Integer32Property("sampleOffset"));
|
|
1170 AddProperty( /* 5 */
|
|
1171 new MP4Integer16Property("bytesPerBlock"));
|
|
1172 AddProperty( /* 6 */
|
|
1173 new MP4Integer16Property("samplesPerBlock"));
|
|
1174
|
|
1175 ((MP4Integer16Property*)m_pProperties[5])->SetValue(1);
|
|
1176 ((MP4Integer16Property*)m_pProperties[6])->SetValue(1);
|
|
1177
|
|
1178 m_pRefData = NULL;
|
|
1179 m_pRefTrack = NULL;
|
|
1180 m_refSampleId = MP4_INVALID_SAMPLE_ID;
|
|
1181 m_refSampleOffset = 0;
|
|
1182 }
|
|
1183
|
|
1184 void MP4RtpSampleData::SetEmbeddedImmediate(MP4SampleId sampleId,
|
|
1185 u_int8_t* pData, u_int16_t dataLength)
|
|
1186 {
|
|
1187 ((MP4Integer8Property*)m_pProperties[1])->SetValue((u_int8_t)-1);
|
|
1188 ((MP4Integer16Property*)m_pProperties[2])->SetValue(dataLength);
|
|
1189 ((MP4Integer32Property*)m_pProperties[3])->SetValue(sampleId);
|
|
1190 ((MP4Integer32Property*)m_pProperties[4])->SetValue(0);
|
|
1191 CHECK_AND_FREE(m_pRefData);
|
|
1192 m_pRefData = pData;
|
|
1193 }
|
|
1194
|
|
1195 void MP4RtpSampleData::SetReferenceSample(
|
|
1196 MP4SampleId refSampleId, u_int32_t refSampleOffset,
|
|
1197 u_int16_t sampleLength)
|
|
1198 {
|
|
1199 ((MP4Integer8Property*)m_pProperties[1])->SetValue(0);
|
|
1200 ((MP4Integer16Property*)m_pProperties[2])->SetValue(sampleLength);
|
|
1201 ((MP4Integer32Property*)m_pProperties[3])->SetValue(refSampleId);
|
|
1202 ((MP4Integer32Property*)m_pProperties[4])->SetValue(refSampleOffset);
|
|
1203 }
|
|
1204
|
|
1205 void MP4RtpSampleData::SetEmbeddedSample(
|
|
1206 MP4SampleId sampleId, MP4Track* pRefTrack,
|
|
1207 MP4SampleId refSampleId, u_int32_t refSampleOffset,
|
|
1208 u_int16_t sampleLength)
|
|
1209 {
|
|
1210 ((MP4Integer8Property*)m_pProperties[1])->SetValue((u_int8_t)-1);
|
|
1211 ((MP4Integer16Property*)m_pProperties[2])->SetValue(sampleLength);
|
|
1212 ((MP4Integer32Property*)m_pProperties[3])->SetValue(sampleId);
|
|
1213 ((MP4Integer32Property*)m_pProperties[4])->SetValue(0);
|
|
1214 m_pRefTrack = pRefTrack;
|
|
1215 m_refSampleId = refSampleId;
|
|
1216 m_refSampleOffset = refSampleOffset;
|
|
1217 }
|
|
1218
|
|
1219 u_int16_t MP4RtpSampleData::GetDataSize()
|
|
1220 {
|
|
1221 return ((MP4Integer16Property*)m_pProperties[2])->GetValue();
|
|
1222 }
|
|
1223
|
|
1224 void MP4RtpSampleData::GetData(u_int8_t* pDest)
|
|
1225 {
|
|
1226 u_int8_t trackRefIndex =
|
|
1227 ((MP4Integer8Property*)m_pProperties[1])->GetValue();
|
|
1228
|
|
1229 MP4Track* pSampleTrack =
|
|
1230 FindTrackFromRefIndex(trackRefIndex);
|
|
1231
|
|
1232 pSampleTrack->ReadSampleFragment(
|
|
1233 ((MP4Integer32Property*)m_pProperties[3])->GetValue(), // sampleId
|
|
1234 ((MP4Integer32Property*)m_pProperties[4])->GetValue(), // sampleOffset
|
|
1235 ((MP4Integer16Property*)m_pProperties[2])->GetValue(), // sampleLength
|
|
1236 pDest);
|
|
1237 }
|
|
1238
|
|
1239 void MP4RtpSampleData::WriteEmbeddedData(MP4File* pFile, u_int64_t startPos)
|
|
1240 {
|
|
1241 // if not using embedded data, nothing to do
|
|
1242 if (((MP4Integer8Property*)m_pProperties[1])->GetValue() != (u_int8_t)-1) {
|
|
1243 return;
|
|
1244 }
|
|
1245
|
|
1246 // figure out the offset within this hint sample for this embedded data
|
|
1247 u_int64_t offset = pFile->GetPosition() - startPos;
|
|
1248 ASSERT(offset <= 0xFFFFFFFF);
|
|
1249 ((MP4Integer32Property*)m_pProperties[4])->SetValue((u_int32_t)offset);
|
|
1250
|
|
1251 u_int16_t length = ((MP4Integer16Property*)m_pProperties[2])->GetValue();
|
|
1252
|
|
1253 if (m_pRefData) {
|
|
1254 pFile->WriteBytes(m_pRefData, length);
|
|
1255 return;
|
|
1256 }
|
|
1257
|
|
1258 if (m_refSampleId != MP4_INVALID_SAMPLE_ID) {
|
|
1259 u_int8_t* pSample = NULL;
|
|
1260 u_int32_t sampleSize = 0;
|
|
1261
|
|
1262 ASSERT(m_pRefTrack);
|
|
1263 m_pRefTrack->ReadSample(m_refSampleId, &pSample, &sampleSize);
|
|
1264
|
|
1265 ASSERT(m_refSampleOffset + length <= sampleSize);
|
|
1266
|
|
1267 pFile->WriteBytes(&pSample[m_refSampleOffset], length);
|
|
1268
|
|
1269 MP4Free(pSample);
|
|
1270 return;
|
|
1271 }
|
|
1272 }
|
|
1273
|
|
1274 MP4RtpSampleDescriptionData::MP4RtpSampleDescriptionData(MP4RtpPacket* pPacket)
|
|
1275 : MP4RtpData(pPacket)
|
|
1276 {
|
|
1277 ((MP4Integer8Property*)m_pProperties[0])->SetValue(3);
|
|
1278
|
|
1279 AddProperty( /* 1 */
|
|
1280 new MP4Integer8Property("trackRefIndex"));
|
|
1281 AddProperty( /* 2 */
|
|
1282 new MP4Integer16Property("length"));
|
|
1283 AddProperty( /* 3 */
|
|
1284 new MP4Integer32Property("sampleDescriptionIndex"));
|
|
1285 AddProperty( /* 4 */
|
|
1286 new MP4Integer32Property("sampleDescriptionOffset"));
|
|
1287 AddProperty( /* 5 */
|
|
1288 new MP4Integer32Property("reserved"));
|
|
1289 }
|
|
1290
|
|
1291 void MP4RtpSampleDescriptionData::Set(u_int32_t sampleDescrIndex,
|
|
1292 u_int32_t offset, u_int16_t length)
|
|
1293 {
|
|
1294 ((MP4Integer16Property*)m_pProperties[2])->SetValue(length);
|
|
1295 ((MP4Integer32Property*)m_pProperties[3])->SetValue(sampleDescrIndex);
|
|
1296 ((MP4Integer32Property*)m_pProperties[4])->SetValue(offset);
|
|
1297 }
|
|
1298
|
|
1299 u_int16_t MP4RtpSampleDescriptionData::GetDataSize()
|
|
1300 {
|
|
1301 return ((MP4Integer16Property*)m_pProperties[2])->GetValue();
|
|
1302 }
|
|
1303
|
|
1304 void MP4RtpSampleDescriptionData::GetData(u_int8_t* pDest)
|
|
1305 {
|
|
1306 // we start with the index into our track references
|
|
1307 u_int8_t trackRefIndex =
|
|
1308 ((MP4Integer8Property*)m_pProperties[1])->GetValue();
|
|
1309
|
|
1310 // from which we can find the track structure
|
|
1311 MP4Track* pSampleTrack =
|
|
1312 FindTrackFromRefIndex(trackRefIndex);
|
|
1313
|
|
1314 // next find the desired atom in the track's sample description table
|
|
1315 u_int32_t sampleDescrIndex =
|
|
1316 ((MP4Integer32Property*)m_pProperties[3])->GetValue();
|
|
1317
|
|
1318 MP4Atom* pTrakAtom =
|
|
1319 pSampleTrack->GetTrakAtom();
|
|
1320
|
|
1321 char sdName[64];
|
|
1322 sprintf(sdName, "trak.mdia.minf.stbl.stsd.*[%u]", sampleDescrIndex);
|
|
1323
|
|
1324 MP4Atom* pSdAtom =
|
|
1325 pTrakAtom->FindAtom(sdName);
|
|
1326
|
|
1327 // bad reference
|
|
1328 if (pSdAtom == NULL) {
|
|
1329 throw new MP4Error("invalid sample description index",
|
|
1330 "MP4RtpSampleDescriptionData::GetData");
|
|
1331 }
|
|
1332
|
|
1333 // check validity of the upcoming copy
|
|
1334 u_int16_t length =
|
|
1335 ((MP4Integer16Property*)m_pProperties[2])->GetValue();
|
|
1336 u_int32_t offset =
|
|
1337 ((MP4Integer32Property*)m_pProperties[4])->GetValue();
|
|
1338
|
|
1339 if (offset + length > pSdAtom->GetSize()) {
|
|
1340 throw new MP4Error("offset and/or length are too large",
|
|
1341 "MP4RtpSampleDescriptionData::GetData");
|
|
1342 }
|
|
1343
|
|
1344 // now we use the raw file to get the desired bytes
|
|
1345
|
|
1346 MP4File* pFile = GetPacket()->GetHint()->GetTrack()->GetFile();
|
|
1347
|
|
1348 u_int64_t orgPos = pFile->GetPosition();
|
|
1349
|
|
1350 // It's not entirely clear from the spec whether the offset is from
|
|
1351 // the start of the sample descirption atom, or the start of the atom's
|
|
1352 // data. I believe it is the former, but the commented out code will
|
|
1353 // realize the latter interpretation if I turn out to be wrong.
|
|
1354 u_int64_t dataPos = pSdAtom->GetStart();
|
|
1355 //u_int64_t dataPos = pSdAtom->GetEnd() - pSdAtom->GetSize();
|
|
1356
|
|
1357 pFile->SetPosition(dataPos + offset);
|
|
1358
|
|
1359 pFile->ReadBytes(pDest, length);
|
|
1360
|
|
1361 pFile->SetPosition(orgPos);
|
|
1362 }
|
|
1363
|