ZLMediaKit/src/RtmpCodec/H264RtmpCodec.h

68 lines
1.5 KiB
C
Raw Normal View History

2018-10-24 15:49:51 +08:00
//
2018-10-24 12:01:40 +08:00
// Created by xzl on 2018/10/24.
//
#ifndef ZLMEDIAKIT_H264RTMPCODEC_H
#define ZLMEDIAKIT_H264RTMPCODEC_H
#include "RtmpCodec.h"
#include "Util/ResourcePool.h"
using namespace ZL::Rtmp;
/**
* h264 Rtmp解码类
*/
class H264RtmpDecoder : public RtmpCodec ,public ResourcePoolHelper<H264Frame> {
public:
typedef std::shared_ptr<H264RtmpDecoder> Ptr;
H264RtmpDecoder();
~H264RtmpDecoder() {}
/**
* 264 Rtmp包
* @param rtmp Rtmp包
* @param key_pos
*/
bool inputRtmp(const RtmpPacket::Ptr &rtmp, bool key_pos = true) override;
TrackType getTrackType() const override{
return TrackVideo;
}
CodecId getCodecId() const override{
return CodecH264;
}
2018-10-24 14:21:59 +08:00
protected:
2018-10-24 12:01:40 +08:00
bool decodeRtmp(const RtmpPacket::Ptr &Rtmp);
void onGetH264_l(const char *pcData, int iLen, uint32_t ui32TimeStamp);
void onGetH264(const char *pcData, int iLen, uint32_t ui32TimeStamp);
H264Frame::Ptr obtainFrame();
2018-10-24 14:21:59 +08:00
protected:
2018-10-24 15:43:52 +08:00
H264Frame::Ptr _h264frame;
string _sps;
string _pps;
2018-10-24 12:01:40 +08:00
};
/**
* 264 Rtmp打包类
*/
class H264RtmpEncoder : public H264RtmpDecoder, public ResourcePoolHelper<RtmpPacket> {
public:
typedef std::shared_ptr<H264RtmpEncoder> Ptr;
H264RtmpEncoder();
~H264RtmpEncoder() {}
/**
* 264,sps pps帧
* @param frame
*/
void inputFrame(const Frame::Ptr &frame) override;
private:
void makeVideoConfigPkt();
};
#endif //ZLMEDIAKIT_H264RTMPCODEC_H