1 files changed, 91 insertions, 0 deletions
diff --git a/jpa_diagram_editor/plugins/org.eclipse.jpt.jpadiagrameditor.ui/src/org/eclipse/jpt/jpadiagrameditor/ui/internal/modelintegration/ui/JPADiagramEditorInput.java b/jpa_diagram_editor/plugins/org.eclipse.jpt.jpadiagrameditor.ui/src/org/eclipse/jpt/jpadiagrameditor/ui/internal/modelintegration/ui/JPADiagramEditorInput.java
new file mode 100644
index 0000000000..160da2e0d9
--- /dev/null
+++ b/jpa_diagram_editor/plugins/org.eclipse.jpt.jpadiagrameditor.ui/src/org/eclipse/jpt/jpadiagrameditor/ui/internal/modelintegration/ui/JPADiagramEditorInput.java
@@ -0,0 +1,91 @@
+/*******************************************************************************
+ * <copyright>
+ *
+ * Copyright (c) 2005, 2010 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:
+ *    Stefan Dimov - initial API, implementation and documentation
+ *
+ * </copyright>
+ *
+ *******************************************************************************/
+package org.eclipse.jpt.jpadiagrameditor.ui.internal.modelintegration.ui;
+
+import org.eclipse.emf.common.util.URI;
+import org.eclipse.emf.ecore.EObject;
+import org.eclipse.emf.ecore.resource.Resource;
+import org.eclipse.emf.transaction.TransactionalEditingDomain;
+import org.eclipse.graphiti.mm.pictograms.Diagram;
+import org.eclipse.graphiti.ui.editor.DiagramEditorInput;
+import org.eclipse.jpt.jpadiagrameditor.ui.internal.modelintegration.util.ModelIntegrationUtil;
+import org.eclipse.jpt.jpadiagrameditor.ui.internal.util.IJPADiagramEditorInput;
+
+
+public class JPADiagramEditorInput extends DiagramEditorInput 
+									implements IJPADiagramEditorInput {
+	
+	private Diagram diagram;
+	private String projectName;
+	
+
+	public JPADiagramEditorInput(Diagram diagram, 
+								 String diagramUriString, 
+								 TransactionalEditingDomain domain, 
+								 String providerId, 
+								 boolean disposeEditingDomain) {
+		
+		super(diagramUriString, domain, providerId, disposeEditingDomain);
+		this.diagram = diagram; 
+		this.projectName = ModelIntegrationUtil.getProjectByDiagram(diagram).getName();		
+	}
+
+	public JPADiagramEditorInput(Diagram diagram,
+								 URI diagramUri, 
+								 TransactionalEditingDomain domain, 
+								 String providerId, 
+								 boolean disposeEditingDomain) {
+		
+		super(diagramUri, domain, providerId, disposeEditingDomain);
+		this.diagram = diagram;
+		this.projectName = ModelIntegrationUtil.getProjectByDiagram(diagram).getName();
+	}
+	
+	@SuppressWarnings("rawtypes")
+	public Object getAdapter(Class adapter) {
+		if (adapter.equals(EObject.class)) {
+			return getDiagram();
+		} 
+		if (adapter.equals(Diagram.class)) {
+			return getDiagram();
+		}
+		if (adapter.equals(TransactionalEditingDomain.class))
+			return ModelIntegrationUtil.getTransactionalEditingDomain(diagram);
+		return null;
+	}
+	
+	public Diagram getDiagram() {
+		return diagram;
+	}	
+	
+	public static JPADiagramEditorInput createEditorInput(Diagram diagram, TransactionalEditingDomain domain, String providerId,
+			boolean disposeEditingDomain) {
+		final Resource resource = diagram.eResource();
+		if (resource == null) {
+			throw new IllegalArgumentException();
+		}
+		final String fragment = resource.getURIFragment(diagram);
+		final URI fragmentUri = resource.getURI().appendFragment(fragment);
+		JPADiagramEditorInput diagramEditorInput;
+		diagramEditorInput = new JPADiagramEditorInput(diagram, fragmentUri, domain, providerId, disposeEditingDomain);
+		return diagramEditorInput;
+	}
+	
+
+	public String getProjectName() {
+		return projectName;
+	}
+}