Go语言解码h264 SPS

import (
	"math"
)

//
func Ue(pBuff []byte, nLen uint, nStartBit *uint) uint {
	//计算0bit的个数
	var nZeroNum int = 0
	for *nStartBit < nLen*8 {
		if (pBuff[*nStartBit/8] & (0x80 >> (*nStartBit % 8))) > 0 {
			break
		}
		nZeroNum++
		*nStartBit++
	}
	*nStartBit++

	//计算结果
	var dwRet uint = 0
	for i := 0; i < nZeroNum; i++ {
		dwRet <<= 1
		if (pBuff[*nStartBit/8] & (0x80 >> (*nStartBit % 8))) > 0 {
			dwRet += 1
		}
		*nStartBit++
	}
	return (1 << nZeroNum) - 1 + dwRet
}

func Se(pBuff []byte, nLen uint, nStartBit *uint) int {
	UeVal := Ue(pBuff, nLen, nStartBit)
	var k int64 = (int64)(UeVal)
	var nValue int = (int)(math.Ceil((float64)(k / 2))) //ceil函数：ceil函数的作用是求不小于给定实数的最小整数。ceil(2)=ceil(1.2)=cei(1.5)=2.00
	if UeVal%2 == 0 {
		nValue = -nValue
	}
	return nValue
}

func u(BitCount uint, buf []byte, nStartBit *uint) uint {
	var dwRet uint = 0
	var i uint = 0
	for i = 0; i < BitCount; i++ {
		dwRet <<= 1
		if (buf[*nStartBit/8] & (0x80 >> (*nStartBit % 8))) > 0 {
			dwRet += 1
		}
		*nStartBit++
	}
	return dwRet
}

/**
 * H264的NAL起始码防竞争机制
 *
 * @param buf SPS数据内容
 *
 * @无返回值
 */
func de_emulation_prevention(buf []byte, buf_size *uint) {
	i := 0
	j := 0
	tmp_ptr := make([]byte, len(buf))
	var tmp_buf_size uint = 0
	var val int = 0

	tmp_ptr = buf
	tmp_buf_size = *buf_size
	for i = 0; i < (int)(tmp_buf_size-2); i++ {
		//check for 0x000003
		val = (int)((tmp_ptr[i] ^ 0x00) + (tmp_ptr[i+1] ^ 0x00) + (tmp_ptr[i+2] ^ 0x03))
		if val == 0 {
			//kick out 0x03
			for j = i + 2; j < (int)(tmp_buf_size-1); j++ {
				tmp_ptr[j] = tmp_ptr[j+1]
			}

			//and so we should devrease bufsize
			*buf_size--
		}
	}
}

/**
 * 解码SPS,获取视频图像宽、高和帧率信息
 *
 * @param buf SPS数据内容
 * @param nLen SPS数据的长度
 * @param width 图像宽度
 * @param height 图像高度
 * @成功则返回true , 失败则返回false
 */
func H264_decode_sps(buf []byte, nLen uint) (int, int, uint, bool) {
	var StartBit uint = 0
	var fps uint = 0
	de_emulation_prevention(buf, &nLen)

	u(1, buf, &StartBit) //forbidden_zero_bit :=
	u(2, buf, &StartBit) //nal_ref_idc :=
	nal_unit_type := u(5, buf, &StartBit)
	if nal_unit_type == 7 {
		profile_idc := u(8, buf, &StartBit) //
		_ = u(1, buf, &StartBit)            //(buf[1] & 0x80)>>7  constraint_set0_flag
		_ = u(1, buf, &StartBit)            //(buf[1] & 0x40)>>6;constraint_set1_flag
		_ = u(1, buf, &StartBit)            //(buf[1] & 0x20)>>5;constraint_set2_flag
		_ = u(1, buf, &StartBit)            //(buf[1] & 0x10)>>4;constraint_set3_flag
		_ = u(4, buf, &StartBit)            //reserved_zero_4bits
		u(8, buf, &StartBit)                //level_idc :=

		Ue(buf, nLen, &StartBit) //seq_parameter_set_id :=

		if profile_idc == 100 || profile_idc == 110 || profile_idc == 122 || profile_idc == 144 {
			chroma_format_idc := Ue(buf, nLen, &StartBit)
			if chroma_format_idc == 3 {
				u(1, buf, &StartBit) //residual_colour_transform_flag :=
			}

			Ue(buf, nLen, &StartBit) //bit_depth_luma_minus8 :=
			Ue(buf, nLen, &StartBit) //bit_depth_chroma_minus8 :=
			u(1, buf, &StartBit)     //qpprime_y_zero_transform_bypass_flag :=
			seq_scaling_matrix_present_flag := u(1, buf, &StartBit)

			seq_scaling_list_present_flag := make([]int, 8)
			if seq_scaling_matrix_present_flag > 0 {
				for i := 0; i < 8; i++ {
					seq_scaling_list_present_flag[i] = (int)(u(1, buf, &StartBit))
				}
			}
		}
		Ue(buf, nLen, &StartBit) //log2_max_frame_num_minus4 :=
		pic_order_cnt_type := Ue(buf, nLen, &StartBit)
		if pic_order_cnt_type == 0 {
			Ue(buf, nLen, &StartBit) //log2_max_pic_order_cnt_lsb_minus4 :=
		} else if pic_order_cnt_type == 1 {
			u(1, buf, &StartBit)     //delta_pic_order_always_zero_flag :=
			Se(buf, nLen, &StartBit) //offset_for_non_ref_pic :=
			Se(buf, nLen, &StartBit) //offset_for_top_to_bottom_field :=
			num_ref_frames_in_pic_order_cnt_cycle := Ue(buf, nLen, &StartBit)

			offset_for_ref_frame := make([]int, num_ref_frames_in_pic_order_cnt_cycle)
			for i := 0; i < (int)(num_ref_frames_in_pic_order_cnt_cycle); i++ {
				offset_for_ref_frame[i] = Se(buf, nLen, &StartBit)
			}
		}
		Ue(buf, nLen, &StartBit) //num_ref_frames :=
		u(1, buf, &StartBit)     //gaps_in_frame_num_value_allowed_flag :=
		pic_width_in_mbs_minus1 := Ue(buf, nLen, &StartBit)
		pic_height_in_map_units_minus1 := Ue(buf, nLen, &StartBit)

		width := (pic_width_in_mbs_minus1 + 1) * 16
		height := (pic_height_in_map_units_minus1 + 1) * 16

		frame_mbs_only_flag := u(1, buf, &StartBit)
		if frame_mbs_only_flag <= 0 {
			u(1, buf, &StartBit) //mb_adaptive_frame_field_flag :=
		}

		u(1, buf, &StartBit) //direct_8x8_inference_flag :=
		frame_cropping_flag := u(1, buf, &StartBit)
		if frame_cropping_flag > 0 {
			Ue(buf, nLen, &StartBit) //frame_crop_left_offset:=
			Ue(buf, nLen, &StartBit) //frame_crop_right_offset:=
			Ue(buf, nLen, &StartBit) //frame_crop_top_offset:=
			Ue(buf, nLen, &StartBit) //frame_crop_bottom_offset:=
		}
		vui_parameter_present_flag := u(1, buf, &StartBit)
		if vui_parameter_present_flag > 0 {
			aspect_ratio_info_present_flag := u(1, buf, &StartBit)
			if aspect_ratio_info_present_flag > 0 {
				aspect_ratio_idc := u(8, buf, &StartBit)
				if aspect_ratio_idc == 255 {
					u(16, buf, &StartBit) //sar_width:=
					u(16, buf, &StartBit) //sar_height:=
				}
			}
			overscan_info_present_flag := u(1, buf, &StartBit)
			if overscan_info_present_flag > 0 {
				u(1, buf, &StartBit) //overscan_appropriate_flagu:=
			}
			video_signal_type_present_flag := u(1, buf, &StartBit)
			if video_signal_type_present_flag > 0 {
				u(3, buf, &StartBit) //video_format:=
				u(1, buf, &StartBit) //video_full_range_flag:=
				colour_description_present_flag := u(1, buf, &StartBit)
				if colour_description_present_flag > 0 {
					u(8, buf, &StartBit) //colour_primaries:=
					u(8, buf, &StartBit) //transfer_characteristics:=
					u(8, buf, &StartBit) //matrix_coefficients:=
				}
			}
			chroma_loc_info_present_flag := u(1, buf, &StartBit)
			if chroma_loc_info_present_flag > 0 {
				Ue(buf, nLen, &StartBit) //chroma_sample_loc_type_top_field:=
				Ue(buf, nLen, &StartBit) //chroma_sample_loc_type_bottom_field:=
			}
			timing_info_present_flag := u(1, buf, &StartBit)

			if timing_info_present_flag > 0 {
				num_units_in_tick := u(32, buf, &StartBit)
				time_scale := u(32, buf, &StartBit)
				fps = time_scale / num_units_in_tick
				fixed_frame_rate_flag := u(1, buf, &StartBit)
				if fixed_frame_rate_flag > 0 {
					fps = fps / 2
				}
			}
		}
		return int(width), int(height), fps, true
	} else {
		return 0, 0, 0, false
	}
}