--- /dev/null
+/*******************************************************************************\r
+ * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
+ * in Industry THTH ry.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ * VTT Technical Research Centre of Finland - initial API and implementation\r
+ *******************************************************************************/\r
+package org.simantics.diagram.symbolcontribution;\r
+\r
+import java.util.ArrayList;\r
+import java.util.Collection;\r
+import java.util.HashSet;\r
+import java.util.List;\r
+import java.util.Map;\r
+import java.util.Set;\r
+import java.util.TreeMap;\r
+import java.util.TreeSet;\r
+\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.diagram.symbollibrary.ISymbolGroup;\r
+import org.simantics.utils.datastructures.MapList;\r
+import org.simantics.utils.strings.EString;\r
+\r
+/**\r
+ * @author Tuukka Lehtonen\r
+ */\r
+public class CompositeSymbolProviderFactory implements SymbolProviderFactory {\r
+\r
+ Collection<SymbolProviderFactory> factories;\r
+\r
+ public CompositeSymbolProviderFactory(Collection<SymbolProviderFactory> factories) {\r
+ if (factories == null)\r
+ throw new NullPointerException("null factories");\r
+ this.factories = factories;\r
+ }\r
+\r
+ @Override\r
+ public int hashCode() {\r
+ return factories.hashCode();\r
+ }\r
+\r
+ @Override\r
+ public boolean equals(Object obj) {\r
+ if (this == obj)\r
+ return true;\r
+ if (obj == null)\r
+ return false;\r
+ if (getClass() != obj.getClass())\r
+ return false;\r
+ CompositeSymbolProviderFactory other = (CompositeSymbolProviderFactory) obj;\r
+ return factories.equals(other.factories);\r
+ }\r
+\r
+ @Override\r
+ public ISymbolProvider create(ReadGraph graph) throws DatabaseException {\r
+ ArrayList<ISymbolProvider> providers = new ArrayList<ISymbolProvider>(factories.size());\r
+ for (SymbolProviderFactory factory : factories) {\r
+ ISymbolProvider provider = factory.create(graph);\r
+ providers.add(provider);\r
+ }\r
+ return new CompositeSymbolProvider(providers);\r
+ }\r
+\r
+ static class CompositeSymbolProvider extends AbstractSymbolProvider {\r
+\r
+ Collection<ISymbolProvider> providers;\r
+\r
+ public CompositeSymbolProvider(Collection<ISymbolProvider> providers) {\r
+ this.providers = providers;\r
+ }\r
+\r
+ @Override\r
+ public Collection<ISymbolGroup> getSymbolGroups() {\r
+ MapList<String, ISymbolGroup> groupsByName = new MapList<String, ISymbolGroup>();\r
+ Set<Object> processedIds = new HashSet<Object>();\r
+ Map<String, ISymbolGroup> groups = new TreeMap<String, ISymbolGroup>();\r
+ for(ISymbolProvider provider : providers) {\r
+ for(ISymbolGroup group : provider.getSymbolGroups()) {\r
+ String name = group.getName();\r
+ if (group instanceof IIdentifiedObject) {\r
+ Object id = ((IIdentifiedObject) group).getId();\r
+ if (!processedIds.add(id))\r
+ continue;\r
+ }\r
+ groupsByName.add(name, group);\r
+ }\r
+ }\r
+ // Process groups with multiple ISymbolGroup contributions\r
+ // into CompositeSymbolGroupsand leave single ISymbolGroups as such.\r
+ for (String name : groupsByName.getKeys()) {\r
+ List<ISymbolGroup> grps = groupsByName.getValues(name);\r
+ int grpCount = grps.size();\r
+ if (grpCount == 1) {\r
+ groups.put(name, grps.get(0));\r
+ } else if (grpCount > 1) {\r
+ Object id = combineIdentification(grps);\r
+ String description = combineDescriptions(grps);\r
+ CompositeSymbolGroup cgroup = new CompositeSymbolGroup(id, name, description);\r
+ for (ISymbolGroup grp : grps)\r
+ cgroup.add(grp);\r
+ groups.put(name, cgroup);\r
+ }\r
+ }\r
+ return new ArrayList<ISymbolGroup>(groups.values());\r
+ }\r
+\r
+ private Object combineIdentification(List<ISymbolGroup> grps) {\r
+ Set<Object> id = new HashSet<Object>();\r
+ for (ISymbolGroup grp : grps) {\r
+ if (grp instanceof IIdentifiedObject) { \r
+ id.add( ((IIdentifiedObject) grp).getId() );\r
+ }\r
+ }\r
+ return id;\r
+ }\r
+\r
+ private String combineDescriptions(List<ISymbolGroup> grps) {\r
+ Set<String> descs = new TreeSet<String>();\r
+ for (ISymbolGroup grp : grps)\r
+ descs.add(grp.getDescription());\r
+ if (descs.isEmpty())\r
+ return "";\r
+ if (descs.size() == 1)\r
+ return descs.iterator().next();\r
+ return EString.implode(descs);\r
+ }\r
+\r
+ @Override\r
+ public void setListener(Runnable listener) {\r
+ for (ISymbolProvider provider : providers)\r
+ provider.setListener(listener);\r
+ }\r
+\r
+ }\r
+\r
+}\r