summaryrefslogtreecommitdiffstats
path: root/ext/tarkin
diff options
context:
space:
mode:
authorWim Taymans <wim.taymans@gmail.com>2002-02-03 16:30:31 +0000
committerWim Taymans <wim.taymans@gmail.com>2002-02-03 16:30:31 +0000
commit57ce0321c8971f964008920003a3e8b19649df85 (patch)
treebabe3c6e4473b4f06142b587db73f2212ec44c42 /ext/tarkin
parentb1fea046869baeed3c6835f20db9f5de9fd912d2 (diff)
downloadgst-plugins-bad-57ce0321c8971f964008920003a3e8b19649df85.tar.gz
gst-plugins-bad-57ce0321c8971f964008920003a3e8b19649df85.tar.bz2
gst-plugins-bad-57ce0321c8971f964008920003a3e8b19649df85.zip
Added a tarkin encoder/decoder plugin.
Original commit message from CVS: Added a tarkin encoder/decoder plugin. I moved the tarking CVS code in here temporarily until they have a library (hence this plugin is in ext) test with: ./gst-launch filesrc location=/opt/data/shihad.mpg ! mpegdemux video_00! { queue ! mpeg2dec ! colorspace ! tarkinenc bitrate=3000 ! disksink location=out.ogg } ./gst-launch filesrc location=out.ogg ! tarkindec ! colorspace ! xvideosink
Diffstat (limited to 'ext/tarkin')
-rw-r--r--ext/tarkin/Makefile.am15
-rw-r--r--ext/tarkin/README8
-rw-r--r--ext/tarkin/TODO42
-rw-r--r--ext/tarkin/WHAT_THE_HECK_IS_THIS_CODE_DOING61
-rw-r--r--ext/tarkin/golomb.h132
-rw-r--r--ext/tarkin/gsttarkin.c156
-rw-r--r--ext/tarkin/gsttarkindec.c301
-rw-r--r--ext/tarkin/gsttarkindec.h83
-rw-r--r--ext/tarkin/gsttarkinenc.c371
-rw-r--r--ext/tarkin/gsttarkinenc.h84
-rw-r--r--ext/tarkin/info.c551
-rw-r--r--ext/tarkin/mem.c143
-rw-r--r--ext/tarkin/mem.h32
-rw-r--r--ext/tarkin/rle.h143
-rw-r--r--ext/tarkin/tarkin.c418
-rw-r--r--ext/tarkin/tarkin.h239
-rw-r--r--ext/tarkin/wavelet.c121
-rw-r--r--ext/tarkin/wavelet.h55
-rw-r--r--ext/tarkin/wavelet_coeff.c506
-rw-r--r--ext/tarkin/wavelet_xform.c391
-rw-r--r--ext/tarkin/yuv.c224
-rw-r--r--ext/tarkin/yuv.h21
22 files changed, 4097 insertions, 0 deletions
diff --git a/ext/tarkin/Makefile.am b/ext/tarkin/Makefile.am
new file mode 100644
index 00000000..86acce1f
--- /dev/null
+++ b/ext/tarkin/Makefile.am
@@ -0,0 +1,15 @@
+plugindir = $(libdir)/gst
+
+plugin_LTLIBRARIES = libgsttarkin.la
+
+libgsttarkin_la_SOURCES = tarkin.c \
+ mem.c wavelet.c wavelet_xform.c \
+ wavelet_coeff.c yuv.c info.c \
+ gsttarkin.c gsttarkinenc.c gsttarkindec.c
+
+libgsttarkin_la_CFLAGS = $(GST_CFLAGS) $(VORBIS_CFLAGS) -DTYPE_BITS=10 -DTYPE=int16_t -DRLECODER
+## AM_PATH_VORBIS also sets VORBISENC_LIBS
+libgsttarkin_la_LIBADD = $(GST_LIBS) $(VORBIS_LIBS) $(VORBISENC_LIBS)
+libgsttarkin_la_LDFLAGS = @GST_PLUGIN_LDFLAGS@
+
+noinst_HEADERS = gsttarkinenc.h
diff --git a/ext/tarkin/README b/ext/tarkin/README
new file mode 100644
index 00000000..ee91356d
--- /dev/null
+++ b/ext/tarkin/README
@@ -0,0 +1,8 @@
+This is a video codec based on an integer wavelet in 3 dimensions (x,
+y, and time/frame). What documentation exists so far is on the
+vorbis-dev and (now) tarkin-dev mailing lists at xiph.org. Some brief
+documentation can be found in the w3d/docs directory.
+
+For sample / test streams, see http://media.xiph.org/
+(and feel free to submit more streams if you have them).
+
diff --git a/ext/tarkin/TODO b/ext/tarkin/TODO
new file mode 100644
index 00000000..909d3771
--- /dev/null
+++ b/ext/tarkin/TODO
@@ -0,0 +1,42 @@
+
+
+Most important things:
+
+ - the entropy coder, replace static huffman
+ - clean up the pnsr tools
+ - write docs and do some performance analysis, compare to other codecs
+ - think about a multiresolution multidimensional motion flow detection scheme,
+ Marco posted a good paper comparing different algorithms to do this
+
+
+Open bugs and stuff required to fix them:
+
+ - wavelet xform bug at short rows, see workaround in wavelet_xform.c
+ - (4,x) and (x,4) wavelet implementations have a bug which causes round-off
+ errors in the two least significand bits
+
+
+Wavelet-related TODO's:
+
+ - remove unecessairy copying in inverse xform
+ - improve truncation table setup
+ - try other approaches to encode coefficients, jack was talking about VQ
+ and reuse vorbis code
+ - write avitotarkin/quicktimetotarkin/mpegtotarkin/player/recorder
+ (a libsndfile/libaudiofile/libao alike video library would be great !)
+ - profile
+ - add special transform functions for large strides to prevent cache misses
+ - mmx/3dnow/sse/altivec
+
+
+Other:
+
+ - u and v buffers could get quarter size already at color conversion
+ this would speed up the whole algorithm; perhaps this should get
+ configurable
+ - fast internal 16bitY/U/V->15/16bitRGB for display could make sense
+ - the wavelet codec could be used for still image compression too
+ (we just have to define a file format with all goodies you can imagine;)
+ - to make it perfect someone has to write a good bilevel compressor and
+ mask seperation algorithm
+
diff --git a/ext/tarkin/WHAT_THE_HECK_IS_THIS_CODE_DOING b/ext/tarkin/WHAT_THE_HECK_IS_THIS_CODE_DOING
new file mode 100644
index 00000000..46602605
--- /dev/null
+++ b/ext/tarkin/WHAT_THE_HECK_IS_THIS_CODE_DOING
@@ -0,0 +1,61 @@
+
+This is deprecated. Take a look in the w3d/docs directory.
+
+The command line semantics are changed. You have to call the test program
+now like this:
+
+./tarkin_enc ../clips/venuscubes-ppm/AnimSpace00%03d.ppm 5000 4 4
+./tarkin_dec
+
+------------------------------------------------------------------------------
+
+Hi,
+
+this is a experimental 3d-integer-wavelet-video compression codec. Since the
+integer wavelet transform is reversible and a reversible rgb-yuv conversion
+is used (you can understand it as (1,2) integer wavelet transform, too), this
+codec should be lossless if you transmit the whole bitstream.
+The Y/U/V-bitstreams are embedded, thus you can simply get lossy compression
+and shape the used bandwith by cutting bitstreams, when a user defined limit
+is reached.
+
+
+Here is how the current code works:
+
+First we grab a block of N_FRAMES frames (defined in main.c) of .ppm files.
+Then each pixel becomes transformed into a YUV-alike colorspace. Take a look in
+yuv.c to see how it is done. Each component is then transformed into frequency
+space by applying the wavelet transform in x, y and frame direction.
+The frame-direction transform is our high-order 'motion compensation'.
+At boundaries we use (1,1)-Wavelets (== HAAR transform), inside the image
+(2,2)-Wavelets. (4,4)-Wavelets should be easy to add. See wavelet.c for details.
+
+The resulting coefficients are scanned bitplane by bitplane and
+runlength-encoded. Runlengths are Huffman-compressed and written into the
+bitstreams. The bitplanes of higher-frequency scales are offset'ed to ensure a
+fast transmission of high-energy-low-frequency coefficients. (coder.c)
+The huffman coder is quite simple and uses a hardcoded table, this can be done
+much better, but I wanted to get it working fast.
+
+Decompression works exactly like compression but in reversed direction.
+
+The test program writes for each frame the grabbed original image, the y/u/v
+component (may look strange, since u/v can be negative and are not clamped to
+the [0:255] range), the coefficients (look much more like usual wavelet
+coefficients if you add 128 to each pixel), the coefficients after they are
+runlength/huffman encoded and decoded, the y/u/v components when inverse wavelet
+transform is done and the output image in .ppm format.
+
+You can call the test program like this:
+
+ $ ./main 20000 5000 5000 ../clips/%i.ppm
+
+which means: images are grabbed from directory ../clips/0.ppm, ../clips/1.ppm,
+etc. The Y component bitstream is limited to 20000 Bytes, the U and V bitstreams
+to 5000 Bytes. If the last argument is omitted, frames are taken from current
+directory.
+
+Good Luck,
+
+- Holger <hwaechtler@users.sourceforge.net>
+
diff --git a/ext/tarkin/golomb.h b/ext/tarkin/golomb.h
new file mode 100644
index 00000000..fa66d750
--- /dev/null
+++ b/ext/tarkin/golomb.h
@@ -0,0 +1,132 @@
+#ifndef __GOLOMB_H
+#define __GOLOMB_H
+
+
+#include "bitcoder.h"
+
+
+static inline
+unsigned int required_bits (unsigned int x)
+{
+ int bits = 31;
+
+ while ((x & (1 << bits)) == 0 && bits)
+ bits--;
+
+ return bits;
+}
+
+
+static inline
+void write_number_binary (BitCoderState *b, unsigned int x, int bits, int u)
+{
+//printf ("wrote %i with %i bits (%i+%i)\n", x, u+bits, u, bits);
+ while (bits) {
+ bits--;
+ bitcoder_write_bit (b, (x >> bits) & 1);
+ }
+}
+
+
+static inline
+unsigned int read_number_binary (BitCoderState *b, int bits)
+{
+ unsigned int x = 0;
+
+ while (bits) {
+ bits--;
+ x |= bitcoder_read_bit (b) << bits;
+ }
+
+ return x;
+}
+
+
+static inline
+void golomb_write_number (BitCoderState *b, unsigned int x, int bits)
+{
+ unsigned int q, r;
+int i = 0;
+
+ assert (x > 0);
+
+ while ((q = (x - 1) >> bits) > 0) {
+ bitcoder_write_bit (b, 1); /* fast temporary adaption, write */
+ bits++; /* unary representation of q */
+i++;
+ };
+
+ bitcoder_write_bit (b, 0);
+
+ r = x - 1 - (q << bits);
+
+ write_number_binary (b, r, bits, i+1);
+}
+
+
+static inline
+unsigned int golomb_read_number (BitCoderState *b, int bits)
+{
+ unsigned int q = 0, r, x;
+
+ while (bitcoder_read_bit (b) != 0) {
+ bits++;
+ }
+
+ r = read_number_binary (b, bits);
+ x = (q << bits) + r + 1;
+
+ return x;
+}
+
+
+typedef struct {
+ uint8_t count;
+ uint8_t bits; /* a 5.3 fixed point integer */
+} GolombAdaptiveCoderState;
+
+#define GOLOMB_ADAPTIVE_CODER_STATE_INITIALIZER { 8<<3, 0 }
+
+
+static const int golomb_w_tab [] = { 256, 128, 64 };
+
+
+
+
+static inline
+void golombcoder_encode_number (GolombAdaptiveCoderState *g,
+ BitCoderState *b,
+ unsigned int x)
+{
+ golomb_write_number (b, x, g->bits >> 3);
+
+ g->bits = ((256 - golomb_w_tab[g->count]) * (int) g->bits +
+ golomb_w_tab[g->count] * (required_bits(x)<<3)) / 256;
+ g->count++;
+
+ if (g->count > 2)
+ g->count = 2;
+}
+
+
+static inline
+unsigned int golombcoder_decode_number (GolombAdaptiveCoderState *g,
+ BitCoderState *b)
+{
+ unsigned int x;
+
+ x = golomb_read_number (b, g->bits >> 3);
+
+ g->bits = ((256 - golomb_w_tab[g->count]) * g->bits +
+ golomb_w_tab[g->count] * (required_bits(x)<<3)) / 256;
+ g->count++;
+
+ if (g->count > 2)
+ g->count = 2;
+
+ return x;
+}
+
+
+#endif
+
diff --git a/ext/tarkin/gsttarkin.c b/ext/tarkin/gsttarkin.c
new file mode 100644
index 00000000..7bfb78f3
--- /dev/null
+++ b/ext/tarkin/gsttarkin.c
@@ -0,0 +1,156 @@
+/* Gnome-Streamer
+ * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
+ *
+ * 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 "gsttarkinenc.h"
+#include "gsttarkindec.h"
+
+extern GstElementDetails tarkinenc_details;
+extern GstElementDetails tarkindec_details;
+
+static GstCaps* tarkin_typefind (GstBuffer *buf, gpointer private);
+
+GstPadTemplate *enc_src_template, *enc_sink_template;
+GstPadTemplate *dec_src_template, *dec_sink_template;
+
+static GstCaps*
+tarkin_caps_factory (void)
+{
+ return
+ gst_caps_new (
+ "tarkin_tarkin",
+ "video/x-ogg",
+ NULL);
+}
+
+static GstCaps*
+raw_caps_factory (void)
+{
+ return
+ GST_CAPS_NEW (
+ "tarkin_raw",
+ "video/raw",
+ "format", GST_PROPS_FOURCC (GST_STR_FOURCC ("RGB ")),
+ "bpp", GST_PROPS_INT (24),
+ "depth", GST_PROPS_INT (24),
+ "endianness", GST_PROPS_INT (G_BYTE_ORDER),
+ "red_mask", GST_PROPS_INT (0xff0000),
+ "green_mask", GST_PROPS_INT (0xff00),
+ "blue_mask", GST_PROPS_INT (0xff),
+ "width", GST_PROPS_INT_RANGE (0, G_MAXINT),
+ "height", GST_PROPS_INT_RANGE (0, G_MAXINT)
+ );
+}
+
+static GstTypeDefinition tarkindefinition =
+{
+ "tarkin_video/x-ogg",
+ "video/x-ogg",
+ ".ogg",
+ tarkin_typefind,
+};
+
+static GstCaps*
+tarkin_typefind (GstBuffer *buf, gpointer private)
+{
+ gulong head = GULONG_FROM_BE (*((gulong *)GST_BUFFER_DATA (buf)));
+
+ /* FIXME */
+ return NULL;
+
+ if (head != 0x4F676753)
+ return NULL;
+
+ return gst_caps_new ("tarkin_typefind", "video/x-ogg", NULL);
+}
+
+
+static gboolean
+plugin_init (GModule *module, GstPlugin *plugin)
+{
+ GstElementFactory *enc, *dec;
+ GstTypeFactory *type;
+ GstCaps *raw_caps, *tarkin_caps;
+
+ gst_plugin_set_longname (plugin, "The OGG Vorbis Codec");
+
+ /* create an elementfactory for the tarkinenc element */
+ enc = gst_elementfactory_new ("tarkinenc", GST_TYPE_TARKINENC,
+ &tarkinenc_details);
+ g_return_val_if_fail (enc != NULL, FALSE);
+
+ raw_caps = raw_caps_factory ();
+ tarkin_caps = tarkin_caps_factory ();
+
+ /* register sink pads */
+ enc_sink_template = gst_padtemplate_new ("sink",
+ GST_PAD_SINK,
+ GST_PAD_ALWAYS,
+ raw_caps,
+ NULL);
+ gst_elementfactory_add_padtemplate (enc, enc_sink_template);
+
+ /* register src pads */
+ enc_src_template = gst_padtemplate_new ("src",
+ GST_PAD_SRC,
+ GST_PAD_ALWAYS,
+ tarkin_caps,
+ NULL);
+ gst_elementfactory_add_padtemplate (enc, enc_src_template);
+
+ gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (enc));
+
+ /* create an elementfactory for the tarkindec element */
+ dec = gst_elementfactory_new ("tarkindec", GST_TYPE_TARKINDEC,
+ &tarkindec_details);
+ g_return_val_if_fail (dec != NULL, FALSE);
+
+ raw_caps = raw_caps_factory ();
+ tarkin_caps = tarkin_caps_factory ();
+
+ /* register sink pads */
+ dec_sink_template = gst_padtemplate_new ("sink",
+ GST_PAD_SINK,
+ GST_PAD_ALWAYS,
+ tarkin_caps,
+ NULL);
+ gst_elementfactory_add_padtemplate (dec, dec_sink_template);
+
+ /* register src pads */
+ dec_src_template = gst_padtemplate_new ("src",
+ GST_PAD_SRC,
+ GST_PAD_ALWAYS,
+ raw_caps,
+ NULL);
+ gst_elementfactory_add_padtemplate (dec, dec_src_template);
+
+ gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (dec));
+
+ type = gst_typefactory_new (&tarkindefinition);
+ gst_plugin_add_feature (plugin, GST_PLUGIN_FEATURE (type));
+
+ return TRUE;
+}
+
+GstPluginDesc plugin_desc = {
+ GST_VERSION_MAJOR,
+ GST_VERSION_MINOR,
+ "tarkin",
+ plugin_init
+};
diff --git a/ext/tarkin/gsttarkindec.c b/ext/tarkin/gsttarkindec.c
new file mode 100644
index 00000000..39c58d76
--- /dev/null
+++ b/ext/tarkin/gsttarkindec.c
@@ -0,0 +1,301 @@
+/* Gnome-Streamer
+ * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
+ *
+ * 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 <stdlib.h>
+#include <string.h>
+
+#include "gsttarkindec.h"
+
+extern GstPadTemplate *dec_src_template, *dec_sink_template;
+
+/* elementfactory information */
+GstElementDetails tarkindec_details = {
+ "Ogg Tarkin decoder",
+ "Filter/Video/Decoder",
+ "Decodes video in OGG Tarkin format",
+ VERSION,
+ "Monty <monty@xiph.org>, "
+ "Wim Taymans <wim.taymans@chello.be>",
+ "(C) 2002",
+};
+
+/* TarkinDec signals and args */
+enum
+{
+ /* FILL ME */
+ LAST_SIGNAL
+};
+
+enum
+{
+ ARG_0,
+ ARG_BITRATE,
+};
+
+static void gst_tarkindec_class_init (TarkinDecClass *klass);
+static void gst_tarkindec_init (TarkinDec *arkindec);
+
+static void gst_tarkindec_chain (GstPad *pad, GstBuffer *buf);
+static void gst_tarkindec_setup (TarkinDec *tarkindec);
+static GstElementStateReturn
+ gst_tarkindec_change_state (GstElement *element);
+
+static void gst_tarkindec_get_property (GObject *object, guint prop_id, GValue *value,
+ GParamSpec *pspec);
+static void gst_tarkindec_set_property (GObject *object, guint prop_id, const GValue *value,
+ GParamSpec *pspec);
+
+static GstElementClass *parent_class = NULL;
+/*static guint gst_tarkindec_signals[LAST_SIGNAL] = { 0 }; */
+
+GType
+tarkindec_get_type (void)
+{
+ static GType tarkindec_type = 0;
+
+ if (!tarkindec_type) {
+ static const GTypeInfo tarkindec_info = {
+ sizeof (TarkinDecClass),
+ NULL,
+ NULL,
+ (GClassInitFunc) gst_tarkindec_class_init,
+ NULL,
+ NULL,
+ sizeof (TarkinDec),
+ 0,
+ (GInstanceInitFunc) gst_tarkindec_init,
+ };
+
+ tarkindec_type = g_type_register_static (GST_TYPE_ELEMENT, "TarkinDec", &tarkindec_info, 0);
+ }
+ return tarkindec_type;
+}
+
+static void
+gst_tarkindec_class_init (TarkinDecClass *klass)
+{
+ GObjectClass *gobject_class;
+ GstElementClass *gstelement_class;
+
+ gobject_class = (GObjectClass *) klass;
+ gstelement_class = (GstElementClass *) klass;
+
+ g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_BITRATE,
+ g_param_spec_int ("bitrate", "bitrate", "bitrate",
+ G_MININT, G_MAXINT, 3000, G_PARAM_READWRITE));
+
+ parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
+
+ gobject_class->set_property = gst_tarkindec_set_property;
+ gobject_class->get_property = gst_tarkindec_get_property;
+
+ gstelement_class->change_state = gst_tarkindec_change_state;
+}
+
+static void
+gst_tarkindec_init (TarkinDec * tarkindec)
+{
+ tarkindec->sinkpad = gst_pad_new_from_template (dec_sink_template, "sink");
+ gst_element_add_pad (GST_ELEMENT (tarkindec), tarkindec->sinkpad);
+ gst_pad_set_chain_function (tarkindec->sinkpad, gst_tarkindec_chain);
+
+ tarkindec->srcpad = gst_pad_new_from_template (dec_src_template, "src");
+ gst_element_add_pad (GST_ELEMENT (tarkindec), tarkindec->srcpad);
+
+ tarkindec->bitrate = 3000;
+ tarkindec->setup = FALSE;
+ tarkindec->nheader = 0;
+
+ /* we're chained and we can deal with events */
+ GST_FLAG_SET (tarkindec, GST_ELEMENT_EVENT_AWARE);
+}
+
+static void
+gst_tarkindec_setup (TarkinDec *tarkindec)
+{
+ tarkindec->tarkin_stream = tarkin_stream_new ();
+
+ ogg_sync_init (&tarkindec->oy);
+ ogg_stream_init (&tarkindec->os, 1);
+ tarkin_info_init (&tarkindec->ti);
+ tarkin_comment_init (&tarkindec->tc);
+
+ tarkindec->setup = TRUE;
+}
+
+static void
+gst_tarkindec_chain (GstPad *pad, GstBuffer *buf)
+{
+ TarkinDec *tarkindec;
+
+ g_return_if_fail (pad != NULL);
+ g_return_if_fail (GST_IS_PAD (pad));
+ g_return_if_fail (buf != NULL);
+
+ tarkindec = GST_TARKINDEC (gst_pad_get_parent (pad));
+
+ if (!tarkindec->setup) {
+ gst_element_error (GST_ELEMENT (tarkindec), "decoder not initialized (input is not audio?)");
+ if (GST_IS_BUFFER (buf))
+ gst_buffer_unref (buf);
+ else
+ gst_pad_event_default (pad, GST_EVENT (buf));
+ return;
+ }
+
+ if (GST_IS_EVENT (buf)) {
+ switch (GST_EVENT_TYPE (buf)) {
+ case GST_EVENT_EOS:
+ default:
+ gst_pad_event_default (pad, GST_EVENT (buf));
+ break;
+ }
+ }
+ else {
+ gchar *data;
+ gulong size;
+ gchar *buffer;
+ guchar *rgb;
+ TarkinTime date;
+ TarkinVideoLayerDesc *layer;
+
+ /* data to decode */
+ data = GST_BUFFER_DATA (buf);
+ size = GST_BUFFER_SIZE (buf);
+
+ buffer = ogg_sync_buffer(&tarkindec->oy, size);
+ memcpy (buffer, data, size);
+ ogg_sync_wrote(&tarkindec->oy, size);
+
+ if (ogg_sync_pageout (&tarkindec->oy, &tarkindec->og)) {
+ ogg_stream_pagein (&tarkindec->os, &tarkindec->og);
+
+ while (ogg_stream_packetout (&tarkindec->os, &tarkindec->op)) {
+ if (tarkindec->op.e_o_s)
+ break;
+ if (tarkindec->nheader < 3) { /* 3 first packets to headerin */
+ tarkin_synthesis_headerin (&tarkindec->ti, &tarkindec->tc, &tarkindec->op);
+
+ if (tarkindec->nheader == 2) {
+ tarkin_synthesis_init (tarkindec->tarkin_stream, &tarkindec->ti);
+ }
+ tarkindec->nheader++;
+ } else {
+ tarkin_synthesis_packetin (tarkindec->tarkin_stream, &tarkindec->op);
+
+ while (tarkin_synthesis_frameout (tarkindec->tarkin_stream, &rgb, 0, &date) == 0) {
+ GstBuffer *outbuf;
+
+ layer = &tarkindec->tarkin_stream->layer->desc;
+
+ if (!GST_PAD_CAPS (tarkindec->srcpad)) {
+ if (!gst_pad_try_set_caps (tarkindec->srcpad,
+ GST_CAPS_NEW (
+ "tarkin_raw",
+ "video/raw",
+ "format", GST_PROPS_FOURCC (GST_STR_FOURCC ("RGB ")),
+ "bpp", GST_PROPS_INT (24),
+ "depth", GST_PROPS_INT (24),
+ "endianness", GST_PROPS_INT (G_BYTE_ORDER),
+ "red_mask", GST_PROPS_INT (0xff0000),
+ "green_mask", GST_PROPS_INT (0xff00),
+ "blue_mask", GST_PROPS_INT (0xff),
+ "width", GST_PROPS_INT (layer->width),
+ "height", GST_PROPS_INT (layer->height)
+ )))
+ {
+ gst_element_error (GST_ELEMENT (tarkindec), "could not output format");
+ gst_buffer_unref (buf);
+ return;
+ }
+ }
+ outbuf = gst_buffer_new ();
+ GST_BUFFER_DATA (outbuf) = rgb;
+ GST_BUFFER_SIZE (outbuf) = layer->width * layer->height * 3;
+ GST_BUFFER_FLAG_SET (outbuf, GST_BUFFER_DONTFREE);
+ gst_pad_push (tarkindec->srcpad, outbuf);
+
+ tarkin_synthesis_freeframe (tarkindec->tarkin_stream, rgb);
+ }
+ }
+ }
+ }
+ gst_buffer_unref (buf);
+ }
+}
+
+static GstElementStateReturn
+gst_tarkindec_change_state (GstElement *element)
+{
+ TarkinDec *tarkindec;
+
+ tarkindec = GST_TARKINDEC (element);
+
+ switch (GST_STATE_TRANSITION (element)) {
+ case GST_STATE_READY_TO_PAUSED:
+ gst_tarkindec_setup (tarkindec);
+ break;
+ case GST_STATE_PAUSED_TO_READY:
+ break;
+ default:
+ break;
+ }
+
+ return parent_class->change_state (element);
+}
+
+static void
+gst_tarkindec_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
+{
+ TarkinDec *tarkindec;
+
+ /* it's not null if we got it, but it might not be ours */
+ g_return_if_fail (GST_IS_TARKINDEC (object));
+
+ tarkindec = GST_TARKINDEC (object);
+
+ switch (prop_id) {
+ case ARG_BITRATE:
+ g_value_set_int (value, tarkindec->bitrate);
+ break;
+ default:
+ break;
+ }
+}
+
+static void
+gst_tarkindec_set_property (GObject *object, guint prop_id, const GValue *value,
+ GParamSpec *pspec)
+{
+ TarkinDec *tarkindec;
+
+ /* it's not null if we got it, but it might not be ours */
+ g_return_if_fail (GST_IS_TARKINDEC (object));
+
+ tarkindec = GST_TARKINDEC (object);
+
+ switch (prop_id) {
+ case ARG_BITRATE:
+ tarkindec->bitrate = g_value_get_int (value);
+ break;
+ default:
+ break;
+ }
+}
diff --git a/ext/tarkin/gsttarkindec.h b/ext/tarkin/gsttarkindec.h
new file mode 100644
index 00000000..78629851
--- /dev/null
+++ b/ext/tarkin/gsttarkindec.h
@@ -0,0 +1,83 @@
+/* Gnome-Streamer
+ * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
+ *
+ * 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.
+ */
+
+
+#ifndef __TARKINDEC_H__
+#define __TARKINDEC_H__
+
+
+#include <config.h>
+#include <gst/gst.h>
+
+#include "tarkin.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif /* __cplusplus */
+
+#define GST_TYPE_TARKINDEC \
+ (tarkindec_get_type())
+#define GST_TARKINDEC(obj) \
+ (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_TARKINDEC,TarkinDec))
+#define GST_TARKINDEC_CLASS(klass) \
+ (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_TARKINDEC,TarkinDecClass))
+#define GST_IS_TARKINDEC(obj) \
+ (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_TARKINDEC))
+#define GST_IS_TARKINDEC_CLASS(obj) \
+ (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_TARKINDEC))
+
+typedef struct _TarkinDec TarkinDec;
+typedef struct _TarkinDecClass TarkinDecClass;
+
+struct _TarkinDec {
+ GstElement element;
+
+ GstPad *sinkpad,*srcpad;
+
+ ogg_sync_state oy;
+ ogg_stream_state os;
+ ogg_page og;
+ ogg_packet op;
+
+ TarkinStream *tarkin_stream;
+ TarkinComment tc;
+ TarkinInfo ti;
+ TarkinVideoLayerDesc layer[1];
+
+ gint frame_num;
+ gint nheader;
+
+ gboolean eos;
+ gint bitrate;
+ gboolean setup;
+};
+
+struct _TarkinDecClass {
+ GstElementClass parent_class;
+};
+
+GType tarkindec_get_type(void);
+
+
+#ifdef __cplusplus
+}
+#endif /* __cplusplus */
+
+
+#endif /* __TARKINDEC_H__ */
diff --git a/ext/tarkin/gsttarkinenc.c b/ext/tarkin/gsttarkinenc.c
new file mode 100644
index 00000000..8042430a
--- /dev/null
+++ b/ext/tarkin/gsttarkinenc.c
@@ -0,0 +1,371 @@
+/* Gnome-Streamer
+ * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
+ *
+ * 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 <stdlib.h>
+#include <string.h>
+
+#include "gsttarkinenc.h"
+
+extern GstPadTemplate *enc_src_template, *enc_sink_template;
+
+/* elementfactory information */
+GstElementDetails tarkinenc_details = {
+ "Ogg Tarkin encoder",
+ "Filter/Video/Encoder",
+ "Encodes video in OGG Tarkin format",
+ VERSION,
+ "Monty <monty@xiph.org>, "
+ "Wim Taymans <wim.taymans@chello.be>",
+ "(C) 2002",
+};
+
+/* TarkinEnc signals and args */
+enum
+{
+ /* FILL ME */
+ LAST_SIGNAL
+};
+
+enum
+{
+ ARG_0,
+ ARG_BITRATE,
+ ARG_S_MOMENTS,
+ ARG_A_MOMENTS,
+};
+
+static void gst_tarkinenc_class_init (TarkinEncClass *klass);
+static void gst_tarkinenc_init (TarkinEnc *arkinenc);
+
+static void gst_tarkinenc_chain (GstPad *pad, GstBuffer *buf);
+static void gst_tarkinenc_setup (TarkinEnc *tarkinenc);
+
+static void gst_tarkinenc_get_property (GObject *object, guint prop_id, GValue *value,
+ GParamSpec *pspec);
+static void gst_tarkinenc_set_property (GObject *object, guint prop_id, const GValue *value,
+ GParamSpec *pspec);
+
+static GstElementClass *parent_class = NULL;
+/*static guint gst_tarkinenc_signals[LAST_SIGNAL] = { 0 }; */
+
+GType
+tarkinenc_get_type (void)
+{
+ static GType tarkinenc_type = 0;
+
+ if (!tarkinenc_type) {
+ static const GTypeInfo tarkinenc_info = {
+ sizeof (TarkinEncClass),
+ NULL,
+ NULL,
+ (GClassInitFunc) gst_tarkinenc_class_init,
+ NULL,
+ NULL,
+ sizeof (TarkinEnc),
+ 0,
+ (GInstanceInitFunc) gst_tarkinenc_init,
+ };
+
+ tarkinenc_type = g_type_register_static (GST_TYPE_ELEMENT, "TarkinEnc", &tarkinenc_info, 0);
+ }
+ return tarkinenc_type;
+}
+
+static void
+gst_tarkinenc_class_init (TarkinEncClass *klass)
+{
+ GObjectClass *gobject_class;
+ GstElementClass *gstelement_class;
+
+ gobject_class = (GObjectClass *) klass;
+ gstelement_class = (GstElementClass *) klass;
+
+ g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_BITRATE,
+ g_param_spec_int ("bitrate", "bitrate", "bitrate",
+ G_MININT, G_MAXINT, 3000, G_PARAM_READWRITE));
+ g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_S_MOMENTS,
+ g_param_spec_int ("s_moments", "Synthesis Moments",
+ "Number of vanishing moments for the synthesis filter",
+ 1, 4, 2, G_PARAM_READWRITE));
+ g_object_class_install_property (G_OBJECT_CLASS (klass), ARG_A_MOMENTS,
+ g_param_spec_int ("a_moments", "Analysis Moments",
+ "Number of vanishing moments for the analysis filter",
+ 1, 4, 2, G_PARAM_READWRITE));
+
+ parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
+
+ gobject_class->set_property = gst_tarkinenc_set_property;
+ gobject_class->get_property = gst_tarkinenc_get_property;
+}
+
+static GstPadConnectReturn
+gst_tarkinenc_sinkconnect (GstPad *pad, GstCaps *caps)
+{
+ TarkinEnc *tarkinenc;
+
+ tarkinenc = GST_TARKINENC (gst_pad_get_parent (pad));
+
+ if (!GST_CAPS_IS_FIXED (caps))
+ return GST_PAD_CONNECT_DELAYED;
+
+ gst_caps_debug (caps);
+
+ tarkinenc->layer[0].bitstream_len = tarkinenc->bitrate;
+ tarkinenc->layer[0].a_moments = tarkinenc->a_moments;
+ tarkinenc->layer[0].s_moments = tarkinenc->s_moments;
+ tarkinenc->layer[0].width = gst_caps_get_int (caps, "width");
+ tarkinenc->layer[0].height = gst_caps_get_int (caps, "height");
+ tarkinenc->layer[0].format = TARKIN_RGB24;
+ tarkinenc->layer[0].frames_per_buf = TARKIN_RGB24;
+
+ gst_tarkinenc_setup (tarkinenc);
+
+ if (tarkinenc->setup)
+ return GST_PAD_CONNECT_OK;
+
+ return GST_PAD_CONNECT_REFUSED;
+}
+
+static void
+gst_tarkinenc_init (TarkinEnc * tarkinenc)
+{
+ tarkinenc->sinkpad = gst_pad_new_from_template (enc_sink_template, "sink");
+ gst_element_add_pad (GST_ELEMENT (tarkinenc), tarkinenc->sinkpad);
+ gst_pad_set_chain_function (tarkinenc->sinkpad, gst_tarkinenc_chain);
+ gst_pad_set_connect_function (tarkinenc->sinkpad, gst_tarkinenc_sinkconnect);
+
+ tarkinenc->srcpad = gst_pad_new_from_template (enc_src_template, "src");
+ gst_element_add_pad (GST_ELEMENT (tarkinenc), tarkinenc->srcpad);
+
+ tarkinenc->bitrate = 3000;
+ tarkinenc->s_moments = 2;
+ tarkinenc->a_moments = 2;
+ tarkinenc->setup = FALSE;
+}
+
+TarkinError free_frame (void *s, void *ptr)
+{
+ return(TARKIN_OK);
+}
+
+TarkinError packet_out (void *stream, ogg_packet *op)
+{
+ ogg_page og;
+ TarkinStream *s = stream;
+ TarkinEnc *te = s->user_ptr;
+ GstBuffer *outbuf;
+
+ ogg_stream_packetin (&te->os, op);
+
+ if(op->e_o_s){
+ ogg_stream_flush (&te->os, &og);
+ outbuf = gst_buffer_new ();
+ GST_BUFFER_DATA (outbuf) = og.header;
+ GST_BUFFER_SIZE (outbuf) = og.header_len;
+ GST_BUFFER_FLAG_SET (outbuf, GST_BUFFER_DONTFREE);
+ gst_pad_push (te->srcpad, outbuf);
+ outbuf = gst_buffer_new ();
+ GST_BUFFER_DATA (outbuf) = og.body;
+ GST_BUFFER_SIZE (outbuf) = og.body_len;
+ GST_BUFFER_FLAG_SET (outbuf, GST_BUFFER_DONTFREE);
+ gst_pad_push (te->srcpad, outbuf);
+ } else {
+ while (ogg_stream_pageout (&te->os, &og)){
+ outbuf = gst_buffer_new ();
+ GST_BUFFER_DATA (outbuf) = og.header;
+ GST_BUFFER_SIZE (outbuf) = og.header_len;
+ GST_BUFFER_FLAG_SET (outbuf, GST_BUFFER_DONTFREE);
+ gst_pad_push (te->srcpad, outbuf);
+ outbuf = gst_buffer_new ();
+ GST_BUFFER_DATA (outbuf) = og.body;
+ GST_BUFFER_SIZE (outbuf) = og.body_len;
+ GST_BUFFER_FLAG_SET (outbuf, GST_BUFFER_DONTFREE);
+ gst_pad_push (te->srcpad, outbuf);
+ }
+ }
+ return (TARKIN_OK);
+}
+
+
+static void
+gst_tarkinenc_setup (TarkinEnc *tarkinenc)
+{
+ gint i;
+ GstBuffer *outbuf;
+
+ ogg_stream_init (&tarkinenc->os, 1);
+ tarkin_info_init (&tarkinenc->ti);
+
+ tarkinenc->ti.inter.numerator = 1;
+ tarkinenc->ti.inter.denominator = 1;
+
+ tarkin_comment_init (&tarkinenc->tc);
+ tarkin_comment_add_tag (&tarkinenc->tc, "TITLE", "GStreamer produced file");
+ tarkin_comment_add_tag (&tarkinenc->tc, "ARTIST", "C coders ;)");
+
+ tarkinenc->tarkin_stream = tarkin_stream_new ();
+ tarkin_analysis_init (tarkinenc->tarkin_stream,
+ &tarkinenc->ti, free_frame, packet_out, (void*)tarkinenc);
+ tarkin_analysis_add_layer(tarkinenc->tarkin_stream, &tarkinenc->layer[0]);
+
+ tarkin_analysis_headerout (tarkinenc->tarkin_stream, &tarkinenc->tc,
+ tarkinenc->op, &tarkinenc->op[1], &tarkinenc->op[2]);
+ for(i = 0; i < 3; i++){
+ ogg_stream_packetin(&tarkinenc->os, &tarkinenc->op[i]);
+ }
+
+ ogg_stream_flush (&tarkinenc->os, &tarkinenc->og);
+
+ tarkinenc->frame_num = 0;
+
+ outbuf = gst_buffer_new ();
+ GST_BUFFER_DATA (outbuf) = tarkinenc->og.header;
+ GST_BUFFER_SIZE (outbuf) = tarkinenc->og.header_len;
+ GST_BUFFER_FLAG_SET (outbuf, GST_BUFFER_DONTFREE);
+ gst_pad_push (tarkinenc->srcpad, outbuf);
+
+ outbuf = gst_buffer_new ();
+ GST_BUFFER_DATA (outbuf) = tarkinenc->og.body;
+ GST_BUFFER_SIZE (outbuf) = tarkinenc->og.body_len;
+ GST_BUFFER_FLAG_SET (outbuf, GST_BUFFER_DONTFREE);
+ gst_pad_push (tarkinenc->srcpad, outbuf);
+
+ tarkinenc->setup = TRUE;
+}
+
+static void
+gst_tarkinenc_chain (GstPad *pad, GstBuffer *buf)
+{
+ TarkinEnc *tarkinenc;
+
+ g_return_if_fail (pad != NULL);
+ g_return_if_fail (GST_IS_PAD (pad));
+ g_return_if_fail (buf != NULL);
+
+ tarkinenc = GST_TARKINENC (gst_pad_get_parent (pad));
+
+ if (!tarkinenc->setup) {
+ gst_element_error (GST_ELEMENT (tarkinenc), "encoder not initialized (input is not audio?)");
+ if (GST_IS_BUFFER (buf))
+ gst_buffer_unref (buf);
+ else
+ gst_pad_event_default (pad, GST_EVENT (buf));
+ return;
+ }
+
+ if (GST_IS_EVENT (buf)) {
+ switch (GST_EVENT_TYPE (buf)) {
+ case GST_EVENT_EOS:
+ tarkin_analysis_framein (tarkinenc->tarkin_stream, NULL, 0, NULL); /* EOS */
+ tarkin_comment_clear (&tarkinenc->tc);
+ tarkin_stream_destroy (tarkinenc->tarkin_stream);
+ default:
+ gst_pad_event_default (pad, GST_EVENT (buf));
+ break;
+ }
+ }
+ else {
+ gchar *data;
+ gulong size;
+ TarkinTime date;
+
+ /* data to encode */
+ data = GST_BUFFER_DATA (buf);
+ size = GST_BUFFER_SIZE (buf);
+
+ date.numerator = tarkinenc->frame_num;
+ date.denominator = 1;
+ tarkin_analysis_framein (tarkinenc->tarkin_stream, data, 0, &date);
+ tarkinenc->frame_num++;
+
+ gst_buffer_unref (buf);
+ }
+}
+
+static void
+gst_tarkinenc_get_property (GObject *object, guint prop_id, GValue *value, GParamSpec *pspec)
+{
+ TarkinEnc *tarkinenc;
+
+ /* it's not null if we got it, but it might not be ours */
+ g_return_if_fail (GST_IS_TARKINENC (object));
+
+ tarkinenc = GST_TARKINENC (object);
+
+ switch (prop_id) {
+ case ARG_BITRATE:
+ g_value_set_int (value, tarkinenc->bitrate);
+ break;
+ case ARG_S_MOMENTS:
+ g_value_set_int (value, tarkinenc->s_moments);
+ break;
+ case ARG_A_MOMENTS:
+ g_value_set_int (value, tarkinenc->a_moments);
+ break;
+ default:
+ break;
+ }
+}
+
+static void
+gst_tarkinenc_set_property (GObject *object, guint prop_id, const GValue *value,
+ GParamSpec *pspec)
+{
+ TarkinEnc *tarkinenc;
+
+ /* it's not null if we got it, but it might not be ours */
+ g_return_if_fail (GST_IS_TARKINENC (object));
+
+ tarkinenc = GST_TARKINENC (object);
+
+ switch (prop_id) {
+ case ARG_BITRATE:
+ tarkinenc->bitrate = g_value_get_int (value);
+ break;
+ case ARG_S_MOMENTS:
+ {
+ gint s_moments;
+
+ s_moments = g_value_get_int (value);
+ if (s_moments != 1 || s_moments != 2 || s_moments != 4) {
+ g_warning ("tarkinenc: s_moments must be 1, 2 or 4");
+ }
+ else {
+ tarkinenc->s_moments = s_moments;
+ }
+ break;
+ }
+ case ARG_A_MOMENTS:
+ {
+ gint a_moments;
+
+ a_moments = g_value_get_int (value);
+ if (a_moments != 1 || a_moments != 2 || a_moments != 4) {
+ g_warning ("tarkinenc: a_moments must be 1, 2 or 4");
+ }
+ else {
+ tarkinenc->a_moments = a_moments;
+ }
+ break;
+ }
+ default:
+ break;
+ }
+}
diff --git a/ext/tarkin/gsttarkinenc.h b/ext/tarkin/gsttarkinenc.h
new file mode 100644
index 00000000..ca74c989
--- /dev/null
+++ b/ext/tarkin/gsttarkinenc.h
@@ -0,0 +1,84 @@
+/* Gnome-Streamer
+ * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
+ *
+ * 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.
+ */
+
+
+#ifndef __TARKINENC_H__
+#define __TARKINENC_H__
+
+
+#include <config.h>
+#include <gst/gst.h>
+
+#include "tarkin.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif /* __cplusplus */
+
+#define GST_TYPE_TARKINENC \
+ (tarkinenc_get_type())
+#define GST_TARKINENC(obj) \
+ (G_TYPE_CHECK_INSTANCE_CAST((obj),GST_TYPE_TARKINENC,TarkinEnc))
+#define GST_TARKINENC_CLASS(klass) \
+ (G_TYPE_CHECK_CLASS_CAST((klass),GST_TYPE_TARKINENC,TarkinEncClass))
+#define GST_IS_TARKINENC(obj) \
+ (G_TYPE_CHECK_INSTANCE_TYPE((obj),GST_TYPE_TARKINENC))
+#define GST_IS_TARKINENC_CLASS(obj) \
+ (G_TYPE_CHECK_CLASS_TYPE((klass),GST_TYPE_TARKINENC))
+
+typedef struct _TarkinEnc TarkinEnc;
+typedef struct _TarkinEncClass TarkinEncClass;
+
+struct _TarkinEnc {
+ GstElement element;
+
+ GstPad *sinkpad,*srcpad;
+
+ ogg_stream_state os; /* take physical pages, weld into a logical
+ stream of packets */
+ ogg_page og; /* one Ogg bitstream page. Tarkin packets are inside */
+ ogg_packet op[3]; /* one raw packet of data for decode */
+
+ TarkinStream *tarkin_stream;
+ TarkinComment tc;
+ TarkinInfo ti;
+ TarkinVideoLayerDesc layer[1];
+
+ gint frame_num;
+
+ gboolean eos;
+ gint bitrate;
+ gint s_moments;
+ gint a_moments;
+ gboolean setup;
+};
+
+struct _TarkinEncClass {
+ GstElementClass parent_class;
+};
+
+GType tarkinenc_get_type(void);
+
+
+#ifdef __cplusplus
+}
+#endif /* __cplusplus */
+
+
+#endif /* __TARKINENC_H__ */
diff --git a/ext/tarkin/info.c b/ext/tarkin/info.c
new file mode 100644
index 00000000..e555e381
--- /dev/null
+++ b/ext/tarkin/info.c
@@ -0,0 +1,551 @@
+/********************************************************************
+ * *
+ * THIS FILE IS PART OF THE OggVorbis SOFTWARE CODEC SOURCE CODE. *
+ * USE, DISTRIBUTION AND REPRODUCTION OF THIS LIBRARY SOURCE IS *
+ * GOVERNED BY A BSD-STYLE SOURCE LICENSE INCLUDED WITH THIS SOURCE *
+ * IN 'COPYING'. PLEASE READ THESE TERMS BEFORE DISTRIBUTING. *
+ * *
+ * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2001 *
+ * by the XIPHOPHORUS Company http://www.xiph.org/ *
+
+ ********************************************************************
+
+ function: maintain the info structure, info <-> header packets
+ last mod: $Id$
+
+ ********************************************************************/
+
+/* general handling of the header and the TarkinInfo structure (and
+ substructures) */
+
+#include <stdlib.h>
+#include <string.h>
+#include <ctype.h>
+#include <ogg/ogg.h>
+#include "tarkin.h"
+#include "yuv.h"
+#include "mem.h"
+
+/* helpers */
+static void _v_writestring(oggpack_buffer *o,char *s, int bytes){
+ while(bytes--){
+ oggpack_write(o,*s++,8);
+ }
+}
+
+static void _v_readstring(oggpack_buffer *o,char *buf,int bytes){
+ while(bytes--){
+ *buf++=oggpack_read(o,8);
+ }
+}
+
+void tarkin_comment_init(TarkinComment *vc){
+ memset(vc,0,sizeof(*vc));
+}
+
+void tarkin_comment_add(TarkinComment *vc,char *comment){
+ vc->user_comments=REALLOC(vc->user_comments,
+ (vc->comments+2)*sizeof(*vc->user_comments));
+ vc->comment_lengths=REALLOC(vc->comment_lengths,
+ (vc->comments+2)*sizeof(*vc->comment_lengths));
+ vc->comment_lengths[vc->comments]=strlen(comment);
+ vc->user_comments[vc->comments]=MALLOC(vc->comment_lengths[vc->comments]+1);
+ strcpy(vc->user_comments[vc->comments], comment);
+ vc->comments++;
+ vc->user_comments[vc->comments]=NULL;
+}
+
+void tarkin_comment_add_tag(TarkinComment *vc, char *tag, char *contents){
+ char *comment=alloca(strlen(tag)+strlen(contents)+2); /* +2 for = and \0 */
+ strcpy(comment, tag);
+ strcat(comment, "=");
+ strcat(comment, contents);
+ tarkin_comment_add(vc, comment);
+}
+
+/* This is more or less the same as strncasecmp - but that doesn't exist
+ * everywhere, and this is a fairly trivial function, so we include it */
+static int tagcompare(const char *s1, const char *s2, int n){
+ int c=0;
+ while(c < n){
+ if(toupper(s1[c]) != toupper(s2[c]))
+ return !0;
+ c++;
+ }
+ return 0;
+}
+
+char *tarkin_comment_query(TarkinComment *vc, char *tag, int count){
+ long i;
+ int found = 0;
+ int taglen = strlen(tag)+1; /* +1 for the = we append */
+ char *fulltag = alloca(taglen+ 1);
+
+ strcpy(fulltag, tag);
+ strcat(fulltag, "=");
+
+ for(i=0;i<vc->comments;i++){
+ if(!tagcompare(vc->user_comments[i], fulltag, taglen)){
+ if(count == found)
+ /* We return a pointer to the data, not a copy */
+ return vc->user_comments[i] + taglen;
+ else
+ found++;
+ }
+ }
+ return NULL; /* didn't find anything */
+}
+
+int tarkin_comment_query_count(TarkinComment *vc, char *tag){
+ int i,count=0;
+ int taglen = strlen(tag)+1; /* +1 for the = we append */
+ char *fulltag = alloca(taglen+1);
+ strcpy(fulltag,tag);
+ strcat(fulltag, "=");
+
+ for(i=0;i<vc->comments;i++){
+ if(!tagcompare(vc->user_comments[i], fulltag, taglen))
+ count++;
+ }
+
+ return count;
+}
+
+void tarkin_comment_clear(TarkinComment *vc){
+ if(vc){
+ long i;
+ for(i=0;i<vc->comments;i++)
+ if(vc->user_comments[i])FREE(vc->user_comments[i]);
+ if(vc->user_comments)FREE(vc->user_comments);
+ if(vc->comment_lengths)FREE(vc->comment_lengths);
+ if(vc->vendor)FREE(vc->vendor);
+ }
+ memset(vc,0,sizeof(*vc));
+}
+
+/* used by synthesis, which has a full, alloced vi */
+void tarkin_info_init(TarkinInfo *vi){
+ memset(vi,0,sizeof(*vi));
+}
+
+void tarkin_info_clear(TarkinInfo *vi){
+ memset(vi,0,sizeof(*vi));
+}
+
+/* Header packing/unpacking ********************************************/
+
+static int _tarkin_unpack_info(TarkinInfo *vi,oggpack_buffer *opb)
+{
+#ifdef DBG_OGG
+ printf("dbg_ogg: Decoding Info: ");
+#endif
+ vi->version=oggpack_read(opb,32);
+ if(vi->version!=0)return(-TARKIN_VERSION);
+
+ vi->n_layers=oggpack_read(opb,8);
+ vi->inter.numerator=oggpack_read(opb,32);
+ vi->inter.denominator=oggpack_read(opb,32);
+
+ vi->bitrate_upper=oggpack_read(opb,32);
+ vi->bitrate_nominal=oggpack_read(opb,32);
+ vi->bitrate_lower=oggpack_read(opb,32);
+
+#ifdef DBG_OGG
+ printf(" n_layers %d, interleave: %d/%d, ",
+ vi->n_layers, vi->inter.numerator, vi->inter.denominator);
+#endif
+
+ if(vi->inter.numerator<1)goto err_out;
+ if(vi->inter.denominator<1)goto err_out;
+ if(vi->n_layers<1)goto err_out;
+
+ if(oggpack_read(opb,1)!=1)goto err_out; /* EOP check */
+
+#ifdef DBG_OGG
+ printf("Success\n");
+#endif
+ return(0);
+ err_out:
+#ifdef DBG_OGG
+ printf("Failed\n");
+#endif
+ tarkin_info_clear(vi);
+ return(-TARKIN_BAD_HEADER);
+}
+
+static int _tarkin_unpack_comment(TarkinComment *vc,oggpack_buffer *opb)
+{
+ int i;
+ int vendorlen=oggpack_read(opb,32);
+
+#ifdef DBG_OGG
+ printf("dbg_ogg: Decoding comment: ");
+#endif
+ if(vendorlen<0)goto err_out;
+ vc->vendor=_ogg_calloc(vendorlen+1,1);
+ _v_readstring(opb,vc->vendor,vendorlen);
+ vc->comments=oggpack_read(opb,32);
+ if(vc->comments<0)goto err_out;
+ vc->user_comments=_ogg_calloc(vc->comments+1,sizeof(*vc->user_comments));
+ vc->comment_lengths=_ogg_calloc(vc->comments+1,
+ sizeof(*vc->comment_lengths));
+
+ for(i=0;i<vc->comments;i++){
+ int len=oggpack_read(opb,32);
+ if(len<0)goto err_out;
+ vc->comment_lengths[i]=len;
+ vc->user_comments[i]=_ogg_calloc(len+1,1);
+ _v_readstring(opb,vc->user_comments[i],len);
+ }
+ if(oggpack_read(opb,1)!=1)goto err_out; /* EOP check */
+
+#ifdef DBG_OGG
+ printf("Success, read %d comments\n", vc->comments);
+#endif
+ return(0);
+ err_out:
+#ifdef DBG_OGG
+ printf("Failed\n");
+#endif
+ tarkin_comment_clear(vc);
+ return(-TARKIN_BAD_HEADER);
+}
+
+/* the real encoding details are here, currently TarkinVideoLayerDesc. */
+static int _tarkin_unpack_layer_desc(TarkinInfo *vi,oggpack_buffer *opb){
+ int i,j;
+ vi->layer = CALLOC (vi->n_layers, (sizeof(*vi->layer)));
+ memset(vi->layer,0, vi->n_layers * sizeof(*vi->layer));
+
+#ifdef DBG_OGG
+ printf("ogg: Decoding layers description: ");
+#endif
+ for(i=0;i<vi->n_layers;i++){
+ TarkinVideoLayer *layer = vi->layer + i;
+ layer->desc.width = oggpack_read(opb,32);
+ layer->desc.height = oggpack_read(opb,32);
+ layer->desc.a_moments = oggpack_read(opb,32);
+ layer->desc.s_moments = oggpack_read(opb,32);
+ layer->desc.frames_per_buf = oggpack_read(opb,32);
+ layer->desc.bitstream_len = oggpack_read(opb,32);
+ layer->desc.format = oggpack_read(opb,32);
+
+ switch (layer->desc.format) {
+ case TARKIN_GRAYSCALE:
+ layer->n_comp = 1;
+ layer->color_fwd_xform = grayscale_to_y;
+ layer->color_inv_xform = y_to_grayscale;
+ break;
+ case TARKIN_RGB24:
+ layer->n_comp = 3;
+ layer->color_fwd_xform = rgb24_to_yuv;
+ layer->color_inv_xform = yuv_to_rgb24;
+ break;
+ case TARKIN_RGB32:
+ layer->n_comp = 3;
+ layer->color_fwd_xform = rgb32_to_yuv;
+ layer->color_inv_xform = yuv_to_rgb32;
+ break;
+ case TARKIN_RGBA:
+ layer->n_comp = 4;
+ layer->color_fwd_xform = rgba_to_yuv;
+ layer->color_inv_xform = yuv_to_rgba;
+ break;
+ default:
+ return -TARKIN_INVALID_COLOR_FORMAT;
+ };
+
+ layer->waveletbuf = (Wavelet3DBuf**) CALLOC (layer->n_comp,
+ sizeof(Wavelet3DBuf*));
+
+ layer->packet = MALLOC (layer->n_comp * sizeof(*layer->packet));
+ memset(layer->packet, 0, layer->n_comp * sizeof(*layer->packet));
+
+ for (j=0; j<layer->n_comp; j++){
+ layer->waveletbuf[j] = wavelet_3d_buf_new (layer->desc.width,
+ layer->desc.height,
+ layer->desc.frames_per_buf);
+ layer->packet[j].data = MALLOC(layer->desc.bitstream_len);
+ layer->packet[j].storage = layer->desc.bitstream_len;
+ }
+
+ vi->max_bitstream_len += layer->desc.bitstream_len
+ + 2 * 10 * sizeof(uint32_t) * layer->n_comp; // truncation tables
+
+#ifdef DBG_OGG
+ printf("\n layer%d: size %dx%dx%d, format %d, a_m %d, s_m %d, %d fpb\n",
+ i, layer->desc.width, layer->desc.height, layer->n_comp,
+ layer->desc.format, layer->desc.a_moments, layer->desc.s_moments,
+ layer->desc.frames_per_buf);
+#endif
+ } /* for each layer */
+
+ if(oggpack_read(opb,1)!=1)goto err_out; /* EOP check */
+
+#ifdef DBG_OGG
+ printf("Success\n");
+#endif
+
+ return(0);
+ err_out:
+#ifdef DBG_OGG
+ printf("Failed\n");
+#endif
+ tarkin_info_clear(vi);
+ return(-TARKIN_BAD_HEADER);
+}
+
+/* The Tarkin header is in three packets; the initial small packet in
+ the first page that identifies basic parameters, a second packet
+ with bitstream comments and a third packet that holds the
+ layer description structures. */
+
+TarkinError tarkin_synthesis_headerin(TarkinInfo *vi,TarkinComment *vc,ogg_packet *op){
+ oggpack_buffer opb;
+
+ if(op){
+ oggpack_readinit(&opb,op->packet,op->bytes);
+
+ /* Which of the three types of header is this? */
+ /* Also verify header-ness, tarkin */
+ {
+ char buffer[6];
+ int packtype=oggpack_read(&opb,8);
+ memset(buffer,0,6);
+ _v_readstring(&opb,buffer,6);
+ if(memcmp(buffer,"tarkin",6)){
+ /* not a tarkin header */
+ return(-TARKIN_NOT_TARKIN);
+ }
+ switch(packtype){
+ case 0x01: /* least significant *bit* is read first */
+ if(!op->b_o_s){
+ /* Not the initial packet */
+ return(-TARKIN_BAD_HEADER);
+ }
+ if(vi->inter.numerator!=0){
+ /* previously initialized info header */
+ return(-TARKIN_BAD_HEADER);
+ }
+
+ return(_tarkin_unpack_info(vi,&opb));
+
+ case 0x03: /* least significant *bit* is read first */
+ if(vi->inter.denominator==0){
+ /* um... we didn't get the initial header */
+ return(-TARKIN_BAD_HEADER);
+ }
+
+ return(_tarkin_unpack_comment(vc,&opb));
+
+ case 0x05: /* least significant *bit* is read first */
+ if(vi->inter.numerator == 0 || vc->vendor==NULL){
+ /* um... we didn;t get the initial header or comments yet */
+ return(-TARKIN_BAD_HEADER);
+ }
+
+ return(_tarkin_unpack_layer_desc(vi,&opb));
+
+ default:
+ /* Not a valid tarkin header type */
+ return(-TARKIN_BAD_HEADER);
+ break;
+ }
+ }
+ }
+ return(-TARKIN_BAD_HEADER);
+}
+
+/* pack side **********************************************************/
+
+static int _tarkin_pack_info(oggpack_buffer *opb,TarkinInfo *vi){
+
+ /* preamble */
+ oggpack_write(opb,0x01,8);
+ _v_writestring(opb,"tarkin", 6);
+
+ /* basic information about the stream */
+ oggpack_write(opb,0x00,32);
+ oggpack_write(opb,vi->n_layers,8);
+ oggpack_write(opb,vi->inter.numerator,32);
+ oggpack_write(opb,vi->inter.denominator,32);
+
+ oggpack_write(opb,vi->bitrate_upper,32);
+ oggpack_write(opb,vi->bitrate_nominal,32);
+ oggpack_write(opb,vi->bitrate_lower,32);
+
+ oggpack_write(opb,1,1);
+
+#ifdef DBG_OGG
+ printf("dbg_ogg: Putting out info, inter %d/%d, n_layers %d\n",
+ vi->inter.numerator,vi->inter.denominator,vi->n_layers);
+#endif
+ return(0);
+}
+
+static int _tarkin_pack_comment(oggpack_buffer *opb,TarkinComment *vc){
+ char temp[]="libTarkin debugging edition 20011104";
+ int bytes = strlen(temp);
+
+ /* preamble */
+ oggpack_write(opb,0x03,8);
+ _v_writestring(opb,"tarkin", 6);
+
+ /* vendor */
+ oggpack_write(opb,bytes,32);
+ _v_writestring(opb,temp, bytes);
+
+ /* comments */
+
+ oggpack_write(opb,vc->comments,32);
+ if(vc->comments){
+ int i;
+ for(i=0;i<vc->comments;i++){
+ if(vc->user_comments[i]){
+ oggpack_write(opb,vc->comment_lengths[i],32);
+ _v_writestring(opb,vc->user_comments[i], vc->comment_lengths[i]);
+ }else{
+ oggpack_write(opb,0,32);
+ }
+ }
+ }
+ oggpack_write(opb,1,1);
+
+#ifdef DBG_OGG
+ printf("dbg_ogg: Putting out %d comments\n", vc->comments);
+#endif
+
+ return(0);
+}
+
+static int _tarkin_pack_layer_desc(oggpack_buffer *opb,TarkinInfo *vi)
+{
+ int i;
+ TarkinVideoLayer *layer;
+
+#ifdef DBG_OGG
+ printf("dbg_ogg: Putting out layers description:\n");
+#endif
+
+ oggpack_write(opb,0x05,8);
+ _v_writestring(opb,"tarkin", 6);
+
+ for(i=0;i<vi->n_layers;i++){
+ layer = vi->layer + i;
+ oggpack_write(opb,layer->desc.width,32);
+ oggpack_write(opb,layer->desc.height,32);
+ oggpack_write(opb,layer->desc.a_moments,32);
+ oggpack_write(opb,layer->desc.s_moments,32);
+ oggpack_write(opb,layer->desc.frames_per_buf,32);
+ oggpack_write(opb,layer->desc.bitstream_len,32);
+ oggpack_write(opb,layer->desc.format,32);
+
+#ifdef DBG_OGG
+ printf(" res. %dx%d, format %d, a_m %d, s_m %d, fpb %d\n",
+ layer->desc.width, layer->desc.height, layer->desc.format,
+ layer->desc.a_moments, layer->desc.s_moments,
+ layer->desc.frames_per_buf);
+#endif
+
+ }
+ oggpack_write(opb,1,1);
+
+#ifdef DBG_OGG
+ printf(" wrote %ld bytes.\n", oggpack_bytes(opb));
+#endif
+
+ return(0);
+}
+
+int tarkin_comment_header_out(TarkinComment *vc, ogg_packet *op)
+{
+
+ oggpack_buffer opb;
+
+ oggpack_writeinit(&opb);
+ if(_tarkin_pack_comment(&opb,vc)) return -TARKIN_NOT_IMPLEMENTED;
+
+ op->packet = MALLOC(oggpack_bytes(&opb));
+ memcpy(op->packet, opb.buffer, oggpack_bytes(&opb));
+
+ op->bytes=oggpack_bytes(&opb);
+ op->b_o_s=0;
+ op->e_o_s=0;
+ op->granulepos=0;
+
+ return 0;
+}
+
+TarkinError tarkin_analysis_headerout(TarkinStream *v,
+ TarkinComment *vc,
+ ogg_packet *op,
+ ogg_packet *op_comm,
+ ogg_packet *op_code)
+{
+ int ret=-TARKIN_NOT_IMPLEMENTED;
+ TarkinInfo * vi;
+ oggpack_buffer opb;
+ tarkin_header_store *b=&v->headers;
+
+ vi = v->ti;
+
+ /* first header packet **********************************************/
+
+ oggpack_writeinit(&opb);
+ if(_tarkin_pack_info(&opb,vi))goto err_out;
+
+ /* build the packet */
+ if(b->header)FREE(b->header);
+ b->header=MALLOC(oggpack_bytes(&opb));
+ memcpy(b->header,opb.buffer,oggpack_bytes(&opb));
+ op->packet=b->header;
+ op->bytes=oggpack_bytes(&opb);
+ op->b_o_s=1;
+ op->e_o_s=0;
+ op->granulepos=0;
+
+ /* second header packet (comments) **********************************/
+
+ oggpack_reset(&opb);
+ if(_tarkin_pack_comment(&opb,vc))goto err_out;
+
+ if(b->header1)FREE(b->header1);
+ b->header1=MALLOC(oggpack_bytes(&opb));
+ memcpy(b->header1,opb.buffer,oggpack_bytes(&opb));
+ op_comm->packet=b->header1;
+ op_comm->bytes=oggpack_bytes(&opb);
+ op_comm->b_o_s=0;
+ op_comm->e_o_s=0;
+ op_comm->granulepos=0;
+
+ /* third header packet (modes/codebooks) ****************************/
+
+ oggpack_reset(&opb);
+ if(_tarkin_pack_layer_desc(&opb,vi))goto err_out;
+
+ if(b->header2)FREE(b->header2);
+ b->header2=MALLOC(oggpack_bytes(&opb));
+ memcpy(b->header2,opb.buffer,oggpack_bytes(&opb));
+ op_code->packet=b->header2;
+ op_code->bytes=oggpack_bytes(&opb);
+ op_code->b_o_s=0;
+ op_code->e_o_s=0;
+ op_code->granulepos=0;
+
+ oggpack_writeclear(&opb);
+ return(0);
+ err_out:
+ oggpack_writeclear(&opb);
+ memset(op,0,sizeof(*op));
+ memset(op_comm,0,sizeof(*op_comm));
+ memset(op_code,0,sizeof(*op_code));
+
+ if(b->header)FREE(b->header);
+ if(b->header1)FREE(b->header1);
+ if(b->header2)FREE(b->header2);
+ b->header=NULL;
+ b->header1=NULL;
+ b->header2=NULL;
+ return(ret);
+}
+
diff --git a/ext/tarkin/mem.c b/ext/tarkin/mem.c
new file mode 100644
index 00000000..c2cf18c0
--- /dev/null
+++ b/ext/tarkin/mem.c
@@ -0,0 +1,143 @@
+/*
+ * Debugging implementation of MALLOC and friends
+ */
+
+
+#include "mem.h"
+
+#if defined(DBG_MEMLEAKS)
+
+typedef struct {
+ void *mem;
+ char *allocated_in_func;
+ char *allocated_in_file;
+ unsigned int allocated_in_line;
+} MemDesc;
+
+
+static int initialized = 0;
+static int alloc_count = 0;
+static MemDesc *alloc_list = NULL;
+
+
+static
+void dbg_memleaks_done (int exitcode, void *dummy)
+{
+ unsigned int i;
+ (void) dummy;
+
+ if (exitcode == 0 && alloc_count != 0) {
+ fprintf (stderr, "\nmemory leak detected !!!\n");
+ fprintf (stderr, "\nalloc_count == %i\n\n", alloc_count);
+ for (i=0; i<alloc_count; i++) {
+ MemDesc *d = &alloc_list[i];
+ fprintf (stderr, "chunk %p allocated in %s (%s: %u) not free'd !!\n",
+ d->mem, d->allocated_in_func, d->allocated_in_file,
+ d->allocated_in_line);
+ }
+ free(alloc_list);
+ }
+ fprintf (stderr, "\n");
+}
+
+
+static
+void dbg_memleaks_init (void)
+{
+ on_exit (dbg_memleaks_done, NULL);
+ initialized = 1;
+}
+
+
+void* dbg_malloc (char* file, int line, char *func, size_t bytes)
+{
+ void *mem = (void*) malloc (bytes);
+ MemDesc *d;
+
+ if (!initialized)
+ dbg_memleaks_init();
+
+ alloc_count++;
+ alloc_list = realloc (alloc_list, alloc_count * sizeof(MemDesc));
+
+ d = &alloc_list[alloc_count-1];
+ d->mem = mem;
+ d->allocated_in_func = func;
+ d->allocated_in_file = file;
+ d->allocated_in_line = line;
+
+ return mem;
+}
+
+
+void* dbg_calloc (char* file, int line, char *func, size_t count, size_t bytes)
+{
+ void *mem = (void*) calloc (count, bytes);
+ MemDesc *d;
+
+ if (!initialized)
+ dbg_memleaks_init();
+
+ alloc_count++;
+ alloc_list = realloc (alloc_list, alloc_count * sizeof(MemDesc));
+
+ d = &alloc_list[alloc_count-1];
+ d->mem = mem;
+ d->allocated_in_func = func;
+ d->allocated_in_file = file;
+ d->allocated_in_line = line;
+
+ return mem;
+}
+
+
+void* dbg_realloc (char *file, int line, char *func, char *what,
+ void *mem, size_t bytes)
+{
+ unsigned int i;
+
+ for (i=0; i<alloc_count; i++) {
+ if (alloc_list[i].mem == mem) {
+ alloc_list[i].mem = (void*) realloc (mem, bytes);
+ return alloc_list[i].mem;
+ }
+ }
+
+ if (mem != NULL) {
+ fprintf (stderr,
+ "%s: trying to reallocate unknown chunk %p (%s)\n"
+ " in %s (%s: %u) !!!\n",
+ __FUNCTION__, mem, what, func, file, line);
+ exit (-1);
+ }
+
+ return dbg_malloc(file, line, func, bytes);
+}
+
+
+void dbg_free (char *file, int line, char *func, char *what, void *mem)
+{
+ unsigned int i;
+
+ if (!initialized)
+ dbg_memleaks_init();
+
+ for (i=0; i<alloc_count; i++) {
+ if (alloc_list[i].mem == mem) {
+ free (mem);
+ alloc_count--;
+ memmove (&alloc_list[i], &alloc_list[i+1],
+ (alloc_count - i) * sizeof(MemDesc));
+ return;
+ }
+ }
+
+ fprintf (stderr, "%s: trying to free unknown chunk %p (%s)\n"
+ " in %s (%s: %u) !!!\n",
+ __FUNCTION__, mem, what, func, file, line);
+ exit (-1);
+}
+
+
+#endif
+
diff --git a/ext/tarkin/mem.h b/ext/tarkin/mem.h
new file mode 100644
index 00000000..f218f8e4
--- /dev/null
+++ b/ext/tarkin/mem.h
@@ -0,0 +1,32 @@
+#ifndef __MEM_H
+#define __MEM_H
+
+#include <stdint.h>
+#include <stdlib.h>
+#include <string.h>
+#include <stdio.h>
+
+
+#if defined(DBG_MEMLEAKS)
+
+extern void* dbg_malloc (char *file, int line, char *func, size_t bytes);
+extern void* dbg_calloc (char *file, int line, char *func, size_t count, size_t bytes);
+extern void* dbg_realloc (char *file, int line, char *func, char *what, void *mem, size_t bytes);
+extern void dbg_free (char *file, int line, char *func, char *what, void *mem);
+
+#define MALLOC(bytes) dbg_malloc(__FILE__,__LINE__,__FUNCTION__,bytes)
+#define CALLOC(count,bytes) dbg_calloc(__FILE__,__LINE__,__FUNCTION__,count,bytes)
+#define FREE(mem) dbg_free(__FILE__,__LINE__,__FUNCTION__,#mem,mem)
+#define REALLOC(mem,bytes) dbg_realloc(__FILE__,__LINE__,__FUNCTION__,#mem,mem,bytes)
+
+#else
+
+#define MALLOC malloc
+#define CALLOC calloc
+#define REALLOC realloc
+#define FREE free
+
+#endif
+
+#endif
+
diff --git a/ext/tarkin/rle.h b/ext/tarkin/rle.h
new file mode 100644
index 00000000..7cf79517
--- /dev/null
+++ b/ext/tarkin/rle.h
@@ -0,0 +1,143 @@
+#ifndef __RLE_H
+#define __RLE_H
+
+#include <string.h>
+#include <assert.h>
+#include "mem.h"
+#include "bitcoder.h"
+#include "golomb.h"
+
+#if defined(RLECODER)
+
+#define OUTPUT_BIT(rlecoder,bit) rlecoder_write_bit(rlecoder,bit)
+#define INPUT_BIT(rlecoder) rlecoder_read_bit(rlecoder)
+#define OUTPUT_BIT_DIRECT(coder,bit) bitcoder_write_bit(&(coder)->bitcoder,bit)
+#define INPUT_BIT_DIRECT(rlecoder) bitcoder_read_bit(&(rlecoder)->bitcoder)
+#define ENTROPY_CODER RLECoderState
+#define ENTROPY_ENCODER_INIT(coder,limit) rlecoder_encoder_init(coder,limit)
+#define ENTROPY_ENCODER_DONE(coder) rlecoder_encoder_done(coder)
+#define ENTROPY_ENCODER_FLUSH(coder) rlecoder_encoder_flush(coder)
+#define ENTROPY_DECODER_INIT(coder,bitstream,limit) \
+ rlecoder_decoder_init(coder,bitstream,limit)
+#define ENTROPY_DECODER_DONE(coder) /* nothing to do ... */
+#define ENTROPY_CODER_BITSTREAM(coder) ((coder)->bitcoder.bitstream)
+#define ENTROPY_CODER_EOS(coder) ((coder)->bitcoder.eos)
+
+#define ENTROPY_CODER_SYMBOL(coder) ((coder)->symbol)
+#define ENTROPY_CODER_RUNLENGTH(coder) ((coder)->count)
+#define ENTROPY_CODER_SKIP(coder,skip) do { (coder)->count -= skip; } while (0)
+#endif
+
+
+
+
+typedef struct {
+ int symbol;
+ uint32_t count; /* have seen count symbol's */
+ BitCoderState bitcoder;
+ GolombAdaptiveCoderState golomb_state [2]; /* 2 states for 2 symbols... */
+ int have_seen_1;
+} RLECoderState;
+
+
+
+/*
+ * bit should be 0 or 1 !!!
+ */
+static inline
+void rlecoder_write_bit (RLECoderState *s, int bit)
+{
+ assert (bit == 0 || bit == 1);
+
+ if (s->symbol == -1) {
+ s->symbol = bit & 1;
+ s->count = 1;
+ s->have_seen_1 = bit;
+ bitcoder_write_bit (&s->bitcoder, bit);
+ }
+
+ if (s->symbol != bit) {
+ golombcoder_encode_number (&s->golomb_state[s->symbol],
+ &s->bitcoder, s->count);
+ s->symbol = ~s->symbol & 1;
+ s->have_seen_1 = 1;
+ s->count = 1;
+ } else
+ s->count++;
+}
+
+static inline
+int rlecoder_read_bit (RLECoderState *s)
+{
+ if (s->count == 0) {
+ s->symbol = ~s->symbol & 1;
+ s->count = golombcoder_decode_number (&s->golomb_state[s->symbol],
+ &s->bitcoder);
+ if (s->bitcoder.eos) {
+ s->symbol = 0;
+ s->count = ~0;
+ }
+ }
+ s->count--;
+ return (s->symbol);
+}
+
+
+int coder_id = 0;
+FILE *file = NULL;
+
+static inline
+void rlecoder_encoder_init (RLECoderState *s, uint32_t limit)
+{
+ bitcoder_encoder_init (&s->bitcoder, limit);
+ s->symbol = -1;
+ s->have_seen_1 = 0;
+ s->golomb_state[0].count = 0;
+ s->golomb_state[1].count = 0;
+ s->golomb_state[0].bits = 5 << 3;
+ s->golomb_state[1].bits = 5 << 3;
+}
+
+
+/**
+ * once you called this, you better should not encode any more symbols ...
+ */
+static inline
+uint32_t rlecoder_encoder_flush (RLECoderState *s)
+{
+ if (s->symbol == -1 || !s->have_seen_1)
+ return 0;
+
+ golombcoder_encode_number (&s->golomb_state[s->symbol],
+ &s->bitcoder, s->count);
+ return bitcoder_flush (&s->bitcoder);
+}
+
+
+static inline
+void rlecoder_decoder_init (RLECoderState *s, uint8_t *bitstream, uint32_t limit)
+{
+ bitcoder_decoder_init (&s->bitcoder, bitstream, limit);
+ s->golomb_state[0].count = 0;
+ s->golomb_state[1].count = 0;
+ s->golomb_state[0].bits = 5 << 3;
+ s->golomb_state[1].bits = 5 << 3;
+ s->symbol = bitcoder_read_bit (&s->bitcoder);
+ s->count = golombcoder_decode_number (&s->golomb_state[s->symbol],
+ &s->bitcoder) - 1;
+ if (s->bitcoder.eos) {
+ s->symbol = 0;
+ s->count = ~0;
+ }
+}
+
+
+static inline
+void rlecoder_encoder_done (RLECoderState *s)
+{
+ bitcoder_encoder_done (&s->bitcoder);
+}
+
+
+#endif
+
diff --git a/ext/tarkin/tarkin.c b/ext/tarkin/tarkin.c
new file mode 100644
index 00000000..053c03f7
--- /dev/null
+++ b/ext/tarkin/tarkin.c
@@ -0,0 +1,418 @@
+/*
+ * The real io-stuff is in tarkin-io.c
+ * (this one has to be rewritten to write ogg streams ...)
+ */
+
+#include "mem.h"
+#include "tarkin.h"
+#include "yuv.h"
+
+
+#define N_FRAMES 1
+
+
+
+TarkinStream* tarkin_stream_new ()
+{
+ TarkinStream *s = (TarkinStream*) CALLOC (1, sizeof(TarkinStream));
+
+ if (!s)
+ return NULL;
+ memset(s,0,sizeof(*s));
+
+ s->frames_per_buf = N_FRAMES;
+
+ return s;
+}
+
+
+void tarkin_stream_destroy (TarkinStream *s)
+{
+ uint32_t i, j;
+
+ if (!s)
+ return;
+
+ for (i=0; i<s->n_layers; i++) {
+ if (s->layer[i].waveletbuf) {
+ for (j=0; j<s->layer[i].n_comp; j++) {
+ wavelet_3d_buf_destroy (s->layer[i].waveletbuf[j]);
+ FREE (s->layer[i].packet[j].data);
+ }
+ FREE(s->layer[i].waveletbuf);
+ FREE(s->layer[i].packet);
+ }
+ }
+
+ if (s->layer)
+ FREE(s->layer);
+
+ if (s->headers.header)
+ FREE(s->headers.header);
+
+ if (s->headers.header1)
+ FREE(s->headers.header1);
+
+ if (s->headers.header2)
+ FREE(s->headers.header2);
+
+
+ FREE(s);
+}
+
+
+int tarkin_analysis_init(TarkinStream *s, TarkinInfo *ti,
+ TarkinError (*free_frame)(void *s, void *ptr),
+ TarkinError (*packet_out)(void *s, ogg_packet *ptr),
+ void *user_ptr)
+{
+ if((!ti->inter.numerator)||(!ti->inter.denominator))return (-TARKIN_FAULT);
+ if((!free_frame) || (!packet_out)) return (-TARKIN_FAULT);
+ s->ti = ti;
+ s->free_frame = free_frame;
+ s->packet_out = packet_out;
+ s->user_ptr = user_ptr;
+ return(0);
+}
+
+
+extern int tarkin_analysis_add_layer(TarkinStream *s,
+ TarkinVideoLayerDesc *tvld)
+{
+ int i;
+ TarkinVideoLayer *layer;
+ if(s->n_layers) {
+ s->layer = REALLOC(s->layer,(s->n_layers+1) * sizeof(*s->layer));
+ } else {
+ s->layer = MALLOC(sizeof(*s->layer));
+ }
+ layer = s->layer + s->n_layers;
+ memset(layer,0,sizeof(*s->layer));
+ memcpy (&layer->desc , tvld, sizeof(TarkinVideoLayerDesc));
+
+ s->n_layers++;
+ s->ti->n_layers = s->n_layers;
+ s->ti->layer = s->layer;
+
+ switch (layer->desc.format) {
+ case TARKIN_GRAYSCALE:
+ layer->n_comp = 1;
+ layer->color_fwd_xform = grayscale_to_y;
+ layer->color_inv_xform = y_to_grayscale;
+ break;
+ case TARKIN_RGB24:
+ layer->n_comp = 3;
+ layer->color_fwd_xform = rgb24_to_yuv;
+ layer->color_inv_xform = yuv_to_rgb24;
+ break;
+ case TARKIN_RGB32:
+ layer->n_comp = 3;
+ layer->color_fwd_xform = rgb32_to_yuv;
+ layer->color_inv_xform = yuv_to_rgb32;
+ break;
+ case TARKIN_RGBA:
+ layer->n_comp = 4;
+ layer->color_fwd_xform = rgba_to_yuv;
+ layer->color_inv_xform = yuv_to_rgba;
+ break;
+ default:
+ return -TARKIN_INVALID_COLOR_FORMAT;
+ };
+
+#ifdef DBG_OGG
+ printf("dbg_ogg:add_layer %d with %d components\n",
+ s->n_layers, layer->n_comp);
+#endif
+
+ layer->waveletbuf = (Wavelet3DBuf**) CALLOC (layer->n_comp,
+ sizeof(Wavelet3DBuf*));
+
+ layer->packet = MALLOC (layer->n_comp * sizeof(*layer->packet));
+ memset(layer->packet, 0, layer->n_comp * sizeof(*layer->packet));
+
+ for (i=0; i<layer->n_comp; i++){
+ layer->waveletbuf[i] = wavelet_3d_buf_new (layer->desc.width,
+ layer->desc.height,
+ layer->desc.frames_per_buf);
+ layer->packet[i].data = MALLOC(layer->desc.bitstream_len);
+ layer->packet[i].storage = layer->desc.bitstream_len;
+ }
+ /*
+ max_bitstream_len += layer->desc.bitstream_len
+ + 2 * 10 * sizeof(uint32_t) * layer->n_comp; // truncation tables
+ */
+ return (TARKIN_OK);
+}
+
+TarkinError _analysis_packetout(TarkinStream *s, uint32_t layer_id,
+ uint32_t comp)
+{
+ ogg_packet op;
+ oggpack_buffer opb;
+ uint8_t *data;
+ uint32_t data_len;
+ int i;
+ data = s->layer[layer_id].packet[comp].data;
+ data_len = s->layer[layer_id].packet[comp].data_len;
+
+ oggpack_writeinit(&opb);
+ oggpack_write(&opb,0,8); /* No feature flags for now */
+ oggpack_write(&opb,layer_id,12);
+ oggpack_write(&opb,comp,12);
+ for(i=0;i<data_len;i++)
+ oggpack_write(&opb,*(data + i), 8);
+
+ op.b_o_s = 0;
+ op.e_o_s = data_len?0:1;
+ op.granulepos = 0;
+ op.bytes = oggpack_bytes(&opb)+4;
+ op.packet = opb.buffer;
+#ifdef DBG_OGG
+ printf("dbg_ogg: writing packet layer %d, comp %d, data_len %d %s\n",
+ layer_id, comp, data_len, op.e_o_s?"eos":"");
+#endif
+ s->layer[layer_id].packet[comp].data_len = 0; /* so direct call => eos */
+ return(s->packet_out(s,&op));
+}
+
+void _stream_flush (TarkinStream *s)
+{
+ uint32_t i, j;
+
+ s->current_frame_in_buf=0;
+
+ for (i=0; i<s->n_layers; i++) {
+ TarkinVideoLayer *layer = &s->layer[i];
+
+ for (j=0; j<layer->n_comp; j++) {
+ uint32_t comp_bitstream_len;
+ TarkinPacket *packet = layer->packet + j;
+
+ /**
+ * implicit 6:1:1 subsampling
+ */
+ if (j == 0)
+ comp_bitstream_len = 6*layer->desc.bitstream_len/(layer->n_comp+5);
+ else
+ comp_bitstream_len = layer->desc.bitstream_len/(layer->n_comp+5);
+
+ if(packet->storage < comp_bitstream_len) {
+ packet->storage = comp_bitstream_len;
+ packet->data = REALLOC (packet->data, comp_bitstream_len);
+ }
+
+ wavelet_3d_buf_dump ("color-%d-%03d.pgm",
+ s->current_frame, j,
+ layer->waveletbuf[j], j == 0 ? 0 : 128);
+
+ wavelet_3d_buf_fwd_xform (layer->waveletbuf[j],
+ layer->desc.a_moments,
+ layer->desc.s_moments);
+
+ wavelet_3d_buf_dump ("coeff-%d-%03d.pgm",
+ s->current_frame, j,
+ layer->waveletbuf[j], 128);
+
+ packet->data_len = wavelet_3d_buf_encode_coeff (layer->waveletbuf[j],
+ packet->data,
+ comp_bitstream_len);
+
+ _analysis_packetout (s, i, j);
+ }
+ }
+}
+
+
+uint32_t tarkin_analysis_framein (TarkinStream *s, uint8_t *frame,
+ uint32_t layer_id, TarkinTime *date)
+{
+ TarkinVideoLayer *layer;
+
+ if(!frame) return (_analysis_packetout(s, 0, 0)); /* eos */
+ if((layer_id>=s->n_layers) || (date->denominator==0)) return (TARKIN_FAULT);
+
+ layer = s->layer + layer_id;
+ layer->color_fwd_xform (frame, layer->waveletbuf,
+ s->current_frame_in_buf);
+ /* We don't use this feature for now, neither date... */
+ s->free_frame(s,frame);
+
+ s->current_frame_in_buf++;
+
+ if (s->current_frame_in_buf == s->frames_per_buf)
+ _stream_flush (s);
+
+#ifdef DBG_OGG
+ printf("dbg_ogg: framein at pos %d/%d, n° %d,%d on layer %d\n",
+ date->numerator, date->denominator,
+ layer->frameno, s->current_frame, layer_id);
+#endif
+
+ layer->frameno++;
+ return (++s->current_frame);
+}
+
+
+
+
+/**
+ * tarkin_stream_read_header() is now info.c:_tarkin_unpack_layer_desc()
+ */
+
+
+
+TarkinError tarkin_stream_get_layer_desc (TarkinStream *s,
+ uint32_t layer_id,
+ TarkinVideoLayerDesc *desc)
+{
+ if (layer_id > s->n_layers-1)
+ return -TARKIN_INVALID_LAYER;
+
+ memcpy (desc, &(s->layer[layer_id].desc), sizeof(TarkinVideoLayerDesc));
+
+ return TARKIN_OK;
+}
+
+TarkinError tarkin_synthesis_init (TarkinStream *s, TarkinInfo *ti)
+{
+ s->ti = ti;
+ s->layer = ti->layer; /* It was malloc()ed by headerin() */
+ s->n_layers = ti->n_layers;
+ return (TARKIN_OK);
+}
+
+TarkinError tarkin_synthesis_packetin (TarkinStream *s, ogg_packet *op)
+{
+ uint32_t i, layer_id, comp, data_len;
+ uint32_t flags, junk;
+ int nread;
+ oggpack_buffer opb;
+ TarkinPacket *packet;
+#ifdef DBG_OGG
+ printf("dbg_ogg: Reading packet n° %lld, granulepos %lld, len %ld, %s%s\n",
+ op->packetno, op->granulepos, op->bytes,
+ op->b_o_s?"b_o_s":"", op->e_o_s?"e_o_s":"");
+#endif
+ oggpack_readinit(&opb,op->packet,op->bytes);
+ flags = oggpack_read(&opb,8);
+ layer_id = oggpack_read(&opb,12); /* Theses are required for */
+ comp = oggpack_read(&opb,12); /* data hole handling (or maybe
+ * packetno would be enough ?) */
+ nread = 4;
+
+ if(flags){ /* This is void "infinite future features" feature ;) */
+ if(flags & 1<<7){
+ junk = flags;
+ while (junk & 1<<7)
+ junk = oggpack_read(&opb,8); /* allow for many future flags
+ that must be correctly ordonned. */
+ }
+ /* This shows how to get a feature's data:
+ if (flags & TARKIN_FLAGS_EXAMPLE){
+ tp->example = oggpack_read(&opb,32);
+ junk = tp->example & 3<<30;
+ tp->example &= 0x4fffffff;
+ }
+ */
+ for(junk=1<<31;junk & 1<<31;) /* and many future data */
+ while((junk=oggpack_read(&opb,32)) & 1<<30);
+ /* That is, feature data comes in 30 bit chunks. We also have
+ * 31 potentially usefull bits in last chunk. */
+ }
+
+ nread = (opb.ptr - opb.buffer);
+ data_len = op->bytes - nread;
+
+#ifdef DBG_OGG
+ printf(" layer_id %d, comp %d, meta-data %dB, w3d data %dB.\n",
+ layer_id, comp,nread, data_len);
+#endif
+
+ /* We now have for shure our data. */
+ packet = &s->layer[layer_id].packet[comp];
+ if(packet->data_len)return(-TARKIN_UNUSED); /* Previous data wasn't used */
+
+ if(packet->storage < data_len){
+ packet->storage = data_len + 255;
+ packet->data = REALLOC (packet->data, packet->storage);
+ }
+
+ for(i=0;i < data_len ; i++)
+ packet->data[i] = oggpack_read(&opb,8);
+
+ packet->data_len = data_len;
+
+ return(TARKIN_OK);
+}
+
+TarkinError tarkin_synthesis_frameout(TarkinStream *s,
+ uint8_t **frame,
+ uint32_t layer_id, TarkinTime *date)
+{
+ int j;
+ TarkinVideoLayer *layer = &s->layer[layer_id];
+ if (s->current_frame_in_buf == 0) {
+ *frame = MALLOC (layer->desc.width * layer->desc.height * layer->n_comp);
+ for (j=0; j<layer->n_comp; j++) {
+ TarkinPacket *packet = layer->packet + j;
+
+ if(packet->data_len == 0)goto err_out ;
+
+ wavelet_3d_buf_decode_coeff (layer->waveletbuf[j], packet->data,
+ packet->data_len);
+
+ wavelet_3d_buf_dump ("rcoeff-%d-%03d.pgm",
+ s->current_frame, j, layer->waveletbuf[j],
+ 128);
+
+ wavelet_3d_buf_inv_xform (layer->waveletbuf[j],
+ layer->desc.a_moments,
+ layer->desc.s_moments);
+
+ wavelet_3d_buf_dump ("rcolor-%d-%03d.pgm",
+ s->current_frame, j,
+ layer->waveletbuf[j], j == 0 ? 0 : 128);
+ }
+
+ /* We did successfylly read a block from this layer, acknowledge it. */
+ for (j=0; j < layer->n_comp; j++)
+ layer->packet[j].data_len = 0;
+ }
+
+ layer->color_inv_xform (layer->waveletbuf, *frame,
+ s->current_frame_in_buf);
+ s->current_frame_in_buf++;
+ s->current_frame++;
+
+ if (s->current_frame_in_buf == s->frames_per_buf)
+ s->current_frame_in_buf=0;
+
+ date->numerator = layer->frameno * s->ti->inter.numerator;
+ date->denominator = s->ti->inter.denominator;
+#ifdef DBG_OGG
+ printf("dbg_ogg: outputting frame pos %d/%d from layer %d.\n",
+ date->numerator, date->denominator, layer_id);
+#endif
+ layer->frameno++;
+ return (TARKIN_OK);
+err_out:
+ FREE(*frame);
+ return (TARKIN_NEED_MORE);
+}
+
+int tarkin_synthesis_freeframe(TarkinStream *s, uint8_t *frame)
+{
+ FREE(frame);
+
+ return(TARKIN_OK);
+}
+
+
+
+
+
+
+
+
+
+
diff --git a/ext/tarkin/tarkin.h b/ext/tarkin/tarkin.h
new file mode 100644
index 00000000..633f9a14
--- /dev/null
+++ b/ext/tarkin/tarkin.h
@@ -0,0 +1,239 @@
+#ifndef __TARKIN_H
+#define __TARKIN_H
+
+#include <stdio.h>
+#include "wavelet.h"
+#include <ogg/ogg.h>
+
+
+#define BUG(x...) \
+ do { \
+ printf("BUG in %s (%s: line %i): ", __FUNCTION__, __FILE__, __LINE__); \
+ printf(#x); \
+ printf("\n"); \
+ exit (-1); \
+ } while (0);
+
+
+/* Theses determine what infos the packet comes with */
+#define TARKIN_PACK_EXAMPLE 1
+
+typedef struct {
+ uint8_t *data;
+ uint32_t data_len;
+ uint32_t storage;
+} TarkinPacket;
+
+
+typedef enum {
+ TARKIN_GRAYSCALE,
+ TARKIN_RGB24, /* tight packed RGB */
+ TARKIN_RGB32, /* 32bit, no alphachannel */
+ TARKIN_RGBA, /* dito w/ alphachannel */
+ TARKIN_YUV2, /* 16 bits YUV */
+ TARKIN_YUV12, /* 12 bits YUV */
+ TARKIN_FYUV, /* Tarkin's Fast YUV-like? */
+} TarkinColorFormat;
+
+#define TARKIN_INTERNAL_FORMAT TARKIN_FYUV
+
+typedef enum {
+ TARKIN_OK = 0,
+ TARKIN_IO_ERROR,
+ TARKIN_SIGNATURE_NOT_FOUND,
+ TARKIN_INVALID_LAYER,
+ TARKIN_INVALID_COLOR_FORMAT,
+ TARKIN_VERSION,
+ TARKIN_BAD_HEADER,
+ TARKIN_NOT_TARKIN,
+ TARKIN_FAULT,
+ TARKIN_UNUSED,
+ TARKIN_NEED_MORE,
+ TARKIN_NOT_IMPLEMENTED
+} TarkinError;
+
+
+
+typedef struct {
+ uint32_t width;
+ uint32_t height;
+ uint32_t a_moments;
+ uint32_t s_moments;
+ uint32_t frames_per_buf;
+ uint32_t bitstream_len; /* for all color components, bytes */
+ TarkinColorFormat format;
+} TarkinVideoLayerDesc;
+
+
+typedef struct {
+ TarkinVideoLayerDesc desc;
+ uint32_t n_comp; /* number of color components */
+ Wavelet3DBuf **waveletbuf;
+ TarkinPacket *packet;
+ uint32_t current_frame_in_buf;
+ uint32_t frameno;
+
+ void (*color_fwd_xform) (uint8_t *rgba, Wavelet3DBuf *yuva [], uint32_t count);
+ void (*color_inv_xform) (Wavelet3DBuf *yuva [], uint8_t *rgba, uint32_t count);
+} TarkinVideoLayer;
+
+typedef struct {
+ uint32_t numerator;
+ uint32_t denominator;
+} TarkinTime; /* Let's say the unit is 1 second */
+
+typedef struct TarkinInfo {
+ int version;
+ int n_layers;
+ TarkinVideoLayer *layer;
+ TarkinTime inter; /* numerator == O if per-frame time info. */
+ int frames_per_block;
+ int comp_per_block; /* AKA "packets per block" for now */
+ uint32_t max_bitstream_len;
+
+ /* The below bitrate declarations are *hints*.
+ Combinations of the three values carry the following implications:
+
+ all three set to the same value:
+ implies a fixed rate bitstream
+ only nominal set:
+ implies a VBR stream that averages the nominal bitrate. No hard
+ upper/lower limit
+ upper and or lower set:
+ implies a VBR bitstream that obeys the bitrate limits. nominal
+ may also be set to give a nominal rate.
+ none set:
+ the coder does not care to speculate.
+ */
+
+ long bitrate_upper;
+ long bitrate_nominal;
+ long bitrate_lower;
+ long bitrate_window;
+} TarkinInfo;
+
+/* This is used for encoding */
+typedef struct {
+ unsigned char *header;
+ unsigned char *header1;
+ unsigned char *header2;
+} tarkin_header_store;
+
+
+
+
+ /* Some of the fields in TarkinStream are redundent with TarkinInfo ones
+ * and will probably get deleted, namely n_layers and frames_per_buf */
+typedef struct TarkinStream {
+ uint32_t n_layers;
+ TarkinVideoLayer *layer;
+ uint32_t current_frame;
+ uint32_t current_frame_in_buf;
+ ogg_int64_t packetno;
+ uint32_t frames_per_buf;
+ uint32_t max_bitstream_len;
+ TarkinInfo *ti;
+ tarkin_header_store headers;
+ /* These callbacks are only used for encoding */
+ TarkinError (*free_frame)(void *tarkinstream, void *ptr);
+ /* These thing allows not to buffer but it needs global var in caller. */
+ TarkinError (*packet_out)(void *tarkinstream, ogg_packet *ptr);
+ void * user_ptr;
+} TarkinStream;
+
+
+typedef struct TarkinComment{
+ /* unlimited user comment fields. libtarkin writes 'libtarkin'
+ whatever vendor is set to in encode */
+ char **user_comments;
+ int *comment_lengths;
+ int comments;
+ char *vendor;
+
+} TarkinComment;
+
+/* Tarkin PRIMITIVES: general ***************************************/
+
+/* The Tarkin header is in three packets, the initial small packet in
+ the first page that identifies basic parameters, that is a TarkinInfo
+ structure, a second packet with bitstream comments and a third packet
+ that holds the layers description structures. */
+
+
+/* Theses are the very same than Vorbis versions, they could be shared. */
+extern TarkinStream* tarkin_stream_new ();
+extern void tarkin_stream_destroy (TarkinStream *s);
+extern void tarkin_info_init(TarkinInfo *vi);
+extern void tarkin_info_clear(TarkinInfo *vi);
+extern void tarkin_comment_init(TarkinComment *vc);
+extern void tarkin_comment_add(TarkinComment *vc, char *comment);
+extern void tarkin_comment_add_tag(TarkinComment *vc,
+ char *tag, char *contents);
+extern char *tarkin_comment_query(TarkinComment *vc, char *tag, int count);
+extern int tarkin_comment_query_count(TarkinComment *vc, char *tag);
+extern void tarkin_comment_clear(TarkinComment *vc);
+
+/* Tarkin PRIMITIVES: analysis layer ****************************/
+/* Tarkin encoding is done this way : you init it passing a fresh
+ * TarkinStream and a fresh TarkinInfo which has at least the rate_num
+ * field renseigned. You also pass it two callback functions: free_frame()
+ * is called when the lib doesn't need a frame anymore, and packet_out
+ * is called when a packet is ready. The pointers given as arguments to
+ * these callback functions are of course only valid at the function call
+ * time. The user_ptr is stored in s and can be used by packet_out(). */
+extern int tarkin_analysis_init(TarkinStream *s,
+ TarkinInfo *ti,
+ TarkinError (*free_frame)(void *tarkinstream, void *ptr),
+ TarkinError (*packet_out)(void *tarkinstream, ogg_packet *ptr),
+ void *user_ptr
+ );
+/* Then you need to add at least a layer in your stream, passing a
+ * TarkinVideoLayerDesc renseigned at least on the width, height and
+ * format parameters. */
+extern int tarkin_analysis_add_layer(TarkinStream *s,
+ TarkinVideoLayerDesc *tvld);
+/* At that point you are ready to get headers out the lib by calling
+ * tarkin_analysis_headerout() passing it a renseigned TarkinComment
+ * structure. It does fill your 3 ogg_packet headers, which are valid
+ * till next call */
+extern int TarkinCommentheader_out(TarkinComment *vc, ogg_packet *op);
+extern TarkinError tarkin_analysis_headerout(TarkinStream *s,
+ TarkinComment *vc,
+ ogg_packet *op,
+ ogg_packet *op_comm,
+ ogg_packet *op_code);
+/* You are now ready to pass in frames to the codec, however don't free
+ * them before the codec told you so. It'll tell you when packets are
+ * ready to be taken out. When you have no more frame, simply pass NULL.
+ * If you encode multiple layers you have to do it synchronously, putting
+ * one frame from each layer at a time. */
+extern uint32_t tarkin_analysis_framein(TarkinStream *s,
+ uint8_t *frame, /* NULL for EOS */
+ uint32_t layer,
+ TarkinTime *date);
+
+/* Tarkin PRIMITIVES: synthesis layer *******************************/
+/* For decoding, you needs first to give the three first packet of the
+ * stream to tarkin_synthesis_headerin() which will fill for you blank
+ * TarkinInfo and TarkinComment. */
+extern TarkinError tarkin_synthesis_headerin(TarkinInfo *vi,TarkinComment *vc,
+ ogg_packet *op);
+/* Then you can init your stream with your TarkinInfo struct. */
+extern TarkinError tarkin_synthesis_init(TarkinStream *s,TarkinInfo *ti);
+/* All subsequent packets are to this be passed to tarkin_synthesis_packetin*/
+extern TarkinError tarkin_synthesis_packetin(TarkinStream *s, ogg_packet *op);
+/* and then tarkin_synthesis_frameout gives you ptr on next frame, or NULL. It
+ * also fills for you date. */
+extern TarkinError tarkin_synthesis_frameout(TarkinStream *s,
+ uint8_t **frame, uint32_t layer_id, TarkinTime *date);
+/* When you're done with a frame, tell it to the codec with this. */
+extern int tarkin_synthesis_freeframe(TarkinStream *s, uint8_t *frame);
+
+
+#endif
+
+
+
+
+
+
diff --git a/ext/tarkin/wavelet.c b/ext/tarkin/wavelet.c
new file mode 100644
index 00000000..a83aa499
--- /dev/null
+++ b/ext/tarkin/wavelet.c
@@ -0,0 +1,121 @@
+#include "mem.h"
+#include "wavelet.h"
+
+/**
+ * (The transform code is in wavelet_xform.c)
+ */
+
+
+#define MAX(a,b) ((a) > (b) ? (a) : (b))
+#define MAX3(a,b,c) (MAX(a,MAX(b,c)))
+
+
+
+Wavelet3DBuf* wavelet_3d_buf_new (uint32_t width, uint32_t height,
+ uint32_t frames)
+{
+ Wavelet3DBuf* buf = (Wavelet3DBuf*) MALLOC (sizeof (Wavelet3DBuf));
+ uint32_t _w = width;
+ uint32_t _h = height;
+ uint32_t _f = frames;
+ int level;
+
+ if (!buf)
+ return NULL;
+
+ buf->data = (TYPE*) MALLOC (width * height * frames * sizeof (TYPE));
+
+ if (!buf->data) {
+ wavelet_3d_buf_destroy (buf);
+ return NULL;
+ }
+
+ buf->width = width;
+ buf->height = height;
+ buf->frames = frames;
+ buf->scales = 1;
+
+ while (_w > 1 || _h > 1 || _f > 1) {
+ buf->scales++;
+ _w = (_w+1)/2;
+ _h = (_h+1)/2;
+ _f = (_f+1)/2;
+ }
+
+ buf->w = (uint32_t*) MALLOC (buf->scales * sizeof (uint32_t));
+ buf->h = (uint32_t*) MALLOC (buf->scales * sizeof (uint32_t));
+ buf->f = (uint32_t*) MALLOC (buf->scales * sizeof (uint32_t));
+ buf->offset = (uint32_t (*) [8]) MALLOC (8 * buf->scales * sizeof (uint32_t));
+
+ buf->scratchbuf = (TYPE*) MALLOC (MAX3(width, height, frames) * sizeof (TYPE));
+
+ if (!buf->w || !buf->h || !buf->f || !buf->offset || !buf->scratchbuf) {
+ wavelet_3d_buf_destroy (buf);
+ return NULL;
+ }
+
+ buf->w [buf->scales-1] = width;
+ buf->h [buf->scales-1] = height;
+ buf->f [buf->scales-1] = frames;
+
+ for (level=buf->scales-2; level>=0; level--) {
+ buf->w [level] = (buf->w [level+1] + 1) / 2;
+ buf->h [level] = (buf->h [level+1] + 1) / 2;
+ buf->f [level] = (buf->f [level+1] + 1) / 2;
+ buf->offset[level][0] = 0;
+ buf->offset[level][1] = buf->w [level];
+ buf->offset[level][2] = buf->h [level] * width;
+ buf->offset[level][3] = buf->f [level] * width * height;
+ buf->offset[level][4] = buf->offset [level][2] + buf->w [level];
+ buf->offset[level][5] = buf->offset [level][3] + buf->w [level];
+ buf->offset[level][6] = buf->offset [level][3] + buf->offset [level][2];
+ buf->offset[level][7] = buf->offset [level][6] + buf->w [level];
+ }
+
+ return buf;
+}
+
+
+void wavelet_3d_buf_destroy (Wavelet3DBuf* buf)
+{
+ if (buf) {
+ if (buf->data)
+ FREE (buf->data);
+ if (buf->w)
+ FREE (buf->w);
+ if (buf->h)
+ FREE (buf->h);
+ if (buf->f)
+ FREE (buf->f);
+ if (buf->offset)
+ FREE (buf->offset);
+ if (buf->scratchbuf)
+ FREE (buf->scratchbuf);
+ FREE (buf);
+ }
+}
+
+
+#if defined(DBG_XFORM)
+
+#include "pnm.h"
+
+void wavelet_3d_buf_dump (char *fmt,
+ uint32_t first_frame_in_buf,
+ uint32_t id,
+ Wavelet3DBuf* buf,
+ int16_t offset)
+{
+ char fname [256];
+ uint32_t f;
+
+ for (f=0; f<buf->frames; f++) {
+ snprintf (fname, 256, fmt, id, first_frame_in_buf + f);
+
+ write_pgm16 (fname, buf->data + f * buf->width * buf->height,
+ buf->width, buf->height, offset);
+ }
+}
+#endif
+
+
diff --git a/ext/tarkin/wavelet.h b/ext/tarkin/wavelet.h
new file mode 100644
index 00000000..914c2799
--- /dev/null
+++ b/ext/tarkin/wavelet.h
@@ -0,0 +1,55 @@
+#ifndef __WAVELET_H
+#define __WAVELET_H
+
+#include <stdint.h>
+
+
+typedef struct {
+ TYPE *data;
+ uint32_t width;
+ uint32_t height;
+ uint32_t frames;
+ uint32_t scales;
+ uint32_t *w;
+ uint32_t *h;
+ uint32_t *f;
+ uint32_t (*offset)[8];
+ TYPE *scratchbuf;
+} Wavelet3DBuf;
+
+
+extern Wavelet3DBuf* wavelet_3d_buf_new (uint32_t width, uint32_t height,
+ uint32_t frames);
+
+extern void wavelet_3d_buf_destroy (Wavelet3DBuf* buf);
+
+/**
+ * transform buf->data
+ * a_moments is the number of vanishing moments of the analyzing
+ * highpass filter,
+ * s_moments the one of the synthesizing lowpass filter.
+ */
+extern void wavelet_3d_buf_fwd_xform (Wavelet3DBuf* buf,
+ int a_moments, int s_moments);
+extern void wavelet_3d_buf_inv_xform (Wavelet3DBuf* buf,
+ int a_moments, int s_moments);
+
+extern int wavelet_3d_buf_encode_coeff (const Wavelet3DBuf* buf,
+ uint8_t *bitstream,
+ uint32_t limit);
+
+extern void wavelet_3d_buf_decode_coeff (Wavelet3DBuf* buf,
+ uint8_t *bitstream,
+ uint32_t limit);
+
+#if defined(DBG_XFORM)
+extern void wavelet_3d_buf_dump (char *fmt,
+ uint32_t first_frame_in_buf,
+ uint32_t id,
+ Wavelet3DBuf* buf,
+ int16_t offset);
+#else
+#define wavelet_3d_buf_dump(x...)
+#endif
+
+#endif
diff --git a/ext/tarkin/wavelet_coeff.c b/ext/tarkin/wavelet_coeff.c
new file mode 100644
index 00000000..2cc0a8f0
--- /dev/null
+++ b/ext/tarkin/wavelet_coeff.c
@@ -0,0 +1,506 @@
+#include "mem.h"
+#include "wavelet.h"
+#include "rle.h"
+
+#define printf(args...)
+
+#define GRAY_CODES 1
+
+#if defined(GRAY_CODES)
+static inline
+uint16_t binary_to_gray (uint16_t x) { return x ^ (x >> 1); }
+
+static inline
+uint16_t gray_to_binary (uint16_t x)
+{ int i; for (i=1; i<16; i+=i) x ^= x >> i; return x; }
+#endif
+
+
+static inline
+void encode_coeff (ENTROPY_CODER significand_bitstream [],
+ ENTROPY_CODER insignificand_bitstream [],
+ TYPE coeff)
+{
+ int sign = (coeff >> (8*sizeof(TYPE)-1)) & 1;
+#if defined(GRAY_CODES)
+ TYPE significance = binary_to_gray(coeff);
+#else
+ static TYPE mask [2] = { 0, ~0 };
+ TYPE significance = coeff ^ mask[sign];
+#endif
+ int i = TYPE_BITS;
+
+ do {
+ i--;
+ OUTPUT_BIT(&significand_bitstream[i], (significance >> i) & 1);
+ } while (!((significance >> i) & 1) && i > 0);
+
+ OUTPUT_BIT(&significand_bitstream[i], sign);
+
+ while (--i >= 0)
+ OUTPUT_BIT(&insignificand_bitstream[i], (significance >> i) & 1);
+}
+
+
+
+static inline
+TYPE decode_coeff (ENTROPY_CODER significand_bitstream [],
+ ENTROPY_CODER insignificand_bitstream [])
+{
+#if !defined(GRAY_CODES)
+ static TYPE mask [2] = { 0, ~0 };
+#endif
+ TYPE significance = 0;
+ int sign;
+ int i = TYPE_BITS;
+
+ do {
+ i--;
+ significance |= INPUT_BIT(&significand_bitstream[i]) << i;
+// if (ENTROPY_CODER_EOS(&significand_bitstream[i]))
+// return 0;
+ } while (!significance && i > 0);
+
+ sign = INPUT_BIT(&significand_bitstream[i]);
+// if (ENTROPY_CODER_EOS(&significand_bitstream[i]))
+// return 0;
+
+ while (--i >= 0)
+ significance |= INPUT_BIT(&insignificand_bitstream[i]) << i;
+
+#if defined(GRAY_CODES)
+ significance |= sign << (8*sizeof(TYPE)-1);
+ return gray_to_binary(significance);
+#else
+ return (significance ^ mask[sign]);
+#endif
+}
+
+
+static inline
+uint32_t skip_0coeffs (Wavelet3DBuf* buf,
+ ENTROPY_CODER s_stream [],
+ ENTROPY_CODER i_stream [],
+ uint32_t limit)
+{
+ int i;
+ uint32_t skip = limit;
+
+ for (i=0; i<TYPE_BITS; i++) {
+ if (ENTROPY_CODER_SYMBOL(&s_stream[i]) != 0) {
+ return 0;
+ } else {
+ uint32_t runlength = ENTROPY_CODER_RUNLENGTH(&s_stream[i]);
+ if (i==0)
+ runlength /= 2; /* sign bits are in this bitplane ... */
+ if (skip > runlength)
+ skip = runlength;
+ if (skip <= 2)
+ return 0;
+ }
+ }
+
+ ENTROPY_CODER_SKIP(&s_stream[0], 2*skip); /* kill sign+significance bits */
+
+ for (i=1; i<TYPE_BITS; i++)
+ ENTROPY_CODER_SKIP(&s_stream[i], skip);
+
+ return skip;
+}
+
+
+
+#if 1
+static inline
+void encode_quadrant (const Wavelet3DBuf* buf,
+ int level, int quadrant, uint32_t w, uint32_t h, uint32_t f,
+ ENTROPY_CODER significand_bitstream [],
+ ENTROPY_CODER insignificand_bitstream [])
+{
+ uint32_t x, y, z;
+
+ for (z=0; z<f; z++) {
+ for (y=0; y<h; y++) {
+ for (x=0; x<w; x++) {
+ unsigned int index = buf->offset [level] [quadrant]
+ + z * buf->width * buf->height
+ + y * buf->width + x;
+
+ encode_coeff (significand_bitstream, insignificand_bitstream,
+ buf->data [index]);
+ }
+ }
+ }
+}
+
+
+static
+void encode_coefficients (const Wavelet3DBuf* buf,
+ ENTROPY_CODER s_stream [],
+ ENTROPY_CODER i_stream [])
+{
+ int level;
+
+ encode_coeff (s_stream, i_stream, buf->data[0]);
+
+ for (level=0; level<buf->scales-1; level++) {
+ uint32_t w, h, f, w1, h1, f1;
+
+ w = buf->w [level];
+ h = buf->h [level];
+ f = buf->f [level];
+ w1 = buf->w [level+1] - w;
+ h1 = buf->h [level+1] - h;
+ f1 = buf->f [level+1] - f;
+
+ if (w1 > 0) encode_quadrant(buf,level,1,w1,h,f,s_stream,i_stream);
+ if (h1 > 0) encode_quadrant(buf,level,2,w,h1,f,s_stream,i_stream);
+ if (f1 > 0) encode_quadrant(buf,level,3,w,h,f1,s_stream,i_stream);
+ if (w1 > 0 && h1 > 0) encode_quadrant(buf,level,4,w1,h1,f,s_stream,i_stream);
+ if (w1 > 0 && f1 > 0) encode_quadrant(buf,level,5,w1,h,f1,s_stream,i_stream);
+ if (h1 > 0 && f1 > 0) encode_quadrant(buf,level,6,w,h1,f1,s_stream,i_stream);
+ if (h1 > 0 && f1 > 0 && f1 > 0)
+ encode_quadrant (buf,level,7,w1,h1,f1,s_stream,i_stream);
+ }
+}
+
+
+static inline
+void decode_quadrant (Wavelet3DBuf* buf,
+ int level, int quadrant, uint32_t w, uint32_t h, uint32_t f,
+ ENTROPY_CODER s_stream [],
+ ENTROPY_CODER i_stream [])
+{
+ uint32_t x, y, z;
+
+ z = 0;
+ do {
+ y = 0;
+ do {
+ x = 0;
+ do {
+ uint32_t skip;
+ uint32_t index = buf->offset [level] [quadrant]
+ + z * buf->width * buf->height
+ + y * buf->width + x;
+
+ buf->data [index] = decode_coeff (s_stream, i_stream);
+
+ skip = skip_0coeffs (buf, s_stream, i_stream,
+ (w-x-1)+(h-y-1)*w+(f-z-1)*w*h);
+ if (skip > 0) {
+ x += skip;
+ while (x >= w) {
+ y++; x -= w;
+ while (y >= h) {
+ z++; y -= h;
+ if (z >= f)
+ return;
+ }
+ }
+ }
+ x++;
+ } while (x < w);
+ y++;
+ } while (y < h);
+ z++;
+ } while (z < f);
+}
+
+
+static
+void decode_coefficients (Wavelet3DBuf* buf,
+ ENTROPY_CODER s_stream [],
+ ENTROPY_CODER i_stream [])
+{
+ int level;
+
+ buf->data[0] = decode_coeff (s_stream, i_stream);
+
+ for (level=0; level<buf->scales-1; level++) {
+ uint32_t w, h, f, w1, h1, f1;
+
+ w = buf->w [level];
+ h = buf->h [level];
+ f = buf->f [level];
+ w1 = buf->w [level+1] - w;
+ h1 = buf->h [level+1] - h;
+ f1 = buf->f [level+1] - f;
+
+ if (w1 > 0) decode_quadrant(buf,level,1,w1,h,f,s_stream,i_stream);
+ if (h1 > 0) decode_quadrant(buf,level,2,w,h1,f,s_stream,i_stream);
+ if (f1 > 0) decode_quadrant(buf,level,3,w,h,f1,s_stream,i_stream);
+ if (w1 > 0 && h1 > 0) decode_quadrant(buf,level,4,w1,h1,f,s_stream,i_stream);
+ if (w1 > 0 && f1 > 0) decode_quadrant(buf,level,5,w1,h,f1,s_stream,i_stream);
+ if (h1 > 0 && f1 > 0) decode_quadrant(buf,level,6,w,h1,f1,s_stream,i_stream);
+ if (h1 > 0 && f1 > 0 && f1 > 0)
+ decode_quadrant (buf,level,7,w1,h1,f1,s_stream,i_stream);
+ }
+}
+#else
+
+static
+void encode_coefficients (const Wavelet3DBuf* buf,
+ ENTROPY_CODER s_stream [],
+ ENTROPY_CODER i_stream [])
+{
+ uint32_t i;
+
+ for (i=0; i<buf->width*buf->height*buf->frames; i++)
+ encode_coeff(s_stream, i_stream, buf->data[i]);
+}
+
+
+
+
+static
+void decode_coefficients (Wavelet3DBuf* buf,
+ ENTROPY_CODER s_stream [],
+ ENTROPY_CODER i_stream [])
+{
+ uint32_t i;
+
+ for (i=0; i<buf->width*buf->height*buf->frames; i++) {
+ uint32_t skip;
+
+ buf->data[i] = decode_coeff(s_stream, i_stream);
+
+ skip = skip_0coeffs (buf, s_stream, i_stream,
+ buf->width*buf->height*buf->frames - i);
+ i += skip;
+ }
+}
+#endif
+
+
+
+static
+uint32_t setup_limittabs (ENTROPY_CODER significand_bitstream [],
+ ENTROPY_CODER insignificand_bitstream [],
+ uint32_t significand_limittab [],
+ uint32_t insignificand_limittab [],
+ uint32_t limit)
+{
+ uint32_t significand_limit;
+ uint32_t insignificand_limit;
+ uint32_t byte_count;
+ int i;
+
+ assert (limit > 2 * TYPE_BITS * sizeof(uint32_t)); /* limit too small */
+
+ printf ("%s: limit == %u\n", __FUNCTION__, limit);
+ byte_count = 2 * TYPE_BITS * sizeof(uint32_t); /* 2 binary coded limittabs */
+ limit -= byte_count;
+ printf ("%s: rem. limit == %u\n", __FUNCTION__, limit);
+
+ significand_limit = limit * 7 / 8;
+ insignificand_limit = limit - significand_limit;
+
+ printf ("%s: limit == %u\n", __FUNCTION__, limit);
+ printf ("significand limit == %u\n", significand_limit);
+ printf ("insignificand limit == %u\n", insignificand_limit);
+
+ for (i=TYPE_BITS-1; i>=0; i--) {
+ uint32_t s_bytes, i_bytes;
+
+ if (i > 0) {
+ significand_limittab[i] = (significand_limit + 1) / 2;
+ insignificand_limittab[i] = (insignificand_limit + 1) / 2;
+ } else {
+ significand_limittab[0] = significand_limit;
+ insignificand_limittab[0] = insignificand_limit;
+ }
+
+ s_bytes = ENTROPY_ENCODER_FLUSH(&significand_bitstream[i]);
+ i_bytes = ENTROPY_ENCODER_FLUSH(&insignificand_bitstream[i]);
+
+ if (s_bytes < significand_limittab[i])
+ significand_limittab[i] = s_bytes;
+
+ if (i_bytes < insignificand_limittab[i])
+ insignificand_limittab[i] = i_bytes;
+
+ byte_count += significand_limittab[i];
+ byte_count += insignificand_limittab[i];
+
+ printf ("insignificand_limittab[%i] == %u / %u\n",
+ i, insignificand_limittab[i], i_bytes);
+ printf (" significand_limittab[%i] == %u / %u\n",
+ i, significand_limittab[i], s_bytes);
+
+ significand_limit -= significand_limittab[i];
+ insignificand_limit -= insignificand_limittab[i];
+ }
+
+ printf ("byte_count == %u\n", byte_count);
+
+ return byte_count;
+}
+
+
+/**
+ * write 'em binary for now, should be easy to compress ...
+ */
+static
+uint8_t* write_limittabs (uint8_t *bitstream,
+ uint32_t significand_limittab [],
+ uint32_t insignificand_limittab [])
+{
+ int i;
+
+ for (i=0; i<TYPE_BITS; i++) {
+ *(uint32_t*) bitstream = significand_limittab[i];
+ bitstream += 4;
+ }
+
+ for (i=0; i<TYPE_BITS; i++) {
+ *(uint32_t*) bitstream = insignificand_limittab[i];
+ bitstream += 4;
+ }
+
+ return bitstream;
+}
+
+
+static
+uint8_t* read_limittabs (uint8_t *bitstream,
+ uint32_t significand_limittab [],
+ uint32_t insignificand_limittab [])
+{
+ int i;
+
+ for (i=0; i<TYPE_BITS; i++) {
+ significand_limittab[i] = *(uint32_t*) bitstream;
+ printf ("significand_limittab[%i] == %u\n", i, significand_limittab[i]);
+ bitstream += 4;
+ }
+
+ for (i=0; i<TYPE_BITS; i++) {
+ insignificand_limittab[i] = *(uint32_t*) bitstream;
+ printf ("insignificand_limittab[%i] == %u\n", i, insignificand_limittab[i]);
+ bitstream += 4;
+ }
+
+ return bitstream;
+}
+
+
+/**
+ * concatenate entropy coder bitstreams
+ */
+static
+void merge_bitstreams (uint8_t *bitstream,
+ ENTROPY_CODER significand_bitstream [],
+ ENTROPY_CODER insignificand_bitstream [],
+ uint32_t significand_limittab [],
+ uint32_t insignificand_limittab [])
+{
+ int i;
+
+ for (i=TYPE_BITS-1; i>=0; i--) {
+ memcpy (bitstream,
+ ENTROPY_CODER_BITSTREAM(&significand_bitstream[i]),
+ significand_limittab[i]);
+
+ bitstream += significand_limittab[i];
+ }
+
+ for (i=TYPE_BITS-1; i>=0; i--) {
+ memcpy (bitstream,
+ ENTROPY_CODER_BITSTREAM(&insignificand_bitstream[i]),
+ insignificand_limittab[i]);
+
+ bitstream += insignificand_limittab[i];
+ }
+}
+
+
+static
+void split_bitstreams (uint8_t *bitstream,
+ ENTROPY_CODER significand_bitstream [],
+ ENTROPY_CODER insignificand_bitstream [],
+ uint32_t significand_limittab [],
+ uint32_t insignificand_limittab [])
+{
+ uint32_t byte_count;
+ int i;
+
+ for (i=TYPE_BITS-1; i>=0; i--) {
+ byte_count = significand_limittab[i];
+ ENTROPY_DECODER_INIT(&significand_bitstream[i], bitstream, byte_count);
+ bitstream += byte_count;
+ }
+
+ for (i=TYPE_BITS-1; i>=0; i--) {
+ byte_count = insignificand_limittab[i];
+ ENTROPY_DECODER_INIT(&insignificand_bitstream[i], bitstream, byte_count);
+ bitstream += byte_count;
+ }
+}
+
+
+int wavelet_3d_buf_encode_coeff (const Wavelet3DBuf* buf,
+ uint8_t *bitstream,
+ uint32_t limit)
+{
+ ENTROPY_CODER significand_bitstream [TYPE_BITS];
+ ENTROPY_CODER insignificand_bitstream [TYPE_BITS];
+ uint32_t significand_limittab [TYPE_BITS];
+ uint32_t insignificand_limittab [TYPE_BITS];
+ uint32_t byte_count;
+ int i;
+
+ for (i=0; i<TYPE_BITS; i++)
+ ENTROPY_ENCODER_INIT(&significand_bitstream[i], limit);
+ for (i=0; i<TYPE_BITS; i++)
+ ENTROPY_ENCODER_INIT(&insignificand_bitstream[i], limit);
+
+ encode_coefficients (buf, significand_bitstream, insignificand_bitstream);
+
+ byte_count = setup_limittabs (significand_bitstream, insignificand_bitstream,
+ significand_limittab, insignificand_limittab,
+ limit);
+
+ bitstream = write_limittabs (bitstream,
+ significand_limittab, insignificand_limittab);
+
+ merge_bitstreams (bitstream, significand_bitstream, insignificand_bitstream,
+ significand_limittab, insignificand_limittab);
+
+ for (i=0; i<TYPE_BITS; i++) {
+ ENTROPY_ENCODER_DONE(&significand_bitstream[i]);
+ ENTROPY_ENCODER_DONE(&insignificand_bitstream[i]);
+ }
+
+ return byte_count;
+}
+
+
+void wavelet_3d_buf_decode_coeff (Wavelet3DBuf* buf,
+ uint8_t *bitstream,
+ uint32_t byte_count)
+{
+ ENTROPY_CODER significand_bitstream [TYPE_BITS];
+ ENTROPY_CODER insignificand_bitstream [TYPE_BITS];
+ uint32_t significand_limittab [TYPE_BITS];
+ uint32_t insignificand_limittab [TYPE_BITS];
+ int i;
+
+ memset (buf->data, 0,
+ buf->width * buf->height * buf->frames * sizeof(TYPE));
+
+ bitstream = read_limittabs (bitstream,
+ significand_limittab, insignificand_limittab);
+
+ split_bitstreams (bitstream, significand_bitstream, insignificand_bitstream,
+ significand_limittab, insignificand_limittab);
+
+ decode_coefficients (buf, significand_bitstream, insignificand_bitstream);
+
+ for (i=0; i<TYPE_BITS; i++) {
+ ENTROPY_DECODER_DONE(&significand_bitstream[i]);
+ ENTROPY_DECODER_DONE(&insignificand_bitstream[i]);
+ }
+}
+
+
diff --git a/ext/tarkin/wavelet_xform.c b/ext/tarkin/wavelet_xform.c
new file mode 100644
index 00000000..768446a2
--- /dev/null
+++ b/ext/tarkin/wavelet_xform.c
@@ -0,0 +1,391 @@
+#include "mem.h"
+#include <assert.h>
+#include "wavelet.h"
+
+
+
+static
+void fwd_analyze_1 (const TYPE *x, TYPE *d, int stride, int n)
+{
+ int i, k=n/2;
+
+ for (i=0; i<k; i++)
+ d[i] = x[(2*i+1)*stride] - x[2*i*stride];
+}
+
+
+static
+void fwd_synthesize_1 (const TYPE *x, TYPE *s, const TYPE *d, int stride, int n)
+{
+ int i, k=n/2;
+
+ for (i=0; i<k; i++)
+ s[i*stride] = x[2*i*stride] + d[i] / 2;
+ if (n & 1)
+ s[k*stride] = x[2*k*stride] + d[k-1] / 2;
+}
+
+
+static
+void inv_analyze_1 (TYPE *x, const TYPE *d, int stride, int n)
+{
+ int i, k=n/2;
+
+ for (i=0; i<k; i++)
+ x[(2*i+1)*stride] = d[i] + x[2*i*stride];
+}
+
+
+static
+void inv_synthesize_1 (TYPE *x, const TYPE *s, const TYPE *d, int stride, int n)
+{
+ int i, k=n/2;
+
+ for (i=0; i<k; i++)
+ x[2*i*stride] = s[i] - d[i] / 2;
+ if (n & 1)
+ x[2*k*stride] = s[k] - d[k-1] / 2;
+}
+
+
+
+static
+void fwd_analyze_2 (const TYPE *x, TYPE *d, int stride, int n)
+{
+ int i, k=n/2;
+
+ if (n & 1) {
+ for (i=0; i<k; i++)
+ d[i] = x[(2*i+1)*stride] - (x[2*i*stride] + x[(2*i+2)*stride]) / 2;
+ } else {
+ for (i=0; i<k-1; i++)
+ d[i] = x[(2*i+1)*stride] - (x[2*i*stride] + x[(2*i+2)*stride]) / 2;
+ d[k-1] = x[(n-1)*stride] - x[(n-2)*stride];
+ }
+}
+
+
+
+static
+void fwd_synthesize_2 (const TYPE *x, TYPE *s, const TYPE *d, int stride, int n)
+{
+ int i, k=n/2;
+
+ s[0] = x[0] + d[1] / 2;
+ for (i=1; i<k; i++)
+ s[i*stride] = x[2*i*stride] + (d[i-1] + d[i]) / 4;
+ if (n & 1)
+ s[k*stride] = x[2*k*stride] + d[k-1] / 2;
+}
+
+
+static inline
+void inv_analyze_2 (TYPE *x, const TYPE *d, int stride, int n)
+{
+ int i, k=n/2;
+
+ if (n & 1) {
+ for (i=0; i<k; i++)
+ x[(2*i+1)*stride] = d[i] + (x[2*i*stride] + x[(2*i+2)*stride]) / 2;
+ } else {
+ for (i=0; i<k-1; i++)
+ x[(2*i+1)*stride] = d[i] + (x[2*i*stride] + x[(2*i+2)*stride]) / 2;
+ x[(n-1)*stride] = d[k-1] + x[(n-2)*stride];
+ }
+}
+
+
+static inline
+void inv_synthesize_2 (TYPE *x, const TYPE *s, const TYPE *d, int stride, int n)
+{
+ int i, k=n/2;
+
+ x[0] = s[0] - d[1] / 2;
+ for (i=1; i<k; i++)
+ x[2*i*stride] = s[i] - (d[i-1] + d[i]) / 4;
+ if (n & 1)
+ x[2*k*stride] = s[k] - d[k-1] / 2;
+}
+
+
+
+static
+void fwd_analyze_4 (const TYPE *x, TYPE *d, int stride, int n)
+{
+ int i, k=n/2;
+
+ d[0] = x[stride] - (x[0] + x[2*stride]) / 2;
+
+ if (n & 1) {
+ for (i=1; i<k-1; i++)
+ d[i] = x[(2*i+1)*stride]
+ - ((uint32_t) 9 * (x[2*i*stride] + x[(2*i+2)*stride])
+ - (x[(2*i-2)*stride] + x[(2*i+4)*stride])) / 16;
+ if (k > 1)
+ d[k-1] = x[(2*k-1)*stride] - (x[(2*k-2)*stride] + x[2*k*stride]) / 2;
+ } else {
+ for (i=1; i<k-2; i++)
+ d[i] = x[(2*i+1)*stride]
+ - ((uint32_t) 9 * (x[2*i*stride] + x[(2*i+2)*stride])
+ - (x[(2*i-2)*stride] + x[(2*i+4)*stride])) / 16;
+ if (k > 2)
+ d[k-2] = x[(2*k-3)*stride] - (x[(2*k-4)*stride]
+ + x[(2*k-2)*stride]) / 2;
+ if (k > 1)
+ d[k-1] = x[(n-1)*stride] - x[(n-2)*stride];
+ }
+}
+
+
+static
+void fwd_synthesize_4 (const TYPE *x, TYPE *s, const TYPE *d, int stride, int n)
+{
+ int i, k=n/2;
+
+ s[0] = x[0] + d[1] / 2;
+ if (k > 1)
+ s[stride] = x[2*stride] + (d[0] + d[1]) / 4;
+ for (i=2; i<k-1; i++)
+ s[i*stride] = x[2*i*stride]
+ + ((uint32_t) 9 * (d[i-1] + d[i]) - (d[i-2] + d[i+1])) / 32;
+ if (k > 2)
+ s[(k-1)*stride] = x[(2*k-2)*stride] + (d[k-2] + d[k-1]) / 4;
+ if (n & 1)
+ s[k*stride] = x[2*k*stride] + d[k-1] / 2;
+}
+
+
+static
+void inv_analyze_4 (TYPE *x, const TYPE *d, int stride, int n)
+{
+ int i, k=n/2;
+
+ x[stride] = d[0] + (x[0] + x[2*stride]) / 2;
+
+ if (n & 1) {
+ for (i=1; i<k-1; i++)
+ x[(2*i+1)*stride] = d[i]
+ + ((uint32_t) 9 * (x[2*i*stride] + x[(2*i+2)*stride])
+ - (x[(2*i-2)*stride] + x[(2*i+4)*stride])) / 16;
+ if (k > 1)
+ x[(2*k-1)*stride] = d[k-1] + (x[(2*k-2)*stride] + x[2*k*stride]) / 2;
+ } else {
+ for (i=1; i<k-2; i++)
+ x[(2*i+1)*stride] = d[i]
+ + (9 * (x[2*i*stride] + x[(2*i+2)*stride])
+ - (x[(2*i-2)*stride] + x[(2*i+4)*stride])) / 16;
+ if (k > 2)
+ x[(2*k-3)*stride] = d[k-2] + (x[(2*k-4)*stride]
+ + x[(2*k-2)*stride]) / 2;
+ if (k > 1)
+ x[(n-1)*stride] = d[k-1] + x[(n-2)*stride];
+ }
+}
+
+
+static
+void inv_synthesize_4 (TYPE *x, const TYPE *s, const TYPE *d, int stride, int n)
+{
+ int i, k=n/2;
+
+ x[0] = s[0] - d[1] / 2;
+ if (k > 1)
+ x[2*stride] = s[1] - (d[0] + d[1]) / 4;
+ for (i=2; i<k-1; i++)
+ x[2*i*stride] = s[i] - ((uint32_t) 9 * (d[i-1] + d[i])
+ - (d[i-2] + d[i+1])) / 32;
+ if (k > 2)
+ x[(2*k-2)*stride] = s[k-1] - (d[k-2] + d[k-1]) / 4;
+ if (n & 1)
+ x[2*k*stride] = s[k] - d[k-1] / 2;
+}
+
+
+static inline
+void copyback_d (TYPE *x, const TYPE *d, int stride, int n)
+{
+ int i, j, k=n/2;
+
+ for (i=n-k, j=0; i<n; i++, j++)
+ x [i*stride] = d[j];
+}
+
+
+static inline
+void copy_s_d (const TYPE *x, TYPE *s_d, int stride, int n)
+{
+ int i;
+
+ for (i=0; i<n; i++)
+ s_d[i] = x [i*stride];
+}
+
+
+
+typedef
+void (*FwdSFnc) (const TYPE *x, TYPE *s, const TYPE *d, int stride, int n);
+
+typedef
+void (*FwdAFnc) (const TYPE *x, TYPE *d, int stride, int n);
+
+typedef
+void (*InvSFnc) (TYPE *x, const TYPE *s, const TYPE *d, int stride, int n);
+
+typedef
+void (*InvAFnc) (TYPE *x, const TYPE *d, int stride, int n);
+
+
+
+static FwdSFnc fwd_synthesize [] = { NULL, fwd_synthesize_1, fwd_synthesize_2,
+ NULL, fwd_synthesize_4 };
+
+static FwdAFnc fwd_analyze [] = { NULL, fwd_analyze_1, fwd_analyze_2,
+ NULL, fwd_analyze_4 };
+
+static InvSFnc inv_synthesize [] = { NULL, inv_synthesize_1, inv_synthesize_2,
+ NULL, inv_synthesize_4 };
+
+static InvAFnc inv_analyze [] = { NULL, inv_analyze_1, inv_analyze_2,
+ NULL, inv_analyze_4 };
+
+
+
+static inline
+void fwd_xform (TYPE *scratchbuf, TYPE *data, int stride, int n,
+ int a_moments, int s_moments)
+{
+ TYPE *x = data;
+ TYPE *d = scratchbuf;
+ TYPE *s = data;
+
+ assert (a_moments == 1 || a_moments == 2 || a_moments == 4);
+ assert (s_moments == 1 || s_moments == 2 || s_moments == 4);
+
+ /* XXX FIXME: Ugly hack to work around */
+ /* the short-row bug in high */
+ /* order xform functions */
+ if (n < 9)
+ a_moments = s_moments = 2;
+ if (n < 5)
+ a_moments = s_moments = 1;
+
+ fwd_analyze [a_moments] (x, d, stride, n);
+ fwd_synthesize [s_moments] (x, s, d, stride, n);
+ copyback_d (x, d, stride, n);
+}
+
+
+static inline
+void inv_xform (TYPE *scratchbuf, TYPE *data, int stride, int n,
+ int a_moments, int s_moments)
+{
+ int k=n/2;
+ TYPE *x = data;
+ TYPE *s = scratchbuf;
+ TYPE *d = scratchbuf + n - k;
+
+ assert (a_moments == 1 || a_moments == 2 || a_moments == 4);
+ assert (s_moments == 1 || s_moments == 2 || s_moments == 4);
+
+ /* XXX FIXME: Ugly hack to work around */
+ /* the short-row bug in high */
+ /* order xform functions */
+ if (n < 9)
+ a_moments = s_moments = 2;
+ if (n < 5)
+ a_moments = s_moments = 1;
+
+ copy_s_d (data, scratchbuf, stride, n);
+ inv_synthesize [s_moments] (x, s, d, stride, n);
+ inv_analyze [a_moments] (x, d, stride, n);
+}
+
+
+
+void wavelet_3d_buf_fwd_xform (Wavelet3DBuf* buf, int a_moments, int s_moments)
+{
+ int level;
+
+ for (level=buf->scales-1; level>0; level--) {
+ uint32_t w = buf->w[level];
+ uint32_t h = buf->h[level];
+ uint32_t f = buf->f[level];
+
+ if (w > 1) {
+ int row, frame;
+ for (frame=0; frame<f; frame++) {
+ for (row=0; row<h; row++) {
+ TYPE *data = buf->data + (frame * buf->height + row) * buf->width;
+ fwd_xform (buf->scratchbuf, data, 1, w, a_moments, s_moments);
+ }
+ }
+ }
+
+ if (h > 1) {
+ int col, frame;
+ for (frame=0; frame<f; frame++) {
+ for (col=0; col<w; col++) {
+ TYPE *data = buf->data + frame * buf->width * buf->height + col;
+ fwd_xform (buf->scratchbuf, data, buf->width, h,
+ a_moments, s_moments);
+ }
+ }
+ }
+
+ if (f > 1) {
+ int i, j;
+ for (j=0; j<h; j++) {
+ for (i=0; i<w; i++) {
+ TYPE *data = buf->data + j*buf->width + i;
+ fwd_xform (buf->scratchbuf, data, buf->width * buf->height, f,
+ a_moments, s_moments);
+ }
+ }
+ }
+ }
+}
+
+
+void wavelet_3d_buf_inv_xform (Wavelet3DBuf* buf, int a_moments, int s_moments)
+{
+ int level;
+
+ for (level=1; level<buf->scales; level++) {
+ uint32_t w = buf->w[level];
+ uint32_t h = buf->h[level];
+ uint32_t f = buf->f[level];
+
+ if (f > 1) {
+ int i, j;
+ for (j=0; j<h; j++) {
+ for (i=0; i<w; i++) {
+ TYPE *data = buf->data + j*buf->width + i;
+ inv_xform (buf->scratchbuf, data, buf->width * buf->height, f,
+ a_moments, s_moments);
+ }
+ }
+ }
+
+ if (h > 1) {
+ int col, frame;
+ for (frame=0; frame<f; frame++) {
+ for (col=0; col<w; col++) {
+ TYPE *data = buf->data + frame * buf->width * buf->height + col;
+ inv_xform (buf->scratchbuf, data, buf->width, h,
+ a_moments, s_moments);
+ }
+ }
+ }
+
+ if (w > 1) {
+ int row, frame;
+ for (frame=0; frame<f; frame++) {
+ for (row=0; row<h; row++) {
+ TYPE *data = buf->data + (frame * buf->height + row) * buf->width;
+ inv_xform (buf->scratchbuf, data, 1, w, a_moments, s_moments);
+ }
+ }
+ }
+ }
+}
+
diff --git a/ext/tarkin/yuv.c b/ext/tarkin/yuv.c
new file mode 100644
index 00000000..6f5fb5c0
--- /dev/null
+++ b/ext/tarkin/yuv.c
@@ -0,0 +1,224 @@
+#include "yuv.h"
+
+/*#define TARKIN_YUV_EXACT*/
+/*#define TARKIN_YUV_LXY*/
+
+
+static inline
+uint8_t CLAMP(int16_t x)
+{
+ return ((x > 255) ? 255 : (x < 0) ? 0 : x);
+}
+
+
+
+void rgb24_to_yuv (uint8_t *rgb, Wavelet3DBuf *yuv [], uint32_t frame)
+{
+ int count = yuv[0]->width * yuv[0]->height;
+ int16_t *y = yuv[0]->data + frame * count;
+ int16_t *u = yuv[1]->data + frame * count;
+ int16_t *v = yuv[2]->data + frame * count;
+ int i;
+
+
+#if defined(TARKIN_YUV_EXACT)
+ for (i=0; i<count; i++, rgb+=3) {
+ y [i] = ((int16_t) 77 * rgb [0] + 150 * rgb [1] + 29 * rgb [2]) / 256;
+ u [i] = ((int16_t) -44 * rgb [0] - 87 * rgb [1] + 131 * rgb [2]) / 256;
+ v [i] = ((int16_t) 131 * rgb [0] - 110 * rgb [1] - 21 * rgb [2]) / 256;
+ }
+#elif defined(TARKIN_YUV_LXY)
+ for (i=0; i<count; i++, rgb+=3) {
+ y [i] = ((int16_t) 54 * rgb [0] + 182 * rgb [1] + 18 * rgb [2]) / 256;
+ u [i] = rgb [0] - y [i];
+ v [i] = rgb [2] - y [i];
+ }
+#else
+ for (i=0; i<count; i++, rgb+=3) {
+ v [i] = rgb [0] - rgb [1];
+ u [i] = rgb [2] - rgb [1];
+ y [i] = rgb [1] + (u [i] + v [i]) / 4;
+ }
+#endif
+}
+
+
+void yuv_to_rgb24 (Wavelet3DBuf *yuv [], uint8_t *rgb, uint32_t frame)
+{
+ int count = yuv[0]->width * yuv[0]->height;
+ int16_t *y = yuv[0]->data + frame * count;
+ int16_t *u = yuv[1]->data + frame * count;
+ int16_t *v = yuv[2]->data + frame * count;
+ int i;
+
+#if defined(TARKIN_YUV_EXACT)
+ for (i=0; i<count; i++, rgb+=3) {
+ rgb [0] = CLAMP(y [i] + 1.371 * v [i]);
+ rgb [1] = CLAMP(y [i] - 0.698 * v [i] - 0.336 * u [i]);
+ rgb [2] = CLAMP(y [i] + 1.732 * u [i]);
+ }
+#elif defined(TARKIN_YUV_LXY)
+ for (i=0; i<count; i++, rgb+=3) {
+ rgb [1] = CLAMP(y [i] - (76 * u [i] - 26 * v [i]) / 256);
+ rgb [0] = CLAMP(y [i] + u [i]);
+ rgb [2] = CLAMP(y [i] + v [i]);
+ }
+#else
+ for (i=0; i<count; i++, rgb+=3) {
+ rgb [1] = CLAMP(y [i] - (u [i] + v [i]) / 4);
+ rgb [2] = CLAMP(u [i] + rgb [1]);
+ rgb [0] = CLAMP(v [i] + rgb [1]);
+ }
+#endif
+}
+
+
+void rgb32_to_yuv (uint8_t *rgb, Wavelet3DBuf *yuv [], uint32_t frame)
+{
+ int count = yuv[0]->width * yuv[0]->height;
+ int16_t *y = yuv[0]->data + frame * count;
+ int16_t *u = yuv[1]->data + frame * count;
+ int16_t *v = yuv[2]->data + frame * count;
+ int i;
+
+#if defined(TARKIN_YUV_EXACT)
+ for (i=0; i<count; i++, rgb+=4) {
+ y [i] = ((int16_t) 77 * rgb [0] + 150 * rgb [1] + 29 * rgb [2]) / 256;
+ u [i] = ((int16_t) -44 * rgb [0] - 87 * rgb [1] + 131 * rgb [2]) / 256;
+ v [i] = ((int16_t) 131 * rgb [0] - 110 * rgb [1] - 21 * rgb [2]) / 256;
+ }
+#elif defined(TARKIN_YUV_LXY)
+ for (i=0; i<count; i++, rgb+=4) {
+ y [i] = ((int16_t) 54 * rgb [0] + 182 * rgb [1] + 18 * rgb [2]) / 256;
+ u [i] = rgb [0] - y [i];
+ v [i] = rgb [2] - y [i];
+ }
+#else
+ for (i=0; i<count; i++, rgb+=4) {
+ v [i] = rgb [0] - rgb [1];
+ u [i] = rgb [2] - rgb [1];
+ y [i] = rgb [1] + (u [i] + v [i]) / 4;
+ }
+#endif
+}
+
+
+void yuv_to_rgb32 (Wavelet3DBuf *yuv [], uint8_t *rgb, uint32_t frame)
+{
+ int count = yuv[0]->width * yuv[0]->height;
+ int16_t *y = yuv[0]->data + frame * count;
+ int16_t *u = yuv[1]->data + frame * count;
+ int16_t *v = yuv[2]->data + frame * count;
+ int i;
+
+#if defined(TARKIN_YUV_EXACT)
+ for (i=0; i<count; i++, rgb+=4) {
+ rgb [0] = CLAMP(y [i] + 1.371 * v [i]);
+ rgb [1] = CLAMP(y [i] - 0.698 * v [i] - 0.336 * u [i]);
+ rgb [2] = CLAMP(y [i] + 1.732 * u [i]);
+ }
+#elif defined(TARKIN_YUV_LXY)
+ for (i=0; i<count; i++, rgb+=4) {
+ rgb [1] = CLAMP(y [i] - (76 * u [i] - 26 * v [i]) / 256);
+ rgb [0] = CLAMP(y [i] + u [i]);
+ rgb [2] = CLAMP(y [i] + v [i]);
+ }
+#else
+ for (i=0; i<count; i++, rgb+=4) {
+ rgb [1] = CLAMP(y [i] - (u [i] + v [i]) / 4);
+ rgb [2] = CLAMP(u [i] + rgb [1]);
+ rgb [0] = CLAMP(v [i] + rgb [1]);
+ }
+#endif
+}
+
+
+void rgba_to_yuv (uint8_t *rgba, Wavelet3DBuf *yuva [], uint32_t frame)
+{
+ int count = yuva[0]->width * yuva[0]->height;
+ int16_t *y = yuva[0]->data + frame * count;
+ int16_t *u = yuva[1]->data + frame * count;
+ int16_t *v = yuva[2]->data + frame * count;
+ int16_t *a = yuva[3]->data + frame * count;
+ int i;
+
+#if defined(TARKIN_YUV_EXACT)
+ for (i=0; i<count; i++, rgba+=4) {
+ y [i] = ((int16_t) 77 * rgba [0] + 150 * rgba [1] + 29 * rgba [2]) / 256;
+ u [i] = ((int16_t) -44 * rgba [0] - 87 * rgba [1] + 131 * rgba [2]) / 256;
+ v [i] = ((int16_t) 131 * rgba [0] - 110 * rgba [1] - 21 * rgba [2]) / 256;
+ a [i] = rgba [3];
+ }
+#elif defined(TARKIN_YUV_LXY)
+ for (i=0; i<count; i++, rgba+=4) {
+ y [i] = ((int16_t) 54 * rgba [0] + 182 * rgba [1] + 18 * rgba [2]) / 256;
+ u [i] = rgba [0] - y [i];
+ v [i] = rgba [2] - y [i];
+ a [i] = rgba [3];
+ }
+#else
+ for (i=0; i<count; i++, rgba+=4) {
+ v [i] = rgba [0] - rgba [1];
+ u [i] = rgba [2] - rgba [1];
+ y [i] = rgba [1] + (u [i] + v [i]) / 4;
+ a [i] = rgba [3];
+ }
+#endif
+}
+
+
+void yuv_to_rgba (Wavelet3DBuf *yuva [], uint8_t *rgba, uint32_t frame)
+{
+ int count = yuva[0]->width * yuva[0]->height;
+ int16_t *y = yuva[0]->data + frame * count;
+ int16_t *u = yuva[1]->data + frame * count;
+ int16_t *v = yuva[2]->data + frame * count;
+ int16_t *a = yuva[3]->data + frame * count;
+ int i;
+
+#if defined(TARKIN_YUV_EXACT)
+ for (i=0; i<count; i++, rgba+=4) {
+ rgba [0] = CLAMP(y [i] + 1.371 * v [i]);
+ rgba [1] = CLAMP(y [i] - 0.698 * v [i] - 0.336 * u [i]);
+ rgba [2] = CLAMP(y [i] + 1.732 * u [i]);
+ rgba [3] = a [i];
+ }
+#elif defined(TARKIN_YUV_LXY)
+ for (i=0; i<count; i++, rgba+=4) {
+ rgba [1] = CLAMP(y [i] - (76 * u [i] - 26 * v [i]) / 256);
+ rgba [0] = CLAMP(y [i] + u [i]);
+ rgba [2] = CLAMP(y [i] + v [i]);
+ rgba [3] = a [i];
+ }
+#else
+ for (i=0; i<count; i++, rgba+=4) {
+ rgba [1] = CLAMP(y [i] - (u [i] + v [i]) / 4);
+ rgba [2] = CLAMP(u [i] + rgba [1]);
+ rgba [0] = CLAMP(v [i] + rgba [1]);
+ rgba [3] = a [i];
+ }
+#endif
+}
+
+void grayscale_to_y (uint8_t *rgba, Wavelet3DBuf *y [], uint32_t frame)
+{
+ int count = y[0]->width * y[0]->height;
+ int16_t *_y = y[0]->data + frame * count;
+ int i;
+
+ for (i=0; i<count; i++)
+ _y [i] = rgba [i];
+}
+
+
+void y_to_grayscale (Wavelet3DBuf *y [], uint8_t *rgba, uint32_t frame)
+{
+ int count = y[0]->width * y[0]->height;
+ int16_t *_y = y[0]->data + frame * count;
+ int i;
+
+ for (i=0; i<count; i++)
+ rgba [i] = CLAMP(_y[i]);
+}
+
+
diff --git a/ext/tarkin/yuv.h b/ext/tarkin/yuv.h
new file mode 100644
index 00000000..42ceb072
--- /dev/null
+++ b/ext/tarkin/yuv.h
@@ -0,0 +1,21 @@
+
+#ifndef __YUV_H
+#define __YUV_H
+
+#include <stdint.h>
+#include "wavelet.h"
+
+extern void rgb24_to_yuv (uint8_t *rgb, Wavelet3DBuf *yuv [], uint32_t frame);
+extern void yuv_to_rgb24 (Wavelet3DBuf *yuv [], uint8_t *rgb, uint32_t frame);
+
+extern void rgb32_to_yuv (uint8_t *rgb, Wavelet3DBuf *yuv [], uint32_t frame);
+extern void yuv_to_rgb32 (Wavelet3DBuf *yuv [], uint8_t *rgb, uint32_t frame);
+
+extern void rgba_to_yuv (uint8_t *rgba, Wavelet3DBuf *yuva [], uint32_t frame);
+extern void yuv_to_rgba (Wavelet3DBuf *yuva [], uint8_t *rgba, uint32_t frame);
+
+extern void grayscale_to_y (uint8_t *rgba, Wavelet3DBuf *y [], uint32_t frame);
+extern void y_to_grayscale (Wavelet3DBuf *y [], uint8_t *rgba, uint32_t frame);
+
+#endif
+