AMPEG4ElementaryAssembler.cpp revision 8d342970108926c4ea355c90d26a2a353ec0fd47
1/* 2 * Copyright (C) 2010 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17#include "AMPEG4ElementaryAssembler.h" 18 19#include "ARTPSource.h" 20 21#include <media/stagefright/foundation/ABuffer.h> 22#include <media/stagefright/foundation/ADebug.h> 23#include <media/stagefright/foundation/AMessage.h> 24#include <media/stagefright/foundation/hexdump.h> 25 26#include <stdint.h> 27 28#define BE_VERBOSE 0 29 30namespace android { 31 32// static 33AMPEG4ElementaryAssembler::AMPEG4ElementaryAssembler(const sp<AMessage> ¬ify) 34 : mNotifyMsg(notify), 35 mAccessUnitRTPTime(0), 36 mNextExpectedSeqNoValid(false), 37 mNextExpectedSeqNo(0), 38 mAccessUnitDamaged(false) { 39} 40 41AMPEG4ElementaryAssembler::~AMPEG4ElementaryAssembler() { 42} 43 44ARTPAssembler::AssemblyStatus AMPEG4ElementaryAssembler::addPacket( 45 const sp<ARTPSource> &source) { 46 List<sp<ABuffer> > *queue = source->queue(); 47 48 if (queue->empty()) { 49 return NOT_ENOUGH_DATA; 50 } 51 52 if (mNextExpectedSeqNoValid) { 53 List<sp<ABuffer> >::iterator it = queue->begin(); 54 while (it != queue->end()) { 55 if ((uint32_t)(*it)->int32Data() >= mNextExpectedSeqNo) { 56 break; 57 } 58 59 it = queue->erase(it); 60 } 61 62 if (queue->empty()) { 63 return NOT_ENOUGH_DATA; 64 } 65 } 66 67 sp<ABuffer> buffer = *queue->begin(); 68 69 if (!mNextExpectedSeqNoValid) { 70 mNextExpectedSeqNoValid = true; 71 mNextExpectedSeqNo = (uint32_t)buffer->int32Data(); 72 } else if ((uint32_t)buffer->int32Data() != mNextExpectedSeqNo) { 73#if BE_VERBOSE 74 LOG(VERBOSE) << "Not the sequence number I expected"; 75#endif 76 77 return WRONG_SEQUENCE_NUMBER; 78 } 79 80 uint32_t rtpTime; 81 CHECK(buffer->meta()->findInt32("rtp-time", (int32_t *)&rtpTime)); 82 83 if (mPackets.size() > 0 && rtpTime != mAccessUnitRTPTime) { 84 submitAccessUnit(); 85 } 86 mAccessUnitRTPTime = rtpTime; 87 88 mPackets.push_back(buffer); 89 // hexdump(buffer->data(), buffer->size()); 90 91 queue->erase(queue->begin()); 92 ++mNextExpectedSeqNo; 93 94 return OK; 95} 96 97void AMPEG4ElementaryAssembler::submitAccessUnit() { 98 CHECK(!mPackets.empty()); 99 100#if BE_VERBOSE 101 LOG(VERBOSE) << "Access unit complete (" << mPackets.size() << " nal units)"; 102#endif 103 104 size_t totalSize = 0; 105 for (List<sp<ABuffer> >::iterator it = mPackets.begin(); 106 it != mPackets.end(); ++it) { 107 totalSize += (*it)->size(); 108 } 109 110 sp<ABuffer> accessUnit = new ABuffer(totalSize); 111 size_t offset = 0; 112 for (List<sp<ABuffer> >::iterator it = mPackets.begin(); 113 it != mPackets.end(); ++it) { 114 sp<ABuffer> nal = *it; 115 memcpy(accessUnit->data() + offset, nal->data(), nal->size()); 116 offset += nal->size(); 117 } 118 119 CopyTimes(accessUnit, *mPackets.begin()); 120 121#if 0 122 printf(mAccessUnitDamaged ? "X" : "."); 123 fflush(stdout); 124#endif 125 126 if (mAccessUnitDamaged) { 127 accessUnit->meta()->setInt32("damaged", true); 128 } 129 130 mPackets.clear(); 131 mAccessUnitDamaged = false; 132 133 sp<AMessage> msg = mNotifyMsg->dup(); 134 msg->setObject("access-unit", accessUnit); 135 msg->post(); 136} 137 138ARTPAssembler::AssemblyStatus AMPEG4ElementaryAssembler::assembleMore( 139 const sp<ARTPSource> &source) { 140 AssemblyStatus status = addPacket(source); 141 if (status == MALFORMED_PACKET) { 142 mAccessUnitDamaged = true; 143 } 144 return status; 145} 146 147void AMPEG4ElementaryAssembler::packetLost() { 148 CHECK(mNextExpectedSeqNoValid); 149 LOG(VERBOSE) << "packetLost (expected " << mNextExpectedSeqNo << ")"; 150 151 ++mNextExpectedSeqNo; 152 153 mAccessUnitDamaged = true; 154} 155 156void AMPEG4ElementaryAssembler::onByeReceived() { 157 sp<AMessage> msg = mNotifyMsg->dup(); 158 msg->setInt32("eos", true); 159 msg->post(); 160} 161 162} // namespace android 163