bundles/org.simantics.diagram/src/org/simantics/diagram/adapter/ConnectionInfoRequest.java

   1 /*******************************************************************************\r
   2  * Copyright (c) 2007, 2010 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.diagram.adapter;\r
  13 \r
  14 import java.util.concurrent.atomic.AtomicInteger;\r
  15 \r
  16 import org.simantics.db.AsyncReadGraph;\r
  17 import org.simantics.db.Resource;\r
  18 import org.simantics.db.common.request.UnaryAsyncRead;\r
  19 import org.simantics.db.procedure.AsyncProcedure;\r
  20 import org.simantics.diagram.connection.ConnectionSegmentEnd;\r
  21 import org.simantics.diagram.connection.ConnectionSegmentType;\r
  22 import org.simantics.diagram.content.EdgeResource;\r
  23 import org.simantics.diagram.stubs.DiagramResource;\r
  24 \r
  25 /**\r
  26  * @author Antti Villberg\r
  27  *\r
  28  */\r
  29 public class ConnectionInfoRequest extends UnaryAsyncRead<EdgeResource, ConnectionInfo> {\r
  30 \r
  31     static class SingleConnectionInfo {\r
  32         public Resource connection;\r
  33         public ConnectionSegmentEnd end;\r
  34     }\r
  35 \r
  36     final DiagramResource dr;\r
  37 \r
  38     private SingleConnectionInfo first;\r
  39     private SingleConnectionInfo second;\r
  40 \r
  41     public ConnectionInfoRequest(DiagramResource dr, EdgeResource resource) {\r
  42         super(resource);\r
  43         this.dr = dr;\r
  44     }\r
  45 \r
  46     public void fillPossibleType(AsyncReadGraph graph, Resource resource, final AsyncProcedure<ConnectionInfo> procedure, final ConnectionInfo result) {\r
  47         if (resource == null) {\r
  48             procedure.execute(graph, result);\r
  49         } else {\r
  50             graph.forSingleType(resource, dr.Connection, new AsyncProcedure<Resource>() {\r
  51                 @Override\r
  52                 public void exception(AsyncReadGraph graph, Throwable throwable) {\r
  53                     procedure.exception(graph, throwable);\r
  54                 }\r
  55 \r
  56                 @Override\r
  57                 public void execute(AsyncReadGraph graph, Resource type) {\r
  58                     result.connectionType = type;\r
  59                     procedure.execute(graph, result);\r
  60                 }\r
  61             });\r
  62         }\r
  63     }\r
  64 \r
  65     ConnectionInfo createConnectionInfo(Resource connection) {\r
  66         ConnectionInfo info = new ConnectionInfo();\r
  67         info.connection = connection;\r
  68         info.firstEnd = first.end;\r
  69         info.secondEnd = second.end;\r
  70         info.segmentType = ConnectionSegmentType.toSegmentType(first.end, second.end);\r
  71         return info;\r
  72     }\r
  73 \r
  74     public void isBranchPointOf(AsyncReadGraph graph, Resource resource, final AsyncProcedure<ConnectionInfo> procedure, final SingleConnectionInfo result, final AtomicInteger ready) {\r
  75 \r
  76         graph.forPossibleObject(resource, dr.IsBranchPointOf, new AsyncProcedure<Resource>() {\r
  77             @Override\r
  78             public void exception(AsyncReadGraph graph, Throwable throwable) {\r
  79                 procedure.exception(graph, throwable);\r
  80             }\r
  81 \r
  82             @Override\r
  83             public void execute(AsyncReadGraph graph, Resource connection) {\r
  84                 result.end = ConnectionSegmentEnd.BRANCH;\r
  85                 result.connection = connection;\r
  86                 if (ready.decrementAndGet() == 0) {\r
  87                     ConnectionInfo info = createConnectionInfo(connection);\r
  88                     fillPossibleType(graph, connection, procedure, info);\r
  89                 }\r
  90             }\r
  91         });\r
  92 \r
  93     }\r
  94 \r
  95     public void isConnectorOf(AsyncReadGraph graph, final Resource resource, final AsyncProcedure<ConnectionInfo> procedure, final SingleConnectionInfo result, final AtomicInteger ready) {\r
  96 \r
  97         graph.forPossibleObject(resource, dr.IsConnectorOf, new AsyncProcedure<Resource>() {\r
  98             @Override\r
  99             public void exception(AsyncReadGraph graph, Throwable throwable) {\r
 100                 procedure.exception(graph, throwable);\r
 101             }\r
 102 \r
 103             @Override\r
 104             public void execute(AsyncReadGraph graph, Resource connection) {\r
 105                 if (connection != null) {\r
 106                     result.end = ConnectionSegmentEnd.CONNECTOR;\r
 107                     result.connection = connection;\r
 108                     if (ready.decrementAndGet() == 0) {\r
 109                         ConnectionInfo info = createConnectionInfo(connection);\r
 110                         fillPossibleType(graph, connection, procedure, info);\r
 111                     }\r
 112                 } else {\r
 113                     isBranchPointOf(graph, resource, procedure, result, ready);\r
 114                 }\r
 115             }\r
 116         });\r
 117 \r
 118     }\r
 119 \r
 120     @Override\r
 121     public void perform(AsyncReadGraph graph, AsyncProcedure<ConnectionInfo> procedure) {\r
 122         AtomicInteger ready = new AtomicInteger(2);\r
 123         first = new SingleConnectionInfo();\r
 124         second = new SingleConnectionInfo();\r
 125         isConnectorOf(graph, parameter.first(), procedure, first, ready);\r
 126         isConnectorOf(graph, parameter.second(), procedure, second, ready);\r
 127     }\r
 128 \r
 129 }\r