// Copyright 2019 Fuzhou Rockchip Electronics Co., Ltd. All rights reserved.
|
// Use of this source code is governed by a BSD-style license that can be
|
// found in the LICENSE file.
|
|
#include "h264_server_media_subsession.hh"
|
|
#include <liveMedia/H264VideoRTPSink.hh>
|
#include <liveMedia/H264VideoStreamDiscreteFramer.hh>
|
|
#include "media_type.h"
|
#include "utils.h"
|
|
namespace easymedia
|
{
|
H264ServerMediaSubsession* H264ServerMediaSubsession::createNew(UsageEnvironment& env, Live555MediaInput& wisInput)
|
{
|
return new H264ServerMediaSubsession(env, wisInput);
|
}
|
|
H264ServerMediaSubsession::H264ServerMediaSubsession(UsageEnvironment& env, Live555MediaInput& mediaInput)
|
: OnDemandServerMediaSubsession(env, True /*reuse the first source*/), fMediaInput(mediaInput),
|
fEstimatedKbps(1000), fDoneFlag(0), fDummyRTPSink(NULL), fGetSdpCount(10), fAuxSDPLine(NULL)
|
{
|
}
|
|
H264ServerMediaSubsession::~H264ServerMediaSubsession()
|
{
|
LOG_FILE_FUNC_LINE();
|
if (fAuxSDPLine != NULL) {
|
delete[] fAuxSDPLine;
|
fAuxSDPLine = NULL;
|
}
|
}
|
|
// std::mutex H264ServerMediaSubsession::kMutex;
|
// std::list<unsigned int> H264ServerMediaSubsession::kSessionIdList;
|
|
void H264ServerMediaSubsession::startStream(
|
unsigned clientSessionId, void* streamToken, TaskFunc* rtcpRRHandler, void* rtcpRRHandlerClientData,
|
unsigned short& rtpSeqNum, unsigned& rtpTimestamp,
|
ServerRequestAlternativeByteHandler* serverRequestAlternativeByteHandler,
|
void* serverRequestAlternativeByteHandlerClientData)
|
{
|
OnDemandServerMediaSubsession::startStream(clientSessionId, streamToken, rtcpRRHandler, rtcpRRHandlerClientData,
|
rtpSeqNum, rtpTimestamp, serverRequestAlternativeByteHandler,
|
serverRequestAlternativeByteHandlerClientData);
|
// kMutex.lock();
|
if (fMediaInput.GetStartVideoStreamCallback() != NULL) {
|
fMediaInput.GetStartVideoStreamCallback()();
|
}
|
if (kSessionIdList.empty()) {
|
fMediaInput.Start(envir());
|
}
|
LOG("%s:%s:%p - clientSessionId: 0x%08x\n", __FILE__, __func__, this, clientSessionId);
|
kSessionIdList.push_back(clientSessionId);
|
// kMutex.unlock();
|
}
|
void H264ServerMediaSubsession::deleteStream(unsigned clientSessionId, void*& streamToken)
|
{
|
// kMutex.lock();
|
LOG("%s - clientSessionId: 0x%08x\n", __func__, clientSessionId);
|
kSessionIdList.remove(clientSessionId);
|
if (kSessionIdList.empty()) {
|
fMediaInput.Stop(envir());
|
}
|
// kMutex.unlock();
|
OnDemandServerMediaSubsession::deleteStream(clientSessionId, streamToken);
|
}
|
|
static void afterPlayingDummy(void* clientData)
|
{
|
H264ServerMediaSubsession* subsess = (H264ServerMediaSubsession*)clientData;
|
LOG("%s, set done.\n", __func__);
|
// Signal the event loop that we're done:
|
subsess->afterPlayingDummy1();
|
}
|
|
void H264ServerMediaSubsession::afterPlayingDummy1()
|
{
|
// Unschedule any pending 'checking' task:
|
envir().taskScheduler().unscheduleDelayedTask(nextTask());
|
// Signal the event loop that we're done:
|
setDoneFlag();
|
}
|
|
static void checkForAuxSDPLine(void* clientData)
|
{
|
H264ServerMediaSubsession* subsess = (H264ServerMediaSubsession*)clientData;
|
subsess->checkForAuxSDPLine1();
|
}
|
|
void H264ServerMediaSubsession::checkForAuxSDPLine1()
|
{
|
nextTask() = NULL;
|
|
char const* dasl;
|
if (fAuxSDPLine != NULL) {
|
// Signal the event loop that we're done:
|
setDoneFlag();
|
} else if (fDummyRTPSink != NULL && (dasl = fDummyRTPSink->auxSDPLine()) != NULL) {
|
fAuxSDPLine = strDup(dasl);
|
fDummyRTPSink = NULL;
|
|
// Signal the event loop that we're done:
|
setDoneFlag();
|
} else if (!fDoneFlag) {
|
if (fGetSdpCount-- < 0) {
|
setDoneFlag();
|
LOG("%s:%s:%p: get sdp time out.\n", __FILE__, __func__, this);
|
} else {
|
// try again after a brief delay:
|
int uSecsToDelay = 100000; // 100 ms
|
LOG_FILE_FUNC_LINE();
|
nextTask() =
|
envir().taskScheduler().scheduleDelayedTask(uSecsToDelay, (TaskFunc*)checkForAuxSDPLine, this);
|
}
|
}
|
}
|
|
char const* H264ServerMediaSubsession::getAuxSDPLine(RTPSink* rtpSink, FramedSource* inputSource)
|
{
|
// Note: For MPEG-4 video buffer, the 'config' information isn't known
|
// until we start reading the Buffer. This means that "rtpSink"s
|
// "auxSDPLine()" will be NULL initially, and we need to start reading
|
// data from our buffer until this changes.
|
if (fAuxSDPLine != NULL) {
|
return fAuxSDPLine;
|
}
|
if (fDummyRTPSink == NULL) {
|
// force I framed
|
if (fMediaInput.GetStartVideoStreamCallback() != NULL) {
|
fMediaInput.GetStartVideoStreamCallback()();
|
}
|
fDummyRTPSink = rtpSink;
|
fGetSdpCount = 10;
|
fDummyRTPSink->startPlaying(*inputSource, afterPlayingDummy, this);
|
checkForAuxSDPLine(this);
|
}
|
envir().taskScheduler().doEventLoop(&fDoneFlag);
|
return fAuxSDPLine;
|
}
|
|
FramedSource* H264ServerMediaSubsession::createNewStreamSource(unsigned clientSessionId, unsigned& estBitrate)
|
{
|
LOG("%s:%s:%p - clientSessionId: 0x%08x\n", __FILE__, __func__, this, clientSessionId);
|
estBitrate = fMediaInput.getMaxIdrSize();
|
if (estBitrate < fEstimatedKbps) {
|
estBitrate = fEstimatedKbps;
|
}
|
|
// Create a framer for the Video Elementary Stream:
|
FramedSource* source =
|
H264VideoStreamDiscreteFramer::createNew(envir(), fMediaInput.videoSource(CODEC_TYPE_H264));
|
LOG("h264 framedsource : %p, estBitrate = %u \n", source, estBitrate);
|
return source;
|
}
|
|
RTPSink* H264ServerMediaSubsession::createNewRTPSink(Groupsock* rtpGroupsock, unsigned char rtpPayloadTypeIfDynamic,
|
FramedSource* inputSource)
|
{
|
if (!inputSource) {
|
LOG("inputSource is not ready, can not create new rtp sink\n");
|
return NULL;
|
}
|
setVideoRTPSinkBufferSize();
|
LOG_FILE_FUNC_LINE();
|
RTPSink* rtp_sink = H264VideoRTPSink::createNew(envir(), rtpGroupsock, rtpPayloadTypeIfDynamic);
|
LOG("h264 rtp sink : %p\n", rtp_sink);
|
return rtp_sink;
|
}
|
|
} // namespace easymedia
|
