diff --git a/gst/arg-types.py b/gst/arg-types.py index dfba16950f..aee1abb0c3 100644 --- a/gst/arg-types.py +++ b/gst/arg-types.py @@ -335,6 +335,40 @@ class StringArrayArg(ArgType): " }\n" " return PyTuple_New (0);\n") +class Uint8PArg(ArgType): + """guint8* arguments (strings)""" + def write_param(self, ptype, pname, pdflt, pnull, keeprefcount, info): + if pdflt: + if pdflt != 'NULL': pdflt = '"' + pdflt + '"' + info.varlist.add('guint8', '*' + pname + ' = ' + pdflt) + else: + info.varlist.add('guint8', '*' + pname) + info.arglist.append(pname) + if pnull: + info.add_parselist('z', ['&' + pname], [pname]) + else: + info.add_parselist('s', ['&' + pname], [pname]) + + def write_return(self, ptype, ownsreturn, info): + if ownsreturn: + # have to free result ... + info.varlist.add('guint8', '*ret') + info.codeafter.append(' if (ret) {\n' + + ' PyObject *py_ret = PyString_FromString((gchar*) ret);\n' + + ' g_free(ret);\n' + + ' return py_ret;\n' + + ' }\n' + + ' Py_INCREF(Py_None);\n' + + ' return Py_None;') + else: + info.varlist.add('const guint8', '*ret') + info.codeafter.append(' if (ret)\n' + + ' return PyString_FromString((gchar*) ret);\n'+ + ' Py_INCREF(Py_None);\n' + + ' return Py_None;') + + + matcher.register('GstClockTime', UInt64Arg()) matcher.register('GstClockTimeDiff', Int64Arg()) matcher.register('xmlNodePtr', XmlNodeArg()) @@ -343,6 +377,8 @@ matcher.register('GstCaps', GstCapsArg()) #FIXME: does this work? matcher.register('GstCaps*', GstCapsArg()) #FIXME: does this work? matcher.register('const-GstCaps*', GstCapsArg()) matcher.register('GstIterator*', GstIteratorArg()) +matcher.register('guint8*', Uint8PArg()) +matcher.register('const-guint8*', Uint8PArg()) arg = PointerArg('gpointer', 'G_TYPE_POINTER') matcher.register('GstClockID', arg) diff --git a/testsuite/Makefile.am b/testsuite/Makefile.am index 831f5d2528..35298a03b8 100644 --- a/testsuite/Makefile.am +++ b/testsuite/Makefile.am @@ -19,6 +19,7 @@ testhelper.la: $(testhelper_la_OBJECTS) $(testhelper_la_DEPENDENCIES) $(LINK) -rpath $(pkgpyexecdir) $(testhelper_la_LDFLAGS) $(testhelper_la_OBJECTS) $(testhelper_la_LIBADD) $(LIBS) tests = \ + test_adapter.py \ test_bin.py \ test_buffer.py \ test_caps.py \ diff --git a/testsuite/test_adapter.py b/testsuite/test_adapter.py new file mode 100644 index 0000000000..eec78b2fa2 --- /dev/null +++ b/testsuite/test_adapter.py @@ -0,0 +1,83 @@ +# -*- Mode: Python -*- +# vi:si:et:sw=4:sts=4:ts=4 +# +# gst-python - Python bindings for GStreamer +# Copyright (C) 2009 Edward Hervey +# +# This library is free software; you can redistribute it and/or +# modify it under the terms of the GNU Lesser General Public +# License as published by the Free Software Foundation; either +# version 2.1 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 +# Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser 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 + +from common import gobject, gst, unittest, TestCase + +class AdapterTest(TestCase): + + def setUp(self): + TestCase.setUp(self) + self.adapter = gst.Adapter() + + def tearDown(self): + self.adapter = None + TestCase.tearDown(self) + + def testAvailable(self): + # starts empty + self.assertEquals(self.adapter.available(), 0) + self.assertEquals(self.adapter.available_fast(), 0) + + # let's give it 4 bytes + self.adapter.push(gst.Buffer("1234")) + self.assertEquals(self.adapter.available_fast(), 4) + + # let's give it another 5 bytes + self.adapter.push(gst.Buffer("56789")) + # we now have 9 bytes + self.assertEquals(self.adapter.available(), 9) + # but can only do a fast take of 4 bytes (the first buffer) + self.assertEquals(self.adapter.available_fast(), 4) + + def testPeek(self): + self.adapter.push(gst.Buffer("0123456789")) + + # let's peek at 5 bytes + b = self.adapter.peek(5) + # it can return more than 5 bytes + self.assert_(len(b) >= 5) + self.assertEquals(b[:5], "01234") + + # it's still 10 bytes big + self.assertEquals(self.adapter.available(), 10) + + # if we try to peek more than what's available, we'll have None + self.assertEquals(self.adapter.peek(11), None) + + def testFlush(self): + self.adapter.push(gst.Buffer("0123456789")) + self.assertEquals(self.adapter.available(), 10) + + self.adapter.flush(5) + self.assertEquals(self.adapter.available(), 5) + + # it flushed the first 5 bytes + self.assertEquals(self.adapter.peek(5)[:5], "56789") + + self.adapter.flush(5) + self.assertEquals(self.adapter.available(), 0) + + def testTake(self): + self.adapter.push(gst.Buffer("0123456789")) + self.assertEquals(self.adapter.available(), 10) + + s = self.adapter.take(5) + self.assertEquals(s[:5], "01234") + self.assertEquals(self.adapter.available(), 5)