将GEF图形另存为图像文件(转)

在实际中,有需求将我们在GEF中画的图形保存为标准格式的图像文件.在网上找到了一个这样的方法.

public boolean saveImageFile(GraphicalViewer vierer,String fileName){
//save diagram to image file
//根据root part类型,cast 为对应的RootEditPart
ScalableFreeformRootEditPart rootEditPart = (ScalableFreeformRootEditPart)viewer.getEditPartRegistry().get(LayerManager.ID);
IFigure rootFigure = ((LayerManager)rootEditPart).getLayer(LayerConstants.PRINTABLE_LAYERS);//rootEditPart.getFigure();
//用其它的方式获取到root figure也可以...
Rectangle rootFigureBounds = rootFigure.getBounds();

Control figureCanvas = viewer.getControl();
GC figureCanvasGC = new GC(figureCanvas);

Image img = new Image(null, rootFigureBounds.width, rootFigureBounds.height);
GC imageGC = new GC(img);
imageGC.setBackground(figureCanvasGC.getBackground());
imageGC.setForeground(figureCanvasGC.getForeground());
imageGC.setFont(figureCanvasGC.getFont());
imageGC.setLineStyle(figureCanvasGC.getLineStyle());
imageGC.setLineWidth(figureCanvasGC.getLineWidth());
imageGC.setXORMode(figureCanvasGC.getXORMode());
Graphics imgGraphics = new SWTGraphics(imageGC);

rootFigure.paint(imgGraphics);

ImageData[] imgData = new ImageData[1];
imgData[0] = img.getImageData();

ImageLoader imgLoader = new ImageLoader();
imgLoader.data = imgData;

//具体格式见:FileFormat,4为JPEG类型
imgLoader.save(fileName, 4);

figureCanvasGC.dispose();
imageGC.dispose();
img.dispose();
}