开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 多媒体 > 流媒体/Mpeg4/MP4 > 查看源码
mpeg3video.c
资源名称：NETVIDEO.rar　[点击查看]
上传用户：sun1608
上传日期：2007-02-02
资源大小：6116k
文件大小：16k
源码类别：

流媒体/Mpeg4/MP4

开发平台：

Visual C++

mpeg3video.c：源码内容
  1. #include "../libmpeg3.h"
  2. #include "../mpeg3private.h"
  3. #include "../mpeg3protos.h"
  4. #include "mpeg3video.h"
  5. #include "mpeg3videoprotos.h"
  6. #ifndef SDL_THREADS
  7. #include <pthread.h>
  8. #else
  9. #include "SDL.h"
  10. #include "SDL_thread.h"
  11. #endif
  12. #include <stdlib.h>
  14. // "脚悼绕" <doogle@shinbiro.com>
  16. unsigned char mpeg3_zig_zag_scan_mmx[64] =
  17. {
  18.     0*8+0 /* 0*/, 1*8+0 /* 1*/, 0*8+1 /* 8*/, 0*8+2 /*16*/, 1*8+1 /* 9*/, 2*8+0 /* 2*/, 3*8+0 /* 3*/, 2*8+1 /*10*/,
  19.     1*8+2 /*17*/, 0*8+3 /*24*/, 0*8+4 /*32*/, 1*8+3 /*25*/, 2*8+2 /*18*/, 3*8+1 /*11*/, 4*8+0 /* 4*/, 5*8+0 /* 5*/,
  20.     4*8+1 /*12*/, 3*8+2 /*19*/, 2*8+3 /*26*/, 1*8+4 /*33*/, 0*8+5 /*40*/, 0*8+6 /*48*/, 1*8+5 /*41*/, 2*8+4 /*34*/,
  21.     3*8+3 /*27*/, 4*8+2 /*20*/, 5*8+1 /*13*/, 6*8+0 /* 6*/, 7*8+0 /* 7*/, 6*8+1 /*14*/, 5*8+2 /*21*/, 4*8+3 /*28*/,
  22.     3*8+4 /*35*/, 2*8+5 /*42*/, 1*8+6 /*49*/, 0*8+7 /*56*/, 1*8+7 /*57*/, 2*8+6 /*50*/, 3*8+5 /*43*/, 4*8+4 /*36*/,
  23.     5*8+3 /*29*/, 6*8+2 /*22*/, 7*8+1 /*15*/, 7*8+2 /*23*/, 6*8+3 /*30*/, 5*8+4 /*37*/, 4*8+5 /*44*/, 3*8+6 /*51*/,
  24.     2*8+7 /*58*/, 3*8+7 /*59*/, 4*8+6 /*52*/, 5*8+5 /*45*/, 6*8+4 /*38*/, 7*8+3 /*31*/, 7*8+4 /*39*/, 6*8+5 /*46*/,
  25.     5*8+6 /*53*/, 4*8+7 /*60*/, 5*8+7 /*61*/, 6*8+6 /*54*/, 7*8+5 /*47*/, 7*8+6 /*55*/, 6*8+7 /*62*/, 7*8+7 /*63*/
  26. };
  28. /* alternate scan */
  29. unsigned char mpeg3_alternate_scan_mmx[64] =
  30. {
  31.      0*8+0 /*0 */, 0*8+1 /* 8*/, 0*8+2 /*16*/, 0*8+3 /*24*/, 1*8+0 /* 1*/, 1*8+1 /* 9*/, 2*8+0 /* 2*/, 2*8+1 /*10*/,
  32.      1*8+2 /*17*/, 1*8+3 /*25*/, 0*8+4 /*32*/, 0*8+5 /*40*/, 0*8+6 /*48*/, 0*8+7 /*56*/, 1*8+7 /*57*/, 1*8+6 /*49*/,
  33.      1*8+5 /*41*/, 1*8+4 /*33*/, 2*8+3 /*26*/, 2*8+2 /*18*/, 3*8+0 /* 3*/, 3*8+1 /*11*/, 4*8+0 /* 4*/, 4*8+1 /*12*/,
  34.      3*8+2 /*19*/, 3*8+3 /*27*/, 2*8+4 /*34*/, 2*8+5 /*42*/, 2*8+6 /*50*/, 2*8+7 /*58*/, 3*8+4 /*35*/, 3*8+5 /*43*/,
  35.      3*8+6 /*51*/, 3*8+7 /*59*/, 4*8+2 /*20*/, 4*8+3 /*28*/, 5*8+0 /* 5*/, 5*8+1 /*13*/, 6*8+0 /* 6*/, 6*8+1 /*14*/,
  36.      5*8+2 /*21*/, 5*8+3 /*29*/, 4*8+4 /*36*/, 4*8+5 /*44*/, 4*8+6 /*52*/, 4*8+7 /*60*/, 5*8+4 /*37*/, 5*8+5 /*45*/,
  37.      5*8+6 /*53*/, 5*8+7 /*61*/, 6*8+2 /*22*/, 6*8+3 /*30*/, 7*8+0 /* 7*/, 7*8+1 /*15*/, 7*8+2 /*23*/, 7*8+3 /*31*/,
  38.      6*8+4 /*38*/, 6*8+5 /*46*/, 6*8+6 /*54*/, 6*8+7 /*62*/, 7*8+4 /*39*/, 7*8+5 /*47*/, 7*8+6 /*55*/, 7*8+7 /*63*/
  39. };
  42. /* zig-zag scan */
  43. unsigned char mpeg3_zig_zag_scan_nommx[64] =
  44. {
  45.   0, 1, 8, 16, 9, 2, 3, 10, 17, 24, 32, 25, 18, 11, 4, 5, 
  46.   12, 19, 26, 33, 40, 48, 41, 34, 27, 20, 13, 6, 7, 14, 21, 28, 
  47.   35, 42, 49, 56, 57, 50, 43, 36, 29, 22, 15, 23, 30, 37, 44, 51, 
  48.   58, 59, 52, 45, 38, 31, 39, 46, 53, 60, 61, 54, 47, 55, 62, 63
  49. };
  51. /* alternate scan */
  52. unsigned char mpeg3_alternate_scan_nommx[64] =
  53. {
  54.   0, 8, 16, 24, 1, 9, 2, 10, 17, 25, 32, 40, 48, 56, 57, 49, 
  55.   41, 33, 26, 18, 3, 11, 4, 12, 19, 27, 34, 42, 50, 58, 35, 43, 
  56.   51, 59, 20, 28, 5, 13, 6, 14, 21, 29, 36, 44, 52, 60, 37, 45, 
  57.   53, 61, 22, 30, 7, 15, 23, 31, 38, 46, 54, 62, 39, 47, 55, 63
  58. };
  60. /* default intra quantization matrix */
  61. unsigned char mpeg3_default_intra_quantizer_matrix[64] =
  62. {
  63.   8, 16, 19, 22, 26, 27, 29, 34,
  64.   16, 16, 22, 24, 27, 29, 34, 37,
  65.   19, 22, 26, 27, 29, 34, 34, 38,
  66.   22, 22, 26, 27, 29, 34, 37, 40,
  67.   22, 26, 27, 29, 32, 35, 40, 48,
  68.   26, 27, 29, 32, 35, 40, 48, 58,
  69.   26, 27, 29, 34, 38, 46, 56, 69,
  70.   27, 29, 35, 38, 46, 56, 69, 83
  71. };
  73. unsigned char mpeg3_non_linear_mquant_table[32] = 
  74. {
  75.    0, 1, 2, 3, 4, 5, 6, 7,
  76.    8, 10, 12, 14, 16, 18, 20, 22, 
  77.   24, 28, 32, 36, 40, 44, 48, 52, 
  78.   56, 64, 72, 80, 88, 96, 104, 112
  79. };
  81. double mpeg3_frame_rate_table[16] =
  82. {
  83.   0.0,   /* Pad */
  84.   24000.0/1001.0,       /* Official frame rates */
  85.   24.0,
  86.   25.0,
  87.   30000.0/1001.0,
  88.   30.0,
  89.   50.0,
  90.   ((60.0*1000.0)/1001.0),
  91.   60.0,
  93.   1,                    /* Unofficial economy rates */
  94.   5, 
  95.   10,
  96.   12,
  97.   15,
  98.   0,
  99.   0,
  100. };
  102. int mpeg3video_initdecoder(mpeg3video_t *video)
  103. {
  104. int blk_cnt_tab[3] = {6, 8, 12};
  105. int cc;
  106.    int i;
  107. long size[4], padding[2];         /* Size of Y, U, and V buffers */
  109. if(!video->mpeg2)
  110. {
  111. /* force MPEG-1 parameters */
  112.      video->prog_seq = 1;
  113.      video->prog_frame = 1;
  114.      video->pict_struct = FRAME_PICTURE;
  115.      video->frame_pred_dct = 1;
  116.      video->chroma_format = CHROMA420;
  117.      video->matrix_coefficients = 5;
  118. }
  120. /* Get dimensions rounded to nearest multiple of coded macroblocks */
  121. video->mb_width = (video->horizontal_size + 15) / 16;
  122. video->mb_height = (video->mpeg2 && !video->prog_seq) ? 
  123. (2 * ((video->vertical_size + 31) / 32)) : 
  124. ((video->vertical_size + 15) / 16);
  125. video->coded_picture_width = 16 * video->mb_width;
  126. video->coded_picture_height = 16 * video->mb_height;
  127. video->chrom_width = (video->chroma_format == CHROMA444) ? 
  128. video->coded_picture_width : 
  129. (video->coded_picture_width >> 1);
  130. video->chrom_height = (video->chroma_format != CHROMA420) ? 
  131. video->coded_picture_height : 
  132.                     (video->coded_picture_height >> 1);
  133. video->blk_cnt = blk_cnt_tab[video->chroma_format - 1];
  135. /* Get sizes of YUV buffers */
  136. padding[0] = 16 * video->coded_picture_width;
  137. size[0] = video->coded_picture_width * video->coded_picture_height + padding[0] * 2;
  139. padding[1] = 16 * video->chrom_width;
  140. size[1] = video->chrom_width * video->chrom_height + 2 * padding[1];
  142. size[2] = (video->llw * video->llh);
  143. size[3] = (video->llw * video->llh) / 4;
  145. /* Allocate contiguous fragments for YUV buffers for hardware YUV decoding */
  146. video->yuv_buffer[0] = (unsigned char*)calloc(1, (size[0] + padding[0]) + 2 * (size[1] + padding[1]));
  147. video->yuv_buffer[1] = (unsigned char*)calloc(1, (size[0] + padding[0]) + 2 * (size[1] + padding[1]));
  148. video->yuv_buffer[2] = (unsigned char*)calloc(1, (size[0] + padding[0]) + 2 * (size[1] + padding[1]));
  150.     if(video->scalable_mode == SC_SPAT)
  151. {
  152. video->yuv_buffer[3] = (unsigned char*)calloc(1, size[2] + 2 * size[3]);
  153. video->yuv_buffer[4] = (unsigned char*)calloc(1, size[2] + 2 * size[3]);
  154. }
  156. /* Direct pointers to areas of contiguous fragments in YVU order per Microsoft */
  157. for(cc = 0; cc < 3; cc++)
  158. {
  159. video->llframe0[cc] = 0;
  160. video->llframe1[cc] = 0;
  161. video->newframe[cc] = 0;
  162. }
  164. video->refframe[0]    = video->yuv_buffer[0];
  165. video->oldrefframe[0] = video->yuv_buffer[1];
  166. video->auxframe[0]    = video->yuv_buffer[2];
  167. video->refframe[2]    = video->yuv_buffer[0] + size[0] + padding[0];
  168. video->oldrefframe[2] = video->yuv_buffer[1] + size[0] + padding[0];
  169. video->auxframe[2]    = video->yuv_buffer[2] + size[0] + padding[0];
  170. video->refframe[1]    = video->yuv_buffer[0] + size[0] + padding[0] + size[1] + padding[1];
  171. video->oldrefframe[1] = video->yuv_buffer[1] + size[0] + padding[0] + size[1] + padding[1];
  172. video->auxframe[1]    = video->yuv_buffer[2] + size[0] + padding[0] + size[1] + padding[1];
  174.     if(video->scalable_mode == SC_SPAT)
  175. {
  176. /* this assumes lower layer is 4:2:0 */
  177. video->llframe0[0] = video->yuv_buffer[3] + padding[0]     ;
  178. video->llframe1[0] = video->yuv_buffer[4] + padding[0]     ;
  179. video->llframe0[2] = video->yuv_buffer[3] + padding[1] + size[2]    ;
  180. video->llframe1[2] = video->yuv_buffer[4] + padding[1] + size[2]    ;
  181. video->llframe0[1] = video->yuv_buffer[3] + padding[1] + size[2] + size[3];
  182. video->llframe1[1] = video->yuv_buffer[4] + padding[1] + size[2] + size[3];
  183.     }
  185. /* Initialize the YUV tables for software YUV decoding */
  186. video->cr_to_r = malloc(sizeof(long) * 256);
  187. video->cr_to_g = malloc(sizeof(long) * 256);
  188. video->cb_to_g = malloc(sizeof(long) * 256);
  189. video->cb_to_b = malloc(sizeof(long) * 256);
  190. video->cr_to_r_ptr = video->cr_to_r + 128;
  191. video->cr_to_g_ptr = video->cr_to_g + 128;
  192. video->cb_to_g_ptr = video->cb_to_g + 128;
  193. video->cb_to_b_ptr = video->cb_to_b + 128;
  195. for(i = -128; i < 128; i++)
  196. {
  197. video->cr_to_r_ptr[i] = (long)( 1.371 * 65536 * i);
  198. video->cr_to_g_ptr[i] = (long)(-0.698 * 65536 * i);
  199. video->cb_to_g_ptr[i] = (long)(-0.336 * 65536 * i);
  200. video->cb_to_b_ptr[i] = (long)( 1.732 * 65536 * i);
  201. }
  203. return 0;
  204. }
  206. int mpeg3video_deletedecoder(mpeg3video_t *video)
  207. {
  208. int i, padding;
  210. free(video->yuv_buffer[0]);
  211. free(video->yuv_buffer[1]);
  212. free(video->yuv_buffer[2]);
  214. if(video->llframe0[0])
  215. {
  216. free(video->yuv_buffer[3]);
  217. free(video->yuv_buffer[4]);
  218. }
  220. free(video->cr_to_r);
  221. free(video->cr_to_g);
  222. free(video->cb_to_g);
  223. free(video->cb_to_b);
  224. return 0;
  225. }
  227. void mpeg3video_init_scantables(mpeg3video_t *video)
  228. {
  229. #ifdef HAVE_MMX
  230. if(video->have_mmx)
  231. {
  232. video->mpeg3_zigzag_scan_table = mpeg3_zig_zag_scan_mmx;
  233. video->mpeg3_alternate_scan_table = mpeg3_alternate_scan_mmx;
  234. }
  235. else
  236. #endif
  237. {
  238. video->mpeg3_zigzag_scan_table = mpeg3_zig_zag_scan_nommx;
  239. video->mpeg3_alternate_scan_table = mpeg3_alternate_scan_nommx;
  240. }
  241. }
  243. mpeg3video_t *mpeg3video_allocate_struct(void)
  244. {
  245. int i;
  246. mpeg3video_t *video = calloc(1, sizeof(mpeg3video_t));
  247. #ifndef SDL_THREADS
  248. pthread_mutexattr_t mutex_attr;
  249. #endif
  251. video->vstream = mpeg3bits_new_stream();
  253. //printf("mpeg3video_allocate_struct %dn", mpeg3bits_eof(video->vstream));
  254. video->last_number = -1;
  256. /* First frame is all green */
  257. video->framenum = -1;
  258. video->have_mmx = mpeg3_mmx_test();
  260. video->percentage_seek = -1;
  261. video->frame_seek = -1;
  263. mpeg3video_init_scantables(video);
  264. mpeg3video_init_output();
  266. #ifndef SDL_THREADS
  267. pthread_mutexattr_init(&mutex_attr);
  268. // pthread_mutexattr_setkind_np(&mutex_attr, PTHREAD_MUTEX_FAST_NP);
  269. pthread_mutex_init(&(video->test_lock), &mutex_attr);
  270. pthread_mutex_init(&(video->slice_lock), &mutex_attr);
  271. #else
  272. video->test_lock = SDL_CreateMutex();
  273. video->slice_lock = SDL_CreateMutex();
  274. #endif
  275. return video;
  276. }
  278. int mpeg3video_delete_struct(mpeg3video_t *video)
  279. {
  280. int i;
  281. mpeg3bits_delete_stream(video->vstream);
  282. #ifndef SDL_THREADS
  283. pthread_mutex_destroy(&(video->test_lock));
  284. pthread_mutex_destroy(&(video->slice_lock));
  285. #else
  286. SDL_DestroyMutex(video->test_lock);
  287. SDL_DestroyMutex(video->slice_lock);
  288. #endif
  289. if(video->x_table)
  290. {
  291. free(video->x_table);
  292. free(video->y_table);
  293. }
  294. if(video->total_slice_decoders)
  295. {
  296. for(i = 0; i < video->total_slice_decoders; i++)
  297. mpeg3_delete_slice_decoder(&(video->slice_decoders[i]));
  298. }
  299. for(i = 0; i < video->slice_buffers_initialized; i++)
  300. mpeg3_delete_slice_buffer(&(video->slice_buffers[i]));
  302. free(video);
  303. return 0;
  304. }
  307. int mpeg3video_read_frame_backend(mpeg3video_t *video, int skip_bframes)
  308. {
  309. int result = 0;
  311. if(mpeg3bits_eof(video->vstream)) result = 1;
  313. if(!result) result = mpeg3video_get_header(video, 0);
  315. //printf("frame type %dn", video->pict_type);
  316. /* skip_bframes is the number of bframes we can skip successfully. */
  317. /* This is in case a skipped B-frame is repeated and the second repeat happens */
  318. /* to be a B frame we need. */
  319. video->skip_bframes = skip_bframes;
  321. if(!result)
  322. result = mpeg3video_getpicture(video, video->framenum);
  324. #ifdef HAVE_MMX
  325. if(video->have_mmx)
  326. __asm__ __volatile__ ("emms");
  327. #endif
  329. if(!result)
  330. {
  331. video->last_number = video->framenum;
  332. video->framenum++;
  333. }
  334. return result;
  335. }
  337. int* mpeg3video_get_scaletable(int input_w, int output_w)
  338. {
  339. int *result = malloc(sizeof(int) * output_w);
  340. float i;
  341. float scale = (float)input_w / output_w;
  342. for(i = 0; i < output_w; i++)
  343. {
  344. result[(int)i] = (int)(scale * i);
  345. }
  346. return result;
  347. }
  349. /* Get the first I frame. */
  350. int mpeg3video_get_firstframe(mpeg3video_t *video,
  351.       unsigned char *input, 
  352.       long size)
  353. {
  354. int result = 0;
  355. video->repeat_count = video->current_repeat = 0;
  356. return mpeg3video_process_frame(video, input, size);
  357. }
  359. int mpeg3video_process_frame(mpeg3video_t *video,
  360.      unsigned char *input, 
  361.      long size)
  362. {
  363. mpeg3bits_use_ptr_len(video->vstream, input, size);
  364. return mpeg3video_read_frame_backend(video, 0);
  365. }
  367. static long gop_to_frame(mpeg3video_t *video, mpeg3_timecode_t *gop_timecode)
  368. {
  369. int hour, minute, second, frame, fps;
  370. long result;
  372. // Mirror of what mpeg2enc does
  373. fps = (int)(video->frame_rate + 0.5);
  376. hour = gop_timecode->hour;
  377. minute = gop_timecode->minute;
  378. second = gop_timecode->second;
  379. frame = gop_timecode->frame;
  381. result = (long)hour * 60 * 60 * fps + 
  382. minute * 60 * fps + 
  383. second * fps +
  384. frame;
  386. return result;
  387. }
  390. /* ======================================================================= */
  391. /*                                    ENTRY POINTS */
  392. /* ======================================================================= */
  396. mpeg3video_t* mpeg3video_new(int is_video_stream,
  397.      int cpus)
  398. {
  399. mpeg3video_t *video;
  400. int result = 0;
  402. video = mpeg3video_allocate_struct();
  403. video->cpus = cpus;
  405. return video;
  406. }
  408. int mpeg3video_delete(mpeg3video_t *video)
  409. {
  410. if(video->decoder_initted)
  411. {
  412. mpeg3video_deletedecoder(video);
  413. }
  414. mpeg3video_delete_struct(video);
  415. return 0;
  416. }
  418. int mpeg3video_set_cpus(mpeg3video_t *video, int cpus)
  419. {
  420. return 0;
  421. }
  423. int mpeg3video_set_mmx(mpeg3video_t *video, int use_mmx)
  424. {
  425. video->have_mmx = use_mmx;
  426. mpeg3video_init_scantables(video);
  427. return 0;
  428. }
  430. int mpeg3video_read_frame(mpeg3video_t *video, 
  431.   unsigned char *input,
  432.   long input_size,
  433.   unsigned char **output_rows,
  434.   int in_x, 
  435.   int in_y, 
  436.   int in_w, 
  437.   int in_h, 
  438.   int out_w, 
  439.   int out_h, 
  440.   int color_model)
  441. {
  442. int result = 0;
  445. mpeg3bits_use_ptr_len(video->vstream, input, input_size); 
  447. video->want_yvu = 0;
  448. video->output_rows = output_rows;
  449. video->color_model = color_model;
  451. //printf("mpeg3video_read_frame 1n");
  452. /* Get scaling tables */
  453. if(video->out_w != out_w || video->out_h != out_h ||
  454. video->in_w != in_w || video->in_h != in_h ||
  455. video->in_x != in_x || video->in_y != in_y)
  456. {
  457. if(video->x_table)
  458. {
  459. free(video->x_table);
  460. free(video->y_table);
  461. video->x_table = 0;
  462. video->y_table = 0;
  463. }
  464. }
  466. video->out_w = out_w;
  467. video->out_h = out_h;
  468. video->in_w = in_w;
  469. video->in_h = in_h;
  470. video->in_x = in_x;
  471. video->in_y = in_y;
  473. if(!video->x_table)
  474. {
  475. video->x_table = mpeg3video_get_scaletable(video->in_w, video->out_w);
  476. video->y_table = mpeg3video_get_scaletable(video->in_h, video->out_h);
  477. }
  479. //printf("mpeg3video_read_frame 4n");
  480. if(!result) result = mpeg3video_read_frame_backend(video, 0);
  482. //printf("mpeg3video_read_frame 5n");
  483. if(video->output_src) mpeg3video_present_frame(video);
  485. //printf("mpeg3video_read_frame 6n");
  486. return result;
  487. }
  489. int mpeg3video_read_yuvframe(mpeg3video_t *video, 
  490.      unsigned char *input,
  491.      long input_size,
  492.      char *y_output,
  493.      char *u_output,
  494.      char *v_output,
  495.      int in_x,
  496.      int in_y,
  497.      int in_w,
  498.      int in_h)
  499. {
  500. int result = 0;
  502. mpeg3bits_use_ptr_len(video->vstream, input, input_size); 
  504. video->want_yvu = 1;
  505. video->y_output = y_output;
  506. video->u_output = u_output;
  507. video->v_output = v_output;
  508. video->in_x = in_x;
  509. video->in_y = in_y;
  510. video->in_w = in_w;
  511. video->in_h = in_h;
  513. if(!result) result = mpeg3video_read_frame_backend(video, 0);
  515. if(video->output_src) mpeg3video_present_frame(video);
  517. video->want_yvu = 0;
  518. video->percentage_seek = -1;
  519. return result;
  520. }
  522. int mpeg3video_read_yuvframe_ptr(mpeg3video_t *video, 
  523.  unsigned char *input,
  524.  long input_size,
  525.  char **y_output,
  526.  char **u_output,
  527.  char **v_output)
  528. {
  529. int result = 0;
  531. mpeg3bits_use_ptr_len(video->vstream, input, input_size); 
  533. video->want_yvu = 1;
  535. if(!result) result = mpeg3video_read_frame_backend(video, 0);
  537. if (video->output_src) {
  538. *y_output = video->output_src[0];
  539. *u_output = video->output_src[1];
  540. *v_output = video->output_src[2];
  541. }
  543. video->want_yvu = 0;
  544. video->percentage_seek = -1;
  547. return result;
  548. }
  550. int mpeg3video_colormodel(mpeg3video_t *video)
  551. {
  552. switch(video->chroma_format)
  553. {
  554. case CHROMA422:
  555. return MPEG3_YUV422P;
  556. break;
  558. case CHROMA420:
  559. return MPEG3_YUV420P;
  560. break;
  561. }
  563. return MPEG3_YUV420P;
  564. }