/* GStreamer divx encoder plugin * Copyright (C) 2003 Ronald Bultje * * 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. */ #ifdef HAVE_CONFIG_H #include "config.h" #endif #include #include "gstdivxenc.h" #include #include /* elementfactory information */ GstElementDetails gst_divxenc_details = { "Divx encoder", "Codec/Video/Encoder", "Divx encoder based on divxencore", "Ronald Bultje " }; GST_PAD_TEMPLATE_FACTORY(sink_template, "sink", GST_PAD_SINK, GST_PAD_ALWAYS, gst_caps_new( "divxdec_src", "video/x-raw-yuv", GST_VIDEO_YUV_PAD_TEMPLATE_PROPS( GST_PROPS_LIST( GST_PROPS_FOURCC(GST_MAKE_FOURCC('R','G','B',' ')), GST_PROPS_FOURCC(GST_MAKE_FOURCC('I','4','2','0')), GST_PROPS_FOURCC(GST_MAKE_FOURCC('Y','U','Y','2')), GST_PROPS_FOURCC(GST_MAKE_FOURCC('Y','V','1','2')), GST_PROPS_FOURCC(GST_MAKE_FOURCC('U','Y','V','Y')) ) ) ), gst_caps_new( "divxdec_src_rgb1", "video/x-raw-rgb", GST_VIDEO_RGB_PAD_TEMPLATE_PROPS_24_32 ), gst_caps_new( "divxdec_src_rgb2", "video/x-raw-rgb", GST_VIDEO_RGB_PAD_TEMPLATE_PROPS_15_16 ) ) GST_PAD_TEMPLATE_FACTORY(src_template, "src", GST_PAD_SRC, GST_PAD_ALWAYS, GST_CAPS_NEW( "divxenc_sink", "video/x-divx", "divxversion", GST_PROPS_INT(5), "width", GST_PROPS_INT_RANGE(0, G_MAXINT), "height", GST_PROPS_INT_RANGE(0, G_MAXINT), "framerate", GST_PROPS_FLOAT_RANGE(0, G_MAXFLOAT) ) ) /* DivxEnc signals and args */ enum { FRAME_ENCODED, LAST_SIGNAL }; enum { ARG_0, ARG_BITRATE, ARG_MAXKEYINTERVAL, ARG_BUFSIZE, ARG_QUALITY }; static void gst_divxenc_class_init (GstDivxEncClass *klass); static void gst_divxenc_base_init (GstDivxEncClass *klass); static void gst_divxenc_init (GstDivxEnc *divxenc); static void gst_divxenc_dispose (GObject *object); static void gst_divxenc_chain (GstPad *pad, GstData *data); static GstPadLinkReturn gst_divxenc_connect (GstPad *pad, GstCaps *vscapslist); /* properties */ static void gst_divxenc_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec); static void gst_divxenc_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec); static GstElementClass *parent_class = NULL; static guint gst_divxenc_signals[LAST_SIGNAL] = { 0 }; static gchar * gst_divxenc_error (int errorcode) { gchar *error; switch (errorcode) { case ENC_BUFFER: error = "Invalid buffer"; break; case ENC_FAIL: error = "Operation failed"; break; case ENC_OK: error = "No error"; break; case ENC_MEMORY: error = "Bad memory location"; break; case ENC_BAD_FORMAT: error = "Invalid format"; break; case ENC_INTERNAL: error = "Internal error"; break; default: error = "Unknown error"; break; } return error; } GType gst_divxenc_get_type(void) { static GType divxenc_type = 0; if (!divxenc_type) { static const GTypeInfo divxenc_info = { sizeof(GstDivxEncClass), (GBaseInitFunc) gst_divxenc_base_init, NULL, (GClassInitFunc) gst_divxenc_class_init, NULL, NULL, sizeof(GstDivxEnc), 0, (GInstanceInitFunc) gst_divxenc_init, }; divxenc_type = g_type_register_static(GST_TYPE_ELEMENT, "GstDivxEnc", &divxenc_info, 0); } return divxenc_type; } static void gst_divxenc_base_init (GstDivxEncClass *klass) { GstElementClass *element_class = GST_ELEMENT_CLASS (klass); gst_element_class_add_pad_template (element_class, GST_PAD_TEMPLATE_GET (sink_template)); gst_element_class_add_pad_template (element_class, GST_PAD_TEMPLATE_GET (src_template)); gst_element_class_set_details (element_class, &gst_divxenc_details); } static void gst_divxenc_class_init (GstDivxEncClass *klass) { GstElementClass *gstelement_class; GObjectClass *gobject_class; gobject_class = (GObjectClass *) klass; gstelement_class = (GstElementClass *) klass; parent_class = g_type_class_ref(GST_TYPE_ELEMENT); g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_BITRATE, g_param_spec_ulong("bitrate","Bitrate", "Target video bitrate", 0,G_MAXULONG,0,G_PARAM_READWRITE)); g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_MAXKEYINTERVAL, g_param_spec_int("max_key_interval","Max. Key Interval", "Maximum number of frames between two keyframes", 0,G_MAXINT,0,G_PARAM_READWRITE)); g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_BUFSIZE, g_param_spec_ulong("buffer_size", "Buffer Size", "Size of the video buffers", 0,G_MAXULONG,0,G_PARAM_READABLE)); g_object_class_install_property(G_OBJECT_CLASS(klass), ARG_QUALITY, g_param_spec_int("quality", "Quality", "Amount of Motion Estimation", 1,5,3,G_PARAM_READWRITE)); gobject_class->set_property = gst_divxenc_set_property; gobject_class->get_property = gst_divxenc_get_property; gobject_class->dispose = gst_divxenc_dispose; gst_divxenc_signals[FRAME_ENCODED] = g_signal_new ("frame_encoded", G_TYPE_FROM_CLASS(klass), G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstDivxEncClass, frame_encoded), NULL, NULL, g_cclosure_marshal_VOID__VOID, G_TYPE_NONE, 0); } static void gst_divxenc_init (GstDivxEnc *divxenc) { /* create the sink pad */ divxenc->sinkpad = gst_pad_new_from_template( GST_PAD_TEMPLATE_GET(sink_template), "sink"); gst_element_add_pad(GST_ELEMENT(divxenc), divxenc->sinkpad); gst_pad_set_chain_function(divxenc->sinkpad, gst_divxenc_chain); gst_pad_set_link_function(divxenc->sinkpad, gst_divxenc_connect); /* create the src pad */ divxenc->srcpad = gst_pad_new_from_template( GST_PAD_TEMPLATE_GET(src_template), "src"); gst_element_add_pad(GST_ELEMENT(divxenc), divxenc->srcpad); /* bitrate, etc. */ divxenc->width = divxenc->height = divxenc->csp = divxenc->bitcnt = -1; divxenc->bitrate = 512 * 1024; divxenc->max_key_interval = -1; /* default - 2*fps */ divxenc->buffer_size = 512 * 1024; divxenc->quality = 3; /* set divx handle to NULL */ divxenc->handle = NULL; } static gboolean gst_divxenc_setup (GstDivxEnc *divxenc) { void *handle = NULL; SETTINGS output; DivXBitmapInfoHeader input; int ret; /* set it up */ memset(&input, 0, sizeof(DivXBitmapInfoHeader)); input.biSize = sizeof(DivXBitmapInfoHeader); input.biWidth = divxenc->width; input.biHeight = divxenc->height; input.biBitCount = divxenc->bitcnt; input.biCompression = divxenc->csp; memset(&output, 0, sizeof(SETTINGS)); output.vbr_mode = RCMODE_VBV_1PASS; output.bitrate = divxenc->bitrate; output.quantizer = 0; output.use_bidirect = 1; output.input_clock = 0; output.input_frame_period = 1000000; output.internal_timescale = divxenc->fps * 1000000; output.max_key_interval = (divxenc->max_key_interval == -1) ? 150 : divxenc->max_key_interval; output.key_frame_threshold = 50; output.vbv_bitrate = 0; output.vbv_size = 0; output.vbv_occupancy = 0; output.complexity_modulation = 0; output.deinterlace = 0; output.quality = divxenc->quality; output.data_partitioning = 0; output.quarter_pel = 1; output.use_gmc = 1; output.psychovisual = 0; output.pv_strength_frame = 0; output.pv_strength_MB = 0; output.interlace_mode = 0; output.enable_crop = 0; output.enable_resize = 0; output.temporal_enable = 1; output.spatial_passes = 3; output.spatial_level = 1.0; output.temporal_level = 1.0; if ((ret = encore(&handle, ENC_OPT_INIT, &input, &output))) { gst_element_error(GST_ELEMENT(divxenc), "Error setting up divx encoder: %s (%d)", gst_divxenc_error(ret), ret); return FALSE; } divxenc->handle = handle; /* set buffer size to theoretical limit (see docs on divx.com) */ divxenc->buffer_size = 6 * divxenc->width * divxenc->height; return TRUE; } static void gst_divxenc_unset (GstDivxEnc *divxenc) { if (divxenc->handle) { encore(divxenc->handle, ENC_OPT_RELEASE, NULL, NULL); divxenc->handle = NULL; } } static void gst_divxenc_dispose (GObject *object) { GstDivxEnc *divxenc = GST_DIVXENC(object); gst_divxenc_unset(divxenc); } static void gst_divxenc_chain (GstPad *pad, GstData *_data) { GstBuffer *buf = GST_BUFFER (_data); GstDivxEnc *divxenc; GstBuffer *outbuf; ENC_FRAME xframe; ENC_RESULT xres; int ret; g_return_if_fail(pad != NULL); g_return_if_fail(GST_IS_PAD(pad)); g_return_if_fail(buf != NULL); divxenc = GST_DIVXENC(GST_OBJECT_PARENT(pad)); outbuf = gst_buffer_new_and_alloc(divxenc->buffer_size); GST_BUFFER_TIMESTAMP(outbuf) = GST_BUFFER_TIMESTAMP(buf); /* encode and so ... */ xframe.image = GST_BUFFER_DATA(buf); xframe.bitstream = (void *) GST_BUFFER_DATA(outbuf); xframe.length = GST_BUFFER_MAXSIZE(outbuf); xframe.produce_empty_frame = 0; if ((ret = encore(divxenc->handle, ENC_OPT_ENCODE, &xframe, &xres))) { gst_element_error(GST_ELEMENT(divxenc), "Error encoding divx frame: %s (%d)", gst_divxenc_error(ret), ret); gst_buffer_unref(buf); return; } GST_BUFFER_SIZE(outbuf) = xframe.length; if (xres.cType == 'I') GST_BUFFER_FLAG_SET(outbuf, GST_BUFFER_KEY_UNIT); /* go out, multiply! */ gst_pad_push(divxenc->srcpad, GST_DATA (outbuf)); /* proclaim destiny */ g_signal_emit(G_OBJECT(divxenc),gst_divxenc_signals[FRAME_ENCODED], 0); /* until the final judgement */ gst_buffer_unref(buf); } static GstPadLinkReturn gst_divxenc_connect (GstPad *pad, GstCaps *vscaps) { GstDivxEnc *divxenc; GstCaps *caps; divxenc = GST_DIVXENC(gst_pad_get_parent (pad)); /* if there's something old around, remove it */ gst_divxenc_unset(divxenc); /* we are not going to act on variable caps */ if (!GST_CAPS_IS_FIXED(vscaps)) return GST_PAD_LINK_DELAYED; for (caps = vscaps; caps != NULL; caps = caps->next) { gint w,h,d; gfloat fps; guint32 fourcc; guint32 divx_cs; gint bitcnt = 0; gst_caps_get_int(caps, "width", &w); gst_caps_get_int(caps, "height", &h); gst_caps_get_float(caps, "framerate", &fps); gst_caps_get_fourcc_int(caps, "format", &fourcc); switch (fourcc) { case GST_MAKE_FOURCC('I','4','2','0'): divx_cs = GST_MAKE_FOURCC('I','4','2','0'); break; case GST_MAKE_FOURCC('Y','U','Y','2'): divx_cs = GST_MAKE_FOURCC('Y','U','Y','2'); break; case GST_MAKE_FOURCC('Y','V','1','2'): divx_cs = GST_MAKE_FOURCC('Y','V','1','2'); break; case GST_MAKE_FOURCC('Y','V','Y','U'): divx_cs = GST_MAKE_FOURCC('Y','V','Y','U'); break; case GST_MAKE_FOURCC('U','Y','V','Y'): divx_cs = GST_MAKE_FOURCC('U','Y','V','Y'); break; case GST_MAKE_FOURCC('R','G','B',' '): gst_caps_get_int(caps, "depth", &d); switch (d) { case 24: divx_cs = 0; bitcnt = 24; break; case 32: divx_cs = 0; bitcnt = 32; break; default: goto trynext; } break; default: goto trynext; } divxenc->csp = divx_cs; divxenc->bitcnt = bitcnt; divxenc->width = w; divxenc->height = h; divxenc->fps = fps; /* try it */ if (gst_divxenc_setup(divxenc)) { GstPadLinkReturn ret; GstCaps *new_caps; new_caps = GST_CAPS_NEW("divxenc_src_caps", "video/x-divx", "divxversion", GST_PROPS_INT(5), "width", GST_PROPS_INT(w), "height", GST_PROPS_INT(h), "framerate", GST_PROPS_FLOAT(fps)); ret = gst_pad_try_set_caps(divxenc->srcpad, new_caps); if (ret <= 0) { gst_divxenc_unset(divxenc); } return ret; } trynext: continue; } /* if we got here - it's not good */ return GST_PAD_LINK_REFUSED; } static void gst_divxenc_set_property (GObject *object, guint prop_id, const GValue *value, GParamSpec *pspec) { GstDivxEnc *divxenc; /* it's not null if we got it, but it might not be ours */ g_return_if_fail (GST_IS_DIVXENC (object)); divxenc = GST_DIVXENC(object); switch (prop_id) { case ARG_BITRATE: divxenc->bitrate = g_value_get_ulong(value); break; case ARG_MAXKEYINTERVAL: divxenc->max_key_interval = g_value_get_int(value); break; case ARG_QUALITY: divxenc->quality = g_value_get_int(value); break; default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } } static void gst_divxenc_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec) { GstDivxEnc *divxenc; /* it's not null if we got it, but it might not be ours */ g_return_if_fail (GST_IS_DIVXENC (object)); divxenc = GST_DIVXENC(object); switch (prop_id) { case ARG_BITRATE: g_value_set_ulong(value, divxenc->bitrate); break; case ARG_BUFSIZE: g_value_set_ulong(value, divxenc->buffer_size); break; case ARG_MAXKEYINTERVAL: g_value_set_int(value, divxenc->max_key_interval); break; case ARG_QUALITY: g_value_set_int(value, divxenc->quality); break; default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } } static gboolean plugin_init (GstPlugin *plugin) { int lib_version; lib_version = encore(NULL, ENC_OPT_VERSION, 0, 0); if (lib_version != ENCORE_VERSION) { g_warning("Version mismatch! This plugin was compiled for " "DivX version %d, while your library has version %d!", ENCORE_VERSION, lib_version); return FALSE; } /* create an elementfactory for the v4lmjpegsrcparse element */ return gst_element_register(plugin, "divxenc", GST_RANK_PRIMARY, GST_TYPE_DIVXENC); } GST_PLUGIN_DEFINE ( GST_VERSION_MAJOR, GST_VERSION_MINOR, "divxenc", "DivX encoder", plugin_init, "5.03", GST_LICENSE_UNKNOWN, "divx4linux", "http://www.divx.com/" )