|
1 /* |
|
2 * Copyright (c) 2000, 2015, 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 package com.sun.imageio.plugins.jpeg; |
|
27 |
|
28 import javax.imageio.IIOException; |
|
29 import javax.imageio.ImageWriter; |
|
30 import javax.imageio.ImageWriteParam; |
|
31 import javax.imageio.IIOImage; |
|
32 import javax.imageio.ImageTypeSpecifier; |
|
33 import javax.imageio.metadata.IIOMetadata; |
|
34 import javax.imageio.metadata.IIOMetadataFormatImpl; |
|
35 import javax.imageio.metadata.IIOInvalidTreeException; |
|
36 import javax.imageio.spi.ImageWriterSpi; |
|
37 import javax.imageio.stream.ImageOutputStream; |
|
38 import javax.imageio.plugins.jpeg.JPEGImageWriteParam; |
|
39 import javax.imageio.plugins.jpeg.JPEGQTable; |
|
40 import javax.imageio.plugins.jpeg.JPEGHuffmanTable; |
|
41 |
|
42 import org.w3c.dom.Node; |
|
43 |
|
44 import java.awt.image.Raster; |
|
45 import java.awt.image.WritableRaster; |
|
46 import java.awt.image.DataBufferByte; |
|
47 import java.awt.image.ColorModel; |
|
48 import java.awt.image.IndexColorModel; |
|
49 import java.awt.image.ColorConvertOp; |
|
50 import java.awt.image.RenderedImage; |
|
51 import java.awt.image.BufferedImage; |
|
52 import java.awt.color.ColorSpace; |
|
53 import java.awt.color.ICC_ColorSpace; |
|
54 import java.awt.color.ICC_Profile; |
|
55 import java.awt.Dimension; |
|
56 import java.awt.Rectangle; |
|
57 import java.awt.Transparency; |
|
58 |
|
59 import java.io.IOException; |
|
60 |
|
61 import java.util.List; |
|
62 import java.util.ArrayList; |
|
63 import java.util.Iterator; |
|
64 |
|
65 import sun.java2d.Disposer; |
|
66 import sun.java2d.DisposerRecord; |
|
67 |
|
68 public class JPEGImageWriter extends ImageWriter { |
|
69 |
|
70 ///////// Private variables |
|
71 |
|
72 private boolean debug = false; |
|
73 |
|
74 /** |
|
75 * The following variable contains a pointer to the IJG library |
|
76 * structure for this reader. It is assigned in the constructor |
|
77 * and then is passed in to every native call. It is set to 0 |
|
78 * by dispose to avoid disposing twice. |
|
79 */ |
|
80 private long structPointer = 0; |
|
81 |
|
82 |
|
83 /** The output stream we write to */ |
|
84 private ImageOutputStream ios = null; |
|
85 |
|
86 /** The Raster we will write from */ |
|
87 private Raster srcRas = null; |
|
88 |
|
89 /** An intermediate Raster holding compressor-friendly data */ |
|
90 private WritableRaster raster = null; |
|
91 |
|
92 /** |
|
93 * Set to true if we are writing an image with an |
|
94 * indexed ColorModel |
|
95 */ |
|
96 private boolean indexed = false; |
|
97 private IndexColorModel indexCM = null; |
|
98 |
|
99 private boolean convertTosRGB = false; // Used by PhotoYCC only |
|
100 private WritableRaster converted = null; |
|
101 |
|
102 private boolean isAlphaPremultiplied = false; |
|
103 private ColorModel srcCM = null; |
|
104 |
|
105 /** |
|
106 * If there are thumbnails to be written, this is the list. |
|
107 */ |
|
108 private List<? extends BufferedImage> thumbnails = null; |
|
109 |
|
110 /** |
|
111 * If metadata should include an icc profile, store it here. |
|
112 */ |
|
113 private ICC_Profile iccProfile = null; |
|
114 |
|
115 private int sourceXOffset = 0; |
|
116 private int sourceYOffset = 0; |
|
117 private int sourceWidth = 0; |
|
118 private int [] srcBands = null; |
|
119 private int sourceHeight = 0; |
|
120 |
|
121 /** Used when calling listeners */ |
|
122 private int currentImage = 0; |
|
123 |
|
124 private ColorConvertOp convertOp = null; |
|
125 |
|
126 private JPEGQTable [] streamQTables = null; |
|
127 private JPEGHuffmanTable[] streamDCHuffmanTables = null; |
|
128 private JPEGHuffmanTable[] streamACHuffmanTables = null; |
|
129 |
|
130 // Parameters for writing metadata |
|
131 private boolean ignoreJFIF = false; // If it's there, use it |
|
132 private boolean forceJFIF = false; // Add one for the thumbnails |
|
133 private boolean ignoreAdobe = false; // If it's there, use it |
|
134 private int newAdobeTransform = JPEG.ADOBE_IMPOSSIBLE; // Change if needed |
|
135 private boolean writeDefaultJFIF = false; |
|
136 private boolean writeAdobe = false; |
|
137 private JPEGMetadata metadata = null; |
|
138 |
|
139 private boolean sequencePrepared = false; |
|
140 |
|
141 private int numScans = 0; |
|
142 |
|
143 /** The referent to be registered with the Disposer. */ |
|
144 private Object disposerReferent = new Object(); |
|
145 |
|
146 /** The DisposerRecord that handles the actual disposal of this writer. */ |
|
147 private DisposerRecord disposerRecord; |
|
148 |
|
149 ///////// End of Private variables |
|
150 |
|
151 ///////// Protected variables |
|
152 |
|
153 protected static final int WARNING_DEST_IGNORED = 0; |
|
154 protected static final int WARNING_STREAM_METADATA_IGNORED = 1; |
|
155 protected static final int WARNING_DEST_METADATA_COMP_MISMATCH = 2; |
|
156 protected static final int WARNING_DEST_METADATA_JFIF_MISMATCH = 3; |
|
157 protected static final int WARNING_DEST_METADATA_ADOBE_MISMATCH = 4; |
|
158 protected static final int WARNING_IMAGE_METADATA_JFIF_MISMATCH = 5; |
|
159 protected static final int WARNING_IMAGE_METADATA_ADOBE_MISMATCH = 6; |
|
160 protected static final int WARNING_METADATA_NOT_JPEG_FOR_RASTER = 7; |
|
161 protected static final int WARNING_NO_BANDS_ON_INDEXED = 8; |
|
162 protected static final int WARNING_ILLEGAL_THUMBNAIL = 9; |
|
163 protected static final int WARNING_IGNORING_THUMBS = 10; |
|
164 protected static final int WARNING_FORCING_JFIF = 11; |
|
165 protected static final int WARNING_THUMB_CLIPPED = 12; |
|
166 protected static final int WARNING_METADATA_ADJUSTED_FOR_THUMB = 13; |
|
167 protected static final int WARNING_NO_RGB_THUMB_AS_INDEXED = 14; |
|
168 protected static final int WARNING_NO_GRAY_THUMB_AS_INDEXED = 15; |
|
169 |
|
170 private static final int MAX_WARNING = WARNING_NO_GRAY_THUMB_AS_INDEXED; |
|
171 |
|
172 ///////// End of Protected variables |
|
173 |
|
174 ///////// static initializer |
|
175 |
|
176 static { |
|
177 java.security.AccessController.doPrivileged( |
|
178 new java.security.PrivilegedAction<Void>() { |
|
179 public Void run() { |
|
180 System.loadLibrary("javajpeg"); |
|
181 return null; |
|
182 } |
|
183 }); |
|
184 initWriterIDs(JPEGQTable.class, |
|
185 JPEGHuffmanTable.class); |
|
186 } |
|
187 |
|
188 //////// Public API |
|
189 |
|
190 public JPEGImageWriter(ImageWriterSpi originator) { |
|
191 super(originator); |
|
192 structPointer = initJPEGImageWriter(); |
|
193 disposerRecord = new JPEGWriterDisposerRecord(structPointer); |
|
194 Disposer.addRecord(disposerReferent, disposerRecord); |
|
195 } |
|
196 |
|
197 public void setOutput(Object output) { |
|
198 setThreadLock(); |
|
199 try { |
|
200 cbLock.check(); |
|
201 |
|
202 super.setOutput(output); // validates output |
|
203 resetInternalState(); |
|
204 ios = (ImageOutputStream) output; // so this will always work |
|
205 // Set the native destination |
|
206 setDest(structPointer); |
|
207 } finally { |
|
208 clearThreadLock(); |
|
209 } |
|
210 } |
|
211 |
|
212 public ImageWriteParam getDefaultWriteParam() { |
|
213 return new JPEGImageWriteParam(null); |
|
214 } |
|
215 |
|
216 public IIOMetadata getDefaultStreamMetadata(ImageWriteParam param) { |
|
217 setThreadLock(); |
|
218 try { |
|
219 return new JPEGMetadata(param, this); |
|
220 } finally { |
|
221 clearThreadLock(); |
|
222 } |
|
223 } |
|
224 |
|
225 public IIOMetadata |
|
226 getDefaultImageMetadata(ImageTypeSpecifier imageType, |
|
227 ImageWriteParam param) { |
|
228 setThreadLock(); |
|
229 try { |
|
230 return new JPEGMetadata(imageType, param, this); |
|
231 } finally { |
|
232 clearThreadLock(); |
|
233 } |
|
234 } |
|
235 |
|
236 public IIOMetadata convertStreamMetadata(IIOMetadata inData, |
|
237 ImageWriteParam param) { |
|
238 // There isn't much we can do. If it's one of ours, then |
|
239 // return it. Otherwise just return null. We use it only |
|
240 // for tables, so we can't get a default and modify it, |
|
241 // as this will usually not be what is intended. |
|
242 if (inData instanceof JPEGMetadata) { |
|
243 JPEGMetadata jpegData = (JPEGMetadata) inData; |
|
244 if (jpegData.isStream) { |
|
245 return inData; |
|
246 } |
|
247 } |
|
248 return null; |
|
249 } |
|
250 |
|
251 public IIOMetadata |
|
252 convertImageMetadata(IIOMetadata inData, |
|
253 ImageTypeSpecifier imageType, |
|
254 ImageWriteParam param) { |
|
255 setThreadLock(); |
|
256 try { |
|
257 return convertImageMetadataOnThread(inData, imageType, param); |
|
258 } finally { |
|
259 clearThreadLock(); |
|
260 } |
|
261 } |
|
262 |
|
263 private IIOMetadata |
|
264 convertImageMetadataOnThread(IIOMetadata inData, |
|
265 ImageTypeSpecifier imageType, |
|
266 ImageWriteParam param) { |
|
267 // If it's one of ours, just return it |
|
268 if (inData instanceof JPEGMetadata) { |
|
269 JPEGMetadata jpegData = (JPEGMetadata) inData; |
|
270 if (!jpegData.isStream) { |
|
271 return inData; |
|
272 } else { |
|
273 // Can't convert stream metadata to image metadata |
|
274 // XXX Maybe this should put out a warning? |
|
275 return null; |
|
276 } |
|
277 } |
|
278 // If it's not one of ours, create a default and set it from |
|
279 // the standard tree from the input, if it exists. |
|
280 if (inData.isStandardMetadataFormatSupported()) { |
|
281 String formatName = |
|
282 IIOMetadataFormatImpl.standardMetadataFormatName; |
|
283 Node tree = inData.getAsTree(formatName); |
|
284 if (tree != null) { |
|
285 JPEGMetadata jpegData = new JPEGMetadata(imageType, |
|
286 param, |
|
287 this); |
|
288 try { |
|
289 jpegData.setFromTree(formatName, tree); |
|
290 } catch (IIOInvalidTreeException e) { |
|
291 // Other plug-in generates bogus standard tree |
|
292 // XXX Maybe this should put out a warning? |
|
293 return null; |
|
294 } |
|
295 |
|
296 return jpegData; |
|
297 } |
|
298 } |
|
299 return null; |
|
300 } |
|
301 |
|
302 public int getNumThumbnailsSupported(ImageTypeSpecifier imageType, |
|
303 ImageWriteParam param, |
|
304 IIOMetadata streamMetadata, |
|
305 IIOMetadata imageMetadata) { |
|
306 if (jfifOK(imageType, param, streamMetadata, imageMetadata)) { |
|
307 return Integer.MAX_VALUE; |
|
308 } |
|
309 return 0; |
|
310 } |
|
311 |
|
312 static final Dimension [] preferredThumbSizes = {new Dimension(1, 1), |
|
313 new Dimension(255, 255)}; |
|
314 |
|
315 public Dimension[] getPreferredThumbnailSizes(ImageTypeSpecifier imageType, |
|
316 ImageWriteParam param, |
|
317 IIOMetadata streamMetadata, |
|
318 IIOMetadata imageMetadata) { |
|
319 if (jfifOK(imageType, param, streamMetadata, imageMetadata)) { |
|
320 return preferredThumbSizes.clone(); |
|
321 } |
|
322 return null; |
|
323 } |
|
324 |
|
325 private boolean jfifOK(ImageTypeSpecifier imageType, |
|
326 ImageWriteParam param, |
|
327 IIOMetadata streamMetadata, |
|
328 IIOMetadata imageMetadata) { |
|
329 // If the image type and metadata are JFIF compatible, return true |
|
330 if ((imageType != null) && |
|
331 (!JPEG.isJFIFcompliant(imageType, true))) { |
|
332 return false; |
|
333 } |
|
334 if (imageMetadata != null) { |
|
335 JPEGMetadata metadata = null; |
|
336 if (imageMetadata instanceof JPEGMetadata) { |
|
337 metadata = (JPEGMetadata) imageMetadata; |
|
338 } else { |
|
339 metadata = (JPEGMetadata)convertImageMetadata(imageMetadata, |
|
340 imageType, |
|
341 param); |
|
342 } |
|
343 // metadata must have a jfif node |
|
344 if (metadata.findMarkerSegment |
|
345 (JFIFMarkerSegment.class, true) == null){ |
|
346 return false; |
|
347 } |
|
348 } |
|
349 return true; |
|
350 } |
|
351 |
|
352 public boolean canWriteRasters() { |
|
353 return true; |
|
354 } |
|
355 |
|
356 public void write(IIOMetadata streamMetadata, |
|
357 IIOImage image, |
|
358 ImageWriteParam param) throws IOException { |
|
359 setThreadLock(); |
|
360 try { |
|
361 cbLock.check(); |
|
362 |
|
363 writeOnThread(streamMetadata, image, param); |
|
364 } finally { |
|
365 clearThreadLock(); |
|
366 } |
|
367 } |
|
368 |
|
369 private void writeOnThread(IIOMetadata streamMetadata, |
|
370 IIOImage image, |
|
371 ImageWriteParam param) throws IOException { |
|
372 |
|
373 if (ios == null) { |
|
374 throw new IllegalStateException("Output has not been set!"); |
|
375 } |
|
376 |
|
377 if (image == null) { |
|
378 throw new IllegalArgumentException("image is null!"); |
|
379 } |
|
380 |
|
381 // if streamMetadata is not null, issue a warning |
|
382 if (streamMetadata != null) { |
|
383 warningOccurred(WARNING_STREAM_METADATA_IGNORED); |
|
384 } |
|
385 |
|
386 // Obtain the raster and image, if there is one |
|
387 boolean rasterOnly = image.hasRaster(); |
|
388 |
|
389 RenderedImage rimage = null; |
|
390 if (rasterOnly) { |
|
391 srcRas = image.getRaster(); |
|
392 } else { |
|
393 rimage = image.getRenderedImage(); |
|
394 if (rimage instanceof BufferedImage) { |
|
395 // Use the Raster directly. |
|
396 srcRas = ((BufferedImage)rimage).getRaster(); |
|
397 } else if (rimage.getNumXTiles() == 1 && |
|
398 rimage.getNumYTiles() == 1) |
|
399 { |
|
400 // Get the unique tile. |
|
401 srcRas = rimage.getTile(rimage.getMinTileX(), |
|
402 rimage.getMinTileY()); |
|
403 |
|
404 // Ensure the Raster has dimensions of the image, |
|
405 // as the tile dimensions might differ. |
|
406 if (srcRas.getWidth() != rimage.getWidth() || |
|
407 srcRas.getHeight() != rimage.getHeight()) |
|
408 { |
|
409 srcRas = srcRas.createChild(srcRas.getMinX(), |
|
410 srcRas.getMinY(), |
|
411 rimage.getWidth(), |
|
412 rimage.getHeight(), |
|
413 srcRas.getMinX(), |
|
414 srcRas.getMinY(), |
|
415 null); |
|
416 } |
|
417 } else { |
|
418 // Image is tiled so get a contiguous raster by copying. |
|
419 srcRas = rimage.getData(); |
|
420 } |
|
421 } |
|
422 |
|
423 // Now determine if we are using a band subset |
|
424 |
|
425 // By default, we are using all source bands |
|
426 int numSrcBands = srcRas.getNumBands(); |
|
427 indexed = false; |
|
428 indexCM = null; |
|
429 ColorModel cm = null; |
|
430 ColorSpace cs = null; |
|
431 isAlphaPremultiplied = false; |
|
432 srcCM = null; |
|
433 if (!rasterOnly) { |
|
434 cm = rimage.getColorModel(); |
|
435 if (cm != null) { |
|
436 cs = cm.getColorSpace(); |
|
437 if (cm instanceof IndexColorModel) { |
|
438 indexed = true; |
|
439 indexCM = (IndexColorModel) cm; |
|
440 numSrcBands = cm.getNumComponents(); |
|
441 } |
|
442 if (cm.isAlphaPremultiplied()) { |
|
443 isAlphaPremultiplied = true; |
|
444 srcCM = cm; |
|
445 } |
|
446 } |
|
447 } |
|
448 |
|
449 srcBands = JPEG.bandOffsets[numSrcBands-1]; |
|
450 int numBandsUsed = numSrcBands; |
|
451 // Consult the param to determine if we're writing a subset |
|
452 |
|
453 if (param != null) { |
|
454 int[] sBands = param.getSourceBands(); |
|
455 if (sBands != null) { |
|
456 if (indexed) { |
|
457 warningOccurred(WARNING_NO_BANDS_ON_INDEXED); |
|
458 } else { |
|
459 srcBands = sBands; |
|
460 numBandsUsed = srcBands.length; |
|
461 if (numBandsUsed > numSrcBands) { |
|
462 throw new IIOException |
|
463 ("ImageWriteParam specifies too many source bands"); |
|
464 } |
|
465 } |
|
466 } |
|
467 } |
|
468 |
|
469 boolean usingBandSubset = (numBandsUsed != numSrcBands); |
|
470 boolean fullImage = ((!rasterOnly) && (!usingBandSubset)); |
|
471 |
|
472 int [] bandSizes = null; |
|
473 if (!indexed) { |
|
474 bandSizes = srcRas.getSampleModel().getSampleSize(); |
|
475 // If this is a subset, we must adjust bandSizes |
|
476 if (usingBandSubset) { |
|
477 int [] temp = new int [numBandsUsed]; |
|
478 for (int i = 0; i < numBandsUsed; i++) { |
|
479 temp[i] = bandSizes[srcBands[i]]; |
|
480 } |
|
481 bandSizes = temp; |
|
482 } |
|
483 } else { |
|
484 int [] tempSize = srcRas.getSampleModel().getSampleSize(); |
|
485 bandSizes = new int [numSrcBands]; |
|
486 for (int i = 0; i < numSrcBands; i++) { |
|
487 bandSizes[i] = tempSize[0]; // All the same |
|
488 } |
|
489 } |
|
490 |
|
491 for (int i = 0; i < bandSizes.length; i++) { |
|
492 // 4450894 part 1: The IJG libraries are compiled so they only |
|
493 // handle <= 8-bit samples. We now check the band sizes and throw |
|
494 // an exception for images, such as USHORT_GRAY, with > 8 bits |
|
495 // per sample. |
|
496 if (bandSizes[i] <= 0 || bandSizes[i] > 8) { |
|
497 throw new IIOException("Illegal band size: should be 0 < size <= 8"); |
|
498 } |
|
499 // 4450894 part 2: We expand IndexColorModel images to full 24- |
|
500 // or 32-bit in grabPixels() for each scanline. For indexed |
|
501 // images such as BYTE_BINARY, we need to ensure that we update |
|
502 // bandSizes to account for the scaling from 1-bit band sizes |
|
503 // to 8-bit. |
|
504 if (indexed) { |
|
505 bandSizes[i] = 8; |
|
506 } |
|
507 } |
|
508 |
|
509 if (debug) { |
|
510 System.out.println("numSrcBands is " + numSrcBands); |
|
511 System.out.println("numBandsUsed is " + numBandsUsed); |
|
512 System.out.println("usingBandSubset is " + usingBandSubset); |
|
513 System.out.println("fullImage is " + fullImage); |
|
514 System.out.print("Band sizes:"); |
|
515 for (int i = 0; i< bandSizes.length; i++) { |
|
516 System.out.print(" " + bandSizes[i]); |
|
517 } |
|
518 System.out.println(); |
|
519 } |
|
520 |
|
521 // Destination type, if there is one |
|
522 ImageTypeSpecifier destType = null; |
|
523 if (param != null) { |
|
524 destType = param.getDestinationType(); |
|
525 // Ignore dest type if we are writing a complete image |
|
526 if ((fullImage) && (destType != null)) { |
|
527 warningOccurred(WARNING_DEST_IGNORED); |
|
528 destType = null; |
|
529 } |
|
530 } |
|
531 |
|
532 // Examine the param |
|
533 |
|
534 sourceXOffset = srcRas.getMinX(); |
|
535 sourceYOffset = srcRas.getMinY(); |
|
536 int imageWidth = srcRas.getWidth(); |
|
537 int imageHeight = srcRas.getHeight(); |
|
538 sourceWidth = imageWidth; |
|
539 sourceHeight = imageHeight; |
|
540 int periodX = 1; |
|
541 int periodY = 1; |
|
542 int gridX = 0; |
|
543 int gridY = 0; |
|
544 JPEGQTable [] qTables = null; |
|
545 JPEGHuffmanTable[] DCHuffmanTables = null; |
|
546 JPEGHuffmanTable[] ACHuffmanTables = null; |
|
547 boolean optimizeHuffman = false; |
|
548 JPEGImageWriteParam jparam = null; |
|
549 int progressiveMode = ImageWriteParam.MODE_DISABLED; |
|
550 |
|
551 if (param != null) { |
|
552 |
|
553 Rectangle sourceRegion = param.getSourceRegion(); |
|
554 if (sourceRegion != null) { |
|
555 Rectangle imageBounds = new Rectangle(sourceXOffset, |
|
556 sourceYOffset, |
|
557 sourceWidth, |
|
558 sourceHeight); |
|
559 sourceRegion = sourceRegion.intersection(imageBounds); |
|
560 sourceXOffset = sourceRegion.x; |
|
561 sourceYOffset = sourceRegion.y; |
|
562 sourceWidth = sourceRegion.width; |
|
563 sourceHeight = sourceRegion.height; |
|
564 } |
|
565 |
|
566 if (sourceWidth + sourceXOffset > imageWidth) { |
|
567 sourceWidth = imageWidth - sourceXOffset; |
|
568 } |
|
569 if (sourceHeight + sourceYOffset > imageHeight) { |
|
570 sourceHeight = imageHeight - sourceYOffset; |
|
571 } |
|
572 |
|
573 periodX = param.getSourceXSubsampling(); |
|
574 periodY = param.getSourceYSubsampling(); |
|
575 gridX = param.getSubsamplingXOffset(); |
|
576 gridY = param.getSubsamplingYOffset(); |
|
577 |
|
578 switch(param.getCompressionMode()) { |
|
579 case ImageWriteParam.MODE_DISABLED: |
|
580 throw new IIOException("JPEG compression cannot be disabled"); |
|
581 case ImageWriteParam.MODE_EXPLICIT: |
|
582 float quality = param.getCompressionQuality(); |
|
583 quality = JPEG.convertToLinearQuality(quality); |
|
584 qTables = new JPEGQTable[2]; |
|
585 qTables[0] = JPEGQTable.K1Luminance.getScaledInstance |
|
586 (quality, true); |
|
587 qTables[1] = JPEGQTable.K2Chrominance.getScaledInstance |
|
588 (quality, true); |
|
589 break; |
|
590 case ImageWriteParam.MODE_DEFAULT: |
|
591 qTables = new JPEGQTable[2]; |
|
592 qTables[0] = JPEGQTable.K1Div2Luminance; |
|
593 qTables[1] = JPEGQTable.K2Div2Chrominance; |
|
594 break; |
|
595 // We'll handle the metadata case later |
|
596 } |
|
597 |
|
598 progressiveMode = param.getProgressiveMode(); |
|
599 |
|
600 if (param instanceof JPEGImageWriteParam) { |
|
601 jparam = (JPEGImageWriteParam)param; |
|
602 optimizeHuffman = jparam.getOptimizeHuffmanTables(); |
|
603 } |
|
604 } |
|
605 |
|
606 // Now examine the metadata |
|
607 IIOMetadata mdata = image.getMetadata(); |
|
608 if (mdata != null) { |
|
609 if (mdata instanceof JPEGMetadata) { |
|
610 metadata = (JPEGMetadata) mdata; |
|
611 if (debug) { |
|
612 System.out.println |
|
613 ("We have metadata, and it's JPEG metadata"); |
|
614 } |
|
615 } else { |
|
616 if (!rasterOnly) { |
|
617 ImageTypeSpecifier type = destType; |
|
618 if (type == null) { |
|
619 type = new ImageTypeSpecifier(rimage); |
|
620 } |
|
621 metadata = (JPEGMetadata) convertImageMetadata(mdata, |
|
622 type, |
|
623 param); |
|
624 } else { |
|
625 warningOccurred(WARNING_METADATA_NOT_JPEG_FOR_RASTER); |
|
626 } |
|
627 } |
|
628 } |
|
629 |
|
630 // First set a default state |
|
631 |
|
632 ignoreJFIF = false; // If it's there, use it |
|
633 ignoreAdobe = false; // If it's there, use it |
|
634 newAdobeTransform = JPEG.ADOBE_IMPOSSIBLE; // Change if needed |
|
635 writeDefaultJFIF = false; |
|
636 writeAdobe = false; |
|
637 |
|
638 // By default we'll do no conversion: |
|
639 int inCsType = JPEG.JCS_UNKNOWN; |
|
640 int outCsType = JPEG.JCS_UNKNOWN; |
|
641 |
|
642 JFIFMarkerSegment jfif = null; |
|
643 AdobeMarkerSegment adobe = null; |
|
644 SOFMarkerSegment sof = null; |
|
645 |
|
646 if (metadata != null) { |
|
647 jfif = (JFIFMarkerSegment) metadata.findMarkerSegment |
|
648 (JFIFMarkerSegment.class, true); |
|
649 adobe = (AdobeMarkerSegment) metadata.findMarkerSegment |
|
650 (AdobeMarkerSegment.class, true); |
|
651 sof = (SOFMarkerSegment) metadata.findMarkerSegment |
|
652 (SOFMarkerSegment.class, true); |
|
653 } |
|
654 |
|
655 iccProfile = null; // By default don't write one |
|
656 convertTosRGB = false; // PhotoYCC does this |
|
657 converted = null; |
|
658 |
|
659 if (destType != null) { |
|
660 if (numBandsUsed != destType.getNumBands()) { |
|
661 throw new IIOException |
|
662 ("Number of source bands != number of destination bands"); |
|
663 } |
|
664 cs = destType.getColorModel().getColorSpace(); |
|
665 // Check the metadata against the destination type |
|
666 if (metadata != null) { |
|
667 checkSOFBands(sof, numBandsUsed); |
|
668 |
|
669 checkJFIF(jfif, destType, false); |
|
670 // Do we want to write an ICC profile? |
|
671 if ((jfif != null) && (ignoreJFIF == false)) { |
|
672 if (JPEG.isNonStandardICC(cs)) { |
|
673 iccProfile = ((ICC_ColorSpace) cs).getProfile(); |
|
674 } |
|
675 } |
|
676 checkAdobe(adobe, destType, false); |
|
677 |
|
678 } else { // no metadata, but there is a dest type |
|
679 // If we can add a JFIF or an Adobe marker segment, do so |
|
680 if (JPEG.isJFIFcompliant(destType, false)) { |
|
681 writeDefaultJFIF = true; |
|
682 // Do we want to write an ICC profile? |
|
683 if (JPEG.isNonStandardICC(cs)) { |
|
684 iccProfile = ((ICC_ColorSpace) cs).getProfile(); |
|
685 } |
|
686 } else { |
|
687 int transform = JPEG.transformForType(destType, false); |
|
688 if (transform != JPEG.ADOBE_IMPOSSIBLE) { |
|
689 writeAdobe = true; |
|
690 newAdobeTransform = transform; |
|
691 } |
|
692 } |
|
693 // re-create the metadata |
|
694 metadata = new JPEGMetadata(destType, null, this); |
|
695 } |
|
696 inCsType = getSrcCSType(destType); |
|
697 outCsType = getDefaultDestCSType(destType); |
|
698 } else { // no destination type |
|
699 if (metadata == null) { |
|
700 if (fullImage) { // no dest, no metadata, full image |
|
701 // Use default metadata matching the image and param |
|
702 metadata = new JPEGMetadata(new ImageTypeSpecifier(rimage), |
|
703 param, this); |
|
704 if (metadata.findMarkerSegment |
|
705 (JFIFMarkerSegment.class, true) != null) { |
|
706 cs = rimage.getColorModel().getColorSpace(); |
|
707 if (JPEG.isNonStandardICC(cs)) { |
|
708 iccProfile = ((ICC_ColorSpace) cs).getProfile(); |
|
709 } |
|
710 } |
|
711 |
|
712 inCsType = getSrcCSType(rimage); |
|
713 outCsType = getDefaultDestCSType(rimage); |
|
714 } |
|
715 // else no dest, no metadata, not an image, |
|
716 // so no special headers, no color conversion |
|
717 } else { // no dest type, but there is metadata |
|
718 checkSOFBands(sof, numBandsUsed); |
|
719 if (fullImage) { // no dest, metadata, image |
|
720 // Check that the metadata and the image match |
|
721 |
|
722 ImageTypeSpecifier inputType = |
|
723 new ImageTypeSpecifier(rimage); |
|
724 |
|
725 inCsType = getSrcCSType(rimage); |
|
726 |
|
727 if (cm != null) { |
|
728 boolean alpha = cm.hasAlpha(); |
|
729 switch (cs.getType()) { |
|
730 case ColorSpace.TYPE_GRAY: |
|
731 if (!alpha) { |
|
732 outCsType = JPEG.JCS_GRAYSCALE; |
|
733 } else { |
|
734 if (jfif != null) { |
|
735 ignoreJFIF = true; |
|
736 warningOccurred |
|
737 (WARNING_IMAGE_METADATA_JFIF_MISMATCH); |
|
738 } |
|
739 // out colorspace remains unknown |
|
740 } |
|
741 if ((adobe != null) |
|
742 && (adobe.transform != JPEG.ADOBE_UNKNOWN)) { |
|
743 newAdobeTransform = JPEG.ADOBE_UNKNOWN; |
|
744 warningOccurred |
|
745 (WARNING_IMAGE_METADATA_ADOBE_MISMATCH); |
|
746 } |
|
747 break; |
|
748 case ColorSpace.TYPE_RGB: |
|
749 if (!alpha) { |
|
750 if (jfif != null) { |
|
751 outCsType = JPEG.JCS_YCbCr; |
|
752 if (JPEG.isNonStandardICC(cs) |
|
753 || ((cs instanceof ICC_ColorSpace) |
|
754 && (jfif.iccSegment != null))) { |
|
755 iccProfile = |
|
756 ((ICC_ColorSpace) cs).getProfile(); |
|
757 } |
|
758 } else if (adobe != null) { |
|
759 switch (adobe.transform) { |
|
760 case JPEG.ADOBE_UNKNOWN: |
|
761 outCsType = JPEG.JCS_RGB; |
|
762 break; |
|
763 case JPEG.ADOBE_YCC: |
|
764 outCsType = JPEG.JCS_YCbCr; |
|
765 break; |
|
766 default: |
|
767 warningOccurred |
|
768 (WARNING_IMAGE_METADATA_ADOBE_MISMATCH); |
|
769 newAdobeTransform = JPEG.ADOBE_UNKNOWN; |
|
770 outCsType = JPEG.JCS_RGB; |
|
771 break; |
|
772 } |
|
773 } else { |
|
774 // consult the ids |
|
775 int outCS = sof.getIDencodedCSType(); |
|
776 // if they don't resolve it, |
|
777 // consult the sampling factors |
|
778 if (outCS != JPEG.JCS_UNKNOWN) { |
|
779 outCsType = outCS; |
|
780 } else { |
|
781 boolean subsampled = |
|
782 isSubsampled(sof.componentSpecs); |
|
783 if (subsampled) { |
|
784 outCsType = JPEG.JCS_YCbCr; |
|
785 } else { |
|
786 outCsType = JPEG.JCS_RGB; |
|
787 } |
|
788 } |
|
789 } |
|
790 } else { // RGBA |
|
791 if (jfif != null) { |
|
792 ignoreJFIF = true; |
|
793 warningOccurred |
|
794 (WARNING_IMAGE_METADATA_JFIF_MISMATCH); |
|
795 } |
|
796 if (adobe != null) { |
|
797 if (adobe.transform |
|
798 != JPEG.ADOBE_UNKNOWN) { |
|
799 newAdobeTransform = JPEG.ADOBE_UNKNOWN; |
|
800 warningOccurred |
|
801 (WARNING_IMAGE_METADATA_ADOBE_MISMATCH); |
|
802 } |
|
803 outCsType = JPEG.JCS_RGBA; |
|
804 } else { |
|
805 // consult the ids |
|
806 int outCS = sof.getIDencodedCSType(); |
|
807 // if they don't resolve it, |
|
808 // consult the sampling factors |
|
809 if (outCS != JPEG.JCS_UNKNOWN) { |
|
810 outCsType = outCS; |
|
811 } else { |
|
812 boolean subsampled = |
|
813 isSubsampled(sof.componentSpecs); |
|
814 outCsType = subsampled ? |
|
815 JPEG.JCS_YCbCrA : JPEG.JCS_RGBA; |
|
816 } |
|
817 } |
|
818 } |
|
819 break; |
|
820 case ColorSpace.TYPE_3CLR: |
|
821 if (cs == JPEG.JCS.getYCC()) { |
|
822 if (!alpha) { |
|
823 if (jfif != null) { |
|
824 convertTosRGB = true; |
|
825 convertOp = |
|
826 new ColorConvertOp(cs, |
|
827 JPEG.JCS.sRGB, |
|
828 null); |
|
829 outCsType = JPEG.JCS_YCbCr; |
|
830 } else if (adobe != null) { |
|
831 if (adobe.transform |
|
832 != JPEG.ADOBE_YCC) { |
|
833 newAdobeTransform = JPEG.ADOBE_YCC; |
|
834 warningOccurred |
|
835 (WARNING_IMAGE_METADATA_ADOBE_MISMATCH); |
|
836 } |
|
837 outCsType = JPEG.JCS_YCC; |
|
838 } else { |
|
839 outCsType = JPEG.JCS_YCC; |
|
840 } |
|
841 } else { // PhotoYCCA |
|
842 if (jfif != null) { |
|
843 ignoreJFIF = true; |
|
844 warningOccurred |
|
845 (WARNING_IMAGE_METADATA_JFIF_MISMATCH); |
|
846 } else if (adobe != null) { |
|
847 if (adobe.transform |
|
848 != JPEG.ADOBE_UNKNOWN) { |
|
849 newAdobeTransform |
|
850 = JPEG.ADOBE_UNKNOWN; |
|
851 warningOccurred |
|
852 (WARNING_IMAGE_METADATA_ADOBE_MISMATCH); |
|
853 } |
|
854 } |
|
855 outCsType = JPEG.JCS_YCCA; |
|
856 } |
|
857 } |
|
858 } |
|
859 } |
|
860 } // else no dest, metadata, not an image. Defaults ok |
|
861 } |
|
862 } |
|
863 |
|
864 boolean metadataProgressive = false; |
|
865 int [] scans = null; |
|
866 |
|
867 if (metadata != null) { |
|
868 if (sof == null) { |
|
869 sof = (SOFMarkerSegment) metadata.findMarkerSegment |
|
870 (SOFMarkerSegment.class, true); |
|
871 } |
|
872 if ((sof != null) && (sof.tag == JPEG.SOF2)) { |
|
873 metadataProgressive = true; |
|
874 if (progressiveMode == ImageWriteParam.MODE_COPY_FROM_METADATA) { |
|
875 scans = collectScans(metadata, sof); // Might still be null |
|
876 } else { |
|
877 numScans = 0; |
|
878 } |
|
879 } |
|
880 if (jfif == null) { |
|
881 jfif = (JFIFMarkerSegment) metadata.findMarkerSegment |
|
882 (JFIFMarkerSegment.class, true); |
|
883 } |
|
884 } |
|
885 |
|
886 thumbnails = image.getThumbnails(); |
|
887 int numThumbs = image.getNumThumbnails(); |
|
888 forceJFIF = false; |
|
889 // determine if thumbnails can be written |
|
890 // If we are going to add a default JFIF marker segment, |
|
891 // then thumbnails can be written |
|
892 if (!writeDefaultJFIF) { |
|
893 // If there is no metadata, then we can't write thumbnails |
|
894 if (metadata == null) { |
|
895 thumbnails = null; |
|
896 if (numThumbs != 0) { |
|
897 warningOccurred(WARNING_IGNORING_THUMBS); |
|
898 } |
|
899 } else { |
|
900 // There is metadata |
|
901 // If we are writing a raster or subbands, |
|
902 // then the user must specify JFIF on the metadata |
|
903 if (fullImage == false) { |
|
904 if (jfif == null) { |
|
905 thumbnails = null; // Or we can't include thumbnails |
|
906 if (numThumbs != 0) { |
|
907 warningOccurred(WARNING_IGNORING_THUMBS); |
|
908 } |
|
909 } |
|
910 } else { // It is a full image, and there is metadata |
|
911 if (jfif == null) { // Not JFIF |
|
912 // Can it have JFIF? |
|
913 if ((outCsType == JPEG.JCS_GRAYSCALE) |
|
914 || (outCsType == JPEG.JCS_YCbCr)) { |
|
915 if (numThumbs != 0) { |
|
916 forceJFIF = true; |
|
917 warningOccurred(WARNING_FORCING_JFIF); |
|
918 } |
|
919 } else { // Nope, not JFIF-compatible |
|
920 thumbnails = null; |
|
921 if (numThumbs != 0) { |
|
922 warningOccurred(WARNING_IGNORING_THUMBS); |
|
923 } |
|
924 } |
|
925 } |
|
926 } |
|
927 } |
|
928 } |
|
929 |
|
930 // Set up a boolean to indicate whether we need to call back to |
|
931 // write metadata |
|
932 boolean haveMetadata = |
|
933 ((metadata != null) || writeDefaultJFIF || writeAdobe); |
|
934 |
|
935 // Now that we have dealt with metadata, finalize our tables set up |
|
936 |
|
937 // Are we going to write tables? By default, yes. |
|
938 boolean writeDQT = true; |
|
939 boolean writeDHT = true; |
|
940 |
|
941 // But if the metadata has no tables, no. |
|
942 DQTMarkerSegment dqt = null; |
|
943 DHTMarkerSegment dht = null; |
|
944 |
|
945 int restartInterval = 0; |
|
946 |
|
947 if (metadata != null) { |
|
948 dqt = (DQTMarkerSegment) metadata.findMarkerSegment |
|
949 (DQTMarkerSegment.class, true); |
|
950 dht = (DHTMarkerSegment) metadata.findMarkerSegment |
|
951 (DHTMarkerSegment.class, true); |
|
952 DRIMarkerSegment dri = |
|
953 (DRIMarkerSegment) metadata.findMarkerSegment |
|
954 (DRIMarkerSegment.class, true); |
|
955 if (dri != null) { |
|
956 restartInterval = dri.restartInterval; |
|
957 } |
|
958 |
|
959 if (dqt == null) { |
|
960 writeDQT = false; |
|
961 } |
|
962 if (dht == null) { |
|
963 writeDHT = false; // Ignored if optimizeHuffman is true |
|
964 } |
|
965 } |
|
966 |
|
967 // Whether we write tables or not, we need to figure out which ones |
|
968 // to use |
|
969 if (qTables == null) { // Get them from metadata, or use defaults |
|
970 if (dqt != null) { |
|
971 qTables = collectQTablesFromMetadata(metadata); |
|
972 } else if (streamQTables != null) { |
|
973 qTables = streamQTables; |
|
974 } else if ((jparam != null) && (jparam.areTablesSet())) { |
|
975 qTables = jparam.getQTables(); |
|
976 } else { |
|
977 qTables = JPEG.getDefaultQTables(); |
|
978 } |
|
979 |
|
980 } |
|
981 |
|
982 // If we are optimizing, we don't want any tables. |
|
983 if (optimizeHuffman == false) { |
|
984 // If they were for progressive scans, we can't use them. |
|
985 if ((dht != null) && (metadataProgressive == false)) { |
|
986 DCHuffmanTables = collectHTablesFromMetadata(metadata, true); |
|
987 ACHuffmanTables = collectHTablesFromMetadata(metadata, false); |
|
988 } else if (streamDCHuffmanTables != null) { |
|
989 DCHuffmanTables = streamDCHuffmanTables; |
|
990 ACHuffmanTables = streamACHuffmanTables; |
|
991 } else if ((jparam != null) && (jparam.areTablesSet())) { |
|
992 DCHuffmanTables = jparam.getDCHuffmanTables(); |
|
993 ACHuffmanTables = jparam.getACHuffmanTables(); |
|
994 } else { |
|
995 DCHuffmanTables = JPEG.getDefaultHuffmanTables(true); |
|
996 ACHuffmanTables = JPEG.getDefaultHuffmanTables(false); |
|
997 } |
|
998 } |
|
999 |
|
1000 // By default, ids are 1 - N, no subsampling |
|
1001 int [] componentIds = new int[numBandsUsed]; |
|
1002 int [] HsamplingFactors = new int[numBandsUsed]; |
|
1003 int [] VsamplingFactors = new int[numBandsUsed]; |
|
1004 int [] QtableSelectors = new int[numBandsUsed]; |
|
1005 for (int i = 0; i < numBandsUsed; i++) { |
|
1006 componentIds[i] = i+1; // JFIF compatible |
|
1007 HsamplingFactors[i] = 1; |
|
1008 VsamplingFactors[i] = 1; |
|
1009 QtableSelectors[i] = 0; |
|
1010 } |
|
1011 |
|
1012 // Now override them with the contents of sof, if there is one, |
|
1013 if (sof != null) { |
|
1014 for (int i = 0; i < numBandsUsed; i++) { |
|
1015 if (forceJFIF == false) { // else use JFIF-compatible default |
|
1016 componentIds[i] = sof.componentSpecs[i].componentId; |
|
1017 } |
|
1018 HsamplingFactors[i] = sof.componentSpecs[i].HsamplingFactor; |
|
1019 VsamplingFactors[i] = sof.componentSpecs[i].VsamplingFactor; |
|
1020 QtableSelectors[i] = sof.componentSpecs[i].QtableSelector; |
|
1021 } |
|
1022 } |
|
1023 |
|
1024 sourceXOffset += gridX; |
|
1025 sourceWidth -= gridX; |
|
1026 sourceYOffset += gridY; |
|
1027 sourceHeight -= gridY; |
|
1028 |
|
1029 int destWidth = (sourceWidth + periodX - 1)/periodX; |
|
1030 int destHeight = (sourceHeight + periodY - 1)/periodY; |
|
1031 |
|
1032 // Create an appropriate 1-line databuffer for writing |
|
1033 int lineSize = sourceWidth*numBandsUsed; |
|
1034 |
|
1035 DataBufferByte buffer = new DataBufferByte(lineSize); |
|
1036 |
|
1037 // Create a raster from that |
|
1038 int [] bandOffs = JPEG.bandOffsets[numBandsUsed-1]; |
|
1039 |
|
1040 raster = Raster.createInterleavedRaster(buffer, |
|
1041 sourceWidth, 1, |
|
1042 lineSize, |
|
1043 numBandsUsed, |
|
1044 bandOffs, |
|
1045 null); |
|
1046 |
|
1047 // Call the writer, who will call back for every scanline |
|
1048 |
|
1049 clearAbortRequest(); |
|
1050 cbLock.lock(); |
|
1051 try { |
|
1052 processImageStarted(currentImage); |
|
1053 } finally { |
|
1054 cbLock.unlock(); |
|
1055 } |
|
1056 |
|
1057 boolean aborted = false; |
|
1058 |
|
1059 if (debug) { |
|
1060 System.out.println("inCsType: " + inCsType); |
|
1061 System.out.println("outCsType: " + outCsType); |
|
1062 } |
|
1063 |
|
1064 // Note that getData disables acceleration on buffer, but it is |
|
1065 // just a 1-line intermediate data transfer buffer that does not |
|
1066 // affect the acceleration of the source image. |
|
1067 aborted = writeImage(structPointer, |
|
1068 buffer.getData(), |
|
1069 inCsType, outCsType, |
|
1070 numBandsUsed, |
|
1071 bandSizes, |
|
1072 sourceWidth, |
|
1073 destWidth, destHeight, |
|
1074 periodX, periodY, |
|
1075 qTables, |
|
1076 writeDQT, |
|
1077 DCHuffmanTables, |
|
1078 ACHuffmanTables, |
|
1079 writeDHT, |
|
1080 optimizeHuffman, |
|
1081 (progressiveMode |
|
1082 != ImageWriteParam.MODE_DISABLED), |
|
1083 numScans, |
|
1084 scans, |
|
1085 componentIds, |
|
1086 HsamplingFactors, |
|
1087 VsamplingFactors, |
|
1088 QtableSelectors, |
|
1089 haveMetadata, |
|
1090 restartInterval); |
|
1091 |
|
1092 cbLock.lock(); |
|
1093 try { |
|
1094 if (aborted) { |
|
1095 processWriteAborted(); |
|
1096 } else { |
|
1097 processImageComplete(); |
|
1098 } |
|
1099 |
|
1100 ios.flush(); |
|
1101 } finally { |
|
1102 cbLock.unlock(); |
|
1103 } |
|
1104 currentImage++; // After a successful write |
|
1105 } |
|
1106 |
|
1107 @Override |
|
1108 public boolean canWriteSequence() { |
|
1109 return true; |
|
1110 } |
|
1111 |
|
1112 public void prepareWriteSequence(IIOMetadata streamMetadata) |
|
1113 throws IOException { |
|
1114 setThreadLock(); |
|
1115 try { |
|
1116 cbLock.check(); |
|
1117 |
|
1118 prepareWriteSequenceOnThread(streamMetadata); |
|
1119 } finally { |
|
1120 clearThreadLock(); |
|
1121 } |
|
1122 } |
|
1123 |
|
1124 private void prepareWriteSequenceOnThread(IIOMetadata streamMetadata) |
|
1125 throws IOException { |
|
1126 if (ios == null) { |
|
1127 throw new IllegalStateException("Output has not been set!"); |
|
1128 } |
|
1129 |
|
1130 /* |
|
1131 * from jpeg_metadata.html: |
|
1132 * If no stream metadata is supplied to |
|
1133 * {@code ImageWriter.prepareWriteSequence}, then no |
|
1134 * tables-only image is written. If stream metadata containing |
|
1135 * no tables is supplied to |
|
1136 * {@code ImageWriter.prepareWriteSequence}, then a tables-only |
|
1137 * image containing default visually lossless tables is written. |
|
1138 */ |
|
1139 if (streamMetadata != null) { |
|
1140 if (streamMetadata instanceof JPEGMetadata) { |
|
1141 // write a complete tables-only image at the beginning of |
|
1142 // the stream. |
|
1143 JPEGMetadata jmeta = (JPEGMetadata) streamMetadata; |
|
1144 if (jmeta.isStream == false) { |
|
1145 throw new IllegalArgumentException |
|
1146 ("Invalid stream metadata object."); |
|
1147 } |
|
1148 // Check that we are |
|
1149 // at the beginning of the stream, or can go there, and haven't |
|
1150 // written out the metadata already. |
|
1151 if (currentImage != 0) { |
|
1152 throw new IIOException |
|
1153 ("JPEG Stream metadata must precede all images"); |
|
1154 } |
|
1155 if (sequencePrepared == true) { |
|
1156 throw new IIOException("Stream metadata already written!"); |
|
1157 } |
|
1158 |
|
1159 // Set the tables |
|
1160 // If the metadata has no tables, use default tables. |
|
1161 streamQTables = collectQTablesFromMetadata(jmeta); |
|
1162 if (debug) { |
|
1163 System.out.println("after collecting from stream metadata, " |
|
1164 + "streamQTables.length is " |
|
1165 + streamQTables.length); |
|
1166 } |
|
1167 if (streamQTables == null) { |
|
1168 streamQTables = JPEG.getDefaultQTables(); |
|
1169 } |
|
1170 streamDCHuffmanTables = |
|
1171 collectHTablesFromMetadata(jmeta, true); |
|
1172 if (streamDCHuffmanTables == null) { |
|
1173 streamDCHuffmanTables = JPEG.getDefaultHuffmanTables(true); |
|
1174 } |
|
1175 streamACHuffmanTables = |
|
1176 collectHTablesFromMetadata(jmeta, false); |
|
1177 if (streamACHuffmanTables == null) { |
|
1178 streamACHuffmanTables = JPEG.getDefaultHuffmanTables(false); |
|
1179 } |
|
1180 |
|
1181 // Now write them out |
|
1182 writeTables(structPointer, |
|
1183 streamQTables, |
|
1184 streamDCHuffmanTables, |
|
1185 streamACHuffmanTables); |
|
1186 } else { |
|
1187 throw new IIOException("Stream metadata must be JPEG metadata"); |
|
1188 } |
|
1189 } |
|
1190 sequencePrepared = true; |
|
1191 } |
|
1192 |
|
1193 public void writeToSequence(IIOImage image, ImageWriteParam param) |
|
1194 throws IOException { |
|
1195 setThreadLock(); |
|
1196 try { |
|
1197 cbLock.check(); |
|
1198 |
|
1199 if (sequencePrepared == false) { |
|
1200 throw new IllegalStateException("sequencePrepared not called!"); |
|
1201 } |
|
1202 // In the case of JPEG this does nothing different from write |
|
1203 write(null, image, param); |
|
1204 } finally { |
|
1205 clearThreadLock(); |
|
1206 } |
|
1207 } |
|
1208 |
|
1209 public void endWriteSequence() throws IOException { |
|
1210 setThreadLock(); |
|
1211 try { |
|
1212 cbLock.check(); |
|
1213 |
|
1214 if (sequencePrepared == false) { |
|
1215 throw new IllegalStateException("sequencePrepared not called!"); |
|
1216 } |
|
1217 sequencePrepared = false; |
|
1218 } finally { |
|
1219 clearThreadLock(); |
|
1220 } |
|
1221 } |
|
1222 |
|
1223 public synchronized void abort() { |
|
1224 setThreadLock(); |
|
1225 try { |
|
1226 /** |
|
1227 * NB: we do not check the call back lock here, we allow to abort |
|
1228 * the reader any time. |
|
1229 */ |
|
1230 super.abort(); |
|
1231 abortWrite(structPointer); |
|
1232 } finally { |
|
1233 clearThreadLock(); |
|
1234 } |
|
1235 } |
|
1236 |
|
1237 @Override |
|
1238 protected synchronized void clearAbortRequest() { |
|
1239 setThreadLock(); |
|
1240 try { |
|
1241 cbLock.check(); |
|
1242 if (abortRequested()) { |
|
1243 super.clearAbortRequest(); |
|
1244 // reset C structures |
|
1245 resetWriter(structPointer); |
|
1246 // reset the native destination |
|
1247 setDest(structPointer); |
|
1248 } |
|
1249 } finally { |
|
1250 clearThreadLock(); |
|
1251 } |
|
1252 } |
|
1253 |
|
1254 private void resetInternalState() { |
|
1255 // reset C structures |
|
1256 resetWriter(structPointer); |
|
1257 |
|
1258 // reset local Java structures |
|
1259 srcRas = null; |
|
1260 raster = null; |
|
1261 convertTosRGB = false; |
|
1262 currentImage = 0; |
|
1263 numScans = 0; |
|
1264 metadata = null; |
|
1265 } |
|
1266 |
|
1267 public void reset() { |
|
1268 setThreadLock(); |
|
1269 try { |
|
1270 cbLock.check(); |
|
1271 |
|
1272 super.reset(); |
|
1273 } finally { |
|
1274 clearThreadLock(); |
|
1275 } |
|
1276 } |
|
1277 |
|
1278 public void dispose() { |
|
1279 setThreadLock(); |
|
1280 try { |
|
1281 cbLock.check(); |
|
1282 |
|
1283 if (structPointer != 0) { |
|
1284 disposerRecord.dispose(); |
|
1285 structPointer = 0; |
|
1286 } |
|
1287 } finally { |
|
1288 clearThreadLock(); |
|
1289 } |
|
1290 } |
|
1291 |
|
1292 ////////// End of public API |
|
1293 |
|
1294 ///////// Package-access API |
|
1295 |
|
1296 /** |
|
1297 * Called by the native code or other classes to signal a warning. |
|
1298 * The code is used to lookup a localized message to be used when |
|
1299 * sending warnings to listeners. |
|
1300 */ |
|
1301 void warningOccurred(int code) { |
|
1302 cbLock.lock(); |
|
1303 try { |
|
1304 if ((code < 0) || (code > MAX_WARNING)){ |
|
1305 throw new InternalError("Invalid warning index"); |
|
1306 } |
|
1307 processWarningOccurred |
|
1308 (currentImage, |
|
1309 "com.sun.imageio.plugins.jpeg.JPEGImageWriterResources", |
|
1310 Integer.toString(code)); |
|
1311 } finally { |
|
1312 cbLock.unlock(); |
|
1313 } |
|
1314 } |
|
1315 |
|
1316 /** |
|
1317 * The library has it's own error facility that emits warning messages. |
|
1318 * This routine is called by the native code when it has already |
|
1319 * formatted a string for output. |
|
1320 * XXX For truly complete localization of all warning messages, |
|
1321 * the sun_jpeg_output_message routine in the native code should |
|
1322 * send only the codes and parameters to a method here in Java, |
|
1323 * which will then format and send the warnings, using localized |
|
1324 * strings. This method will have to deal with all the parameters |
|
1325 * and formats (%u with possibly large numbers, %02d, %02x, etc.) |
|
1326 * that actually occur in the JPEG library. For now, this prevents |
|
1327 * library warnings from being printed to stderr. |
|
1328 */ |
|
1329 void warningWithMessage(String msg) { |
|
1330 cbLock.lock(); |
|
1331 try { |
|
1332 processWarningOccurred(currentImage, msg); |
|
1333 } finally { |
|
1334 cbLock.unlock(); |
|
1335 } |
|
1336 } |
|
1337 |
|
1338 void thumbnailStarted(int thumbnailIndex) { |
|
1339 cbLock.lock(); |
|
1340 try { |
|
1341 processThumbnailStarted(currentImage, thumbnailIndex); |
|
1342 } finally { |
|
1343 cbLock.unlock(); |
|
1344 } |
|
1345 } |
|
1346 |
|
1347 // Provide access to protected superclass method |
|
1348 void thumbnailProgress(float percentageDone) { |
|
1349 cbLock.lock(); |
|
1350 try { |
|
1351 processThumbnailProgress(percentageDone); |
|
1352 } finally { |
|
1353 cbLock.unlock(); |
|
1354 } |
|
1355 } |
|
1356 |
|
1357 // Provide access to protected superclass method |
|
1358 void thumbnailComplete() { |
|
1359 cbLock.lock(); |
|
1360 try { |
|
1361 processThumbnailComplete(); |
|
1362 } finally { |
|
1363 cbLock.unlock(); |
|
1364 } |
|
1365 } |
|
1366 |
|
1367 ///////// End of Package-access API |
|
1368 |
|
1369 ///////// Private methods |
|
1370 |
|
1371 ///////// Metadata handling |
|
1372 |
|
1373 private void checkSOFBands(SOFMarkerSegment sof, int numBandsUsed) |
|
1374 throws IIOException { |
|
1375 // Does the metadata frame header, if any, match numBandsUsed? |
|
1376 if (sof != null) { |
|
1377 if (sof.componentSpecs.length != numBandsUsed) { |
|
1378 throw new IIOException |
|
1379 ("Metadata components != number of destination bands"); |
|
1380 } |
|
1381 } |
|
1382 } |
|
1383 |
|
1384 private void checkJFIF(JFIFMarkerSegment jfif, |
|
1385 ImageTypeSpecifier type, |
|
1386 boolean input) { |
|
1387 if (jfif != null) { |
|
1388 if (!JPEG.isJFIFcompliant(type, input)) { |
|
1389 ignoreJFIF = true; // type overrides metadata |
|
1390 warningOccurred(input |
|
1391 ? WARNING_IMAGE_METADATA_JFIF_MISMATCH |
|
1392 : WARNING_DEST_METADATA_JFIF_MISMATCH); |
|
1393 } |
|
1394 } |
|
1395 } |
|
1396 |
|
1397 private void checkAdobe(AdobeMarkerSegment adobe, |
|
1398 ImageTypeSpecifier type, |
|
1399 boolean input) { |
|
1400 if (adobe != null) { |
|
1401 int rightTransform = JPEG.transformForType(type, input); |
|
1402 if (adobe.transform != rightTransform) { |
|
1403 warningOccurred(input |
|
1404 ? WARNING_IMAGE_METADATA_ADOBE_MISMATCH |
|
1405 : WARNING_DEST_METADATA_ADOBE_MISMATCH); |
|
1406 if (rightTransform == JPEG.ADOBE_IMPOSSIBLE) { |
|
1407 ignoreAdobe = true; |
|
1408 } else { |
|
1409 newAdobeTransform = rightTransform; |
|
1410 } |
|
1411 } |
|
1412 } |
|
1413 } |
|
1414 |
|
1415 /** |
|
1416 * Collect all the scan info from the given metadata, and |
|
1417 * organize it into the scan info array required by the |
|
1418 * IJG libray. It is much simpler to parse out this |
|
1419 * data in Java and then just copy the data in C. |
|
1420 */ |
|
1421 private int [] collectScans(JPEGMetadata metadata, |
|
1422 SOFMarkerSegment sof) { |
|
1423 List<SOSMarkerSegment> segments = new ArrayList<>(); |
|
1424 int SCAN_SIZE = 9; |
|
1425 int MAX_COMPS_PER_SCAN = 4; |
|
1426 for (Iterator<MarkerSegment> iter = metadata.markerSequence.iterator(); |
|
1427 iter.hasNext();) { |
|
1428 MarkerSegment seg = iter.next(); |
|
1429 if (seg instanceof SOSMarkerSegment) { |
|
1430 segments.add((SOSMarkerSegment) seg); |
|
1431 } |
|
1432 } |
|
1433 int [] retval = null; |
|
1434 numScans = 0; |
|
1435 if (!segments.isEmpty()) { |
|
1436 numScans = segments.size(); |
|
1437 retval = new int [numScans*SCAN_SIZE]; |
|
1438 int index = 0; |
|
1439 for (int i = 0; i < numScans; i++) { |
|
1440 SOSMarkerSegment sos = segments.get(i); |
|
1441 retval[index++] = sos.componentSpecs.length; // num comps |
|
1442 for (int j = 0; j < MAX_COMPS_PER_SCAN; j++) { |
|
1443 if (j < sos.componentSpecs.length) { |
|
1444 int compSel = sos.componentSpecs[j].componentSelector; |
|
1445 for (int k = 0; k < sof.componentSpecs.length; k++) { |
|
1446 if (compSel == sof.componentSpecs[k].componentId) { |
|
1447 retval[index++] = k; |
|
1448 break; // out of for over sof comps |
|
1449 } |
|
1450 } |
|
1451 } else { |
|
1452 retval[index++] = 0; |
|
1453 } |
|
1454 } |
|
1455 retval[index++] = sos.startSpectralSelection; |
|
1456 retval[index++] = sos.endSpectralSelection; |
|
1457 retval[index++] = sos.approxHigh; |
|
1458 retval[index++] = sos.approxLow; |
|
1459 } |
|
1460 } |
|
1461 return retval; |
|
1462 } |
|
1463 |
|
1464 /** |
|
1465 * Finds all DQT marker segments and returns all the q |
|
1466 * tables as a single array of JPEGQTables. |
|
1467 */ |
|
1468 private JPEGQTable [] collectQTablesFromMetadata |
|
1469 (JPEGMetadata metadata) { |
|
1470 ArrayList<DQTMarkerSegment.Qtable> tables = new ArrayList<>(); |
|
1471 Iterator<MarkerSegment> iter = metadata.markerSequence.iterator(); |
|
1472 while (iter.hasNext()) { |
|
1473 MarkerSegment seg = iter.next(); |
|
1474 if (seg instanceof DQTMarkerSegment) { |
|
1475 DQTMarkerSegment dqt = |
|
1476 (DQTMarkerSegment) seg; |
|
1477 tables.addAll(dqt.tables); |
|
1478 } |
|
1479 } |
|
1480 JPEGQTable [] retval = null; |
|
1481 if (tables.size() != 0) { |
|
1482 retval = new JPEGQTable[tables.size()]; |
|
1483 for (int i = 0; i < retval.length; i++) { |
|
1484 retval[i] = |
|
1485 new JPEGQTable(tables.get(i).data); |
|
1486 } |
|
1487 } |
|
1488 return retval; |
|
1489 } |
|
1490 |
|
1491 /** |
|
1492 * Finds all DHT marker segments and returns all the q |
|
1493 * tables as a single array of JPEGQTables. The metadata |
|
1494 * must not be for a progressive image, or an exception |
|
1495 * will be thrown when two Huffman tables with the same |
|
1496 * table id are encountered. |
|
1497 */ |
|
1498 private JPEGHuffmanTable[] collectHTablesFromMetadata |
|
1499 (JPEGMetadata metadata, boolean wantDC) throws IIOException { |
|
1500 ArrayList<DHTMarkerSegment.Htable> tables = new ArrayList<>(); |
|
1501 Iterator<MarkerSegment> iter = metadata.markerSequence.iterator(); |
|
1502 while (iter.hasNext()) { |
|
1503 MarkerSegment seg = iter.next(); |
|
1504 if (seg instanceof DHTMarkerSegment) { |
|
1505 DHTMarkerSegment dht = (DHTMarkerSegment) seg; |
|
1506 for (int i = 0; i < dht.tables.size(); i++) { |
|
1507 DHTMarkerSegment.Htable htable = dht.tables.get(i); |
|
1508 if (htable.tableClass == (wantDC ? 0 : 1)) { |
|
1509 tables.add(htable); |
|
1510 } |
|
1511 } |
|
1512 } |
|
1513 } |
|
1514 JPEGHuffmanTable [] retval = null; |
|
1515 if (tables.size() != 0) { |
|
1516 DHTMarkerSegment.Htable [] htables = |
|
1517 new DHTMarkerSegment.Htable[tables.size()]; |
|
1518 tables.toArray(htables); |
|
1519 retval = new JPEGHuffmanTable[tables.size()]; |
|
1520 for (int i = 0; i < retval.length; i++) { |
|
1521 retval[i] = null; |
|
1522 for (int j = 0; j < tables.size(); j++) { |
|
1523 if (htables[j].tableID == i) { |
|
1524 if (retval[i] != null) { |
|
1525 throw new IIOException("Metadata has duplicate Htables!"); |
|
1526 } |
|
1527 retval[i] = new JPEGHuffmanTable(htables[j].numCodes, |
|
1528 htables[j].values); |
|
1529 } |
|
1530 } |
|
1531 } |
|
1532 } |
|
1533 |
|
1534 return retval; |
|
1535 } |
|
1536 |
|
1537 /////////// End of metadata handling |
|
1538 |
|
1539 ////////////// ColorSpace conversion |
|
1540 |
|
1541 private int getSrcCSType(ImageTypeSpecifier type) { |
|
1542 return getSrcCSType(type.getColorModel()); |
|
1543 } |
|
1544 |
|
1545 private int getSrcCSType(RenderedImage rimage) { |
|
1546 return getSrcCSType(rimage.getColorModel()); |
|
1547 } |
|
1548 |
|
1549 private int getSrcCSType(ColorModel cm) { |
|
1550 int retval = JPEG.JCS_UNKNOWN; |
|
1551 if (cm != null) { |
|
1552 boolean alpha = cm.hasAlpha(); |
|
1553 ColorSpace cs = cm.getColorSpace(); |
|
1554 switch (cs.getType()) { |
|
1555 case ColorSpace.TYPE_GRAY: |
|
1556 retval = JPEG.JCS_GRAYSCALE; |
|
1557 break; |
|
1558 case ColorSpace.TYPE_RGB: |
|
1559 if (alpha) { |
|
1560 retval = JPEG.JCS_RGBA; |
|
1561 } else { |
|
1562 retval = JPEG.JCS_RGB; |
|
1563 } |
|
1564 break; |
|
1565 case ColorSpace.TYPE_YCbCr: |
|
1566 if (alpha) { |
|
1567 retval = JPEG.JCS_YCbCrA; |
|
1568 } else { |
|
1569 retval = JPEG.JCS_YCbCr; |
|
1570 } |
|
1571 break; |
|
1572 case ColorSpace.TYPE_3CLR: |
|
1573 if (cs == JPEG.JCS.getYCC()) { |
|
1574 if (alpha) { |
|
1575 retval = JPEG.JCS_YCCA; |
|
1576 } else { |
|
1577 retval = JPEG.JCS_YCC; |
|
1578 } |
|
1579 } |
|
1580 break; |
|
1581 case ColorSpace.TYPE_CMYK: |
|
1582 retval = JPEG.JCS_CMYK; |
|
1583 break; |
|
1584 } |
|
1585 } |
|
1586 return retval; |
|
1587 } |
|
1588 |
|
1589 private int getDestCSType(ImageTypeSpecifier destType) { |
|
1590 ColorModel cm = destType.getColorModel(); |
|
1591 boolean alpha = cm.hasAlpha(); |
|
1592 ColorSpace cs = cm.getColorSpace(); |
|
1593 int retval = JPEG.JCS_UNKNOWN; |
|
1594 switch (cs.getType()) { |
|
1595 case ColorSpace.TYPE_GRAY: |
|
1596 retval = JPEG.JCS_GRAYSCALE; |
|
1597 break; |
|
1598 case ColorSpace.TYPE_RGB: |
|
1599 if (alpha) { |
|
1600 retval = JPEG.JCS_RGBA; |
|
1601 } else { |
|
1602 retval = JPEG.JCS_RGB; |
|
1603 } |
|
1604 break; |
|
1605 case ColorSpace.TYPE_YCbCr: |
|
1606 if (alpha) { |
|
1607 retval = JPEG.JCS_YCbCrA; |
|
1608 } else { |
|
1609 retval = JPEG.JCS_YCbCr; |
|
1610 } |
|
1611 break; |
|
1612 case ColorSpace.TYPE_3CLR: |
|
1613 if (cs == JPEG.JCS.getYCC()) { |
|
1614 if (alpha) { |
|
1615 retval = JPEG.JCS_YCCA; |
|
1616 } else { |
|
1617 retval = JPEG.JCS_YCC; |
|
1618 } |
|
1619 } |
|
1620 break; |
|
1621 case ColorSpace.TYPE_CMYK: |
|
1622 retval = JPEG.JCS_CMYK; |
|
1623 break; |
|
1624 } |
|
1625 return retval; |
|
1626 } |
|
1627 |
|
1628 private int getDefaultDestCSType(ImageTypeSpecifier type) { |
|
1629 return getDefaultDestCSType(type.getColorModel()); |
|
1630 } |
|
1631 |
|
1632 private int getDefaultDestCSType(RenderedImage rimage) { |
|
1633 return getDefaultDestCSType(rimage.getColorModel()); |
|
1634 } |
|
1635 |
|
1636 private int getDefaultDestCSType(ColorModel cm) { |
|
1637 int retval = JPEG.JCS_UNKNOWN; |
|
1638 if (cm != null) { |
|
1639 boolean alpha = cm.hasAlpha(); |
|
1640 ColorSpace cs = cm.getColorSpace(); |
|
1641 switch (cs.getType()) { |
|
1642 case ColorSpace.TYPE_GRAY: |
|
1643 retval = JPEG.JCS_GRAYSCALE; |
|
1644 break; |
|
1645 case ColorSpace.TYPE_RGB: |
|
1646 if (alpha) { |
|
1647 retval = JPEG.JCS_YCbCrA; |
|
1648 } else { |
|
1649 retval = JPEG.JCS_YCbCr; |
|
1650 } |
|
1651 break; |
|
1652 case ColorSpace.TYPE_YCbCr: |
|
1653 if (alpha) { |
|
1654 retval = JPEG.JCS_YCbCrA; |
|
1655 } else { |
|
1656 retval = JPEG.JCS_YCbCr; |
|
1657 } |
|
1658 break; |
|
1659 case ColorSpace.TYPE_3CLR: |
|
1660 if (cs == JPEG.JCS.getYCC()) { |
|
1661 if (alpha) { |
|
1662 retval = JPEG.JCS_YCCA; |
|
1663 } else { |
|
1664 retval = JPEG.JCS_YCC; |
|
1665 } |
|
1666 } |
|
1667 break; |
|
1668 case ColorSpace.TYPE_CMYK: |
|
1669 retval = JPEG.JCS_YCCK; |
|
1670 break; |
|
1671 } |
|
1672 } |
|
1673 return retval; |
|
1674 } |
|
1675 |
|
1676 private boolean isSubsampled(SOFMarkerSegment.ComponentSpec [] specs) { |
|
1677 int hsamp0 = specs[0].HsamplingFactor; |
|
1678 int vsamp0 = specs[0].VsamplingFactor; |
|
1679 for (int i = 1; i < specs.length; i++) { |
|
1680 if ((specs[i].HsamplingFactor != hsamp0) || |
|
1681 (specs[i].VsamplingFactor != vsamp0)) |
|
1682 return true; |
|
1683 } |
|
1684 return false; |
|
1685 } |
|
1686 |
|
1687 ////////////// End of ColorSpace conversion |
|
1688 |
|
1689 ////////////// Native methods and callbacks |
|
1690 |
|
1691 /** Sets up static native structures. */ |
|
1692 private static native void initWriterIDs(Class<?> qTableClass, |
|
1693 Class<?> huffClass); |
|
1694 |
|
1695 /** Sets up per-writer native structure and returns a pointer to it. */ |
|
1696 private native long initJPEGImageWriter(); |
|
1697 |
|
1698 /** Sets up native structures for output stream */ |
|
1699 private native void setDest(long structPointer); |
|
1700 |
|
1701 /** |
|
1702 * Returns {@code true} if the write was aborted. |
|
1703 */ |
|
1704 private native boolean writeImage(long structPointer, |
|
1705 byte [] data, |
|
1706 int inCsType, int outCsType, |
|
1707 int numBands, |
|
1708 int [] bandSizes, |
|
1709 int srcWidth, |
|
1710 int destWidth, int destHeight, |
|
1711 int stepX, int stepY, |
|
1712 JPEGQTable [] qtables, |
|
1713 boolean writeDQT, |
|
1714 JPEGHuffmanTable[] DCHuffmanTables, |
|
1715 JPEGHuffmanTable[] ACHuffmanTables, |
|
1716 boolean writeDHT, |
|
1717 boolean optimizeHuffman, |
|
1718 boolean progressive, |
|
1719 int numScans, |
|
1720 int [] scans, |
|
1721 int [] componentIds, |
|
1722 int [] HsamplingFactors, |
|
1723 int [] VsamplingFactors, |
|
1724 int [] QtableSelectors, |
|
1725 boolean haveMetadata, |
|
1726 int restartInterval); |
|
1727 |
|
1728 |
|
1729 /** |
|
1730 * Writes the metadata out when called by the native code, |
|
1731 * which will have already written the header to the stream |
|
1732 * and established the library state. This is simpler than |
|
1733 * breaking the write call in two. |
|
1734 */ |
|
1735 private void writeMetadata() throws IOException { |
|
1736 if (metadata == null) { |
|
1737 if (writeDefaultJFIF) { |
|
1738 JFIFMarkerSegment.writeDefaultJFIF(ios, |
|
1739 thumbnails, |
|
1740 iccProfile, |
|
1741 this); |
|
1742 } |
|
1743 if (writeAdobe) { |
|
1744 AdobeMarkerSegment.writeAdobeSegment(ios, newAdobeTransform); |
|
1745 } |
|
1746 } else { |
|
1747 metadata.writeToStream(ios, |
|
1748 ignoreJFIF, |
|
1749 forceJFIF, |
|
1750 thumbnails, |
|
1751 iccProfile, |
|
1752 ignoreAdobe, |
|
1753 newAdobeTransform, |
|
1754 this); |
|
1755 } |
|
1756 } |
|
1757 |
|
1758 /** |
|
1759 * Write out a tables-only image to the stream. |
|
1760 */ |
|
1761 private native void writeTables(long structPointer, |
|
1762 JPEGQTable [] qtables, |
|
1763 JPEGHuffmanTable[] DCHuffmanTables, |
|
1764 JPEGHuffmanTable[] ACHuffmanTables); |
|
1765 |
|
1766 /** |
|
1767 * Put the scanline y of the source ROI view Raster into the |
|
1768 * 1-line Raster for writing. This handles ROI and band |
|
1769 * rearrangements, and expands indexed images. Subsampling is |
|
1770 * done in the native code. |
|
1771 * This is called by the native code. |
|
1772 */ |
|
1773 private void grabPixels(int y) { |
|
1774 |
|
1775 Raster sourceLine = null; |
|
1776 if (indexed) { |
|
1777 sourceLine = srcRas.createChild(sourceXOffset, |
|
1778 sourceYOffset+y, |
|
1779 sourceWidth, 1, |
|
1780 0, 0, |
|
1781 new int [] {0}); |
|
1782 // If the image has BITMASK transparency, we need to make sure |
|
1783 // it gets converted to 32-bit ARGB, because the JPEG encoder |
|
1784 // relies upon the full 8-bit alpha channel. |
|
1785 boolean forceARGB = |
|
1786 (indexCM.getTransparency() != Transparency.OPAQUE); |
|
1787 BufferedImage temp = indexCM.convertToIntDiscrete(sourceLine, |
|
1788 forceARGB); |
|
1789 sourceLine = temp.getRaster(); |
|
1790 } else { |
|
1791 sourceLine = srcRas.createChild(sourceXOffset, |
|
1792 sourceYOffset+y, |
|
1793 sourceWidth, 1, |
|
1794 0, 0, |
|
1795 srcBands); |
|
1796 } |
|
1797 if (convertTosRGB) { |
|
1798 if (debug) { |
|
1799 System.out.println("Converting to sRGB"); |
|
1800 } |
|
1801 // The first time through, converted is null, so |
|
1802 // a new raster is allocated. It is then reused |
|
1803 // on subsequent lines. |
|
1804 converted = convertOp.filter(sourceLine, converted); |
|
1805 sourceLine = converted; |
|
1806 } |
|
1807 if (isAlphaPremultiplied) { |
|
1808 WritableRaster wr = sourceLine.createCompatibleWritableRaster(); |
|
1809 int[] data = null; |
|
1810 data = sourceLine.getPixels(sourceLine.getMinX(), sourceLine.getMinY(), |
|
1811 sourceLine.getWidth(), sourceLine.getHeight(), |
|
1812 data); |
|
1813 wr.setPixels(sourceLine.getMinX(), sourceLine.getMinY(), |
|
1814 sourceLine.getWidth(), sourceLine.getHeight(), |
|
1815 data); |
|
1816 srcCM.coerceData(wr, false); |
|
1817 sourceLine = wr.createChild(wr.getMinX(), wr.getMinY(), |
|
1818 wr.getWidth(), wr.getHeight(), |
|
1819 0, 0, |
|
1820 srcBands); |
|
1821 } |
|
1822 raster.setRect(sourceLine); |
|
1823 if ((y > 7) && (y%8 == 0)) { // Every 8 scanlines |
|
1824 cbLock.lock(); |
|
1825 try { |
|
1826 processImageProgress((float) y / (float) sourceHeight * 100.0F); |
|
1827 } finally { |
|
1828 cbLock.unlock(); |
|
1829 } |
|
1830 } |
|
1831 } |
|
1832 |
|
1833 /** Aborts the current write in the native code */ |
|
1834 private native void abortWrite(long structPointer); |
|
1835 |
|
1836 /** Resets native structures */ |
|
1837 private native void resetWriter(long structPointer); |
|
1838 |
|
1839 /** Releases native structures */ |
|
1840 private static native void disposeWriter(long structPointer); |
|
1841 |
|
1842 private static class JPEGWriterDisposerRecord implements DisposerRecord { |
|
1843 private long pData; |
|
1844 |
|
1845 public JPEGWriterDisposerRecord(long pData) { |
|
1846 this.pData = pData; |
|
1847 } |
|
1848 |
|
1849 public synchronized void dispose() { |
|
1850 if (pData != 0) { |
|
1851 disposeWriter(pData); |
|
1852 pData = 0; |
|
1853 } |
|
1854 } |
|
1855 } |
|
1856 |
|
1857 /** |
|
1858 * This method is called from native code in order to write encoder |
|
1859 * output to the destination. |
|
1860 * |
|
1861 * We block any attempt to change the writer state during this |
|
1862 * method, in order to prevent a corruption of the native encoder |
|
1863 * state. |
|
1864 */ |
|
1865 private void writeOutputData(byte[] data, int offset, int len) |
|
1866 throws IOException |
|
1867 { |
|
1868 cbLock.lock(); |
|
1869 try { |
|
1870 ios.write(data, offset, len); |
|
1871 } finally { |
|
1872 cbLock.unlock(); |
|
1873 } |
|
1874 } |
|
1875 |
|
1876 private Thread theThread = null; |
|
1877 private int theLockCount = 0; |
|
1878 |
|
1879 private synchronized void setThreadLock() { |
|
1880 Thread currThread = Thread.currentThread(); |
|
1881 if (theThread != null) { |
|
1882 if (theThread != currThread) { |
|
1883 // it looks like that this reader instance is used |
|
1884 // by multiple threads. |
|
1885 throw new IllegalStateException("Attempt to use instance of " + |
|
1886 this + " locked on thread " + |
|
1887 theThread + " from thread " + |
|
1888 currThread); |
|
1889 } else { |
|
1890 theLockCount ++; |
|
1891 } |
|
1892 } else { |
|
1893 theThread = currThread; |
|
1894 theLockCount = 1; |
|
1895 } |
|
1896 } |
|
1897 |
|
1898 private synchronized void clearThreadLock() { |
|
1899 Thread currThread = Thread.currentThread(); |
|
1900 if (theThread == null || theThread != currThread) { |
|
1901 throw new IllegalStateException("Attempt to clear thread lock form wrong thread. " + |
|
1902 "Locked thread: " + theThread + |
|
1903 "; current thread: " + currThread); |
|
1904 } |
|
1905 theLockCount --; |
|
1906 if (theLockCount == 0) { |
|
1907 theThread = null; |
|
1908 } |
|
1909 } |
|
1910 |
|
1911 private CallBackLock cbLock = new CallBackLock(); |
|
1912 |
|
1913 private static class CallBackLock { |
|
1914 |
|
1915 private State lockState; |
|
1916 |
|
1917 CallBackLock() { |
|
1918 lockState = State.Unlocked; |
|
1919 } |
|
1920 |
|
1921 void check() { |
|
1922 if (lockState != State.Unlocked) { |
|
1923 throw new IllegalStateException("Access to the writer is not allowed"); |
|
1924 } |
|
1925 } |
|
1926 |
|
1927 private void lock() { |
|
1928 lockState = State.Locked; |
|
1929 } |
|
1930 |
|
1931 private void unlock() { |
|
1932 lockState = State.Unlocked; |
|
1933 } |
|
1934 |
|
1935 private static enum State { |
|
1936 Unlocked, |
|
1937 Locked |
|
1938 } |
|
1939 } |
|
1940 } |