/Mac/Modules/cf/pycfbridge.c
http://unladen-swallow.googlecode.com/ · C · 303 lines · 263 code · 28 blank · 12 comment · 78 complexity · 46a8cd4e91920186b684548c50c9917e MD5 · raw file
- /*
- ** Convert objects from Python to CoreFoundation and vice-versa.
- */
- #include <CoreServices/CoreServices.h>
- #include "Python.h"
- #include "pymactoolbox.h"
- #include "pycfbridge.h"
- /* ---------------------------------------- */
- /* CoreFoundation objects to Python objects */
- /* ---------------------------------------- */
- PyObject *
- PyCF_CF2Python(CFTypeRef src) {
- CFTypeID typeid;
-
- if( src == NULL ) {
- Py_INCREF(Py_None);
- return Py_None;
- }
- typeid = CFGetTypeID(src);
- if (typeid == CFArrayGetTypeID())
- return PyCF_CF2Python_sequence((CFArrayRef)src);
- if (typeid == CFDictionaryGetTypeID())
- return PyCF_CF2Python_mapping((CFDictionaryRef)src);
- return PyCF_CF2Python_simple(src);
- }
- PyObject *
- PyCF_CF2Python_sequence(CFArrayRef src) {
- int size = CFArrayGetCount(src);
- PyObject *rv;
- CFTypeRef item_cf;
- PyObject *item_py = NULL;
- int i;
-
- if ( (rv=PyList_New(size)) == NULL )
- return NULL;
- for(i=0; i<size; i++) {
- item_cf = CFArrayGetValueAtIndex(src, i);
- if (item_cf == NULL ) goto err;
- item_py = PyCF_CF2Python(item_cf);
- if (item_py == NULL ) goto err;
- if (PyList_SetItem(rv, i, item_py) < 0) goto err;
- item_py = NULL;
- }
- return rv;
- err:
- Py_XDECREF(item_py);
- Py_DECREF(rv);
- return NULL;
- }
- PyObject *
- PyCF_CF2Python_mapping(CFTypeRef src) {
- int size = CFDictionaryGetCount(src);
- PyObject *rv = NULL;
- CFTypeRef *allkeys = NULL, *allvalues = NULL;
- CFTypeRef key_cf, value_cf;
- PyObject *key_py = NULL, *value_py = NULL;
- int i;
-
- allkeys = malloc(size*sizeof(CFTypeRef *));
- if (allkeys == NULL) {
- PyErr_NoMemory();
- goto err;
- }
- allvalues = malloc(size*sizeof(CFTypeRef *));
- if (allvalues == NULL) {
- PyErr_NoMemory();
- goto err;
- }
- if ( (rv=PyDict_New()) == NULL ) goto err;
- CFDictionaryGetKeysAndValues(src, allkeys, allvalues);
- for(i=0; i<size; i++) {
- key_cf = allkeys[i];
- value_cf = allvalues[i];
- key_py = PyCF_CF2Python(key_cf);
- if (key_py == NULL ) goto err;
- value_py = PyCF_CF2Python(value_cf);
- if (value_py == NULL ) goto err;
- if (PyDict_SetItem(rv, key_py, value_py) < 0) goto err;
- key_py = NULL;
- value_py = NULL;
- }
- return rv;
- err:
- Py_XDECREF(key_py);
- Py_XDECREF(value_py);
- Py_XDECREF(rv);
- free(allkeys);
- free(allvalues);
- return NULL;
- }
- PyObject *
- PyCF_CF2Python_simple(CFTypeRef src) {
- CFTypeID typeid;
-
- typeid = CFGetTypeID(src);
- if (typeid == CFStringGetTypeID())
- return PyCF_CF2Python_string((CFStringRef)src);
- if (typeid == CFBooleanGetTypeID())
- return PyBool_FromLong((long)CFBooleanGetValue(src));
- if (typeid == CFNumberGetTypeID()) {
- if (CFNumberIsFloatType(src)) {
- double d;
- CFNumberGetValue(src, kCFNumberDoubleType, &d);
- return PyFloat_FromDouble(d);
- } else {
- long l;
- if (!CFNumberGetValue(src, kCFNumberLongType, &l))
- /* XXXX Out of range! */;
- return PyInt_FromLong(l);
- }
- }
- /* XXXX Should return as CFTypeRef, really... */
- PyMac_Error(resNotFound);
- return NULL;
- }
- /* Unsure - Return unicode or 8 bit strings? */
- PyObject *
- PyCF_CF2Python_string(CFStringRef src) {
- int size = CFStringGetLength(src)+1;
- Py_UNICODE *data = malloc(size*sizeof(Py_UNICODE));
- CFRange range;
- PyObject *rv;
- range.location = 0;
- range.length = size;
- if( data == NULL ) return PyErr_NoMemory();
- CFStringGetCharacters(src, range, data);
- rv = (PyObject *)PyUnicode_FromUnicode(data, size-1);
- free(data);
- return rv;
- }
- /* ---------------------------------------- */
- /* Python objects to CoreFoundation objects */
- /* ---------------------------------------- */
- int
- PyCF_Python2CF(PyObject *src, CFTypeRef *dst) {
- if (PyString_Check(src) || PyUnicode_Check(src))
- return PyCF_Python2CF_simple(src, dst);
- if (PySequence_Check(src))
- return PyCF_Python2CF_sequence(src, (CFArrayRef *)dst);
- if (PyMapping_Check(src))
- return PyCF_Python2CF_mapping(src, (CFDictionaryRef *)dst);
- return PyCF_Python2CF_simple(src, dst);
- }
- int
- PyCF_Python2CF_sequence(PyObject *src, CFArrayRef *dst) {
- CFMutableArrayRef rv = NULL;
- CFTypeRef item_cf = NULL;
- PyObject *item_py = NULL;
- int size, i;
-
- if( !PySequence_Check(src) ) {
- PyErr_Format(PyExc_TypeError,
- "Cannot convert %.500s objects to CFArray",
- src->ob_type->tp_name);
- return 0;
- }
- size = PySequence_Size(src);
- rv = CFArrayCreateMutable((CFAllocatorRef)NULL, size, &kCFTypeArrayCallBacks);
- if (rv == NULL) {
- PyMac_Error(resNotFound);
- goto err;
- }
- for( i=0; i<size; i++) {
- item_py = PySequence_GetItem(src, i);
- if (item_py == NULL) goto err;
- if ( !PyCF_Python2CF(item_py, &item_cf)) goto err;
- Py_DECREF(item_py);
- CFArraySetValueAtIndex(rv, i, item_cf);
- CFRelease(item_cf);
- item_cf = NULL;
- }
- *dst = rv;
- return 1;
- err:
- Py_XDECREF(item_py);
- if (rv) CFRelease(rv);
- if (item_cf) CFRelease(item_cf);
- return 0;
- }
- int
- PyCF_Python2CF_mapping(PyObject *src, CFDictionaryRef *dst) {
- CFMutableDictionaryRef rv = NULL;
- PyObject *aslist = NULL;
- CFTypeRef key_cf = NULL, value_cf = NULL;
- PyObject *item_py = NULL, *key_py = NULL, *value_py = NULL;
- int size, i;
-
- if( !PyMapping_Check(src) ) {
- PyErr_Format(PyExc_TypeError,
- "Cannot convert %.500s objects to CFDictionary",
- src->ob_type->tp_name);
- return 0;
- }
- size = PyMapping_Size(src);
- rv = CFDictionaryCreateMutable((CFAllocatorRef)NULL, size,
- &kCFTypeDictionaryKeyCallBacks,
- &kCFTypeDictionaryValueCallBacks);
- if (rv == NULL) {
- PyMac_Error(resNotFound);
- goto err;
- }
- if ( (aslist = PyMapping_Items(src)) == NULL ) goto err;
- for( i=0; i<size; i++) {
- item_py = PySequence_GetItem(aslist, i);
- if (item_py == NULL) goto err;
- if (!PyArg_ParseTuple(item_py, "OO", &key_py, &value_py)) goto err;
- if ( !PyCF_Python2CF(key_py, &key_cf) ) goto err;
- if ( !PyCF_Python2CF(value_py, &value_cf) ) goto err;
- CFDictionaryAddValue(rv, key_cf, value_cf);
- CFRelease(key_cf);
- key_cf = NULL;
- CFRelease(value_cf);
- value_cf = NULL;
- }
- *dst = rv;
- return 1;
- err:
- Py_XDECREF(item_py);
- Py_XDECREF(aslist);
- if (rv) CFRelease(rv);
- if (key_cf) CFRelease(key_cf);
- if (value_cf) CFRelease(value_cf);
- return 0;
- }
- int
- PyCF_Python2CF_simple(PyObject *src, CFTypeRef *dst) {
-
- #if 0
- if (PyObject_HasAttrString(src, "CFType")) {
- *dst = PyObject_CallMethod(src, "CFType", "");
- return (*dst != NULL);
- }
- #endif
- if (PyString_Check(src) || PyUnicode_Check(src))
- return PyCF_Python2CF_string(src, (CFStringRef *)dst);
- if (PyBool_Check(src)) {
- if (src == Py_True)
- *dst = kCFBooleanTrue;
- else
- *dst = kCFBooleanFalse;
- return 1;
- }
- if (PyInt_Check(src)) {
- long v = PyInt_AsLong(src);
- *dst = CFNumberCreate(NULL, kCFNumberLongType, &v);
- return 1;
- }
- if (PyFloat_Check(src)) {
- double d = PyFloat_AsDouble(src);
- *dst = CFNumberCreate(NULL, kCFNumberDoubleType, &d);
- return 1;
- }
-
- PyErr_Format(PyExc_TypeError,
- "Cannot convert %.500s objects to CFType",
- src->ob_type->tp_name);
- return 0;
- }
- int
- PyCF_Python2CF_string(PyObject *src, CFStringRef *dst) {
- char *chars;
- CFIndex size;
- UniChar *unichars;
-
- if (PyString_Check(src)) {
- if (!PyArg_Parse(src, "es", "ascii", &chars))
- return 0; /* This error is more descriptive than the general one below */
- *dst = CFStringCreateWithCString((CFAllocatorRef)NULL, chars, kCFStringEncodingASCII);
- PyMem_Free(chars);
- return 1;
- }
- if (PyUnicode_Check(src)) {
- /* We use the CF types here, if Python was configured differently that will give an error */
- size = PyUnicode_GetSize(src);
- if ((unichars = PyUnicode_AsUnicode(src)) == NULL ) goto err;
- *dst = CFStringCreateWithCharacters((CFAllocatorRef)NULL, unichars, size);
- return 1;
- }
- err:
- PyErr_Format(PyExc_TypeError,
- "Cannot convert %.500s objects to CFString",
- src->ob_type->tp_name);
- return 0;
- }