bundles/org.simantics.objmap2/src/org/simantics/objmap/structural/rules/domain/RelatedObjectsAccessor.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.rules.domain;\r
  13 \r
  14 import java.util.ArrayList;\r
  15 import java.util.Collection;\r
  16 import java.util.Collections;\r
  17 import java.util.List;\r
  18 \r
  19 import org.apache.log4j.Logger;\r
  20 import org.simantics.db.ReadGraph;\r
  21 import org.simantics.db.Resource;\r
  22 import org.simantics.db.WriteGraph;\r
  23 import org.simantics.db.exception.DatabaseException;\r
  24 import org.simantics.objmap.exceptions.MappingException;\r
  25 import org.simantics.objmap.graph.rules.domain.IDomainAccessor;\r
  26 import org.simantics.objmap.graph.rules.domain.MappingUtils;\r
  27 import org.simantics.objmap.structural.StructuralResource;\r
  28 \r
  29 \r
  30 public class RelatedObjectsAccessor implements IDomainAccessor<StructuralResource,Collection<StructuralResource>> {\r
  31 \r
  32     static Logger LOGGER = Logger.getLogger("org.simantics.objmap");\r
  33     \r
  34         Resource relation;\r
  35         boolean deleteExtraObjects;\r
  36         boolean useTypeResource;\r
  37         \r
  38         private boolean preventStructuralChanges = true;\r
  39         private boolean preventStructuralRootChanges = true;\r
  40 \r
  41         public RelatedObjectsAccessor(Resource relation, boolean deleteExtraObjects, boolean useTypeResource) {\r
  42         super();\r
  43         this.relation = relation;\r
  44         this.deleteExtraObjects = deleteExtraObjects;\r
  45         this.useTypeResource = useTypeResource;\r
  46     }\r
  47         \r
  48         public RelatedObjectsAccessor(Resource relation, boolean deleteExtraObjects, boolean useTypeResource, boolean preventStructuralChanges, boolean preventStructuralRootChanges) {\r
  49         super();\r
  50         this.relation = relation;\r
  51         this.deleteExtraObjects = deleteExtraObjects;\r
  52         this.useTypeResource = useTypeResource;\r
  53                 this.preventStructuralChanges = preventStructuralChanges;\r
  54                 this.preventStructuralRootChanges = preventStructuralRootChanges;\r
  55     }\r
  56 \r
  57     @Override\r
  58         public Collection<StructuralResource> get(ReadGraph g, StructuralResource element) throws MappingException {\r
  59                 try {\r
  60                     LOGGER.info("        RelatedObjectsAccessor.get");\r
  61                         \r
  62                    \r
  63                     Resource res = getServiceResource(g, element);\r
  64                     if (res == null)\r
  65                         return Collections.emptyList();\r
  66                     Collection<Resource> coll = g.getObjects(res, relation);\r
  67                         List<StructuralResource> result = new ArrayList<StructuralResource>(coll.size());\r
  68                         for (Resource r : coll) {\r
  69                                 if (StructuralUtils.isStructuralInstance(g, r)) {\r
  70                                         result.add(new StructuralResource(g, r, element.getContext(),r));\r
  71                                 } else {\r
  72                                         result.add(new StructuralResource(g, r, element.getContext()));\r
  73                                 }\r
  74                         }\r
  75                         return result;\r
  76                 } catch (DatabaseException e) {\r
  77                         throw new MappingException(e);\r
  78                 }\r
  79         }\r
  80     \r
  81     private boolean preventChange(StructuralResource element) {\r
  82         return preventStructuralChanges && element.isStructural() && (!element.isStructuralRoot()||preventStructuralRootChanges);       \r
  83     }\r
  84         \r
  85         @Override\r
  86         public boolean set(WriteGraph g, StructuralResource element, Collection<StructuralResource> value)\r
  87                         throws MappingException {\r
  88                 try {\r
  89                     LOGGER.info("        RelatedObjectsAccessor.set");\r
  90                     Resource res = getServiceResource(g, element);\r
  91                     if (res == null)\r
  92                         return false;\r
  93                     if (preventChange(element))\r
  94                                 return false;\r
  95                     Resource[] arr = new Resource[value.size()];\r
  96                     int i = 0; \r
  97                     for (StructuralResource sr : value) {\r
  98                         arr[i++] = sr.getResource();\r
  99                     }\r
 100                         return MappingUtils.synchronizeStatements(g, res, relation, \r
 101                                 arr, deleteExtraObjects);\r
 102                 } catch (DatabaseException e) {\r
 103                         throw new MappingException(e);\r
 104                 }\r
 105                 \r
 106         }\r
 107 \r
 108         private Resource getServiceResource(ReadGraph g, StructuralResource element) {\r
 109                 if (!useTypeResource)\r
 110                         return element.getResource();\r
 111                 return element.getTypeResource();\r
 112         }\r
 113 }\r