/********************************************************************************
 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;
};