1 /*
2  * Copyright (c) 1997, 2014, Oracle and/or its affiliates. All rights reserved.
3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4  *
5  * This code is free software; you can redistribute it and/or modify it
6  * under the terms of the GNU General Public License version 2 only, as
7  * published by the Free Software Foundation.  Oracle designates this
8  * particular file as subject to the "Classpath" exception as provided
9  * by Oracle in the LICENSE file that accompanied this code.
10  *
11  * This code is distributed in the hope that it will be useful, but WITHOUT
12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
14  * version 2 for more details (a copy is included in the LICENSE file that
15  * accompanied this code).
16  *
17  * You should have received a copy of the GNU General Public License version
18  * 2 along with this work; if not, write to the Free Software Foundation,
19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20  *
21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22  * or visit www.oracle.com if you need additional information or have any
23  * questions.
24  */

25
26 /* ****************************************************************
27  ******************************************************************
28  ******************************************************************
29  *** COPYRIGHT (c) Eastman Kodak Company, 1997
30  *** As  an unpublished  work pursuant to Title 17 of the United
31  *** States Code.  All rights reserved.
32  ******************************************************************
33  ******************************************************************
34  ******************************************************************/

35
36 package java.awt.image;
37
38 import java.util.Arrays;
39
40 /**
41  *  This class represents pixel data packed such that the N samples which make
42  *  up a single pixel are stored in a single data array element, and each data
43  *  data array element holds samples for only one pixel.
44  *  This class supports
45  *  {@link DataBuffer#TYPE_BYTE TYPE_BYTE},
46  *  {@link DataBuffer#TYPE_USHORT TYPE_USHORT},
47  *  {@link DataBuffer#TYPE_INT TYPE_INT} data types.
48  *  All data array elements reside
49  *  in the first bank of a DataBuffer.  Accessor methods are provided so
50  *  that the image data can be manipulated directly. Scanline stride is the
51  *  number of data array elements between a given sample and the corresponding
52  *  sample in the same column of the next scanline. Bit masks are the masks
53  *  required to extract the samples representing the bands of the pixel.
54  *  Bit offsets are the offsets in bits into the data array
55  *  element of the samples representing the bands of the pixel.
56  * <p>
57  * The following code illustrates extracting the bits of the sample
58  * representing band {@code b} for pixel {@code x,y}
59  * from DataBuffer {@code data}:
60  * <pre>{@code
61  *      int sample = data.getElem(y * scanlineStride + x);
62  *      sample = (sample & bitMasks[b]) >>> bitOffsets[b];
63  * }</pre>
64  */

65
66 public class SinglePixelPackedSampleModel extends SampleModel
67 {
68     /** Bit masks for all bands of the image data. */
69     private int bitMasks[];
70
71     /** Bit Offsets for all bands of the image data. */
72     private int bitOffsets[];
73
74     /** Bit sizes for all the bands of the image data. */
75     private int bitSizes[];
76
77     /** Maximum bit size. */
78     private int maxBitSize;
79
80     /** Line stride of the region of image data described by this
81      *  SinglePixelPackedSampleModel.
82      */

83     private int scanlineStride;
84
85     private static native void initIDs();
86     static {
87         ColorModel.loadLibraries();
88         initIDs();
89     }
90
91     /**
92      * Constructs a SinglePixelPackedSampleModel with bitMasks.length bands.
93      * Each sample is stored in a data array element in the position of
94      * its corresponding bit mask.  Each bit mask must be contiguous and
95      * masks must not overlap. Bit masks exceeding data type capacity are
96      * truncated.
97      * @param dataType  The data type for storing samples.
98      * @param w         The width (in pixels) of the region of the
99      *                  image data described.
100      * @param h         The height (in pixels) of the region of the
101      *                  image data described.
102      * @param bitMasks  The bit masks for all bands.
103      * @throws IllegalArgumentException if {@code dataType} is not
104      *         either {@code DataBuffer.TYPE_BYTE},
105      *         {@code DataBuffer.TYPE_USHORT}, or
106      *         {@code DataBuffer.TYPE_INT}
107      */

108     public SinglePixelPackedSampleModel(int dataType, int w, int h,
109                                    int bitMasks[]) {
110         this(dataType, w, h, w, bitMasks);
111         if (dataType != DataBuffer.TYPE_BYTE &&
112             dataType != DataBuffer.TYPE_USHORT &&
113             dataType != DataBuffer.TYPE_INT) {
114             throw new IllegalArgumentException("Unsupported data type "+
115                                                dataType);
116         }
117     }
118
119     /**
120      * Constructs a SinglePixelPackedSampleModel with bitMasks.length bands
121      * and a scanline stride equal to scanlineStride data array elements.
122      * Each sample is stored in a data array element in the position of
123      * its corresponding bit mask.  Each bit mask must be contiguous and
124      * masks must not overlap. Bit masks exceeding data type capacity are
125      * truncated.
126      * @param dataType  The data type for storing samples.
127      * @param w         The width (in pixels) of the region of
128      *                  image data described.
129      * @param h         The height (in pixels) of the region of
130      *                  image data described.
131      * @param scanlineStride The line stride of the image data.
132      * @param bitMasks The bit masks for all bands.
133      * @throws IllegalArgumentException if {@code w} or
134      *         {@code h} is not greater than 0
135      * @throws IllegalArgumentException if any mask in
136      *         {@code bitMask} is not contiguous
137      * @throws IllegalArgumentException if {@code dataType} is not
138      *         either {@code DataBuffer.TYPE_BYTE},
139      *         {@code DataBuffer.TYPE_USHORT}, or
140      *         {@code DataBuffer.TYPE_INT}
141      */

142     public SinglePixelPackedSampleModel(int dataType, int w, int h,
143                                    int scanlineStride, int bitMasks[]) {
144         super(dataType, w, h, bitMasks.length);
145         if (dataType != DataBuffer.TYPE_BYTE &&
146             dataType != DataBuffer.TYPE_USHORT &&
147             dataType != DataBuffer.TYPE_INT) {
148             throw new IllegalArgumentException("Unsupported data type "+
149                                                dataType);
150         }
151         this.dataType = dataType;
152         this.bitMasks = bitMasks.clone();
153         this.scanlineStride = scanlineStride;
154
155         this.bitOffsets = new int[numBands];
156         this.bitSizes = new int[numBands];
157
158         int maxMask = (int)((1L << DataBuffer.getDataTypeSize(dataType)) - 1);
159
160         this.maxBitSize = 0;
161         for (int i=0; i<numBands; i++) {
162             int bitOffset = 0, bitSize = 0, mask;
163             this.bitMasks[i] &= maxMask;
164             mask = this.bitMasks[i];
165             if (mask != 0) {
166                 while ((mask & 1) == 0) {
167                     mask = mask >>> 1;
168                     bitOffset++;
169                 }
170                 while ((mask & 1) == 1) {
171                     mask = mask >>> 1;
172                     bitSize++;
173                 }
174                 if (mask != 0) {
175                     throw new IllegalArgumentException("Mask "+bitMasks[i]+
176                                                        " must be contiguous");
177                 }
178             }
179             bitOffsets[i] = bitOffset;
180             bitSizes[i] = bitSize;
181             if (bitSize > maxBitSize) {
182                 maxBitSize = bitSize;
183             }
184         }
185     }
186
187     /**
188      * Returns the number of data elements needed to transfer one pixel
189      * via the getDataElements and setDataElements methods.
190      * For a SinglePixelPackedSampleModel, this is one.
191      */

192     public int getNumDataElements() {
193         return 1;
194     }
195
196     /**
197      * Returns the size of the buffer (in data array elements)
198      * needed for a data buffer that matches this
199      * SinglePixelPackedSampleModel.
200      */

201     private long getBufferSize() {
202       long size = scanlineStride * (height-1) + width;
203       return size;
204     }
205
206     /**
207      * Creates a new SinglePixelPackedSampleModel with the specified
208      * width and height.  The new SinglePixelPackedSampleModel will have the
209      * same storage data type and bit masks as this
210      * SinglePixelPackedSampleModel.
211      * @param w the width of the resulting {@code SampleModel}
212      * @param h the height of the resulting {@code SampleModel}
213      * @return a {@code SinglePixelPackedSampleModel} with the
214      *         specified width and height.
215      * @throws IllegalArgumentException if {@code w} or
216      *         {@code h} is not greater than 0
217      */

218     public SampleModel createCompatibleSampleModel(int w, int h) {
219       SampleModel sampleModel = new SinglePixelPackedSampleModel(dataType, w, h,
220                                                               bitMasks);
221       return sampleModel;
222     }
223
224     /**
225      * Creates a DataBuffer that corresponds to this
226      * SinglePixelPackedSampleModel.  The DataBuffer's data type and size
227      * will be consistent with this SinglePixelPackedSampleModel.  The
228      * DataBuffer will have a single bank.
229      */

230     public DataBuffer createDataBuffer() {
231         DataBuffer dataBuffer = null;
232
233         int size = (int)getBufferSize();
234         switch (dataType) {
235         case DataBuffer.TYPE_BYTE:
236             dataBuffer = new DataBufferByte(size);
237             break;
238         case DataBuffer.TYPE_USHORT:
239             dataBuffer = new DataBufferUShort(size);
240             break;
241         case DataBuffer.TYPE_INT:
242             dataBuffer = new DataBufferInt(size);
243             break;
244         }
245         return dataBuffer;
246     }
247
248     /** Returns the number of bits per sample for all bands. */
249     public int[] getSampleSize() {
250         return bitSizes.clone();
251     }
252
253     /** Returns the number of bits per sample for the specified band. */
254     public int getSampleSize(int band) {
255         return bitSizes[band];
256     }
257
258     /** Returns the offset (in data array elements) of pixel (x,y).
259      *  The data element containing pixel {@code x,y}
260      *  can be retrieved from a DataBuffer {@code data} with a
261      *  SinglePixelPackedSampleModel {@code sppsm} as:
262      * <pre>
263      *        data.getElem(sppsm.getOffset(x, y));
264      * </pre>
265      * @param x the X coordinate of the specified pixel
266      * @param y the Y coordinate of the specified pixel
267      * @return the offset of the specified pixel.
268      */

269     public int getOffset(int x, int y) {
270         int offset = y * scanlineStride + x;
271         return offset;
272     }
273
274     /** Returns the bit offsets into the data array element representing
275      *  a pixel for all bands.
276      *  @return the bit offsets representing a pixel for all bands.
277      */

278     public int [] getBitOffsets() {
279       return bitOffsets.clone();
280     }
281
282     /** Returns the bit masks for all bands.
283      *  @return the bit masks for all bands.
284      */

285     public int [] getBitMasks() {
286       return bitMasks.clone();
287     }
288
289     /** Returns the scanline stride of this SinglePixelPackedSampleModel.
290      *  @return the scanline stride of this
291      *          {@code SinglePixelPackedSampleModel}.
292      */

293     public int getScanlineStride() {
294       return scanlineStride;
295     }
296
297     /**
298      * This creates a new SinglePixelPackedSampleModel with a subset of the
299      * bands of this SinglePixelPackedSampleModel.  The new
300      * SinglePixelPackedSampleModel can be used with any DataBuffer that the
301      * existing SinglePixelPackedSampleModel can be used with.  The new
302      * SinglePixelPackedSampleModel/DataBuffer combination will represent
303      * an image with a subset of the bands of the original
304      * SinglePixelPackedSampleModel/DataBuffer combination.
305      * @exception RasterFormatException if the length of the bands argument is
306      *                                  greater than the number of bands in
307      *                                  the sample model.
308      */

309     public SampleModel createSubsetSampleModel(int bands[]) {
310         if (bands.length > numBands)
311             throw new RasterFormatException("There are only " +
312                                             numBands +
313                                             " bands");
314         int newBitMasks[] = new int[bands.length];
315         for (int i=0; i<bands.length; i++)
316             newBitMasks[i] = bitMasks[bands[i]];
317
318         return new SinglePixelPackedSampleModel(this.dataType, width, height,
319                                            this.scanlineStride, newBitMasks);
320     }
321
322     /**
323      * Returns data for a single pixel in a primitive array of type
324      * TransferType.  For a SinglePixelPackedSampleModel, the array will
325      * have one element, and the type will be the same as the storage
326      * data type.  Generally, obj
327      * should be passed in as null, so that the Object will be created
328      * automatically and will be of the right primitive data type.
329      * <p>
330      * The following code illustrates transferring data for one pixel from
331      * DataBuffer {@code db1}, whose storage layout is described by
332      * SinglePixelPackedSampleModel {@code sppsm1}, to
333      * DataBuffer {@code db2}, whose storage layout is described by
334      * SinglePixelPackedSampleModel {@code sppsm2}.
335      * The transfer will generally be more efficient than using
336      * getPixel/setPixel.
337      * <pre>
338      *       SinglePixelPackedSampleModel sppsm1, sppsm2;
339      *       DataBufferInt db1, db2;
340      *       sppsm2.setDataElements(x, y, sppsm1.getDataElements(x, y, null,
341      *                              db1), db2);
342      * </pre>
343      * Using getDataElements/setDataElements to transfer between two
344      * DataBuffer/SampleModel pairs is legitimate if the SampleModels have
345      * the same number of bands, corresponding bands have the same number of
346      * bits per sample, and the TransferTypes are the same.
347      * <p>
348      * If obj is non-null, it should be a primitive array of type TransferType.
349      * Otherwise, a ClassCastException is thrown.  An
350      * ArrayIndexOutOfBoundsException may be thrown if the coordinates are
351      * not in bounds, or if obj is non-null and is not large enough to hold
352      * the pixel data.
353      * @param x         The X coordinate of the pixel location.
354      * @param y         The Y coordinate of the pixel location.
355      * @param obj       If non-null, a primitive array in which to return
356      *                  the pixel data.
357      * @param data      The DataBuffer containing the image data.
358      * @return the data for the specified pixel.
359      * @see #setDataElements(intint, Object, DataBuffer)
360      */

361     public Object getDataElements(int x, int y, Object obj, DataBuffer data) {
362         // Bounds check for 'b' will be performed automatically
363         if ((x < 0) || (y < 0) || (x >= width) || (y >= height)) {
364             throw new ArrayIndexOutOfBoundsException
365                 ("Coordinate out of bounds!");
366         }
367
368         int type = getTransferType();
369
370         switch(type) {
371
372         case DataBuffer.TYPE_BYTE:
373
374             byte[] bdata;
375
376             if (obj == null)
377                 bdata = new byte[1];
378             else
379                 bdata = (byte[])obj;
380
381             bdata[0] = (byte)data.getElem(y * scanlineStride + x);
382
383             obj = (Object)bdata;
384             break;
385
386         case DataBuffer.TYPE_USHORT:
387
388             short[] sdata;
389
390             if (obj == null)
391                 sdata = new short[1];
392             else
393                 sdata = (short[])obj;
394
395             sdata[0] = (short)data.getElem(y * scanlineStride + x);
396
397             obj = (Object)sdata;
398             break;
399
400         case DataBuffer.TYPE_INT:
401
402             int[] idata;
403
404             if (obj == null)
405                 idata = new int[1];
406             else
407                 idata = (int[])obj;
408
409             idata[0] = data.getElem(y * scanlineStride + x);
410
411             obj = (Object)idata;
412             break;
413         }
414
415         return obj;
416     }
417
418     /**
419      * Returns all samples in for the specified pixel in an int array.
420      * ArrayIndexOutOfBoundsException may be thrown if the coordinates are
421      * not in bounds.
422      * @param x         The X coordinate of the pixel location.
423      * @param y         The Y coordinate of the pixel location.
424      * @param iArray    If non-null, returns the samples in this array
425      * @param data      The DataBuffer containing the image data.
426      * @return all samples for the specified pixel.
427      * @see #setPixel(intintint[], DataBuffer)
428      */

429     public int [] getPixel(int x, int y, int iArray[], DataBuffer data) {
430         if ((x < 0) || (y < 0) || (x >= width) || (y >= height)) {
431             throw new ArrayIndexOutOfBoundsException
432                 ("Coordinate out of bounds!");
433         }
434         int pixels[];
435         if (iArray == null) {
436             pixels = new int [numBands];
437         } else {
438             pixels = iArray;
439         }
440
441         int value = data.getElem(y * scanlineStride + x);
442         for (int i=0; i<numBands; i++) {
443             pixels[i] = (value & bitMasks[i]) >>> bitOffsets[i];
444         }
445         return pixels;
446     }
447
448     /**
449      * Returns all samples for the specified rectangle of pixels in
450      * an int array, one sample per array element.
451      * ArrayIndexOutOfBoundsException may be thrown if the coordinates are
452      * not in bounds.
453      * @param x         The X coordinate of the upper left pixel location.
454      * @param y         The Y coordinate of the upper left pixel location.
455      * @param w         The width of the pixel rectangle.
456      * @param h         The height of the pixel rectangle.
457      * @param iArray    If non-null, returns the samples in this array.
458      * @param data      The DataBuffer containing the image data.
459      * @return all samples for the specified region of pixels.
460      * @see #setPixels(intintintintint[], DataBuffer)
461      */

462     public int[] getPixels(int x, int y, int w, int h,
463                            int iArray[], DataBuffer data) {
464         int x1 = x + w;
465         int y1 = y + h;
466
467         if (x < 0 || x >= width || w > width || x1 < 0 || x1 > width ||
468             y < 0 || y >= height || h > height || y1 < 0 || y1 >  height)
469         {
470             throw new ArrayIndexOutOfBoundsException
471                 ("Coordinate out of bounds!");
472         }
473         int pixels[];
474         if (iArray != null) {
475            pixels = iArray;
476         } else {
477            pixels = new int [w*h*numBands];
478         }
479         int lineOffset = y*scanlineStride + x;
480         int dstOffset = 0;
481
482         for (int i = 0; i < h; i++) {
483            for (int j = 0; j < w; j++) {
484               int value = data.getElem(lineOffset+j);
485               for (int k=0; k < numBands; k++) {
486                   pixels[dstOffset++] =
487                      ((value & bitMasks[k]) >>> bitOffsets[k]);
488               }
489            }
490            lineOffset += scanlineStride;
491         }
492         return pixels;
493     }
494
495     /**
496      * Returns as int the sample in a specified band for the pixel
497      * located at (x,y).
498      * ArrayIndexOutOfBoundsException may be thrown if the coordinates are
499      * not in bounds.
500      * @param x         The X coordinate of the pixel location.
501      * @param y         The Y coordinate of the pixel location.
502      * @param b         The band to return.
503      * @param data      The DataBuffer containing the image data.
504      * @return the sample in a specified band for the specified
505      *         pixel.
506      * @see #setSample(intintintint, DataBuffer)
507      */

508     public int getSample(int x, int y, int b, DataBuffer data) {
509         // Bounds check for 'b' will be performed automatically
510         if ((x < 0) || (y < 0) || (x >= width) || (y >= height)) {
511             throw new ArrayIndexOutOfBoundsException
512                 ("Coordinate out of bounds!");
513         }
514         int sample = data.getElem(y * scanlineStride + x);
515         return ((sample & bitMasks[b]) >>> bitOffsets[b]);
516     }
517
518     /**
519      * Returns the samples for a specified band for the specified rectangle
520      * of pixels in an int array, one sample per array element.
521      * ArrayIndexOutOfBoundsException may be thrown if the coordinates are
522      * not in bounds.
523      * @param x         The X coordinate of the upper left pixel location.
524      * @param y         The Y coordinate of the upper left pixel location.
525      * @param w         The width of the pixel rectangle.
526      * @param h         The height of the pixel rectangle.
527      * @param b         The band to return.
528      * @param iArray    If non-null, returns the samples in this array.
529      * @param data      The DataBuffer containing the image data.
530      * @return the samples for the specified band for the specified
531      *         region of pixels.
532      * @see #setSamples(intintintintintint[], DataBuffer)
533      */

534     public int[] getSamples(int x, int y, int w, int h, int b,
535                            int iArray[], DataBuffer data) {
536         // Bounds check for 'b' will be performed automatically
537         if ((x < 0) || (y < 0) || (x + w > width) || (y + h > height)) {
538             throw new ArrayIndexOutOfBoundsException
539                 ("Coordinate out of bounds!");
540         }
541         int samples[];
542         if (iArray != null) {
543            samples = iArray;
544         } else {
545            samples = new int [w*h];
546         }
547         int lineOffset = y*scanlineStride + x;
548         int dstOffset = 0;
549
550         for (int i = 0; i < h; i++) {
551            for (int j = 0; j < w; j++) {
552               int value = data.getElem(lineOffset+j);
553               samples[dstOffset++] =
554                  ((value & bitMasks[b]) >>> bitOffsets[b]);
555            }
556            lineOffset += scanlineStride;
557         }
558         return samples;
559     }
560
561     /**
562      * Sets the data for a single pixel in the specified DataBuffer from a
563      * primitive array of type TransferType.  For a
564      * SinglePixelPackedSampleModel, only the first element of the array
565      * will hold valid data, and the type of the array must be the same as
566      * the storage data type of the SinglePixelPackedSampleModel.
567      * <p>
568      * The following code illustrates transferring data for one pixel from
569      * DataBuffer {@code db1}, whose storage layout is described by
570      * SinglePixelPackedSampleModel {@code sppsm1},
571      * to DataBuffer {@code db2}, whose storage layout is described by
572      * SinglePixelPackedSampleModel {@code sppsm2}.
573      * The transfer will generally be more efficient than using
574      * getPixel/setPixel.
575      * <pre>
576      *       SinglePixelPackedSampleModel sppsm1, sppsm2;
577      *       DataBufferInt db1, db2;
578      *       sppsm2.setDataElements(x, y, sppsm1.getDataElements(x, y, null,
579      *                              db1), db2);
580      * </pre>
581      * Using getDataElements/setDataElements to transfer between two
582      * DataBuffer/SampleModel pairs is legitimate if the SampleModels have
583      * the same number of bands, corresponding bands have the same number of
584      * bits per sample, and the TransferTypes are the same.
585      * <p>
586      * obj must be a primitive array of type TransferType.  Otherwise,
587      * a ClassCastException is thrown.  An
588      * ArrayIndexOutOfBoundsException may be thrown if the coordinates are
589      * not in bounds, or if obj is not large enough to hold the pixel data.
590      * @param x         The X coordinate of the pixel location.
591      * @param y         The Y coordinate of the pixel location.
592      * @param obj       A primitive array containing pixel data.
593      * @param data      The DataBuffer containing the image data.
594      * @see #getDataElements(intint, Object, DataBuffer)
595      */

596     public void setDataElements(int x, int y, Object obj, DataBuffer data) {
597         if ((x < 0) || (y < 0) || (x >= width) || (y >= height)) {
598             throw new ArrayIndexOutOfBoundsException
599                 ("Coordinate out of bounds!");
600         }
601
602         int type = getTransferType();
603
604         switch(type) {
605
606         case DataBuffer.TYPE_BYTE:
607
608             byte[] barray = (byte[])obj;
609             data.setElem(y*scanlineStride+x, ((int)barray[0])&0xff);
610             break;
611
612         case DataBuffer.TYPE_USHORT:
613
614             short[] sarray = (short[])obj;
615             data.setElem(y*scanlineStride+x, ((int)sarray[0])&0xffff);
616             break;
617
618         case DataBuffer.TYPE_INT:
619
620             int[] iarray = (int[])obj;
621             data.setElem(y*scanlineStride+x, iarray[0]);
622             break;
623         }
624     }
625
626     /**
627      * Sets a pixel in the DataBuffer using an int array of samples for input.
628      * ArrayIndexOutOfBoundsException may be thrown if the coordinates are
629      * not in bounds.
630      * @param x         The X coordinate of the pixel location.
631      * @param y         The Y coordinate of the pixel location.
632      * @param iArray    The input samples in an int array.
633      * @param data      The DataBuffer containing the image data.
634      * @see #getPixel(intintint[], DataBuffer)
635      */

636     public void setPixel(int x, int y,
637                          int iArray[],
638                          DataBuffer data) {
639         if ((x < 0) || (y < 0) || (x >= width) || (y >= height)) {
640             throw new ArrayIndexOutOfBoundsException
641                 ("Coordinate out of bounds!");
642         }
643         int lineOffset = y * scanlineStride + x;
644         int value = data.getElem(lineOffset);
645         for (int i=0; i < numBands; i++) {
646             value &= ~bitMasks[i];
647             value |= ((iArray[i] << bitOffsets[i]) & bitMasks[i]);
648         }
649         data.setElem(lineOffset, value);
650     }
651
652     /**
653      * Sets all samples for a rectangle of pixels from an int array containing
654      * one sample per array element.
655      * ArrayIndexOutOfBoundsException may be thrown if the coordinates are
656      * not in bounds.
657      * @param x         The X coordinate of the upper left pixel location.
658      * @param y         The Y coordinate of the upper left pixel location.
659      * @param w         The width of the pixel rectangle.
660      * @param h         The height of the pixel rectangle.
661      * @param iArray    The input samples in an int array.
662      * @param data      The DataBuffer containing the image data.
663      * @see #getPixels(intintintintint[], DataBuffer)
664      */

665     public void setPixels(int x, int y, int w, int h,
666                           int iArray[], DataBuffer data) {
667         int x1 = x + w;
668         int y1 = y + h;
669
670         if (x < 0 || x >= width || w > width || x1 < 0 || x1 > width ||
671             y < 0 || y >= height || h > height || y1 < 0 || y1 >  height)
672         {
673             throw new ArrayIndexOutOfBoundsException
674                 ("Coordinate out of bounds!");
675         }
676
677         int lineOffset = y*scanlineStride + x;
678         int srcOffset = 0;
679
680         for (int i = 0; i < h; i++) {
681            for (int j = 0; j < w; j++) {
682                int value = data.getElem(lineOffset+j);
683                for (int k=0; k < numBands; k++) {
684                    value &= ~bitMasks[k];
685                    int srcValue = iArray[srcOffset++];
686                    value |= ((srcValue << bitOffsets[k])
687                              & bitMasks[k]);
688                }
689                data.setElem(lineOffset+j, value);
690            }
691            lineOffset += scanlineStride;
692         }
693     }
694
695     /**
696      * Sets a sample in the specified band for the pixel located at (x,y)
697      * in the DataBuffer using an int for input.
698      * ArrayIndexOutOfBoundsException may be thrown if the coordinates are
699      * not in bounds.
700      * @param x         The X coordinate of the pixel location.
701      * @param y         The Y coordinate of the pixel location.
702      * @param b         The band to set.
703      * @param s         The input sample as an int.
704      * @param data      The DataBuffer containing the image data.
705      * @see #getSample(intintint, DataBuffer)
706      */

707     public void setSample(int x, int y, int b, int s,
708                           DataBuffer data) {
709         // Bounds check for 'b' will be performed automatically
710         if ((x < 0) || (y < 0) || (x >= width) || (y >= height)) {
711             throw new ArrayIndexOutOfBoundsException
712                 ("Coordinate out of bounds!");
713         }
714         int value = data.getElem(y*scanlineStride + x);
715         value &= ~bitMasks[b];
716         value |= (s << bitOffsets[b]) & bitMasks[b];
717         data.setElem(y*scanlineStride + x,value);
718     }
719
720     /**
721      * Sets the samples in the specified band for the specified rectangle
722      * of pixels from an int array containing one sample per array element.
723      * ArrayIndexOutOfBoundsException may be thrown if the coordinates are
724      * not in bounds.
725      * @param x         The X coordinate of the upper left pixel location.
726      * @param y         The Y coordinate of the upper left pixel location.
727      * @param w         The width of the pixel rectangle.
728      * @param h         The height of the pixel rectangle.
729      * @param b         The band to set.
730      * @param iArray    The input samples in an int array.
731      * @param data      The DataBuffer containing the image data.
732      * @see #getSamples(intintintintintint[], DataBuffer)
733      */

734     public void setSamples(int x, int y, int w, int h, int b,
735                           int iArray[], DataBuffer data) {
736         // Bounds check for 'b' will be performed automatically
737         if ((x < 0) || (y < 0) || (x + w > width) || (y + h > height)) {
738             throw new ArrayIndexOutOfBoundsException
739                 ("Coordinate out of bounds!");
740         }
741         int lineOffset = y*scanlineStride + x;
742         int srcOffset = 0;
743
744         for (int i = 0; i < h; i++) {
745            for (int j = 0; j < w; j++) {
746               int value = data.getElem(lineOffset+j);
747               value &= ~bitMasks[b];
748               int sample = iArray[srcOffset++];
749               value |= (sample << bitOffsets[b]) & bitMasks[b];
750               data.setElem(lineOffset+j,value);
751            }
752            lineOffset += scanlineStride;
753         }
754     }
755
756     public boolean equals(Object o) {
757         if ((o == null) || !(o instanceof SinglePixelPackedSampleModel)) {
758             return false;
759         }
760
761         SinglePixelPackedSampleModel that = (SinglePixelPackedSampleModel)o;
762         return this.width == that.width &&
763             this.height == that.height &&
764             this.numBands == that.numBands &&
765             this.dataType == that.dataType &&
766             Arrays.equals(this.bitMasks, that.bitMasks) &&
767             Arrays.equals(this.bitOffsets, that.bitOffsets) &&
768             Arrays.equals(this.bitSizes, that.bitSizes) &&
769             this.maxBitSize == that.maxBitSize &&
770             this.scanlineStride == that.scanlineStride;
771     }
772
773     // If we implement equals() we must also implement hashCode
774     public int hashCode() {
775         int hash = 0;
776         hash = width;
777         hash <<= 8;
778         hash ^= height;
779         hash <<= 8;
780         hash ^= numBands;
781         hash <<= 8;
782         hash ^= dataType;
783         hash <<= 8;
784         for (int i = 0; i < bitMasks.length; i++) {
785             hash ^= bitMasks[i];
786             hash <<= 8;
787         }
788         for (int i = 0; i < bitOffsets.length; i++) {
789             hash ^= bitOffsets[i];
790             hash <<= 8;
791         }
792         for (int i = 0; i < bitSizes.length; i++) {
793             hash ^= bitSizes[i];
794             hash <<= 8;
795         }
796         hash ^= maxBitSize;
797         hash <<= 8;
798         hash ^= scanlineStride;
799         return hash;
800     }
801 }
802