MPEG2/TS 获取PMT PID

一个简单的从指定MPEG2/TS文件中获取PMT PID的函数，

注意代码中的注释。

 

#define MAKE_PID(p) (((p[0]<<8) + p[1]) & 0x1fff)

[html]  view plain  copy

1. ushort GetPmtPid(const char* filename)  
2. {  
3. #define TSPACKET_SIZE     188    
4.     ushort pid;  
5.     uchar buf[TSPACKET_SIZE];  
6.     uchar* p = buf;  
7.     FILE*  fp;  
8.   
9.     fp = fopen(filename, "rb");  
10.     if (fp == NULL) {  
11.         fprintf(stderr, "Failed to open the file %s: %s\n", filename, strerror(errno));  
12.         return 0;  
13.     }  
14.   
15.     while (fread(buf, 1, TSPACKET_SIZE, fp) == TSPACKET_SIZE) {  
16.         if (buf[0] != MPEG_TS_SYNC_BYTE) {  
17.             fprintf(stderr, "No sync byte where one was expected!\n");  
18.             break;  
19.         }  
20.   
21.         p = buf;  
22.         pid = MAKE_PID((p+1));  
23.         if ( pid != 0 ) {  
24.             // 不是PAT包  
25.             continue;  
26.         }  
27.   
28.         // Skip the TS header  
29.         p += MPEG_TS_HEADER_SIZE; //@MPEG_TS_HEADER_SIZE = 4  
30.   
31.         // Adaptation field exist or not  
32.         if ((buf[3] >> 4 & 0x03) & 0x2) {  
33.             fprintf(stderr, "Adaptation field exist\n");  
34.             p += 1 + p[0];  // p[0]是Adaptation field长度, 1为长度域  
35.         }  
36.   
37.         // p[0] 为 Point field长度， 1为Point field  
38.         p += 1 + p[0];  
39.   
40.         // 8 为PAT表头长度，如果定义了表头，则用sizeof运算符  
41.         p += 8;  
42.   
43.         // NIT Program exist??  
44.         if( *((ushort*)p) == 0) {  
45.             p += 4; // PAT的节目结构为4字节  
46.         }  
47.   
48.         p += 2; // 节目号为两个字节  
49.   
50.         fclose(fp);  
51.         return MAKE_PID(p);  
52.     }  
53.   
54.     fclose(fp);  
55.     return 0;  
56. }