+++ /dev/null
-/* GStreamer
- * Copyright (C) 2008 Pioneers of the Inevitable <songbird@songbirdnest.com>
- *
- * Authors: Michael Smith <msmith@songbirdnest.com>
- *
- * 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., 51 Franklin St, Fifth Floor,
- * Boston, MA 02110-1301, USA.
- */
-
-#ifdef HAVE_CONFIG_H
-#include "config.h"
-#endif
-
-#include <windows.h>
-#include <mmreg.h>
-#include <msacm.h>
-
-#include <gst/gst.h>
-#include <gst/riff/riff-media.h>
-
-/* This has to be bigger than some unspecified minimum size or things
- * break; I don't understand why (4kB isn't enough). Make it nice and
- * big.
- */
-#define ACM_BUFFER_SIZE (64 * 1024)
-enum
-{
- ARG_0,
- ARG_BITRATE
-};
-
-#define DEFAULT_BITRATE 128000
-
-#define ACMENC_PARAMS_QDATA g_quark_from_static_string("acmenc-params")
-
-#define GST_CAT_DEFAULT acmenc_debug
-GST_DEBUG_CATEGORY_STATIC (acmenc_debug);
-
-static GstStaticPadTemplate acmenc_sink_template =
-GST_STATIC_PAD_TEMPLATE ("sink",
- GST_PAD_SINK, GST_PAD_ALWAYS, GST_STATIC_CAPS ("audio/x-raw, "
- "depth = (int)16, "
- "width = (int)16, "
- "endianness = (int)" G_STRINGIFY (G_BYTE_ORDER) ", "
- "signed = (boolean)TRUE, "
- "channels = (int) [1,2], " "rate = (int)[1, MAX]"));
-
-static GstStaticPadTemplate acmenc_src_template =
-GST_STATIC_PAD_TEMPLATE ("src", GST_PAD_SRC, GST_PAD_ALWAYS,
- GST_STATIC_CAPS_ANY);
-
-static GstElementClass *parent_class = NULL;
-typedef struct _ACMEncClass
-{
- GstElementClass parent_class;
- HACMDRIVERID driverId;
-} ACMEncClass;
-
-typedef struct _ACMEnc
-{
- GstElement parent;
- GstPad *sinkpad;
- GstPad *srcpad;
- gboolean is_setup;
- WAVEFORMATEX infmt;
- WAVEFORMATEX *outfmt;
- HACMDRIVER driver;
- HACMSTREAM stream;
- ACMSTREAMHEADER header;
-
- /* Offset into input buffer to write next data */
- int offset;
-
- /* Number of bytes written */
- int bytes_output;
-
- /* From received caps */
- int rate;
- int channels;
-
- /* Set in properties */
- int selected_bitrate;
- GstCaps *output_caps;
-} ACMEnc;
-
-typedef struct _ACMEncParams
-{
- HACMDRIVERID driverId;
- HMODULE dll;
- gchar *name;
-} ACMEncParams;
-
-static GstCaps *
-acmenc_caps_from_format (WAVEFORMATEX * fmt)
-{
- return gst_riff_create_audio_caps (fmt->wFormatTag, NULL,
- (gst_riff_strf_auds *) fmt, NULL, NULL, NULL, NULL);
-}
-
-static gboolean
-acmenc_set_input_format (ACMEnc * enc, WAVEFORMATEX * infmt)
-{
- infmt->wFormatTag = WAVE_FORMAT_PCM;
- infmt->nChannels = enc->channels;
- infmt->nSamplesPerSec = enc->rate;
- infmt->nAvgBytesPerSec = 2 * enc->channels * enc->rate;
- infmt->nBlockAlign = 2 * enc->channels;
- infmt->wBitsPerSample = 16;
- infmt->cbSize = 0;
- return TRUE;
-}
-
-static BOOL CALLBACK
-acmenc_format_enum (HACMDRIVERID driverId, LPACMFORMATDETAILS fd,
- DWORD_PTR dwInstance, DWORD fdwSupport)
-{
- ACMEnc *enc = (ACMEnc *) dwInstance;
- int oldbrdiff, newbrdiff;
- gboolean oldmatch, newmatch;
- if (!enc->outfmt) {
-
- /* First one is always the best so far */
- enc->outfmt =
- (WAVEFORMATEX *) g_malloc (fd->pwfx->cbSize + sizeof (WAVEFORMATEX));
- memcpy (enc->outfmt, fd->pwfx, fd->pwfx->cbSize + sizeof (WAVEFORMATEX));
- return TRUE;
- }
-
- /* We prefer a format with exact rate/channels match,
- * and the closest bitrate match. Otherwise, we accept
- * the closest bitrate match.
- */
- newmatch = (enc->channels == fd->pwfx->nChannels) &&
- (enc->rate == fd->pwfx->nSamplesPerSec);
- oldmatch = (enc->outfmt->nChannels == enc->channels) &&
- (enc->outfmt->nSamplesPerSec == enc->rate);
- newbrdiff = abs (enc->selected_bitrate - fd->pwfx->nAvgBytesPerSec * 8);
- oldbrdiff = abs (enc->selected_bitrate - enc->outfmt->nAvgBytesPerSec * 8);
- if ((newmatch && (!oldmatch || (newbrdiff < oldbrdiff))) ||
- (!newmatch && !oldmatch && (newbrdiff < oldbrdiff))) {
- g_free (enc->outfmt);
- enc->outfmt =
- (WAVEFORMATEX *) g_malloc (fd->pwfx->cbSize + sizeof (WAVEFORMATEX));
- memcpy (enc->outfmt, fd->pwfx, fd->pwfx->cbSize + sizeof (WAVEFORMATEX));
- }
-
- /* Always return TRUE to continue enumeration */
- return TRUE;
-}
-
-static gboolean
-acmenc_set_format (ACMEnc * enc)
-{
- WAVEFORMATEX *in = NULL;
- ACMFORMATDETAILS details;
- MMRESULT res;
- int maxSize;
-
- /* Input is fixed */
- acmenc_set_input_format (enc, &enc->infmt);
-
- /* Select the closest output that matches */
- res =
- acmMetrics ((HACMOBJ) enc->driver, ACM_METRIC_MAX_SIZE_FORMAT, &maxSize);
-
- /* Set the format to match what we'll be providing */
- in = (WAVEFORMATEX *) g_malloc (maxSize);
- acmenc_set_input_format (enc, in);
- in->cbSize = (WORD) (maxSize - sizeof (*in));
- details.cbStruct = sizeof (details);
- details.dwFormatIndex = 0;
- details.dwFormatTag = WAVE_FORMAT_UNKNOWN;
- details.fdwSupport = 0;
- details.pwfx = in;
- details.cbwfx = maxSize;
-
- /* We set enc->outfmt to the closest match in the callback */
- res =
- acmFormatEnum (enc->driver, &details, acmenc_format_enum, (DWORD_PTR) enc,
- ACM_FORMATENUMF_CONVERT);
- g_free (in);
- if (res) {
- GST_WARNING_OBJECT (enc, "Failed to enumerate formats");
- return FALSE;
- }
- if (!enc->outfmt) {
- GST_WARNING_OBJECT (enc, "No compatible output for input format");
- return FALSE;
- }
- return TRUE;
-}
-
-static gboolean
-acmenc_setup (ACMEnc * enc)
-{
- MMRESULT res;
- ACMEncClass *encclass = (ACMEncClass *) G_OBJECT_GET_CLASS (enc);
- int destBufferSize;
- res = acmDriverOpen (&enc->driver, encclass->driverId, 0);
- if (res) {
- GST_WARNING ("Failed to open ACM driver: %d", res);
- return FALSE;
- }
- if (!acmenc_set_format (enc))
- return FALSE;
- res =
- acmStreamOpen (&enc->stream, enc->driver, &enc->infmt, enc->outfmt, 0, 0,
- 0, ACM_STREAMOPENF_NONREALTIME);
- if (res) {
- GST_WARNING_OBJECT (enc, "Failed to open ACM stream");
- return FALSE;
- }
- enc->header.cbStruct = sizeof (ACMSTREAMHEADER);
- enc->header.fdwStatus = 0;
- enc->header.dwUser = 0;
- enc->header.pbSrc = (BYTE *) g_malloc (ACM_BUFFER_SIZE);
- enc->header.cbSrcLength = ACM_BUFFER_SIZE;
- enc->header.cbSrcLengthUsed = 0;
- enc->header.dwSrcUser = 0;
-
- /* Ask what buffer size we need to use for our output */
- acmStreamSize (enc->stream, ACM_BUFFER_SIZE, (LPDWORD) & destBufferSize,
- ACM_STREAMSIZEF_SOURCE);
- enc->header.pbDst = (BYTE *) g_malloc (destBufferSize);
- enc->header.cbDstLength = destBufferSize;
- enc->header.cbDstLengthUsed = 0;
- enc->header.dwDstUser = 0;
- res = acmStreamPrepareHeader (enc->stream, &enc->header, 0);
- if (res) {
- GST_WARNING_OBJECT (enc, "Failed to prepare ACM stream");
- return FALSE;
- }
- enc->output_caps = acmenc_caps_from_format (enc->outfmt);
- if (enc->output_caps) {
- gst_pad_set_caps (enc->srcpad, enc->output_caps);
- }
- enc->is_setup = TRUE;
- return TRUE;
-}
-
-static void
-acmenc_teardown (ACMEnc * enc)
-{
- if (enc->outfmt) {
- g_free (enc->outfmt);
- enc->outfmt = NULL;
- }
- if (enc->output_caps) {
- gst_caps_unref (enc->output_caps);
- enc->output_caps = NULL;
- }
- g_free (enc->header.pbSrc);
- g_free (enc->header.pbDst);
- memset (&enc->header, 0, sizeof (enc->header));
- if (enc->stream) {
- acmStreamClose (enc->stream, 0);
- enc->stream = 0;
- }
- if (enc->driver) {
- acmDriverClose (enc->driver, 0);
- enc->driver = 0;
- }
- enc->bytes_output = 0;
- enc->offset = 0;
- enc->is_setup = FALSE;
-}
-
-static gboolean
-acmenc_sink_setcaps (GstPad * pad, GstCaps * caps)
-{
- ACMEnc *enc = (ACMEnc *) GST_PAD_PARENT (pad);
- GstStructure *structure;
- gboolean ret;
- structure = gst_caps_get_structure (caps, 0);
- gst_structure_get_int (structure, "channels", &enc->channels);
- gst_structure_get_int (structure, "rate", &enc->rate);
- if (enc->is_setup)
- acmenc_teardown (enc);
- ret = acmenc_setup (enc);
- return ret;
-}
-
-static GstFlowReturn
-acmenc_push_output (ACMEnc * enc)
-{
- GstFlowReturn ret = GST_FLOW_OK;
- if (enc->header.cbDstLengthUsed > 0) {
- GstBuffer *outbuf = gst_buffer_new_and_alloc (enc->header.cbDstLengthUsed);
- if (!outbuf) {
- GST_WARNING_OBJECT (enc, "cannot allocate a new GstBuffer");
- goto done_push_output;
- }
-
- if (gst_buffer_fill (outbuf, 0, enc->header.pbDst,
- enc->header.cbDstLengthUsed) != enc->header.cbDstLengthUsed) {
- gst_buffer_unref (outbuf);
- GST_WARNING_OBJECT (enc, "unable to fill output buffer");
- goto done_push_output;
- }
-
- if (enc->outfmt->nAvgBytesPerSec > 0) {
-
- /* We have a bitrate, so we can create a timestamp, hopefully */
- GST_BUFFER_TIMESTAMP (outbuf) =
- gst_util_uint64_scale_int (enc->bytes_output, GST_SECOND,
- enc->outfmt->nAvgBytesPerSec);
- }
- enc->bytes_output += enc->header.cbDstLengthUsed;
- GST_DEBUG_OBJECT (enc, "Pushing %lu byte encoded buffer",
- enc->header.cbDstLengthUsed);
- ret = gst_pad_push (enc->srcpad, outbuf);
- }
-
-done_push_output:
- return ret;
-}
-
-static GstFlowReturn
-acmenc_chain (GstPad * pad, GstObject * parent, GstBuffer * buf)
-{
- MMRESULT res;
- ACMEnc *enc = (ACMEnc *) GST_PAD_PARENT (pad);
- GstMapInfo map;
- guchar *data;
- gint len;
- int chunklen;
- GstFlowReturn ret = GST_FLOW_OK;
- gst_buffer_map (buf, &map, GST_MAP_READ);
- len = map.size;
- data = map.data;
- while (len) {
- chunklen = MIN (len, ACM_BUFFER_SIZE - enc->offset);
- memcpy (enc->header.pbSrc + enc->offset, data, chunklen);
- enc->header.cbSrcLength = enc->offset + chunklen;
- data += chunklen;
- len -= chunklen;
-
- /* Now we have a buffer ready to go */
- res =
- acmStreamConvert (enc->stream, &enc->header,
- ACM_STREAMCONVERTF_BLOCKALIGN);
- if (res) {
- GST_WARNING_OBJECT (enc, "Failed to encode data");
- break;
- }
- if (enc->header.cbSrcLengthUsed > 0) {
- if (enc->header.cbSrcLengthUsed != enc->header.cbSrcLength) {
-
- /* Only part of input consumed */
- memmove (enc->header.pbSrc,
- enc->header.pbSrc + enc->header.cbSrcLengthUsed,
- enc->header.cbSrcLengthUsed);
- enc->offset -= enc->header.cbSrcLengthUsed;
- }
-
- else
- enc->offset = 0;
- }
-
- /* Write out any data produced */
- acmenc_push_output (enc);
- }
- gst_buffer_unmap (buf, &map);
- return ret;
-}
-
-static GstFlowReturn
-acmenc_finish_stream (ACMEnc * enc)
-{
- MMRESULT res;
- GstFlowReturn ret = GST_FLOW_OK;
- int len;
-
- /* Ensure any remaining input data is consumed */
- len = enc->offset;
- enc->header.cbSrcLength = len;
- res =
- acmStreamConvert (enc->stream, &enc->header,
- ACM_STREAMCONVERTF_BLOCKALIGN | ACM_STREAMCONVERTF_END);
- if (res) {
- GST_WARNING_OBJECT (enc, "Failed to encode data");
- return ret;
- }
- ret = acmenc_push_output (enc);
- return ret;
-}
-
-static gboolean
-acmenc_sink_event (GstPad * pad, GstObject * parent, GstEvent * event)
-{
- ACMEnc *enc = (ACMEnc *) GST_PAD_PARENT (pad);
- gboolean res;
- switch (GST_EVENT_TYPE (event)) {
- case GST_EVENT_CAPS:{
- GstCaps *caps;
- gst_event_parse_caps (event, &caps);
- return acmenc_sink_setcaps (pad, caps);
- }
- case GST_EVENT_EOS:
- acmenc_finish_stream (enc);
- res = gst_pad_push_event (enc->srcpad, event);
- break;
- default:
- res = gst_pad_push_event (enc->srcpad, event);
- break;
- }
- return res;
-}
-
-static void
-acmenc_dispose (GObject * obj)
-{
- ACMEnc *enc = (ACMEnc *) obj;
- acmenc_teardown (enc);
- G_OBJECT_CLASS (parent_class)->dispose (obj);
-} static void
-
-acmenc_init (ACMEnc * enc)
-{
- enc->sinkpad =
- gst_pad_new_from_static_template (&acmenc_sink_template, "sink");
- gst_pad_set_chain_function (enc->sinkpad, GST_DEBUG_FUNCPTR (acmenc_chain));
- gst_pad_set_event_function (enc->sinkpad,
- GST_DEBUG_FUNCPTR (acmenc_sink_event));
- gst_element_add_pad (GST_ELEMENT (enc), enc->sinkpad);
- enc->srcpad = gst_pad_new_from_static_template (&acmenc_src_template, "src");
- gst_element_add_pad (GST_ELEMENT (enc), enc->srcpad);
- enc->selected_bitrate = DEFAULT_BITRATE;
-} static void
-
-acmenc_set_property (GObject * obj, guint prop_id, const GValue * value,
- GParamSpec * pspec)
-{
- ACMEnc *enc = (ACMEnc *) obj;
- switch (prop_id) {
- case ARG_BITRATE:
- enc->selected_bitrate = g_value_get_int (value);
- break;
- default:
- G_OBJECT_WARN_INVALID_PROPERTY_ID (obj, prop_id, pspec);
- }
-}
-
-static void
-acmenc_get_property (GObject * obj, guint prop_id, GValue * value,
- GParamSpec * pspec)
-{
- ACMEnc *enc = (ACMEnc *) obj;
- switch (prop_id) {
- case ARG_BITRATE:
- g_value_set_int (value, enc->selected_bitrate);
- break;
- default:
- G_OBJECT_WARN_INVALID_PROPERTY_ID (obj, prop_id, pspec);
- }
-}
-
-static void
-acmenc_class_init (ACMEncClass * klass)
-{
- GObjectClass *gobjectclass = (GObjectClass *) klass;
- GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
- ACMEncParams *params;
- ACMDRIVERDETAILS driverdetails;
- gchar *shortname, *longname, *detail, *description;
- MMRESULT res;
-
- parent_class = (GstElementClass *) g_type_class_peek_parent (klass);
- gobjectclass->dispose = acmenc_dispose;
- gobjectclass->set_property = acmenc_set_property;
- gobjectclass->get_property = acmenc_get_property;
- g_object_class_install_property (gobjectclass, ARG_BITRATE,
- g_param_spec_int ("bitrate", "Bitrate", "Bitrate to encode at (in bps)",
- 0, 1000000, DEFAULT_BITRATE, G_PARAM_READWRITE));
-
- gst_element_class_add_static_pad_template (element_class,
- &acmenc_sink_template);
- gst_element_class_add_static_pad_template (element_class,
- &acmenc_src_template);
- params =
- (ACMEncParams *) g_type_get_qdata (G_OBJECT_CLASS_TYPE (klass),
- ACMENC_PARAMS_QDATA);
- g_assert (params);
- memset (&driverdetails, 0, sizeof (driverdetails));
- driverdetails.cbStruct = sizeof (driverdetails);
- res = acmDriverDetails (params->driverId, &driverdetails, 0);
- if (res) {
- GST_WARNING ("Could not get driver details: %d", res);
- }
- shortname =
- g_utf16_to_utf8 ((gunichar2 *) driverdetails.szShortName, -1, NULL, NULL,
- NULL);
- longname =
- g_utf16_to_utf8 ((gunichar2 *) driverdetails.szLongName, -1, NULL, NULL,
- NULL);
- detail = g_strdup_printf ("ACM Encoder: %s", (shortname
- && *shortname) ? shortname : params->name);
- description = g_strdup_printf ("ACM Encoder: %s", (longname
- && *longname) ? longname : params->name);
- gst_element_class_set_static_metadata (element_class, detail,
- "Codec/Encoder/Audio", description,
- "Pioneers of the Inevitable <songbird@songbirdnest.com>");
- g_free (shortname);
- g_free (longname);
- g_free (description);
- g_free (detail);
- klass->driverId = params->driverId;
-}
-
-static ACMEncParams *
-acmenc_open_driver (wchar_t * filename)
-{
- HACMDRIVERID driverid = NULL;
- HMODULE mod = NULL;
- FARPROC func;
- MMRESULT res;
- ACMEncParams *params;
- mod = LoadLibraryW (filename);
- if (!mod) {
- GST_WARNING ("Failed to load ACM");
- goto done;
- }
- func = GetProcAddress (mod, "DriverProc");
- if (!func) {
- GST_WARNING ("Failed to find 'DriverProc' in ACM");
- goto done;
- }
- res =
- acmDriverAdd (&driverid, mod, (LPARAM) func, 0, ACM_DRIVERADDF_FUNCTION);
- if (res) {
- GST_WARNING ("Failed to add ACM driver: %d", res);
- goto done;
- }
- params = g_new0 (ACMEncParams, 1);
- params->dll = mod;
- params->driverId = driverid;
- return params;
-done:if (driverid)
- acmDriverRemove (driverid, 0);
- if (mod)
- FreeLibrary (mod);
- return NULL;
-}
-
-static gboolean
-acmenc_register_file (GstPlugin * plugin, wchar_t * filename)
-{
- ACMEncParams *params;
- gchar *type_name, *name;
- GType type;
- GTypeInfo typeinfo = {
- sizeof (ACMEncClass),
- NULL, NULL,
- (GClassInitFunc) acmenc_class_init, NULL, NULL, sizeof (ACMEnc),
- 0, (GInstanceInitFunc) acmenc_init,
- };
- params = acmenc_open_driver (filename);
- if (!params)
- return FALSE;
-
- /* Strip .acm off the end, convert to utf-8 */
- name =
- g_utf16_to_utf8 (filename, (glong) wcslen (filename) - 4, NULL, NULL,
- NULL);
- params->name = name;
- type_name = g_strdup_printf ("acmenc_%s", name);
- type = g_type_register_static (GST_TYPE_ELEMENT, type_name, &typeinfo, 0);
-
- /* Store params in type qdata */
- g_type_set_qdata (type, ACMENC_PARAMS_QDATA, (gpointer) params);
-
- /* register type */
- if (!gst_element_register (plugin, type_name, GST_RANK_NONE, type)) {
- g_warning ("Failed to register %s", type_name);
- g_type_set_qdata (type, ACMENC_PARAMS_QDATA, NULL);
- g_free (name);
- g_free (type_name);
- g_free (params);
- return FALSE;
- }
- g_free (type_name);
- return TRUE;
-}
-
-static gboolean
-acmenc_register (GstPlugin * plugin)
-{
- int res;
- wchar_t dirname[1024];
- WIN32_FIND_DATAW filedata;
- HANDLE find;
- res = GetSystemDirectoryW (dirname, sizeof (dirname) / sizeof (wchar_t));
- if (!res || res > 1000) {
- GST_WARNING ("Couldn't get system directory");
- return FALSE;
- }
- wcscat (dirname, L"\\*.acm");
- find = FindFirstFileW (dirname, &filedata);
- if (find == INVALID_HANDLE_VALUE) {
- GST_WARNING ("Failed to find ACM files: %lx", GetLastError ());
- return FALSE;
- }
-
- do {
- char *filename =
- g_utf16_to_utf8 ((gunichar2 *) filedata.cFileName, -1, NULL, NULL,
- NULL);
- GST_INFO ("Registering ACM filter from file %s", filename);
- if (acmenc_register_file (plugin, filedata.cFileName))
- GST_INFO ("Loading filter from ACM '%s' succeeded", filename);
-
- else
- GST_WARNING ("Loading filter from ACM '%s' failed", filename);
- g_free (filename);
- } while (FindNextFileW (find, &filedata));
- FindClose (find);
- return TRUE;
-}
-
-static gboolean
-plugin_init (GstPlugin * plugin)
-{
- gboolean res;
- GST_DEBUG_CATEGORY_INIT (acmenc_debug, "acmenc", 0, "ACM Encoders");
- GST_INFO ("Registering ACM encoders");
- res = acmenc_register (plugin);
- return res;
-}
-
-GST_PLUGIN_DEFINE (GST_VERSION_MAJOR, GST_VERSION_MINOR, acmenc,
- "ACM Encoder wrapper", plugin_init, VERSION, "LGPL", GST_PACKAGE_NAME,
- GST_PACKAGE_ORIGIN)
+++ /dev/null
-/* GStreamer
- * Copyright (C) 2008 Pioneers of the Inevitable <songbird@songbirdnest.com>
- *
- * Authors: Michael Smith <msmith@songbirdnest.com>
- *
- * 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., 51 Franklin St, Fifth Floor,
- * Boston, MA 02110-1301, USA.
- */
-
-#ifdef HAVE_CONFIG_H
-#include "config.h"
-#endif
-
-#include <windows.h>
-#include <mmreg.h>
-#include <msacm.h>
-
-#include <gst/gst.h>
-#include <gst/riff/riff-media.h>
-
-#define ACM_BUFFER_SIZE (64 * 1024)
-
-#define GST_TYPE_ACM_MP3_DEC \
- (acmmp3dec_get_type())
-#define GST_ACM_MP3_DEC(obj) \
- (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_ACM_MP3_DEC,ACMMP3Dec))
-#define GST_ACM_MP3_DEC_CLASS(klass) \
- (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_ACM_MP3_DEC,ACMMP3DecClass))
-#define GST_IS_ACM_MP3_DEC(obj) \
- (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_ACM_MP3_DEC))
-#define GST_IS_ACM_MP3_DEC_CLASS(klass) \
- (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_ACM_MP3_DEC))
-
-#define GST_CAT_DEFAULT acmmp3dec_debug
-GST_DEBUG_CATEGORY_STATIC (acmmp3dec_debug);
-
-static GstStaticPadTemplate acmmp3dec_src_template =
-GST_STATIC_PAD_TEMPLATE ("src",
- GST_PAD_SRC,
- GST_PAD_ALWAYS,
- GST_STATIC_CAPS ("audio/x-raw, "
- "depth = (int)16, "
- "width = (int)16, "
- "endianness = (int)" G_STRINGIFY (G_BYTE_ORDER) ", "
- "signed = (boolean)TRUE, "
- "channels = (int) [1,2], " "rate = (int)[1, MAX]")
- );
-
-static GstStaticPadTemplate acmmp3dec_sink_template =
-GST_STATIC_PAD_TEMPLATE ("sink",
- GST_PAD_SINK,
- GST_PAD_ALWAYS,
- GST_STATIC_CAPS ("audio/mpeg, "
- "mpegversion = (int)1, "
- "layer = (int)3, "
- "rate = (int){ 8000, 11025, 12000, 16000, 22050, 24000, "
- " 32000, 44100, 48000 }, "
- "channels = (int)[1,2], " "parsed = (boolean) true")
- );
-
-typedef struct _ACMMP3DecClass
-{
- GstElementClass parent_class;
-} ACMMP3DecClass;
-
-typedef struct _ACMMP3Dec
-{
- GstElement parent;
-
- GstPad *sinkpad;
- GstPad *srcpad;
-
- gboolean is_setup;
-
- MPEGLAYER3WAVEFORMAT infmt;
- WAVEFORMATEX outfmt;
-
- HACMSTREAM stream;
- ACMSTREAMHEADER header;
-
- /* Next timestamp to push */
- GstClockTime timestamp;
-
- /* From received caps */
- int rate;
- int channels;
-
- GstCaps *output_caps;
-} ACMMP3Dec;
-
-GType acmmp3dec_get_type (void);
-
-#define acmmp3dec_parent_class parent_class
-G_DEFINE_TYPE (ACMMP3Dec, acmmp3dec, GST_TYPE_ELEMENT);
-
-static GstCaps *
-acmmp3dec_caps_from_format (WAVEFORMATEX * fmt)
-{
- return gst_riff_create_audio_caps (fmt->wFormatTag,
- NULL, (gst_riff_strf_auds *) fmt, NULL, NULL, NULL, NULL);
-}
-
-static gboolean
-acmmp3dec_set_input_format (ACMMP3Dec * dec)
-{
- dec->infmt.wfx.wFormatTag = WAVE_FORMAT_MPEGLAYER3;
- dec->infmt.wfx.nChannels = dec->channels;
- dec->infmt.wfx.nSamplesPerSec = dec->rate;
- dec->infmt.wfx.nAvgBytesPerSec = 8000; /* Has to be set, but actual
- value doesn't matter */
- dec->infmt.wfx.nBlockAlign = 1;
- dec->infmt.wfx.wBitsPerSample = 0;
- dec->infmt.wfx.cbSize = MPEGLAYER3_WFX_EXTRA_BYTES;
- dec->infmt.wID = MPEGLAYER3_ID_MPEG;
- dec->infmt.fdwFlags = MPEGLAYER3_FLAG_PADDING_OFF;
- dec->infmt.nBlockSize = 1; /* Needs to be non-zero, actual
- value doesn't matter */
- dec->infmt.nFramesPerBlock = 1;
- dec->infmt.nCodecDelay = 0;
-
- return TRUE;
-}
-
-static gboolean
-acmmp3dec_set_output_format (ACMMP3Dec * dec)
-{
- dec->outfmt.wFormatTag = WAVE_FORMAT_PCM;
- dec->outfmt.nChannels = dec->channels;
- dec->outfmt.nSamplesPerSec = dec->rate;
- dec->outfmt.nAvgBytesPerSec = 2 * dec->channels * dec->rate;
- dec->outfmt.nBlockAlign = 2 * dec->channels;
- dec->outfmt.wBitsPerSample = 16;
- dec->outfmt.cbSize = 0;
-
- return TRUE;
-}
-
-static gboolean
-acmmp3dec_setup (ACMMP3Dec * dec)
-{
- MMRESULT res;
- int destBufferSize;
-
- acmmp3dec_set_input_format (dec);
- acmmp3dec_set_output_format (dec);
-
- res =
- acmStreamOpen (&dec->stream, NULL, (LPWAVEFORMATEX) & dec->infmt,
- &dec->outfmt, 0, 0, 0, 0);
- if (res) {
- GST_WARNING_OBJECT (dec, "Failed to open ACM stream: %d", res);
- return FALSE;
- }
-
- dec->header.cbStruct = sizeof (ACMSTREAMHEADER);
- dec->header.fdwStatus = 0;
- dec->header.dwUser = 0;
-
- dec->header.pbSrc = (BYTE *) g_malloc (ACM_BUFFER_SIZE);
- dec->header.cbSrcLength = ACM_BUFFER_SIZE;
- dec->header.cbSrcLengthUsed = 0;
- dec->header.dwSrcUser = 0;
-
- /* Ask what buffer size we need to use for our output */
- acmStreamSize (dec->stream, ACM_BUFFER_SIZE,
- (LPDWORD) & destBufferSize, ACM_STREAMSIZEF_SOURCE);
-
- dec->header.pbDst = (BYTE *) g_malloc (destBufferSize);
- dec->header.cbDstLength = destBufferSize;
- dec->header.cbDstLengthUsed = 0;
- dec->header.dwDstUser = 0;
-
- res = acmStreamPrepareHeader (dec->stream, &dec->header, 0);
- if (res) {
- GST_WARNING_OBJECT (dec, "Failed to prepare ACM stream: %x", res);
- return FALSE;
- }
-
- dec->output_caps = acmmp3dec_caps_from_format (&dec->outfmt);
- if (dec->output_caps) {
- gst_pad_set_caps (dec->srcpad, dec->output_caps);
- }
-
- dec->timestamp = GST_CLOCK_TIME_NONE;
- dec->is_setup = TRUE;
- return TRUE;
-}
-
-static void
-acmmp3dec_teardown (ACMMP3Dec * dec)
-{
- if (dec->output_caps) {
- gst_caps_unref (dec->output_caps);
- dec->output_caps = NULL;
- }
-
- if (dec->header.fdwStatus & ACMSTREAMHEADER_STATUSF_PREPARED)
- acmStreamUnprepareHeader (dec->stream, &dec->header, 0);
- g_free (dec->header.pbSrc);
- g_free (dec->header.pbDst);
- memset (&dec->header, 0, sizeof (dec->header));
-
- if (dec->stream) {
- acmStreamClose (dec->stream, 0);
- dec->stream = 0;
- }
-
- dec->timestamp = GST_CLOCK_TIME_NONE;
- dec->is_setup = FALSE;
-}
-
-static gboolean
-acmmp3dec_sink_setcaps (GstPad * pad, GstCaps * caps)
-{
- ACMMP3Dec *dec = (ACMMP3Dec *) GST_PAD_PARENT (pad);
- GstStructure *structure;
- gboolean ret;
-
- structure = gst_caps_get_structure (caps, 0);
- gst_structure_get_int (structure, "channels", &dec->channels);
- gst_structure_get_int (structure, "rate", &dec->rate);
-
- if (dec->is_setup)
- acmmp3dec_teardown (dec);
- ret = acmmp3dec_setup (dec);
-
- return ret;
-}
-
-static GstFlowReturn
-acmmp3dec_push_output (ACMMP3Dec * dec)
-{
- GstFlowReturn ret = GST_FLOW_OK;
-
- if (dec->header.cbDstLengthUsed > 0) {
- GstBuffer *outbuf = gst_buffer_new_and_alloc (dec->header.cbDstLengthUsed);
- if (!outbuf) {
- GST_WARNING_OBJECT (dec, "cannot allocate a new GstBuffer");
- goto done_push_output;
- }
-
- if (gst_buffer_fill (outbuf, 0, dec->header.pbDst,
- dec->header.cbDstLengthUsed) != dec->header.cbDstLengthUsed) {
- gst_buffer_unref (outbuf);
- GST_WARNING_OBJECT (dec, "unable to fill output buffer");
- goto done_push_output;
- }
-
- if (dec->timestamp != GST_CLOCK_TIME_NONE)
- GST_BUFFER_TIMESTAMP (outbuf) = dec->timestamp;
- GST_BUFFER_DURATION (outbuf) =
- gst_util_uint64_scale_int (dec->header.cbDstLengthUsed, GST_SECOND,
- dec->rate * dec->channels * 2);
-
- GST_DEBUG_OBJECT (dec, "decoded buffer has ts %d, duration %d",
- (int) (GST_BUFFER_TIMESTAMP (outbuf)),
- (int) (GST_BUFFER_DURATION (outbuf)));
-
- if (dec->timestamp != GST_CLOCK_TIME_NONE)
- dec->timestamp += GST_BUFFER_DURATION (outbuf);
-
- GST_DEBUG_OBJECT (dec, "Pushing %lu byte decoded buffer",
- dec->header.cbDstLengthUsed);
- ret = gst_pad_push (dec->srcpad, outbuf);
- } else
- GST_DEBUG_OBJECT (dec, "Not pushing decoded buffer, no output");
-
-done_push_output:
- return ret;
-}
-
-static GstFlowReturn
-acmmp3dec_chain (GstPad * pad, GstObject * parent, GstBuffer * buf)
-{
- MMRESULT res;
- ACMMP3Dec *dec = (ACMMP3Dec *) GST_PAD_PARENT (pad);
- GstMapInfo map;
- GstFlowReturn ret = GST_FLOW_OK;
-
- gst_buffer_map (buf, &map, GST_MAP_READ);
-
- if (map.size > ACM_BUFFER_SIZE) {
- GST_WARNING_OBJECT (dec, "Impossibly large mp3 frame!");
- ret = GST_FLOW_ERROR;
- goto done;
- }
-
- if (GST_BUFFER_TIMESTAMP (buf) != GST_CLOCK_TIME_NONE &&
- (dec->timestamp == GST_CLOCK_TIME_NONE ||
- GST_BUFFER_FLAG_IS_SET (buf, GST_BUFFER_FLAG_DISCONT))) {
- dec->timestamp = GST_BUFFER_TIMESTAMP (buf);
- }
-
- memcpy (dec->header.pbSrc, map.data, map.size);
- dec->header.cbSrcLength = map.size;
-
- /* Now we have a buffer ready to go */
- res = acmStreamConvert (dec->stream, &dec->header,
- ACM_STREAMCONVERTF_BLOCKALIGN);
- if (res) {
- GST_WARNING_OBJECT (dec, "Failed to decode data");
- ret = GST_FLOW_OK; /* Maybe it was just a corrupt frame */
- goto done;
- }
-
- if (dec->header.cbSrcLengthUsed > 0) {
- if (dec->header.cbSrcLengthUsed != dec->header.cbSrcLength) {
- GST_WARNING_OBJECT (dec, "ACM decoder didn't consume all data!");
- /* We could handle this, but it shouldn't be possible, so don't try
- * for now */
- ret = GST_FLOW_ERROR;
- goto done;
- }
-
- /* Write out any data produced */
- acmmp3dec_push_output (dec);
- }
-
-done:
- gst_buffer_unmap (buf, &map);
- gst_buffer_unref (buf);
-
- return ret;
-}
-
-static GstFlowReturn
-acmmp3dec_finish_stream (ACMMP3Dec * dec)
-{
- MMRESULT res;
- GstFlowReturn ret = GST_FLOW_OK;
-
- dec->header.cbSrcLength = 0;
-
- /* Flush out any remaining data internal to the decoder */
- res = acmStreamConvert (dec->stream, &dec->header,
- ACM_STREAMCONVERTF_BLOCKALIGN | ACM_STREAMCONVERTF_END);
- if (res) {
- GST_WARNING_OBJECT (dec, "Failed to decode data");
- return ret;
- }
-
- ret = acmmp3dec_push_output (dec);
- return ret;
-}
-
-static gboolean
-acmmp3dec_sink_event (GstPad * pad, GstObject * parent, GstEvent * event)
-{
- ACMMP3Dec *dec = (ACMMP3Dec *) GST_PAD_PARENT (pad);
-
- switch (GST_EVENT_TYPE (event)) {
- case GST_EVENT_CAPS:{
- GstCaps *caps;
- gst_event_parse_caps (event, &caps);
- return acmmp3dec_sink_setcaps (pad, caps);
- }
- case GST_EVENT_EOS:
- acmmp3dec_finish_stream (dec);
- break;
- case GST_EVENT_FLUSH_STOP:
- if (dec->stream)
- acmStreamReset (dec->stream, 0);
- dec->timestamp = GST_CLOCK_TIME_NONE;
- break;
- default:
- break;
- }
-
-
- return gst_pad_push_event (dec->srcpad, event);
-}
-
-static void
-acmmp3dec_dispose (GObject * obj)
-{
- ACMMP3Dec *dec = (ACMMP3Dec *) obj;
-
- if (dec->is_setup)
- acmmp3dec_teardown (dec);
-
- G_OBJECT_CLASS (parent_class)->dispose (obj);
-}
-
-static void
-acmmp3dec_init (ACMMP3Dec * dec)
-{
- dec->sinkpad =
- gst_pad_new_from_static_template (&acmmp3dec_sink_template, "sink");
- gst_pad_set_chain_function (dec->sinkpad,
- GST_DEBUG_FUNCPTR (acmmp3dec_chain));
- gst_pad_set_event_function (dec->sinkpad,
- GST_DEBUG_FUNCPTR (acmmp3dec_sink_event));
- gst_element_add_pad (GST_ELEMENT (dec), dec->sinkpad);
-
- dec->srcpad =
- gst_pad_new_from_static_template (&acmmp3dec_src_template, "src");
- gst_element_add_pad (GST_ELEMENT (dec), dec->srcpad);
-
-}
-
-static void
-acmmp3dec_class_init (ACMMP3DecClass * klass)
-{
- GObjectClass *gobjectclass = (GObjectClass *) klass;
- GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
- gobjectclass->dispose = acmmp3dec_dispose;
-
- gst_element_class_add_static_pad_template (element_class,
- &acmmp3dec_sink_template);
- gst_element_class_add_static_pad_template (element_class,
- &acmmp3dec_src_template);
- gst_element_class_set_static_metadata (element_class, "ACM MP3 decoder",
- "Codec/Decoder/Audio", "Decode MP3 using ACM decoder",
- "Pioneers of the Inevitable <songbird@songbirdnest.com");
-}
-
-static gboolean
-plugin_init (GstPlugin * plugin)
-{
- GST_DEBUG_CATEGORY_INIT (acmmp3dec_debug, "acmmp3dec", 0, "ACM Decoders");
-
- GST_INFO ("Registering ACM MP3 decoder");
- if (!gst_element_register (plugin, "acmmp3dec", GST_RANK_PRIMARY,
- GST_TYPE_ACM_MP3_DEC)) {
- return FALSE;
- }
-
- return TRUE;
-}
-
-GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,
- GST_VERSION_MINOR,
- acmmp3dec,
- "ACM MP3 Decoder",
- plugin_init, VERSION, "LGPL", GST_PACKAGE_NAME, GST_PACKAGE_ORIGIN)
projects / platform / upstream / gstreamer.git / commitdiff