swscale.h 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436
  1. /*
  2. * Copyright (C) 2001-2011 Michael Niedermayer <michaelni@gmx.at>
  3. *
  4. * This file is part of FFmpeg.
  5. *
  6. * FFmpeg is free software; you can redistribute it and/or
  7. * modify it under the terms of the GNU Lesser General Public
  8. * License as published by the Free Software Foundation; either
  9. * version 2.1 of the License, or (at your option) any later version.
  10. *
  11. * FFmpeg is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14. * Lesser General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU Lesser General Public
  17. * License along with FFmpeg; if not, write to the Free Software
  18. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19. */
  20. #ifndef SWSCALE_SWSCALE_H
  21. #define SWSCALE_SWSCALE_H
  22. /**
  23. * @file
  24. * @ingroup libsws
  25. * external API header
  26. */
  27. #include <stdint.h>
  28. #include "libavutil/avutil.h"
  29. #include "libavutil/frame.h"
  30. #include "libavutil/log.h"
  31. #include "libavutil/pixfmt.h"
  32. #include "version_major.h"
  33. #ifndef HAVE_AV_CONFIG_H
  34. /* When included as part of the ffmpeg build, only include the major version
  35. * to avoid unnecessary rebuilds. When included externally, keep including
  36. * the full version information. */
  37. #include "version.h"
  38. #endif
  39. /**
  40. * @defgroup libsws libswscale
  41. * Color conversion and scaling library.
  42. *
  43. * @{
  44. *
  45. * Return the LIBSWSCALE_VERSION_INT constant.
  46. */
  47. unsigned swscale_version(void);
  48. /**
  49. * Return the libswscale build-time configuration.
  50. */
  51. const char *swscale_configuration(void);
  52. /**
  53. * Return the libswscale license.
  54. */
  55. const char *swscale_license(void);
  56. /* values for the flags, the stuff on the command line is different */
  57. #define SWS_FAST_BILINEAR 1
  58. #define SWS_BILINEAR 2
  59. #define SWS_BICUBIC 4
  60. #define SWS_X 8
  61. #define SWS_POINT 0x10
  62. #define SWS_AREA 0x20
  63. #define SWS_BICUBLIN 0x40
  64. #define SWS_GAUSS 0x80
  65. #define SWS_SINC 0x100
  66. #define SWS_LANCZOS 0x200
  67. #define SWS_SPLINE 0x400
  68. #define SWS_SRC_V_CHR_DROP_MASK 0x30000
  69. #define SWS_SRC_V_CHR_DROP_SHIFT 16
  70. #define SWS_PARAM_DEFAULT 123456
  71. #define SWS_PRINT_INFO 0x1000
  72. //the following 3 flags are not completely implemented
  73. //internal chrominance subsampling info
  74. #define SWS_FULL_CHR_H_INT 0x2000
  75. //input subsampling info
  76. #define SWS_FULL_CHR_H_INP 0x4000
  77. #define SWS_DIRECT_BGR 0x8000
  78. #define SWS_ACCURATE_RND 0x40000
  79. #define SWS_BITEXACT 0x80000
  80. #define SWS_ERROR_DIFFUSION 0x800000
  81. #define SWS_MAX_REDUCE_CUTOFF 0.002
  82. #define SWS_CS_ITU709 1
  83. #define SWS_CS_FCC 4
  84. #define SWS_CS_ITU601 5
  85. #define SWS_CS_ITU624 5
  86. #define SWS_CS_SMPTE170M 5
  87. #define SWS_CS_SMPTE240M 7
  88. #define SWS_CS_DEFAULT 5
  89. #define SWS_CS_BT2020 9
  90. /**
  91. * Return a pointer to yuv<->rgb coefficients for the given colorspace
  92. * suitable for sws_setColorspaceDetails().
  93. *
  94. * @param colorspace One of the SWS_CS_* macros. If invalid,
  95. * SWS_CS_DEFAULT is used.
  96. */
  97. const int *sws_getCoefficients(int colorspace);
  98. // when used for filters they must have an odd number of elements
  99. // coeffs cannot be shared between vectors
  100. typedef struct SwsVector {
  101. double *coeff; ///< pointer to the list of coefficients
  102. int length; ///< number of coefficients in the vector
  103. } SwsVector;
  104. // vectors can be shared
  105. typedef struct SwsFilter {
  106. SwsVector *lumH;
  107. SwsVector *lumV;
  108. SwsVector *chrH;
  109. SwsVector *chrV;
  110. } SwsFilter;
  111. struct SwsContext;
  112. /**
  113. * Return a positive value if pix_fmt is a supported input format, 0
  114. * otherwise.
  115. */
  116. int sws_isSupportedInput(enum AVPixelFormat pix_fmt);
  117. /**
  118. * Return a positive value if pix_fmt is a supported output format, 0
  119. * otherwise.
  120. */
  121. int sws_isSupportedOutput(enum AVPixelFormat pix_fmt);
  122. /**
  123. * @param[in] pix_fmt the pixel format
  124. * @return a positive value if an endianness conversion for pix_fmt is
  125. * supported, 0 otherwise.
  126. */
  127. int sws_isSupportedEndiannessConversion(enum AVPixelFormat pix_fmt);
  128. /**
  129. * Allocate an empty SwsContext. This must be filled and passed to
  130. * sws_init_context(). For filling see AVOptions, options.c and
  131. * sws_setColorspaceDetails().
  132. */
  133. struct SwsContext *sws_alloc_context(void);
  134. /**
  135. * Initialize the swscaler context sws_context.
  136. *
  137. * @return zero or positive value on success, a negative value on
  138. * error
  139. */
  140. av_warn_unused_result
  141. int sws_init_context(struct SwsContext *sws_context, SwsFilter *srcFilter, SwsFilter *dstFilter);
  142. /**
  143. * Free the swscaler context swsContext.
  144. * If swsContext is NULL, then does nothing.
  145. */
  146. void sws_freeContext(struct SwsContext *swsContext);
  147. /**
  148. * Allocate and return an SwsContext. You need it to perform
  149. * scaling/conversion operations using sws_scale().
  150. *
  151. * @param srcW the width of the source image
  152. * @param srcH the height of the source image
  153. * @param srcFormat the source image format
  154. * @param dstW the width of the destination image
  155. * @param dstH the height of the destination image
  156. * @param dstFormat the destination image format
  157. * @param flags specify which algorithm and options to use for rescaling
  158. * @param param extra parameters to tune the used scaler
  159. * For SWS_BICUBIC param[0] and [1] tune the shape of the basis
  160. * function, param[0] tunes f(1) and param[1] f´(1)
  161. * For SWS_GAUSS param[0] tunes the exponent and thus cutoff
  162. * frequency
  163. * For SWS_LANCZOS param[0] tunes the width of the window function
  164. * @return a pointer to an allocated context, or NULL in case of error
  165. * @note this function is to be removed after a saner alternative is
  166. * written
  167. */
  168. struct SwsContext *sws_getContext(int srcW, int srcH, enum AVPixelFormat srcFormat,
  169. int dstW, int dstH, enum AVPixelFormat dstFormat,
  170. int flags, SwsFilter *srcFilter,
  171. SwsFilter *dstFilter, const double *param);
  172. /**
  173. * Scale the image slice in srcSlice and put the resulting scaled
  174. * slice in the image in dst. A slice is a sequence of consecutive
  175. * rows in an image.
  176. *
  177. * Slices have to be provided in sequential order, either in
  178. * top-bottom or bottom-top order. If slices are provided in
  179. * non-sequential order the behavior of the function is undefined.
  180. *
  181. * @param c the scaling context previously created with
  182. * sws_getContext()
  183. * @param srcSlice the array containing the pointers to the planes of
  184. * the source slice
  185. * @param srcStride the array containing the strides for each plane of
  186. * the source image
  187. * @param srcSliceY the position in the source image of the slice to
  188. * process, that is the number (counted starting from
  189. * zero) in the image of the first row of the slice
  190. * @param srcSliceH the height of the source slice, that is the number
  191. * of rows in the slice
  192. * @param dst the array containing the pointers to the planes of
  193. * the destination image
  194. * @param dstStride the array containing the strides for each plane of
  195. * the destination image
  196. * @return the height of the output slice
  197. */
  198. int sws_scale(struct SwsContext *c, const uint8_t *const srcSlice[],
  199. const int srcStride[], int srcSliceY, int srcSliceH,
  200. uint8_t *const dst[], const int dstStride[]);
  201. /**
  202. * Scale source data from src and write the output to dst.
  203. *
  204. * This is merely a convenience wrapper around
  205. * - sws_frame_start()
  206. * - sws_send_slice(0, src->height)
  207. * - sws_receive_slice(0, dst->height)
  208. * - sws_frame_end()
  209. *
  210. * @param c The scaling context
  211. * @param dst The destination frame. See documentation for sws_frame_start() for
  212. * more details.
  213. * @param src The source frame.
  214. *
  215. * @return 0 on success, a negative AVERROR code on failure
  216. */
  217. int sws_scale_frame(struct SwsContext *c, AVFrame *dst, const AVFrame *src);
  218. /**
  219. * Initialize the scaling process for a given pair of source/destination frames.
  220. * Must be called before any calls to sws_send_slice() and sws_receive_slice().
  221. *
  222. * This function will retain references to src and dst, so they must both use
  223. * refcounted buffers (if allocated by the caller, in case of dst).
  224. *
  225. * @param c The scaling context
  226. * @param dst The destination frame.
  227. *
  228. * The data buffers may either be already allocated by the caller or
  229. * left clear, in which case they will be allocated by the scaler.
  230. * The latter may have performance advantages - e.g. in certain cases
  231. * some output planes may be references to input planes, rather than
  232. * copies.
  233. *
  234. * Output data will be written into this frame in successful
  235. * sws_receive_slice() calls.
  236. * @param src The source frame. The data buffers must be allocated, but the
  237. * frame data does not have to be ready at this point. Data
  238. * availability is then signalled by sws_send_slice().
  239. * @return 0 on success, a negative AVERROR code on failure
  240. *
  241. * @see sws_frame_end()
  242. */
  243. int sws_frame_start(struct SwsContext *c, AVFrame *dst, const AVFrame *src);
  244. /**
  245. * Finish the scaling process for a pair of source/destination frames previously
  246. * submitted with sws_frame_start(). Must be called after all sws_send_slice()
  247. * and sws_receive_slice() calls are done, before any new sws_frame_start()
  248. * calls.
  249. *
  250. * @param c The scaling context
  251. */
  252. void sws_frame_end(struct SwsContext *c);
  253. /**
  254. * Indicate that a horizontal slice of input data is available in the source
  255. * frame previously provided to sws_frame_start(). The slices may be provided in
  256. * any order, but may not overlap. For vertically subsampled pixel formats, the
  257. * slices must be aligned according to subsampling.
  258. *
  259. * @param c The scaling context
  260. * @param slice_start first row of the slice
  261. * @param slice_height number of rows in the slice
  262. *
  263. * @return a non-negative number on success, a negative AVERROR code on failure.
  264. */
  265. int sws_send_slice(struct SwsContext *c, unsigned int slice_start,
  266. unsigned int slice_height);
  267. /**
  268. * Request a horizontal slice of the output data to be written into the frame
  269. * previously provided to sws_frame_start().
  270. *
  271. * @param c The scaling context
  272. * @param slice_start first row of the slice; must be a multiple of
  273. * sws_receive_slice_alignment()
  274. * @param slice_height number of rows in the slice; must be a multiple of
  275. * sws_receive_slice_alignment(), except for the last slice
  276. * (i.e. when slice_start+slice_height is equal to output
  277. * frame height)
  278. *
  279. * @return a non-negative number if the data was successfully written into the output
  280. * AVERROR(EAGAIN) if more input data needs to be provided before the
  281. * output can be produced
  282. * another negative AVERROR code on other kinds of scaling failure
  283. */
  284. int sws_receive_slice(struct SwsContext *c, unsigned int slice_start,
  285. unsigned int slice_height);
  286. /**
  287. * Get the alignment required for slices
  288. *
  289. * @param c The scaling context
  290. * @return alignment required for output slices requested with sws_receive_slice().
  291. * Slice offsets and sizes passed to sws_receive_slice() must be
  292. * multiples of the value returned from this function.
  293. */
  294. unsigned int sws_receive_slice_alignment(const struct SwsContext *c);
  295. /**
  296. * @param c the scaling context
  297. * @param dstRange flag indicating the while-black range of the output (1=jpeg / 0=mpeg)
  298. * @param srcRange flag indicating the while-black range of the input (1=jpeg / 0=mpeg)
  299. * @param table the yuv2rgb coefficients describing the output yuv space, normally ff_yuv2rgb_coeffs[x]
  300. * @param inv_table the yuv2rgb coefficients describing the input yuv space, normally ff_yuv2rgb_coeffs[x]
  301. * @param brightness 16.16 fixed point brightness correction
  302. * @param contrast 16.16 fixed point contrast correction
  303. * @param saturation 16.16 fixed point saturation correction
  304. *
  305. * @return A negative error code on error, non negative otherwise.
  306. * If `LIBSWSCALE_VERSION_MAJOR < 7`, returns -1 if not supported.
  307. */
  308. int sws_setColorspaceDetails(struct SwsContext *c, const int inv_table[4],
  309. int srcRange, const int table[4], int dstRange,
  310. int brightness, int contrast, int saturation);
  311. /**
  312. * @return A negative error code on error, non negative otherwise.
  313. * If `LIBSWSCALE_VERSION_MAJOR < 7`, returns -1 if not supported.
  314. */
  315. int sws_getColorspaceDetails(struct SwsContext *c, int **inv_table,
  316. int *srcRange, int **table, int *dstRange,
  317. int *brightness, int *contrast, int *saturation);
  318. /**
  319. * Allocate and return an uninitialized vector with length coefficients.
  320. */
  321. SwsVector *sws_allocVec(int length);
  322. /**
  323. * Return a normalized Gaussian curve used to filter stuff
  324. * quality = 3 is high quality, lower is lower quality.
  325. */
  326. SwsVector *sws_getGaussianVec(double variance, double quality);
  327. /**
  328. * Scale all the coefficients of a by the scalar value.
  329. */
  330. void sws_scaleVec(SwsVector *a, double scalar);
  331. /**
  332. * Scale all the coefficients of a so that their sum equals height.
  333. */
  334. void sws_normalizeVec(SwsVector *a, double height);
  335. void sws_freeVec(SwsVector *a);
  336. SwsFilter *sws_getDefaultFilter(float lumaGBlur, float chromaGBlur,
  337. float lumaSharpen, float chromaSharpen,
  338. float chromaHShift, float chromaVShift,
  339. int verbose);
  340. void sws_freeFilter(SwsFilter *filter);
  341. /**
  342. * Check if context can be reused, otherwise reallocate a new one.
  343. *
  344. * If context is NULL, just calls sws_getContext() to get a new
  345. * context. Otherwise, checks if the parameters are the ones already
  346. * saved in context. If that is the case, returns the current
  347. * context. Otherwise, frees context and gets a new context with
  348. * the new parameters.
  349. *
  350. * Be warned that srcFilter and dstFilter are not checked, they
  351. * are assumed to remain the same.
  352. */
  353. struct SwsContext *sws_getCachedContext(struct SwsContext *context,
  354. int srcW, int srcH, enum AVPixelFormat srcFormat,
  355. int dstW, int dstH, enum AVPixelFormat dstFormat,
  356. int flags, SwsFilter *srcFilter,
  357. SwsFilter *dstFilter, const double *param);
  358. /**
  359. * Convert an 8-bit paletted frame into a frame with a color depth of 32 bits.
  360. *
  361. * The output frame will have the same packed format as the palette.
  362. *
  363. * @param src source frame buffer
  364. * @param dst destination frame buffer
  365. * @param num_pixels number of pixels to convert
  366. * @param palette array with [256] entries, which must match color arrangement (RGB or BGR) of src
  367. */
  368. void sws_convertPalette8ToPacked32(const uint8_t *src, uint8_t *dst, int num_pixels, const uint8_t *palette);
  369. /**
  370. * Convert an 8-bit paletted frame into a frame with a color depth of 24 bits.
  371. *
  372. * With the palette format "ABCD", the destination frame ends up with the format "ABC".
  373. *
  374. * @param src source frame buffer
  375. * @param dst destination frame buffer
  376. * @param num_pixels number of pixels to convert
  377. * @param palette array with [256] entries, which must match color arrangement (RGB or BGR) of src
  378. */
  379. void sws_convertPalette8ToPacked24(const uint8_t *src, uint8_t *dst, int num_pixels, const uint8_t *palette);
  380. /**
  381. * Get the AVClass for swsContext. It can be used in combination with
  382. * AV_OPT_SEARCH_FAKE_OBJ for examining options.
  383. *
  384. * @see av_opt_find().
  385. */
  386. const AVClass *sws_get_class(void);
  387. /**
  388. * @}
  389. */
  390. #endif /* SWSCALE_SWSCALE_H */