1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
#include <Python.h>
#ifdef HAVE_STAT
#undef HAVE_STAT
#endif
//#include "/usr/include/python2.3/Imaging.h"
#include "../../config.h"
#ifdef HAVE_PYTHON_IMAGING
#include <Imaging.h>
#endif
#include "pyutils.h"
#undef HAVE_STAT
#include "../rfxswf.h"
/* redefine the ImagingObject struct defined in _imagingmodule.c */
/* there should be a better way to do this... */
typedef struct {
PyObject_HEAD
#ifdef HAVE_PYTHON_IMAGING
Imaging image;
#endif
} ImagingObject;
int image_getWidth(PyObject*_image) {
#ifdef HAVE_PYTHON_IMAGING
if(strcmp(_image->ob_type->tp_name, "ImagingCore")) {
PyErr_SetString(PyExc_Exception, setError("not an image: %s", _image->ob_type->tp_name));
return 0;
}
ImagingObject*image = (ImagingObject*)_image;
return image->image->xsize;
#else
PyErr_SetString(PyExc_Exception, "imaging not compiled in");
return 0;
#endif
}
int image_getHeight(PyObject*_image) {
#ifdef HAVE_PYTHON_IMAGING
if(strcmp(_image->ob_type->tp_name, "ImagingCore")) {
PyErr_SetString(PyExc_Exception, setError("not an image: %s", _image->ob_type->tp_name));
return 0;
}
ImagingObject*image = (ImagingObject*)_image;
return image->image->ysize;
#else
PyErr_SetString(PyExc_Exception, "imaging not compiled in");
return 0;
#endif
}
int image_getBPP(PyObject*_image) {
#ifdef HAVE_PYTHON_IMAGING
if(strcmp(_image->ob_type->tp_name, "ImagingCore")) {
PyErr_SetString(PyExc_Exception, setError("not an image: %s", _image->ob_type->tp_name));
return 0;
}
ImagingObject*image = (ImagingObject*)_image;
if(!strcmp(image->image->mode, "1") ||
!strcmp(image->image->mode, "L") ||
!strcmp(image->image->mode, "P")) {
return 8;
}
if(!strcmp(image->image->mode, "I") ||
!strcmp(image->image->mode, "F")) {
return 32;
}
if(!strcmp(image->image->mode, "RGB") ||
!strcmp(image->image->mode, "RGBA") ||
!strcmp(image->image->mode, "CMYK") ||
!strcmp(image->image->mode, "YCbCr")) {
return 32;
}
PyErr_SetString(PyExc_Exception, setError("Unknown image format (%s).", image->image->mode));
return 0;
#else
PyErr_SetString(PyExc_Exception, "imaging not compiled in");
return 0;
#endif
}
RGBA* image_toRGBA(PyObject*_image)
{
#ifdef HAVE_PYTHON_IMAGING
if(strcmp(_image->ob_type->tp_name, "ImagingCore")) {
PyErr_SetString(PyExc_Exception, setError("not an image: %s", _image->ob_type->tp_name));
return 0;
}
ImagingObject*image = (ImagingObject*)_image;
printf("mode: %s\n", image->image->mode);
printf("depth: %d\n", image->image->depth);
printf("bands: %d\n", image->image->bands);
printf("xsize: %d\n", image->image->xsize);
printf("ysize: %d\n", image->image->ysize);
int bpp = image_getBPP(_image);
if(!bpp)
return 0;
RGBA*rgba = (RGBA*)malloc(image->image->xsize * image->image->ysize * sizeof(RGBA));
if(!strcmp(image->image->mode, "RGBA")) {
int y,ymax=image->image->ysize;
int width = image->image->xsize;
RGBA*dest = rgba;
for(y=0;y<ymax;y++) {
U8* src = (U8*)(image->image->image32[y]);
int x;
for(x=0;x<width;x++) {
dest[x].r = src[x*4+0];
dest[x].g = src[x*4+1];
dest[x].b = src[x*4+2];
dest[x].a = src[x*4+3];
}
dest+=width;
}
return rgba;
}
PyErr_SetString(PyExc_Exception, setError("Unsupported image format: %s (try .convert(\"RGBA\")", image->image->mode));
#else
PyErr_SetString(PyExc_Exception, "imaging not compiled in");
#endif
return 0;
}
#ifdef HAVE_PYTHON_IMAGING
extern PyObject*PyImagingNew(Imaging imOut);
#endif
PyObject* rgba_to_image(RGBA*rgba, int width, int height)
{
#ifdef HAVE_PYTHON_IMAGING
#ifndef WIN32
Imaging img = ImagingNew("RGBA", width, height);
int y;
if(!img->image32) {
fprintf(stderr, "No array allocated!\n");
return 0;
}
for(y=0;y<height;y++) {
U8* dest = (U8*)(img->image32[y]);
RGBA* src = &rgba[width*y];
int x;
for(x=0;x<width;x++) {
dest[x+0] = src[x].r;
dest[x+1] = src[x].g;
dest[x+2] = src[x].b;
dest[x+3] = src[x].a;
}
}
return PyImagingNew(img);
#else
fprintf(stderr, "This image extraction is not yet supported on non-linux systems\n");
return 0;
#endif
#else
PyErr_SetString(PyExc_Exception, "imaging not compiled in");
return 0;
#endif
}
|
