BufferredImage 变换为ImageData

BufferredImage 转换为ImageData
public void setImage(final BufferedImage bufferedImage) {

BufferedImage scaledImage = new BufferedImage(bufferedImage.getWidth(),
bufferedImage.getHeight(), BufferedImage.TYPE_INT_ARGB);//把现有的bufferedImage的颜色模式转换为DirectColorModel
Graphics2D localGraphics2D = scaledImage.createGraphics();
localGraphics2D.drawImage(bufferedImage, 0, 0, scaledImage.getWidth(),
scaledImage.getHeight(), null);
localGraphics2D.dispose();
ImageData imageData = convertToSWT(scaledImage);
//label.setImage(new Image(Display.getDefault(), imageData));

}



public static ImageData convertToSWT(BufferedImage bufferedImage) {
if (bufferedImage.getColorModel() instanceof DirectColorModel) {
DirectColorModel colorModel = (DirectColorModel) bufferedImage
.getColorModel();
PaletteData palette = new PaletteData(colorModel.getRedMask(),
colorModel.getGreenMask(), colorModel.getBlueMask());
ImageData data = new ImageData(bufferedImage.getWidth(),
bufferedImage.getHeight(), colorModel.getPixelSize(),
palette);
WritableRaster raster = bufferedImage.getRaster();
int[] pixelArray = new int[4];
for (int y = 0; y < data.height; y++) {
for (int x = 0; x < data.width; x++) {
raster.getPixel(x, y, pixelArray);
int pixel = palette.getPixel(new RGB(pixelArray[0],
pixelArray[1], pixelArray[2]));
data.setPixel(x, y, pixel);
}
}
return data;
} else if (bufferedImage.getColorModel() instanceof IndexColorModel) {
IndexColorModel colorModel = (IndexColorModel) bufferedImage
.getColorModel();
int size = colorModel.getMapSize();
byte[] reds = new byte[size];
byte[] greens = new byte[size];
byte[] blues = new byte[size];
colorModel.getReds(reds);
colorModel.getGreens(greens);
colorModel.getBlues(blues);
RGB[] rgbs = new RGB[size];
for (int i = 0; i < rgbs.length; i++) {
rgbs[i] = new RGB(reds[i] & 0xFF, greens[i] & 0xFF,
blues[i] & 0xFF);
}
PaletteData palette = new PaletteData(rgbs);
ImageData data = new ImageData(bufferedImage.getWidth(),
bufferedImage.getHeight(), colorModel.getPixelSize(),
palette);
data.transparentPixel = colorModel.getTransparentPixel();
WritableRaster raster = bufferedImage.getRaster();
int[] pixelArray = new int[1];
for (int y = 0; y < data.height; y++) {
for (int x = 0; x < data.width; x++) {
raster.getPixel(x, y, pixelArray);
data.setPixel(x, y, pixelArray[0]);
}
}
return data;
}
return null;
}


/**
* Converts an AWT image to SWT.
*
* @param image
*            the image (<code>null</code> not permitted).
*
* @return Image data.
*/
/*
* public static ImageData convertAWTImageToSWT(Image image) { if (image ==
* null) { throw new IllegalArgumentException("Null 'image' argument."); }
* int w = image.getWidth(null); int h = image.getHeight(null); if (w == -1
* || h == -1) { return null; } BufferedImage bi = new BufferedImage(w, h,
* BufferedImage.TYPE_INT_RGB); Graphics g = bi.getGraphics();
* g.drawImage(image, 0, 0, null); g.dispose(); return convertToSWT(bi); }
*/

/**
* Not entirely sure this method is necessary for indicating 'no cache', but
* since the claim was specific to byte arrays, we'll do it.
*/
/*
* private Image convertToFromBytes(BufferedImage image) { try {
* ByteArrayOutputStream baos = new ByteArrayOutputStream();
* ImageIO.write(image, "png", baos); return
* Toolkit.getDefaultToolkit().createImage(baos.toByteArray()); } catch
* (Exception e) { return null; } }
*/