You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

894 lines
26KB

  1. /*
  2. * Misc image conversion routines
  3. * Copyright (c) 2001, 2002, 2003 Fabrice Bellard
  4. *
  5. * This file is part of FFmpeg.
  6. *
  7. * FFmpeg is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU Lesser General Public
  9. * License as published by the Free Software Foundation; either
  10. * version 2.1 of the License, or (at your option) any later version.
  11. *
  12. * FFmpeg is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with FFmpeg; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. /**
  22. * @file
  23. * misc image conversion routines
  24. */
  25. /* TODO:
  26. * - write 'ffimg' program to test all the image related stuff
  27. * - move all api to slice based system
  28. * - integrate deinterlacing, postprocessing and scaling in the conversion process
  29. */
  30. #include "avcodec.h"
  31. #include "dsputil.h"
  32. #include "internal.h"
  33. #include "imgconvert.h"
  34. #include "libavutil/colorspace.h"
  35. #include "libavutil/pixdesc.h"
  36. #include "libavutil/imgutils.h"
  37. #if HAVE_MMX && HAVE_YASM
  38. #include "x86/dsputil_mmx.h"
  39. #endif
  40. #define FF_COLOR_RGB 0 /**< RGB color space */
  41. #define FF_COLOR_GRAY 1 /**< gray color space */
  42. #define FF_COLOR_YUV 2 /**< YUV color space. 16 <= Y <= 235, 16 <= U, V <= 240 */
  43. #define FF_COLOR_YUV_JPEG 3 /**< YUV color space. 0 <= Y <= 255, 0 <= U, V <= 255 */
  44. #if HAVE_MMX && HAVE_YASM
  45. #define deinterlace_line_inplace ff_deinterlace_line_inplace_mmx
  46. #define deinterlace_line ff_deinterlace_line_mmx
  47. #else
  48. #define deinterlace_line_inplace deinterlace_line_inplace_c
  49. #define deinterlace_line deinterlace_line_c
  50. #endif
  51. #define pixdesc_has_alpha(pixdesc) \
  52. ((pixdesc)->nb_components == 2 || (pixdesc)->nb_components == 4 || (pixdesc)->flags & PIX_FMT_PAL)
  53. typedef struct PixFmtInfo {
  54. uint8_t color_type; /**< color type (see FF_COLOR_xxx constants) */
  55. uint8_t padded_size; /**< padded size in bits if different from the non-padded size */
  56. } PixFmtInfo;
  57. /* this table gives more information about formats */
  58. static const PixFmtInfo pix_fmt_info[PIX_FMT_NB] = {
  59. /* YUV formats */
  60. [PIX_FMT_YUV420P] = {
  61. .color_type = FF_COLOR_YUV,
  62. },
  63. [PIX_FMT_YUV422P] = {
  64. .color_type = FF_COLOR_YUV,
  65. },
  66. [PIX_FMT_YUV444P] = {
  67. .color_type = FF_COLOR_YUV,
  68. },
  69. [PIX_FMT_YUYV422] = {
  70. .color_type = FF_COLOR_YUV,
  71. },
  72. [PIX_FMT_UYVY422] = {
  73. .color_type = FF_COLOR_YUV,
  74. },
  75. [PIX_FMT_YUV410P] = {
  76. .color_type = FF_COLOR_YUV,
  77. },
  78. [PIX_FMT_YUV411P] = {
  79. .color_type = FF_COLOR_YUV,
  80. },
  81. [PIX_FMT_YUV440P] = {
  82. .color_type = FF_COLOR_YUV,
  83. },
  84. [PIX_FMT_YUV420P16LE] = {
  85. .color_type = FF_COLOR_YUV,
  86. },
  87. [PIX_FMT_YUV422P16LE] = {
  88. .color_type = FF_COLOR_YUV,
  89. },
  90. [PIX_FMT_YUV444P16LE] = {
  91. .color_type = FF_COLOR_YUV,
  92. },
  93. [PIX_FMT_YUV420P16BE] = {
  94. .color_type = FF_COLOR_YUV,
  95. },
  96. [PIX_FMT_YUV422P16BE] = {
  97. .color_type = FF_COLOR_YUV,
  98. },
  99. [PIX_FMT_YUV444P16BE] = {
  100. .color_type = FF_COLOR_YUV,
  101. },
  102. /* YUV formats with alpha plane */
  103. [PIX_FMT_YUVA420P] = {
  104. .color_type = FF_COLOR_YUV,
  105. },
  106. [PIX_FMT_YUVA422P] = {
  107. .color_type = FF_COLOR_YUV,
  108. },
  109. [PIX_FMT_YUVA444P] = {
  110. .color_type = FF_COLOR_YUV,
  111. },
  112. /* JPEG YUV */
  113. [PIX_FMT_YUVJ420P] = {
  114. .color_type = FF_COLOR_YUV_JPEG,
  115. },
  116. [PIX_FMT_YUVJ422P] = {
  117. .color_type = FF_COLOR_YUV_JPEG,
  118. },
  119. [PIX_FMT_YUVJ444P] = {
  120. .color_type = FF_COLOR_YUV_JPEG,
  121. },
  122. [PIX_FMT_YUVJ440P] = {
  123. .color_type = FF_COLOR_YUV_JPEG,
  124. },
  125. /* RGB formats */
  126. [PIX_FMT_RGB24] = {
  127. .color_type = FF_COLOR_RGB,
  128. },
  129. [PIX_FMT_BGR24] = {
  130. .color_type = FF_COLOR_RGB,
  131. },
  132. [PIX_FMT_ARGB] = {
  133. .color_type = FF_COLOR_RGB,
  134. },
  135. [PIX_FMT_RGB48BE] = {
  136. .color_type = FF_COLOR_RGB,
  137. },
  138. [PIX_FMT_RGB48LE] = {
  139. .color_type = FF_COLOR_RGB,
  140. },
  141. [PIX_FMT_RGBA64BE] = {
  142. .color_type = FF_COLOR_RGB,
  143. },
  144. [PIX_FMT_RGBA64LE] = {
  145. .color_type = FF_COLOR_RGB,
  146. },
  147. [PIX_FMT_RGB565BE] = {
  148. .color_type = FF_COLOR_RGB,
  149. },
  150. [PIX_FMT_RGB565LE] = {
  151. .color_type = FF_COLOR_RGB,
  152. },
  153. [PIX_FMT_RGB555BE] = {
  154. .color_type = FF_COLOR_RGB,
  155. .padded_size = 16,
  156. },
  157. [PIX_FMT_RGB555LE] = {
  158. .color_type = FF_COLOR_RGB,
  159. .padded_size = 16,
  160. },
  161. [PIX_FMT_RGB444BE] = {
  162. .color_type = FF_COLOR_RGB,
  163. .padded_size = 16,
  164. },
  165. [PIX_FMT_RGB444LE] = {
  166. .color_type = FF_COLOR_RGB,
  167. .padded_size = 16,
  168. },
  169. /* gray / mono formats */
  170. [PIX_FMT_GRAY16BE] = {
  171. .color_type = FF_COLOR_GRAY,
  172. },
  173. [PIX_FMT_GRAY16LE] = {
  174. .color_type = FF_COLOR_GRAY,
  175. },
  176. [PIX_FMT_GRAY8] = {
  177. .color_type = FF_COLOR_GRAY,
  178. },
  179. [PIX_FMT_GRAY8A] = {
  180. .color_type = FF_COLOR_GRAY,
  181. },
  182. [PIX_FMT_MONOWHITE] = {
  183. .color_type = FF_COLOR_GRAY,
  184. },
  185. [PIX_FMT_MONOBLACK] = {
  186. .color_type = FF_COLOR_GRAY,
  187. },
  188. /* paletted formats */
  189. [PIX_FMT_PAL8] = {
  190. .color_type = FF_COLOR_RGB,
  191. },
  192. [PIX_FMT_UYYVYY411] = {
  193. .color_type = FF_COLOR_YUV,
  194. },
  195. [PIX_FMT_ABGR] = {
  196. .color_type = FF_COLOR_RGB,
  197. },
  198. [PIX_FMT_BGR48BE] = {
  199. .color_type = FF_COLOR_RGB,
  200. },
  201. [PIX_FMT_BGR48LE] = {
  202. .color_type = FF_COLOR_RGB,
  203. },
  204. [PIX_FMT_BGRA64BE] = {
  205. .color_type = FF_COLOR_RGB,
  206. },
  207. [PIX_FMT_BGRA64LE] = {
  208. .color_type = FF_COLOR_RGB,
  209. },
  210. [PIX_FMT_BGR565BE] = {
  211. .color_type = FF_COLOR_RGB,
  212. .padded_size = 16,
  213. },
  214. [PIX_FMT_BGR565LE] = {
  215. .color_type = FF_COLOR_RGB,
  216. .padded_size = 16,
  217. },
  218. [PIX_FMT_BGR555BE] = {
  219. .color_type = FF_COLOR_RGB,
  220. .padded_size = 16,
  221. },
  222. [PIX_FMT_BGR555LE] = {
  223. .color_type = FF_COLOR_RGB,
  224. .padded_size = 16,
  225. },
  226. [PIX_FMT_BGR444BE] = {
  227. .color_type = FF_COLOR_RGB,
  228. .padded_size = 16,
  229. },
  230. [PIX_FMT_BGR444LE] = {
  231. .color_type = FF_COLOR_RGB,
  232. .padded_size = 16,
  233. },
  234. [PIX_FMT_RGB8] = {
  235. .color_type = FF_COLOR_RGB,
  236. },
  237. [PIX_FMT_RGB4] = {
  238. .color_type = FF_COLOR_RGB,
  239. },
  240. [PIX_FMT_RGB4_BYTE] = {
  241. .color_type = FF_COLOR_RGB,
  242. .padded_size = 8,
  243. },
  244. [PIX_FMT_BGR8] = {
  245. .color_type = FF_COLOR_RGB,
  246. },
  247. [PIX_FMT_BGR4] = {
  248. .color_type = FF_COLOR_RGB,
  249. },
  250. [PIX_FMT_BGR4_BYTE] = {
  251. .color_type = FF_COLOR_RGB,
  252. .padded_size = 8,
  253. },
  254. [PIX_FMT_NV12] = {
  255. .color_type = FF_COLOR_YUV,
  256. },
  257. [PIX_FMT_NV21] = {
  258. .color_type = FF_COLOR_YUV,
  259. },
  260. [PIX_FMT_BGRA] = {
  261. .color_type = FF_COLOR_RGB,
  262. },
  263. [PIX_FMT_RGBA] = {
  264. .color_type = FF_COLOR_RGB,
  265. },
  266. };
  267. void avcodec_get_chroma_sub_sample(enum PixelFormat pix_fmt, int *h_shift, int *v_shift)
  268. {
  269. *h_shift = av_pix_fmt_descriptors[pix_fmt].log2_chroma_w;
  270. *v_shift = av_pix_fmt_descriptors[pix_fmt].log2_chroma_h;
  271. }
  272. int ff_is_hwaccel_pix_fmt(enum PixelFormat pix_fmt)
  273. {
  274. return av_pix_fmt_descriptors[pix_fmt].flags & PIX_FMT_HWACCEL;
  275. }
  276. int avpicture_fill(AVPicture *picture, uint8_t *ptr,
  277. enum PixelFormat pix_fmt, int width, int height)
  278. {
  279. int ret;
  280. if ((ret = av_image_check_size(width, height, 0, NULL)) < 0)
  281. return ret;
  282. if ((ret = av_image_fill_linesizes(picture->linesize, pix_fmt, width)) < 0)
  283. return ret;
  284. return av_image_fill_pointers(picture->data, pix_fmt, height, ptr, picture->linesize);
  285. }
  286. int avpicture_layout(const AVPicture* src, enum PixelFormat pix_fmt, int width, int height,
  287. unsigned char *dest, int dest_size)
  288. {
  289. int i, j, nb_planes = 0, linesizes[4];
  290. const AVPixFmtDescriptor *desc = &av_pix_fmt_descriptors[pix_fmt];
  291. int size = avpicture_get_size(pix_fmt, width, height);
  292. if (size > dest_size || size < 0)
  293. return AVERROR(EINVAL);
  294. for (i = 0; i < desc->nb_components; i++)
  295. nb_planes = FFMAX(desc->comp[i].plane, nb_planes);
  296. nb_planes++;
  297. av_image_fill_linesizes(linesizes, pix_fmt, width);
  298. for (i = 0; i < nb_planes; i++) {
  299. int h, shift = (i == 1 || i == 2) ? desc->log2_chroma_h : 0;
  300. const unsigned char *s = src->data[i];
  301. h = (height + (1 << shift) - 1) >> shift;
  302. for (j = 0; j < h; j++) {
  303. memcpy(dest, s, linesizes[i]);
  304. dest += linesizes[i];
  305. s += src->linesize[i];
  306. }
  307. }
  308. if (desc->flags & PIX_FMT_PAL) {
  309. uint32_t *d32 = (unsigned char *)(((size_t)dest + 3) & ~3);
  310. for (i = 0; i<256; i++)
  311. AV_WL32(d32 + i, AV_RN32(src->data[1] + 4*i));
  312. }
  313. return size;
  314. }
  315. int avpicture_get_size(enum PixelFormat pix_fmt, int width, int height)
  316. {
  317. AVPicture dummy_pict;
  318. if(av_image_check_size(width, height, 0, NULL))
  319. return -1;
  320. if (av_pix_fmt_descriptors[pix_fmt].flags & PIX_FMT_PSEUDOPAL)
  321. // do not include palette for these pseudo-paletted formats
  322. return width * height;
  323. return avpicture_fill(&dummy_pict, NULL, pix_fmt, width, height);
  324. }
  325. static int get_pix_fmt_depth(int *min, int *max, enum PixelFormat pix_fmt)
  326. {
  327. const AVPixFmtDescriptor *desc = &av_pix_fmt_descriptors[pix_fmt];
  328. int i;
  329. if (!desc->nb_components) {
  330. *min = *max = 0;
  331. return AVERROR(EINVAL);
  332. }
  333. *min = INT_MAX, *max = -INT_MAX;
  334. for (i = 0; i < desc->nb_components; i++) {
  335. *min = FFMIN(desc->comp[i].depth_minus1+1, *min);
  336. *max = FFMAX(desc->comp[i].depth_minus1+1, *max);
  337. }
  338. return 0;
  339. }
  340. int avcodec_get_pix_fmt_loss(enum PixelFormat dst_pix_fmt, enum PixelFormat src_pix_fmt,
  341. int has_alpha)
  342. {
  343. const PixFmtInfo *pf, *ps;
  344. const AVPixFmtDescriptor *src_desc;
  345. const AVPixFmtDescriptor *dst_desc;
  346. int src_min_depth, src_max_depth, dst_min_depth, dst_max_depth;
  347. int ret, loss;
  348. if (dst_pix_fmt >= PIX_FMT_NB || dst_pix_fmt <= PIX_FMT_NONE)
  349. return ~0;
  350. src_desc = &av_pix_fmt_descriptors[src_pix_fmt];
  351. dst_desc = &av_pix_fmt_descriptors[dst_pix_fmt];
  352. ps = &pix_fmt_info[src_pix_fmt];
  353. /* compute loss */
  354. loss = 0;
  355. if ((ret = get_pix_fmt_depth(&src_min_depth, &src_max_depth, src_pix_fmt)) < 0)
  356. return ret;
  357. if ((ret = get_pix_fmt_depth(&dst_min_depth, &dst_max_depth, dst_pix_fmt)) < 0)
  358. return ret;
  359. if (dst_min_depth < src_min_depth ||
  360. dst_max_depth < src_max_depth)
  361. loss |= FF_LOSS_DEPTH;
  362. if (dst_desc->log2_chroma_w > src_desc->log2_chroma_w ||
  363. dst_desc->log2_chroma_h > src_desc->log2_chroma_h)
  364. loss |= FF_LOSS_RESOLUTION;
  365. pf = &pix_fmt_info[dst_pix_fmt];
  366. switch(pf->color_type) {
  367. case FF_COLOR_RGB:
  368. if (ps->color_type != FF_COLOR_RGB &&
  369. ps->color_type != FF_COLOR_GRAY)
  370. loss |= FF_LOSS_COLORSPACE;
  371. break;
  372. case FF_COLOR_GRAY:
  373. if (ps->color_type != FF_COLOR_GRAY)
  374. loss |= FF_LOSS_COLORSPACE;
  375. break;
  376. case FF_COLOR_YUV:
  377. if (ps->color_type != FF_COLOR_YUV)
  378. loss |= FF_LOSS_COLORSPACE;
  379. break;
  380. case FF_COLOR_YUV_JPEG:
  381. if (ps->color_type != FF_COLOR_YUV_JPEG &&
  382. ps->color_type != FF_COLOR_YUV &&
  383. ps->color_type != FF_COLOR_GRAY)
  384. loss |= FF_LOSS_COLORSPACE;
  385. break;
  386. default:
  387. /* fail safe test */
  388. if (ps->color_type != pf->color_type)
  389. loss |= FF_LOSS_COLORSPACE;
  390. break;
  391. }
  392. if (pf->color_type == FF_COLOR_GRAY &&
  393. ps->color_type != FF_COLOR_GRAY)
  394. loss |= FF_LOSS_CHROMA;
  395. if (!pixdesc_has_alpha(dst_desc) && (pixdesc_has_alpha(src_desc) && has_alpha))
  396. loss |= FF_LOSS_ALPHA;
  397. if (dst_pix_fmt == PIX_FMT_PAL8 &&
  398. (src_pix_fmt != PIX_FMT_PAL8 && (ps->color_type != FF_COLOR_GRAY || (pixdesc_has_alpha(src_desc) && has_alpha))))
  399. loss |= FF_LOSS_COLORQUANT;
  400. return loss;
  401. }
  402. static int avg_bits_per_pixel(enum PixelFormat pix_fmt)
  403. {
  404. const PixFmtInfo *info = &pix_fmt_info[pix_fmt];
  405. const AVPixFmtDescriptor *desc = &av_pix_fmt_descriptors[pix_fmt];
  406. return info->padded_size ?
  407. info->padded_size : av_get_bits_per_pixel(desc);
  408. }
  409. enum PixelFormat avcodec_find_best_pix_fmt(int64_t pix_fmt_mask, enum PixelFormat src_pix_fmt,
  410. int has_alpha, int *loss_ptr)
  411. {
  412. enum PixelFormat dst_pix_fmt;
  413. int i;
  414. if (loss_ptr) /* all losses count (for backward compatibility) */
  415. *loss_ptr = 0;
  416. dst_pix_fmt = PIX_FMT_NONE; /* so first iteration doesn't have to be treated special */
  417. for(i = 0; i< FFMIN(PIX_FMT_NB, 64); i++){
  418. if (pix_fmt_mask & (1ULL << i))
  419. dst_pix_fmt = avcodec_find_best_pix_fmt2(dst_pix_fmt, i, src_pix_fmt, has_alpha, loss_ptr);
  420. }
  421. return dst_pix_fmt;
  422. }
  423. enum PixelFormat avcodec_find_best_pix_fmt2(enum PixelFormat dst_pix_fmt1, enum PixelFormat dst_pix_fmt2,
  424. enum PixelFormat src_pix_fmt, int has_alpha, int *loss_ptr)
  425. {
  426. enum PixelFormat dst_pix_fmt;
  427. int loss1, loss2, loss_order1, loss_order2, i, loss_mask;
  428. static const int loss_mask_order[] = {
  429. ~0, /* no loss first */
  430. ~FF_LOSS_ALPHA,
  431. ~FF_LOSS_RESOLUTION,
  432. ~(FF_LOSS_COLORSPACE | FF_LOSS_RESOLUTION),
  433. ~FF_LOSS_COLORQUANT,
  434. ~FF_LOSS_DEPTH,
  435. ~(FF_LOSS_DEPTH|FF_LOSS_COLORSPACE),
  436. ~(FF_LOSS_RESOLUTION | FF_LOSS_DEPTH | FF_LOSS_COLORSPACE | FF_LOSS_ALPHA |
  437. FF_LOSS_COLORQUANT | FF_LOSS_CHROMA),
  438. 0x80000, //non zero entry that combines all loss variants including future additions
  439. 0,
  440. };
  441. loss_mask= loss_ptr?~*loss_ptr:~0; /* use loss mask if provided */
  442. dst_pix_fmt = PIX_FMT_NONE;
  443. loss1 = avcodec_get_pix_fmt_loss(dst_pix_fmt1, src_pix_fmt, has_alpha) & loss_mask;
  444. loss2 = avcodec_get_pix_fmt_loss(dst_pix_fmt2, src_pix_fmt, has_alpha) & loss_mask;
  445. /* try with successive loss */
  446. for(i = 0;loss_mask_order[i] != 0 && dst_pix_fmt == PIX_FMT_NONE;i++) {
  447. loss_order1 = loss1 & loss_mask_order[i];
  448. loss_order2 = loss2 & loss_mask_order[i];
  449. if (loss_order1 == 0 && loss_order2 == 0){ /* use format with smallest depth */
  450. dst_pix_fmt = avg_bits_per_pixel(dst_pix_fmt2) < avg_bits_per_pixel(dst_pix_fmt1) ? dst_pix_fmt2 : dst_pix_fmt1;
  451. } else if (loss_order1 == 0 || loss_order2 == 0) { /* use format with no loss */
  452. dst_pix_fmt = loss_order2 ? dst_pix_fmt1 : dst_pix_fmt2;
  453. }
  454. }
  455. if (loss_ptr)
  456. *loss_ptr = avcodec_get_pix_fmt_loss(dst_pix_fmt, src_pix_fmt, has_alpha);
  457. return dst_pix_fmt;
  458. }
  459. void av_picture_copy(AVPicture *dst, const AVPicture *src,
  460. enum PixelFormat pix_fmt, int width, int height)
  461. {
  462. av_image_copy(dst->data, dst->linesize, src->data,
  463. src->linesize, pix_fmt, width, height);
  464. }
  465. /* 2x2 -> 1x1 */
  466. void ff_shrink22(uint8_t *dst, int dst_wrap,
  467. const uint8_t *src, int src_wrap,
  468. int width, int height)
  469. {
  470. int w;
  471. const uint8_t *s1, *s2;
  472. uint8_t *d;
  473. for(;height > 0; height--) {
  474. s1 = src;
  475. s2 = s1 + src_wrap;
  476. d = dst;
  477. for(w = width;w >= 4; w-=4) {
  478. d[0] = (s1[0] + s1[1] + s2[0] + s2[1] + 2) >> 2;
  479. d[1] = (s1[2] + s1[3] + s2[2] + s2[3] + 2) >> 2;
  480. d[2] = (s1[4] + s1[5] + s2[4] + s2[5] + 2) >> 2;
  481. d[3] = (s1[6] + s1[7] + s2[6] + s2[7] + 2) >> 2;
  482. s1 += 8;
  483. s2 += 8;
  484. d += 4;
  485. }
  486. for(;w > 0; w--) {
  487. d[0] = (s1[0] + s1[1] + s2[0] + s2[1] + 2) >> 2;
  488. s1 += 2;
  489. s2 += 2;
  490. d++;
  491. }
  492. src += 2 * src_wrap;
  493. dst += dst_wrap;
  494. }
  495. }
  496. /* 4x4 -> 1x1 */
  497. void ff_shrink44(uint8_t *dst, int dst_wrap,
  498. const uint8_t *src, int src_wrap,
  499. int width, int height)
  500. {
  501. int w;
  502. const uint8_t *s1, *s2, *s3, *s4;
  503. uint8_t *d;
  504. for(;height > 0; height--) {
  505. s1 = src;
  506. s2 = s1 + src_wrap;
  507. s3 = s2 + src_wrap;
  508. s4 = s3 + src_wrap;
  509. d = dst;
  510. for(w = width;w > 0; w--) {
  511. d[0] = (s1[0] + s1[1] + s1[2] + s1[3] +
  512. s2[0] + s2[1] + s2[2] + s2[3] +
  513. s3[0] + s3[1] + s3[2] + s3[3] +
  514. s4[0] + s4[1] + s4[2] + s4[3] + 8) >> 4;
  515. s1 += 4;
  516. s2 += 4;
  517. s3 += 4;
  518. s4 += 4;
  519. d++;
  520. }
  521. src += 4 * src_wrap;
  522. dst += dst_wrap;
  523. }
  524. }
  525. /* 8x8 -> 1x1 */
  526. void ff_shrink88(uint8_t *dst, int dst_wrap,
  527. const uint8_t *src, int src_wrap,
  528. int width, int height)
  529. {
  530. int w, i;
  531. for(;height > 0; height--) {
  532. for(w = width;w > 0; w--) {
  533. int tmp=0;
  534. for(i=0; i<8; i++){
  535. tmp += src[0] + src[1] + src[2] + src[3] + src[4] + src[5] + src[6] + src[7];
  536. src += src_wrap;
  537. }
  538. *(dst++) = (tmp + 32)>>6;
  539. src += 8 - 8*src_wrap;
  540. }
  541. src += 8*src_wrap - 8*width;
  542. dst += dst_wrap - width;
  543. }
  544. }
  545. int avpicture_alloc(AVPicture *picture,
  546. enum PixelFormat pix_fmt, int width, int height)
  547. {
  548. int ret;
  549. if ((ret = av_image_alloc(picture->data, picture->linesize, width, height, pix_fmt, 1)) < 0) {
  550. memset(picture, 0, sizeof(AVPicture));
  551. return ret;
  552. }
  553. return 0;
  554. }
  555. void avpicture_free(AVPicture *picture)
  556. {
  557. av_free(picture->data[0]);
  558. }
  559. /* return true if yuv planar */
  560. static inline int is_yuv_planar(enum PixelFormat fmt)
  561. {
  562. const PixFmtInfo *info = &pix_fmt_info[fmt];
  563. const AVPixFmtDescriptor *desc = &av_pix_fmt_descriptors[fmt];
  564. int i;
  565. int planes[4] = { 0 };
  566. if (info->color_type != FF_COLOR_YUV &&
  567. info->color_type != FF_COLOR_YUV_JPEG)
  568. return 0;
  569. /* set the used planes */
  570. for (i = 0; i < desc->nb_components; i++)
  571. planes[desc->comp[i].plane] = 1;
  572. /* if there is an unused plane, the format is not planar */
  573. for (i = 0; i < desc->nb_components; i++)
  574. if (!planes[i])
  575. return 0;
  576. return 1;
  577. }
  578. int av_picture_crop(AVPicture *dst, const AVPicture *src,
  579. enum PixelFormat pix_fmt, int top_band, int left_band)
  580. {
  581. int y_shift;
  582. int x_shift;
  583. if (pix_fmt < 0 || pix_fmt >= PIX_FMT_NB)
  584. return -1;
  585. y_shift = av_pix_fmt_descriptors[pix_fmt].log2_chroma_h;
  586. x_shift = av_pix_fmt_descriptors[pix_fmt].log2_chroma_w;
  587. if (is_yuv_planar(pix_fmt)) {
  588. dst->data[0] = src->data[0] + (top_band * src->linesize[0]) + left_band;
  589. dst->data[1] = src->data[1] + ((top_band >> y_shift) * src->linesize[1]) + (left_band >> x_shift);
  590. dst->data[2] = src->data[2] + ((top_band >> y_shift) * src->linesize[2]) + (left_band >> x_shift);
  591. } else{
  592. if(top_band % (1<<y_shift) || left_band % (1<<x_shift))
  593. return -1;
  594. if(left_band) //FIXME add support for this too
  595. return -1;
  596. dst->data[0] = src->data[0] + (top_band * src->linesize[0]) + left_band;
  597. }
  598. dst->linesize[0] = src->linesize[0];
  599. dst->linesize[1] = src->linesize[1];
  600. dst->linesize[2] = src->linesize[2];
  601. return 0;
  602. }
  603. int av_picture_pad(AVPicture *dst, const AVPicture *src, int height, int width,
  604. enum PixelFormat pix_fmt, int padtop, int padbottom, int padleft, int padright,
  605. int *color)
  606. {
  607. uint8_t *optr;
  608. int y_shift;
  609. int x_shift;
  610. int yheight;
  611. int i, y;
  612. if (pix_fmt < 0 || pix_fmt >= PIX_FMT_NB ||
  613. !is_yuv_planar(pix_fmt)) return -1;
  614. for (i = 0; i < 3; i++) {
  615. x_shift = i ? av_pix_fmt_descriptors[pix_fmt].log2_chroma_w : 0;
  616. y_shift = i ? av_pix_fmt_descriptors[pix_fmt].log2_chroma_h : 0;
  617. if (padtop || padleft) {
  618. memset(dst->data[i], color[i],
  619. dst->linesize[i] * (padtop >> y_shift) + (padleft >> x_shift));
  620. }
  621. if (padleft || padright) {
  622. optr = dst->data[i] + dst->linesize[i] * (padtop >> y_shift) +
  623. (dst->linesize[i] - (padright >> x_shift));
  624. yheight = (height - 1 - (padtop + padbottom)) >> y_shift;
  625. for (y = 0; y < yheight; y++) {
  626. memset(optr, color[i], (padleft + padright) >> x_shift);
  627. optr += dst->linesize[i];
  628. }
  629. }
  630. if (src) { /* first line */
  631. uint8_t *iptr = src->data[i];
  632. optr = dst->data[i] + dst->linesize[i] * (padtop >> y_shift) +
  633. (padleft >> x_shift);
  634. memcpy(optr, iptr, (width - padleft - padright) >> x_shift);
  635. iptr += src->linesize[i];
  636. optr = dst->data[i] + dst->linesize[i] * (padtop >> y_shift) +
  637. (dst->linesize[i] - (padright >> x_shift));
  638. yheight = (height - 1 - (padtop + padbottom)) >> y_shift;
  639. for (y = 0; y < yheight; y++) {
  640. memset(optr, color[i], (padleft + padright) >> x_shift);
  641. memcpy(optr + ((padleft + padright) >> x_shift), iptr,
  642. (width - padleft - padright) >> x_shift);
  643. iptr += src->linesize[i];
  644. optr += dst->linesize[i];
  645. }
  646. }
  647. if (padbottom || padright) {
  648. optr = dst->data[i] + dst->linesize[i] *
  649. ((height - padbottom) >> y_shift) - (padright >> x_shift);
  650. memset(optr, color[i],dst->linesize[i] *
  651. (padbottom >> y_shift) + (padright >> x_shift));
  652. }
  653. }
  654. return 0;
  655. }
  656. #if !(HAVE_MMX && HAVE_YASM)
  657. /* filter parameters: [-1 4 2 4 -1] // 8 */
  658. static void deinterlace_line_c(uint8_t *dst,
  659. const uint8_t *lum_m4, const uint8_t *lum_m3,
  660. const uint8_t *lum_m2, const uint8_t *lum_m1,
  661. const uint8_t *lum,
  662. int size)
  663. {
  664. uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
  665. int sum;
  666. for(;size > 0;size--) {
  667. sum = -lum_m4[0];
  668. sum += lum_m3[0] << 2;
  669. sum += lum_m2[0] << 1;
  670. sum += lum_m1[0] << 2;
  671. sum += -lum[0];
  672. dst[0] = cm[(sum + 4) >> 3];
  673. lum_m4++;
  674. lum_m3++;
  675. lum_m2++;
  676. lum_m1++;
  677. lum++;
  678. dst++;
  679. }
  680. }
  681. static void deinterlace_line_inplace_c(uint8_t *lum_m4, uint8_t *lum_m3,
  682. uint8_t *lum_m2, uint8_t *lum_m1,
  683. uint8_t *lum, int size)
  684. {
  685. uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
  686. int sum;
  687. for(;size > 0;size--) {
  688. sum = -lum_m4[0];
  689. sum += lum_m3[0] << 2;
  690. sum += lum_m2[0] << 1;
  691. lum_m4[0]=lum_m2[0];
  692. sum += lum_m1[0] << 2;
  693. sum += -lum[0];
  694. lum_m2[0] = cm[(sum + 4) >> 3];
  695. lum_m4++;
  696. lum_m3++;
  697. lum_m2++;
  698. lum_m1++;
  699. lum++;
  700. }
  701. }
  702. #endif
  703. /* deinterlacing : 2 temporal taps, 3 spatial taps linear filter. The
  704. top field is copied as is, but the bottom field is deinterlaced
  705. against the top field. */
  706. static void deinterlace_bottom_field(uint8_t *dst, int dst_wrap,
  707. const uint8_t *src1, int src_wrap,
  708. int width, int height)
  709. {
  710. const uint8_t *src_m2, *src_m1, *src_0, *src_p1, *src_p2;
  711. int y;
  712. src_m2 = src1;
  713. src_m1 = src1;
  714. src_0=&src_m1[src_wrap];
  715. src_p1=&src_0[src_wrap];
  716. src_p2=&src_p1[src_wrap];
  717. for(y=0;y<(height-2);y+=2) {
  718. memcpy(dst,src_m1,width);
  719. dst += dst_wrap;
  720. deinterlace_line(dst,src_m2,src_m1,src_0,src_p1,src_p2,width);
  721. src_m2 = src_0;
  722. src_m1 = src_p1;
  723. src_0 = src_p2;
  724. src_p1 += 2*src_wrap;
  725. src_p2 += 2*src_wrap;
  726. dst += dst_wrap;
  727. }
  728. memcpy(dst,src_m1,width);
  729. dst += dst_wrap;
  730. /* do last line */
  731. deinterlace_line(dst,src_m2,src_m1,src_0,src_0,src_0,width);
  732. }
  733. static void deinterlace_bottom_field_inplace(uint8_t *src1, int src_wrap,
  734. int width, int height)
  735. {
  736. uint8_t *src_m1, *src_0, *src_p1, *src_p2;
  737. int y;
  738. uint8_t *buf;
  739. buf = av_malloc(width);
  740. src_m1 = src1;
  741. memcpy(buf,src_m1,width);
  742. src_0=&src_m1[src_wrap];
  743. src_p1=&src_0[src_wrap];
  744. src_p2=&src_p1[src_wrap];
  745. for(y=0;y<(height-2);y+=2) {
  746. deinterlace_line_inplace(buf,src_m1,src_0,src_p1,src_p2,width);
  747. src_m1 = src_p1;
  748. src_0 = src_p2;
  749. src_p1 += 2*src_wrap;
  750. src_p2 += 2*src_wrap;
  751. }
  752. /* do last line */
  753. deinterlace_line_inplace(buf,src_m1,src_0,src_0,src_0,width);
  754. av_free(buf);
  755. }
  756. int avpicture_deinterlace(AVPicture *dst, const AVPicture *src,
  757. enum PixelFormat pix_fmt, int width, int height)
  758. {
  759. int i;
  760. if (pix_fmt != PIX_FMT_YUV420P &&
  761. pix_fmt != PIX_FMT_YUVJ420P &&
  762. pix_fmt != PIX_FMT_YUV422P &&
  763. pix_fmt != PIX_FMT_YUVJ422P &&
  764. pix_fmt != PIX_FMT_YUV444P &&
  765. pix_fmt != PIX_FMT_YUV411P &&
  766. pix_fmt != PIX_FMT_GRAY8)
  767. return -1;
  768. if ((width & 3) != 0 || (height & 3) != 0)
  769. return -1;
  770. for(i=0;i<3;i++) {
  771. if (i == 1) {
  772. switch(pix_fmt) {
  773. case PIX_FMT_YUVJ420P:
  774. case PIX_FMT_YUV420P:
  775. width >>= 1;
  776. height >>= 1;
  777. break;
  778. case PIX_FMT_YUV422P:
  779. case PIX_FMT_YUVJ422P:
  780. width >>= 1;
  781. break;
  782. case PIX_FMT_YUV411P:
  783. width >>= 2;
  784. break;
  785. default:
  786. break;
  787. }
  788. if (pix_fmt == PIX_FMT_GRAY8) {
  789. break;
  790. }
  791. }
  792. if (src == dst) {
  793. deinterlace_bottom_field_inplace(dst->data[i], dst->linesize[i],
  794. width, height);
  795. } else {
  796. deinterlace_bottom_field(dst->data[i],dst->linesize[i],
  797. src->data[i], src->linesize[i],
  798. width, height);
  799. }
  800. }
  801. emms_c();
  802. return 0;
  803. }