处理h264数据//处理到那算一帧数据呢?

//Extract the NAL data from the mdat box mdat int WvH264DataExtract::NAL_extract( BYTE * buf, int index, int mdat_size) { int nal_size = 0; int mdat_size_count = 8; while (mdat_size_count < mdat_size) { WriteBuffers(( BYTE *)hex_SCP, 4); nal_size = GetSize(buf + index, 4); WriteBuffers(buf + index + 4, nal_size); mdat_size_count += nal_size + 4; index += nal_size + 4; } return index; } //Extract the NAL data from the H.264 video data acquired from the camera UINT WvH264DataExtract::ExtractData() { int i; int read_size = 0; int buf_size = 0; int bufBack_size = 0; BYTE * buf = NULL; BYTE * bufBack = NULL; int mdat_size = 0; bool SPSPPSfound = false ; UINT ret = 0; printf ( "H.264 data extracting." ); while (stop == false ) { read_size = 0; buf_size = H264BUFFERSIZE; buf = ( BYTE *) malloc (buf_size); if (bufBack_size > 0) { memcpy (buf, bufBack, bufBack_size); free (bufBack); bufBack = NULL; } if (GetBuffers(buf + bufBack_size, buf_size - bufBack_size) == 0) { break ; } bufBack_size = 0; if (SPSPPSfound == false ) { //Extract the SPS and PPS hearders from the avcC box for (i = 0; i <= buf_size - 4; i++) { if (buf[i] == hex_avcC[0] && buf[i + 1] == hex_avcC[1] && buf[i + 2] == hex_avcC[2] && buf[i + 3] == hex_avcC[3]) { if ((read_size = SPSPPS_extract(buf, i - 4)) == -1) { ret = -1; goto EXIT; } SPSPPSfound = true ; break ; } } } //Search the mdat box and acquire the data size for (i = read_size; i <= buf_size - 4; i++) { if (buf[i] == hex_mdat[0] && buf[i + 1] == hex_mdat[1] && buf[i + 2] == hex_mdat[2] && buf[i + 3] == hex_mdat[3]) { mdat_size = GetSize(buf + i - 4, 4); read_size = i + 4; break ; } } if (i > buf_size - 4) { bufBack_size = 4; bufBack = ( BYTE *) malloc (bufBack_size); memcpy (bufBack, buf + (buf_size - 4), bufBack_size); continue ; } //Acquire the data as much as the mdat box size from the buffer if (buf_size - read_size >= mdat_size) { if (buf_size - read_size == mdat_size) { bufBack_size = 0; } else { bufBack_size = buf_size - (read_size + mdat_size); bufBack = ( BYTE *) malloc (bufBack_size); memcpy (bufBack, buf + read_size + mdat_size, bufBack_size); } } else { bufBack_size = buf_size - read_size; memmove (buf, buf + read_size, bufBack_size); buf_size = mdat_size; bufBack = ( BYTE *) realloc (buf, buf_size); buf = bufBack; bufBack = NULL; if (GetBuffers(buf + bufBack_size, buf_size - bufBack_size) == 0) { break ; } read_size = 0; bufBack_size = 0; } //Extract the NAL data from the mdat box read_size = NAL_extract(buf, read_size, mdat_size); free (buf); buf = NULL; printf ( "." ); } printf ( "\n" ); printf ( "H.264 data extracted.\n" ); EXIT: if (buf) { free (buf); buf = NULL; } if (bufBack) { free (bufBack); bufBack = NULL; } return ret; } #define SEARCH_SPSPPS 0 #define WRITE_SPS 1 #define WRITE_PPS 2 const BYTE hex_avcC[4] = {0x61, 0x76, 0x63, 0x43}; const BYTE hex_mdat[4] = {0x6D, 0x64, 0x61, 0x74}; const BYTE hex_SCP[4] = {0x00, 0x00, 0x00, 0x01}; WvH264DataExtract::WvH264DataExtract( void ) { this ->stop = false ; this ->mp4Buffer = NULL; this ->h264Buffer = NULL; this ->keyMP4 = NULL; this ->keyH264 = NULL; this -> thread = NULL; } WvH264DataExtract::~WvH264DataExtract( void ) { this ->mp4Buffer = NULL; this ->h264Buffer = NULL; this ->keyMP4 = NULL; this ->keyH264 = NULL; this -> thread = NULL; } static size_t hex2dec( char * hex) { size_t num = 0; int buf; int i; for (i = 0; ; i++) { if (hex[i] >= '0' && hex[i] <= '9' ) { buf = hex[i] - '0' ; } else if (hex[i] >= 'a' && hex[i] <= 'f' ) { buf = hex[i] - 'a' + 10; } else if (hex[i] >= 'A' && hex[i] <= 'F' ) { buf = hex[i] - 'A' + 10; } else if (hex[i] == '\0' ) { break ; } else { num = 0xFFFFFFFF; break ; } num *= 16; num += buf; if (i >= 8) { //Overflow num = 0xFFFFFFFF; break ; } } return (num); } static int GetSize( BYTE * buf, int size) { int i; int data_size = 0; BYTE size_hex_buf[4]; char size_hex[10]; char format[20] = "" ; if (size > 4) { return 0; } for (i = 0; i < size; i++) { size_hex_buf[i] = buf[i]; strcat (format, "%02x" ); } sprintf (size_hex, format, size_hex_buf[0], size_hex_buf[1], size_hex_buf[2], size_hex_buf[3]); data_size = hex2dec(size_hex); return data_size; } //Write the 1-byte data into the buffer void WvH264DataExtract::WriteBuffer( BYTE buf) { this ->keyH264->Lock(); this ->h264Buffer->push_back(buf); this ->keyH264->Unlock(); } //Write the data as much as the specified size into the buffer void WvH264DataExtract::WriteBuffers( BYTE * buf, int size) { if (size < 0) { return ; } this ->keyH264->Lock(); this ->h264Buffer->insert(h264Buffer->end(), buf, buf + size); this ->keyH264->Unlock(); } //Read the data as much as the specified size from the buffer int WvH264DataExtract::GetBuffers( BYTE * buf, int size) { int i; if (size < 0) { return 0; } while (stop == false ) { if (mp4Buffer->empty() == false && ( int )mp4Buffer->size() > size) { keyMP4->Lock(); for (i = 0; i < size; i++) { buf[i] = ( BYTE )mp4Buffer->front(); mp4Buffer->pop_front(); } keyMP4->Unlock(); return size; } else { ::Sleep(1); continue ; } } return 0; } //Extract the SPS and PPS headers from the avcC box int WvH264DataExtract::SPSPPS_extract( BYTE * buf, int index) { bool SPSflg = false ; bool PPSflg = false ; BYTE dat; int i; int cnt = 0; int condition = SEARCH_SPSPPS; int avcC_size = 0; int SPSPPS_size = 0; avcC_size = GetSize(buf + index, 4); for (i = 4; i < avcC_size; i++) { dat = buf[index + i]; switch (condition) { case SEARCH_SPSPPS: if (dat == 0x67) { WriteBuffers(( BYTE *)hex_SCP, 4); SPSPPS_size = GetSize(buf + (index + i - 2), 2); condition = WRITE_SPS; cnt = 0; i--; } else if (dat == 0x68) { WriteBuffers(( BYTE *)hex_SCP, 4); SPSPPS_size = GetSize(buf + (index + i - 2), 2); condition = WRITE_PPS; cnt = 0; i--; } break ; case WRITE_SPS: WriteBuffer(dat); cnt++; if (cnt >= SPSPPS_size) { SPSflg = true ; condition = SEARCH_SPSPPS; cnt = 0; } break ; case WRITE_PPS: WriteBuffer(dat); cnt++; if (cnt >= SPSPPS_size) { PPSflg = true ; condition = SEARCH_SPSPPS; cnt = 0; } break ; default : printf ( "error : writing SPS or PPS\n" ); return -1; } } if (!SPSflg | !PPSflg) { printf ( "error : not found SPS or PPS\n" ); return -1; } return index + i; } 分享到：

对我有用[0]  丢个板砖[0]  引用 |  举报 |  管理 回复次数：4

wer85121430 wer85121430 等级：