summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAnton Khirnov <anton@khirnov.net>2012-12-25 22:11:36 +0100
committerAnton Khirnov <anton@khirnov.net>2013-03-08 07:37:01 +0100
commit77b2cd7b41d7ec8008b6fac753c04f77824c514c (patch)
tree67016b52528cb2564f8ab5075d3814e29c12170d
parent7ecc2d403ce5c7b6ea3b1f368dccefd105209c7e (diff)
downloadffmpeg-77b2cd7b41d7ec8008b6fac753c04f77824c514c.tar.gz
AVFrame: add side data.
-rw-r--r--libavutil/frame.c72
-rw-r--r--libavutil/frame.h38
2 files changed, 110 insertions, 0 deletions
diff --git a/libavutil/frame.c b/libavutil/frame.c
index e84052eda0..560bab99a7 100644
--- a/libavutil/frame.c
+++ b/libavutil/frame.c
@@ -20,6 +20,7 @@
#include "audioconvert.h"
#include "buffer.h"
#include "common.h"
+#include "dict.h"
#include "frame.h"
#include "imgutils.h"
#include "mem.h"
@@ -273,6 +274,13 @@ void av_frame_unref(AVFrame *frame)
{
int i;
+ for (i = 0; i < frame->nb_side_data; i++) {
+ av_freep(&frame->side_data[i]->data);
+ av_dict_free(&frame->side_data[i]->metadata);
+ av_freep(&frame->side_data[i]);
+ }
+ av_freep(&frame->side_data);
+
for (i = 0; i < FF_ARRAY_ELEMS(frame->buf); i++)
av_buffer_unref(&frame->buf[i]);
for (i = 0; i < frame->nb_extended_buf; i++)
@@ -353,6 +361,8 @@ int av_frame_make_writable(AVFrame *frame)
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
{
+ int i;
+
dst->key_frame = src->key_frame;
dst->pict_type = src->pict_type;
dst->sample_aspect_ratio = src->sample_aspect_ratio;
@@ -367,6 +377,23 @@ int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
dst->coded_picture_number = src->coded_picture_number;
dst->display_picture_number = src->display_picture_number;
+ for (i = 0; i < src->nb_side_data; i++) {
+ const AVFrameSideData *sd_src = src->side_data[i];
+ AVFrameSideData *sd_dst = av_frame_new_side_data(dst, sd_src->type,
+ sd_src->size);
+ if (!sd_dst) {
+ for (i = 0; i < dst->nb_side_data; i++) {
+ av_freep(&dst->side_data[i]->data);
+ av_freep(&dst->side_data[i]);
+ av_dict_free(&dst->side_data[i]->metadata);
+ }
+ av_freep(&dst->side_data);
+ return AVERROR(ENOMEM);
+ }
+ memcpy(sd_dst->data, sd_src->data, sd_src->size);
+ av_dict_copy(&sd_dst->metadata, sd_src->metadata, 0);
+ }
+
return 0;
}
@@ -399,3 +426,48 @@ AVBufferRef *av_frame_get_plane_buffer(AVFrame *frame, int plane)
}
return NULL;
}
+
+AVFrameSideData *av_frame_new_side_data(AVFrame *frame,
+ enum AVFrameSideDataType type,
+ int size)
+{
+ AVFrameSideData *ret, **tmp;
+
+ if (frame->nb_side_data > INT_MAX / sizeof(*frame->side_data) - 1)
+ return NULL;
+
+ tmp = av_realloc(frame->side_data,
+ (frame->nb_side_data + 1) * sizeof(*frame->side_data));
+ if (!tmp)
+ return NULL;
+ frame->side_data = tmp;
+
+ ret = av_mallocz(sizeof(*ret));
+ if (!ret)
+ return NULL;
+
+ ret->data = av_malloc(size);
+ if (!ret->data) {
+ av_freep(&ret);
+ return NULL;
+ }
+
+ ret->size = size;
+ ret->type = type;
+
+ frame->side_data[frame->nb_side_data++] = ret;
+
+ return ret;
+}
+
+AVFrameSideData *av_frame_get_side_data(AVFrame *frame,
+ enum AVFrameSideDataType type)
+{
+ int i;
+
+ for (i = 0; i < frame->nb_side_data; i++) {
+ if (frame->side_data[i]->type == type)
+ return frame->side_data[i];
+ }
+ return NULL;
+}
diff --git a/libavutil/frame.h b/libavutil/frame.h
index ed410ba122..5e50aae11a 100644
--- a/libavutil/frame.h
+++ b/libavutil/frame.h
@@ -26,9 +26,23 @@
#include "avutil.h"
#include "buffer.h"
+#include "dict.h"
#include "rational.h"
#include "samplefmt.h"
+enum AVFrameSideDataType {
+ /**
+ * The data is the AVPanScan struct defined in libavcodec.
+ */
+ AV_FRAME_DATA_PANSCAN,
+};
+
+typedef struct AVFrameSideData {
+ enum AVFrameSideDataType type;
+ uint8_t *data;
+ int size;
+ AVDictionary *metadata;
+} AVFrameSideData;
/**
* This structure describes decoded (raw) audio or video data.
@@ -306,6 +320,9 @@ typedef struct AVFrame {
* Number of elements in extended_buf.
*/
int nb_extended_buf;
+
+ AVFrameSideData **side_data;
+ int nb_side_data;
} AVFrame;
/**
@@ -413,6 +430,7 @@ int av_frame_make_writable(AVFrame *frame);
* Metadata for the purpose of this function are those fields that do not affect
* the data layout in the buffers. E.g. pts, sample rate (for audio) or sample
* aspect ratio (for video), but not width/height or channel layout.
+ * Side data is also copied.
*/
int av_frame_copy_props(AVFrame *dst, const AVFrame *src);
@@ -426,4 +444,24 @@ int av_frame_copy_props(AVFrame *dst, const AVFrame *src);
*/
AVBufferRef *av_frame_get_plane_buffer(AVFrame *frame, int plane);
+/**
+ * Add a new side data to a frame.
+ *
+ * @param frame a frame to which the side data should be added
+ * @param type type of the added side data
+ * @param size size of the side data
+ *
+ * @return newly added side data on success, NULL on error
+ */
+AVFrameSideData *av_frame_new_side_data(AVFrame *frame,
+ enum AVFrameSideDataType type,
+ int size);
+
+/**
+ * @return a pointer to the side data of a given type on success, NULL if there
+ * is no side data with such type in this frame.
+ */
+AVFrameSideData *av_frame_get_side_data(AVFrame *frame,
+ enum AVFrameSideDataType type);
+
#endif /* AVUTIL_FRAME_H */