Pillow

1
/*
2
 * The Python Imaging Library
3
 * $Id$
4
 *
5
 * stuff to extract and paste back individual bands
6
 *
7
 * history:
8
 * 1996-03-20 fl   Created
9
 * 1997-08-27 fl   Fixed putband for single band targets.
10
 * 2003-09-26 fl   Fixed getband/putband for 2-band images (LA, PA).
11
 *
12
 * Copyright (c) 1997-2003 by Secret Labs AB.
13
 * Copyright (c) 1996-1997 by Fredrik Lundh.
14
 *
15
 * See the README file for details on usage and redistribution.
16
 */
17

18
#include "Imaging.h"
19

20
Imaging
21
ImagingGetBand(Imaging imIn, int band) {
22
    Imaging imOut;
23
    int x, y;
24

25
    /* Check arguments */
26
    if (!imIn || imIn->type != IMAGING_TYPE_UINT8) {
27
        return (Imaging)ImagingError_ModeError();
28
    }
29

30
    if (band < 0 || band >= imIn->bands) {
31
        return (Imaging)ImagingError_ValueError("band index out of range");
32
    }
33

34
    /* Shortcuts */
35
    if (imIn->bands == 1) {
36
        return ImagingCopy(imIn);
37
    }
38

39
    /* Special case for LXXA etc */
40
    if (imIn->bands == 2 && band == 1) {
41
        band = 3;
42
    }
43

44
    imOut = ImagingNewDirty("L", imIn->xsize, imIn->ysize);
45
    if (!imOut) {
46
        return NULL;
47
    }
48

49
    /* Extract band from image */
50
    for (y = 0; y < imIn->ysize; y++) {
51
        UINT8 *in = (UINT8 *)imIn->image[y] + band;
52
        UINT8 *out = imOut->image8[y];
53
        x = 0;
54
        for (; x < imIn->xsize - 3; x += 4) {
55
            UINT32 v = MAKE_UINT32(in[0], in[4], in[8], in[12]);
56
            memcpy(out + x, &v, sizeof(v));
57
            in += 16;
58
        }
59
        for (; x < imIn->xsize; x++) {
60
            out[x] = *in;
61
            in += 4;
62
        }
63
    }
64

65
    return imOut;
66
}
67

68
int
69
ImagingSplit(Imaging imIn, Imaging bands[4]) {
70
    int i, j, x, y;
71

72
    /* Check arguments */
73
    if (!imIn || imIn->type != IMAGING_TYPE_UINT8) {
74
        (void)ImagingError_ModeError();
75
        return 0;
76
    }
77

78
    /* Shortcuts */
79
    if (imIn->bands == 1) {
80
        bands[0] = ImagingCopy(imIn);
81
        return imIn->bands;
82
    }
83

84
    for (i = 0; i < imIn->bands; i++) {
85
        bands[i] = ImagingNewDirty("L", imIn->xsize, imIn->ysize);
86
        if (!bands[i]) {
87
            for (j = 0; j < i; ++j) {
88
                ImagingDelete(bands[j]);
89
            }
90
            return 0;
91
        }
92
    }
93

94
    /* Extract bands from image */
95
    if (imIn->bands == 2) {
96
        for (y = 0; y < imIn->ysize; y++) {
97
            UINT8 *in = (UINT8 *)imIn->image[y];
98
            UINT8 *out0 = bands[0]->image8[y];
99
            UINT8 *out1 = bands[1]->image8[y];
100
            x = 0;
101
            for (; x < imIn->xsize - 3; x += 4) {
102
                UINT32 v = MAKE_UINT32(in[0], in[4], in[8], in[12]);
103
                memcpy(out0 + x, &v, sizeof(v));
104
                v = MAKE_UINT32(in[0 + 3], in[4 + 3], in[8 + 3], in[12 + 3]);
105
                memcpy(out1 + x, &v, sizeof(v));
106
                in += 16;
107
            }
108
            for (; x < imIn->xsize; x++) {
109
                out0[x] = in[0];
110
                out1[x] = in[3];
111
                in += 4;
112
            }
113
        }
114
    } else if (imIn->bands == 3) {
115
        for (y = 0; y < imIn->ysize; y++) {
116
            UINT8 *in = (UINT8 *)imIn->image[y];
117
            UINT8 *out0 = bands[0]->image8[y];
118
            UINT8 *out1 = bands[1]->image8[y];
119
            UINT8 *out2 = bands[2]->image8[y];
120
            x = 0;
121
            for (; x < imIn->xsize - 3; x += 4) {
122
                UINT32 v = MAKE_UINT32(in[0], in[4], in[8], in[12]);
123
                memcpy(out0 + x, &v, sizeof(v));
124
                v = MAKE_UINT32(in[0 + 1], in[4 + 1], in[8 + 1], in[12 + 1]);
125
                memcpy(out1 + x, &v, sizeof(v));
126
                v = MAKE_UINT32(in[0 + 2], in[4 + 2], in[8 + 2], in[12 + 2]);
127
                memcpy(out2 + x, &v, sizeof(v));
128
                in += 16;
129
            }
130
            for (; x < imIn->xsize; x++) {
131
                out0[x] = in[0];
132
                out1[x] = in[1];
133
                out2[x] = in[2];
134
                in += 4;
135
            }
136
        }
137
    } else {
138
        for (y = 0; y < imIn->ysize; y++) {
139
            UINT8 *in = (UINT8 *)imIn->image[y];
140
            UINT8 *out0 = bands[0]->image8[y];
141
            UINT8 *out1 = bands[1]->image8[y];
142
            UINT8 *out2 = bands[2]->image8[y];
143
            UINT8 *out3 = bands[3]->image8[y];
144
            x = 0;
145
            for (; x < imIn->xsize - 3; x += 4) {
146
                UINT32 v = MAKE_UINT32(in[0], in[4], in[8], in[12]);
147
                memcpy(out0 + x, &v, sizeof(v));
148
                v = MAKE_UINT32(in[0 + 1], in[4 + 1], in[8 + 1], in[12 + 1]);
149
                memcpy(out1 + x, &v, sizeof(v));
150
                v = MAKE_UINT32(in[0 + 2], in[4 + 2], in[8 + 2], in[12 + 2]);
151
                memcpy(out2 + x, &v, sizeof(v));
152
                v = MAKE_UINT32(in[0 + 3], in[4 + 3], in[8 + 3], in[12 + 3]);
153
                memcpy(out3 + x, &v, sizeof(v));
154
                in += 16;
155
            }
156
            for (; x < imIn->xsize; x++) {
157
                out0[x] = in[0];
158
                out1[x] = in[1];
159
                out2[x] = in[2];
160
                out3[x] = in[3];
161
                in += 4;
162
            }
163
        }
164
    }
165

166
    return imIn->bands;
167
}
168

169
Imaging
170
ImagingPutBand(Imaging imOut, Imaging imIn, int band) {
171
    int x, y;
172

173
    /* Check arguments */
174
    if (!imIn || imIn->bands != 1 || !imOut) {
175
        return (Imaging)ImagingError_ModeError();
176
    }
177

178
    if (band < 0 || band >= imOut->bands) {
179
        return (Imaging)ImagingError_ValueError("band index out of range");
180
    }
181

182
    if (imIn->type != imOut->type || imIn->xsize != imOut->xsize ||
183
        imIn->ysize != imOut->ysize) {
184
        return (Imaging)ImagingError_Mismatch();
185
    }
186

187
    /* Shortcuts */
188
    if (imOut->bands == 1) {
189
        return ImagingCopy2(imOut, imIn);
190
    }
191

192
    /* Special case for LXXA etc */
193
    if (imOut->bands == 2 && band == 1) {
194
        band = 3;
195
    }
196

197
    /* Insert band into image */
198
    for (y = 0; y < imIn->ysize; y++) {
199
        UINT8 *in = imIn->image8[y];
200
        UINT8 *out = (UINT8 *)imOut->image[y] + band;
201
        for (x = 0; x < imIn->xsize; x++) {
202
            *out = in[x];
203
            out += 4;
204
        }
205
    }
206

207
    return imOut;
208
}
209

210
Imaging
211
ImagingFillBand(Imaging imOut, int band, int color) {
212
    int x, y;
213

214
    /* Check arguments */
215
    if (!imOut || imOut->type != IMAGING_TYPE_UINT8) {
216
        return (Imaging)ImagingError_ModeError();
217
    }
218

219
    if (band < 0 || band >= imOut->bands) {
220
        return (Imaging)ImagingError_ValueError("band index out of range");
221
    }
222

223
    /* Special case for LXXA etc */
224
    if (imOut->bands == 2 && band == 1) {
225
        band = 3;
226
    }
227

228
    color = CLIP8(color);
229

230
    /* Insert color into image */
231
    for (y = 0; y < imOut->ysize; y++) {
232
        UINT8 *out = (UINT8 *)imOut->image[y] + band;
233
        for (x = 0; x < imOut->xsize; x++) {
234
            *out = (UINT8)color;
235
            out += 4;
236
        }
237
    }
238

239
    return imOut;
240
}
241

242
Imaging
243
ImagingMerge(const char *mode, Imaging bands[4]) {
244
    int i, x, y;
245
    int bandsCount = 0;
246
    Imaging imOut;
247
    Imaging firstBand;
248

249
    firstBand = bands[0];
250
    if (!firstBand) {
251
        return (Imaging)ImagingError_ValueError("wrong number of bands");
252
    }
253

254
    for (i = 0; i < 4; ++i) {
255
        if (!bands[i]) {
256
            break;
257
        }
258
        if (bands[i]->bands != 1) {
259
            return (Imaging)ImagingError_ModeError();
260
        }
261
        if (bands[i]->xsize != firstBand->xsize ||
262
            bands[i]->ysize != firstBand->ysize) {
263
            return (Imaging)ImagingError_Mismatch();
264
        }
265
    }
266
    bandsCount = i;
267

268
    imOut = ImagingNewDirty(mode, firstBand->xsize, firstBand->ysize);
269
    if (!imOut) {
270
        return NULL;
271
    }
272

273
    if (imOut->bands != bandsCount) {
274
        ImagingDelete(imOut);
275
        return (Imaging)ImagingError_ValueError("wrong number of bands");
276
    }
277

278
    if (imOut->bands == 1) {
279
        return ImagingCopy2(imOut, firstBand);
280
    }
281

282
    if (imOut->bands == 2) {
283
        for (y = 0; y < imOut->ysize; y++) {
284
            UINT8 *in0 = bands[0]->image8[y];
285
            UINT8 *in1 = bands[1]->image8[y];
286
            UINT32 *out = (UINT32 *)imOut->image32[y];
287
            for (x = 0; x < imOut->xsize; x++) {
288
                out[x] = MAKE_UINT32(in0[x], 0, 0, in1[x]);
289
            }
290
        }
291
    } else if (imOut->bands == 3) {
292
        for (y = 0; y < imOut->ysize; y++) {
293
            UINT8 *in0 = bands[0]->image8[y];
294
            UINT8 *in1 = bands[1]->image8[y];
295
            UINT8 *in2 = bands[2]->image8[y];
296
            UINT32 *out = (UINT32 *)imOut->image32[y];
297
            for (x = 0; x < imOut->xsize; x++) {
298
                out[x] = MAKE_UINT32(in0[x], in1[x], in2[x], 0);
299
            }
300
        }
301
    } else if (imOut->bands == 4) {
302
        for (y = 0; y < imOut->ysize; y++) {
303
            UINT8 *in0 = bands[0]->image8[y];
304
            UINT8 *in1 = bands[1]->image8[y];
305
            UINT8 *in2 = bands[2]->image8[y];
306
            UINT8 *in3 = bands[3]->image8[y];
307
            UINT32 *out = (UINT32 *)imOut->image32[y];
308
            for (x = 0; x < imOut->xsize; x++) {
309
                out[x] = MAKE_UINT32(in0[x], in1[x], in2[x], in3[x]);
310
            }
311
        }
312
    }
313

314
    return imOut;
315
}
316