bool FECDecoder::AddPacket(RTPPacket* packet)
{
//Check if it is a redundant packet
if (packet->GetCodec()==VideoCodec::RED)
{
//Get the redundant packet
RTPRedundantPacket *red = (RTPRedundantPacket *)packet;
//Check primary redundant type
if (red->GetPrimaryCodec()==VideoCodec::ULPFEC)
{
//Create new FEC data
FECData *fec = new FECData(red->GetPrimaryPayloadData(),red->GetPrimaryPayloadSize());
//Log
Debug("-fec primary red data at %d\n",fec->GetBaseExtSeq());
//Append it
codes.insert(std::pair<DWORD,FECData*>(fec->GetBaseExtSeq(),fec));
//Packet contained no media
return false;
}
//Ensure we don't have it already
if (medias.find(packet->GetExtSeqNum())==medias.end())
//Add media packet
medias[packet->GetExtSeqNum()] = red->CreatePrimaryPacket();
//For each redundant data
for (int i=0;i<red->GetRedundantCount();++i)
{
//Check if it is a FEC pacekt
if (red->GetRedundantCodec(i)==VideoCodec::ULPFEC)
{
//Create new FEC data
FECData *fec = new FECData(red->GetRedundantPayloadData(i),red->GetRedundantPayloadSize(i));
//Log
Debug("-fec red data at %d\n",fec->GetBaseExtSeq());
//Append it
codes.insert(std::pair<DWORD,FECData*>(fec->GetBaseExtSeq(),fec));
}
}
} else if (packet->GetCodec()==VideoCodec::ULPFEC ) {
//Create new FEC data
FECData *fec = new FECData(packet->GetMediaData(),packet->GetMediaLength());
//Log
Debug("-fec data at %d\n",fec->GetBaseExtSeq());
//Append it
codes.insert(std::pair<DWORD,FECData*>(fec->GetBaseExtSeq(),fec));
//Packet contained no media
return false;
} else if (packet->GetCodec()==VideoCodec::FLEXFEC) {
//TODO
return false;
} else {
//Check if we have it already
if (medias.find(packet->GetExtSeqNum())!=medias.end())
//Do nothing
return false;
//Add media packet
medias[packet->GetExtSeqNum()] = packet->Clone();
}
//Get last seq number
DWORD seq = medias.rbegin()->first;
//Remove unused packets
RTPOrderedPackets::iterator it = medias.begin();
//Delete everything until seq-63
while (it->first<seq-63 && it!=medias.end())
{
//Delete rtp packet
delete (it->second);
//Erase it
medias.erase(it++);
}
//Now clean recovery codes
FECOrderedData::iterator it2 = codes.begin();
//Check base sequence
while (it2->first<seq-63 && it2!=codes.end())
{
//Delete object
delete(it2->second);
//Erase and move to next
codes.erase(it2++);
}
//Packet had media
return true;
}
