git.sesse.net Git - ffmpeg/blob - libavfilter/internal.h

   1 /*
   2  * This file is part of FFmpeg.
   3  *
   4  * FFmpeg is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * FFmpeg is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with FFmpeg; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  17  */
  18
  19 #ifndef AVFILTER_INTERNAL_H
  20 #define AVFILTER_INTERNAL_H
  21
  22 /**
  23  * @file
  24  * internal API functions
  25  */
  26
  27 #include "libavutil/internal.h"
  28 #include "avfilter.h"
  29 #include "avfiltergraph.h"
  30 #include "formats.h"
  31 #include "framepool.h"
  32 #include "thread.h"
  33 #include "version.h"
  34 #include "video.h"
  35 #include "libavcodec/avcodec.h"
  36
  37 typedef struct AVFilterCommand {
  38     double time;                ///< time expressed in seconds
  39     char *command;              ///< command
  40     char *arg;                  ///< optional argument for the command
  41     int flags;
  42     struct AVFilterCommand *next;
  43 } AVFilterCommand;
  44
  45 /**
  46  * Update the position of a link in the age heap.
  47  */
  48 void ff_avfilter_graph_update_heap(AVFilterGraph *graph, AVFilterLink *link);
  49
  50 /**
  51  * A filter pad used for either input or output.
  52  */
  53 struct AVFilterPad {
  54     /**
  55      * Pad name. The name is unique among inputs and among outputs, but an
  56      * input may have the same name as an output. This may be NULL if this
  57      * pad has no need to ever be referenced by name.
  58      */
  59     const char *name;
  60
  61     /**
  62      * AVFilterPad type.
  63      */
  64     enum AVMediaType type;
  65
  66     /**
  67      * Callback function to get a video buffer. If NULL, the filter system will
  68      * use ff_default_get_video_buffer().
  69      *
  70      * Input video pads only.
  71      */
  72     AVFrame *(*get_video_buffer)(AVFilterLink *link, int w, int h);
  73
  74     /**
  75      * Callback function to get an audio buffer. If NULL, the filter system will
  76      * use ff_default_get_audio_buffer().
  77      *
  78      * Input audio pads only.
  79      */
  80     AVFrame *(*get_audio_buffer)(AVFilterLink *link, int nb_samples);
  81
  82     /**
  83      * Filtering callback. This is where a filter receives a frame with
  84      * audio/video data and should do its processing.
  85      *
  86      * Input pads only.
  87      *
  88      * @return >= 0 on success, a negative AVERROR on error. This function
  89      * must ensure that samplesref is properly unreferenced on error if it
  90      * hasn't been passed on to another filter.
  91      */
  92     int (*filter_frame)(AVFilterLink *link, AVFrame *frame);
  93
  94     /**
  95      * Frame poll callback. This returns the number of immediately available
  96      * samples. It should return a positive value if the next request_frame()
  97      * is guaranteed to return one frame (with no delay).
  98      *
  99      * Defaults to just calling the source poll_frame() method.
 100      *
 101      * Output pads only.
 102      */
 103     int (*poll_frame)(AVFilterLink *link);
 104
 105     /**
 106      * Frame request callback. A call to this should result in some progress
 107      * towards producing output over the given link. This should return zero
 108      * on success, and another value on error.
 109      *
 110      * Output pads only.
 111      */
 112     int (*request_frame)(AVFilterLink *link);
 113
 114     /**
 115      * Link configuration callback.
 116      *
 117      * For output pads, this should set the link properties such as
 118      * width/height. This should NOT set the format property - that is
 119      * negotiated between filters by the filter system using the
 120      * query_formats() callback before this function is called.
 121      *
 122      * For input pads, this should check the properties of the link, and update
 123      * the filter's internal state as necessary.
 124      *
 125      * For both input and output filters, this should return zero on success,
 126      * and another value on error.
 127      */
 128     int (*config_props)(AVFilterLink *link);
 129
 130     /**
 131      * The filter expects a fifo to be inserted on its input link,
 132      * typically because it has a delay.
 133      *
 134      * input pads only.
 135      */
 136     int needs_fifo;
 137
 138     /**
 139      * The filter expects writable frames from its input link,
 140      * duplicating data buffers if needed.
 141      *
 142      * input pads only.
 143      */
 144     int needs_writable;
 145 };
 146
 147 struct AVFilterGraphInternal {
 148     void *thread;
 149     avfilter_execute_func *thread_execute;
 150 };
 151
 152 struct AVFilterInternal {
 153     avfilter_execute_func *execute;
 154 };
 155
 156 /**
 157  * Tell if an integer is contained in the provided -1-terminated list of integers.
 158  * This is useful for determining (for instance) if an AVPixelFormat is in an
 159  * array of supported formats.
 160  *
 161  * @param fmt provided format
 162  * @param fmts -1-terminated list of formats
 163  * @return 1 if present, 0 if absent
 164  */
 165 int ff_fmt_is_in(int fmt, const int *fmts);
 166
 167 /* Functions to parse audio format arguments */
 168
 169 /**
 170  * Parse a pixel format.
 171  *
 172  * @param ret pixel format pointer to where the value should be written
 173  * @param arg string to parse
 174  * @param log_ctx log context
 175  * @return >= 0 in case of success, a negative AVERROR code on error
 176  */
 177 av_warn_unused_result
 178 int ff_parse_pixel_format(enum AVPixelFormat *ret, const char *arg, void *log_ctx);
 179
 180 /**
 181  * Parse a sample rate.
 182  *
 183  * @param ret unsigned integer pointer to where the value should be written
 184  * @param arg string to parse
 185  * @param log_ctx log context
 186  * @return >= 0 in case of success, a negative AVERROR code on error
 187  */
 188 av_warn_unused_result
 189 int ff_parse_sample_rate(int *ret, const char *arg, void *log_ctx);
 190
 191 /**
 192  * Parse a time base.
 193  *
 194  * @param ret unsigned AVRational pointer to where the value should be written
 195  * @param arg string to parse
 196  * @param log_ctx log context
 197  * @return >= 0 in case of success, a negative AVERROR code on error
 198  */
 199 av_warn_unused_result
 200 int ff_parse_time_base(AVRational *ret, const char *arg, void *log_ctx);
 201
 202 /**
 203  * Parse a sample format name or a corresponding integer representation.
 204  *
 205  * @param ret integer pointer to where the value should be written
 206  * @param arg string to parse
 207  * @param log_ctx log context
 208  * @return >= 0 in case of success, a negative AVERROR code on error
 209  */
 210 av_warn_unused_result
 211 int ff_parse_sample_format(int *ret, const char *arg, void *log_ctx);
 212
 213 /**
 214  * Parse a channel layout or a corresponding integer representation.
 215  *
 216  * @param ret 64bit integer pointer to where the value should be written.
 217  * @param nret integer pointer to the number of channels;
 218  *             if not NULL, then unknown channel layouts are accepted
 219  * @param arg string to parse
 220  * @param log_ctx log context
 221  * @return >= 0 in case of success, a negative AVERROR code on error
 222  */
 223 av_warn_unused_result
 224 int ff_parse_channel_layout(int64_t *ret, int *nret, const char *arg,
 225                             void *log_ctx);
 226
 227 void ff_update_link_current_pts(AVFilterLink *link, int64_t pts);
 228
 229 void ff_command_queue_pop(AVFilterContext *filter);
 230
 231 /* misc trace functions */
 232
 233 /* #define FF_AVFILTER_TRACE */
 234
 235 #ifdef FF_AVFILTER_TRACE
 236 #    define ff_tlog(pctx, ...) av_log(pctx, AV_LOG_DEBUG, __VA_ARGS__)
 237 #else
 238 #    define ff_tlog(pctx, ...) do { if (0) av_log(pctx, AV_LOG_DEBUG, __VA_ARGS__); } while (0)
 239 #endif
 240
 241 #define FF_TPRINTF_START(ctx, func) ff_tlog(NULL, "%-16s: ", #func)
 242
 243 char *ff_get_ref_perms_string(char *buf, size_t buf_size, int perms);
 244
 245 void ff_tlog_ref(void *ctx, AVFrame *ref, int end);
 246
 247 void ff_tlog_link(void *ctx, AVFilterLink *link, int end);
 248
 249 /**
 250  * Insert a new pad.
 251  *
 252  * @param idx Insertion point. Pad is inserted at the end if this point
 253  *            is beyond the end of the list of pads.
 254  * @param count Pointer to the number of pads in the list
 255  * @param padidx_off Offset within an AVFilterLink structure to the element
 256  *                   to increment when inserting a new pad causes link
 257  *                   numbering to change
 258  * @param pads Pointer to the pointer to the beginning of the list of pads
 259  * @param links Pointer to the pointer to the beginning of the list of links
 260  * @param newpad The new pad to add. A copy is made when adding.
 261  * @return >= 0 in case of success, a negative AVERROR code on error
 262  */
 263 int ff_insert_pad(unsigned idx, unsigned *count, size_t padidx_off,
 264                    AVFilterPad **pads, AVFilterLink ***links,
 265                    AVFilterPad *newpad);
 266
 267 /** Insert a new input pad for the filter. */
 268 static inline int ff_insert_inpad(AVFilterContext *f, unsigned index,
 269                                    AVFilterPad *p)
 270 {
 271     return ff_insert_pad(index, &f->nb_inputs, offsetof(AVFilterLink, dstpad),
 272                   &f->input_pads, &f->inputs, p);
 273 }
 274
 275 /** Insert a new output pad for the filter. */
 276 static inline int ff_insert_outpad(AVFilterContext *f, unsigned index,
 277                                     AVFilterPad *p)
 278 {
 279     return ff_insert_pad(index, &f->nb_outputs, offsetof(AVFilterLink, srcpad),
 280                   &f->output_pads, &f->outputs, p);
 281 }
 282
 283 /**
 284  * Poll a frame from the filter chain.
 285  *
 286  * @param  link the input link
 287  * @return the number of immediately available frames, a negative
 288  * number in case of error
 289  */
 290 int ff_poll_frame(AVFilterLink *link);
 291
 292 /**
 293  * Request an input frame from the filter at the other end of the link.
 294  *
 295  * The input filter may pass the request on to its inputs, fulfill the
 296  * request from an internal buffer or any other means specific to its function.
 297  *
 298  * When the end of a stream is reached AVERROR_EOF is returned and no further
 299  * frames are returned after that.
 300  *
 301  * When a filter is unable to output a frame for example due to its sources
 302  * being unable to do so or because it depends on external means pushing data
 303  * into it then AVERROR(EAGAIN) is returned.
 304  * It is important that a AVERROR(EAGAIN) return is returned all the way to the
 305  * caller (generally eventually a user application) as this step may (but does
 306  * not have to be) necessary to provide the input with the next frame.
 307  *
 308  * If a request is successful then some progress has been made towards
 309  * providing a frame on the link (through ff_filter_frame()). A filter that
 310  * needs several frames to produce one is allowed to return success if one
 311  * more frame has been processed but no output has been produced yet. A
 312  * filter is also allowed to simply forward a success return value.
 313  *
 314  * @param link the input link
 315  * @return     zero on success
 316  *             AVERROR_EOF on end of file
 317  *             AVERROR(EAGAIN) if the previous filter cannot output a frame
 318  *             currently and can neither guarantee that EOF has been reached.
 319  */
 320 int ff_request_frame(AVFilterLink *link);
 321
 322 #define AVFILTER_DEFINE_CLASS(fname)            \
 323     static const AVClass fname##_class = {      \
 324         .class_name = #fname,                   \
 325         .item_name  = av_default_item_name,     \
 326         .option     = fname##_options,          \
 327         .version    = LIBAVUTIL_VERSION_INT,    \
 328         .category   = AV_CLASS_CATEGORY_FILTER, \
 329     }
 330
 331 /**
 332  * Find the index of a link.
 333  *
 334  * I.e. find i such that link == ctx->(in|out)puts[i]
 335  */
 336 #define FF_INLINK_IDX(link)  ((int)((link)->dstpad - (link)->dst->input_pads))
 337 #define FF_OUTLINK_IDX(link) ((int)((link)->srcpad - (link)->src->output_pads))
 338
 339 /**
 340  * Send a frame of data to the next filter.
 341  *
 342  * @param link   the output link over which the data is being sent
 343  * @param frame a reference to the buffer of data being sent. The
 344  *              receiving filter will free this reference when it no longer
 345  *              needs it or pass it on to the next filter.
 346  *
 347  * @return >= 0 on success, a negative AVERROR on error. The receiving filter
 348  * is responsible for unreferencing frame in case of error.
 349  */
 350 int ff_filter_frame(AVFilterLink *link, AVFrame *frame);
 351
 352 /**
 353  * Allocate a new filter context and return it.
 354  *
 355  * @param filter what filter to create an instance of
 356  * @param inst_name name to give to the new filter context
 357  *
 358  * @return newly created filter context or NULL on failure
 359  */
 360 AVFilterContext *ff_filter_alloc(const AVFilter *filter, const char *inst_name);
 361
 362 /**
 363  * Remove a filter from a graph;
 364  */
 365 void ff_filter_graph_remove_filter(AVFilterGraph *graph, AVFilterContext *filter);
 366
 367 /**
 368  * Normalize the qscale factor
 369  * FIXME the H264 qscale is a log based scale, mpeg1/2 is not, the code below
 370  *       cannot be optimal
 371  */
 372 static inline int ff_norm_qscale(int qscale, int type)
 373 {
 374     switch (type) {
 375     case FF_QSCALE_TYPE_MPEG1: return qscale;
 376     case FF_QSCALE_TYPE_MPEG2: return qscale >> 1;
 377     case FF_QSCALE_TYPE_H264:  return qscale >> 2;
 378     case FF_QSCALE_TYPE_VP56:  return (63 - qscale + 2) >> 2;
 379     }
 380     return qscale;
 381 }
 382
 383 #endif /* AVFILTER_INTERNAL_H */