瀏覽代碼

Messing around with complex algebra

master
Pete Shadbolt 10 年之前
父節點
當前提交
b946190670
共有 2 個文件被更改,包括 43 次插入10 次删除
  1. +21
    -0
      makefile
  2. +22
    -10
      src/permanent.c

+ 21
- 0
makefile 查看文件

@@ -0,0 +1,21 @@
srcdir = src
objdir = ./

target = permanent.so
setup = setup.py

default : $(target) test

# Compile
$(target): $(srcdir)/permanent.c
python $(setup) build_ext --inplace

test :
python ./run-tests.py

clean :
@ rm permanent.so
@ rm -rf build

tar :
@ tar czf permanent.tar.gz $(srcdir) $(utils)

+ 22
- 10
src/permanent.c 查看文件

@@ -1,5 +1,6 @@
/* Computes the permanent, given a numpy array */

#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
#include <Python.h>
#include <numpy/arrayobject.h>
#include <math.h>
@@ -23,7 +24,6 @@ PyMODINIT_FUNC initpermanent(void) {
/*****************************************************************************
* Array access macros. *
*****************************************************************************/
#define NPY_NO_DEPRECATED_API NPY_1_7_API_VERSION
#define m(x0) (*(npy_float64*)((PyArray_DATA(py_m) + \
(x0) * PyArray_STRIDES(py_m)[0])))
#define m_shape(i) (py_m->dimensions[(i)])
@@ -44,25 +44,37 @@ PyMODINIT_FUNC initpermanent(void) {
#define F_shape(i) (py_F->dimensions[(i)])

static PyObject *permanent(PyObject *self, PyObject *args) {
// Declare variables.
npy_int64 d, i, j;
double complex sum;
// Parse input
PyArrayObject *submatrix;

// Parse variables.
if (!PyArg_ParseTuple(args, "O!", &PyArray_Type, &submatrix)) {
return NULL;
}

// Declare variables.
/*npy_int64 d, i, j;*/


Py_complex a;
Py_complex b;

a.real=1; a.imag=1;
b.real=1; b.imag=1;

a.real=a.real+b.real;
a.imag=a.imag+b.imag;

// Convert to a python complex number
PyObject *output=PyComplex_FromDoubles(a.real,a.imag);
return output;

/*

// Compute the permanent
for(i = 0; i < d; ++i) {
for(j = 0; j<d; ++j) {
sum+=1j;
}
}
*/

// Convert to a python complex number
/*Py_complex psum = struct {};*/ //fromdouble
PyObject *output=PyComplex_FromCComplex(psum);
return output;
}

Loading…
取消
儲存