66 lines
1.9 KiB
Plaintext
66 lines
1.9 KiB
Plaintext
%module ckzg
|
|
%{
|
|
#include "c_kzg.h"
|
|
#include "bls12_381.h"
|
|
#include "test_util.h"
|
|
%}
|
|
%include "stdint.i"
|
|
%include "carrays.i"
|
|
%include "cpointer.i"
|
|
%rename(alloc_poly) new_poly;
|
|
%rename(alloc_poly_l) new_poly_l;
|
|
|
|
%typemap(in, numinputs=0) OBJECT *OUTPUT($1_basetype tmp) { $1 = &tmp; }
|
|
%typemap(argout) OBJECT *OUTPUT {
|
|
PyObject *obj = SWIG_NewPointerObj(memcpy(malloc(sizeof($1_basetype)),$1,sizeof($1_basetype)),
|
|
$descriptor, SWIG_POINTER_NEW);
|
|
$result = ($result == NULL) ? obj : SWIG_Python_AppendOutput($result, obj);
|
|
}
|
|
%apply OBJECT *OUTPUT {
|
|
fr_t *out, poly *out, poly_l *out,
|
|
g1_t *out, g2_t *out,
|
|
blst_scalar *out,
|
|
FFTSettings *out, KZGSettings *out
|
|
}
|
|
|
|
%typemap(in, numinputs=0) bool *OUTPUT(bool tmp) { $1 = &tmp; }
|
|
%typemap(argout) bool *OUTPUT {
|
|
PyObject *obj = *$1 ? Py_True : Py_False;
|
|
$result = ($result == NULL) ? obj : SWIG_Python_AppendOutput($result, obj);
|
|
}
|
|
%apply bool *OUTPUT { bool *out }
|
|
|
|
%typemap(in, numinputs=0) uint64_t out[4] (uint64_t tmp[4]) { $1 = tmp; }
|
|
%typemap(argout) uint64_t out[4] {
|
|
PyObject *obj = PyTuple_Pack(4,
|
|
PyLong_FromUnsignedLong($1[0]),
|
|
PyLong_FromUnsignedLong($1[1]),
|
|
PyLong_FromUnsignedLong($1[2]),
|
|
PyLong_FromUnsignedLong($1[3]));
|
|
$result = ($result == NULL) ? obj : SWIG_Python_AppendOutput($result, obj);
|
|
}
|
|
|
|
%typemap(in) const uint64_t[4] (uint64_t tmp[4]) {
|
|
if (PyTuple_Check($input)) {
|
|
if (!PyArg_ParseTuple($input, "kkkk", tmp, tmp+1, tmp+2, tmp+3)) {
|
|
PyErr_SetString(PyExc_TypeError, "tuple must have 4 elements");
|
|
SWIG_fail;
|
|
}
|
|
$1 = &tmp[0];
|
|
} else {
|
|
PyErr_SetString(PyExc_TypeError, "expected a tuple.");
|
|
SWIG_fail;
|
|
}
|
|
}
|
|
|
|
%array_class(fr_t, frArray)
|
|
%array_class(g1_t, g1Array)
|
|
%array_class(g2_t, g2Array)
|
|
%pointer_class(poly, polyp)
|
|
%pointer_class(poly_l, poly_lp)
|
|
|
|
%include "../src/c_kzg.h"
|
|
%include "../src/bls12_381.h"
|
|
%include "../inc/blst.h"
|
|
%include "../src/test_util.h"
|