bundles/org.simantics.annotation.ui/src/org/simantics/annotation/ui/actions/AnnotationDropActionFactory.java

   1 package org.simantics.annotation.ui.actions;\r
   2 \r
   3 import java.util.HashSet;\r
   4 import java.util.List;\r
   5 import java.util.Set;\r
   6 \r
   7 import org.simantics.Simantics;\r
   8 import org.simantics.annotation.ontology.AnnotationResource;\r
   9 import org.simantics.db.ReadGraph;\r
  10 import org.simantics.db.Resource;\r
  11 import org.simantics.db.WriteGraph;\r
  12 import org.simantics.db.common.CommentMetadata;\r
  13 import org.simantics.db.common.request.WriteRequest;\r
  14 import org.simantics.db.common.utils.NameUtils;\r
  15 import org.simantics.db.exception.DatabaseException;\r
  16 import org.simantics.db.layer0.SelectionHints;\r
  17 import org.simantics.db.layer0.adapter.DropActionFactory;\r
  18 import org.simantics.db.layer0.request.PossibleModel;\r
  19 import org.simantics.layer0.Layer0;\r
  20 import org.simantics.utils.ObjectUtils;\r
  21 import org.simantics.utils.datastructures.Callback;\r
  22 import org.simantics.utils.ui.ErrorLogger;\r
  23 import org.simantics.utils.ui.ISelectionUtils;\r
  24 \r
  25 /**\r
  26  * @author Jani Simomaa\r
  27  */\r
  28 \r
  29 public class AnnotationDropActionFactory implements DropActionFactory {\r
  30 \r
  31         @Override\r
  32         public Runnable create(ReadGraph graph, Object target, Object source, int operation)\r
  33                         throws DatabaseException {\r
  34 \r
  35                 final Resource targetFolder = ISelectionUtils.getSinglePossibleKey(target, SelectionHints.KEY_MAIN, Resource.class);\r
  36                 if (targetFolder == null) {\r
  37                         return null;\r
  38                 }\r
  39                 Resource targetModel = graph.syncRequest(new PossibleModel(targetFolder));\r
  40                 \r
  41                 List<Resource> resources = ISelectionUtils.getPossibleKeys(source, SelectionHints.KEY_MAIN, Resource.class);\r
  42                 AnnotationResource ANNO = AnnotationResource.getInstance(graph);\r
  43                 Set<Resource> movedItems = new HashSet<Resource>();\r
  44                 for (Resource resource : resources) {\r
  45                         if (graph.isInstanceOf(resource, ANNO.AnnotationType) || graph.isInstanceOf(resource, ANNO.Annotation)) {\r
  46                                 Resource model = graph.syncRequest(new PossibleModel(resource));\r
  47                                 if (ObjectUtils.objectEquals(targetModel, model)) {\r
  48                                         movedItems.add(resource);\r
  49                                 }\r
  50                         }\r
  51                 }\r
  52                 if (!movedItems.isEmpty()) {\r
  53                         return addAnnotations(targetFolder, movedItems);\r
  54                 }\r
  55                 \r
  56                 return null;\r
  57         }\r
  58         \r
  59         \r
  60         private Runnable addAnnotations(final Resource targetFolder, final Set<Resource> movedAnnotationItems) {\r
  61                 \r
  62                 return new Runnable() {\r
  63                         \r
  64                         @Override\r
  65                         public void run() {\r
  66                                 Simantics.getSession().asyncRequest(new WriteRequest() {\r
  67                                         \r
  68                                         @Override\r
  69                                         public void perform(WriteGraph graph) throws DatabaseException {\r
  70                                                 \r
  71                                                 Layer0 L0 = Layer0.getInstance(graph);\r
  72                                                 for (Resource annotation : movedAnnotationItems) {\r
  73                                                         graph.deny(annotation, L0.PartOf);\r
  74                                                         graph.claim(targetFolder, L0.ConsistsOf, annotation);\r
  75                                                 }\r
  76                                                         \r
  77                                                 String name = NameUtils.getSafeName(graph, targetFolder);\r
  78                                                 CommentMetadata cm = graph.getMetadata(CommentMetadata.class);\r
  79                                         graph.addMetadata(cm.add("Moved " + movedAnnotationItems.size() + " annotations to folder " + name + ", resource " + targetFolder));\r
  80                                         }\r
  81                                 }, new Callback<DatabaseException>() {\r
  82                                         \r
  83                                         @Override\r
  84                                         public void run(DatabaseException e) {\r
  85                                                 if (e!= null)\r
  86                                                         ErrorLogger.defaultLogError(e);\r
  87                                                 \r
  88                                         }\r
  89                                 });\r
  90                                 \r
  91                         }\r
  92                 };\r
  93         }\r
  94         \r
  95         \r
  96 }\r