46 lines
1.3 KiB
Cython
46 lines
1.3 KiB
Cython
cimport libav as lib
|
|
from libc.stdint cimport uint8_t, uintptr_t
|
|
from libc.string cimport memcpy
|
|
|
|
|
|
cdef void key_free(void *opaque, uint8_t *data) noexcept nogil:
|
|
cdef char *name = <char *>data
|
|
with gil:
|
|
opaque_container.pop(name)
|
|
|
|
|
|
cdef class OpaqueContainer:
|
|
def __cinit__(self):
|
|
self._objects = {}
|
|
|
|
cdef lib.AVBufferRef *add(self, object v):
|
|
# Use object's memory address as key
|
|
cdef uintptr_t key = <uintptr_t><long long>id(v)
|
|
self._objects[key] = v
|
|
|
|
cdef uint8_t *data = <uint8_t *>lib.av_malloc(sizeof(uintptr_t))
|
|
if data == NULL:
|
|
raise MemoryError("Failed to allocate memory for key")
|
|
|
|
memcpy(data, &key, sizeof(uintptr_t))
|
|
|
|
# Create the buffer with our free callback
|
|
cdef lib.AVBufferRef *buffer_ref = lib.av_buffer_create(
|
|
data, sizeof(uintptr_t), key_free, NULL, 0
|
|
)
|
|
|
|
if buffer_ref == NULL:
|
|
raise MemoryError("Failed to create AVBufferRef")
|
|
|
|
return buffer_ref
|
|
|
|
cdef object get(self, char *name):
|
|
cdef uintptr_t key = (<uintptr_t *>name)[0]
|
|
return self._objects.get(key)
|
|
|
|
cdef object pop(self, char *name):
|
|
cdef uintptr_t key = (<uintptr_t *>name)[0]
|
|
return self._objects.pop(key, None)
|
|
|
|
|
|
cdef OpaqueContainer opaque_container = OpaqueContainer()
|