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

源码开发语言/平台
当前位置：首页 > 电子书籍 > VC书籍 > 查看源码
mp4_predict.c
资源名称：VC++视频传输.rar　[点击查看]
上传用户：hxb_1234
上传日期：2010-03-30
资源大小：8328k
文件大小：8k
源码类别：

VC书籍

开发平台：

Visual C++

mp4_predict.c：源码内容
  3. #include <math.h>
  5. #include "mp4_vars.h"
  7. #include "mp4_predict.h"
  9. /**
  10.  *
  11. **/
  13. static void rescue_predict();
  17. void dc_recon(int block_num, short * dc_value)
  18. {
  19. if (mp4_state->hdr.prediction_type == P_VOP) {
  20. rescue_predict();
  21. }
  23. if (block_num < 4)
  24. {
  25. int b_xpos = (mp4_state->hdr.mb_xpos << 1) + (block_num & 1);
  26. int b_ypos = (mp4_state->hdr.mb_ypos << 1) + ((block_num & 2) >> 1);
  27. int dc_pred;
  30. if (abs(mp4_state->coeff_pred.dc_store_lum[b_ypos+1-1][b_xpos+1-1] -
  31. mp4_state->coeff_pred.dc_store_lum[b_ypos+1][b_xpos+1-1]) < // Fa - Fb
  32. abs(mp4_state->coeff_pred.dc_store_lum[b_ypos+1-1][b_xpos+1-1] -
  33. mp4_state->coeff_pred.dc_store_lum[b_ypos+1-1][b_xpos+1])) // Fb - Fc
  34. {
  35. mp4_state->coeff_pred.predict_dir = TOP;
  36. dc_pred = mp4_state->coeff_pred.dc_store_lum[b_ypos+1-1][b_xpos+1];
  37. }
  38. else
  39. {
  40. mp4_state->coeff_pred.predict_dir = LEFT;
  41. dc_pred = mp4_state->coeff_pred.dc_store_lum[b_ypos+1][b_xpos+1-1];
  42. }
  44. (* dc_value) += _div_div(dc_pred, mp4_state->hdr.dc_scaler);
  45. (* dc_value) *= mp4_state->hdr.dc_scaler;
  48. mp4_state->coeff_pred.dc_store_lum[b_ypos+1][b_xpos+1] = (* dc_value);
  49. }
  50. else 
  51. {
  52. int b_xpos = mp4_state->hdr.mb_xpos;
  53. int b_ypos = mp4_state->hdr.mb_ypos;
  54. int chr_num = block_num - 4;
  55. int dc_pred;
  58. if (abs(mp4_state->coeff_pred.dc_store_chr[chr_num][b_ypos+1-1][b_xpos+1-1] -
  59. mp4_state->coeff_pred.dc_store_chr[chr_num][b_ypos+1][b_xpos+1-1]) < // Fa - Fb
  60. abs(mp4_state->coeff_pred.dc_store_chr[chr_num][b_ypos+1-1][b_xpos+1-1] -
  61. mp4_state->coeff_pred.dc_store_chr[chr_num][b_ypos+1-1][b_xpos+1])) // Fb - Fc
  62. {
  63. mp4_state->coeff_pred.predict_dir = TOP;
  64. dc_pred = mp4_state->coeff_pred.dc_store_chr[chr_num][b_ypos+1-1][b_xpos+1];
  65. }
  66. else
  67. {
  68. mp4_state->coeff_pred.predict_dir = LEFT;
  69. dc_pred = mp4_state->coeff_pred.dc_store_chr[chr_num][b_ypos+1][b_xpos+1-1];
  70. }
  72. (* dc_value) += _div_div(dc_pred, mp4_state->hdr.dc_scaler);
  73. (* dc_value) *= mp4_state->hdr.dc_scaler;
  76. mp4_state->coeff_pred.dc_store_chr[chr_num][b_ypos+1][b_xpos+1] = (* dc_value);
  77. }
  78. }
  80. /***/
  82. static int saiAcLeftIndex[8] = 
  83. {
  84. 0, 8,16,24,32,40,48,56
  85. };
  87. void ac_recon(int block_num, short * psBlock)
  88. {
  89. int b_xpos, b_ypos;
  90. int i;
  92. if (block_num < 4) {
  93. b_xpos = (mp4_state->hdr.mb_xpos << 1) + (block_num & 1);
  94. b_ypos = (mp4_state->hdr.mb_ypos << 1) + ((block_num & 2) >> 1);
  95. }
  96. else {
  97. b_xpos = mp4_state->hdr.mb_xpos;
  98. b_ypos = mp4_state->hdr.mb_ypos;
  99. }
  102. if (mp4_state->hdr.ac_pred_flag) 
  103. {
  104. if (block_num < 4) 
  105. {
  106. if (mp4_state->coeff_pred.predict_dir == TOP)
  107. {
  108. for (i = 1; i < 8; i++) 
  109. psBlock[i] += mp4_state->coeff_pred.ac_top_lum[b_ypos+1-1][b_xpos+1][i-1];
  110. }
  111. else 
  112. {
  113. for (i = 1; i < 8; i++)
  114. psBlock[mp4_tables->saiAcLeftIndex[i]] += mp4_state->coeff_pred.ac_left_lum[b_ypos+1][b_xpos+1-1][i-1];
  115. }
  116. }
  117. else
  118. {
  119. int chr_num = block_num - 4;
  121. if (mp4_state->coeff_pred.predict_dir == TOP)
  122. {
  123. for (i = 1; i < 8; i++)
  124. psBlock[i] += mp4_state->coeff_pred.ac_top_chr[chr_num][b_ypos+1-1][b_xpos+1][i-1];
  125. }
  126. else 
  127. {
  128. for (i = 1; i < 8; i++)
  129. psBlock[mp4_tables->saiAcLeftIndex[i]] += mp4_state->coeff_pred.ac_left_chr[chr_num][b_ypos+1][b_xpos+1-1][i-1];
  130. }
  131. }
  132. }
  133. }
  135. /***/
  137. void ac_store(int block_num, short * psBlock)
  138. {
  139. int b_xpos, b_ypos;
  140. int i;
  142. if (block_num < 4) { 
  143. b_xpos = (mp4_state->hdr.mb_xpos << 1) + (block_num & 1);
  144. b_ypos = (mp4_state->hdr.mb_ypos << 1) + ((block_num & 2) >> 1);
  145. }
  146. else {
  147. b_xpos = mp4_state->hdr.mb_xpos;
  148. b_ypos = mp4_state->hdr.mb_ypos;
  149. }
  151. if (block_num < 4)
  152. {
  153. for (i = 1; i < 8; i++) {
  154. mp4_state->coeff_pred.ac_top_lum[b_ypos+1][b_xpos+1][i-1] = psBlock[i];
  155. mp4_state->coeff_pred.ac_left_lum[b_ypos+1][b_xpos+1][i-1] = psBlock[mp4_tables->saiAcLeftIndex[i]];
  156. }
  157. }
  158. else 
  159. {
  160. int chr_num = block_num - 4;
  162. for (i = 1; i < 8; i++) {
  163. mp4_state->coeff_pred.ac_top_chr[chr_num][b_ypos+1][b_xpos+1][i-1] = psBlock[i];
  164. mp4_state->coeff_pred.ac_left_chr[chr_num][b_ypos+1][b_xpos+1][i-1] = psBlock[mp4_tables->saiAcLeftIndex[i]];
  165. }
  166. }
  167. }
  170. /***/
  172. #define _rescale(predict_quant, current_quant, coeff) (coeff != 0) ?
  173. _div_div((coeff) * (predict_quant), (current_quant)) : 0
  175. int ac_rescaling(int block_num, short * psBlock)
  176. {
  177. int mb_xpos = mp4_state->hdr.mb_xpos;
  178. int mb_ypos = mp4_state->hdr.mb_ypos;
  179. int current_quant = mp4_state->hdr.quantizer;
  180. int predict_quant = (mp4_state->coeff_pred.predict_dir == TOP) ?
  181. mp4_state->quant_store[mb_ypos][mb_xpos+1] : mp4_state->quant_store[mb_ypos+1][mb_xpos];
  182. int b_xpos, b_ypos; 
  183. int i;
  185. if ((! mp4_state->hdr.ac_pred_flag) || (current_quant == predict_quant) || (block_num == 3))
  186. return 0;
  188. if ((mb_ypos == 0) && (mp4_state->coeff_pred.predict_dir == TOP))
  189. return 0;
  190. if ((mb_xpos == 0) && (mp4_state->coeff_pred.predict_dir == LEFT))
  191. return 0;
  192. if ((mb_xpos == 0) && (mb_ypos == 0))
  193. return 0;
  195. if (block_num < 4) {
  196. b_xpos = (mp4_state->hdr.mb_xpos << 1) + (block_num & 1);
  197. b_ypos = (mp4_state->hdr.mb_ypos << 1) + ((block_num & 2) >> 1);
  198. }
  199. else {
  200. b_xpos = mp4_state->hdr.mb_xpos;
  201. b_ypos = mp4_state->hdr.mb_ypos;
  202. }
  204. if (mp4_state->coeff_pred.predict_dir == TOP) 
  205. switch (block_num)
  206. {
  207. case 0: case 1:
  208. for (i = 1; i < 8; i++)
  209. psBlock[i] += _rescale(predict_quant, current_quant, mp4_state->coeff_pred.ac_top_lum[b_ypos][b_xpos+1][i-1]);
  210. return 1;
  211. break;
  212. case 4:
  213. for (i = 1; i < 8; i++)
  214. psBlock[i] += _rescale(predict_quant, current_quant, mp4_state->coeff_pred.ac_top_chr[0][b_ypos][b_xpos+1][i-1]);
  215. return 1;
  216. break;
  217. case 5:
  218. for (i = 1; i < 8; i++)
  219. psBlock[i] += _rescale(predict_quant, current_quant, mp4_state->coeff_pred.ac_top_chr[1][b_ypos][b_xpos+1][i-1]);
  220. return 1;
  221. break;
  222. }
  223. else 
  224. {
  225. switch (block_num)
  226. {
  227. case 0: case 2:
  228. for (i = 1; i < 8; i++)
  229. psBlock[mp4_tables->saiAcLeftIndex[i]] += _rescale(predict_quant, current_quant, mp4_state->coeff_pred.ac_left_lum[b_ypos+1][b_xpos][i-1]);
  230. return 1;
  231. break;
  232. case 4:
  233. for (i = 1; i < 8; i++)
  234. psBlock[mp4_tables->saiAcLeftIndex[i]] += _rescale(predict_quant, current_quant, mp4_state->coeff_pred.ac_left_chr[0][b_ypos+1][b_xpos][i-1]);
  235. return 1;
  236. break;
  237. case 5:
  238. for (i = 1; i < 8; i++)
  239. psBlock[mp4_tables->saiAcLeftIndex[i]] += _rescale(predict_quant, current_quant, mp4_state->coeff_pred.ac_left_chr[1][b_ypos+1][b_xpos][i-1]);
  240. return 1;
  241. break;
  242. }
  243. }
  245. return 0;
  246. }
  248. /***/
  250. #define _IsIntra(mb_y, mb_x) ((mp4_state->modemap[(mb_y)+1][(mb_x)+1] == INTRA) || 
  251. (mp4_state->modemap[(mb_y)+1][(mb_x)+1] == INTRA_Q))
  253. static void rescue_predict() 
  254. {
  255. int mb_xpos = mp4_state->hdr.mb_xpos;
  256. int mb_ypos = mp4_state->hdr.mb_ypos;
  257. int i;
  259. if (! _IsIntra(mb_ypos-1, mb_xpos-1)) {
  261. mp4_state->coeff_pred.dc_store_lum[2*mb_ypos+1-1][2*mb_xpos+1-1] = 1024;
  262. mp4_state->coeff_pred.dc_store_chr[0][mb_ypos+1-1][mb_xpos+1-1] = 1024;
  263. mp4_state->coeff_pred.dc_store_chr[1][mb_ypos+1-1][mb_xpos+1-1] = 1024;
  264. }
  266. if (! _IsIntra(mb_ypos, mb_xpos-1)) {
  268. mp4_state->coeff_pred.dc_store_lum[2*mb_ypos+1][2*mb_xpos+1-1] = 1024;
  269. mp4_state->coeff_pred.dc_store_lum[2*mb_ypos+1+1][2*mb_xpos+1-1] = 1024;
  270. mp4_state->coeff_pred.dc_store_chr[0][mb_ypos+1][mb_xpos+1-1] = 1024;
  271. mp4_state->coeff_pred.dc_store_chr[1][mb_ypos+1][mb_xpos+1-1] = 1024;
  273. for(i = 0; i < 7; i++) {
  274. mp4_state->coeff_pred.ac_left_lum[2*mb_ypos+1][2*mb_xpos+1-1][i] = 0;
  275. mp4_state->coeff_pred.ac_left_lum[2*mb_ypos+1+1][2*mb_xpos+1-1][i] = 0;
  276. mp4_state->coeff_pred.ac_left_chr[0][mb_ypos+1][mb_xpos+1-1][i] = 0;
  277. mp4_state->coeff_pred.ac_left_chr[1][mb_ypos+1][mb_xpos+1-1][i] = 0;
  278. }
  279. }
  281. if (! _IsIntra(mb_ypos-1, mb_xpos)) {
  283. mp4_state->coeff_pred.dc_store_lum[2*mb_ypos+1-1][2*mb_xpos+1] = 1024;
  284. mp4_state->coeff_pred.dc_store_lum[2*mb_ypos+1-1][2*mb_xpos+1+1] = 1024;
  285. mp4_state->coeff_pred.dc_store_chr[0][mb_ypos+1-1][mb_xpos+1] = 1024;
  286. mp4_state->coeff_pred.dc_store_chr[1][mb_ypos+1-1][mb_xpos+1] = 1024;
  288. for(i = 0; i < 7; i++) {
  289. mp4_state->coeff_pred.ac_top_lum[2*mb_ypos+1-1][2*mb_xpos+1][i] = 0;
  290. mp4_state->coeff_pred.ac_top_lum[2*mb_ypos+1-1][2*mb_xpos+1+1][i] = 0;
  291. mp4_state->coeff_pred.ac_top_chr[0][mb_ypos+1-1][mb_xpos+1][i] = 0;
  292. mp4_state->coeff_pred.ac_top_chr[1][mb_ypos+1-1][mb_xpos+1][i] = 0;
  293. }
  294. }
  295. }