/* 
 * File:   h264depacketizer.cpp
 * Author: Sergio
 * 
 * Created on 26 de enero de 2012, 9:46
 */

#include "VP9.h"
#include "VP9Depacketizer.h"
#include "VP9PayloadDescription.h"
#include "media.h"
#include "codecs.h"
#include "rtp.h"
#include "log.h"


VP9Depacketizer::VP9Depacketizer() : RTPDepacketizer(MediaFrame::Video,VideoCodec::VP9), frame(VideoCodec::VP9,0)
{
	//Generic vp9 config
	VP9CodecConfig config;
	//Set config size
	frame.AllocateCodecConfig(config.GetSize());
	//Serialize
	config.Serialize(frame.GetCodecConfigData(),frame.GetCodecConfigSize());
	//Set clock rate
	frame.SetClockRate(90000);
	//Disable shared buffer
	frame.DisableSharedBuffer();
}

VP9Depacketizer::~VP9Depacketizer()
{

}

void VP9Depacketizer::ResetFrame()
{
	//Reset frame data
	frame.Reset();
	//Generic vp9 config
	VP9CodecConfig config;
	//Set config size
	frame.AllocateCodecConfig(config.GetSize());
	//Serialize
	config.Serialize(frame.GetCodecConfigData(),frame.GetCodecConfigSize());
}

MediaFrame* VP9Depacketizer::AddPacket(const RTPPacket::shared& packet)
{
	//Get timestamp in ms
	auto ts = packet->GetExtTimestamp();
	//Check it is from same packet
	if (frame.GetTimeStamp()!=ts)
		//Reset frame
		ResetFrame();
	//If not timestamp
	if (frame.GetTimeStamp()==(DWORD)-1)
	{
		//Set timestamp
		frame.SetTimestamp(ts);
		//Set clock rate
		frame.SetClockRate(packet->GetClockRate());
		//Set time
		frame.SetTime(packet->GetTime());
		//Set sender time
		frame.SetSenderTime(packet->GetSenderTime());

		// Presentation time == DTS as there are no B-frames in RTP at the moment
		frame.SetPresentationTimestamp(packet->GetTimestamp());
	}

	//Set SSRC
	frame.SetSSRC(packet->GetSSRC());
	//Add payload
	AddPayload(packet->GetMediaData(),packet->GetMediaLength());
	//If it is last return frame
	return packet->GetMark() ? &frame : NULL;
}

MediaFrame* VP9Depacketizer::AddPayload(const BYTE* payload, DWORD len)
{
	//Check length
	if (!len)
		//Exit
		return NULL;
    
	//Is first?
	int first = !frame.GetLength(); 
    
	VP9PayloadDescription desc;

	//Decode payload descriptr
	DWORD descLen = desc.Parse(payload,len);
	
	//Check
	if (!descLen || len<descLen)
	{
		//Error
		UltraDebug("-VP9Depacketizer::AddPayload() | Error decoding VP9 payload header [desc:%d,len:%d]\n",descLen,len);
		return NULL;
	}
	
	//Skip desc
	DWORD pos = frame.AppendMedia(payload+descLen, len-descLen);
	
	//If it is the first one
	if (desc.startOfLayerFrame)
	{
		//Set values
		layer.pos =  pos;
		layer.size = len - descLen;
		layer.info.spatialLayerId	= desc.spatialLayerId;
		layer.info.temporalLayerId	= desc.temporalLayerId;
		if (desc.scalabiltiyStructureDataPresent && desc.scalabilityStructure.spatialLayerFrameResolutions.size()>desc.spatialLayerId)
		{
			//Set layer info
			layer.width  = desc.scalabilityStructure.spatialLayerFrameResolutions[desc.spatialLayerId].first;
			layer.height = desc.scalabilityStructure.spatialLayerFrameResolutions[desc.spatialLayerId].second;
		} else {
			layer.width = 0;
			layer.height = 0;
		}
	//If it is from the same layer frame
	} else if (layer.info.spatialLayerId == desc.spatialLayerId && layer.info.temporalLayerId == desc.temporalLayerId) {
		//Increase layer info
		layer.size += len - descLen;
		
	}
	//If it is latest
	if (desc.endOfLayerFrame)
		//Add it
		frame.AddLayerFrame(layer);
	
	BYTE inmediate[14];
	
	//If it has the scalability structure
	if (desc.scalabiltiyStructureDataPresent)
	{
		//Get topmost spatial layer
		auto it = desc.scalabilityStructure.spatialLayerFrameResolutions.rbegin();
		//Ensure we have sizes
		if (it!=desc.scalabilityStructure.spatialLayerFrameResolutions.rend())
		{
			//Set sizes
			frame.SetWidth(it->first);
			frame.SetHeight(it->second);
		}
		//Disable it, as it can't be bigger than 14 buytes
		desc.scalabiltiyStructureDataPresent = false;
	}
	
	//Serialize it
	DWORD size = desc.Serialize(inmediate,sizeof(inmediate));
	
	//Add RTP packet
	frame.AddRtpPacket(pos,len-descLen,inmediate,size);
	
	//If it is first
	if (first)
		//calculate if it is an iframe
		frame.SetIntra(!desc.interPicturePredictedLayerFrame);
    
	return &frame;
}