yading@10: /* yading@10: * Copyright Stefano Sabatini yading@10: * Copyright Anton Khirnov yading@10: * Copyright Michael Niedermayer yading@10: * yading@10: * This file is part of FFmpeg. yading@10: * yading@10: * FFmpeg is free software; you can redistribute it and/or yading@10: * modify it under the terms of the GNU Lesser General Public yading@10: * License as published by the Free Software Foundation; either yading@10: * version 2.1 of the License, or (at your option) any later version. yading@10: * yading@10: * FFmpeg is distributed in the hope that it will be useful, yading@10: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@10: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@10: * Lesser General Public License for more details. yading@10: * yading@10: * You should have received a copy of the GNU Lesser General Public yading@10: * License along with FFmpeg; if not, write to the Free Software yading@10: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@10: */ yading@10: yading@10: #include "libavutil/channel_layout.h" yading@10: #include "libavutil/avassert.h" yading@10: #include "libavutil/common.h" yading@10: #include "libavutil/imgutils.h" yading@10: #include "libavcodec/avcodec.h" yading@10: yading@10: #include "avfilter.h" yading@10: #include "internal.h" yading@10: #include "audio.h" yading@10: #include "avcodec.h" yading@10: yading@10: void ff_avfilter_default_free_buffer(AVFilterBuffer *ptr) yading@10: { yading@10: if (ptr->extended_data != ptr->data) yading@10: av_freep(&ptr->extended_data); yading@10: av_free(ptr->data[0]); yading@10: av_free(ptr); yading@10: } yading@10: yading@10: static void copy_video_props(AVFilterBufferRefVideoProps *dst, AVFilterBufferRefVideoProps *src) { yading@10: *dst = *src; yading@10: if (src->qp_table) { yading@10: int qsize = src->qp_table_size; yading@10: dst->qp_table = av_malloc(qsize); yading@10: memcpy(dst->qp_table, src->qp_table, qsize); yading@10: } yading@10: } yading@10: yading@10: AVFilterBufferRef *avfilter_ref_buffer(AVFilterBufferRef *ref, int pmask) yading@10: { yading@10: AVFilterBufferRef *ret = av_malloc(sizeof(AVFilterBufferRef)); yading@10: if (!ret) yading@10: return NULL; yading@10: *ret = *ref; yading@10: yading@10: ret->metadata = NULL; yading@10: av_dict_copy(&ret->metadata, ref->metadata, 0); yading@10: yading@10: if (ref->type == AVMEDIA_TYPE_VIDEO) { yading@10: ret->video = av_malloc(sizeof(AVFilterBufferRefVideoProps)); yading@10: if (!ret->video) { yading@10: av_free(ret); yading@10: return NULL; yading@10: } yading@10: copy_video_props(ret->video, ref->video); yading@10: ret->extended_data = ret->data; yading@10: } else if (ref->type == AVMEDIA_TYPE_AUDIO) { yading@10: ret->audio = av_malloc(sizeof(AVFilterBufferRefAudioProps)); yading@10: if (!ret->audio) { yading@10: av_free(ret); yading@10: return NULL; yading@10: } yading@10: *ret->audio = *ref->audio; yading@10: yading@10: if (ref->extended_data && ref->extended_data != ref->data) { yading@10: int nb_channels = av_get_channel_layout_nb_channels(ref->audio->channel_layout); yading@10: if (!(ret->extended_data = av_malloc(sizeof(*ret->extended_data) * yading@10: nb_channels))) { yading@10: av_freep(&ret->audio); yading@10: av_freep(&ret); yading@10: return NULL; yading@10: } yading@10: memcpy(ret->extended_data, ref->extended_data, yading@10: sizeof(*ret->extended_data) * nb_channels); yading@10: } else yading@10: ret->extended_data = ret->data; yading@10: } yading@10: ret->perms &= pmask; yading@10: ret->buf->refcount ++; yading@10: return ret; yading@10: } yading@10: yading@10: void avfilter_unref_buffer(AVFilterBufferRef *ref) yading@10: { yading@10: if (!ref) yading@10: return; yading@10: av_assert0(ref->buf->refcount > 0); yading@10: if (!(--ref->buf->refcount)) yading@10: ref->buf->free(ref->buf); yading@10: if (ref->extended_data != ref->data) yading@10: av_freep(&ref->extended_data); yading@10: if (ref->video) yading@10: av_freep(&ref->video->qp_table); yading@10: av_freep(&ref->video); yading@10: av_freep(&ref->audio); yading@10: av_dict_free(&ref->metadata); yading@10: av_free(ref); yading@10: } yading@10: yading@10: void avfilter_unref_bufferp(AVFilterBufferRef **ref) yading@10: { yading@10: avfilter_unref_buffer(*ref); yading@10: *ref = NULL; yading@10: } yading@10: yading@10: int avfilter_copy_frame_props(AVFilterBufferRef *dst, const AVFrame *src) yading@10: { yading@10: dst->pts = src->pts; yading@10: dst->pos = av_frame_get_pkt_pos(src); yading@10: dst->format = src->format; yading@10: yading@10: av_dict_free(&dst->metadata); yading@10: av_dict_copy(&dst->metadata, av_frame_get_metadata(src), 0); yading@10: yading@10: switch (dst->type) { yading@10: case AVMEDIA_TYPE_VIDEO: yading@10: dst->video->w = src->width; yading@10: dst->video->h = src->height; yading@10: dst->video->sample_aspect_ratio = src->sample_aspect_ratio; yading@10: dst->video->interlaced = src->interlaced_frame; yading@10: dst->video->top_field_first = src->top_field_first; yading@10: dst->video->key_frame = src->key_frame; yading@10: dst->video->pict_type = src->pict_type; yading@10: break; yading@10: case AVMEDIA_TYPE_AUDIO: yading@10: dst->audio->sample_rate = src->sample_rate; yading@10: dst->audio->channel_layout = src->channel_layout; yading@10: break; yading@10: default: yading@10: return AVERROR(EINVAL); yading@10: } yading@10: yading@10: return 0; yading@10: } yading@10: yading@10: void avfilter_copy_buffer_ref_props(AVFilterBufferRef *dst, AVFilterBufferRef *src) yading@10: { yading@10: // copy common properties yading@10: dst->pts = src->pts; yading@10: dst->pos = src->pos; yading@10: yading@10: switch (src->type) { yading@10: case AVMEDIA_TYPE_VIDEO: { yading@10: if (dst->video->qp_table) yading@10: av_freep(&dst->video->qp_table); yading@10: copy_video_props(dst->video, src->video); yading@10: break; yading@10: } yading@10: case AVMEDIA_TYPE_AUDIO: *dst->audio = *src->audio; break; yading@10: default: break; yading@10: } yading@10: yading@10: av_dict_free(&dst->metadata); yading@10: av_dict_copy(&dst->metadata, src->metadata, 0); yading@10: }