mirror of https://github.com/python/cpython
304 lines
7.5 KiB
C
304 lines
7.5 KiB
C
/*
|
|
** 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 PyLong_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 (PyLong_Check(src)) {
|
|
long v = PyLong_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;
|
|
}
|