mirror of https://github.com/nostar/urfd.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
145 lines
3.5 KiB
145 lines
3.5 KiB
// Copyright © 2015 Jean-Luc Deltombe (LX3JL). All rights reserved.
|
|
|
|
// urfd -- The universal reflector
|
|
// Copyright © 2021 Thomas A. Early N7TAE
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
#include "Main.h"
|
|
#include "PacketStream.h"
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////
|
|
// constructor
|
|
|
|
#ifdef TRANSCODED_MODULES
|
|
CPacketStream::CPacketStream(std::shared_ptr<CUnixDgramReader> reader)
|
|
#else
|
|
CPacketStream::CPacketStream()
|
|
#endif
|
|
{
|
|
m_bOpen = false;
|
|
m_uiStreamId = 0;
|
|
m_uiPacketCntr = 0;
|
|
m_OwnerClient = nullptr;
|
|
#ifdef TRANSCODED_MODULES
|
|
m_CodecStream = nullptr;
|
|
m_TCReader = reader;
|
|
#endif
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////
|
|
// open / close
|
|
|
|
bool CPacketStream::OpenPacketStream(const CDvHeaderPacket &DvHeader, std::shared_ptr<CClient>client)
|
|
{
|
|
// not already open?
|
|
if ( !m_bOpen )
|
|
{
|
|
// update status
|
|
m_bOpen = true;
|
|
m_uiStreamId = DvHeader.GetStreamId();
|
|
m_uiPacketCntr = 0;
|
|
m_DvHeader = DvHeader;
|
|
m_OwnerClient = client;
|
|
m_LastPacketTime.start();
|
|
#ifdef TRANSCODED_MODULES
|
|
if (DvHeader.IsLocalOrigin()) // we only need transcoding if the source is local
|
|
{
|
|
auto mod = DvHeader.GetRpt2Module();
|
|
if (std::string::npos != std::string(TRANSCODED_MODULES).find(mod))
|
|
{
|
|
m_CodecStream = std::unique_ptr<CCodecStream>(new CCodecStream(this, m_uiStreamId, DvHeader.GetCodecIn(), m_TCReader));
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
std::cerr << "Could not find module '" << mod << " in the transcoded list, '" << TRANSCODED_MODULES << "'" << std::endl;
|
|
return false;
|
|
}
|
|
}
|
|
else
|
|
return true;
|
|
#else
|
|
return true;
|
|
#endif
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void CPacketStream::ClosePacketStream(void)
|
|
{
|
|
// update status
|
|
m_bOpen = false;
|
|
m_uiStreamId = 0;
|
|
m_OwnerClient.reset();
|
|
#ifdef TRANSCODED_MODULES
|
|
m_CodecStream.reset();
|
|
#endif
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////
|
|
// push & pop
|
|
|
|
void CPacketStream::Push(std::unique_ptr<CPacket> Packet)
|
|
{
|
|
// update stream dependent packet data
|
|
m_LastPacketTime.start();
|
|
if (Packet->IsDvFrame())
|
|
{
|
|
Packet->UpdatePids(m_uiPacketCntr++);
|
|
}
|
|
// transcoder avaliable ?
|
|
#ifdef TRANSCODED_MODULES
|
|
if ( m_CodecStream != nullptr )
|
|
{
|
|
// todo: verify no possibilty of double lock here
|
|
m_CodecStream->Lock();
|
|
{
|
|
// transcoder ready & frame need transcoding ?
|
|
if (Packet->IsDvFrame())
|
|
{
|
|
// yes, push packet to trancoder queue
|
|
// trancoder will push it after transcoding
|
|
// is completed
|
|
m_CodecStream->push(Packet);
|
|
}
|
|
else
|
|
{
|
|
// no, just bypass transcoder
|
|
push(Packet);
|
|
}
|
|
}
|
|
m_CodecStream->Unlock();
|
|
}
|
|
else
|
|
#endif
|
|
{
|
|
// otherwise, push direct push
|
|
push(Packet);
|
|
}
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////////////
|
|
// get
|
|
|
|
const CIp *CPacketStream::GetOwnerIp(void)
|
|
{
|
|
if ( m_OwnerClient != nullptr )
|
|
{
|
|
return &(m_OwnerClient->GetIp());
|
|
}
|
|
return nullptr;
|
|
}
|