tg2sip/PacketReassembler.cpp

75 lines
1.7 KiB
C++

//
// Created by Grishka on 19.03.2018.
//
#include "PacketReassembler.h"
#include "logging.h"
#include <assert.h>
using namespace tgvoip;
PacketReassembler::PacketReassembler(){
}
PacketReassembler::~PacketReassembler(){
}
void PacketReassembler::Reset(){
}
void PacketReassembler::AddFragment(Buffer pkt, unsigned int fragmentIndex, unsigned int fragmentCount, uint32_t pts, bool keyframe){
//LOGD("add fragment: ts=%u, index=%u of %u", pts, fragmentIndex, fragmentCount);
if(pts!=currentTimestamp){
assert(fragmentCount<=255);
currentTimestamp=pts;
/*for(Buffer& b:parts){
if(!b.IsEmpty())
b=Buffer();
}*/
currentPacketPartCount=fragmentCount;
currentIsKeyframe=keyframe;
receivedPartCount=0;
}
if(fragmentCount==0 || fragmentCount==1){
callback(std::move(pkt), pts, keyframe);
return;
}
if(fragmentIndex<currentPacketPartCount){
parts[fragmentIndex]=std::move(pkt);
}
if(parts.size()<currentPacketPartCount)
return;
receivedPartCount++;
bool anyAreEmpty=false;
for(int i=0;i<currentPacketPartCount;i++){
if(parts[i].IsEmpty()){
anyAreEmpty=true;
break;
}
}
if(!anyAreEmpty && receivedPartCount==currentPacketPartCount){
//LOGI("part count %u", (unsigned int)parts.size());
BufferOutputStream out(10240);
for(int i=0;i<currentPacketPartCount;i++){
out.WriteBytes(parts[i]);
parts[i]=Buffer();
}
//LOGI("reassembled %u parts, %u bytes", (unsigned int)receivedPartCount, out.GetLength());
callback(Buffer(std::move(out)), currentTimestamp, currentIsKeyframe);
currentPacketPartCount=0;
currentTimestamp=0;
}
}
void PacketReassembler::SetCallback(std::function<void(Buffer packet, uint32_t pts, bool keyframe)> callback){
this->callback=callback;
}