ExportToImageManager.java

package org.insightech.er.editor.model.dbexport.image;

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;

import javax.imageio.ImageIO;

import org.eclipse.swt.SWT;
import org.eclipse.swt.SWTException;
import org.eclipse.swt.graphics.Image;
import org.eclipse.swt.graphics.ImageData;
import org.eclipse.swt.graphics.ImageLoader;
import org.insightech.er.editor.model.dbexport.html.page_generator.HtmlReportPageGenerator;

public class ExportToImageManager {

	protected Image img;

	private int format;

	private String saveFilePath;

	private String formatName;

	public ExportToImageManager(Image img, int format, String saveFilePath) {
		this.img = img;
		this.format = format;
		this.saveFilePath = saveFilePath;
	}

	public void doProcess() throws IOException, InterruptedException {
		if (format == SWT.IMAGE_JPEG || format == SWT.IMAGE_BMP) {
			writeJPGorBMP(img, saveFilePath, format);

		} else if (format == SWT.IMAGE_PNG || format == SWT.IMAGE_GIF) {
			writePNGorGIF(img, saveFilePath, formatName);
		}
	}

	private void writeJPGorBMP(Image image, String saveFilePath, int format)
			throws IOException {
		ImageData[] imgData = new ImageData[1];
		imgData[0] = image.getImageData();

		ImageLoader imgLoader = new ImageLoader();
		imgLoader.data = imgData;
		imgLoader.save(saveFilePath, format);
	}

	private void writePNGorGIF(Image image, String saveFilePath,
			String formatName) throws IOException, InterruptedException {

		try {
			ImageLoader loader = new ImageLoader();
			loader.data = new ImageData[] { image.getImageData() };
			loader.save(saveFilePath, format);

		} catch (SWTException e) {
			// Eclipse 3.2 �ł́A PNG �� Unsupported or unrecognized format �ƂȂ邽�߁A
			// �ȉ��̑�֕��@���g�p����
			// �������A���̕��@�ł͏�肭�o�͂ł��Ȃ��‹�����

			e.printStackTrace();
			BufferedImage bufferedImage = new BufferedImage(
					image.getBounds().width, image.getBounds().height,
					BufferedImage.TYPE_INT_RGB);

			drawAtBufferedImage(bufferedImage, image, 0, 0);

			ImageIO.write(bufferedImage, formatName, new File(saveFilePath));
		}
	}

	private void drawAtBufferedImage(BufferedImage bimg, Image image, int x,
			int y) throws InterruptedException {

		ImageData data = image.getImageData();

		for (int i = 0; i < image.getBounds().width; i++) {

			for (int j = 0; j < image.getBounds().height; j++) {
				int tmp = 4 * (j * image.getBounds().width + i);

				if (data.data.length > tmp + 2) {
					int r = 0xff & data.data[tmp + 2];
					int g = 0xff & data.data[tmp + 1];
					int b = 0xff & data.data[tmp];

					bimg.setRGB(i + x, j + y, 0xFF << 24 | r << 16 | g << 8
							| b << 0);
				}

				this.doPostTask();
			}
		}
	}

	protected void doPreTask(HtmlReportPageGenerator pageGenerator,
			Object object) {
	}

	protected void doPostTask() throws InterruptedException {
	}
}