bundles/org.simantics.modeling/src/org/simantics/modeling/adapters/NewCompositeActionFactory.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2012, 2015 Association for Decentralized Information Management in\r
   3  * 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  *     Semantum Oy - platform #5873: customizable instantiation logic\r
  12  *******************************************************************************/\r
  13 package org.simantics.modeling.adapters;\r
  14 \r
  15 import org.eclipse.core.runtime.IProgressMonitor;\r
  16 import org.eclipse.core.runtime.IStatus;\r
  17 import org.eclipse.core.runtime.Status;\r
  18 import org.eclipse.core.runtime.jobs.Job;\r
  19 import org.simantics.DatabaseJob;\r
  20 import org.simantics.Simantics;\r
  21 import org.simantics.db.ReadGraph;\r
  22 import org.simantics.db.Resource;\r
  23 import org.simantics.db.Session;\r
  24 import org.simantics.db.WriteGraph;\r
  25 import org.simantics.db.common.request.UniqueRead;\r
  26 import org.simantics.db.common.request.WriteResultRequest;\r
  27 import org.simantics.db.common.utils.NameUtils;\r
  28 import org.simantics.db.exception.CancelTransactionException;\r
  29 import org.simantics.db.exception.DatabaseException;\r
  30 import org.simantics.db.layer0.adapter.ActionFactory;\r
  31 import org.simantics.db.request.Read;\r
  32 import org.simantics.db.request.WriteResult;\r
  33 import org.simantics.layer0.Layer0;\r
  34 import org.simantics.modeling.internal.Plugin;\r
  35 import org.simantics.scl.runtime.function.Function3;\r
  36 import org.simantics.structural.stubs.StructuralResource2;\r
  37 import org.simantics.structural2.utils.StructuralUtils;\r
  38 import org.simantics.ui.workbench.action.DefaultActions;\r
  39 import org.simantics.utils.logging.TimeLogger;\r
  40 \r
  41 public class NewCompositeActionFactory implements ActionFactory {\r
  42 \r
  43     Resource compositeType;\r
  44     String defaultName;\r
  45 \r
  46     public NewCompositeActionFactory(Resource compositeType, String defaultName) {\r
  47         this.compositeType = compositeType;\r
  48         this.defaultName = defaultName;\r
  49     }\r
  50 \r
  51     @Override\r
  52     public Runnable create(Object target_) {\r
  53         final Resource target = (Resource)target_;\r
  54         return new Runnable() {\r
  55             @Override\r
  56             public void run() {\r
  57                 TimeLogger.resetTimeAndLog("NewCompositeActionFactory: create composite");\r
  58                 Job job = new DatabaseJob("Create Diagram") {\r
  59                     @Override\r
  60                     protected IStatus run(IProgressMonitor monitor) {\r
  61                         monitor.beginTask("Create Diagram...", IProgressMonitor.UNKNOWN);\r
  62                         try {\r
  63                             createComposite(compositeType, defaultName, target);\r
  64                             return Status.OK_STATUS;\r
  65                         } catch (CancelTransactionException e) {\r
  66                             return new Status(IStatus.INFO , Plugin.PLUGIN_ID, e.getMessage());\r
  67                         } catch (DatabaseException e) {\r
  68                             return new Status(IStatus.ERROR, Plugin.PLUGIN_ID, e.getMessage(), e);\r
  69                         } finally {\r
  70                             monitor.done();\r
  71                         }\r
  72                     }\r
  73                 };\r
  74                 //job.setUser(true);\r
  75                 job.schedule();\r
  76             }\r
  77         };\r
  78     }\r
  79 \r
  80     private static Resource createComposite(Resource compositeType, String defaultName, Resource target) throws DatabaseException {\r
  81         Session session = Simantics.getSession();\r
  82         session.markUndoPoint();\r
  83 \r
  84         Function3<Resource, String, Resource, Resource> instantiator = session.syncRequest( possibleCompositeInstantiator(compositeType) );\r
  85         if (instantiator != null)\r
  86             return instantiator.apply(compositeType, defaultName, target);\r
  87 \r
  88         Resource composite = session.syncRequest( createCompositeRequest( target, defaultName, compositeType ) );\r
  89         DefaultActions.asyncPerformDefaultAction(session, composite, false, false, true);\r
  90         return composite;\r
  91     }\r
  92 \r
  93     private static Read<Function3<Resource, String, Resource, Resource>> possibleCompositeInstantiator(final Resource compositeType) {\r
  94         return new UniqueRead<Function3<Resource, String, Resource, Resource>>() {\r
  95             @Override\r
  96             public Function3<Resource, String, Resource, Resource> perform(ReadGraph graph) throws DatabaseException {\r
  97                 return graph.getPossibleRelatedValue2(\r
  98                         compositeType,\r
  99                         StructuralResource2.getInstance(graph).Composite_instantiationFunction);\r
 100             }\r
 101         };\r
 102     }\r
 103 \r
 104     public static WriteResult<Resource> createCompositeRequest(final Resource target, final String defaultName, final Resource compositeType) {\r
 105         return new WriteResultRequest<Resource>() {\r
 106             @Override\r
 107             public Resource perform(WriteGraph graph) throws DatabaseException {\r
 108                 return createComposite(graph, target, defaultName, compositeType);\r
 109             }\r
 110         };\r
 111     }\r
 112 \r
 113     public static Resource createComposite(WriteGraph graph, Resource target, String defaultName, Resource compositeType) throws DatabaseException {\r
 114         String name = NameUtils.findFreshName(graph, defaultName, target, Layer0.getInstance(graph).ConsistsOf, "%s%d");\r
 115         return StructuralUtils.newComponent(graph, target, name, compositeType);\r
 116     }\r
 117 \r
 118 }\r