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.

730 lines
23KB

  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. /* #define DEBUG */
  22. #include "libavutil/pixdesc.h"
  23. #include "libavutil/rational.h"
  24. #include "libavutil/audioconvert.h"
  25. #include "libavutil/imgutils.h"
  26. #include "libavcodec/avcodec.h"
  27. #include "avfilter.h"
  28. #include "internal.h"
  29. unsigned avfilter_version(void) {
  30. return LIBAVFILTER_VERSION_INT;
  31. }
  32. const char *avfilter_configuration(void)
  33. {
  34. return LIBAV_CONFIGURATION;
  35. }
  36. const char *avfilter_license(void)
  37. {
  38. #define LICENSE_PREFIX "libavfilter license: "
  39. return LICENSE_PREFIX LIBAV_LICENSE + sizeof(LICENSE_PREFIX) - 1;
  40. }
  41. AVFilterBufferRef *avfilter_ref_buffer(AVFilterBufferRef *ref, int pmask)
  42. {
  43. AVFilterBufferRef *ret = av_malloc(sizeof(AVFilterBufferRef));
  44. if (!ret)
  45. return NULL;
  46. *ret = *ref;
  47. if (ref->type == AVMEDIA_TYPE_VIDEO) {
  48. ret->video = av_malloc(sizeof(AVFilterBufferRefVideoProps));
  49. if (!ret->video) {
  50. av_free(ret);
  51. return NULL;
  52. }
  53. *ret->video = *ref->video;
  54. ret->extended_data = ret->data;
  55. } else if (ref->type == AVMEDIA_TYPE_AUDIO) {
  56. ret->audio = av_malloc(sizeof(AVFilterBufferRefAudioProps));
  57. if (!ret->audio) {
  58. av_free(ret);
  59. return NULL;
  60. }
  61. *ret->audio = *ref->audio;
  62. if (ref->extended_data != ref->data) {
  63. int nb_channels = av_get_channel_layout_nb_channels(ref->audio->channel_layout);
  64. if (!(ret->extended_data = av_malloc(sizeof(*ret->extended_data) *
  65. nb_channels))) {
  66. av_freep(&ret->audio);
  67. av_freep(&ret);
  68. return NULL;
  69. }
  70. memcpy(ret->extended_data, ref->extended_data,
  71. sizeof(*ret->extended_data) * nb_channels);
  72. } else
  73. ret->extended_data = ret->data;
  74. }
  75. ret->perms &= pmask;
  76. ret->buf->refcount ++;
  77. return ret;
  78. }
  79. void avfilter_unref_buffer(AVFilterBufferRef *ref)
  80. {
  81. if (!ref)
  82. return;
  83. if (!(--ref->buf->refcount))
  84. ref->buf->free(ref->buf);
  85. if (ref->extended_data != ref->data)
  86. av_freep(&ref->extended_data);
  87. av_free(ref->video);
  88. av_free(ref->audio);
  89. av_free(ref);
  90. }
  91. void avfilter_insert_pad(unsigned idx, unsigned *count, size_t padidx_off,
  92. AVFilterPad **pads, AVFilterLink ***links,
  93. AVFilterPad *newpad)
  94. {
  95. unsigned i;
  96. idx = FFMIN(idx, *count);
  97. *pads = av_realloc(*pads, sizeof(AVFilterPad) * (*count + 1));
  98. *links = av_realloc(*links, sizeof(AVFilterLink*) * (*count + 1));
  99. memmove(*pads +idx+1, *pads +idx, sizeof(AVFilterPad) * (*count-idx));
  100. memmove(*links+idx+1, *links+idx, sizeof(AVFilterLink*) * (*count-idx));
  101. memcpy(*pads+idx, newpad, sizeof(AVFilterPad));
  102. (*links)[idx] = NULL;
  103. (*count)++;
  104. for (i = idx+1; i < *count; i++)
  105. if (*links[i])
  106. (*(unsigned *)((uint8_t *) *links[i] + padidx_off))++;
  107. }
  108. int avfilter_link(AVFilterContext *src, unsigned srcpad,
  109. AVFilterContext *dst, unsigned dstpad)
  110. {
  111. AVFilterLink *link;
  112. if (src->output_count <= srcpad || dst->input_count <= dstpad ||
  113. src->outputs[srcpad] || dst->inputs[dstpad])
  114. return -1;
  115. if (src->output_pads[srcpad].type != dst->input_pads[dstpad].type) {
  116. av_log(src, AV_LOG_ERROR,
  117. "Media type mismatch between the '%s' filter output pad %d and the '%s' filter input pad %d\n",
  118. src->name, srcpad, dst->name, dstpad);
  119. return AVERROR(EINVAL);
  120. }
  121. src->outputs[srcpad] =
  122. dst-> inputs[dstpad] = link = av_mallocz(sizeof(AVFilterLink));
  123. link->src = src;
  124. link->dst = dst;
  125. link->srcpad = &src->output_pads[srcpad];
  126. link->dstpad = &dst->input_pads[dstpad];
  127. link->type = src->output_pads[srcpad].type;
  128. assert(PIX_FMT_NONE == -1 && AV_SAMPLE_FMT_NONE == -1);
  129. link->format = -1;
  130. return 0;
  131. }
  132. int avfilter_insert_filter(AVFilterLink *link, AVFilterContext *filt,
  133. unsigned filt_srcpad_idx, unsigned filt_dstpad_idx)
  134. {
  135. int ret;
  136. unsigned dstpad_idx = link->dstpad - link->dst->input_pads;
  137. av_log(link->dst, AV_LOG_INFO, "auto-inserting filter '%s' "
  138. "between the filter '%s' and the filter '%s'\n",
  139. filt->name, link->src->name, link->dst->name);
  140. link->dst->inputs[dstpad_idx] = NULL;
  141. if ((ret = avfilter_link(filt, filt_dstpad_idx, link->dst, dstpad_idx)) < 0) {
  142. /* failed to link output filter to new filter */
  143. link->dst->inputs[dstpad_idx] = link;
  144. return ret;
  145. }
  146. /* re-hookup the link to the new destination filter we inserted */
  147. link->dst = filt;
  148. link->dstpad = &filt->input_pads[filt_srcpad_idx];
  149. filt->inputs[filt_srcpad_idx] = link;
  150. /* if any information on supported media formats already exists on the
  151. * link, we need to preserve that */
  152. if (link->out_formats)
  153. avfilter_formats_changeref(&link->out_formats,
  154. &filt->outputs[filt_dstpad_idx]->out_formats);
  155. return 0;
  156. }
  157. int avfilter_config_links(AVFilterContext *filter)
  158. {
  159. int (*config_link)(AVFilterLink *);
  160. unsigned i;
  161. int ret;
  162. for (i = 0; i < filter->input_count; i ++) {
  163. AVFilterLink *link = filter->inputs[i];
  164. if (!link) continue;
  165. switch (link->init_state) {
  166. case AVLINK_INIT:
  167. continue;
  168. case AVLINK_STARTINIT:
  169. av_log(filter, AV_LOG_INFO, "circular filter chain detected\n");
  170. return 0;
  171. case AVLINK_UNINIT:
  172. link->init_state = AVLINK_STARTINIT;
  173. if ((ret = avfilter_config_links(link->src)) < 0)
  174. return ret;
  175. if (!(config_link = link->srcpad->config_props))
  176. config_link = avfilter_default_config_output_link;
  177. if ((ret = config_link(link)) < 0)
  178. return ret;
  179. if (link->time_base.num == 0 && link->time_base.den == 0)
  180. link->time_base = link->src && link->src->input_count ?
  181. link->src->inputs[0]->time_base : AV_TIME_BASE_Q;
  182. if (link->sample_aspect_ratio.num == 0 && link->sample_aspect_ratio.den == 0)
  183. link->sample_aspect_ratio = link->src->input_count ?
  184. link->src->inputs[0]->sample_aspect_ratio : (AVRational){1,1};
  185. if (link->sample_rate == 0 && link->src && link->src->input_count)
  186. link->sample_rate = link->src->inputs[0]->sample_rate;
  187. if (link->channel_layout == 0 && link->src && link->src->input_count)
  188. link->channel_layout = link->src->inputs[0]->channel_layout;
  189. if ((config_link = link->dstpad->config_props))
  190. if ((ret = config_link(link)) < 0)
  191. return ret;
  192. link->init_state = AVLINK_INIT;
  193. }
  194. }
  195. return 0;
  196. }
  197. #ifdef DEBUG
  198. static char *ff_get_ref_perms_string(char *buf, size_t buf_size, int perms)
  199. {
  200. snprintf(buf, buf_size, "%s%s%s%s%s%s",
  201. perms & AV_PERM_READ ? "r" : "",
  202. perms & AV_PERM_WRITE ? "w" : "",
  203. perms & AV_PERM_PRESERVE ? "p" : "",
  204. perms & AV_PERM_REUSE ? "u" : "",
  205. perms & AV_PERM_REUSE2 ? "U" : "",
  206. perms & AV_PERM_NEG_LINESIZES ? "n" : "");
  207. return buf;
  208. }
  209. #endif
  210. static void ff_dlog_ref(void *ctx, AVFilterBufferRef *ref, int end)
  211. {
  212. av_unused char buf[16];
  213. av_dlog(ctx,
  214. "ref[%p buf:%p refcount:%d perms:%s data:%p linesize[%d, %d, %d, %d] pts:%"PRId64" pos:%"PRId64,
  215. ref, ref->buf, ref->buf->refcount, ff_get_ref_perms_string(buf, sizeof(buf), ref->perms), ref->data[0],
  216. ref->linesize[0], ref->linesize[1], ref->linesize[2], ref->linesize[3],
  217. ref->pts, ref->pos);
  218. if (ref->video) {
  219. av_dlog(ctx, " a:%d/%d s:%dx%d i:%c iskey:%d type:%c",
  220. ref->video->pixel_aspect.num, ref->video->pixel_aspect.den,
  221. ref->video->w, ref->video->h,
  222. !ref->video->interlaced ? 'P' : /* Progressive */
  223. ref->video->top_field_first ? 'T' : 'B', /* Top / Bottom */
  224. ref->video->key_frame,
  225. av_get_picture_type_char(ref->video->pict_type));
  226. }
  227. if (ref->audio) {
  228. av_dlog(ctx, " cl:%"PRId64"d n:%d r:%d p:%d",
  229. ref->audio->channel_layout,
  230. ref->audio->nb_samples,
  231. ref->audio->sample_rate,
  232. ref->audio->planar);
  233. }
  234. av_dlog(ctx, "]%s", end ? "\n" : "");
  235. }
  236. void ff_dlog_link(void *ctx, AVFilterLink *link, int end)
  237. {
  238. if (link->type == AVMEDIA_TYPE_VIDEO) {
  239. av_dlog(ctx,
  240. "link[%p s:%dx%d fmt:%-16s %-16s->%-16s]%s",
  241. link, link->w, link->h,
  242. av_pix_fmt_descriptors[link->format].name,
  243. link->src ? link->src->filter->name : "",
  244. link->dst ? link->dst->filter->name : "",
  245. end ? "\n" : "");
  246. } else {
  247. char buf[128];
  248. av_get_channel_layout_string(buf, sizeof(buf), -1, link->channel_layout);
  249. av_dlog(ctx,
  250. "link[%p r:%"PRId64" cl:%s fmt:%-16s %-16s->%-16s]%s",
  251. link, link->sample_rate, buf,
  252. av_get_sample_fmt_name(link->format),
  253. link->src ? link->src->filter->name : "",
  254. link->dst ? link->dst->filter->name : "",
  255. end ? "\n" : "");
  256. }
  257. }
  258. AVFilterBufferRef *avfilter_get_video_buffer(AVFilterLink *link, int perms, int w, int h)
  259. {
  260. AVFilterBufferRef *ret = NULL;
  261. av_unused char buf[16];
  262. FF_DPRINTF_START(NULL, get_video_buffer); ff_dlog_link(NULL, link, 0);
  263. av_dlog(NULL, " perms:%s w:%d h:%d\n", ff_get_ref_perms_string(buf, sizeof(buf), perms), w, h);
  264. if (link->dstpad->get_video_buffer)
  265. ret = link->dstpad->get_video_buffer(link, perms, w, h);
  266. if (!ret)
  267. ret = avfilter_default_get_video_buffer(link, perms, w, h);
  268. if (ret)
  269. ret->type = AVMEDIA_TYPE_VIDEO;
  270. FF_DPRINTF_START(NULL, get_video_buffer); ff_dlog_link(NULL, link, 0); av_dlog(NULL, " returning "); ff_dlog_ref(NULL, ret, 1);
  271. return ret;
  272. }
  273. AVFilterBufferRef *
  274. avfilter_get_video_buffer_ref_from_arrays(uint8_t *data[4], int linesize[4], int perms,
  275. int w, int h, enum PixelFormat format)
  276. {
  277. AVFilterBuffer *pic = av_mallocz(sizeof(AVFilterBuffer));
  278. AVFilterBufferRef *picref = av_mallocz(sizeof(AVFilterBufferRef));
  279. if (!pic || !picref)
  280. goto fail;
  281. picref->buf = pic;
  282. picref->buf->free = ff_avfilter_default_free_buffer;
  283. if (!(picref->video = av_mallocz(sizeof(AVFilterBufferRefVideoProps))))
  284. goto fail;
  285. pic->w = picref->video->w = w;
  286. pic->h = picref->video->h = h;
  287. /* make sure the buffer gets read permission or it's useless for output */
  288. picref->perms = perms | AV_PERM_READ;
  289. pic->refcount = 1;
  290. picref->type = AVMEDIA_TYPE_VIDEO;
  291. pic->format = picref->format = format;
  292. memcpy(pic->data, data, 4*sizeof(data[0]));
  293. memcpy(pic->linesize, linesize, 4*sizeof(linesize[0]));
  294. memcpy(picref->data, pic->data, sizeof(picref->data));
  295. memcpy(picref->linesize, pic->linesize, sizeof(picref->linesize));
  296. pic-> extended_data = pic->data;
  297. picref->extended_data = picref->data;
  298. return picref;
  299. fail:
  300. if (picref && picref->video)
  301. av_free(picref->video);
  302. av_free(picref);
  303. av_free(pic);
  304. return NULL;
  305. }
  306. int avfilter_request_frame(AVFilterLink *link)
  307. {
  308. FF_DPRINTF_START(NULL, request_frame); ff_dlog_link(NULL, link, 1);
  309. if (link->srcpad->request_frame)
  310. return link->srcpad->request_frame(link);
  311. else if (link->src->inputs[0])
  312. return avfilter_request_frame(link->src->inputs[0]);
  313. else return -1;
  314. }
  315. int avfilter_poll_frame(AVFilterLink *link)
  316. {
  317. int i, min = INT_MAX;
  318. if (link->srcpad->poll_frame)
  319. return link->srcpad->poll_frame(link);
  320. for (i = 0; i < link->src->input_count; i++) {
  321. int val;
  322. if (!link->src->inputs[i])
  323. return -1;
  324. val = avfilter_poll_frame(link->src->inputs[i]);
  325. min = FFMIN(min, val);
  326. }
  327. return min;
  328. }
  329. /* XXX: should we do the duplicating of the picture ref here, instead of
  330. * forcing the source filter to do it? */
  331. void avfilter_start_frame(AVFilterLink *link, AVFilterBufferRef *picref)
  332. {
  333. void (*start_frame)(AVFilterLink *, AVFilterBufferRef *);
  334. AVFilterPad *dst = link->dstpad;
  335. int perms = picref->perms;
  336. FF_DPRINTF_START(NULL, start_frame); ff_dlog_link(NULL, link, 0); av_dlog(NULL, " "); ff_dlog_ref(NULL, picref, 1);
  337. if (!(start_frame = dst->start_frame))
  338. start_frame = avfilter_default_start_frame;
  339. if (picref->linesize[0] < 0)
  340. perms |= AV_PERM_NEG_LINESIZES;
  341. /* prepare to copy the picture if it has insufficient permissions */
  342. if ((dst->min_perms & perms) != dst->min_perms || dst->rej_perms & perms) {
  343. av_log(link->dst, AV_LOG_DEBUG,
  344. "frame copy needed (have perms %x, need %x, reject %x)\n",
  345. picref->perms,
  346. link->dstpad->min_perms, link->dstpad->rej_perms);
  347. link->cur_buf = avfilter_get_video_buffer(link, dst->min_perms, link->w, link->h);
  348. link->src_buf = picref;
  349. avfilter_copy_buffer_ref_props(link->cur_buf, link->src_buf);
  350. }
  351. else
  352. link->cur_buf = picref;
  353. start_frame(link, link->cur_buf);
  354. }
  355. void avfilter_end_frame(AVFilterLink *link)
  356. {
  357. void (*end_frame)(AVFilterLink *);
  358. if (!(end_frame = link->dstpad->end_frame))
  359. end_frame = avfilter_default_end_frame;
  360. end_frame(link);
  361. /* unreference the source picture if we're feeding the destination filter
  362. * a copied version dues to permission issues */
  363. if (link->src_buf) {
  364. avfilter_unref_buffer(link->src_buf);
  365. link->src_buf = NULL;
  366. }
  367. }
  368. void avfilter_draw_slice(AVFilterLink *link, int y, int h, int slice_dir)
  369. {
  370. uint8_t *src[4], *dst[4];
  371. int i, j, vsub;
  372. void (*draw_slice)(AVFilterLink *, int, int, int);
  373. FF_DPRINTF_START(NULL, draw_slice); ff_dlog_link(NULL, link, 0); av_dlog(NULL, " y:%d h:%d dir:%d\n", y, h, slice_dir);
  374. /* copy the slice if needed for permission reasons */
  375. if (link->src_buf) {
  376. vsub = av_pix_fmt_descriptors[link->format].log2_chroma_h;
  377. for (i = 0; i < 4; i++) {
  378. if (link->src_buf->data[i]) {
  379. src[i] = link->src_buf-> data[i] +
  380. (y >> (i==1 || i==2 ? vsub : 0)) * link->src_buf-> linesize[i];
  381. dst[i] = link->cur_buf->data[i] +
  382. (y >> (i==1 || i==2 ? vsub : 0)) * link->cur_buf->linesize[i];
  383. } else
  384. src[i] = dst[i] = NULL;
  385. }
  386. for (i = 0; i < 4; i++) {
  387. int planew =
  388. av_image_get_linesize(link->format, link->cur_buf->video->w, i);
  389. if (!src[i]) continue;
  390. for (j = 0; j < h >> (i==1 || i==2 ? vsub : 0); j++) {
  391. memcpy(dst[i], src[i], planew);
  392. src[i] += link->src_buf->linesize[i];
  393. dst[i] += link->cur_buf->linesize[i];
  394. }
  395. }
  396. }
  397. if (!(draw_slice = link->dstpad->draw_slice))
  398. draw_slice = avfilter_default_draw_slice;
  399. draw_slice(link, y, h, slice_dir);
  400. }
  401. #define MAX_REGISTERED_AVFILTERS_NB 64
  402. static AVFilter *registered_avfilters[MAX_REGISTERED_AVFILTERS_NB + 1];
  403. static int next_registered_avfilter_idx = 0;
  404. AVFilter *avfilter_get_by_name(const char *name)
  405. {
  406. int i;
  407. for (i = 0; registered_avfilters[i]; i++)
  408. if (!strcmp(registered_avfilters[i]->name, name))
  409. return registered_avfilters[i];
  410. return NULL;
  411. }
  412. int avfilter_register(AVFilter *filter)
  413. {
  414. if (next_registered_avfilter_idx == MAX_REGISTERED_AVFILTERS_NB)
  415. return -1;
  416. registered_avfilters[next_registered_avfilter_idx++] = filter;
  417. return 0;
  418. }
  419. AVFilter **av_filter_next(AVFilter **filter)
  420. {
  421. return filter ? ++filter : &registered_avfilters[0];
  422. }
  423. void avfilter_uninit(void)
  424. {
  425. memset(registered_avfilters, 0, sizeof(registered_avfilters));
  426. next_registered_avfilter_idx = 0;
  427. }
  428. static int pad_count(const AVFilterPad *pads)
  429. {
  430. int count;
  431. for(count = 0; pads->name; count ++) pads ++;
  432. return count;
  433. }
  434. static const char *filter_name(void *p)
  435. {
  436. AVFilterContext *filter = p;
  437. return filter->filter->name;
  438. }
  439. static const AVClass avfilter_class = {
  440. "AVFilter",
  441. filter_name,
  442. NULL,
  443. LIBAVUTIL_VERSION_INT,
  444. };
  445. int avfilter_open(AVFilterContext **filter_ctx, AVFilter *filter, const char *inst_name)
  446. {
  447. AVFilterContext *ret;
  448. *filter_ctx = NULL;
  449. if (!filter)
  450. return AVERROR(EINVAL);
  451. ret = av_mallocz(sizeof(AVFilterContext));
  452. if (!ret)
  453. return AVERROR(ENOMEM);
  454. ret->av_class = &avfilter_class;
  455. ret->filter = filter;
  456. ret->name = inst_name ? av_strdup(inst_name) : NULL;
  457. if (filter->priv_size) {
  458. ret->priv = av_mallocz(filter->priv_size);
  459. if (!ret->priv)
  460. goto err;
  461. }
  462. ret->input_count = pad_count(filter->inputs);
  463. if (ret->input_count) {
  464. ret->input_pads = av_malloc(sizeof(AVFilterPad) * ret->input_count);
  465. if (!ret->input_pads)
  466. goto err;
  467. memcpy(ret->input_pads, filter->inputs, sizeof(AVFilterPad) * ret->input_count);
  468. ret->inputs = av_mallocz(sizeof(AVFilterLink*) * ret->input_count);
  469. if (!ret->inputs)
  470. goto err;
  471. }
  472. ret->output_count = pad_count(filter->outputs);
  473. if (ret->output_count) {
  474. ret->output_pads = av_malloc(sizeof(AVFilterPad) * ret->output_count);
  475. if (!ret->output_pads)
  476. goto err;
  477. memcpy(ret->output_pads, filter->outputs, sizeof(AVFilterPad) * ret->output_count);
  478. ret->outputs = av_mallocz(sizeof(AVFilterLink*) * ret->output_count);
  479. if (!ret->outputs)
  480. goto err;
  481. }
  482. *filter_ctx = ret;
  483. return 0;
  484. err:
  485. av_freep(&ret->inputs);
  486. av_freep(&ret->input_pads);
  487. ret->input_count = 0;
  488. av_freep(&ret->outputs);
  489. av_freep(&ret->output_pads);
  490. ret->output_count = 0;
  491. av_freep(&ret->priv);
  492. av_free(ret);
  493. return AVERROR(ENOMEM);
  494. }
  495. void avfilter_free(AVFilterContext *filter)
  496. {
  497. int i;
  498. AVFilterLink *link;
  499. if (filter->filter->uninit)
  500. filter->filter->uninit(filter);
  501. for (i = 0; i < filter->input_count; i++) {
  502. if ((link = filter->inputs[i])) {
  503. if (link->src)
  504. link->src->outputs[link->srcpad - link->src->output_pads] = NULL;
  505. avfilter_formats_unref(&link->in_formats);
  506. avfilter_formats_unref(&link->out_formats);
  507. }
  508. av_freep(&link);
  509. }
  510. for (i = 0; i < filter->output_count; i++) {
  511. if ((link = filter->outputs[i])) {
  512. if (link->dst)
  513. link->dst->inputs[link->dstpad - link->dst->input_pads] = NULL;
  514. avfilter_formats_unref(&link->in_formats);
  515. avfilter_formats_unref(&link->out_formats);
  516. }
  517. av_freep(&link);
  518. }
  519. av_freep(&filter->name);
  520. av_freep(&filter->input_pads);
  521. av_freep(&filter->output_pads);
  522. av_freep(&filter->inputs);
  523. av_freep(&filter->outputs);
  524. av_freep(&filter->priv);
  525. av_free(filter);
  526. }
  527. int avfilter_init_filter(AVFilterContext *filter, const char *args, void *opaque)
  528. {
  529. int ret=0;
  530. if (filter->filter->init)
  531. ret = filter->filter->init(filter, args, opaque);
  532. return ret;
  533. }
  534. int avfilter_copy_frame_props(AVFilterBufferRef *dst, const AVFrame *src)
  535. {
  536. dst->pts = src->pts;
  537. dst->format = src->format;
  538. switch (dst->type) {
  539. case AVMEDIA_TYPE_VIDEO:
  540. dst->video->w = src->width;
  541. dst->video->h = src->height;
  542. dst->video->pixel_aspect = src->sample_aspect_ratio;
  543. dst->video->interlaced = src->interlaced_frame;
  544. dst->video->top_field_first = src->top_field_first;
  545. dst->video->key_frame = src->key_frame;
  546. dst->video->pict_type = src->pict_type;
  547. break;
  548. case AVMEDIA_TYPE_AUDIO:
  549. dst->audio->sample_rate = src->sample_rate;
  550. dst->audio->channel_layout = src->channel_layout;
  551. break;
  552. default:
  553. return AVERROR(EINVAL);
  554. }
  555. return 0;
  556. }
  557. int avfilter_copy_buf_props(AVFrame *dst, const AVFilterBufferRef *src)
  558. {
  559. int planes, nb_channels;
  560. memcpy(dst->data, src->data, sizeof(dst->data));
  561. memcpy(dst->linesize, src->linesize, sizeof(dst->linesize));
  562. dst->pts = src->pts;
  563. dst->format = src->format;
  564. switch (src->type) {
  565. case AVMEDIA_TYPE_VIDEO:
  566. dst->width = src->video->w;
  567. dst->height = src->video->h;
  568. dst->sample_aspect_ratio = src->video->pixel_aspect;
  569. dst->interlaced_frame = src->video->interlaced;
  570. dst->top_field_first = src->video->top_field_first;
  571. dst->key_frame = src->video->key_frame;
  572. dst->pict_type = src->video->pict_type;
  573. break;
  574. case AVMEDIA_TYPE_AUDIO:
  575. nb_channels = av_get_channel_layout_nb_channels(src->audio->channel_layout);
  576. planes = av_sample_fmt_is_planar(src->format) ? nb_channels : 1;
  577. if (planes > FF_ARRAY_ELEMS(dst->data)) {
  578. dst->extended_data = av_mallocz(planes * sizeof(*dst->extended_data));
  579. if (!dst->extended_data)
  580. return AVERROR(ENOMEM);
  581. memcpy(dst->extended_data, src->extended_data,
  582. planes * sizeof(dst->extended_data));
  583. } else
  584. dst->extended_data = dst->data;
  585. dst->sample_rate = src->audio->sample_rate;
  586. dst->channel_layout = src->audio->channel_layout;
  587. dst->nb_samples = src->audio->nb_samples;
  588. break;
  589. default:
  590. return AVERROR(EINVAL);
  591. }
  592. return 0;
  593. }
  594. void avfilter_copy_buffer_ref_props(AVFilterBufferRef *dst, AVFilterBufferRef *src)
  595. {
  596. // copy common properties
  597. dst->pts = src->pts;
  598. dst->pos = src->pos;
  599. switch (src->type) {
  600. case AVMEDIA_TYPE_VIDEO: *dst->video = *src->video; break;
  601. case AVMEDIA_TYPE_AUDIO: *dst->audio = *src->audio; break;
  602. default: break;
  603. }
  604. }