1 files changed, 67 insertions, 0 deletions
diff --git a/plugins/infra/gmfdiag/assistant/org.eclipse.papyrus.infra.gmfdiag.assistant/src/org/eclipse/papyrus/infra/gmfdiag/assistant/internal/core/DefaultModelingAssistantModelProvider.java b/plugins/infra/gmfdiag/assistant/org.eclipse.papyrus.infra.gmfdiag.assistant/src/org/eclipse/papyrus/infra/gmfdiag/assistant/internal/core/DefaultModelingAssistantModelProvider.java
new file mode 100644
index 00000000000..9a8301eb691
--- /dev/null
+++ b/plugins/infra/gmfdiag/assistant/org.eclipse.papyrus.infra.gmfdiag.assistant/src/org/eclipse/papyrus/infra/gmfdiag/assistant/internal/core/DefaultModelingAssistantModelProvider.java
@@ -0,0 +1,67 @@
+/*****************************************************************************
+ * Copyright (c) 2014 Christian W. Damus 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:
+ *   Christian W. Damus - Initial API and implementation
+ *   
+ *****************************************************************************/
+
+package org.eclipse.papyrus.infra.gmfdiag.assistant.internal.core;
+
+import java.util.Collections;
+
+import org.eclipse.emf.common.util.URI;
+import org.eclipse.emf.ecore.resource.Resource;
+import org.eclipse.emf.ecore.resource.ResourceSet;
+import org.eclipse.emf.ecore.util.EcoreUtil;
+import org.eclipse.gmf.runtime.emf.ui.services.modelingassistant.IModelingAssistantProvider;
+import org.eclipse.papyrus.infra.gmfdiag.assistant.core.IModelingAssistantModelProvider;
+
+import com.google.common.collect.Iterables;
+
+/**
+ * A provider that loads modeling assistant models from EMF resources.
+ */
+public class DefaultModelingAssistantModelProvider implements IModelingAssistantModelProvider {
+	private final URI resourceURI;
+
+	/**
+	 * Initializes me.
+	 */
+	public DefaultModelingAssistantModelProvider(URI resourceURI) {
+		this.resourceURI = resourceURI;
+	}
+
+	@Override
+	public Iterable<? extends IModelingAssistantProvider> loadProviders(ResourceSet resourceSet) {
+		Iterable<? extends IModelingAssistantProvider> result;
+
+		Resource resource = null;
+
+		try {
+			resource = resourceSet.getResource(resourceURI, true);
+		} catch (Exception e) {
+			AssistantPlugin.log.error("Error loading modeling assistants model " + resourceURI, e); //$NON-NLS-1$
+
+			// In case we managed to get something
+			resource = resourceSet.getResource(resourceURI, false);
+		}
+
+		if (resource != null) {
+			// Because we track all the resources loaded by this provider
+			EcoreUtil.resolveAll(resource);
+
+			result = Iterables.filter(resource.getContents(), IModelingAssistantProvider.class);
+		} else {
+			result = Collections.emptyList();
+		}
+
+		return result;
+	}
+
+}