import org.opengis.referencing.operation.TransformException;
import org.simantics.Simantics;
import org.simantics.databoard.Bindings;
-import org.simantics.db.ReadGraph;
import org.simantics.db.Resource;
import org.simantics.db.WriteGraph;
-import org.simantics.db.common.request.ObjectsWithType;
-import org.simantics.db.common.request.UniqueRead;
import org.simantics.db.common.request.WriteRequest;
import org.simantics.db.exception.DatabaseException;
import org.simantics.db.layer0.util.Layer0Utils;
-import org.simantics.diagram.stubs.DiagramResource;
import org.simantics.district.network.DNEdgeBuilder;
import org.simantics.district.network.DistrictNetworkUtil;
import org.simantics.district.network.ontology.DistrictNetworkResource;
-import org.simantics.layer0.Layer0;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
-import org.simantics.district.network.DistrictNetworkUtil.ResourceVertex;
-import com.vividsolutions.jts.geom.Envelope;
import com.vividsolutions.jts.index.quadtree.Quadtree;
public class DistrictImportUtils {
double halfPadding = padding / 2;
- Quadtree vv = Simantics.getSession().syncRequest(new UniqueRead<Quadtree>() {
-
- @Override
- public Quadtree perform(ReadGraph graph) throws DatabaseException {
- Collection<Resource> vertices = graph.syncRequest(new ObjectsWithType(model.getParentDiagram(), Layer0.getInstance(graph).ConsistsOf, DistrictNetworkResource.getInstance(graph).Vertex));
- Quadtree vv = new Quadtree();
- for (Resource vertex : vertices) {
- double[] coords = graph.getRelatedValue2(vertex, DiagramResource.getInstance(graph).HasLocation, Bindings.DOUBLE_ARRAY);
- double x1 = coords[0] - halfPadding;
- double y1= coords[1] - halfPadding;
- double x2 = coords[0] + halfPadding;
- double y2= coords[1] + halfPadding;
- Envelope e = new Envelope(x1, x2, y1, y2);
- vv.insert(e, new ResourceVertex(vertex, coords, true));
- }
- return vv;
- }
- });
+ Quadtree existingVertices = DistrictNetworkUtil.existingVertices(model.getParentDiagram(), halfPadding);
Simantics.getSession().syncRequest(new WriteRequest() {
// Switch to (longitude, latitude)
flipAxes(startCoords);
flipAxes(endCoords);
+
+ if ("6993003.00000".equals(idValue)) {
+ System.err.println("ghere we are!!");
+ }
- Optional<Resource> oedge = DNEdgeBuilder.create(graph, vv, model.getParentDiagram(), model.getComponentMappings().get(mappingValue), startCoords, startZCoord, endCoords, endZCoord, new double[0], padding, true);
+ Optional<Resource> oedge = DNEdgeBuilder.create(graph, existingVertices, model.getParentDiagram(), model.getComponentMappings().get(mappingValue), startCoords, startZCoord, endCoords, endZCoord, new double[0], padding, true);
if (oedge.isPresent()) {
Resource edge = oedge.get();
import java.util.stream.Collectors;
import java.util.stream.Stream;
+import org.simantics.Simantics;
import org.simantics.databoard.Bindings;
import org.simantics.datatypes.literal.RGB;
import org.simantics.datatypes.literal.RGB.Integer;
import org.simantics.db.Resource;
import org.simantics.db.WriteGraph;
import org.simantics.db.common.procedure.adapter.TransientCacheListener;
+import org.simantics.db.common.request.BinaryRead;
import org.simantics.db.common.request.IndexRoot;
+import org.simantics.db.common.request.ObjectsWithType;
import org.simantics.db.common.request.ResourceRead;
import org.simantics.db.common.utils.OrderedSetUtils;
import org.simantics.db.exception.BindingException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
+import com.vividsolutions.jts.geom.Envelope;
+import com.vividsolutions.jts.index.quadtree.Quadtree;
+
public class DistrictNetworkUtil {
private static final Logger LOGGER = LoggerFactory.getLogger(DistrictNetworkUtil.class);
return results;
}
+ public static List<ResourceVertex> nearbyResourceVertices(ReadGraph graph, Resource diagramResource, Resource vertex, Double padding) throws DatabaseException {
+ double halfPadding = padding / 2;
+
+ Quadtree existingVertices = graph.syncRequest(new ExistingVerticesRead(diagramResource, halfPadding), TransientCacheListener.instance());
+ double[] coords = graph.getRelatedValue(vertex, DiagramResource.getInstance(graph).HasLocation, Bindings.DOUBLE_ARRAY);
+ double x1 = coords[0] - halfPadding;
+ double y1= coords[1] - halfPadding;
+ double x2 = coords[0] + halfPadding;
+ double y2= coords[1] + halfPadding;
+ Envelope e = new Envelope(x1, x2, y1, y2);
+
+ List<?> result = existingVertices.query(e);
+ @SuppressWarnings("unchecked")
+ List<ResourceVertex> vertices = (List<ResourceVertex>) result;
+
+ Rectangle2D vertexRectangle = new Rectangle2D.Double(coords[0] - halfPadding, coords[1] - halfPadding, padding, padding);
+
+ // let's sort by distance
+ List<ResourceVertex> sortedVertices = vertices.stream().filter(rv -> {
+ if (rv.vertex.equals(vertex))
+ return false;
+
+ Rectangle2D nearbyRectangle = new Rectangle2D.Double(rv.coords[0] - halfPadding, rv.coords[1] - halfPadding, padding, padding);
+ return vertexRectangle.intersects(nearbyRectangle);
+ }).sorted((o1, o2) -> {
+ double disto1 = Math.sqrt((Math.pow(coords[0] - o1.coords[0], 2) + (Math.pow(coords[1] - o1.coords[1], 2))));
+ double disto2 = Math.sqrt((Math.pow(coords[0] - o2.coords[0], 2) + (Math.pow(coords[1] - o2.coords[1], 2))));
+
+ if (o1.vertex.getResourceId() == 2554883) {
+ System.err.println("here we are");
+ }
+
+ return Double.compare(disto1, disto2);
+ }).collect(Collectors.toList());
+ return sortedVertices;
+ }
+
+ public static List<Resource> nearbyVertices(ReadGraph graph, Resource vertex, double padding) throws DatabaseException {
+ Resource diagramResource = graph.getSingleObject(vertex, Layer0.getInstance(graph).PartOf);
+ return nearbyResourceVertices(graph, diagramResource, vertex, padding)
+ .stream()
+ .map(rv -> rv.vertex)
+ .collect(Collectors.toList());
+ }
+
+ public static Quadtree existingVertices(Resource diagramResource, Double padding) throws DatabaseException {
+ Quadtree vv = Simantics.getSession().syncRequest(new ExistingVerticesRead(diagramResource, padding));
+ return vv;
+ }
+
+ public static class ExistingVerticesRead extends BinaryRead<Resource, Double, Quadtree> {
+
+ public ExistingVerticesRead(Resource diagramResource, Double padding) {
+ super(diagramResource, padding);
+ }
+
+ @Override
+ public Quadtree perform(ReadGraph graph) throws DatabaseException {
+ Collection<Resource> vertices = graph.syncRequest(new ObjectsWithType(parameter, Layer0.getInstance(graph).ConsistsOf, DistrictNetworkResource.getInstance(graph).Vertex));
+ Quadtree vv = new Quadtree();
+ for (Resource vertex : vertices) {
+ double[] coords = graph.getRelatedValue2(vertex, DiagramResource.getInstance(graph).HasLocation, Bindings.DOUBLE_ARRAY);
+ double x1 = coords[0] - parameter2;
+ double y1= coords[1] - parameter2;
+ double x2 = coords[0] + parameter2;
+ double y2= coords[1] + parameter2;
+ Envelope e = new Envelope(x1, x2, y1, y2);
+ vv.insert(e, new ResourceVertex(vertex, coords, true));
+ }
+ return vv;
+ }
+ }
}