X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=org.simantics.plant3d%2Fsrc%2Forg%2Fsimantics%2Fplant3d%2Futils%2FP3DUtil.java;h=71f844d866329fe2dac458d7e9c468c6eb28ce33;hb=bbf50fc505117a5269f0aaff504eaa3dc57b9d31;hp=e6e3385f3a2967aaee826a684136ac98be7dce50;hpb=a7d96d461b5ead5ace673d3f6f9f8ec3d19118eb;p=simantics%2F3d.git

diff --git a/org.simantics.plant3d/src/org/simantics/plant3d/utils/P3DUtil.java b/org.simantics.plant3d/src/org/simantics/plant3d/utils/P3DUtil.java
index e6e3385f..71f844d8 100644
--- a/org.simantics.plant3d/src/org/simantics/plant3d/utils/P3DUtil.java
+++ b/org.simantics.plant3d/src/org/simantics/plant3d/utils/P3DUtil.java
@@ -11,20 +11,30 @@ import org.simantics.db.Resource;
 import org.simantics.db.WriteGraph;
 import org.simantics.db.exception.DatabaseException;
 import org.simantics.db.request.Read;
+import org.simantics.g3d.scenegraph.base.INode;
 import org.simantics.layer0.Layer0;
 import org.simantics.plant3d.ontology.Plant3D;
+import org.simantics.plant3d.scenegraph.Equipment;
+import org.simantics.plant3d.scenegraph.P3DRootNode;
+import org.simantics.plant3d.scenegraph.PipeRun;
+import org.simantics.plant3d.scenegraph.PipelineComponent;
+import org.simantics.plant3d.scenegraph.controlpoint.PipeControlPoint;
+import org.simantics.plant3d.scenegraph.controlpoint.PipingRules;
 import org.simantics.plant3d.utils.Item.Type;
-import org.simantics.ui.SimanticsUI;
 
 public class P3DUtil {
 	
 	public static List<Item> getEquipments() throws DatabaseException {
+		return getEquipments(Plant3D.URIs.Builtin);
+	}
+	
+	public static List<Item> getEquipments(final String libUri) throws DatabaseException {
 		return Simantics.getSession().syncRequest(new Read<List<Item>>() {
 			@Override
 			public List<Item> perform(ReadGraph graph) throws DatabaseException {
 				Plant3D p3d = Plant3D.getInstance(graph);
 				Resource project = Simantics.getProject().get();
-				Resource builtins = graph.getResource(Plant3D.URIs.Builtin);
+				Resource builtins = graph.getResource(libUri);
 				List<Item> actions = getItems(graph, project,p3d.Equipment);
 				actions.addAll(getItems(graph, builtins,p3d.Equipment));
 				return actions;
@@ -39,26 +49,43 @@ public class P3DUtil {
 			@Override
 			public List<Item> perform(ReadGraph graph) throws DatabaseException {
 				Plant3D p3d = Plant3D.getInstance(graph);
-				ItemQuery query = new ItemQuery(p3d.Nozzle);
+				ItemQuery query = new ItemQuery(p3d.Nozzle, Plant3D.URIs.Builtin);
 				return graph.syncRequest(query);
 			}
 		});
 	}
 	
-	private static class ItemQuery implements Read<List<Item>> {
+	public static class ItemQuery implements Read<List<Item>> {
 		private Resource type;
-		public ItemQuery(Resource type) {
+		private String libUri;
+		public ItemQuery(Resource type, String libUri) {
 			this.type = type;
+			this.libUri = libUri;
 		}
 		
 		@Override
 		public List<Item> perform(ReadGraph graph) throws DatabaseException {
 			Resource project = Simantics.getProject().get();
-			Resource builtins = graph.getResource(Plant3D.URIs.Builtin);
+			Resource builtins = graph.getResource(libUri);
 			List<Item> actions = getItems(graph, project,type);
 			actions.addAll(getItems(graph, builtins,type));
 			return actions;
 		}
+		
+		@Override
+		public boolean equals(Object obj) {
+			if (obj.getClass() != this.getClass())
+				return false;
+			ItemQuery other = (ItemQuery)obj;
+			if (!type.equals(other.type))
+				return false;
+			return libUri.equals(other.libUri);
+		}
+		
+		@Override
+		public int hashCode() {
+			return type.hashCode() + libUri.hashCode();
+		}
 	}
 	
 	public static List<Item> getEnds() throws DatabaseException {
@@ -66,7 +93,7 @@ public class P3DUtil {
 			@Override
 			public List<Item> perform(ReadGraph graph) throws DatabaseException {
 				Plant3D p3d = Plant3D.getInstance(graph);
-				ItemQuery query = new ItemQuery(p3d.EndComponent);
+				ItemQuery query = new ItemQuery(p3d.EndComponent, Plant3D.URIs.Builtin);
 				return graph.syncRequest(query);
 			}
 		});
@@ -77,7 +104,7 @@ public class P3DUtil {
 			@Override
 			public List<Item> perform(ReadGraph graph) throws DatabaseException {
 				Plant3D p3d = Plant3D.getInstance(graph);
-				ItemQuery query = new ItemQuery(p3d.TurnComponent);
+				ItemQuery query = new ItemQuery(p3d.TurnComponent, Plant3D.URIs.Builtin);
 				return graph.syncRequest(query);
 			}
 		});
@@ -88,19 +115,19 @@ public class P3DUtil {
 			@Override
 			public List<Item> perform(ReadGraph graph) throws DatabaseException {
 				Plant3D p3d = Plant3D.getInstance(graph);
-				ItemQuery query = new ItemQuery(p3d.InlineComponent);
+				ItemQuery query = new ItemQuery(p3d.InlineComponent, Plant3D.URIs.Builtin);
 				return graph.syncRequest(query);
 			}
 		});
 	}
 	
 	public static List<Item> filterUserComponents(List<Item> items) {
-	    List<Item> result = new ArrayList<Item>(items.size());
-	    for (Item i : items) {
-	        if (!i.isCode())
-	            result.add(i);
-	    }
-	    return result;
+		List<Item> result = new ArrayList<Item>(items.size());
+		for (Item i : items) {
+			if (!i.isCode())
+				result.add(i);
+		}
+		return result;
 	}
 	
 	private static List<Item> getItems(ReadGraph graph, Resource lib, Resource type) throws DatabaseException{
@@ -129,11 +156,11 @@ public class P3DUtil {
 			}
 		}
 		Collections.sort(result, new Comparator<Item>() {
-		    @Override
-		    public int compare(Item o1, Item o2) {
-		        return o1.getName().compareTo(o2.getName());
-		    }
-        });
+			@Override
+			public int compare(Item o1, Item o2) {
+				return o1.getName().compareTo(o2.getName());
+			}
+		});
 		return result;
 	}
 	
@@ -159,7 +186,7 @@ public class P3DUtil {
 		if (graph.hasStatement(r, p3d.CodeComponent))
 			item.setCode(true);
 		if (graph.hasStatement(r, p3d.VariableAngleTurnComponent) ||
-		    graph.hasStatement(r, p3d.VariableLengthInlineComponent))
+			graph.hasStatement(r, p3d.VariableLengthInlineComponent))
 			item.setVariable(true);
 		if (graph.hasStatement(r, p3d.SizeChangeComponent))
 			item.setSizeChange(true);
@@ -175,5 +202,44 @@ public class P3DUtil {
 		
 		return model;
 	}
+	
+	public static void finalizeDBLoad(P3DRootNode rootNode) throws Exception{
+		for (INode node : rootNode.getChild()) {
+			if (node instanceof PipeRun) {
+				for (PipelineComponent pc : ((PipeRun) node).getChild())
+					pc.sync();
+			} else if (node instanceof Equipment) {
+				for (PipelineComponent pc : ((Equipment) node).getChild())
+					pc.sync();
+			}
+		}
+		
+		for (INode node : rootNode.getChild()) {
+			if (node instanceof PipeRun) {
+				for (PipelineComponent pc : ((PipeRun) node).getChild())
+					pc.sync2();
+			} else if (node instanceof Equipment) {
+				for (PipelineComponent pc : ((Equipment) node).getChild())
+					pc.sync2();
+			}
+		}
+		for (INode node : rootNode.getChild()) {
+			if (node instanceof PipeRun) {
+				PipingRules.validate((PipeRun)node);
+			}
+		}
+	}
+	
+   public static void finalizeDBLoad2(P3DRootNode rootNode) throws Exception{
+       PipingRules.setEnabled(true);
+        for (INode node : rootNode.getChild()) {
+            if (node instanceof PipeRun) {
+                PipeRun run = (PipeRun)node;
+                for (PipeControlPoint pcp : run.getControlPoints())
+                    PipingRules.positionUpdate(pcp);
+                    //PipingRules.requestUpdate(pcp);
+            }
+        }
+   }
 
 }