typefindhelper: also unmap collected mapped buffers

This commit is contained in:
Mark Nauwelaerts 2012-03-30 16:53:05 +02:00
parent 096d36e5c5
commit a0649dd357

View file

@ -45,6 +45,12 @@
static void static void
helper_find_suggest (gpointer data, guint probability, GstCaps * caps); helper_find_suggest (gpointer data, guint probability, GstCaps * caps);
typedef struct
{
GstBuffer *buffer;
GstMapInfo map;
} GstMappedBuffer;
typedef struct typedef struct
{ {
GSList *buffers; /* buffer cache */ GSList *buffers; /* buffer cache */
@ -80,7 +86,7 @@ helper_find_peek (gpointer data, gint64 offset, guint size)
GSList *insert_pos = NULL; GSList *insert_pos = NULL;
gsize buf_size; gsize buf_size;
guint64 buf_offset; guint64 buf_offset;
GstMapInfo info; GstMappedBuffer *bmap;
#if 0 #if 0
GstCaps *caps; GstCaps *caps;
#endif #endif
@ -105,7 +111,8 @@ helper_find_peek (gpointer data, gint64 offset, guint size)
GSList *walk; GSList *walk;
for (walk = helper->buffers; walk; walk = walk->next) { for (walk = helper->buffers; walk; walk = walk->next) {
GstBuffer *buf = GST_BUFFER_CAST (walk->data); GstMappedBuffer *bmp = (GstMappedBuffer *) walk->data;
GstBuffer *buf = GST_BUFFER_CAST (bmp->buffer);
guint64 buf_offset = GST_BUFFER_OFFSET (buf); guint64 buf_offset = GST_BUFFER_OFFSET (buf);
guint buf_size = gst_buffer_get_size (buf); guint buf_size = gst_buffer_get_size (buf);
@ -114,10 +121,8 @@ helper_find_peek (gpointer data, gint64 offset, guint size)
* we're after the searched end offset */ * we're after the searched end offset */
if (buf_offset <= offset) { if (buf_offset <= offset) {
if ((offset + size) < (buf_offset + buf_size)) { if ((offset + size) < (buf_offset + buf_size)) {
/* FIXME, unmap after usage */ /* must already have been mapped before */
if (!gst_buffer_map (buf, &info, GST_MAP_READ)) return (guint8 *) bmp->map.data + (offset - buf_offset);
return NULL;
return (guint8 *) info.data + (offset - buf_offset);
} }
} else if (offset + size >= buf_offset + buf_size) { } else if (offset + size >= buf_offset + buf_size) {
insert_pos = walk; insert_pos = walk;
@ -168,21 +173,21 @@ helper_find_peek (gpointer data, gint64 offset, guint size)
return NULL; return NULL;
} }
bmap = g_slice_new0 (GstMappedBuffer);
bmap->buffer = buffer;
if (insert_pos) { if (insert_pos) {
helper->buffers = helper->buffers = g_slist_insert_before (helper->buffers, insert_pos, bmap);
g_slist_insert_before (helper->buffers, insert_pos, buffer);
} else { } else {
/* if insert_pos is not set, our offset is bigger than the largest offset /* if insert_pos is not set, our offset is bigger than the largest offset
* we have so far; since we keep the list sorted with highest offsets * we have so far; since we keep the list sorted with highest offsets
* first, we need to prepend the buffer to the list */ * first, we need to prepend the buffer to the list */
helper->last_offset = GST_BUFFER_OFFSET (buffer) + buf_size; helper->last_offset = GST_BUFFER_OFFSET (buffer) + buf_size;
helper->buffers = g_slist_prepend (helper->buffers, buffer); helper->buffers = g_slist_prepend (helper->buffers, bmap);
} }
/* FIXME, unmap */ gst_buffer_map (buffer, &bmap->map, GST_MAP_READ);
gst_buffer_map (buffer, &info, GST_MAP_READ);
return info.data; return bmap->map.data;
error: error:
{ {
@ -343,8 +348,13 @@ gst_type_find_helper_get_range (GstObject * obj, GstObject * parent,
} }
gst_plugin_feature_list_free (type_list); gst_plugin_feature_list_free (type_list);
for (walk = helper.buffers; walk; walk = walk->next) for (walk = helper.buffers; walk; walk = walk->next) {
gst_buffer_unref (GST_BUFFER_CAST (walk->data)); GstMappedBuffer *bmap = (GstMappedBuffer *) walk->data;
gst_buffer_unmap (bmap->buffer, &bmap->map);
gst_buffer_unref (bmap->buffer);
g_slice_free (GstMappedBuffer, bmap);
}
g_slist_free (helper.buffers); g_slist_free (helper.buffers);
if (helper.best_probability > 0) if (helper.best_probability > 0)