X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;ds=sidebyside;f=org.simantics.g3d%2Fsrc%2Forg%2Fsimantics%2Fg3d%2Fobjmap%2Fschema%2FDefaultSchema.java;fp=org.simantics.g3d%2Fsrc%2Forg%2Fsimantics%2Fg3d%2Fobjmap%2Fschema%2FDefaultSchema.java;h=43ed22fdd1388acc2bdd80e91662d9f017dafbf3;hb=87b3241ec277ba3d8e414b26186a032c9cdcaeed;hp=0000000000000000000000000000000000000000;hpb=1f0bcd66274375f2278d2e6c486cb28257a5f7b2;p=simantics%2F3d.git

diff --git a/org.simantics.g3d/src/org/simantics/g3d/objmap/schema/DefaultSchema.java b/org.simantics.g3d/src/org/simantics/g3d/objmap/schema/DefaultSchema.java
new file mode 100644
index 00000000..43ed22fd
--- /dev/null
+++ b/org.simantics.g3d/src/org/simantics/g3d/objmap/schema/DefaultSchema.java
@@ -0,0 +1,96 @@
+/*******************************************************************************
+ * Copyright (c) 2007, 2010 Association for Decentralized Information Management
+ * in Industry THTH ry.
+ * 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:
+ *     VTT Technical Research Centre of Finland - initial API and implementation
+ *******************************************************************************/
+package org.simantics.g3d.objmap.schema;
+
+import gnu.trove.THashMap;
+
+import org.simantics.db.ReadGraph;
+import org.simantics.db.Resource;
+import org.simantics.db.common.utils.NameUtils;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.objmap.ILinkType;
+import org.simantics.objmap.IMappingSchema;
+import org.simantics.objmap.MappingException;
+import org.simantics.objmap.schema.SimpleLinkType;
+
+/**
+ * 
+ */
+public class DefaultSchema implements IMappingSchema {
+
+    THashMap<Resource, ILinkType> domainLinkTypes = 
+        new THashMap<Resource, ILinkType>();
+    THashMap<Class<?>, ILinkType> rangeLinkTypes = 
+        new THashMap<Class<?>, ILinkType>();
+    
+    public void addLinkType(SimpleLinkType linkType) {
+        domainLinkTypes.put(linkType.domainType, linkType);
+        rangeLinkTypes.put(linkType.rangeType, linkType);
+    }
+    
+    public void addLinkType(AdaptedLinkType linkType) {
+        domainLinkTypes.put(linkType.domainType, linkType);
+        rangeLinkTypes.put(linkType.rangeType, linkType);
+    }
+    
+    @Override
+    public ILinkType linkTypeOfDomainElement(ReadGraph g, Resource element) throws MappingException {        
+        try {
+        	
+        	for(Resource type : g.getTypes(element)) {
+
+        		ILinkType linkType = domainLinkTypes.get(type);
+        		if(linkType != null) return linkType;
+        		
+        	}
+        	
+        	throw new MappingException("Didn't find a link type for " +
+        			NameUtils.getSafeName(g, element) + ".");
+   				
+        } catch (DatabaseException e) {
+            throw new MappingException(e);
+        }
+    }
+
+    @Override
+    public ILinkType linkTypeOfRangeElement(Object element) throws MappingException {
+    	ILinkType type = rangeLinkTypes.get(element.getClass());
+		if(type == null)  {
+			for (Class<?> clazz : element.getClass().getInterfaces()) {
+				type = rangeLinkTypes.get(clazz);
+				if (type != null)
+					return type;
+				
+			}
+			throw new MappingException("Didn't find a link type for " +	element + ".");
+		}
+		return type;
+    }
+
+    
+    public ILinkType linkTypeOfDomainType(ReadGraph g, Resource type)  {        
+    	return domainLinkTypes.get(type);
+    }
+    
+    public ILinkType linkTypeOfRangeType(Class<?> clazz) {
+    	ILinkType type = rangeLinkTypes.get(clazz);
+    	if(type == null)  {
+			for (Class<?> c : clazz.getInterfaces()) {
+				type = rangeLinkTypes.get(clazz);
+				if (type != null)
+					return type;
+				
+			}
+		}
+    	return null;
+    }
+}