X-Git-Url: https://projects.mako.cc/source/harrypotter-wikipedia-cdsw/blobdiff_plain/2da9c10d579b36f785f90fd2aebc57b3b0c3e876..2e473702cd932f0a9f9c05545134c372089ef30b:/win_unicode_console/buffer.py

diff --git a/win_unicode_console/buffer.py b/win_unicode_console/buffer.py
new file mode 100644
index 0000000..cd5853c
--- /dev/null
+++ b/win_unicode_console/buffer.py
@@ -0,0 +1,51 @@
+
+from ctypes import (byref, POINTER, Structure, pythonapi,
+	c_int, c_char, c_char_p, c_void_p, py_object, c_ssize_t)
+import sys
+
+c_ssize_p = POINTER(c_ssize_t)
+
+PyObject_GetBuffer = pythonapi.PyObject_GetBuffer
+PyBuffer_Release = pythonapi.PyBuffer_Release
+
+
+PyBUF_SIMPLE = 0
+PyBUF_WRITABLE = 1
+
+
+class Py_buffer(Structure):
+	_fields_ = [
+		("buf", c_void_p),
+		("obj", py_object),
+		("len", c_ssize_t),
+		("itemsize", c_ssize_t),
+		("readonly", c_int),
+		("ndim", c_int),
+		("format", c_char_p),
+		("shape", c_ssize_p),
+		("strides", c_ssize_p),
+		("suboffsets", c_ssize_p),
+		("internal", c_void_p)
+	]
+	
+	if sys.version_info[0] < 3:
+		_fields_.insert(-1, ("smalltable", c_ssize_t * 2))
+	
+	@classmethod
+	def get_from(cls, obj, flags=PyBUF_SIMPLE):
+		buf = cls()
+		PyObject_GetBuffer(py_object(obj), byref(buf), flags)
+		return buf
+	
+	def release(self):
+		PyBuffer_Release(byref(self))
+
+
+def get_buffer(obj, writable=False):
+	buf = Py_buffer.get_from(obj, PyBUF_WRITABLE if writable else PyBUF_SIMPLE)
+	try:
+		buffer_type = c_char * buf.len
+		return buffer_type.from_address(buf.buf)
+	finally:
+		buf.release()
+