完善Frame相关的接口

src/Player/Frame.h

@@ -50,13 +50,19 @@ public:
     /**
      * 时间戳
      */
-    virtual uint32_t stamp() = 0;
+    virtual uint32_t stamp() const = 0;
 
     /**
      * 前缀长度，譬如264前缀为0x00 00 00 01,那么前缀长度就是4
      * aac前缀则为7个字节
      */
-    virtual uint32_t prefixSize() = 0;
+    virtual uint32_t prefixSize() const = 0;
+
+    /**
+     * 返回是否为关键帧
+     * @return
+     */
+    virtual bool keyFrame() const = 0;
 };
 
 /**
@@ -85,10 +91,8 @@ public:
     /**
      * 写入帧数据
      * @param frame 帧
-     * @param key_pos 是否为关键帧
-     * @return 是否为关键帧
      */
-    virtual bool inputFrame(const Frame::Ptr &frame,bool key_pos) = 0;
+    virtual void inputFrame(const Frame::Ptr &frame) = 0;
 };
 
 
@@ -120,11 +124,9 @@ public:
     /**
      * 输入数据帧
      * @param frame
-     * @param key_pos
      */
-    bool inputFrame(const Frame::Ptr &frame,bool key_pos) override{
-        _frameRing->write(frame,key_pos);
-        return key_pos;
+    void inputFrame(const Frame::Ptr &frame) override{
+        _frameRing->write(frame,frame->keyFrame());
     }
 protected:
     RingType::Ptr _frameRing;
@@ -143,10 +145,10 @@ public:
     uint32_t size() const override {
         return buffer.size();
     }
-    uint32_t stamp() override {
+    uint32_t stamp() const override {
         return timeStamp;
     }
-    uint32_t prefixSize() override{
+    uint32_t prefixSize() const override{
         return iPrefixSize;
     }
 
@@ -157,6 +159,10 @@ public:
     CodecId getCodecId() const override{
         return CodecH264;
     }
+
+    bool keyFrame() const override {
+        return type == 5;
+    }
 public:
     uint16_t sequence;
     uint32_t timeStamp;
@@ -178,10 +184,10 @@ public:
     uint32_t size() const override {
         return aac_frame_length;
     }
-    uint32_t stamp() override {
+    uint32_t stamp() const override {
         return timeStamp;
     }
-    uint32_t prefixSize() override{
+    uint32_t prefixSize() const override{
         return iPrefixSize;
     }
 
@@ -192,6 +198,10 @@ public:
     CodecId getCodecId() const override{
         return CodecAAC;
     }
+
+    bool keyFrame() const override {
+        return false;
+    }
 public:
     unsigned int syncword; //12 bslbf 同步字The bit string ‘1111 1111 1111’，说明一个ADTS帧的开始
     unsigned int id;        //1 bslbf   MPEG 标示符, 设置为1

src/Player/Track.h

@@ -21,6 +21,9 @@ public:
     Track(){}
     virtual ~Track(){}
 
+    /**
+     * 根据sdp生成Track对象
+     */
     static Ptr getTrackBySdp(const string &sdp);
 };
 
@@ -165,9 +168,8 @@ public:
     /**
      * 输入数据帧,并获取sps pps
      * @param frame 数据帧
-     * @param key_pos 是否为关键帧
      */
-    bool inputFrame(const Frame::Ptr &frame,bool key_pos) override{
+    void inputFrame(const Frame::Ptr &frame) override{
         int type = (*((uint8_t *)frame->data() + frame->prefixSize())) & 0x1F;
         switch (type){
             case 7:{
@@ -193,7 +195,7 @@ public:
                     insertFrame->type = 7;
                     insertFrame->buffer = _sps;
                     insertFrame->iPrefixSize = 0;
-                    VideoTrack::inputFrame(insertFrame, true);
+                    VideoTrack::inputFrame(insertFrame);
                 }
 
                 if(!_pps.empty()){
@@ -202,19 +204,18 @@ public:
                     insertFrame->type = 8;
                     insertFrame->buffer = _pps;
                     insertFrame->iPrefixSize = 0;
-                    VideoTrack::inputFrame(insertFrame, false);
+                    VideoTrack::inputFrame(insertFrame);
                 }
-                VideoTrack::inputFrame(frame, false);
+                VideoTrack::inputFrame(frame);
             }
                 break;
 
             case 1:{
                 //B or P
-                VideoTrack::inputFrame(frame, false);
+                VideoTrack::inputFrame(frame);
             }
                 break;
         }
-        return type == 5;
     }
 private:
     /**

src/RTP/AACRtpCodec.cpp

@@ -17,8 +17,8 @@ AACRtpEncoder::AACRtpEncoder(uint32_t ui32Ssrc,
         AACRtpDecoder(ui32SampleRate){
 }
 
-bool AACRtpEncoder::inputFrame(const Frame::Ptr &frame, bool key_pos) {
-    RtpCodec::inputFrame(frame, false);
+void AACRtpEncoder::inputFrame(const Frame::Ptr &frame) {
+    RtpCodec::inputFrame(frame);
 
     GET_CONFIG_AND_REGISTER(uint32_t, cycleMS, Config::Rtp::kCycleMS);
     auto uiStamp = frame->stamp();
@@ -47,7 +47,6 @@ bool AACRtpEncoder::inputFrame(const Frame::Ptr &frame, bool key_pos) {
         ptr += (m_ui32MtuSize - 20);
         iSize -= (m_ui32MtuSize - 20);
     }
-    return false;
 }
 
 void AACRtpEncoder::makeAACRtp(const void *pData, unsigned int uiLen, bool bMark, uint32_t uiStamp) {
@@ -122,7 +121,7 @@ bool AACRtpDecoder::inputRtp(const RtpPacket::Ptr &rtppack, bool key_pos) {
 
 void AACRtpDecoder::onGetAdts(const AACFrame::Ptr &frame) {
     //写入环形缓存
-    RtpCodec::inputFrame(frame, false);
+    RtpCodec::inputFrame(frame);
     m_adts = obtainFrame();
 }
 

src/RTP/AACRtpCodec.h

@@ -68,9 +68,8 @@ public:
     /**
      * 输入aac 数据，必须带dats头
      * @param frame 带dats头的aac数据
-     * @param key_pos 此参数内部强制转换为false,请忽略之
      */
-    bool inputFrame(const Frame::Ptr &frame, bool key_pos = false) override;
+    void inputFrame(const Frame::Ptr &frame) override;
 private:
     void makeAACRtp(const void *pData, unsigned int uiLen, bool bMark, uint32_t uiStamp);
 private:

src/RTP/H264RtpCodec.cpp

@@ -99,7 +99,7 @@ bool H264RtpDecoder::decodeRtp(const RtpPacket::Ptr &rtppack) {
 
 void H264RtpDecoder::onGetH264(const H264Frame::Ptr &frame) {
     //写入环形缓存
-    RtpCodec::inputFrame(frame,frame->type == 5);
+    RtpCodec::inputFrame(frame);
     m_h264frame = obtainFrame();
 }
 
@@ -118,13 +118,11 @@ H264RtpEncoder::H264RtpEncoder(uint32_t ui32Ssrc,
                 ui8Interleaved) {
 }
 
-bool H264RtpEncoder::inputFrame(const Frame::Ptr &frame, bool key_pos) {
-    auto pcData = frame->data() + frame->prefixSize();
-
-    key_pos = (((uint8_t *) (pcData))[0] & 0x1F) == 5;
-    RtpCodec::inputFrame(frame, key_pos);
+void H264RtpEncoder::inputFrame(const Frame::Ptr &frame) {
+    RtpCodec::inputFrame(frame);
 
     GET_CONFIG_AND_REGISTER(uint32_t,cycleMS,Config::Rtp::kCycleMS);
+    auto pcData = frame->data() + frame->prefixSize();
     auto uiStamp = frame->stamp();
     auto iLen = frame->size() - frame->prefixSize();
 
@@ -167,8 +165,6 @@ bool H264RtpEncoder::inputFrame(const Frame::Ptr &frame, bool key_pos) {
     } else {
         makeH264Rtp(pcData, iLen, true, uiStamp);
     }
-
-    return key_pos;
 }
 
 void H264RtpEncoder::makeH264Rtp(const void* data, unsigned int len, bool mark, uint32_t uiStamp) {

src/RTP/H264RtpCodec.h

@@ -67,9 +67,8 @@ public:
     /**
      * 输入264帧
      * @param frame 帧数据，必须
-     * @param key_pos
      */
-    bool inputFrame(const Frame::Ptr &frame, bool key_pos) override;
+    void inputFrame(const Frame::Ptr &frame) override;
 private:
     void makeH264Rtp(const void *pData, unsigned int uiLen, bool bMark, uint32_t uiStamp);
 private:

src/RTP/RtpCodec.h

@@ -162,6 +162,16 @@ public:
     RtpCodec(){}
     virtual ~RtpCodec(){}
 
+    /**
+     * 根据CodecId生成Rtp打包器
+     * @param codecId
+     * @param ui32Ssrc
+     * @param ui32MtuSize
+     * @param ui32SampleRate
+     * @param ui8PlayloadType
+     * @param ui8Interleaved
+     * @return
+     */
     static Ptr getRtpEncoderById(CodecId codecId,
                                  uint32_t ui32Ssrc,
                                  uint32_t ui32MtuSize,
@@ -169,6 +179,12 @@ public:
                                  uint8_t ui8PlayloadType,
                                  uint8_t ui8Interleaved);
 
+    /**
+     * 根据CodecId生成Rtp解包器
+     * @param codecId
+     * @param ui32SampleRate
+     * @return
+     */
     static Ptr getRtpDecoderById(CodecId codecId,uint32_t ui32SampleRate);
 };
 

src/Rtsp/RtspEncoder.h

@@ -71,10 +71,9 @@ public:
     /**
      * 输入帧数据，驱动rtp打包
      * @param frame 帧数据
-     * @param key_pos 是否为关键帧
      */
-    bool inputFrame(const Frame::Ptr &frame,bool key_pos) override{
-        return _encoder->inputFrame(frame,key_pos);
+    void inputFrame(const Frame::Ptr &frame) override{
+        _encoder->inputFrame(frame);
     }
 
     /**
@@ -313,14 +312,13 @@ public:
     /**
      * 写入帧数据然后打包rtp
      * @param frame 帧数据
-     * @param key_pos 是否为关键帧
      */
-    bool inputFrame(const Frame::Ptr &frame,bool key_pos = true) override {
+    void inputFrame(const Frame::Ptr &frame) override {
         auto it = _sdp_map.find(frame->getTrackType());
         if(it == _sdp_map.end()){
-            return false;
+            return ;
         }
-        return it->second->inputFrame(frame,key_pos);
+        it->second->inputFrame(frame);
     }
 
      /**