ZLMediaKit/src/RTP/H264RtpCodec.h

//
// Created by xzl on 2018/10/18.
//

#ifndef ZLMEDIAKIT_H264RTPCODEC_H
#define ZLMEDIAKIT_H264RTPCODEC_H

#include "RtpCodec.h"
#include "Util/ResourcePool.h"

using namespace ZL::Util;

class H264RtpCodec : public RtpEncoder {
public:
    H264RtpCodec(uint32_t ui32Ssrc,
                 uint32_t ui32MtuSize = 1400,
                 uint32_t ui32SampleRate = 90000,
                 uint8_t ui8PlayloadType = 96,
                 uint8_t ui8Interleaved = TrackVideo * 2) :
            RtpEncoder(ui32Ssrc,ui32MtuSize,ui32SampleRate,ui8PlayloadType,ui8Interleaved) {
        m_framePool.setSize(32);
        m_h264frame = m_framePool.obtain();
    }
    ~H264RtpCodec(){}

    void inputFame(const Frame::Ptr &frame,bool key_pos) override;
    void inputRtp(const RtpPacket::Ptr &rtp, bool key_pos) override ;
private:
    void makeH264Rtp(const void *pData, unsigned int uiLen, bool bMark, uint32_t uiStamp);
    bool decodeRtp(const RtpPacket::Ptr &rtp, bool key_pos);
    void onGetH264(const H264Frame::Ptr &frame);
private:
    unsigned char m_aucSectionBuf[1600];
    H264Frame::Ptr m_h264frame;
    ResourcePool<H264Frame> m_framePool;
};


#endif //ZLMEDIAKIT_H264RTPCODEC_H