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

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

VC书籍

开发平台：

Visual C++

mot_code.c：源码内容
  1. #include "vm_common_defs.h"
  2. #include "bitstream.h"
  3. #include "putvlc.h"
  4. #define MBV_H(h,v)      (ph[2*(v)*2*hdim+2*(h)])
  5. #define MBV_V(h,v)      (pv[2*(v)*2*hdim+2*(h)])
  6. #define BV_H(h,v,h2,v2) (ph[(2*(v)+(v2))*2*hdim+2*(h)+(h2)])
  7. #define BV_V(h,v,h2,v2) (pv[(2*(v)+(v2))*2*hdim+2*(h)+(h2)])
  8. #define MB_MODE(h,v)    ( ((h)<0||(h)>=hdim||(v)<0||(v)>=vdim ? MBM_OUT : (pm[(v)*hdim+(h)])) )
  9. #define BV(p,xdim,h,v,h2,v2) (p[(2*(v)+(v2))*(xdim)+2*(h)+(h2)])
  10. Int WriteMVcomponent(Int f_code, Int dmv, Image *bs);
  11. Void ScaleMVD (Int f_code, Int diff_vector, Int *residual, Int *vlc_code_mag);
  12. Void   find_pmvs (Image *mot_x, Image *mot_y, Image *MB_decisions, Image *B_decisions,
  13. Int x, Int y, Int block, Int transparent_value, Int quarter_pel, Int *error_flag,
  14. Int *mvx, Int *mvy, Int newgob);
  15. SInt ModeMB (Image *MB_decision, Int i, Int j);
  16. Int
  17. Bits_CountMB_Motion(
  18. Image   *mot_h,   
  19. Image   *mot_v,   
  20. Image   *alpha,   
  21. Image   *modes,   
  22. Int     h,   
  23. Int     v,   
  24. Int     f_code,   
  25.   
  26. Int     quarter_pel,       
  27. Image   *bs,   
  28. Int     error_res_disable,
  29. Int     after_marker,
  30. Int     **slice_nb,
  31. Int arbitrary_shape
  32. )
  33. {
  34. Int     vdim, hdim;   
  35. Float   *ph, *pv;   
  36. SInt    *pm;   
  37. SInt    *pa;   
  38. Int     mode;
  39. Int     bits_mot = 0;
  41. Int     i, error_flag=0,mvx=0,mvy=0;
  42. Float   pred_h, pred_v;
  43. Float   diff_h, diff_v;
  44. Int     bh, bv;
  45. Int     local_f_code;   
  46. Float   subdim;   
  47. vdim = (Int)modes->y;
  48. hdim = (Int)modes->x;
  49. ph=(Float*)GetImageData(mot_h);
  50. pv=(Float*)GetImageData(mot_v);
  51. pm=(SInt*)GetImageData(modes);
  52. pa=NULL;
  55. if (quarter_pel)
  56. {
  57. local_f_code = f_code+1;
  58. subdim = 4.0;
  59. }
  60. else
  61. {
  62. local_f_code = f_code;
  63. subdim = 2.0;
  64. }
  66. switch (mode=MB_MODE(h,v))
  67. {
  68. case MBM_INTRA:
  69. break;
  70. case MBM_INTER16:
  72. find_pmvs(mot_h,mot_v,modes,alpha,h,v,0,MBM_TRANSPARENT,
  73.   
  74. quarter_pel,&error_flag,&mvx,&mvy,0);
  75. pred_h=((Float)mvx)/subdim;   
  76. pred_v=((Float)mvy)/subdim;   
  77.   
  78. bits_mot += WriteMVcomponent(local_f_code, (Int)(subdim*(MBV_H(h,v) - pred_h)), bs);
  79.   
  80. bits_mot += WriteMVcomponent(local_f_code, (Int)(subdim*(MBV_V(h,v) - pred_v)), bs);
  81. break;
  82. case MBM_INTER8:
  83. i=1;
  84. for (bv=0; bv<=1; bv++)
  85. for (bh=0; bh<=1; bh++)
  86. {
  87. find_pmvs(mot_h,mot_v,modes,alpha,h,v,i,MBM_TRANSPARENT,
  88.   
  89. quarter_pel,&error_flag,&mvx,&mvy,0);
  90. pred_h=((Float)mvx)/subdim;   
  91. pred_v=((Float)mvy)/subdim;   
  92. i++;
  93. diff_h=BV_H(h,v,bh,bv)-pred_h;
  94. diff_v=BV_V(h,v,bh,bv)-pred_v;
  95.   
  96. bits_mot += WriteMVcomponent(local_f_code, (Int)(subdim*diff_h), bs);
  97.   
  98. bits_mot += WriteMVcomponent(local_f_code, (Int)(subdim*diff_v), bs);
  99. }
  100. break;
  101. }
  102. return bits_mot;
  103. }
  104. Int
  105. WriteMVcomponent(
  106. Int     f_code,
  107. Int     dmv,
  108. Image   *bs
  109. )
  110. {
  111. Int   residual, vlc_code_mag, bits, entry;
  112. ScaleMVD(f_code, dmv, &residual, &vlc_code_mag);
  113. if (vlc_code_mag < 0)
  114. entry = vlc_code_mag + 65;
  115. else
  116. entry = vlc_code_mag;
  117. bits = PutMV (entry, bs);
  118. if ((f_code != 1) && (vlc_code_mag != 0))
  119. {
  120. BitstreamPutBits(bs, residual, f_code-1);
  121. bits += f_code - 1;
  122. }
  123. return(bits);
  124. }
  125. Void
  126. ScaleMVD (
  127. Int  f_code,   
  128. Int  diff_vector,   
  129. Int  *residual,   
  130. Int  *vlc_code_mag   
  131. )
  132. {
  133. Int   range;
  134. Int   scale_factor;
  135. Int   r_size;
  136. Int   low;
  137. Int   high;
  138. Int   aux;
  139. r_size = f_code-1;
  140. scale_factor = 1<<r_size;
  141. range = 32*scale_factor;
  142. low   = -range;
  143. high  =  range-1;
  144. if (diff_vector < low)
  145. {
  146. diff_vector += 2*range;
  147. }
  148. else if (diff_vector > high)
  149. {
  150. diff_vector -= 2*range;
  151. }
  152. if (diff_vector==0)
  153. {
  154. *vlc_code_mag = 0;
  155. *residual = 0;
  156. }
  157. else if (scale_factor==1)
  158. {
  159. *vlc_code_mag = diff_vector;
  160. *residual = 0;
  161. }
  162. else
  163. {
  164. aux = ABS(diff_vector) + scale_factor - 1;
  165. *vlc_code_mag = aux>>r_size;
  166. if (diff_vector<0)
  167. *vlc_code_mag = -*vlc_code_mag;
  168. *residual = aux & (scale_factor-1);
  169. }
  170. }
  171. Void   
  172. find_pmvs(
  173. Image  *mot_x,   
  174. Image  *mot_y,   
  175. Image  *MB_decisions,   
  176. Image  *B_decisions,   
  177. Int    x,   
  178. Int    y,   
  179. Int    block,   
  180. Int    transparent_value,   
  181. Int    quarter_pel,      
  182. Int    *error_flag,   
  183. Int    *mvx,   
  184. Int    *mvy,   
  185. Int    newgob
  186. )
  187. {
  188. Float   p1x,p2x,p3x;
  189. Float   p1y,p2y,p3y;
  190. Int     xin1, xin2, xin3;
  191. Int     yin1, yin2, yin3;
  192. Int     vec1, vec2, vec3;
  193. Int     rule1, rule2, rule3;
  194. Int     subdim;   
  195. Float   *motxdata = (Float *) GetImageData(mot_x);
  196. Float   *motydata = (Float *) GetImageData(mot_y);
  197. Int     xM = GetImageSizeX(mot_x);
  198. Int xB = xM;
  199. Int     mb_mode, sum;
  201. if (quarter_pel)
  202. {
  203. subdim=4;
  204. }
  205. else
  206. {
  207. subdim=2;
  208. }
  211. switch (block)
  212. {
  213. case 0:
  214. vec1 = 1 ; yin1 = y  ; xin1 = x-1;
  215. vec2 = 2 ; yin2 = y-1; xin2 = x;
  216. vec3 = 2 ; yin3 = y-1; xin3 = x+1;
  217. break;
  218. case 1:
  219. vec1 = 1 ; yin1 = y  ; xin1 = x-1;
  220. vec2 = 2 ; yin2 = y-1; xin2 = x;
  221. vec3 = 2 ; yin3 = y-1; xin3 = x+1;
  222. break;
  223. case 2:
  224. vec1 = 0 ; yin1 = y  ; xin1 = x;
  225. vec2 = 3 ; yin2 = y-1; xin2 = x;
  226. vec3 = 2 ; yin3 = y-1; xin3 = x+1;
  227. break;
  228. case 3:
  229. vec1 = 3 ; yin1 = y  ; xin1 = x-1;
  230. vec2 = 0 ; yin2 = y  ; xin2 = x;
  231. vec3 = 1 ; yin3 = y  ; xin3 = x;
  232. break;
  233. case 4:
  234. vec1 = 2 ; yin1 = y  ; xin1 = x;
  235. vec2 = 0 ; yin2 = y  ; xin2 = x;
  236. vec3 = 1 ; yin3 = y  ; xin3 = x;
  237. break;
  238. default:
  239. printf("Illegal block number in find_pmv (mot_decode.c)");
  240. *error_flag = 1;
  241. *mvx=*mvy=0;
  242. return ;
  243. }
  244. if (block==0)
  245. {
  247. if (x>0 )
  248. rule1 = 0;
  249. else
  250. rule1 = 1;
  251. if ((y>0 && newgob==0) )
  252. rule2 = 0;
  253. else
  254. rule2 = 1;
  255. if ((x != xB/2 -1) &&
  256. ((y>0 && newgob==0)) )
  257. rule3 = 0;
  258. else
  259. rule3 = 1;
  260. }
  261. else
  262. {
  265.   
  266. if (((block == 1 || block == 3) && (x == 0))  )
  267. rule1 = 1;
  268. else
  269. rule1 = 0;
  271.   
  272. if (((block == 1 || block == 2) && (y == 0))  )
  273. rule2 = 1;
  274. else
  275. rule2 = 0;
  277. if (((block == 1 || block == 2) && (x == xB/2 -1 || y == 0)) )
  278. rule3 = 1;
  279. else
  280. rule3 = 0;
  281. }
  282. if (rule1 )
  283. {
  284. p1x = p1y = 0;
  285. }
  286. else if (((mb_mode = ModeMB(MB_decisions,xin1,yin1)) >= MBM_FIELD00) && (mb_mode <= MBM_FIELD11))
  287. {
  289. sum = subdim*(BV(motxdata, xM, xin1, yin1, 0, 0) + BV(motxdata, xM, xin1, yin1, 1, 0));
  290. p1x = (Float)((sum & 3) ? ((sum | 2) >> 1) : (sum >> 1))/(Float)subdim;
  291. sum = subdim*(BV(motydata, xM, xin1, yin1, 0, 0) + BV(motydata, xM, xin1, yin1, 1, 0));
  292. p1y = (Float)((sum & 3) ? ((sum | 2) >> 1) : (sum >> 1))/(Float)subdim;
  293. }
  294. else
  295. {
  296. p1x = BV(motxdata, xM, xin1, yin1, vec1&0x1, vec1>>1 );
  297. p1y = BV(motydata, xM, xin1, yin1, vec1&0x1, vec1>>1 );
  298. }
  299. if (rule2)
  300. {
  301. p2x = p2y = 0 ;
  302. }
  303. else if (((mb_mode = ModeMB(MB_decisions,xin2,yin2)) >= MBM_FIELD00) && (mb_mode <= MBM_FIELD11))
  304. {
  306. sum = subdim*(BV(motxdata, xM, xin2, yin2, 0, 0) + BV(motxdata, xM, xin2, yin2, 1, 0));
  307. p2x = (Float)((sum & 3) ? ((sum | 2) >> 1) : (sum >> 1))/(Float)subdim;
  308. sum = subdim*(BV(motydata, xM, xin2, yin2, 0, 0) + BV(motydata, xM, xin2, yin2, 1, 0));
  309. p2y = (Float)((sum & 3) ? ((sum | 2) >> 1) : (sum >> 1))/(Float)subdim;
  310. }
  311. else
  312. {
  313. p2x = BV(motxdata, xM, xin2, yin2, vec2&0x1, vec2>>1 );
  314. p2y = BV(motydata, xM, xin2, yin2, vec2&0x1, vec2>>1 );
  315. }
  316. if (rule3 )
  317. {
  318. p3x = p3y =0;
  319. }
  320. else if (((mb_mode = ModeMB(MB_decisions,xin3,yin3)) >= MBM_FIELD00) && (mb_mode <= MBM_FIELD11))
  321. {
  323. sum = subdim*(BV(motxdata, xM, xin3, yin3, 0, 0) + BV(motxdata, xM, xin3, yin3, 1, 0));
  324. p3x = (Float)((sum & 3) ? ((sum | 2) >> 1) : (sum >> 1))/(Float)subdim;
  325. sum = subdim*(BV(motydata, xM, xin3, yin3, 0, 0) + BV(motydata, xM, xin3, yin3, 1, 0));
  326. p3y = (Float)((sum & 3) ? ((sum | 2) >> 1) : (sum >> 1))/(Float)subdim;
  327. }
  328. else
  329. {
  330. p3x = BV(motxdata, xM, xin3, yin3, vec3&0x1, vec3>>1 );
  331. p3y = BV(motydata, xM, xin3, yin3, vec3&0x1, vec3>>1 );
  332. }
  333. if (rule1 && rule2 && rule3 )
  334. {
  336. *mvx=*mvy=0;
  337. }
  338. else if (rule1+rule2+rule3 == 2)
  339. {
  342. *mvx=(Int) subdim*(p1x+p2x+p3x);   
  343. *mvy=(Int) subdim*(p1y+p2y+p3y);   
  344. }
  345. else
  346. {
  348.   
  349. *mvx=(Int)(subdim*(p1x+p2x+p3x-MAX(p1x,MAX(p2x,p3x))-MIN(p1x,MIN(p2x,p3x))));
  350.   
  351. *mvy=(Int)(subdim*(p1y+p2y+p3y-MAX(p1y,MAX(p2y,p3y))-MIN(p1y,MIN(p2y,p3y))));
  352. }
  353. #ifdef _DEBUG_PMVS_
  354. fprintf(stdout,"find_pmvs (%2d,%2d, rule %1d%1d%1d) :np1 %6.2f / %6.2fnp2 %6.2f / %6.2fnp3 %6.2f / %6.2fn",x,y,rule1,rule2,rule3,p1x,p1y,p2x,p2y,p3x,p3y);
  355. #endif
  356. return;
  357. }
  358. SInt ModeMB (Image *MB_decision, Int i, Int j)
  359. {
  360. Int   width = MB_decision->x;
  361. SInt  *p = (SInt *)GetImageData(MB_decision);
  362. return p[width*j+i];
  363. }