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

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