/*
* Copyright (c) 2012-2016 Marsha Chechik, Alessio Di Sandro, Michalis Famelis,
* Rick Salay.
* 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:
* Alessio Di Sandro - Implementation.
*/
package edu.toronto.cs.se.modelepedia.classdiagram.diagram.part;
import java.lang.reflect.InvocationTargetException;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.jface.dialogs.ErrorDialog;
import org.eclipse.jface.operation.IRunnableWithProgress;
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.PartInitException;
import org.eclipse.ui.actions.WorkspaceModifyOperation;
/**
* @generated
*/
public class ClassDiagramCreationWizard extends Wizard implements INewWizard {
/**
* @generated
*/
private IWorkbench workbench;
/**
* @generated
*/
protected IStructuredSelection selection;
/**
* @generated
*/
protected ClassDiagramCreationWizardPage diagramModelFilePage;
/**
* @generated
*/
protected ClassDiagramCreationWizardPage domainModelFilePage;
/**
* @generated
*/
protected Resource diagram;
/**
* @generated
*/
private boolean openNewlyCreatedDiagramEditor = true;
/**
* @generated
*/
public IWorkbench getWorkbench() {
return workbench;
}
/**
* @generated
*/
public IStructuredSelection getSelection() {
return selection;
}
/**
* @generated
*/
public final Resource getDiagram() {
return diagram;
}
/**
* @generated
*/
public final boolean isOpenNewlyCreatedDiagramEditor() {
return openNewlyCreatedDiagramEditor;
}
/**
* @generated
*/
public void setOpenNewlyCreatedDiagramEditor(boolean openNewlyCreatedDiagramEditor) {
this.openNewlyCreatedDiagramEditor = openNewlyCreatedDiagramEditor;
}
/**
* @generated
*/
public void init(IWorkbench workbench, IStructuredSelection selection) {
this.workbench = workbench;
this.selection = selection;
setWindowTitle(Messages.ClassDiagramCreationWizardTitle);
setDefaultPageImageDescriptor(
ClassDiagramDiagramEditorPlugin.getBundledImageDescriptor("icons/wizban/NewClassDiagramWizard.gif")); //$NON-NLS-1$
setNeedsProgressMonitor(true);
}
/**
* @generated
*/
public void addPages() {
diagramModelFilePage = new ClassDiagramCreationWizardPage(
"DiagramModelFile", //$NON-NLS-1$
getSelection(),
"classdiagramdiag"); //$NON-NLS-1$
diagramModelFilePage.setTitle(Messages.ClassDiagramCreationWizard_DiagramModelFilePageTitle);
diagramModelFilePage.setDescription(Messages.ClassDiagramCreationWizard_DiagramModelFilePageDescription);
addPage(diagramModelFilePage);
domainModelFilePage = new ClassDiagramCreationWizardPage("DomainModelFile", getSelection(), "classdiagram") { //$NON-NLS-1$ //$NON-NLS-2$
public void setVisible(boolean visible) {
if (visible) {
String fileName = diagramModelFilePage.getFileName();
fileName = fileName.substring(0, fileName.length() - ".classdiagramdiag".length()); //$NON-NLS-1$
setFileName(
ClassDiagramDiagramEditorUtil
.getUniqueFileName(getContainerFullPath(), fileName, "classdiagram")); //$NON-NLS-1$
}
super.setVisible(visible);
}
};
domainModelFilePage.setTitle(Messages.ClassDiagramCreationWizard_DomainModelFilePageTitle);
domainModelFilePage.setDescription(Messages.ClassDiagramCreationWizard_DomainModelFilePageDescription);
addPage(domainModelFilePage);
}
/**
* @generated
*/
public boolean performFinish() {
IRunnableWithProgress op = new WorkspaceModifyOperation(null) {
protected void execute(IProgressMonitor monitor) throws CoreException, InterruptedException {
diagram = ClassDiagramDiagramEditorUtil
.createDiagram(diagramModelFilePage.getURI(), domainModelFilePage.getURI(), monitor);
if (isOpenNewlyCreatedDiagramEditor() && diagram != null) {
try {
ClassDiagramDiagramEditorUtil.openDiagram(diagram);
}
catch (PartInitException e) {
ErrorDialog.openError(
getContainer().getShell(),
Messages.ClassDiagramCreationWizardOpenEditorError,
null,
e.getStatus());
}
}
}
};
try {
getContainer().run(false, true, op);
}
catch (InterruptedException e) {
return false;
}
catch (InvocationTargetException e) {
if (e.getTargetException() instanceof CoreException) {
ErrorDialog.openError(
getContainer().getShell(),
Messages.ClassDiagramCreationWizardCreationError,
null,
((CoreException) e.getTargetException()).getStatus());
}
else {
ClassDiagramDiagramEditorPlugin
.getInstance()
.logError("Error creating diagram", e.getTargetException()); //$NON-NLS-1$
}
return false;
}
return diagram != null;
}
}