完善Frame相关的接口

This commit is contained in:
xiongziliang
2018-10-23 21:41:45 +08:00
parent 8930dd099e
commit 452f150f22
8 changed files with 61 additions and 43 deletions

View File

@@ -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

View File

@@ -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:
/**