summaryrefslogtreecommitdiff
path: root/demo/extern_python.py
blob: f315cc548159df931849a7f286da74d4f7073aa3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import cffi

ffi = cffi.FFI()

ffi.cdef("""int my_algo(int); extern "Python" int f(int);""")

ffi.set_source("_extern_python_cffi", """
    static int f(int);
    static int my_algo(int n) {
        int i, sum = 0;
        for (i = 0; i < n; i++)
            sum += f(i);
        return sum;
    }
""")

ffi.compile()


from _extern_python_cffi import ffi, lib

@ffi.def_extern()
def f(n):
    return n * n

assert lib.my_algo(10) == 0+1+4+9+16+25+36+49+64+81