← Back to branch summary

~ubuntu-branches/ubuntu/wily/pyzmq/wily

« back to all changes in this revision

Viewing changes to zmq/utils/allocate.pxd

  • Committer: Package Import Robot
  • Author(s): Julian Taylor
  • Date: 2013-02-24 19:23:15 UTC
  • mfrom: (1.2.1) (9 sid)
  • mto: This revision was merged to the branch mainline in revision 10.
  • Revision ID: package-import@ubuntu.com-20130224192315-qhmwp3m3ymk8r60d
Tags: 2.2.0.1-1
* New upstream release
* relicense debian packaging to LGPL-3
* update watch file to use github directly
  thanks to Bart Martens for the file
* add autopkgtests
* drop obsolete DM-Upload-Allowed
* bump standard to 3.9.4, no changes required

Show diffs side-by-side

added added

removed removed

Lines of Context:
zmq/utils/allocate.pxd
1
 
"""A utility to allocate a C array.
2
 
 
3
 
This was copied from mpi4py and is licensed under the BSD license.
4
 
"""
5
 
 
6
 
from libc.stdlib cimport free, malloc
7
 
 
8
 
#-----------------------------------------------------------------------------
9
 
# Python includes.
10
 
#-----------------------------------------------------------------------------
11
 
 
12
 
cdef extern from "Python.h":
13
 
    object PyCObject_FromVoidPtr(void *, void (*)(void*))
14
 
 
15
 
#-----------------------------------------------------------------------------
16
 
# Main functions.
17
 
#-----------------------------------------------------------------------------
18
 
 
19
 
cdef inline void *memnew(size_t n):
20
 
    """malloc a new memory chunk of a given size."""
21
 
    if n == 0: n = 1
22
 
    return malloc(n)
23
 
 
24
 
cdef inline void memdel(void *p):
25
 
    """free a chunk of memory allocated with memnew."""
26
 
    if p != NULL: free(p)
27
 
 
28
 
cdef inline object allocate(size_t n, void **pp):
29
 
    """A wrapper that allocates a C array, but with Python ref-counting."""
30
 
    cdef object cob
31
 
    cdef void *p = memnew(n)
32
 
    if p == NULL:
33
 
        raise MemoryError()
34
 
    try:
35
 
        cob = PyCObject_FromVoidPtr(p, memdel)
36
 
    except:
37
 
        memdel(p)
38
 
        raise
39
 
    pp[0] = p
40
 
    return cob