← Back to branch summary

~malept/ubuntu/lucid/python2.6/dev-dependency-fix

« back to all changes in this revision

Viewing changes to Mac/Modules/cf/pycfbridge.c

  • Committer: Bazaar Package Importer
  • Author(s): Matthias Klose
  • Date: 2009-02-13 12:51:00 UTC
  • Revision ID: james.westby@ubuntu.com-20090213125100-uufgcb9yeqzujpqw
Tags: upstream-2.6.1
Import upstream version 2.6.1

Show diffs side-by-side

added added

removed removed

Lines of Context:
Mac/Modules/cf/pycfbridge.c
 
1
/*
 
2
** Convert objects from Python to CoreFoundation and vice-versa.
 
3
*/
 
4
 
 
5
#include <CoreServices/CoreServices.h>
 
6
 
 
7
#include "Python.h"
 
8
#include "pymactoolbox.h"
 
9
#include "pycfbridge.h"
 
10
 
 
11
 
 
12
/* ---------------------------------------- */
 
13
/* CoreFoundation objects to Python objects */
 
14
/* ---------------------------------------- */
 
15
 
 
16
PyObject *
 
17
PyCF_CF2Python(CFTypeRef src) {
 
18
        CFTypeID typeid;
 
19
        
 
20
        if( src == NULL ) {
 
21
                Py_INCREF(Py_None);
 
22
                return Py_None;
 
23
        }
 
24
        typeid = CFGetTypeID(src);
 
25
        if (typeid == CFArrayGetTypeID())
 
26
                return PyCF_CF2Python_sequence((CFArrayRef)src);
 
27
        if (typeid == CFDictionaryGetTypeID())
 
28
                return PyCF_CF2Python_mapping((CFDictionaryRef)src);
 
29
        return PyCF_CF2Python_simple(src);
 
30
}
 
31
 
 
32
PyObject *
 
33
PyCF_CF2Python_sequence(CFArrayRef src) {
 
34
        int size = CFArrayGetCount(src);
 
35
        PyObject *rv;
 
36
        CFTypeRef item_cf;
 
37
        PyObject *item_py = NULL;
 
38
        int i;
 
39
        
 
40
        if ( (rv=PyList_New(size)) == NULL )
 
41
                return NULL;
 
42
        for(i=0; i<size; i++) {
 
43
                item_cf = CFArrayGetValueAtIndex(src, i);
 
44
                if (item_cf == NULL ) goto err;
 
45
                item_py = PyCF_CF2Python(item_cf);
 
46
                if (item_py == NULL ) goto err;
 
47
                if (PyList_SetItem(rv, i, item_py) < 0) goto err;
 
48
                item_py = NULL;
 
49
        }
 
50
        return rv;
 
51
err:
 
52
        Py_XDECREF(item_py);
 
53
        Py_DECREF(rv);
 
54
        return NULL;
 
55
}
 
56
 
 
57
PyObject *
 
58
PyCF_CF2Python_mapping(CFTypeRef src) {
 
59
        int size = CFDictionaryGetCount(src);
 
60
        PyObject *rv = NULL;
 
61
        CFTypeRef *allkeys = NULL, *allvalues = NULL;
 
62
        CFTypeRef key_cf, value_cf;
 
63
        PyObject *key_py = NULL, *value_py = NULL;
 
64
        int i;
 
65
        
 
66
        allkeys = malloc(size*sizeof(CFTypeRef *));
 
67
        if (allkeys == NULL) {
 
68
                PyErr_NoMemory();
 
69
                goto err;
 
70
        }
 
71
        allvalues = malloc(size*sizeof(CFTypeRef *));
 
72
        if (allvalues == NULL) {
 
73
                PyErr_NoMemory();
 
74
                goto err;
 
75
        }
 
76
        if ( (rv=PyDict_New()) == NULL ) goto err;
 
77
        CFDictionaryGetKeysAndValues(src, allkeys, allvalues);
 
78
        for(i=0; i<size; i++) {
 
79
                key_cf = allkeys[i];
 
80
                value_cf = allvalues[i];
 
81
                key_py = PyCF_CF2Python(key_cf);
 
82
                if (key_py == NULL ) goto err;
 
83
                value_py = PyCF_CF2Python(value_cf);
 
84
                if (value_py == NULL ) goto err;
 
85
                if (PyDict_SetItem(rv, key_py, value_py) < 0) goto err;
 
86
                key_py = NULL;
 
87
                value_py = NULL;
 
88
        }
 
89
        return rv;
 
90
err:
 
91
        Py_XDECREF(key_py);
 
92
        Py_XDECREF(value_py);
 
93
        Py_XDECREF(rv);
 
94
        free(allkeys);
 
95
        free(allvalues);
 
96
        return NULL;
 
97
}
 
98
 
 
99
PyObject *
 
100
PyCF_CF2Python_simple(CFTypeRef src) {
 
101
        CFTypeID typeid;
 
102
        
 
103
        typeid = CFGetTypeID(src);
 
104
        if (typeid == CFStringGetTypeID())
 
105
                return PyCF_CF2Python_string((CFStringRef)src);
 
106
        if (typeid == CFBooleanGetTypeID())
 
107
                return PyBool_FromLong((long)CFBooleanGetValue(src));
 
108
        if (typeid == CFNumberGetTypeID()) {
 
109
                if (CFNumberIsFloatType(src)) {
 
110
                        double d;
 
111
                        CFNumberGetValue(src, kCFNumberDoubleType, &d);
 
112
                        return PyFloat_FromDouble(d);
 
113
                } else {
 
114
                        long l;
 
115
                        if (!CFNumberGetValue(src, kCFNumberLongType, &l))
 
116
                                /* XXXX Out of range! */;
 
117
                        return PyInt_FromLong(l);
 
118
                }
 
119
        }
 
120
        /* XXXX Should return as CFTypeRef, really... */
 
121
        PyMac_Error(resNotFound);
 
122
        return NULL;
 
123
}
 
124
 
 
125
/* Unsure - Return unicode or 8 bit strings? */
 
126
PyObject *
 
127
PyCF_CF2Python_string(CFStringRef src) {
 
128
        int size = CFStringGetLength(src)+1;
 
129
        Py_UNICODE *data = malloc(size*sizeof(Py_UNICODE));
 
130
        CFRange range;
 
131
        PyObject *rv;
 
132
 
 
133
        range.location = 0;
 
134
        range.length = size;
 
135
        if( data == NULL ) return PyErr_NoMemory();
 
136
        CFStringGetCharacters(src, range, data);
 
137
        rv = (PyObject *)PyUnicode_FromUnicode(data, size-1);
 
138
        free(data);
 
139
        return rv;
 
140
}
 
141
 
 
142
/* ---------------------------------------- */
 
143
/* Python objects to CoreFoundation objects */
 
144
/* ---------------------------------------- */
 
145
 
 
146
int
 
147
PyCF_Python2CF(PyObject *src, CFTypeRef *dst) {
 
148
 
 
149
        if (PyString_Check(src) || PyUnicode_Check(src))
 
150
                return PyCF_Python2CF_simple(src, dst);
 
151
        if (PySequence_Check(src))
 
152
                return PyCF_Python2CF_sequence(src, (CFArrayRef *)dst);
 
153
        if (PyMapping_Check(src))
 
154
                return PyCF_Python2CF_mapping(src, (CFDictionaryRef *)dst);
 
155
        return PyCF_Python2CF_simple(src, dst);
 
156
}
 
157
 
 
158
int
 
159
PyCF_Python2CF_sequence(PyObject *src, CFArrayRef *dst) {
 
160
        CFMutableArrayRef rv = NULL;
 
161
        CFTypeRef item_cf = NULL;
 
162
        PyObject *item_py = NULL;
 
163
        int size, i;
 
164
        
 
165
        if( !PySequence_Check(src) ) {
 
166
                PyErr_Format(PyExc_TypeError,
 
167
                        "Cannot convert %.500s objects to CFArray",
 
168
                        src->ob_type->tp_name);
 
169
                return 0;
 
170
        }
 
171
        size = PySequence_Size(src);
 
172
        rv = CFArrayCreateMutable((CFAllocatorRef)NULL, size, &kCFTypeArrayCallBacks);
 
173
        if (rv == NULL) {
 
174
                PyMac_Error(resNotFound); 
 
175
                goto err;
 
176
        }
 
177
 
 
178
        for( i=0; i<size; i++) {
 
179
                item_py = PySequence_GetItem(src, i);
 
180
                if (item_py == NULL) goto err;
 
181
                if ( !PyCF_Python2CF(item_py, &item_cf)) goto err;
 
182
                Py_DECREF(item_py);
 
183
                CFArraySetValueAtIndex(rv, i, item_cf);
 
184
                CFRelease(item_cf);
 
185
                item_cf = NULL;
 
186
        }
 
187
        *dst = rv;
 
188
        return 1;
 
189
err:
 
190
        Py_XDECREF(item_py);
 
191
        if (rv) CFRelease(rv);
 
192
        if (item_cf) CFRelease(item_cf);
 
193
        return 0;               
 
194
}
 
195
 
 
196
int
 
197
PyCF_Python2CF_mapping(PyObject *src, CFDictionaryRef *dst) {
 
198
        CFMutableDictionaryRef rv = NULL;
 
199
        PyObject *aslist = NULL;
 
200
        CFTypeRef key_cf = NULL, value_cf = NULL;
 
201
        PyObject *item_py = NULL, *key_py = NULL, *value_py = NULL;
 
202
        int size, i;
 
203
        
 
204
        if( !PyMapping_Check(src) ) {
 
205
                PyErr_Format(PyExc_TypeError,
 
206
                        "Cannot convert %.500s objects to CFDictionary",
 
207
                        src->ob_type->tp_name);
 
208
                return 0;
 
209
        }
 
210
        size = PyMapping_Size(src);
 
211
        rv = CFDictionaryCreateMutable((CFAllocatorRef)NULL, size,
 
212
                                        &kCFTypeDictionaryKeyCallBacks,
 
213
                                        &kCFTypeDictionaryValueCallBacks);
 
214
        if (rv == NULL) {
 
215
                PyMac_Error(resNotFound); 
 
216
                goto err;
 
217
        }
 
218
        if ( (aslist = PyMapping_Items(src)) == NULL ) goto err;
 
219
 
 
220
        for( i=0; i<size; i++) {
 
221
                item_py = PySequence_GetItem(aslist, i);
 
222
                if (item_py == NULL) goto err;
 
223
                if (!PyArg_ParseTuple(item_py, "OO", &key_py, &value_py)) goto err;
 
224
                if ( !PyCF_Python2CF(key_py, &key_cf) ) goto err;
 
225
                if ( !PyCF_Python2CF(value_py, &value_cf) ) goto err;
 
226
                CFDictionaryAddValue(rv, key_cf, value_cf);
 
227
                CFRelease(key_cf);
 
228
                key_cf = NULL;
 
229
                CFRelease(value_cf);
 
230
                value_cf = NULL;
 
231
        }
 
232
        *dst = rv;
 
233
        return 1;
 
234
err:
 
235
        Py_XDECREF(item_py);
 
236
        Py_XDECREF(aslist);
 
237
        if (rv) CFRelease(rv);
 
238
        if (key_cf) CFRelease(key_cf);
 
239
        if (value_cf) CFRelease(value_cf);
 
240
        return 0;               
 
241
}
 
242
 
 
243
int
 
244
PyCF_Python2CF_simple(PyObject *src, CFTypeRef *dst) {
 
245
        
 
246
#if 0
 
247
        if (PyObject_HasAttrString(src, "CFType")) {
 
248
                *dst = PyObject_CallMethod(src, "CFType", "");
 
249
                return (*dst != NULL);
 
250
        }
 
251
#endif
 
252
        if (PyString_Check(src) || PyUnicode_Check(src)) 
 
253
                return PyCF_Python2CF_string(src, (CFStringRef *)dst);
 
254
        if (PyBool_Check(src)) {
 
255
                if (src == Py_True)
 
256
                        *dst = kCFBooleanTrue;
 
257
                else
 
258
                        *dst = kCFBooleanFalse;
 
259
                return 1;
 
260
        }
 
261
        if (PyInt_Check(src)) {
 
262
                long v = PyInt_AsLong(src);
 
263
                *dst = CFNumberCreate(NULL, kCFNumberLongType, &v);
 
264
                return 1;
 
265
        }
 
266
        if (PyFloat_Check(src)) {
 
267
                double d = PyFloat_AsDouble(src);
 
268
                *dst = CFNumberCreate(NULL, kCFNumberDoubleType, &d);
 
269
                return 1;
 
270
        }
 
271
                        
 
272
        PyErr_Format(PyExc_TypeError,
 
273
                  "Cannot convert %.500s objects to CFType",
 
274
                                     src->ob_type->tp_name);
 
275
        return 0;
 
276
}
 
277
 
 
278
int
 
279
PyCF_Python2CF_string(PyObject *src, CFStringRef *dst) {
 
280
        char *chars;
 
281
        CFIndex size;
 
282
        UniChar *unichars;
 
283
        
 
284
        if (PyString_Check(src)) {
 
285
                if (!PyArg_Parse(src, "es", "ascii", &chars))
 
286
                        return 0; /* This error is more descriptive than the general one below */
 
287
                *dst = CFStringCreateWithCString((CFAllocatorRef)NULL, chars, kCFStringEncodingASCII);
 
288
                PyMem_Free(chars);
 
289
                return 1;
 
290
        }
 
291
        if (PyUnicode_Check(src)) {
 
292
                /* We use the CF types here, if Python was configured differently that will give an error */
 
293
                size = PyUnicode_GetSize(src);
 
294
                if ((unichars = PyUnicode_AsUnicode(src)) == NULL ) goto err;
 
295
                *dst = CFStringCreateWithCharacters((CFAllocatorRef)NULL, unichars, size);
 
296
                return 1;
 
297
        }
 
298
err:
 
299
        PyErr_Format(PyExc_TypeError,
 
300
                  "Cannot convert %.500s objects to CFString",
 
301
                                     src->ob_type->tp_name);
 
302
        return 0;
 
303
}