2023面试高手

这个屌丝很懒，什么也没留下！

热门标签

java ps h264_关于对H264码流的PS的封装的相关代码实现

作者：2023面试高手 | 2024-04-09 10:17:54

踩

java ps流封装

1、写在开始之前：

最近因为新工作要维护别人留下的GB模块代码，先熟悉了流程，然后也试着封装了下ps流，结果也能通过测试正常预览了，当然，其中开发读文档的头疼，预览花屏，卡帧的事情都有遇到，当时慢慢的看文档，整理逻辑，也就都顺利解决了，下面把大致的一些流程代码贴出来分享下。既然是对接国标，自然少不了通读它的标准文档和相关的RFC文档了！具体的我就不说了，可以用百度google下的。

注意：因为是GB要求ps封装后再加上rtp头的格式来的，所以下面代码中我也加上了rtp头，如果不需要的话，直接屏蔽代码中的rtp即可。

2、封装的重点

当我们从读缓冲区中取得一帧音视频数据的时候，封装时其实每一帧数据有且只有一个ps头和psm头，如果是I帧的话，就还多一个system头，一个或者多个pes头和rtp头，

像如果帧数据过长的话，就得进行分片，每片都会包含一个pes头，rtp负载最好长度1460，所以会进行再分包操作！所以每一个包数据至少一个rtp+databuf，每一片数据，至少有个rtp+pes+databuf，每一帧数据至少有rtp+ps+psm+pes+databuf(关键帧的话：多一个system头)

3、具体的各个封装的代码实现

首先给去一个整体的封装rtp->ps->sys->psm->pes(如果只要ps的话，则为ps->sys->psm->pes)的大致流程，

然后再一一罗列出各个部件的封装接口

/***

*@remark: 音视频数据的打包成ps流，并封装成rtp

*@param : pData [in] 需要发送的音视频数据

* nFrameLen [in] 发送数据的长度

* pPacker [in] 数据包的一些信息，包括时间戳，rtp数据buff，发送的socket相关信息

* stream_type[in] 数据类型 0 视频 1 音频

*@return: 0 success others failed

int gb28181_streampackageForH264(char *pData, int nFrameLen, Data_Info_s* pPacker, int stream_type)

{

char szTempPacketHead[256];

int nSizePos = 0;

int nSize = 0;

char *pBuff = NULL;

memset(szTempPacketHead, 0, 256);

// 1 package for ps header

gb28181_make_ps_header(szTempPacketHead + nSizePos, pPacker->s64CurPts);

nSizePos += PS_HDR_LEN;

//2 system header

if( pPacker->IFrame == 1 )

{

// 如果是I帧的话，则添加系统头

gb28181_make_sys_header(szTempPacketHead + nSizePos);

nSizePos += SYS_HDR_LEN;

//这个地方我是不管是I帧还是p帧都加上了map的，貌似只是I帧加也没有问题

// gb28181_make_psm_header(szTempPacketHead + nSizePos);

// nSizePos += PSM_HDR_LEN;

}

// psm头 (也是map)

gb28181_make_psm_header(szTempPacketHead + nSizePos);

nSizePos += PSM_HDR_LEN;

//加上rtp发送出去，这样的话，后面的数据就只要分片分包就只有加上pes头和rtp头了

if(gb28181_send_rtp_pack(szTempPacketHead, nSizePos, 0, pPacker) != 0 )

return -1;

// 这里向后移动是为了方便拷贝pes头

//这里是为了减少后面音视频裸数据的大量拷贝浪费空间，所以这里就向后移动，在实际处理的时候，要注意地址是否越界以及覆盖等问题

pBuff = pData - PES_HDR_LEN;

while(nFrameLen > 0)

{

//每次帧的长度不要超过short类型，过了就得分片进循环行发送

nSize = (nFrameLen > PS_PES_PAYLOAD_SIZE) ? PS_PES_PAYLOAD_SIZE : nFrameLen;

// 添加pes头

gb28181_make_pes_header(pBuff, stream_type ? 0xC0:0xE0, nSize, (pPacker->s64CurPts / 100), (pPacker->s64CurPts/300));

//最后在添加rtp头并发送数据

if( gb28181_send_rtp_pack(pBuff, nSize + PES_HDR_LEN, ((nSize == nFrameLen)?1:0), pPacker) != 0 )

{

printf("gb28181_send_pack failed!\n");

return -1;

}

//分片后每次发送的数据移动指针操作

nFrameLen -= nSize;

//这里也只移动nSize,因为在while向后移动的pes头长度，正好重新填充pes头数据

pBuff += nSize;

}

return 0;

}

上面列出来了整个打包发包的过程，接下来一个一个接口的看

/***

*@remark: ps头的封装,里面的具体数据的填写已经占位，可以参考标准

*@param : pData [in] 填充ps头数据的地址

* s64Src [in] 时间戳

*@return: 0 success, others failed

int gb28181_make_ps_header(char *pData, unsigned long long s64Scr)

{

unsigned long long lScrExt = (s64Scr) % 100;

s64Scr = s64Scr / 100;

// 这里除以100是由于sdp协议返回的video的频率是90000，帧率是25帧/s，所以每次递增的量是3600,

// 所以实际你应该根据你自己编码里的时间戳来处理以保证时间戳的增量为3600即可，

//如果这里不对的话，就可能导致卡顿现象了

bits_buffer_s bitsBuffer;

bitsBuffer.i_size = PS_HDR_LEN;

bitsBuffer.i_data = 0;

bitsBuffer.i_mask = 0x80; // 二进制：10000000 这里是为了后面对一个字节的每一位进行操作，避免大小端夸字节字序错乱

bitsBuffer.p_data = (unsigned char *)(pData);

memset(bitsBuffer.p_data, 0, PS_HDR_LEN);

bits_write(&bitsBuffer, 32, 0x000001BA); /*start codes*/

bits_write(&bitsBuffer, 2, 1); /*marker bits '01b'*/

bits_write(&bitsBuffer, 3, (s64Scr>>30)&0x07); /*System clock [32..30]*/

bits_write(&bitsBuffer, 1, 1); /*marker bit*/

bits_write(&bitsBuffer, 15, (s64Scr>>15)&0x7FFF); /*System clock [29..15]*/

bits_write(&bitsBuffer, 1, 1); /*marker bit*/

bits_write(&bitsBuffer, 15, s64Scr&0x7fff); /*System clock [29..15]*/

bits_write(&bitsBuffer, 1, 1); /*marker bit*/

bits_write(&bitsBuffer, 9, lScrExt&0x01ff); /*System clock [14..0]*/

bits_write(&bitsBuffer, 1, 1); /*marker bit*/

bits_write(&bitsBuffer, 22, (255)&0x3fffff); /*bit rate(n units of 50 bytes per second.)*/

bits_write(&bitsBuffer, 2, 3); /*marker bits '11'*/

bits_write(&bitsBuffer, 5, 0x1f); /*reserved(reserved for future use)*/

bits_write(&bitsBuffer, 3, 0); /*stuffing length*/

return 0;

}

/***

*@remark: sys头的封装,里面的具体数据的填写已经占位，可以参考标准

*@param : pData [in] 填充ps头数据的地址

*@return: 0 success, others failed

int gb28181_make_sys_header(char *pData)

{

bits_buffer_s bitsBuffer;

bitsBuffer.i_size = SYS_HDR_LEN;

bitsBuffer.i_data = 0;

bitsBuffer.i_mask = 0x80;

bitsBuffer.p_data = (unsigned char *)(pData);

memset(bitsBuffer.p_data, 0, SYS_HDR_LEN);

/*system header*/

bits_write( &bitsBuffer, 32, 0x000001BB); /*start code*/

bits_write( &bitsBuffer, 16, SYS_HDR_LEN-6);/*header_length 表示次字节后面的长度，后面的相关头也是次意思*/

bits_write( &bitsBuffer, 1, 1); /*marker_bit*/

bits_write( &bitsBuffer, 22, 50000); /*rate_bound*/

bits_write( &bitsBuffer, 1, 1); /*marker_bit*/

bits_write( &bitsBuffer, 6, 1); /*audio_bound*/

bits_write( &bitsBuffer, 1, 0); /*fixed_flag */

bits_write( &bitsBuffer, 1, 1); /*CSPS_flag */

bits_write( &bitsBuffer, 1, 1); /*system_audio_lock_flag*/

bits_write( &bitsBuffer, 1, 1); /*system_video_lock_flag*/

bits_write( &bitsBuffer, 1, 1); /*marker_bit*/

bits_write( &bitsBuffer, 5, 1); /*video_bound*/

bits_write( &bitsBuffer, 1, 0); /*dif from mpeg1*/

bits_write( &bitsBuffer, 7, 0x7F); /*reserver*/

/*audio stream bound*/

bits_write( &bitsBuffer, 8, 0xC0); /*stream_id*/

bits_write( &bitsBuffer, 2, 3); /*marker_bit */

bits_write( &bitsBuffer, 1, 0); /*PSTD_buffer_bound_scale*/

bits_write( &bitsBuffer, 13, 512); /*PSTD_buffer_size_bound*/

/*video stream bound*/

bits_write( &bitsBuffer, 8, 0xE0); /*stream_id*/

bits_write( &bitsBuffer, 2, 3); /*marker_bit */

bits_write( &bitsBuffer, 1, 1); /*PSTD_buffer_bound_scale*/

bits_write( &bitsBuffer, 13, 2048); /*PSTD_buffer_size_bound*/

return 0;

}

/***

*@remark: psm头的封装,里面的具体数据的填写已经占位，可以参考标准

*@param : pData [in] 填充ps头数据的地址

*@return: 0 success, others failed

int gb28181_make_psm_header(char *pData)

{

bits_buffer_s bitsBuffer;

bitsBuffer.i_size = PSM_HDR_LEN;

bitsBuffer.i_data = 0;

bitsBuffer.i_mask = 0x80;

bitsBuffer.p_data = (unsigned char *)(pData);

memset(bitsBuffer.p_data, 0, PS_SYS_MAP_SIZE);

bits_write(&bitsBuffer, 24,0x000001); /*start code*/

bits_write(&bitsBuffer, 8, 0xBC); /*map stream id*/

bits_write(&bitsBuffer, 16,18); /*program stream map length*/

bits_write(&bitsBuffer, 1, 1); /*current next indicator */

bits_write(&bitsBuffer, 2, 3); /*reserved*/

bits_write(&bitsBuffer, 5, 0); /*program stream map version*/

bits_write(&bitsBuffer, 7, 0x7F); /*reserved */

bits_write(&bitsBuffer, 1, 1); /*marker bit */

bits_write(&bitsBuffer, 16,0); /*programe stream info length*/

bits_write(&bitsBuffer, 16, 8); /*elementary stream map length is*/

/*audio*/

bits_write(&bitsBuffer, 8, 0x90); /*stream_type*/

bits_write(&bitsBuffer, 8, 0xC0); /*elementary_stream_id*/

bits_write(&bitsBuffer, 16, 0); /*elementary_stream_info_length is*/

/*video*/

bits_write(&bitsBuffer, 8, 0x1B); /*stream_type*/

bits_write(&bitsBuffer, 8, 0xE0); /*elementary_stream_id*/

bits_write(&bitsBuffer, 16, 0); /*elementary_stream_info_length */

/*crc (2e b9 0f 3d)*/

bits_write(&bitsBuffer, 8, 0x45); /*crc (24~31) bits*/

bits_write(&bitsBuffer, 8, 0xBD); /*crc (16~23) bits*/

bits_write(&bitsBuffer, 8, 0xDC); /*crc (8~15) bits*/

bits_write(&bitsBuffer, 8, 0xF4); /*crc (0~7) bits*/

return 0;

}

/***

*@remark: pes头的封装,里面的具体数据的填写已经占位，可以参考标准

*@param : pData [in] 填充ps头数据的地址

* stream_id [in] 码流类型

* paylaod_len[in] 负载长度

* pts [in] 时间戳

* dts [in]

*@return: 0 success, others failed

int gb28181_make_pes_header(char *pData, int stream_id, int payload_len, unsigned long long pts, unsigned long long dts)

{

bits_buffer_s bitsBuffer;

bitsBuffer.i_size = PES_HDR_LEN;

bitsBuffer.i_data = 0;

bitsBuffer.i_mask = 0x80;

bitsBuffer.p_data = (unsigned char *)(pData);

memset(bitsBuffer.p_data, 0, PES_HDR_LEN);

/*system header*/

bits_write( &bitsBuffer, 24,0x000001); /*start code*/

bits_write( &bitsBuffer, 8, (stream_id)); /*streamID*/

bits_write( &bitsBuffer, 16,(payload_len)+13); /*packet_len*/ //指出pes分组中数据长度和该字节后的长度和

bits_write( &bitsBuffer, 2, 2 ); /*'10'*/

bits_write( &bitsBuffer, 2, 0 ); /*scrambling_control*/

bits_write( &bitsBuffer, 1, 0 ); /*priority*/

bits_write( &bitsBuffer, 1, 0 ); /*data_alignment_indicator*/

bits_write( &bitsBuffer, 1, 0 ); /*copyright*/

bits_write( &bitsBuffer, 1, 0 ); /*original_or_copy*/

bits_write( &bitsBuffer, 1, 1 ); /*PTS_flag*/

bits_write( &bitsBuffer, 1, 1 ); /*DTS_flag*/

bits_write( &bitsBuffer, 1, 0 ); /*ESCR_flag*/

bits_write( &bitsBuffer, 1, 0 ); /*ES_rate_flag*/

bits_write( &bitsBuffer, 1, 0 ); /*DSM_trick_mode_flag*/

bits_write( &bitsBuffer, 1, 0 ); /*additional_copy_info_flag*/

bits_write( &bitsBuffer, 1, 0 ); /*PES_CRC_flag*/

bits_write( &bitsBuffer, 1, 0 ); /*PES_extension_flag*/

bits_write( &bitsBuffer, 8, 10); /*header_data_length*/

// 指出包含在 PES 分组标题中的可选字段和任何填充字节所占用的总字节数。该字段之前

//的字节指出了有无可选字段。

/*PTS,DTS*/

bits_write( &bitsBuffer, 4, 3 ); /*'0011'*/

bits_write( &bitsBuffer, 3, ((pts)>>30)&0x07 ); /*PTS[32..30]*/