bundles/org.simantics.objmap2/src/org/simantics/objmap/structural/schema/SimpleLinkType.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2013 Association for Decentralized Information Management\r
   3  * in Industry THTH ry.\r
   4  * All rights reserved. This program and the accompanying materials\r
   5  * are made available under the terms of the Eclipse Public License v1.0\r
   6  * which accompanies this distribution, and is available at\r
   7  * http://www.eclipse.org/legal/epl-v10.html\r
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.objmap.structural.schema;\r
  13 \r
  14 import java.util.ArrayList;\r
  15 import java.util.Collections;\r
  16 import java.util.List;\r
  17 \r
  18 import org.apache.log4j.Logger;\r
  19 import org.simantics.db.ReadGraph;\r
  20 import org.simantics.db.Resource;\r
  21 import org.simantics.db.WriteGraph;\r
  22 import org.simantics.db.common.utils.NameUtils;\r
  23 import org.simantics.db.exception.DatabaseException;\r
  24 import org.simantics.layer0.Layer0;\r
  25 import org.simantics.objmap.backward.IBackwardMapping;\r
  26 import org.simantics.objmap.bidirectional.IBidirectionalMappingRule;\r
  27 import org.simantics.objmap.exceptions.MappingException;\r
  28 import org.simantics.objmap.forward.IForwardMapping;\r
  29 import org.simantics.objmap.graph.schema.ILinkType;\r
  30 import org.simantics.objmap.structural.IStructuralObject;\r
  31 import org.simantics.objmap.structural.StructuralResource;\r
  32 \r
  33 \r
  34 \r
  35 public class SimpleLinkType implements ILinkType<StructuralResource,IStructuralObject> {\r
  36     \r
  37     static Logger LOGGER = Logger.getLogger("org.simantics.objmap");\r
  38     \r
  39     public Resource domainType;\r
  40     public Class<?> rangeType;\r
  41     ArrayList<IBidirectionalMappingRule<StructuralResource,IStructuralObject>> rules;\r
  42     \r
  43     public SimpleLinkType(Resource domainType, Class<?> rangeType,\r
  44             ArrayList<IBidirectionalMappingRule<StructuralResource,IStructuralObject>> rules) {\r
  45         this.domainType = domainType;\r
  46         this.rangeType = rangeType;\r
  47         this.rules = rules;\r
  48     }\r
  49 \r
  50     public SimpleLinkType(Resource domainType, Class<?> rangeType) {\r
  51         this(domainType, rangeType, new ArrayList<IBidirectionalMappingRule<StructuralResource,IStructuralObject>>());\r
  52     }\r
  53 \r
  54     /**\r
  55      * Adds a new rule to this link type that is enforced\r
  56      * during updates.\r
  57      */\r
  58     public void addRule(IBidirectionalMappingRule<StructuralResource,IStructuralObject> rule) {\r
  59         rules.add(rule);\r
  60     }\r
  61     \r
  62     @Override\r
  63     public StructuralResource createDomainElement(WriteGraph g, IStructuralObject rangeElement)\r
  64             throws MappingException {\r
  65         try {\r
  66             if(LOGGER.isInfoEnabled())\r
  67                 LOGGER.info("SimpleLinkType.createDomainElement " +\r
  68                         rangeElement.toString()\r
  69                 );\r
  70             if (rangeElement.getContext().size() == 0) {\r
  71                 // there is no context, this not a structural resource / object.\r
  72                 Resource result = newResource(g, domainType);\r
  73                 return new StructuralResource(g,result);\r
  74             } else {\r
  75                 if (rangeElement.getContext().size() == 1 && rangeElement.getContext().get(0).equals(rangeElement)) {\r
  76                         // Structural object's context is itself, we are instantiating a new structural model.\r
  77                         Resource type = rangeElement.getType();\r
  78                         Resource result = newResource(g, type);\r
  79                         return new StructuralResource(g,result,result);\r
  80                 } else {\r
  81                         // Structural object's context is not itself, which means that the object is inside of a structural model.\r
  82                         // At the moment we do not support modifying instantiated structural models.\r
  83                         throw new MappingException("Cannot create a new StucturalObject " + rangeElement + " " + rangeElement.getClass());\r
  84                 }\r
  85             }\r
  86         } catch(DatabaseException e) {\r
  87             throw new MappingException(e);\r
  88         }\r
  89     }\r
  90     \r
  91     protected Resource newResource(WriteGraph g, Resource type) throws DatabaseException {\r
  92         Resource result = g.newResource();\r
  93         g.claim(result, Layer0.getInstance(g).InstanceOf, null, type);\r
  94         return result;\r
  95     }\r
  96     @Override\r
  97     public IStructuralObject createRangeElement(ReadGraph g, StructuralResource domainElement)\r
  98             throws MappingException {\r
  99         try {\r
 100             if(LOGGER.isInfoEnabled())\r
 101                 try { \r
 102                     LOGGER.info("SimpleLinkType.createRangeElement " + NameUtils.getSafeName(g, domainElement.getResource()));\r
 103                 } catch(DatabaseException e) {\r
 104                     throw new MappingException(e);\r
 105                 }\r
 106             IStructuralObject result = (IStructuralObject)rangeType.newInstance();\r
 107             if (domainElement.getContext().size() == 1) {\r
 108                 if (domainElement.getContext().get(0).equals(domainElement.getResource()))\r
 109                         result.setContext(Collections.singletonList(result));\r
 110                 else {\r
 111                         //result.setContext(result); \r
 112                 }\r
 113             }\r
 114             return result;\r
 115         } catch (InstantiationException e) {\r
 116             throw new MappingException(e);\r
 117         } catch (IllegalAccessException e) {\r
 118             throw new MappingException(e);\r
 119         }\r
 120     }\r
 121     \r
 122     @SuppressWarnings("unchecked")\r
 123         public void createDomain(WriteGraph graph, IBackwardMapping<StructuralResource,IStructuralObject> mapping, StructuralResource domainElement, IStructuralObject rangeElement) throws MappingException {\r
 124         if (domainElement.isStructuralRoot())\r
 125                 // FIXME: this is nasty, but when a structural model is instantiated by creating new IStructuralObject, its related objects must be read from the graph first, or otherwise the objects would be deleted from the graph.\r
 126                 //        as a side effect, if the new IStructuralObject has any properties set, those properties are set to null (because the graph does not contain those values).\r
 127                 // \r
 128                 updateRange(graph, (IForwardMapping<StructuralResource, IStructuralObject>)mapping, domainElement, rangeElement);\r
 129         updateDomain(graph, mapping, domainElement, rangeElement);\r
 130     };\r
 131     \r
 132     \r
 133     @Override\r
 134     public void createRange(ReadGraph graph, IForwardMapping<StructuralResource, IStructuralObject> mapping, StructuralResource domainElement, IStructuralObject rangeElement) throws MappingException {\r
 135         if (rangeElement.getContext().size() == 0 && domainElement.getContext().size() > 0) {\r
 136                 List<IStructuralObject> ctx = new ArrayList<IStructuralObject>(domainElement.getContext().size());\r
 137                         try {\r
 138                                 List<Resource> context = new ArrayList<Resource>();\r
 139                                 for (int i = 0; i <domainElement.getContext().size(); i++) {\r
 140                                         context.add(domainElement.getContext().get(i));\r
 141                                         IStructuralObject ctxObj = mapping.get(new StructuralResource(graph,context.get(context.size()-1),context));\r
 142                                         if (ctxObj == null) throw new MappingException("Cannot resolve range context for domain element " + domainElement);\r
 143                                         ctx.add(ctxObj);\r
 144                                 }\r
 145                                 //ctx = mapping.get(new StructuralResource(graph,context.get(context.size()-1),context));\r
 146                         } catch (DatabaseException e) {\r
 147                                 throw new MappingException(e);\r
 148                         }\r
 149                 if (ctx.size() == 0)\r
 150                         throw new MappingException("Cannot find context for structural object, " + domainElement);\r
 151                 rangeElement.setContext(ctx);\r
 152         }\r
 153         updateRange(graph, mapping, domainElement, rangeElement);\r
 154     }\r
 155     \r
 156     public boolean updateDomain(WriteGraph g, IBackwardMapping<StructuralResource,IStructuralObject> map, StructuralResource domainElement, IStructuralObject rangeElement) throws MappingException {\r
 157         if(LOGGER.isInfoEnabled())\r
 158             try { \r
 159                 LOGGER.info("SimpleLinkType.updateDomain " +\r
 160                         NameUtils.getSafeName(g, domainElement.getResource()) + " " +\r
 161                         rangeElement.toString()\r
 162                         );\r
 163             } catch(DatabaseException e) {\r
 164                 throw new MappingException(e);\r
 165             }\r
 166         \r
 167         boolean updated = false;\r
 168         for(IBidirectionalMappingRule<StructuralResource, IStructuralObject> rule : rules)\r
 169                 updated |= rule.updateDomain(g, map, domainElement, rangeElement);\r
 170         return updated;\r
 171     }\r
 172     \r
 173     public boolean updateRange(ReadGraph g, IForwardMapping<StructuralResource, IStructuralObject> map, StructuralResource domainElement, IStructuralObject rangeElement) throws MappingException {\r
 174     \r
 175         if(LOGGER.isInfoEnabled())\r
 176             try { \r
 177                 LOGGER.info("SimpleLinkType.updateRange " +\r
 178                                 NameUtils.getSafeName(g, domainElement.getResource()) + " " +\r
 179                         rangeElement.toString()\r
 180                         );\r
 181             } catch(DatabaseException e) {\r
 182                 throw new MappingException(e);\r
 183             }\r
 184         \r
 185         boolean updated = false;\r
 186         for(IBidirectionalMappingRule<StructuralResource, IStructuralObject> rule : rules)\r
 187             updated |= rule.updateRange(g, map, domainElement, rangeElement);\r
 188         return updated;\r
 189     }\r
 190 }\r