96 lines
2.6 KiB
C++
96 lines
2.6 KiB
C++
/********************************************************************************
|
|
Copyright (C) 2012 Hugh Bailey <obs.jim@gmail.com>
|
|
|
|
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 2 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, write to the Free Software
|
|
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
|
|
********************************************************************************/
|
|
|
|
|
|
struct NetworkPacket
|
|
{
|
|
List<BYTE> data;
|
|
DWORD timestamp;
|
|
PacketType type;
|
|
};
|
|
|
|
struct BitRecord
|
|
{
|
|
UINT bits;
|
|
DWORD timestamp;
|
|
};
|
|
|
|
//max latency in milliseconds allowed when using the send buffer
|
|
const DWORD maxBufferTime = 600;
|
|
|
|
class RTMPPublisher : public NetworkStream
|
|
{
|
|
friend class DelayedPublisher;
|
|
|
|
protected:
|
|
RTMP *rtmp;
|
|
|
|
HANDLE hSendSempahore;
|
|
HANDLE hDataMutex;
|
|
HANDLE hSendThread;
|
|
|
|
bool bStopping;
|
|
|
|
int packetWaitType;
|
|
List<NetworkPacket> Packets;
|
|
List<BitRecord> bitsIn, bitsOut;
|
|
UINT bitsInPerTime, bitsOutPerTime, bitsQueued;
|
|
|
|
UINT numVideoPackets;
|
|
UINT maxBitRate, bufferSize, bufferTime;
|
|
|
|
QWORD bytesSent;
|
|
|
|
UINT numPFramesDumped;
|
|
UINT numBFramesDumped;
|
|
|
|
UINT numVideoPacketsBuffered;
|
|
DWORD firstBufferedVideoFrameTimestamp;
|
|
|
|
bool bPacketDumpMode;
|
|
|
|
BOOL bUseSendBuffer;
|
|
|
|
//all data sending is done in yet another separate thread to prevent blocking in the main capture thread
|
|
|
|
List<BYTE> sendBuffer;
|
|
int curSendBufferLen;
|
|
|
|
void SendLoop();
|
|
static DWORD SendThread(RTMPPublisher *publisher);
|
|
void DoIFrameDelay();
|
|
void DumpBFrame();
|
|
UINT BitsPerTime(List<BitRecord> *list);
|
|
void AddBits(List<BitRecord> *list, UINT bits, DWORD timestamp);
|
|
|
|
int FlushSendBuffer();
|
|
static int BufferedSend(RTMPSockBuf *sb, const char *buf, int len, RTMPPublisher *network);
|
|
|
|
public:
|
|
RTMPPublisher(RTMP *rtmpIn, BOOL bUseSendBuffer, UINT sendBufferSize);
|
|
~RTMPPublisher();
|
|
|
|
void SendPacket(BYTE *data, UINT size, DWORD timestamp, PacketType type);
|
|
|
|
void BeginPublishing();
|
|
|
|
double GetPacketStrain() const;
|
|
QWORD GetCurrentSentBytes();
|
|
DWORD NumDroppedFrames() const;
|
|
};
|