ogg-python/src/pyoggstreamstate.c

   1 #include "_oggmodule.h"
   2 #include "pyoggstreamstate.h"
   3 #include "pyoggpacket.h"
   4 #include "pyoggpage.h"
   5 #include "general.h"
   6
   7 /*****************************************************************
   8                             OggStreamState Object
   9  *****************************************************************/
  10
  11
  12 char py_ogg_stream_state_doc[] = "";
  13
  14 static void py_ogg_stream_state_dealloc(PyObject *);
  15 static PyObject* py_ogg_stream_state_getattr(PyObject *, char *);
  16 static PyObject *py_ogg_stream_repr(PyObject *self);
  17
  18 FDEF(ogg_stream_packetin) "Add a packet to the stream.";
  19 FDEF(ogg_stream_clear) "Clear the contents of the stream state.";
  20 FDEF(ogg_stream_flush) "Produce an ogg page suitable for writing to output.";
  21 FDEF(ogg_stream_eos) "Return whether the end of the stream is reached.";
  22 FDEF(ogg_stream_pageout) "Extract and return an OggPage.";
  23 FDEF(ogg_stream_reset) "Reset the stream state";
  24 FDEF(ogg_stream_pagein) "Write a page to the stream";
  25 FDEF(ogg_stream_packetout) "Extract a packet from the stream";
  26
  27 PyTypeObject py_ogg_stream_state_type = {
  28   PyObject_HEAD_INIT(NULL)
  29   0,
  30   "OggStreamState",
  31   sizeof(py_ogg_stream_state),
  32   0,
  33
  34   /* Standard Methods */
  35   /* (destructor) */ py_ogg_stream_state_dealloc,
  36   /* (printfunc) */ 0,
  37   /* (getattrfunc) */ py_ogg_stream_state_getattr,
  38   /* (setattrfunc) */ 0,
  39   /* (cmpfunc) */ 0,
  40   /* (reprfunc) */ py_ogg_stream_repr,
  41
  42   /* Type Categories */
  43   0, /* as number */
  44   0, /* as sequence */
  45   0, /* as mapping */
  46   0, /* hash */
  47   0, /* binary */
  48   0, /* repr */
  49   0, /* getattro */
  50   0, /* setattro */
  51   0, /* as buffer */
  52   0, /* tp_flags */
  53   py_ogg_stream_state_doc
  54 };
  55
  56 static PyMethodDef py_ogg_stream_state_methods[] = {
  57   {"clear", py_ogg_stream_clear,
  58    METH_VARARGS, py_ogg_stream_clear_doc},
  59   {"flush", py_ogg_stream_flush,
  60    METH_VARARGS, py_ogg_stream_flush_doc},
  61   {"eos", py_ogg_stream_eos,
  62    METH_VARARGS, py_ogg_stream_eos_doc},
  63   {"packetin", py_ogg_stream_packetin,
  64    METH_VARARGS, py_ogg_stream_packetin_doc},
  65   {"pageout", py_ogg_stream_pageout,
  66    METH_VARARGS, py_ogg_stream_pageout_doc},
  67   {"pagein", py_ogg_stream_pagein,
  68    METH_VARARGS, py_ogg_stream_pagein_doc},
  69   {"packetout", py_ogg_stream_packetout,
  70    METH_VARARGS, py_ogg_stream_packetout_doc},
  71   {"reset", py_ogg_stream_reset,
  72    METH_VARARGS, py_ogg_stream_reset_doc},
  73   {NULL, NULL}
  74 };
  75
  76 static void
  77 py_ogg_stream_state_dealloc(PyObject *self)
  78 {
  79   ogg_stream_clear(PY_OGG_STREAM(self));
  80   PyMem_DEL(self);
  81 }
  82
  83 static PyObject*
  84 py_ogg_stream_state_getattr(PyObject *self, char *name)
  85 {
  86   return Py_FindMethod(py_ogg_stream_state_methods, self, name);
  87 }
  88
  89 PyObject *
  90 py_ogg_stream_state_from_serialno(int serialno)
  91 {
  92   py_ogg_stream_state *ret = PyObject_NEW(py_ogg_stream_state,
  93                                           &py_ogg_stream_state_type);
  94
  95   if (ret == NULL)
  96     return NULL;
  97
  98   ogg_stream_init(&ret->os, serialno);
  99   return (PyObject *) ret;
 100 }
 101
 102 PyObject *
 103 py_ogg_stream_state_new(PyObject *self, PyObject *args)
 104 {
 105   int serialno;
 106   if (!PyArg_ParseTuple(args, "i", &serialno))
 107     return NULL;
 108   return py_ogg_stream_state_from_serialno(serialno);
 109 }
 110
 111 static PyObject *
 112 py_ogg_stream_clear(PyObject *self, PyObject *args)
 113 {
 114   if (!PyArg_ParseTuple(args, ""))
 115     return NULL;
 116   ogg_stream_clear(PY_OGG_STREAM(self));
 117   Py_INCREF(Py_None);
 118   return Py_None;
 119 }
 120
 121 static PyObject *
 122 py_ogg_stream_packetin(PyObject *self, PyObject *args)
 123 {
 124   py_ogg_packet *packetobj;
 125
 126   if (!PyArg_ParseTuple(args, "O!", &py_ogg_packet_type,
 127                         (PyObject *) &packetobj))
 128     return NULL;
 129
 130   if (ogg_stream_packetin(PY_OGG_STREAM(self), &packetobj->op)) {
 131     /* currently this can't happen */
 132     PyErr_SetString(Py_OggError, "error in ogg_stream_packetin");
 133     return NULL;
 134   }
 135
 136   Py_INCREF(Py_None);
 137   return Py_None;
 138 }
 139
 140 static PyObject *
 141 py_ogg_stream_flush(PyObject *self, PyObject *args)
 142 {
 143   ogg_page op;
 144   int res;
 145
 146   if (!PyArg_ParseTuple(args, ""))
 147     return NULL;
 148   res = ogg_stream_flush(PY_OGG_STREAM(self), &op);
 149   if (res == 0) {
 150     Py_INCREF(Py_None);
 151     return Py_None;
 152   }
 153   return py_ogg_page_from_page(&op);
 154 }
 155
 156 /* TODO: You can't keep a page between calls to flush or pageout!! */
 157
 158 static PyObject *
 159 py_ogg_stream_pageout(PyObject *self, PyObject *args)
 160 {
 161   ogg_page op;
 162   int res;
 163
 164   if (!PyArg_ParseTuple(args, ""))
 165     return NULL;
 166   res = ogg_stream_pageout(PY_OGG_STREAM(self), &op);
 167   if (res == 0) {
 168     Py_INCREF(Py_None);
 169     return Py_None;
 170   }
 171   return py_ogg_page_from_page(&op);
 172 }
 173
 174 static PyObject*
 175 py_ogg_stream_eos(PyObject *self, PyObject *args)
 176 {
 177   int eos = ogg_stream_eos(PY_OGG_STREAM(self));
 178   if (!PyArg_ParseTuple(args, ""))
 179     return NULL;
 180   return PyInt_FromLong(eos);
 181 }
 182
 183 static PyObject *
 184 py_ogg_stream_reset(PyObject *self, PyObject *args)
 185 {
 186   if (!PyArg_ParseTuple(args, ""))
 187     return NULL;
 188   if (ogg_stream_reset(PY_OGG_STREAM(self))) {
 189     PyErr_SetString(Py_OggError, "Error resetting stream");
 190     return NULL;
 191   } else {
 192     Py_INCREF(Py_None);
 193     return Py_None;
 194   }
 195 }
 196
 197 static PyObject *
 198 py_ogg_stream_pagein(PyObject *self, PyObject *args)
 199 {
 200   py_ogg_page *pageobj;
 201
 202   if (!PyArg_ParseTuple(args, "O!", &py_ogg_page_type,
 203                         (PyObject *) &pageobj))
 204     return NULL;
 205
 206   if (ogg_stream_pagein(PY_OGG_STREAM(self), &pageobj->op)) {
 207     PyErr_SetString(Py_OggError, "error in ogg_stream_pagein (bad page?)");
 208     return NULL;
 209   }
 210
 211   Py_INCREF(Py_None);
 212   return Py_None;
 213 }
 214
 215 static PyObject *
 216 py_ogg_stream_packetout(PyObject *self, PyObject *args)
 217 {
 218   ogg_packet op;
 219   int res;
 220
 221   if (!PyArg_ParseTuple(args, ""))
 222     return NULL;
 223   res = ogg_stream_packetout(PY_OGG_STREAM(self), &op);
 224   if (res == 0) {
 225     Py_INCREF(Py_None);
 226     return Py_None;
 227   } else if (res == -1) {
 228     PyErr_SetString(Py_OggError, "lost sync");
 229     return NULL;
 230   }
 231   return py_ogg_packet_from_packet(&op);
 232 }
 233
 234 static PyObject *
 235 py_ogg_stream_repr(PyObject *self)
 236 {
 237   ogg_stream_state *os = PY_OGG_STREAM(self);
 238   char buf[256];
 239   char *bos = os->b_o_s ? "BOS " : "";
 240   char *eos = os->e_o_s ? "EOS " : "";
 241
 242   sprintf(buf, "<OggStreamState, %s%spageno = %ld, packetno = %lld,"
 243           " granulepos = %lld, serialno = %d, at %p>",
 244           bos, eos, os->pageno, os->packetno, os->granulepos,
 245           os->serialno, self);
 246   return PyString_FromString(buf);
 247 }