Add a first implementation of toBuffered for ColorSeparation to handle CMYK pictures.

The code has been tested with a CMYK picture, but I have not been able to test alpha picture.
The code will probably also need to be improved to handle any kind of ColorSeparation picture.
Though, it's a nice base for further developments.
This commit is contained in:
guinotphil 2013-11-15 16:02:35 +01:00
parent b34770658a
commit b5f2f9dbb8

View File

@ -32,6 +32,8 @@ import magick.*;
import java.awt.*;
import java.awt.color.ColorSpace;
import java.awt.color.ICC_ColorSpace;
import java.awt.color.ICC_Profile;
import java.awt.image.*;
/**
@ -160,7 +162,11 @@ public final class MagickUtil {
image = rgbToBuffered(pImage, true);
break;
case ImageType.ColorSeparationType:
image = cmykToBuffered(pImage, false);
break;
case ImageType.ColorSeparationMatteType:
image = cmykToBuffered(pImage, true); // Not tested actually
break;
case ImageType.OptimizeType:
default:
throw new IllegalArgumentException("Unknown JMagick image type: " + pImage.getImageType());
@ -546,4 +552,62 @@ public final class MagickUtil {
return new BufferedImage(pAlpha ? CM_COLOR_ALPHA : CM_COLOR_OPAQUE, raster, pAlpha, null);
}
/**
* Converts an {@code MagickImage} to a {@code BufferedImage} which holds an CMYK ICC profile
*
* @param pImage the original {@code MagickImage}
* @param pAlpha keep alpha channel
* @return a new {@code BufferedImage}
*
* @throws MagickException if an exception occurs during conversion
*
* @see BufferedImage
*/
private static BufferedImage cmykToBuffered(MagickImage pImage, boolean pAlpha) throws MagickException {
final Dimension size = pImage.getDimension();
final int length = size.width * size.height;
// Retreive the ICC profile
final ICC_Profile profile = ICC_Profile.getInstance(pImage.getColorProfile().getInfo());
final ColorSpace cs = new ICC_ColorSpace(profile);
final int bands = cs.getNumComponents() + (pAlpha ? 1 : 0);
final ColorModel cm;
final int[] bits = new int[bands];
for (int i = 0; i < bands; i++) {
bits[i] = 8;
}
if (pAlpha) {
cm = new ComponentColorModel(cs, bits, true, true, Transparency.TRANSLUCENT, DataBuffer.TYPE_BYTE);
} else {
cm = new ComponentColorModel(cs, bits, false, false, Transparency.OPAQUE, DataBuffer.TYPE_BYTE);
}
final byte[] pixels = new byte[length * bands];
// TODO: If we do multiple dispatches (one per line, typically), we could provide listener
// feedback. But it's currently a lot slower than fetching all the pixels in one go.
// TODO: handle more generic cases if profile is not CMYK
// The "ACMYK" value has not been tested with an alpha picture actually...
pImage.dispatchImage(0, 0, size.width, size.height, pAlpha ? "ACMYK" : "CMYK", pixels);
// Init databuffer with array, to avoid allocation of empty array
final DataBuffer buffer = new DataBufferByte(pixels, pixels.length);
// TODO: build array from bands variable, here it just works for CMYK
// The values has not been tested with an alpha picture actually...
final int[] bandOffsets = pAlpha ? new int[] {0, 1, 2, 3, 4} : new int[] {0, 1, 2, 3};
final WritableRaster raster =
Raster.createInterleavedRaster(buffer, size.width, size.height,
size.width * bands, bands, bandOffsets, LOCATION_UPPER_LEFT);
return new BufferedImage(cm, raster, pAlpha, null);
}
}