motion_comp.c
上传用户:aoeyumen
上传日期:2007-01-06
资源大小:3329k
文件大小:3k
源码类别:

DVD

开发平台:

Unix_Linux

  1. /*
  2.  *  motion_comp.c
  3.  *
  4.  *  Copyright (C) Aaron Holtzman <aholtzma@ess.engr.uvic.ca> - Nov 1999
  5.  *
  6.  *  This file is part of mpeg2dec, a free MPEG-2 video stream decoder.
  7.  *
  8.  *  mpeg2dec is free software; you can redistribute it and/or modify
  9.  *  it under the terms of the GNU General Public License as published by
  10.  *  the Free Software Foundation; either version 2, or (at your option)
  11.  *  any later version.
  12.  *   
  13.  *  mpeg2dec is distributed in the hope that it will be useful,
  14.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  15.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16.  *  GNU General Public License for more details.
  17.  *   
  18.  *  You should have received a copy of the GNU General Public License
  19.  *  along with GNU Make; see the file COPYING.  If not, write to
  20.  *  the Free Software Foundation, 
  21.  *
  22.  */
  23. #include <stdlib.h>
  24. #include <stdio.h>
  25. #include "config.h"
  26. #include "mpeg2.h"
  27. #include "mpeg2_internal.h"
  28. #include "mb_buffer.h"
  29. #include "motion_comp.h"
  30. #include "motion_comp_mmx.h"
  31. // motion_comp main entry point 
  32. void (*motion_comp)(picture_t *picture,mb_buffer_t *mb_buffer);
  33. void motion_comp_c(picture_t *picture,mb_buffer_t *mb_buffer);
  34. static uint_8 clip_lut[1024], *clip;
  35. void
  36. motion_comp_init(void)
  37. {
  38. sint_32 i;
  39. clip = clip_lut + 384;
  40. for(i=-384;i<640;i++)
  41. clip[i] = i < 0 ? 0 : (i > 255 ? 255 : i);
  42. #ifdef __i386__
  43. if(config.flags & MPEG2_MMX_ENABLE)
  44. motion_comp = motion_comp_mmx;
  45. else
  46. #endif
  47. motion_comp = motion_comp_c;
  48. }
  49. //Combine the prediction and intra coded block into current frame
  50. static void
  51. motion_comp_block(uint_8 *pred,sint_16 *block,uint_8 *dst,uint_32 pitch)
  52. {
  53. uint_32 i;
  54. pitch = pitch - 8;
  55. for(i=0;i<8;i++)
  56. {
  57. *dst++ = clip[*block++];
  58. *dst++ = clip[*block++];
  59. *dst++ = clip[*block++];
  60. *dst++ = clip[*block++];
  61. *dst++ = clip[*block++];
  62. *dst++ = clip[*block++];
  63. *dst++ = clip[*block++];
  64. *dst++ = clip[*block++];
  65. dst += pitch;
  66. }
  67. }
  68. void
  69. motion_comp_c(picture_t *picture,mb_buffer_t *mb_buffer)
  70. {
  71. macroblock_t *mb   = mb_buffer->macroblocks;
  72. uint_32 num_blocks = mb_buffer->num_blocks;
  73. uint_32 i;
  74. uint_32 width,x,y;
  75. uint_32 mb_width;
  76. uint_32 pitch;
  77. uint_32 d;
  78. uint_8 *dst;
  79. width = picture->coded_picture_width;
  80. mb_width = picture->coded_picture_width >> 4;
  81. for(i=0;i<num_blocks;i++)
  82. {
  83. if(mb[i].skipped)
  84. continue;
  85. //handle interlaced blocks
  86. if (mb[i].dct_type) 
  87. {
  88. d = 1;
  89. pitch = width *2;
  90. }
  91. else 
  92. {
  93. d = 8;
  94. pitch = width;
  95. }
  96. //FIXME I'd really to take these two divides out.
  97. //maybe do fixed point mult with a LUT of the 16.16 inverse
  98. //of common widths
  99. x = mb[i].mba % mb_width;
  100. y = mb[i].mba / mb_width;
  101. //Do y component
  102. dst = &picture->current_frame[0][x * 16 + y * width * 16];
  103. motion_comp_block(0, mb[i].y_blocks       , dst                , pitch);
  104. motion_comp_block(0, mb[i].y_blocks +   64, dst + 8            , pitch);
  105. motion_comp_block(0, mb[i].y_blocks + 2*64, dst + width * d    , pitch);
  106. motion_comp_block(0, mb[i].y_blocks + 3*64, dst + width * d + 8, pitch);
  107. //Do Cr component
  108. dst = &picture->current_frame[1][x * 8 + y * width/2 * 8];
  109. motion_comp_block(0, mb[i].cr_blocks, dst, width/2);
  110. //Do Cb component
  111. dst = &picture->current_frame[2][x * 8 + y * width/2 * 8];
  112. motion_comp_block(0, mb[i].cb_blocks, dst, width/2);
  113. }
  114. }