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.

672 lines
24KB

  1. /*
  2. * filter layer
  3. * Copyright (c) 2007 Bobby Bingham
  4. *
  5. * This file is part of Libav.
  6. *
  7. * Libav 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. * Libav 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 Libav; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. #ifndef AVFILTER_AVFILTER_H
  22. #define AVFILTER_AVFILTER_H
  23. #include "libavutil/avutil.h"
  24. #include "libavutil/log.h"
  25. #include "libavutil/samplefmt.h"
  26. #include "libavutil/pixfmt.h"
  27. #include "libavutil/rational.h"
  28. #include "libavcodec/avcodec.h"
  29. #include <stddef.h>
  30. #include "libavfilter/version.h"
  31. /**
  32. * Return the LIBAVFILTER_VERSION_INT constant.
  33. */
  34. unsigned avfilter_version(void);
  35. /**
  36. * Return the libavfilter build-time configuration.
  37. */
  38. const char *avfilter_configuration(void);
  39. /**
  40. * Return the libavfilter license.
  41. */
  42. const char *avfilter_license(void);
  43. typedef struct AVFilterContext AVFilterContext;
  44. typedef struct AVFilterLink AVFilterLink;
  45. typedef struct AVFilterPad AVFilterPad;
  46. typedef struct AVFilterFormats AVFilterFormats;
  47. /**
  48. * A reference-counted buffer data type used by the filter system. Filters
  49. * should not store pointers to this structure directly, but instead use the
  50. * AVFilterBufferRef structure below.
  51. */
  52. typedef struct AVFilterBuffer {
  53. uint8_t *data[8]; ///< buffer data for each plane/channel
  54. /**
  55. * pointers to the data planes/channels.
  56. *
  57. * For video, this should simply point to data[].
  58. *
  59. * For planar audio, each channel has a separate data pointer, and
  60. * linesize[0] contains the size of each channel buffer.
  61. * For packed audio, there is just one data pointer, and linesize[0]
  62. * contains the total size of the buffer for all channels.
  63. *
  64. * Note: Both data and extended_data will always be set, but for planar
  65. * audio with more channels that can fit in data, extended_data must be used
  66. * in order to access all channels.
  67. */
  68. uint8_t **extended_data;
  69. int linesize[8]; ///< number of bytes per line
  70. /** private data to be used by a custom free function */
  71. void *priv;
  72. /**
  73. * A pointer to the function to deallocate this buffer if the default
  74. * function is not sufficient. This could, for example, add the memory
  75. * back into a memory pool to be reused later without the overhead of
  76. * reallocating it from scratch.
  77. */
  78. void (*free)(struct AVFilterBuffer *buf);
  79. int format; ///< media format
  80. int w, h; ///< width and height of the allocated buffer
  81. unsigned refcount; ///< number of references to this buffer
  82. } AVFilterBuffer;
  83. #define AV_PERM_READ 0x01 ///< can read from the buffer
  84. #define AV_PERM_WRITE 0x02 ///< can write to the buffer
  85. #define AV_PERM_PRESERVE 0x04 ///< nobody else can overwrite the buffer
  86. #define AV_PERM_REUSE 0x08 ///< can output the buffer multiple times, with the same contents each time
  87. #define AV_PERM_REUSE2 0x10 ///< can output the buffer multiple times, modified each time
  88. #define AV_PERM_NEG_LINESIZES 0x20 ///< the buffer requested can have negative linesizes
  89. /**
  90. * Audio specific properties in a reference to an AVFilterBuffer. Since
  91. * AVFilterBufferRef is common to different media formats, audio specific
  92. * per reference properties must be separated out.
  93. */
  94. typedef struct AVFilterBufferRefAudioProps {
  95. uint64_t channel_layout; ///< channel layout of audio buffer
  96. int nb_samples; ///< number of audio samples
  97. int sample_rate; ///< audio buffer sample rate
  98. int planar; ///< audio buffer - planar or packed
  99. } AVFilterBufferRefAudioProps;
  100. /**
  101. * Video specific properties in a reference to an AVFilterBuffer. Since
  102. * AVFilterBufferRef is common to different media formats, video specific
  103. * per reference properties must be separated out.
  104. */
  105. typedef struct AVFilterBufferRefVideoProps {
  106. int w; ///< image width
  107. int h; ///< image height
  108. AVRational pixel_aspect; ///< pixel aspect ratio
  109. int interlaced; ///< is frame interlaced
  110. int top_field_first; ///< field order
  111. enum AVPictureType pict_type; ///< picture type of the frame
  112. int key_frame; ///< 1 -> keyframe, 0-> not
  113. } AVFilterBufferRefVideoProps;
  114. /**
  115. * A reference to an AVFilterBuffer. Since filters can manipulate the origin of
  116. * a buffer to, for example, crop image without any memcpy, the buffer origin
  117. * and dimensions are per-reference properties. Linesize is also useful for
  118. * image flipping, frame to field filters, etc, and so is also per-reference.
  119. *
  120. * TODO: add anything necessary for frame reordering
  121. */
  122. typedef struct AVFilterBufferRef {
  123. AVFilterBuffer *buf; ///< the buffer that this is a reference to
  124. uint8_t *data[8]; ///< picture/audio data for each plane
  125. /**
  126. * pointers to the data planes/channels.
  127. *
  128. * For video, this should simply point to data[].
  129. *
  130. * For planar audio, each channel has a separate data pointer, and
  131. * linesize[0] contains the size of each channel buffer.
  132. * For packed audio, there is just one data pointer, and linesize[0]
  133. * contains the total size of the buffer for all channels.
  134. *
  135. * Note: Both data and extended_data will always be set, but for planar
  136. * audio with more channels that can fit in data, extended_data must be used
  137. * in order to access all channels.
  138. */
  139. uint8_t **extended_data;
  140. int linesize[8]; ///< number of bytes per line
  141. AVFilterBufferRefVideoProps *video; ///< video buffer specific properties
  142. AVFilterBufferRefAudioProps *audio; ///< audio buffer specific properties
  143. /**
  144. * presentation timestamp. The time unit may change during
  145. * filtering, as it is specified in the link and the filter code
  146. * may need to rescale the PTS accordingly.
  147. */
  148. int64_t pts;
  149. int64_t pos; ///< byte position in stream, -1 if unknown
  150. int format; ///< media format
  151. int perms; ///< permissions, see the AV_PERM_* flags
  152. enum AVMediaType type; ///< media type of buffer data
  153. } AVFilterBufferRef;
  154. /**
  155. * Copy properties of src to dst, without copying the actual data
  156. */
  157. void avfilter_copy_buffer_ref_props(AVFilterBufferRef *dst, AVFilterBufferRef *src);
  158. /**
  159. * Add a new reference to a buffer.
  160. *
  161. * @param ref an existing reference to the buffer
  162. * @param pmask a bitmask containing the allowable permissions in the new
  163. * reference
  164. * @return a new reference to the buffer with the same properties as the
  165. * old, excluding any permissions denied by pmask
  166. */
  167. AVFilterBufferRef *avfilter_ref_buffer(AVFilterBufferRef *ref, int pmask);
  168. /**
  169. * Remove a reference to a buffer. If this is the last reference to the
  170. * buffer, the buffer itself is also automatically freed.
  171. *
  172. * @param ref reference to the buffer, may be NULL
  173. */
  174. void avfilter_unref_buffer(AVFilterBufferRef *ref);
  175. #if FF_API_AVFILTERPAD_PUBLIC
  176. /**
  177. * A filter pad used for either input or output.
  178. *
  179. * @warning this struct will be removed from public API.
  180. * users should call avfilter_pad_get_name() and avfilter_pad_get_type()
  181. * to access the name and type fields; there should be no need to access
  182. * any other fields from outside of libavfilter.
  183. */
  184. struct AVFilterPad {
  185. /**
  186. * Pad name. The name is unique among inputs and among outputs, but an
  187. * input may have the same name as an output. This may be NULL if this
  188. * pad has no need to ever be referenced by name.
  189. */
  190. const char *name;
  191. /**
  192. * AVFilterPad type.
  193. */
  194. enum AVMediaType type;
  195. /**
  196. * Minimum required permissions on incoming buffers. Any buffer with
  197. * insufficient permissions will be automatically copied by the filter
  198. * system to a new buffer which provides the needed access permissions.
  199. *
  200. * Input pads only.
  201. */
  202. int min_perms;
  203. /**
  204. * Permissions which are not accepted on incoming buffers. Any buffer
  205. * which has any of these permissions set will be automatically copied
  206. * by the filter system to a new buffer which does not have those
  207. * permissions. This can be used to easily disallow buffers with
  208. * AV_PERM_REUSE.
  209. *
  210. * Input pads only.
  211. */
  212. int rej_perms;
  213. /**
  214. * Callback called before passing the first slice of a new frame. If
  215. * NULL, the filter layer will default to storing a reference to the
  216. * picture inside the link structure.
  217. *
  218. * Input video pads only.
  219. */
  220. void (*start_frame)(AVFilterLink *link, AVFilterBufferRef *picref);
  221. /**
  222. * Callback function to get a video buffer. If NULL, the filter system will
  223. * use avfilter_default_get_video_buffer().
  224. *
  225. * Input video pads only.
  226. */
  227. AVFilterBufferRef *(*get_video_buffer)(AVFilterLink *link, int perms, int w, int h);
  228. /**
  229. * Callback function to get an audio buffer. If NULL, the filter system will
  230. * use avfilter_default_get_audio_buffer().
  231. *
  232. * Input audio pads only.
  233. */
  234. AVFilterBufferRef *(*get_audio_buffer)(AVFilterLink *link, int perms,
  235. int nb_samples);
  236. /**
  237. * Callback called after the slices of a frame are completely sent. If
  238. * NULL, the filter layer will default to releasing the reference stored
  239. * in the link structure during start_frame().
  240. *
  241. * Input video pads only.
  242. */
  243. void (*end_frame)(AVFilterLink *link);
  244. /**
  245. * Slice drawing callback. This is where a filter receives video data
  246. * and should do its processing.
  247. *
  248. * Input video pads only.
  249. */
  250. void (*draw_slice)(AVFilterLink *link, int y, int height, int slice_dir);
  251. /**
  252. * Samples filtering callback. This is where a filter receives audio data
  253. * and should do its processing.
  254. *
  255. * Input audio pads only.
  256. */
  257. void (*filter_samples)(AVFilterLink *link, AVFilterBufferRef *samplesref);
  258. /**
  259. * Frame poll callback. This returns the number of immediately available
  260. * samples. It should return a positive value if the next request_frame()
  261. * is guaranteed to return one frame (with no delay).
  262. *
  263. * Defaults to just calling the source poll_frame() method.
  264. *
  265. * Output pads only.
  266. */
  267. int (*poll_frame)(AVFilterLink *link);
  268. /**
  269. * Frame request callback. A call to this should result in at least one
  270. * frame being output over the given link. This should return zero on
  271. * success, and another value on error.
  272. *
  273. * Output pads only.
  274. */
  275. int (*request_frame)(AVFilterLink *link);
  276. /**
  277. * Link configuration callback.
  278. *
  279. * For output pads, this should set the link properties such as
  280. * width/height. This should NOT set the format property - that is
  281. * negotiated between filters by the filter system using the
  282. * query_formats() callback before this function is called.
  283. *
  284. * For input pads, this should check the properties of the link, and update
  285. * the filter's internal state as necessary.
  286. *
  287. * For both input and output filters, this should return zero on success,
  288. * and another value on error.
  289. */
  290. int (*config_props)(AVFilterLink *link);
  291. /**
  292. * The filter expects a fifo to be inserted on its input link,
  293. * typically because it has a delay.
  294. *
  295. * input pads only.
  296. */
  297. int needs_fifo;
  298. };
  299. #endif
  300. /**
  301. * Get the name of an AVFilterPad.
  302. *
  303. * @param pads an array of AVFilterPads
  304. * @param pad_idx index of the pad in the array it; is the caller's
  305. * responsibility to ensure the index is valid
  306. *
  307. * @return name of the pad_idx'th pad in pads
  308. */
  309. const char *avfilter_pad_get_name(AVFilterPad *pads, int pad_idx);
  310. /**
  311. * Get the type of an AVFilterPad.
  312. *
  313. * @param pads an array of AVFilterPads
  314. * @param pad_idx index of the pad in the array; it is the caller's
  315. * responsibility to ensure the index is valid
  316. *
  317. * @return type of the pad_idx'th pad in pads
  318. */
  319. enum AVMediaType avfilter_pad_get_type(AVFilterPad *pads, int pad_idx);
  320. /**
  321. * Filter definition. This defines the pads a filter contains, and all the
  322. * callback functions used to interact with the filter.
  323. */
  324. typedef struct AVFilter {
  325. const char *name; ///< filter name
  326. /**
  327. * A description for the filter. You should use the
  328. * NULL_IF_CONFIG_SMALL() macro to define it.
  329. */
  330. const char *description;
  331. const AVFilterPad *inputs; ///< NULL terminated list of inputs. NULL if none
  332. const AVFilterPad *outputs; ///< NULL terminated list of outputs. NULL if none
  333. /*****************************************************************
  334. * All fields below this line are not part of the public API. They
  335. * may not be used outside of libavfilter and can be changed and
  336. * removed at will.
  337. * New public fields should be added right above.
  338. *****************************************************************
  339. */
  340. /**
  341. * Filter initialization function. Args contains the user-supplied
  342. * parameters. FIXME: maybe an AVOption-based system would be better?
  343. */
  344. int (*init)(AVFilterContext *ctx, const char *args);
  345. /**
  346. * Filter uninitialization function. Should deallocate any memory held
  347. * by the filter, release any buffer references, etc. This does not need
  348. * to deallocate the AVFilterContext->priv memory itself.
  349. */
  350. void (*uninit)(AVFilterContext *ctx);
  351. /**
  352. * Queries formats supported by the filter and its pads, and sets the
  353. * in_formats for links connected to its output pads, and out_formats
  354. * for links connected to its input pads.
  355. *
  356. * @return zero on success, a negative value corresponding to an
  357. * AVERROR code otherwise
  358. */
  359. int (*query_formats)(AVFilterContext *);
  360. int priv_size; ///< size of private data to allocate for the filter
  361. } AVFilter;
  362. /** An instance of a filter */
  363. struct AVFilterContext {
  364. const AVClass *av_class; ///< needed for av_log()
  365. AVFilter *filter; ///< the AVFilter of which this is an instance
  366. char *name; ///< name of this filter instance
  367. AVFilterPad *input_pads; ///< array of input pads
  368. AVFilterLink **inputs; ///< array of pointers to input links
  369. #if FF_API_FOO_COUNT
  370. unsigned input_count; ///< @deprecated use nb_inputs
  371. #endif
  372. unsigned nb_inputs; ///< number of input pads
  373. AVFilterPad *output_pads; ///< array of output pads
  374. AVFilterLink **outputs; ///< array of pointers to output links
  375. #if FF_API_FOO_COUNT
  376. unsigned output_count; ///< @deprecated use nb_outputs
  377. #endif
  378. unsigned nb_outputs; ///< number of output pads
  379. void *priv; ///< private data for use by the filter
  380. };
  381. /**
  382. * A link between two filters. This contains pointers to the source and
  383. * destination filters between which this link exists, and the indexes of
  384. * the pads involved. In addition, this link also contains the parameters
  385. * which have been negotiated and agreed upon between the filter, such as
  386. * image dimensions, format, etc.
  387. */
  388. struct AVFilterLink {
  389. AVFilterContext *src; ///< source filter
  390. AVFilterPad *srcpad; ///< output pad on the source filter
  391. AVFilterContext *dst; ///< dest filter
  392. AVFilterPad *dstpad; ///< input pad on the dest filter
  393. enum AVMediaType type; ///< filter media type
  394. /* These parameters apply only to video */
  395. int w; ///< agreed upon image width
  396. int h; ///< agreed upon image height
  397. AVRational sample_aspect_ratio; ///< agreed upon sample aspect ratio
  398. /* These two parameters apply only to audio */
  399. uint64_t channel_layout; ///< channel layout of current buffer (see libavutil/audioconvert.h)
  400. int sample_rate; ///< samples per second
  401. int format; ///< agreed upon media format
  402. /**
  403. * Define the time base used by the PTS of the frames/samples
  404. * which will pass through this link.
  405. * During the configuration stage, each filter is supposed to
  406. * change only the output timebase, while the timebase of the
  407. * input link is assumed to be an unchangeable property.
  408. */
  409. AVRational time_base;
  410. /*****************************************************************
  411. * All fields below this line are not part of the public API. They
  412. * may not be used outside of libavfilter and can be changed and
  413. * removed at will.
  414. * New public fields should be added right above.
  415. *****************************************************************
  416. */
  417. /**
  418. * Lists of formats supported by the input and output filters respectively.
  419. * These lists are used for negotiating the format to actually be used,
  420. * which will be loaded into the format member, above, when chosen.
  421. */
  422. AVFilterFormats *in_formats;
  423. AVFilterFormats *out_formats;
  424. /**
  425. * Lists of channel layouts and sample rates used for automatic
  426. * negotiation.
  427. */
  428. AVFilterFormats *in_samplerates;
  429. AVFilterFormats *out_samplerates;
  430. struct AVFilterChannelLayouts *in_channel_layouts;
  431. struct AVFilterChannelLayouts *out_channel_layouts;
  432. /**
  433. * Audio only, the destination filter sets this to a non-zero value to
  434. * request that buffers with the given number of samples should be sent to
  435. * it. AVFilterPad.needs_fifo must also be set on the corresponding input
  436. * pad.
  437. * Last buffer before EOF will be padded with silence.
  438. */
  439. int request_samples;
  440. /** stage of the initialization of the link properties (dimensions, etc) */
  441. enum {
  442. AVLINK_UNINIT = 0, ///< not started
  443. AVLINK_STARTINIT, ///< started, but incomplete
  444. AVLINK_INIT ///< complete
  445. } init_state;
  446. /**
  447. * The buffer reference currently being sent across the link by the source
  448. * filter. This is used internally by the filter system to allow
  449. * automatic copying of buffers which do not have sufficient permissions
  450. * for the destination. This should not be accessed directly by the
  451. * filters.
  452. */
  453. AVFilterBufferRef *src_buf;
  454. AVFilterBufferRef *cur_buf;
  455. AVFilterBufferRef *out_buf;
  456. };
  457. /**
  458. * Link two filters together.
  459. *
  460. * @param src the source filter
  461. * @param srcpad index of the output pad on the source filter
  462. * @param dst the destination filter
  463. * @param dstpad index of the input pad on the destination filter
  464. * @return zero on success
  465. */
  466. int avfilter_link(AVFilterContext *src, unsigned srcpad,
  467. AVFilterContext *dst, unsigned dstpad);
  468. /**
  469. * Negotiate the media format, dimensions, etc of all inputs to a filter.
  470. *
  471. * @param filter the filter to negotiate the properties for its inputs
  472. * @return zero on successful negotiation
  473. */
  474. int avfilter_config_links(AVFilterContext *filter);
  475. /**
  476. * Create a buffer reference wrapped around an already allocated image
  477. * buffer.
  478. *
  479. * @param data pointers to the planes of the image to reference
  480. * @param linesize linesizes for the planes of the image to reference
  481. * @param perms the required access permissions
  482. * @param w the width of the image specified by the data and linesize arrays
  483. * @param h the height of the image specified by the data and linesize arrays
  484. * @param format the pixel format of the image specified by the data and linesize arrays
  485. */
  486. AVFilterBufferRef *
  487. avfilter_get_video_buffer_ref_from_arrays(uint8_t *data[4], int linesize[4], int perms,
  488. int w, int h, enum PixelFormat format);
  489. /**
  490. * Create an audio buffer reference wrapped around an already
  491. * allocated samples buffer.
  492. *
  493. * @param data pointers to the samples plane buffers
  494. * @param linesize linesize for the samples plane buffers
  495. * @param perms the required access permissions
  496. * @param nb_samples number of samples per channel
  497. * @param sample_fmt the format of each sample in the buffer to allocate
  498. * @param channel_layout the channel layout of the buffer
  499. */
  500. AVFilterBufferRef *avfilter_get_audio_buffer_ref_from_arrays(uint8_t **data,
  501. int linesize,
  502. int perms,
  503. int nb_samples,
  504. enum AVSampleFormat sample_fmt,
  505. uint64_t channel_layout);
  506. /** Initialize the filter system. Register all builtin filters. */
  507. void avfilter_register_all(void);
  508. /** Uninitialize the filter system. Unregister all filters. */
  509. void avfilter_uninit(void);
  510. /**
  511. * Register a filter. This is only needed if you plan to use
  512. * avfilter_get_by_name later to lookup the AVFilter structure by name. A
  513. * filter can still by instantiated with avfilter_open even if it is not
  514. * registered.
  515. *
  516. * @param filter the filter to register
  517. * @return 0 if the registration was succesfull, a negative value
  518. * otherwise
  519. */
  520. int avfilter_register(AVFilter *filter);
  521. /**
  522. * Get a filter definition matching the given name.
  523. *
  524. * @param name the filter name to find
  525. * @return the filter definition, if any matching one is registered.
  526. * NULL if none found.
  527. */
  528. AVFilter *avfilter_get_by_name(const char *name);
  529. /**
  530. * If filter is NULL, returns a pointer to the first registered filter pointer,
  531. * if filter is non-NULL, returns the next pointer after filter.
  532. * If the returned pointer points to NULL, the last registered filter
  533. * was already reached.
  534. */
  535. AVFilter **av_filter_next(AVFilter **filter);
  536. /**
  537. * Create a filter instance.
  538. *
  539. * @param filter_ctx put here a pointer to the created filter context
  540. * on success, NULL on failure
  541. * @param filter the filter to create an instance of
  542. * @param inst_name Name to give to the new instance. Can be NULL for none.
  543. * @return >= 0 in case of success, a negative error code otherwise
  544. */
  545. int avfilter_open(AVFilterContext **filter_ctx, AVFilter *filter, const char *inst_name);
  546. /**
  547. * Initialize a filter.
  548. *
  549. * @param filter the filter to initialize
  550. * @param args A string of parameters to use when initializing the filter.
  551. * The format and meaning of this string varies by filter.
  552. * @param opaque Any extra non-string data needed by the filter. The meaning
  553. * of this parameter varies by filter.
  554. * @return zero on success
  555. */
  556. int avfilter_init_filter(AVFilterContext *filter, const char *args, void *opaque);
  557. /**
  558. * Free a filter context.
  559. *
  560. * @param filter the filter to free
  561. */
  562. void avfilter_free(AVFilterContext *filter);
  563. /**
  564. * Insert a filter in the middle of an existing link.
  565. *
  566. * @param link the link into which the filter should be inserted
  567. * @param filt the filter to be inserted
  568. * @param filt_srcpad_idx the input pad on the filter to connect
  569. * @param filt_dstpad_idx the output pad on the filter to connect
  570. * @return zero on success
  571. */
  572. int avfilter_insert_filter(AVFilterLink *link, AVFilterContext *filt,
  573. unsigned filt_srcpad_idx, unsigned filt_dstpad_idx);
  574. /**
  575. * Copy the frame properties of src to dst, without copying the actual
  576. * image data.
  577. *
  578. * @return 0 on success, a negative number on error.
  579. */
  580. int avfilter_copy_frame_props(AVFilterBufferRef *dst, const AVFrame *src);
  581. /**
  582. * Copy the frame properties and data pointers of src to dst, without copying
  583. * the actual data.
  584. *
  585. * @return 0 on success, a negative number on error.
  586. */
  587. int avfilter_copy_buf_props(AVFrame *dst, const AVFilterBufferRef *src);
  588. #endif /* AVFILTER_AVFILTER_H */