+package org.simantics.district.network.profile;
+
+import org.simantics.db.ReadGraph;
+import org.simantics.db.Resource;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.diagram.profile.StyleBase;
+import org.simantics.district.network.ontology.DistrictNetworkResource;
+import org.simantics.scenegraph.INode;
+import org.simantics.scenegraph.g2d.nodes.ConnectionNode;
+import org.simantics.scenegraph.profile.EvaluationContext;
+import org.simantics.scenegraph.profile.common.ProfileVariables;
+
+public class EdgeNodeStyle extends StyleBase<Double> {
+
+ public EdgeNodeStyle() {
+ System.out.println("EdgeNodeStyle init");
+ }
+
+ @Override
+ public Double calculateStyle(ReadGraph graph, Resource runtimeDiagram, Resource entry, Resource groupItem) throws DatabaseException {
+ DistrictNetworkResource DN = DistrictNetworkResource.getInstance(graph);
+ Double diameter = graph.getPossibleRelatedValue(groupItem, DN.Edge_HasDiameter);
+ if (diameter != null) {
+ return diameter;
+ }
+ return super.calculateStyle(graph, runtimeDiagram, entry, groupItem);
+ }
+
+ @Override
+ public void applyStyleForNode(EvaluationContext observer, INode node, Double result) {
+ if (result != null) {
+ ConnectionNode n = (ConnectionNode) node;
+ for (INode nn : n.getNodes()) {
+ ProfileVariables.claimNodeProperty(nn, "stroke", result, observer);
+ }
+ }
+ }
+
+}