#ifdef SWIGPYTHON

%include "protocols_helper.i"

%define NUMERIC_DENSELABELS(class_type, class_name, type_name, format_str, operator_name, operator)

PyObject* class_name ## _inplace ## operator_name ## (PyObject *self, PyObject *o2)

{

PyObject* resultobj=0;

class_type* arg1=(class_type*) 0; // self in c++ repr

void* argp1=0; // pointer to self

int res1=0; // result for self's casting

PyObject* internal_data=0;

res1 = SWIG_ConvertPtr(self, &argp1, SWIG_TypeQuery("shogun::class_type"), 0 | 0 );

if (!SWIG_IsOK(res1))

{

// TODO fix message

SWIG_exception_fail(SWIG_ArgError(res1),

"in method '" "inplace_#operator_name" "', argument " "1"" of type '" "class_type *""'");

}

arg1=reinterpret_cast< class_type* >(argp1);

internal_data=PySequence_GetSlice(self, 0, arg1->get_num_labels());

PyNumber_InPlace ## operator ## (internal_data, o2);

resultobj=self;

Py_INCREF(resultobj);

return resultobj;

fail:

return NULL;

}

%enddef // NUMERIC_DENSELABELS

%define PROTOCOLS_DENSELABELS(class_type, class_name, type_name, format_str, typecode)

%wrapper

%{

static int class_name ## _getbuffer(PyObject *self, Py_buffer *view, int flags)

{

class_type* arg1=(class_type*) 0; // self in c++ repr

void* argp1=0; // pointer to self

int res1=0; // result for self's casting

int num_labels=0;

Py_ssize_t* shape=NULL;

Py_ssize_t* strides=NULL;

buffer_vector_ ## type_name ## _info* info=NULL;

static char* format=(char *) format_str; // http://docs.python.org/dev/library/struct.html#module-struct

res1 = SWIG_ConvertPtr(self, &argp1, SWIG_TypeQuery("shogun::class_type"), 0 | 0 );

if (!SWIG_IsOK(res1))

{

SWIG_exception_fail(SWIG_ArgError(res1),

"in method '" "getbuffer" "', argument " "1"" of type '" "class_type *""'");

}

if ((flags & PyBUF_C_CONTIGUOUS)==PyBUF_C_CONTIGUOUS)

{

PyErr_SetString(PyExc_ValueError, "class_name is not C-contiguous");

goto fail;

}

if ((flags & PyBUF_STRIDES)!=PyBUF_STRIDES &&

(flags & PyBUF_ND)==PyBUF_ND)

{

PyErr_SetString(PyExc_ValueError, "class_name is not C-contiguous");

goto fail;

}

arg1=reinterpret_cast< class_type* >(argp1);

info=new buffer_vector_ ## type_name ## _info;

info->buf=arg1->get_labels();

num_labels=arg1->get_num_labels();

view->buf=info->buf.vector;

shape=new Py_ssize_t[1];

shape[0]=num_labels;

strides=new Py_ssize_t[1];

strides[0]=sizeof( type_name );

info->shape=shape;

info->strides=strides;

view->ndim=1;

view->format=(char*) format_str;

view->itemsize=sizeof( type_name );

view->len=shape[0]*view->itemsize;

view->shape=shape;

view->strides=strides;

view->readonly=0;

view->suboffsets=NULL;

view->internal=(void*) info;

view->obj=(PyObject*) self;

Py_INCREF(self);

return 0;

fail:

view->obj=NULL;

return -1;

}

static void class_name ## _releasebuffer(PyObject *self, Py_buffer *view)

{

buffer_vector_ ## type_name ## _info* temp=NULL;

if (view->obj!=NULL && view->internal!=NULL)

{

temp=(buffer_vector_ ## type_name ## _info*) view->internal;

if (temp->shape!=NULL)

delete[] temp->shape;

if (temp->strides!=NULL)

delete[] temp->strides;

temp->buf=SGVector< type_name >();

delete temp;

}

}

static PyObject* class_name ## _getitem(PyObject *self, Py_ssize_t idx, bool get_scalar=true)

{

class_type* arg1=0; // self in c++ repr

void* argp1=0; // pointer to self

int res1=0; // result for self's casting

char* data=0; // internal data of self

int num_labels=0;

SGVector< type_name > temp;

Py_ssize_t* shape;

Py_ssize_t* strides;

PyObject* ret;

PyArray_Descr* descr=PyArray_DescrFromType(typecode);

res1 = SWIG_ConvertPtr(self, &argp1, SWIG_TypeQuery("shogun::class_type"), 0 | 0 );

if (!SWIG_IsOK(res1))

{

SWIG_exception_fail(SWIG_ArgError(res1),

"in method '" "getitem" "', argument " "1"" of type '" "class_type *""'");

}

arg1=reinterpret_cast< class_type* >(argp1);

temp=arg1->get_labels();

num_labels=arg1->get_num_labels();

data=(char*) temp.vector;

idx=get_idx_in_bounds(idx, num_labels);

if (idx < 0)

{

goto fail;

}

data+=idx * sizeof( type_name );

shape=new Py_ssize_t[1];

shape[0]=1;

strides=new Py_ssize_t[1];

strides[0]=sizeof( type_name );

if (get_scalar)

{

ret=(PyObject *) PyArray_Scalar(data, descr, (PyObject *) self);

}

else

{

ret=(PyObject *) PyArray_NewFromDescr(&PyArray_Type, descr,

0, shape,

strides, data,

NPY_ARRAY_FARRAY | NPY_ARRAY_WRITEABLE,

(PyObject *) self);

}

if (ret==NULL)

goto fail;

Py_INCREF(self);

return ret;

fail:

return NULL;

}

static int class_name ## _setitem(PyObject *self, Py_ssize_t idx, PyObject *v)

{

PyArrayObject* tmp=NULL;

int ret=0;

if (v==NULL)

{

// TODO error message

goto fail;

}

tmp=(PyArrayObject *) class_name ## _getitem(self, idx, false);

if(tmp==NULL)

{

goto fail;

}

ret=PyArray_CopyObject(tmp, v);

Py_DECREF(tmp);

return ret;

fail:

return -1;

}

static PyObject* class_name ## _getslice(PyObject *self, Py_ssize_t ilow, Py_ssize_t ihigh)

{

class_type* arg1=0; // self in c++ repr

void* argp1=0; // pointer to self

int res1=0 ; // result for self's casting

int num_labels=0;

char* data=0; // internal data of self

SGVector< type_name > temp;

Py_ssize_t* shape;

Py_ssize_t* strides;

PyArrayObject* ret;

PyArray_Descr* descr=PyArray_DescrFromType(typecode);

res1=SWIG_ConvertPtr(self, &argp1, SWIG_TypeQuery("shogun::class_type"), 0 | 0 );

if (!SWIG_IsOK(res1))

{

SWIG_exception_fail(SWIG_ArgError(res1),

"in method '" "slice" "', argument " "1"" of type '" "class_type *""'");

}

arg1=reinterpret_cast< class_type* >(argp1);

temp=arg1->get_labels();

num_labels=arg1->get_num_labels();

data=(char*) temp.vector;

get_slice_in_bounds(&ilow, &ihigh, num_labels);

if (ilow < ihigh)

{

data+=ilow * sizeof( type_name );

}

shape=new Py_ssize_t[1];

shape[0]=ihigh - ilow;

strides=new Py_ssize_t[1];

strides[0]=sizeof( type_name );

ret=(PyArrayObject *) PyArray_NewFromDescr(&PyArray_Type, descr,

1, shape,

strides, data,

NPY_ARRAY_FARRAY | NPY_ARRAY_WRITEABLE,

(PyObject *) self);

if (ret==NULL)

goto fail;

Py_INCREF(self);

return (PyObject *) ret;

fail:

return NULL;

}

static int class_name ## _setslice(PyObject *self, Py_ssize_t ilow, Py_ssize_t ihigh, PyObject* v)

{

PyArrayObject* tmp=NULL;

int ret=0;

if (v==NULL)

{

// TODO error message

goto fail;

}

tmp=(PyArrayObject *) class_name ## _getslice(self, ilow, ihigh);

if(tmp==NULL)

{

goto fail;

}

ret = PyArray_CopyObject(tmp, v);

Py_DECREF(tmp);

return ret;

fail:

return -1;

}

NUMERIC_DENSELABELS(class_type, class_name, type_name, format_str, add, Add)

NUMERIC_DENSELABELS(class_type, class_name, type_name, format_str, sub, Subtract)

NUMERIC_DENSELABELS(class_type, class_name, type_name, format_str, mul, Multiply)

static long class_name ## _flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_NEWBUFFER | Py_TPFLAGS_BASETYPE;

%}

%init

%{

SwigPyBuiltin__shogun__ ## class_type ## _type.ht_type.tp_flags = class_name ## _flags;

%}

%feature("python:bf_getbuffer") class_type #class_name "_getbuffer"

%feature("python:bf_releasebuffer") class_type #class_name "_releasebuffer"

%feature("python:nb_inplace_add") class_type #class_name "_inplaceadd"

%feature("python:nb_inplace_subtract") class_type #class_name "_inplacesub"

%feature("python:nb_inplace_multiply") class_type #class_name "_inplacemul"

%feature("python:sq_item") class_type #class_name "_getitem"

%feature("python:sq_ass_item") class_type #class_name "_setitem"

%feature("python:sq_slice") class_type #class_name "_getslice"

%feature("python:sq_ass_slice") class_type #class_name "_setslice"

%enddef /* PROTOCOLS_DENSELABELS */

#endif /* SWIG_PYTHON */