/*******************************************************************************
* Copyright (c) 2000, 2005 IBM Corporation and others.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* IBM Corporation - initial API and implementation
*******************************************************************************/
package org.eclipse.gef.print;
import java.util.ArrayList;
import java.util.List;
import org.eclipse.swt.printing.Printer;
import org.eclipse.jface.viewers.StructuredSelection;
import org.eclipse.draw2d.IFigure;
import org.eclipse.draw2d.PrintFigureOperation;
import org.eclipse.gef.GraphicalViewer;
import org.eclipse.gef.LayerConstants;
import org.eclipse.gef.editparts.LayerManager;
/**
* @author danlee
*/
public class PrintGraphicalViewerOperation extends PrintFigureOperation {
private GraphicalViewer viewer;
private List selectedEditParts;
/**
* Constructor for PrintGraphicalViewerOperation
* @param p The Printer to print to
* @param g The viewer containing what is to be printed
* NOTE: The GraphicalViewer to be printed must have a
* {@link org.eclipse.draw2d.Layer Layer} with the {@link
* org.eclipse.gef.LayerConstants PRINTABLE_LAYERS} key.
*/
public PrintGraphicalViewerOperation(Printer p, GraphicalViewer g) {
super(p);
viewer = g;
LayerManager lm = (LayerManager)viewer.getEditPartRegistry().get(LayerManager.ID);
IFigure f = lm.getLayer(LayerConstants.PRINTABLE_LAYERS);
setPrintSource(f);
}
/**
* Returns the viewer.
*
* @return GraphicalViewer
*/
public GraphicalViewer getViewer() {
return viewer;
}
/**
* @see org.eclipse.draw2d.PrintOperation#preparePrintSource()
*/
protected void preparePrintSource() {
super.preparePrintSource();
selectedEditParts = new ArrayList(viewer.getSelectedEditParts());
viewer.deselectAll();
}
/**
* @see org.eclipse.draw2d.PrintOperation#restorePrintSource()
*/
protected void restorePrintSource() {
super.restorePrintSource();
viewer.setSelection(new StructuredSelection(selectedEditParts));
}
/**
* Sets the viewer.
*
* @param viewer The viewer to set
*/
public void setViewer(GraphicalViewer viewer) {
this.viewer = viewer;
}
}