#include <Python.h>
#include <structmember.h>
+#include <stddef.h>
+
#if (PY_VERSION_HEX < 0x02050000)
typedef int Py_ssize_t;
PyObject_HEAD
/* expose these */
- short kk;
- short mm;
+ unsigned short kk;
+ unsigned short mm;
/* internal */
fec_t* fec_matrix;
NULL
};
int ink, inm;
- if (!PyArg_ParseTupleAndKeywords(args, kwdict, "ii", kwlist, &ink, &inm))
+ if (!PyArg_ParseTupleAndKeywords(args, kwdict, "ii:Encoder.__init__", kwlist, &ink, &inm))
return -1;
if (ink < 1) {
- PyErr_Format(py_fec_error, "Precondition violation: first argument is required to be greater than or equal to 1, but it was %d", self->kk);
+ PyErr_Format(py_fec_error, "Precondition violation: first argument is required to be greater than or equal to 1, but it was %d", ink);
return -1;
}
if (inm < 1) {
- PyErr_Format(py_fec_error, "Precondition violation: second argument is required to be greater than or equal to 1, but it was %d", self->mm);
+ PyErr_Format(py_fec_error, "Precondition violation: second argument is required to be greater than or equal to 1, but it was %d", inm);
return -1;
}
if (inm > 256) {
- PyErr_Format(py_fec_error, "Precondition violation: second argument is required to be less than or equal to 256, but it was %d", self->mm);
+ PyErr_Format(py_fec_error, "Precondition violation: second argument is required to be less than or equal to 256, but it was %d", inm);
return -1;
}
if (ink > inm) {
PyErr_Format(py_fec_error, "Precondition violation: first argument is required to be less than or equal to the second argument, but they were %d and %d respectively", ink, inm);
return -1;
}
- self->kk = (short)ink;
- self->mm = (short)inm;
+ self->kk = (unsigned short)ink;
+ self->mm = (unsigned short)inm;
self->fec_matrix = fec_new(self->kk, self->mm);
return 0;
Py_ssize_t sz, oldsz = 0;
unsigned char check_block_index = 0; /* index into the check_blocks_produced and (parallel) pystrs_produced arrays */
- if (!PyArg_ParseTuple(args, "O|O", &inblocks, &desired_blocks_nums))
+ if (!PyArg_ParseTuple(args, "O|O:Encoder.encode", &inblocks, &desired_blocks_nums))
return NULL;
for (i=0; i<self->mm - self->kk; i++)
goto err;
if (PySequence_Fast_GET_SIZE(fastinblocks) != self->kk) {
- PyErr_Format(py_fec_error, "Precondition violation: Wrong length -- first argument (the sequence of input blocks) is required to contain exactly k blocks. len(first): %d, k: %d", PySequence_Fast_GET_SIZE(fastinblocks), self->kk);
+ PyErr_Format(py_fec_error, "Precondition violation: Wrong length -- first argument (the sequence of input blocks) is required to contain exactly k blocks. len(first): %Zu, k: %d", PySequence_Fast_GET_SIZE(fastinblocks), self->kk);
goto err;
}
}
oldsz = sz;
}
-
+
/* Allocate space for all of the check blocks. */
for (i=0; i<num_desired_blocks; i++) {
static void
Encoder_dealloc(Encoder * self) {
- fec_free(self->fec_matrix);
+ if (self->fec_matrix)
+ fec_free(self->fec_matrix);
self->ob_type->tp_free((PyObject*)self);
}
PyObject_HEAD
/* expose these */
- short kk;
- short mm;
+ unsigned short kk;
+ unsigned short mm;
/* internal */
fec_t* fec_matrix;
};
int ink, inm;
- if (!PyArg_ParseTupleAndKeywords(args, kwdict, "ii", kwlist, &ink, &inm))
+ if (!PyArg_ParseTupleAndKeywords(args, kwdict, "ii:Decoder.__init__", kwlist, &ink, &inm))
return -1;
if (ink < 1) {
- PyErr_Format(py_fec_error, "Precondition violation: first argument is required to be greater than or equal to 1, but it was %d", self->kk);
+ PyErr_Format(py_fec_error, "Precondition violation: first argument is required to be greater than or equal to 1, but it was %d", ink);
return -1;
}
if (inm < 1) {
- PyErr_Format(py_fec_error, "Precondition violation: second argument is required to be greater than or equal to 1, but it was %d", self->mm);
+ PyErr_Format(py_fec_error, "Precondition violation: second argument is required to be greater than or equal to 1, but it was %d", inm);
return -1;
}
if (inm > 256) {
- PyErr_Format(py_fec_error, "Precondition violation: second argument is required to be less than or equal to 256, but it was %d", self->mm);
+ PyErr_Format(py_fec_error, "Precondition violation: second argument is required to be less than or equal to 256, but it was %d", inm);
return -1;
}
if (ink > inm) {
PyErr_Format(py_fec_error, "Precondition violation: first argument is required to be less than or equal to the second argument, but they were %d and %d respectively", ink, inm);
return -1;
}
- self->kk = (short)ink;
- self->mm = (short)inm;
+ self->kk = (unsigned short)ink;
+ self->mm = (unsigned short)inm;
self->fec_matrix = fec_new(self->kk, self->mm);
return 0;
long tmpl;
unsigned nextrecoveredix=0;
- if (!PyArg_ParseTuple(args, "OO", &blocks, &blocknums))
+ if (!PyArg_ParseTuple(args, "OO:Decoder.decode", &blocks, &blocknums))
return NULL;
for (i=0; i<self->kk; i++)
goto err;
if (PySequence_Fast_GET_SIZE(fastblocks) != self->kk) {
- PyErr_Format(py_fec_error, "Precondition violation: Wrong length -- first argument is required to contain exactly k blocks. len(first): %d, k: %d", PySequence_Fast_GET_SIZE(fastblocks), self->kk);
+ PyErr_Format(py_fec_error, "Precondition violation: Wrong length -- first argument is required to contain exactly k blocks. len(first): %Zu, k: %d", PySequence_Fast_GET_SIZE(fastblocks), self->kk);
goto err;
}
if (PySequence_Fast_GET_SIZE(fastblocknums) != self->kk) {
- PyErr_Format(py_fec_error, "Precondition violation: Wrong length -- blocknums is required to contain exactly k blocks. len(blocknums): %d, k: %d", PySequence_Fast_GET_SIZE(fastblocknums), self->kk);
+ PyErr_Format(py_fec_error, "Precondition violation: Wrong length -- blocknums is required to contain exactly k blocks. len(blocknums): %Zu, k: %d", PySequence_Fast_GET_SIZE(fastblocknums), self->kk);
goto err;
}
oldsz = sz;
}
- /* move src packets into position */
+ /* Move src packets into position. At the end of this loop we want the i'th
+ element of the arrays to be the block with block number i, if that block
+ is among our inputs. */
for (i=0; i<self->kk;) {
if (cblocknums[i] >= self->kk || cblocknums[i] == i)
i++;
static void
Decoder_dealloc(Decoder * self) {
- fec_free(self->fec_matrix);
+ if (self->fec_matrix)
+ fec_free(self->fec_matrix);
self->ob_type->tp_free((PyObject*)self);
}
Decoder_new, /* tp_new */
};
+
+void
+_hexwrite(unsigned char*s, size_t l) {
+ size_t i;
+ for (i = 0; i < l; i++)
+ printf("%.2x", s[i]);
+}
+
+
PyObject*
-test_from_agl() {
- unsigned char b1c[8], b2c[8];
- unsigned char b1[8], b2[8], b3[8], b4[8], b5[8];
- memset(b1, 1, 8);
- memset(b2, 2, 8);
- memset(b3, 3, 8);
- const unsigned char *blocks[3] = {b1, b2, b3};
- unsigned char *outblocks[2] = {b4, b5};
+test_from_agl(PyObject* self, PyObject* args) {
+ unsigned char b0c[8], b1c[8];
+ unsigned char b0[8], b1[8], b2[8], b3[8], b4[8];
+
+ const unsigned char *blocks[3] = {b0, b1, b2};
+ unsigned char *outblocks[2] = {b3, b4};
unsigned block_nums[] = {3, 4};
fec_t *const fec = fec_new(3, 5);
- fec_encode(fec, blocks, outblocks, block_nums, 2, 8);
- write(1, b1, 8);
- write(1, b2, 8);
- write(1, b3, 8);
- write(1, b4, 8);
- write(1, b5, 8);
+ const unsigned char *inpkts[] = {b3, b4, b2};
+ unsigned char *outpkts[] = {b0, b1};
+ unsigned indexes[] = {3, 4, 2};
- memcpy(b1c, b1, 8); memcpy(b2c, b2, 8);
+ memset(b0, 1, 8);
+ memset(b1, 2, 8);
+ memset(b2, 3, 8);
- const unsigned char *inpkts[] = {b3, b4, b5};
- unsigned char *outpkts[] = {b1, b2};
- unsigned indexes[] = {2, 3, 4};
+ /*printf("_from_c before encoding:\n");
+ printf("b0: "); _hexwrite(b0, 8); printf(", ");
+ printf("b1: "); _hexwrite(b1, 8); printf(", ");
+ printf("b2: "); _hexwrite(b2, 8); printf(", ");
+ printf("\n");*/
+
+ fec_encode(fec, blocks, outblocks, block_nums, 2, 8);
+
+ /*printf("after encoding:\n");
+ printf("b3: "); _hexwrite(b3, 8); printf(", ");
+ printf("b4: "); _hexwrite(b4, 8); printf(", ");
+ printf("\n");*/
+
+ memcpy(b0c, b0, 8); memcpy(b1c, b1, 8);
fec_decode(fec, inpkts, outpkts, indexes, 8);
- write(1, b1, 8);
- write(1, b2, 8);
+ /*printf("after decoding:\n");
+ printf("b0: "); _hexwrite(b0, 8); printf(", ");
+ printf("b1: "); _hexwrite(b1, 8);
+ printf("\n");*/
- if ((memcmp(b1, b1c,8) == 0) && (memcmp(b2, b2c,8) == 0))
+ if ((memcmp(b0, b0c,8) == 0) && (memcmp(b1, b1c,8) == 0))
Py_RETURN_TRUE;
else
Py_RETURN_FALSE;
}
-static PyMethodDef fec_functions[] = {
- {"test_from_agl", test_from_agl, METH_NOARGS, NULL},
- {NULL}
+static PyMethodDef fec_functions[] = {
+ {"test_from_agl", test_from_agl, METH_NOARGS, NULL},
+ {NULL}
};
#ifndef PyMODINIT_FUNC /* declarations for DLL import/export */
PyDict_SetItemString(module_dict, "Error", py_fec_error);
}
-/**
- * zfec -- fast forward error correction library with Python interface
- *
- * Copyright (C) 2007 Allmydata, Inc.
- * Author: Zooko Wilcox-O'Hearn
- *
- * This file is part of zfec.
- *
- * See README.txt for licensing information.
- */
-
/**
* originally inspired by fecmodule.c by the Mnet Project, especially Myers
* Carpenter and Hauke Johannknecht
*/
-