int RTPMultiplexerSmoother::Run()
{
timeval prev;
timespec wait;
DWORD sendingTime = 0;
//Calculate first
getUpdDifTime(&prev);
Log(">RTPMultiplexerSmoother run\n");
while(inited)
{
//Wait for new frame
if (!queue.Wait(0))
//Check again
continue;
//Get it
RTPPacketSched *sched = queue.Pop();
//Check it
if (!sched)
//Exit
continue;
//Multiplex
Multiplex(*(RTPPacket*)sched);
//Update sending time
sendingTime = sched->GetSendingTime();
//Lock
pthread_mutex_lock(&mutex);
//Calculate timeout
calcAbsTimeout(&wait,&prev,sendingTime);
//Wait next or stopped
pthread_cond_timedwait(&cond,&mutex,&wait);
//Unlock
pthread_mutex_unlock(&mutex);
//If it was last
if (sched->GetMark())
//Update time of the previous frame
getUpdDifTime(&prev);
//DElete it
delete(sched);
}
Log("<RTPMultiplexerSmoother run\n");
return 1;
}
int FLVEncoder::EncodeVideo()
{
timeval prev;
//Start
Log(">FLVEncoder encode video\n");
//Allocate media frame
RTMPVideoFrame frame(0,262143);
//Check codec
switch(videoCodec)
{
case VideoCodec::SORENSON:
//Ser Video codec
frame.SetVideoCodec(RTMPVideoFrame::FLV1);
break;
case VideoCodec::H264:
//Ser Video codec
frame.SetVideoCodec(RTMPVideoFrame::AVC);
//Set NAL type
frame.SetAVCType(RTMPVideoFrame::AVCNALU);
//No delay
frame.SetAVCTS(0);
break;
default:
return Error("-Wrong codec type %d\n",videoCodec);
}
//Create the encoder
VideoEncoder *encoder = VideoCodecFactory::CreateEncoder(videoCodec,videoProperties);
///Set frame rate
encoder->SetFrameRate(fps,bitrate,intra);
//Set dimensions
encoder->SetSize(width,height);
//Start capturing
videoInput->StartVideoCapture(width,height,fps);
//The time of the first one
gettimeofday(&prev,NULL);
//No wait for first
DWORD frameTime = 0;
Log(">FLVEncoder encode vide\n");
//Mientras tengamos que capturar
while(encodingVideo)
{
//Nos quedamos con el puntero antes de que lo cambien
BYTE* pic=videoInput->GrabFrame(frameTime);
//Ensure we are still encoding
if (!encodingVideo)
break;
//Check pic
if (!pic)
continue;
//Check if we need to send intra
if (sendFPU)
{
//Set it
encoder->FastPictureUpdate();
//Do not send anymore
sendFPU = false;
}
//Encode next frame
VideoFrame *encoded = encoder->EncodeFrame(pic,videoInput->GetBufferSize());
//Check
if (!encoded)
break;
//Check size
if (frame.GetMaxMediaSize()<encoded->GetLength())
{
//Not enougth space
Error("Not enought space to copy FLV encodec frame [frame:%d,encoded:%d",frame.GetMaxMediaSize(),encoded->GetLength());
//NExt
continue;
}
//Check
if (frameTime)
{
timespec ts;
//Lock
pthread_mutex_lock(&mutex);
//Calculate timeout
calcAbsTimeout(&ts,&prev,frameTime);
//Wait next or stopped
int canceled = !pthread_cond_timedwait(&cond,&mutex,&ts);
//Unlock
pthread_mutex_unlock(&mutex);
//Check if we have been canceled
if (canceled)
//Exit
break;
}
//Set sending time of previous frame
getUpdDifTime(&prev);
//Set timestamp
encoded->SetTimestamp(getDifTime(&first)/1000);
//Set next one
frameTime = 1000/fps;
//Set duration
encoded->SetDuration(frameTime);
//Get full frame
frame.SetVideoFrame(encoded->GetData(),encoded->GetLength());
//Set buffer size
frame.SetMediaSize(encoded->GetLength());
//Check type
if (encoded->IsIntra())
//Set type
frame.SetFrameType(RTMPVideoFrame::INTRA);
else
//Set type
frame.SetFrameType(RTMPVideoFrame::INTER);
//If we need desc but yet not have it
if (!frameDesc && encoded->IsIntra() && videoCodec==VideoCodec::H264)
{
//Create new description
AVCDescriptor desc;
//Set values
desc.SetConfigurationVersion(1);
desc.SetAVCProfileIndication(0x42);
desc.SetProfileCompatibility(0x80);
desc.SetAVCLevelIndication(0x0C);
desc.SetNALUnitLength(3);
//Get encoded data
BYTE *data = encoded->GetData();
//Get size
DWORD size = encoded->GetLength();
//get from frame
desc.AddParametersFromFrame(data,size);
//Crete desc frame
frameDesc = new RTMPVideoFrame(getDifTime(&first)/1000,desc);
//Lock
pthread_mutex_lock(&mutex);
//Send it
SendMediaFrame(frameDesc);
//unlock
pthread_mutex_unlock(&mutex);
}
//Lock
pthread_mutex_lock(&mutex);
//Set timestamp
frame.SetTimestamp(encoded->GetTimeStamp());
//Publish it
SendMediaFrame(&frame);
//For each listener
for(MediaFrameListeners::iterator it = mediaListeners.begin(); it!=mediaListeners.end(); ++it)
//Send it
(*it)->onMediaFrame(RTMPMediaStream::id,*encoded);
//unlock
pthread_mutex_unlock(&mutex);
}
Log("-FLVEncoder encode video end of loop\n");
//Stop the capture
videoInput->StopVideoCapture();
//Check
if (encoder)
//Exit
delete(encoder);
Log("<FLVEncoder encode vide\n");
//Exit
return 1;
}
int MP4Streamer::PlayLoop()
{
QWORD audioNext = MP4_INVALID_TIMESTAMP;
QWORD videoNext = MP4_INVALID_TIMESTAMP;
QWORD textNext = MP4_INVALID_TIMESTAMP;
timeval tv ;
timespec ts;
Log(">MP4Streamer::PlayLoop()\n");
//If it is from the begining
if (!seeked)
{
// If we have audio
if (audio)
{
//Reset
audio->Reset();
// Send audio
audioNext = audio->GetNextFrameTime();
}
// If we have video
if (video)
{
//Reset
video->Reset();
// Send video
videoNext = video->GetNextFrameTime();
}
// If we have text
if (text)
{
//Reset
text->Reset();
//Get the next frame time
textNext = text->GetNextFrameTime();
}
} else {
//If we have video
if (video)
//Get nearest i frame
videoNext = video->SeekNearestSyncFrame(seeked);
//If we have audio
if (audio)
//Get nearest frame
audioNext = audio->Seek(seeked);
//If we have text
if (text)
//Get nearest frame
textNext = text->Seek(seeked);
}
//If first text frame is not sent inmediatelly
if (text && textNext!=seeked)
//send previous text subtitle
text->ReadPrevious(seeked,listener);
// Calculate start time
getUpdDifTime(&tv);
//Lock
pthread_mutex_lock(&mutex);
//Time counter
QWORD t = 0;
// Wait control messages or finish of both streams
while ( opened && playing && (!(audioNext==MP4_INVALID_TIMESTAMP && videoNext==MP4_INVALID_TIMESTAMP && textNext==MP4_INVALID_TIMESTAMP)))
{
// Get next time
if (audioNext<videoNext)
{
if (audioNext<textNext)
t = audioNext;
else
t = textNext;
} else {
if (videoNext<textNext)
t = videoNext;
else
t = textNext;
}
// Wait time diff
QWORD now = (QWORD)getDifTime(&tv)/1000+seeked;
if (t>now)
{
//Calculate timeout
calcAbsTimeout(&ts,&tv,t-seeked);
//Wait next or stopped
pthread_cond_timedwait(&cond,&mutex,&ts);
//loop
continue;
}
// if we have to send audio
if (audioNext<=t)
audioNext = audio->Read(listener);
// or video
if (videoNext<=t)
videoNext = video->Read(listener);
// or text
if (textNext<=t)
textNext = text->Read(listener);
}
Log("-MP4Streamer::PlayLoop()\n");
//Check if we were stoped
bool stoped = !opened || !playing;
//Not playing anymore
playing = 0;
//Unlock
pthread_mutex_unlock(&mutex);
//Check end of file
if (!stoped && listener)
//End of file
listener->onEnd();
Log("<MP4Streamer::PlayLoop()\n");
return 1;
}
