identity: add eos-after property

Using `num-buffers` can be unpredictable as buffer sizes are often
arbitrary (filesrc, multifilesrc, etc.). The `error-after` property on
`identity` is better but obviously reports an error afterwards. This
adds `eos-after` which does exactly the same thing but reports EOS
instead.
This commit is contained in:
Jonny Lamb 2018-12-11 16:48:56 +00:00 committed by Nicolas Dufresne
parent e03f086fae
commit 460c0edbfb
2 changed files with 34 additions and 1 deletions

View file

@ -75,6 +75,7 @@ enum
#define DEFAULT_SIGNAL_HANDOFFS TRUE
#define DEFAULT_TS_OFFSET 0
#define DEFAULT_DROP_ALLOCATION FALSE
#define DEFAULT_EOS_AFTER -1
enum
{
@ -93,7 +94,8 @@ enum
PROP_CHECK_IMPERFECT_TIMESTAMP,
PROP_CHECK_IMPERFECT_OFFSET,
PROP_SIGNAL_HANDOFFS,
PROP_DROP_ALLOCATION
PROP_DROP_ALLOCATION,
PROP_EOS_AFTER
};
@ -237,6 +239,18 @@ gst_identity_class_init (GstIdentityClass * klass)
"Don't forward allocation queries", DEFAULT_DROP_ALLOCATION,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
/**
* GstIdentity:eos-after
*
* EOS after N buffers.
*
* Since: 1.16
**/
g_object_class_install_property (gobject_class, PROP_EOS_AFTER,
g_param_spec_int ("eos-after", "EOS After", "EOS after N buffers",
G_MININT, G_MAXINT, DEFAULT_ERROR_AFTER,
G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
/**
* GstIdentity::handoff:
* @identity: the identity instance
@ -291,6 +305,7 @@ gst_identity_init (GstIdentity * identity)
identity->signal_handoffs = DEFAULT_SIGNAL_HANDOFFS;
identity->ts_offset = DEFAULT_TS_OFFSET;
g_cond_init (&identity->blocked_cond);
identity->eos_after = DEFAULT_EOS_AFTER;
gst_base_transform_set_gap_aware (GST_BASE_TRANSFORM_CAST (identity), TRUE);
}
@ -630,6 +645,12 @@ gst_identity_transform_ip (GstBaseTransform * trans, GstBuffer * buf)
goto error_after;
}
if (identity->eos_after >= 0) {
identity->eos_after--;
if (identity->eos_after == 0)
goto eos_after;
}
if (identity->drop_probability > 0.0) {
if ((gfloat) (1.0 * rand () / (RAND_MAX)) < identity->drop_probability)
goto dropped;
@ -699,6 +720,11 @@ error_after:
(_("Failed after iterations as requested.")), (NULL));
return GST_FLOW_ERROR;
}
eos_after:
{
GST_DEBUG_OBJECT (identity, "EOS after iterations as requested.");
return GST_FLOW_EOS;
}
dropped:
{
if (!identity->silent) {
@ -769,6 +795,9 @@ gst_identity_set_property (GObject * object, guint prop_id,
case PROP_DROP_ALLOCATION:
identity->drop_allocation = g_value_get_boolean (value);
break;
case PROP_EOS_AFTER:
identity->eos_after = g_value_get_int (value);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;
@ -835,6 +864,9 @@ gst_identity_get_property (GObject * object, guint prop_id, GValue * value,
case PROP_DROP_ALLOCATION:
g_value_set_boolean (value, identity->drop_allocation);
break;
case PROP_EOS_AFTER:
g_value_set_int (value, identity->eos_after);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break;

View file

@ -79,6 +79,7 @@ struct _GstIdentity {
gboolean blocked;
GstClockTimeDiff ts_offset;
gboolean drop_allocation;
gint eos_after;
};
struct _GstIdentityClass {