/* GStreamer * Copyright (C) <1999> Erik Walthinsen * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Library General Public * License as published by the Free Software Foundation; either * version 2 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Library General Public License for more details. * * You should have received a copy of the GNU Library General Public * License along with this library; if not, write to the * Free Software Foundation, Inc., 59 Temple Place - Suite 330, * Boston, MA 02111-1307, USA. */ #include /* First, include the header file for the plugin, to bring in the * object definition and other useful things. */ #include "gstdvdec.h" #define NTSC_HEIGHT 480 #define NTSC_BUFFER 120000 #define PAL_HEIGHT 576 #define PAL_BUFFER 144000 /* The ElementDetails structure gives a human-readable description * of the plugin, as well as author and version data. */ static GstElementDetails dvdec_details = { "DV (smpte314) decoder plugin", "Decoder/DV", "Uses libdv to decode DV video (libdv.sourceforge.net)", VERSION, "Erik Walthinsen \n" "Wim Taymans ", "(C) 2001-2002", }; /* These are the signals that this element can fire. They are zero- * based because the numbers themselves are private to the object. * LAST_SIGNAL is used for initialization of the signal array. */ enum { ASDF, /* FILL ME */ LAST_SIGNAL }; /* Arguments are identified the same way, but cannot be zero, so you * must leave the ARG_0 entry in as a placeholder. */ enum { ARG_0, /* FILL ME */ }; /* The PadFactory structures describe what pads the element has or * can have. They can be quite complex, but for this dvdec plugin * they are rather simple. */ GST_PAD_TEMPLATE_FACTORY (sink_temp, "sink", GST_PAD_SINK, GST_PAD_ALWAYS, GST_CAPS_NEW ( "dv_dec_sink", "video/dv", "format", GST_PROPS_STRING ("NTSC") ) ) GST_PAD_TEMPLATE_FACTORY (video_src_temp, "video", GST_PAD_SRC, GST_PAD_ALWAYS, GST_CAPS_NEW ( "dv_dec_src", "video/raw", "format", GST_PROPS_FOURCC (GST_MAKE_FOURCC ('Y','U','Y','2')), "width", GST_PROPS_INT (720), "height", GST_PROPS_INT_RANGE (NTSC_HEIGHT, PAL_HEIGHT) ), GST_CAPS_NEW ( "dv_dec_src", "video/raw", "format", GST_PROPS_FOURCC(GST_MAKE_FOURCC('R','G','B',' ')), "bpp", GST_PROPS_INT(24), "depth", GST_PROPS_INT(24), "red_mask", GST_PROPS_INT(0x0000ff), "green_mask", GST_PROPS_INT(0x00ff00), "blue_mask", GST_PROPS_INT(0xff0000), "width", GST_PROPS_INT (720), "height", GST_PROPS_INT_RANGE (NTSC_HEIGHT, PAL_HEIGHT) ) ) GST_PAD_TEMPLATE_FACTORY ( audio_src_temp, "audio", GST_PAD_SRC, GST_PAD_ALWAYS, GST_CAPS_NEW ( "arts_sample", "audio/raw", "format", GST_PROPS_STRING ("int"), "law", GST_PROPS_INT (0), "depth", GST_PROPS_INT (16), "width", GST_PROPS_INT (16), "signed", GST_PROPS_BOOLEAN (TRUE), "channels", GST_PROPS_INT (2), "endianness", GST_PROPS_INT (G_LITTLE_ENDIAN) ) ) /* A number of functon prototypes are given so we can refer to them later. */ static void gst_dvdec_class_init (GstDVDecClass *klass); static void gst_dvdec_init (GstDVDec *dvdec); static void gst_dvdec_loop (GstElement *element); static GstElementStateReturn gst_dvdec_change_state (GstElement *element); static void gst_dvdec_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec); static void gst_dvdec_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec); /* The parent class pointer needs to be kept around for some object * operations. */ static GstElementClass *parent_class = NULL; /* This array holds the ids of the signals registered for this object. * The array indexes are based on the enum up above. */ /*static guint gst_dvdec_signals[LAST_SIGNAL] = { 0 }; */ /* This function is used to register and subsequently return the type * identifier for this object class. On first invocation, it will * register the type, providing the name of the class, struct sizes, * and pointers to the various functions that define the class. */ GType gst_dvdec_get_type (void) { static GType dvdec_type = 0; if (!dvdec_type) { static const GTypeInfo dvdec_info = { sizeof (GstDVDecClass), NULL, NULL, (GClassInitFunc) gst_dvdec_class_init, NULL, NULL, sizeof (GstDVDec), 0, (GInstanceInitFunc) gst_dvdec_init, }; dvdec_type = g_type_register_static (GST_TYPE_ELEMENT, "GstDVDec", &dvdec_info, 0); } return dvdec_type; } /* In order to create an instance of an object, the class must be * initialized by this function. GObject will take care of running * it, based on the pointer to the function provided above. */ static void gst_dvdec_class_init (GstDVDecClass *klass) { /* Class pointers are needed to supply pointers to the private * implementations of parent class methods. */ GObjectClass *gobject_class; GstElementClass *gstelement_class; /* Since the dvdec class contains the parent classes, you can simply * cast the pointer to get access to the parent classes. */ gobject_class = (GObjectClass*) klass; gstelement_class = (GstElementClass*) klass; /* The parent class is needed for class method overrides. */ parent_class = g_type_class_ref (GST_TYPE_ELEMENT); gobject_class->set_property = gst_dvdec_set_property; gobject_class->get_property = gst_dvdec_get_property; gstelement_class->change_state = gst_dvdec_change_state; /* table initialization, only do once */ dv_init(); } /* This function is responsible for initializing a specific instance of * the plugin. */ static void gst_dvdec_init(GstDVDec *dvdec) { dvdec->sinkpad = gst_pad_new_from_template (GST_PAD_TEMPLATE_GET (sink_temp), "sink"); gst_element_add_pad (GST_ELEMENT (dvdec), dvdec->sinkpad); dvdec->videosrcpad = gst_pad_new_from_template (GST_PAD_TEMPLATE_GET (video_src_temp), "video"); gst_element_add_pad (GST_ELEMENT (dvdec), dvdec->videosrcpad); dvdec->audiosrcpad = gst_pad_new_from_template (GST_PAD_TEMPLATE_GET(audio_src_temp), "audio"); gst_element_add_pad(GST_ELEMENT(dvdec),dvdec->audiosrcpad); gst_element_set_loop_function (GST_ELEMENT (dvdec), gst_dvdec_loop); dvdec->decoder = dv_decoder_new (); dvdec->decoder->quality = DV_QUALITY_BEST; dvdec->pool = NULL; } static gboolean gst_dvdec_handle_event (GstDVDec *dvdec) { guint32 remaining; GstEvent *event; GstEventType type; gst_bytestream_get_status (dvdec->bs, &remaining, &event); type = event? GST_EVENT_TYPE (event) : GST_EVENT_UNKNOWN; switch (type) { case GST_EVENT_EOS: gst_pad_event_default (dvdec->sinkpad, event); break; case GST_EVENT_SEEK: g_warning ("seek event\n"); gst_event_free (event); break; case GST_EVENT_FLUSH: g_warning ("flush event\n"); gst_event_free (event); break; case GST_EVENT_DISCONTINUOUS: g_warning ("discont event\n"); gst_event_free (event); break; default: g_warning ("unhandled event %d\n", type); gst_event_free (event); break; } return TRUE; } static void gst_dvdec_loop (GstElement *element) { GstDVDec *dvdec; GstBuffer *buf, *outbuf; guint8 *inframe; guint8 *outframe; guint8 *outframe_ptrs[3]; gint outframe_pitches[3]; gboolean PAL; gint height; dvdec = GST_DVDEC (element); /* first read enough bytes to parse the header */ inframe = gst_bytestream_peek_bytes (dvdec->bs, header_size); if (!inframe) { gst_dvdec_handle_event (dvdec); return; } dv_parse_header (dvdec->decoder, inframe); /* after parsing the header we know the size of the data */ PAL = dv_system_50_fields (dvdec->decoder); height = (PAL ? PAL_HEIGHT : NTSC_HEIGHT); /* then read the read data */ buf = gst_bytestream_read (dvdec->bs, (PAL ? PAL_BUFFER : NTSC_BUFFER)); if (!buf) { gst_dvdec_handle_event (dvdec); return; } /* if we did not negotiate yet, do it now */ if (!GST_PAD_CAPS (dvdec->videosrcpad)) { GstCaps *allowed; GstCaps *trylist; /* we what we are allowed to do */ allowed = gst_pad_get_allowed_caps (dvdec->videosrcpad); /* try to fix our height */ trylist = gst_caps_intersect (allowed, GST_CAPS_NEW ( "dvdec_negotiate", "video/raw", "height", GST_PROPS_INT (height) )); /* prepare for looping */ trylist = gst_caps_normalize (trylist); while (trylist) { GstCaps *to_try = gst_caps_copy_1 (trylist); /* try each format */ if (gst_pad_try_set_caps (dvdec->videosrcpad, to_try)) { guint32 fourcc; /* it worked, try to find what it was again */ gst_caps_get_fourcc_int (to_try, "format", &fourcc); if (fourcc == GST_STR_FOURCC ("RGB ")) { dvdec->space = e_dv_color_rgb; dvdec->bpp = 3; } else { dvdec->space = e_dv_color_yuv; dvdec->bpp = 2; } break; } trylist = trylist->next; } /* oops list exhausted an nothing was found... */ if (!trylist) { gst_element_error (element, "could not negotiate"); return; } } /* try to grab a pool */ if (!dvdec->pool) { dvdec->pool = gst_pad_get_bufferpool (dvdec->videosrcpad); } outbuf = NULL; /* try to get a buffer from the pool if we have one */ if (dvdec->pool) { outbuf = gst_buffer_new_from_pool (dvdec->pool, 0, 0); } /* no buffer from pool, allocate one ourselves */ if (!outbuf) { outbuf = gst_buffer_new (); GST_BUFFER_SIZE (outbuf) = (720 * height) * dvdec->bpp; GST_BUFFER_DATA (outbuf) = g_malloc (GST_BUFFER_SIZE (outbuf)); } outframe = GST_BUFFER_DATA (outbuf); outframe_ptrs[0] = outframe; outframe_ptrs[1] = outframe_ptrs[0] + 720 * height; outframe_ptrs[2] = outframe_ptrs[1] + 360 * height; outframe_pitches[0] = 720 * dvdec->bpp; outframe_pitches[1] = height / 2; outframe_pitches[2] = height / 2; dv_decode_full_frame (dvdec->decoder, GST_BUFFER_DATA (buf), dvdec->space, outframe_ptrs, outframe_pitches); gst_buffer_unref (buf); gst_pad_push (dvdec->videosrcpad, outbuf); } static GstElementStateReturn gst_dvdec_change_state (GstElement *element) { GstDVDec *dvdec = GST_DVDEC (element); switch (GST_STATE_TRANSITION (element)) { case GST_STATE_NULL_TO_READY: break; case GST_STATE_READY_TO_PAUSED: dvdec->bs = gst_bytestream_new (dvdec->sinkpad); break; case GST_STATE_PAUSED_TO_PLAYING: break; case GST_STATE_PLAYING_TO_PAUSED: dvdec->pool = NULL; break; case GST_STATE_PAUSED_TO_READY: gst_bytestream_destroy (dvdec->bs); break; case GST_STATE_READY_TO_NULL: break; default: break; } parent_class->change_state (element); return GST_STATE_SUCCESS; } /* Arguments are part of the Gtk+ object system, and these functions * enable the element to respond to various arguments. */ static void gst_dvdec_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec) { GstDVDec *dvdec; /* It's not null if we got it, but it might not be ours */ g_return_if_fail(GST_IS_DVDEC(object)); /* Get a pointer of the right type. */ dvdec = GST_DVDEC(object); /* Check the argument id to see which argument we're setting. */ switch (prop_id) { default: break; } } /* The set function is simply the inverse of the get fuction. */ static void gst_dvdec_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec) { GstDVDec *dvdec; /* It's not null if we got it, but it might not be ours */ g_return_if_fail(GST_IS_DVDEC(object)); dvdec = GST_DVDEC(object); switch (prop_id) { default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } } /* This is the entry into the plugin itself. When the plugin loads, * this function is called to register everything that the plugin provides. */ static gboolean plugin_init (GModule *module, GstPlugin *plugin) { GstElementFactory *factory; if (!gst_library_load ("gstbytestream")) { gst_info("dvdec: could not load support library: 'gstbytestream'\n"); return FALSE; } /* We need to create an ElementFactory for each element we provide. * This consists of the name of the element, the GType identifier, * and a pointer to the details structure at the top of the file. */ factory = gst_element_factory_new("dvdec", GST_TYPE_DVDEC, &dvdec_details); g_return_val_if_fail(factory != NULL, FALSE); /* The pad templates can be easily generated from the factories above, * and then added to the list of padtemplates for the elementfactory. * Note that the generated padtemplates are stored in static global * variables, for the gst_dvdec_init function to use later on. */ gst_element_factory_add_pad_template (factory, GST_PAD_TEMPLATE_GET(sink_temp)); gst_element_factory_add_pad_template (factory, GST_PAD_TEMPLATE_GET(video_src_temp)); gst_element_factory_add_pad_template (factory, GST_PAD_TEMPLATE_GET(audio_src_temp)); /* The very last thing is to register the elementfactory with the plugin. */ gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (factory)); return TRUE; } GstPluginDesc plugin_desc = { GST_VERSION_MAJOR, GST_VERSION_MINOR, "dvdec", plugin_init };