mirror of
https://gitee.com/xia-chu/ZLMediaKit.git
synced 2026-05-06 19:07:49 +08:00
109 lines
3.2 KiB
C++
109 lines
3.2 KiB
C++
/*
|
|
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
|
*
|
|
* This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
|
|
*
|
|
* Use of this source code is governed by MIT license that can be found in the
|
|
* LICENSE file in the root of the source tree. All contributing project authors
|
|
* may be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#include "H264.h"
|
|
#include "SPSParser.h"
|
|
#include "Util/logger.h"
|
|
using namespace toolkit;
|
|
|
|
namespace mediakit{
|
|
|
|
static bool getAVCInfo(const char * sps,size_t sps_len,int &iVideoWidth, int &iVideoHeight, float &iVideoFps){
|
|
T_GetBitContext tGetBitBuf;
|
|
T_SPS tH264SpsInfo;
|
|
memset(&tGetBitBuf,0,sizeof(tGetBitBuf));
|
|
memset(&tH264SpsInfo,0,sizeof(tH264SpsInfo));
|
|
tGetBitBuf.pu8Buf = (uint8_t*)sps + 1;
|
|
tGetBitBuf.iBufSize = (int)(sps_len - 1);
|
|
if(0 != h264DecSeqParameterSet((void *) &tGetBitBuf, &tH264SpsInfo)){
|
|
return false;
|
|
}
|
|
h264GetWidthHeight(&tH264SpsInfo, &iVideoWidth, &iVideoHeight);
|
|
h264GeFramerate(&tH264SpsInfo, &iVideoFps);
|
|
//ErrorL << iVideoWidth << " " << iVideoHeight << " " << iVideoFps;
|
|
return true;
|
|
}
|
|
|
|
bool getAVCInfo(const string &strSps, int &iVideoWidth, int &iVideoHeight, float &iVideoFps) {
|
|
return getAVCInfo(strSps.data(), strSps.size(), iVideoWidth, iVideoHeight, iVideoFps);
|
|
}
|
|
|
|
static const char *memfind(const char *buf, size_t len, const char *subbuf, size_t sublen) {
|
|
for (ssize_t i = 0; i < (ssize_t)(len - sublen); ++i) {
|
|
if (memcmp(buf + i, subbuf, sublen) == 0) {
|
|
return buf + i;
|
|
}
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
void splitH264(const char *ptr, size_t len, size_t prefix, const std::function<void(const char *, size_t , size_t)> &cb) {
|
|
auto start = ptr + prefix;
|
|
auto end = ptr + len;
|
|
size_t next_prefix;
|
|
while (true) {
|
|
auto next_start = memfind(start, end - start, "\x00\x00\x01", 3);
|
|
if (next_start) {
|
|
//找到下一帧
|
|
if (*(next_start - 1) == 0x00) {
|
|
//这个是00 00 00 01开头
|
|
next_start -= 1;
|
|
next_prefix = 4;
|
|
} else {
|
|
//这个是00 00 01开头
|
|
next_prefix = 3;
|
|
}
|
|
//记得加上本帧prefix长度
|
|
cb(start - prefix, next_start - start + prefix, prefix);
|
|
//搜索下一帧末尾的起始位置
|
|
start = next_start + next_prefix;
|
|
//记录下一帧的prefix长度
|
|
prefix = next_prefix;
|
|
continue;
|
|
}
|
|
//未找到下一帧,这是最后一帧
|
|
cb(start - prefix, end - start + prefix, prefix);
|
|
break;
|
|
}
|
|
}
|
|
|
|
size_t prefixSize(const char *ptr, size_t len){
|
|
if (len < 4) {
|
|
return 0;
|
|
}
|
|
|
|
if (ptr[0] != 0x00 || ptr[1] != 0x00) {
|
|
//不是0x00 00开头
|
|
return 0;
|
|
}
|
|
|
|
if (ptr[2] == 0x00 && ptr[3] == 0x01) {
|
|
//是0x00 00 00 01
|
|
return 4;
|
|
}
|
|
|
|
if (ptr[2] == 0x01) {
|
|
//是0x00 00 01
|
|
return 3;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
Sdp::Ptr H264Track::getSdp() {
|
|
if(!ready()){
|
|
WarnL << getCodecName() << " Track未准备好";
|
|
return nullptr;
|
|
}
|
|
return std::make_shared<H264Sdp>(getSps(), getPps(), getBitRate() / 1024);
|
|
}
|
|
}//namespace mediakit
|
|
|
|
|