| /******************************************************************************* |
| * <copyright> |
| * |
| * Copyright (c) 2013, 2013 SAP AG. |
| * 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: |
| * SAP AG - initial API, implementation and documentation |
| * |
| * </copyright> |
| * |
| *******************************************************************************/ |
| package org.eclipse.fmc.blockdiagram.editor.wizards; |
| |
| import java.io.ByteArrayInputStream; |
| import java.io.IOException; |
| import java.io.InputStream; |
| import java.lang.reflect.InvocationTargetException; |
| |
| import org.eclipse.core.resources.IContainer; |
| import org.eclipse.core.resources.IFile; |
| import org.eclipse.core.resources.IResource; |
| import org.eclipse.core.resources.IWorkspaceRoot; |
| import org.eclipse.core.resources.ResourcesPlugin; |
| import org.eclipse.core.runtime.CoreException; |
| import org.eclipse.core.runtime.IProgressMonitor; |
| import org.eclipse.core.runtime.IStatus; |
| import org.eclipse.core.runtime.Path; |
| import org.eclipse.core.runtime.Status; |
| import org.eclipse.emf.common.util.URI; |
| import org.eclipse.fmc.blockdiagram.editor.util.DiagramFactory; |
| import org.eclipse.jface.dialogs.MessageDialog; |
| import org.eclipse.jface.operation.IRunnableWithProgress; |
| import org.eclipse.jface.viewers.ISelection; |
| import org.eclipse.jface.viewers.IStructuredSelection; |
| import org.eclipse.jface.wizard.Wizard; |
| import org.eclipse.ui.INewWizard; |
| import org.eclipse.ui.IWorkbench; |
| import org.eclipse.ui.IWorkbenchPage; |
| import org.eclipse.ui.IWorkbenchWizard; |
| import org.eclipse.ui.PartInitException; |
| import org.eclipse.ui.PlatformUI; |
| import org.eclipse.ui.ide.IDE; |
| |
| /** |
| * This class is responsible for creating Blockdiagram files with a single graphical |
| * file. |
| * |
| * @author Benjamin Schmeling |
| * @author Rainer Thome |
| * |
| */ |
| public class BlockDiagramFileWizard extends Wizard implements INewWizard { |
| |
| protected BlockDiagramFilePage page; |
| protected int diagramWidth; |
| protected int diagramHeight; |
| protected int gridUnit; |
| protected ISelection selection; |
| |
| /** |
| * Constructor for BlockDiagramFileWizard. |
| */ |
| public BlockDiagramFileWizard() { |
| super(); |
| setNeedsProgressMonitor(true); |
| } |
| |
| /** |
| * Adding the page to the wizard. |
| */ |
| public void addPages() { |
| page = new BlockDiagramFilePage(selection); |
| addPage(page); |
| } |
| |
| /** |
| * This method is called when 'Finish' button is pressed in the wizard. We |
| * will create an operation and run it using wizard as execution context. |
| */ |
| public boolean performFinish() { |
| final String containerName = page.getContainerName(); |
| final String fileName = page.getFileName(); |
| this.gridUnit = page.getGridUnit(); |
| IRunnableWithProgress op = new IRunnableWithProgress() { |
| public void run(IProgressMonitor monitor) |
| throws InvocationTargetException { |
| try { |
| doFinish(containerName, fileName, monitor, |
| page.getExtension()); |
| } catch (CoreException e) { |
| throw new InvocationTargetException(e); |
| } finally { |
| monitor.done(); |
| } |
| } |
| }; |
| try { |
| getContainer().run(true, false, op); |
| } catch (InterruptedException e) { |
| return false; |
| } catch (InvocationTargetException e) { |
| Throwable realException = e.getTargetException(); |
| MessageDialog.openError(getShell(), "Error", |
| realException.getMessage()); |
| return false; |
| } |
| return true; |
| } |
| |
| /** |
| * The worker method. It will find the container, create the file if missing |
| * or just replace its contents, and open the editor on the newly created |
| * file. |
| */ |
| private void doFinish(String containerName, String fileName, |
| IProgressMonitor monitor, String extension) throws CoreException { |
| // create a sample file |
| monitor.beginTask("Creating " + fileName, 2); |
| IWorkspaceRoot root = ResourcesPlugin.getWorkspace().getRoot(); |
| IResource resource = root.findMember(new Path(containerName)); |
| if (!resource.exists() || !(resource instanceof IContainer)) { |
| throwCoreException("Project \"" + containerName |
| + "\" does not exist."); |
| } |
| IContainer container = (IContainer) resource; |
| final IFile file = container.getFile(new Path(fileName)); |
| try { |
| InputStream stream = openContentStream(); |
| if (!file.exists()) { |
| |
| file.create(stream, true, monitor); |
| } |
| stream.close(); |
| } catch (IOException e) { |
| e.printStackTrace(); |
| } |
| createBlockdiagramFile(file); |
| monitor.worked(1); |
| monitor.setTaskName("Opening file for editing..."); |
| getShell().getDisplay().asyncExec(new Runnable() { |
| public void run() { |
| IWorkbenchPage page = PlatformUI.getWorkbench() |
| .getActiveWorkbenchWindow().getActivePage(); |
| try { |
| IDE.openEditor(page, file, true); |
| } catch (PartInitException e) { |
| e.printStackTrace(); |
| } |
| } |
| }); |
| monitor.worked(1); |
| } |
| |
| protected void createBlockdiagramFile(final IFile file) { |
| DiagramFactory.createDefaultBlockdiagramFile(URI.createPlatformResourceURI(file |
| .getFullPath().toString(), true), diagramWidth, diagramHeight, |
| gridUnit); |
| } |
| |
| /** |
| * We will initialize file contents with a sample text. |
| */ |
| private InputStream openContentStream() { |
| return new ByteArrayInputStream(new byte[0]); |
| } |
| |
| private void throwCoreException(String message) throws CoreException { |
| IStatus status = new Status(IStatus.ERROR, "BlockdiagramEditor", IStatus.OK, |
| message, null); |
| throw new CoreException(status); |
| } |
| |
| /** |
| * We will accept the selection in the workbench to see if we can initialize |
| * from it. |
| * |
| * @see IWorkbenchWizard#init(IWorkbench, IStructuredSelection) |
| */ |
| public void init(IWorkbench workbench, IStructuredSelection selection) { |
| this.selection = selection; |
| } |
| } |