#include "general.h"
#include "module.h"
#include "page.h"

/*****************************************************************
			    OggPage Object
 *****************************************************************/


char PyOggPage_Doc[] = "This is an Ogg Page.";

static void PyOggPage_Dealloc(PyObject *);
static PyObject* PyOggPage_Getattr(PyObject *, char *);
static int PyOggPage_Setattr(PyObject *self, char *name, PyObject *value);
static PyObject *PyOggPage_Repr(PyObject *self);
static int PyOggPage_Size(PyObject *self);

PySequenceMethods PyOggPage_SeqMethods = { 
  PyOggPage_Size,	/* (length) */
  0,			/* (concat) */
  0,			/* (repeat) */
  0,			/* (item) */
  0,			/* (slice) */
  0,			/* (ass_item) */
  0,			/* (ass_slice) */
  0,			/* (contains) */
  0,			/* (inplace_concat) */
  0,			/* (inplace_repeat) */ 
};

PyTypeObject PyOggPage_Type = {
  PyObject_HEAD_INIT(NULL)
  0,
  "OggPage",
  sizeof(PyOggPageObject),
  0,
  
  /* Standard Methods */
  PyOggPage_Dealloc,	/* (destructor) */
  0,			/* (printfunc) */ 
  PyOggPage_Getattr,	/* (getattrfunc) */
  PyOggPage_Setattr, 	/* (setattrfunc) */
  0, 			/* (cmpfunc) */ 
  PyOggPage_Repr,	/* (reprfunc) */ 
  
  /* Type Categories */
  0, 			/* as number */ 
  &PyOggPage_SeqMethods,/* as sequence */
  0, 			/* as mapping */ 
  0, 			/* hash */
  0, 			/* binary */
  0, 			/* repr */
  0, 			/* getattro */
  0, 			/* setattro */
  0, 			/* as buffer */
  0,			/* tp_flags */
  PyOggPage_Doc
};

static PyMethodDef PyOggPage_methods[] = {
  {"bos", NULL},
  {"continued", NULL},
  {"eos", NULL},
  {"granulepos", NULL},
  {"packets", NULL},
  {"pageno", NULL},
  {"serialno", NULL},
  {"version", NULL},
  {NULL, NULL}
};

PyOggPageObject *
PyOggPage_Alloc() {
  ogg2_page *page;
  PyOggPageObject *ret;

  ret = (PyOggPageObject *) PyObject_NEW(PyOggPageObject, &PyOggPage_Type);
  if (ret == NULL)
    return NULL;

  ret->valid_flag = 1;
  page = PyMem_New(ogg2_page, 1);
  if (page == NULL) {
    PyObject_Del(ret);
    return NULL;
  }
  memset(page, 0, sizeof(*page));

  ret->page = page;
  return ret;
}

static void 
PyOggPage_Dealloc(PyObject *self) {
  ogg2_page_release(PyOggPage_AsOggPage(self));
  PyMem_Del(PyOggPage_AsOggPage(self));
  PyObject_Del(self);
}


static int
PyOggPage_Size(PyObject *self) {

  if (((PyOggPageObject *) self)->valid_flag == 0) {
    PyErr_SetString(PyOggPage_Error, "this page is no longer usable.");
    return -1;
  } 

  return PyOggPage_AsOggPage(self)->header_len +
         PyOggPage_AsOggPage(self)->body_len;
}

static PyObject* 
PyOggPage_Getattr(PyObject *self, char *name) {
  ogg2_page *page;

  if (((PyOggPageObject *) self)->valid_flag == 0) {
    PyErr_SetString(PyOggPage_Error, "this page is no longer usable.");
    return NULL;
  }

  page = PyOggPage_AsOggPage(self);

  if (strcmp(name, "bos") == 0) 
    return Py_TrueFalse(ogg2_page_bos(page));
  if (strcmp(name, "continued") == 0) 
    return Py_TrueFalse(ogg2_page_continued(page));
  if (strcmp(name, "eos") == 0) 
    return Py_TrueFalse(ogg2_page_eos(page));
  if (strcmp(name, "granulepos") == 0)
    return PyLong_FromLongLong(ogg2_page_granulepos(page));
  if (strcmp(name, "packets") == 0) 
    return PyInt_FromLong(ogg2_page_packets(page));
  if (strcmp(name, "pageno") == 0) 
    return PyInt_FromLong(ogg2_page_pageno(page));
  if (strcmp(name, "serialno") == 0) 
    return PyInt_FromLong(ogg2_page_serialno(page));
  if (strcmp(name, "version") == 0) 
    return PyInt_FromLong(ogg2_page_version(page));
  return Py_FindMethod(PyOggPage_methods, self, name);
}

static int
PyOggPage_Setattr(PyObject *self, char *name, PyObject *value)
{
  if (((PyOggPageObject *) self)->valid_flag == 0) {
    PyErr_SetString(PyOggPage_Error, "this page is no longer usable.");
    return -1;
  }
  if (strcmp(name, "bos") == 0) {
    int v;
    if (!arg_to_int32(value, &v)) {
      PyErr_SetString(PyExc_ValueError, "value must be type INT");  
      return -1;
    }
    ogg2_page_set_bos(PyOggPage_AsOggPage(self), v);
    return 0;
  }
  if (strcmp(name, "continued") == 0) {
    int v;
    if (!arg_to_int32(value, &v)) {
      PyErr_SetString(PyExc_ValueError, "value must be type INT");  
      return -1;
    }
    ogg2_page_set_continued(PyOggPage_AsOggPage(self), v);
    return 0;
  }
  if (strcmp(name, "eos") == 0) {
    int v;
    if (!arg_to_int32(value, &v)) {
      PyErr_SetString(PyExc_ValueError, "value must be type INT");  
      return -1;
    }
    ogg2_page_set_eos(PyOggPage_AsOggPage(self), v);
    return 0;
  }
  if (strcmp(name, "granulepos") == 0) {
    ogg_int64_t v;
    if (!arg_to_int64(value, &v)) {
      PyErr_SetString(PyExc_ValueError, "value must be type INT");  
      return -1;
    }
    ogg2_page_set_granulepos(PyOggPage_AsOggPage(self), v);
    return 0;
  }
  if (strcmp(name, "pageno") == 0) {
    ogg_uint32_t v;
    if (!arg_to_int32(value, &v)) {
      PyErr_SetString(PyExc_ValueError, "value must be type INT");  
      return -1;
    }
    ogg2_page_set_pageno(PyOggPage_AsOggPage(self), v);
    return 0;
  }
  if (strcmp(name, "serialno") == 0) {
    ogg_uint32_t v;
    if (!arg_to_int32(value, &v)) {
      PyErr_SetString(PyExc_ValueError, "value must be type INT");  
      return -1;
    }
    ogg2_page_set_serialno(PyOggPage_AsOggPage(self), v);
    return 0;
  }
  if (strcmp(name, "version") == 0) {
    PyErr_SetString(PyExc_ValueError, "version is read-only");  
    return -1;
  }
  if (strcmp(name, "packets") == 0) {
    PyErr_SetString(PyExc_ValueError, "packets is read-only, use OggStreamState");  
    return -1;
  }
  PyErr_SetString(PyExc_AttributeError, "OggPage object has no such attribute");  
  return -1;
}

static PyObject *
PyOggPage_Repr(PyObject *self)
{
  char buf[256];
  char *bos;
  char *eos;
  char *cont;

  if (((PyOggPageObject *) self)->valid_flag == 0) {
    sprintf(buf, "<OggPage that has been passed back to libogg2>");
    return PyString_FromString(buf);
  }

  bos = ogg2_page_bos(PyOggPage_AsOggPage(self)) ? "BOS, " : "";
  eos = ogg2_page_eos(PyOggPage_AsOggPage(self)) ? "EOS, " : "";
  cont = ogg2_page_continued(PyOggPage_AsOggPage(self)) ? "CONT, " : "";
  sprintf(buf, "<OggPage, %s%s%spageno = %ld, granulepos = %lld,"
	  " packets = %d, serialno = %d, version = %d," 
          " head length = %ld, body length = %ld, at %p (%p)>",
	  cont, bos, eos, ogg2_page_pageno(PyOggPage_AsOggPage(self)),
          ogg2_page_granulepos(PyOggPage_AsOggPage(self)),
	  ogg2_page_packets(PyOggPage_AsOggPage(self)),
          ogg2_page_serialno(PyOggPage_AsOggPage(self)),
          ogg2_page_version(PyOggPage_AsOggPage(self)), 
          PyOggPage_AsOggPage(self)->header_len, 
          PyOggPage_AsOggPage(self)->body_len,
          self, PyOggPage_AsOggPage(self)); 

  return PyString_FromString(buf);
}