move ImageConverter to ImageUtil

1 files changed, 95 insertions, 0 deletions

diff --git a/pdf-over-commons/src/main/java/at/asit/pdfover/commons/utils/ImageUtil.java b/pdf-over-commons/src/main/java/at/asit/pdfover/commons/utils/ImageUtil.java
index e3540a35..4a926226 100644
--- a/pdf-over-commons/src/main/java/at/asit/pdfover/commons/utils/ImageUtil.java
+++ b/pdf-over-commons/src/main/java/at/asit/pdfover/commons/utils/ImageUtil.java
@@ -6,6 +6,15 @@ import java.io.File;
 import java.io.IOException;
 import java.util.Iterator;
 
+import java.awt.image.ComponentColorModel;
+import java.awt.image.DirectColorModel;
+import java.awt.image.IndexColorModel;
+import java.awt.image.WritableRaster;
+
+import org.eclipse.swt.graphics.ImageData;
+import org.eclipse.swt.graphics.PaletteData;
+import org.eclipse.swt.graphics.RGB;
+
 import javax.imageio.ImageIO;
 import javax.imageio.ImageReadParam;
 import javax.imageio.ImageReader;
@@ -156,4 +165,90 @@ public final class ImageUtil {
         JOptionPane.showMessageDialog(null, new JScrollPane(panel));
         return image;
     }
+
+    /**
+	 * Convert AWT Image to SWT Image
+	 *
+	 * @param bufferedImage
+	 * @return SWT Image data
+	 */
+	public static ImageData convertToSWT(final BufferedImage bufferedImage) {
+		if (bufferedImage == null)
+			return null;
+
+		if ((bufferedImage.getWidth() <= 0) || (bufferedImage.getHeight() <= 0))
+			return null;
+
+		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[3];
+			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;
+		} else if (bufferedImage.getColorModel() instanceof ComponentColorModel) {
+		    ComponentColorModel colorModel = (ComponentColorModel)bufferedImage.getColorModel();
+
+		    //ASSUMES: 3 BYTE BGR IMAGE TYPE
+
+		    PaletteData palette = new PaletteData(0x0000FF, 0x00FF00,0xFF0000);
+		    ImageData data = new ImageData(bufferedImage.getWidth(), bufferedImage.getHeight(), colorModel.getPixelSize(), palette);
+
+		    //This is valid because we are using a 3-byte Data model with no transparent pixels
+		    data.transparentPixel = -1;
+
+		    WritableRaster raster = bufferedImage.getRaster();
+		    int[] pixelArray = new int[bufferedImage.getColorModel().getNumComponents()];
+		    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;
+		}
+		return null;
+	}
 }