import java.awt.geom.Path2D;
import java.util.ArrayList;
import java.util.Collection;
+import java.util.Collections;
+import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
+import java.util.Map;
import java.util.Set;
import org.simantics.Simantics;
import org.simantics.db.ReadGraph;
+import org.simantics.db.RequestProcessor;
import org.simantics.db.Resource;
import org.simantics.db.common.request.ResourceRead;
import org.simantics.db.common.utils.ListUtils;
import org.simantics.db.exception.DatabaseException;
-import org.simantics.db.exception.ServiceException;
+import org.simantics.db.layer0.QueryIndexUtils;
import org.simantics.db.layer0.request.ActiveModels;
import org.simantics.db.layer0.request.ActiveRuns;
import org.simantics.db.layer0.request.Configuration;
+import org.simantics.db.layer0.request.PossibleActiveModel;
import org.simantics.db.layer0.variable.RVI;
import org.simantics.db.layer0.variable.Variable;
import org.simantics.db.layer0.variable.Variables;
+import org.simantics.db.request.Read;
import org.simantics.diagram.stubs.DiagramResource;
+import org.simantics.district.network.ontology.DistrictNetworkResource;
import org.simantics.district.region.ontology.DiagramRegionsResource;
import org.simantics.district.selection.ElementSelector.AggregateCondition.Type;
import org.simantics.layer0.Layer0;
public class ElementSelector {
String name;
String expression;
- Integer count;
+
Resource resource;
Generator generator;
static StructuralResource2 STR;
static ModelingResources MOD;
static DiagramResource DIA;
+ static DistrictNetworkResource DN;
+
+ private static Logger LOGGER = LoggerFactory.getLogger(ElementSelector.class);
ElementSelector(ReadGraph graph, Resource resource) throws DatabaseException {
super();
STR = StructuralResource2.getInstance(graph);
MOD = ModelingResources.getInstance(graph);
DIA = DiagramResource.getInstance(graph);
+ DN = DistrictNetworkResource.getInstance(graph);
this.resource = resource;
- this.count = -1;
try {
this.name = graph.getRelatedValue(resource, L0.HasLabel);
- this.expression = buildExpression(graph, resource);
+ this.expression = getExpression(graph, resource);
} catch (DatabaseException e) {
LOG.error("Error reading element selector", e);
throw e;
}
}
- public static ElementSelector getSelector(ReadGraph graph, Resource resource) throws DatabaseException {
+ /**
+ * Instantiate an element selector object for an ES.Selection resource.
+ */
+ public static ElementSelector getSelector(RequestProcessor graph, Resource resource) throws DatabaseException {
return graph.syncRequest(new ElementSelectorQuery(resource));
}
+
+ /**
+ * Get an SQL-like textual description of an ES.Selection resource.
+ */
+ public static String getExpression(RequestProcessor graph, Resource resource) throws DatabaseException {
+ return graph.syncRequest(new SelectionExpressionRequest(resource));
+ }
+
+ /**
+ * Get a Java object representation of an ES.Condition resource.
+ */
+ public static Condition getCondition(RequestProcessor graph, Resource condition) throws DatabaseException {
+ return graph.syncRequest(new SelectionConditionRequest(condition));
+ }
+ /**
+ * Get the name of the element selector.
+ */
public String getName() {
return name;
}
+ /**
+ * Get a textual SQL-like description of the element selector.
+ */
public String getExpression() {
return expression;
}
+ /**
+ * Get the resource that this selector represents.
+ */
public Resource getResource() {
return resource;
}
+ /**
+ * Get the generator component. Use {@link #buildSelection(ReadGraph)} to make it available first.
+ */
+ public Generator getGenerator() {
+ return generator;
+ }
+
+ /**
+ * Get the selector component. Use {@link #buildSelection(ReadGraph)} to make it available first.
+ */
+ public Selector getSelector() {
+ return selector;
+ }
+
+ /**
+ * Get the condition component. Use {@link #buildSelection(ReadGraph)} to make it available first.
+ */
+ public Condition getCondition() {
+ return condition;
+ }
+
+ /**
+ *
+ * @param graph
+ * @throws DatabaseException
+ */
+ public void buildSelection(ReadGraph graph) throws DatabaseException {
+ Resource selector = graph.getSingleObject(resource, ES.Selection_HasSelector);
+ Resource generator = graph.getSingleObject(resource, ES.Selection_HasGenerator);
+ Resource condition = graph.getPossibleObject(resource, ES.Selection_HasCondition);
+
+ this.selector = buildSelector(graph, selector);
+ this.generator = buildGenerator(graph, generator);
+ this.condition = buildCondition(graph, condition);
+ }
+
+ public static Map<Resource, String> findDiagrams() {
+ try {
+ return Simantics.getSession().syncRequest(new Read<Map<Resource, String>>() {
+ @Override
+ public Map<Resource, String> perform(ReadGraph graph) throws DatabaseException {
+ Map<Resource, String> result = new HashMap<>();
+ Resource model = graph.syncRequest(new PossibleActiveModel(Simantics.getProjectResource()));
+ List<Resource> composites = QueryIndexUtils.searchByType(graph, model, StructuralResource2.getInstance(graph).Composite);
+ for (Resource r : composites) {
+ // Get diagram
+ Resource diagram = graph.getPossibleObject(r, ModelingResources.getInstance(graph).CompositeToDiagram);
+ if (diagram == null) continue;
+
+ // Filter out user component diagrams
+ Resource parent = graph.getPossibleObject(r, Layer0.getInstance(graph).PartOf);
+ if (parent == null || graph.isInheritedFrom(parent, StructuralResource2.getInstance(graph).Component))
+ continue;
+
+ result.put(r, graph.getRelatedValue(r, Layer0.getInstance(graph).HasName));
+ }
+
+ return result;
+ }
+ });
+ } catch (DatabaseException e) {
+ LOGGER.error("Query for model diagrams failed", e);
+ return Collections.emptyMap();
+ }
+ }
+
+ public static Variable getVariableForElement(ReadGraph graph, Resource element) throws DatabaseException {
+ Resource component = graph.getPossibleObject(element, MOD.ElementToComponent);
+ if (component != null) {
+ Variable var = Variables.getVariable(graph, component);
+ RVI realRvi = var.getRVI(graph);
+
+ for (Resource activeModel : graph.syncRequest(new ActiveModels(Simantics.getProjectResource()))) {
+ for (Variable run : graph.syncRequest(new ActiveRuns(activeModel))) {
+ Variable v = realRvi.resolvePossible(graph, run);
+ if (v != null) {
+ return v;
+ }
+ }
+ Variable configuration = Variables.getPossibleConfigurationContext(graph, activeModel);
+ if (configuration != null) {
+ Variable v = realRvi.resolvePossible(graph, configuration);
+ if (v != null) {
+ return v;
+ }
+ }
+ }
+ }
+
+ return null;
+ }
+
+ public static Double getPropertyValue(ReadGraph graph, Resource element, String propertyName) {
+ try {
+ Variable v = getVariableForElement(graph, element);
+ if (v != null) {
+ Number value = v.getPossiblePropertyValue(graph, propertyName);
+ if (value != null)
+ return value.doubleValue();
+ }
+
+ // No property found - try possible mapped element property as well
+ Resource mappedElement = graph.getPossibleObject(element, DN.MappedComponent);
+ if (mappedElement != null)
+ return getPropertyValue(graph, mappedElement, propertyName);
+ }
+ catch (DatabaseException e) {
+ }
+
+ return null;
+ }
+
public List<Resource> selectElementsFrom(ReadGraph graph, Resource model) throws DatabaseException {
if (selector == null) {
buildSelection(graph);
String propertyName = graph.getRelatedValue(resource, ES.PropertyCondition_HasPropertyName);
Double lowerLimit = graph.getPossibleRelatedValue(resource, ES.PropertyCondition_HasLowerLimit);
Double upperLimit = graph.getPossibleRelatedValue(resource, ES.PropertyCondition_HasUpperLimit);
- cond = new PropertyCondition(propertyName, lowerLimit, upperLimit);
+ cond = new PropertyCondition(resource, propertyName, lowerLimit, upperLimit);
}
else if (graph.isInstanceOf(resource, ES.RegionCondition)) {
DiagramRegionsResource DR = DiagramRegionsResource.getInstance(graph);
- double[] region = graph.getRelatedValue(graph.getSingleObject(resource, ES.RegionCondition_HasRegion), DR.Region_area);
- cond = new RegionCondition(region);
+ Resource regionResource = graph.getSingleObject(resource, ES.RegionCondition_HasRegion);
+ double[] region = graph.getRelatedValue(regionResource, DR.Region_area);
+ cond = new RegionCondition(resource, regionResource, region);
}
else if (graph.isInstanceOf(resource, ES.RouteCondition)) {
- Set<Resource> routePoints = new HashSet<>(ListUtils.toList(graph, graph.getSingleObject(resource, ES.RouteCondition_HasRoute)));
- cond = new RouteCondition(routePoints);
+ Resource routeResource = graph.getSingleObject(resource, ES.RouteCondition_HasRoute);
+ Set<Resource> routePoints = new HashSet<>(ListUtils.toList(graph, routeResource));
+ cond = new RouteCondition(resource, routeResource, routePoints);
}
else if (graph.isInstanceOf(resource, ES.AggregateCondition)) {
- boolean isDisjunction = graph.isInstanceOf(resource, ES.Disjunction);
Collection<Resource> conditionResources = graph.getObjects(resource, ES.HasSubcondition);
List<Condition> conditions = new ArrayList<>(conditionResources.size());
for (Resource c : conditionResources) {
else
throw new IllegalArgumentException("Unknown aggreate condition type " + graph.getURI(graph.getSingleType(resource)));
- cond = new AggregateCondition(type, conditions);
+ cond = new AggregateCondition(resource, type, conditions);
}
else {
throw new IllegalArgumentException("Unknown condition type " + graph.getURI(graph.getSingleType(resource)));
}
- cond.isInverse = graph.hasStatement(resource, ES.Condition_IsInverse, resource);
return cond;
}
private static String buildExpression(ReadGraph graph, Resource r) throws DatabaseException {
if (graph.isInstanceOf(r, ES.Selection)) {
- String exp = "select " + buildExpression(graph, graph.getSingleObject(r, ES.Selection_HasSelector)) +
- " from " + buildExpression(graph, graph.getSingleObject(r, ES.Selection_HasGenerator));
+ String exp = "select " + getExpression(graph, graph.getSingleObject(r, ES.Selection_HasSelector)) +
+ " from " + getExpression(graph, graph.getSingleObject(r, ES.Selection_HasGenerator));
Resource cond = graph.getPossibleObject(r, ES.Selection_HasCondition);
- return cond != null ? exp + " where {" + buildExpression(graph, cond) + "}" : exp;
+ return cond != null ? exp + " where {" + getExpression(graph, cond) + "}" : exp;
}
else if (graph.isInstanceOf(r, ES.Condition)) {
if (graph.isInstanceOf(r, ES.PropertyCondition)) {
List<String> exps = new ArrayList<>();
Collection<Resource> objects = graph.getObjects(r, ES.HasSubcondition);
for (Resource c : objects) {
- String exp = buildExpression(graph, c);
+ String exp = getExpression(graph, c);
exps.add(objects.size() > 1 ? "{" + exp + "}" : exp);
}
String result = String.join(op, exps);
}
}
- private static Collection<Resource> elementsOfDiagram(ReadGraph graph, Resource diagram) throws ServiceException {
+ private static Collection<Resource> elementsOfDiagram(ReadGraph graph, Resource diagram) throws DatabaseException {
+ if (graph.isInstanceOf(diagram, STR.Composite)) {
+ // Resource is a composite - get diagram
+ return elementsOfDiagram(graph, graph.getSingleObject(diagram, MOD.CompositeToDiagram));
+ }
+
Collection<Resource> elements = graph.getObjects(diagram, L0.ConsistsOf);
Collection<Resource> result = new ArrayList<>();
for (Resource r : elements)
return result;
}
- private void buildSelection(ReadGraph graph) throws DatabaseException {
- Resource selector = graph.getSingleObject(resource, ES.Selection_HasSelector);
- Resource generator = graph.getSingleObject(resource, ES.Selection_HasGenerator);
- Resource condition = graph.getPossibleObject(resource, ES.Selection_HasCondition);
-
- this.selector = buildSelector(graph, selector);
- this.generator = buildGenerator(graph, generator);
- this.condition = buildCondition(graph, condition);
- }
-
private Collection<Resource> filterElementsFrom(ReadGraph graph, Collection<Resource> elements) throws DatabaseException {
if (condition == null) return elements;
return selector.select(graph, filterElementsFrom(graph, generator.generate(graph, model)));
}
- static Variable getVariableForElement(ReadGraph graph, Resource element) throws DatabaseException {
- Resource component = graph.getPossibleObject(element, MOD.ElementToComponent);
- if (component != null) {
- Variable var = Variables.getVariable(graph, component);
- RVI realRvi = var.getRVI(graph);
-
- for (Resource activeModel : graph.syncRequest(new ActiveModels(Simantics.getProjectResource()))) {
- for (Variable run : graph.syncRequest(new ActiveRuns(activeModel))) {
- Variable v = realRvi.resolvePossible(graph, run);
- if (v != null) {
- return v;
- }
- }
- Variable configuration = Variables.getPossibleConfigurationContext(graph, activeModel);
- if (configuration != null) {
- Variable v = realRvi.resolvePossible(graph, configuration);
- if (v != null) {
- return v;
- }
- }
- }
- }
-
- return null;
- }
-
- static Double getPropertyValue(ReadGraph graph, Resource element, String propertyName) {
- try {
- Variable v = getVariableForElement(graph, element);
- if (v != null) {
- Number value = v.getPossiblePropertyValue(graph, propertyName);
- if (value != null)
- return value.doubleValue();
- }
- }
- catch (DatabaseException e) {
- }
-
- return null;
- }
-
// Generators
- static abstract class Generator {
+ public static abstract class Generator {
abstract Collection<Resource> generate(ReadGraph graph, Resource model) throws DatabaseException;
}
- static class ModelGenerator extends Generator {
+ public static class ModelGenerator extends Generator {
@Override
Collection<Resource> generate(ReadGraph graph, Resource model) throws DatabaseException {
Resource conf = graph.syncRequest(new Configuration(model));
}
}
- static class DiagramGenerator extends Generator {
- Resource diagram;
+ public static class DiagramGenerator extends Generator {
+ public Resource diagram;
public DiagramGenerator(Resource diagram) {
this.diagram = diagram;
}
}
- static class ExplicitGenerator extends Generator {
+ public static class ExplicitGenerator extends Generator {
public ExplicitGenerator(Collection<Resource> elements) {
this.elements = elements;
}
- Collection<Resource> elements;
+ public Collection<Resource> elements;
@Override
Collection<Resource> generate(ReadGraph graph, Resource model) {
// Selectors
- static abstract class Selector {
+ public static abstract class Selector {
abstract Collection<Resource> select(ReadGraph graph, Collection<Resource> elements);
}
- static class All extends Selector {
+ public static class All extends Selector {
public All() {
}
}
}
- static class PropertySelector extends Selector {
+ public static class PropertySelector extends Selector {
public PropertySelector(boolean smallest, String propertyName, int resultCount) {
this.smallest = smallest;
this.propertyName = propertyName;
this.resultCount = resultCount;
}
- boolean smallest;
- String propertyName;
- int resultCount;
+ public boolean smallest;
+ public String propertyName;
+ public int resultCount;
@SuppressWarnings("unchecked")
@Override
Collection<Resource> select(ReadGraph graph, Collection<Resource> elements) {
- List<Resource> result = new ArrayList<>(elements);
List<Tuple2> result2 = Lists.map(new FunctionImpl1<Resource, Tuple2>() {
@Override
public Tuple2 apply(Resource r) {
return new Tuple2(r, getPropertyValue(graph, r, propertyName));
}
- }, result);
+ }, new ArrayList<>(elements));
result2 = Lists.filter(new FunctionImpl1<Tuple2, Boolean>() {
@Override
result2.sort((t1, t2) -> smallest ? Double.compare((Double) t1.c1, (Double) t2.c1) : Double.compare((Double) t2.c1, (Double) t1.c1));
- result2 = result2.subList(0, resultCount);
+ if (resultCount < result2.size()) {
+ double limitValue = (double) result2.get(resultCount-1).c1;
+
+ // Expand selection to contain all items with the same value as the nth one
+ int count = resultCount;
+ while (count < result2.size() && (double)result2.get(count).c1 == limitValue) count++;
+ result2 = result2.subList(0, count);
+ }
- result = Lists.map(new FunctionImpl1<Tuple2, Resource>() {
+ return (List<Resource>) Lists.map(new FunctionImpl1<Tuple2, Resource>() {
@Override
public Resource apply(Tuple2 p0) {
return (Resource) p0.c0;
}
}, result2);
-
- return result;
}
}
// Conditions
- static abstract class Condition {
- boolean isInverse;
+ public static abstract class Condition {
+ public Resource resource;
+
+ Condition(Resource r) {
+ resource = r;
+ }
+
abstract boolean match(ReadGraph graph, Resource r) throws DatabaseException;
}
- static class PropertyCondition extends Condition {
- public PropertyCondition(String propertyName, Double lowerLimit, Double upperLimit) {
- super();
+ public static class PropertyCondition extends Condition {
+ public PropertyCondition(Resource r, String propertyName, Double lowerLimit, Double upperLimit) {
+ super(r);
+
this.propertyName = propertyName;
this.lowerLimit = lowerLimit;
this.upperLimit = upperLimit;
}
- String propertyName;
- Double lowerLimit;
- Double upperLimit;
+ public String propertyName;
+ public Double lowerLimit;
+ public Double upperLimit;
@Override
boolean match(ReadGraph graph, Resource r) {
}
}
- static class RegionCondition extends Condition {
- public RegionCondition(double[] region) {
- super();
+ public static class RegionCondition extends Condition {
+ public RegionCondition(Resource r, Resource regionResoruce, double[] region) {
+ super(r);
this.region = region;
Path2D path = new Path2D.Double();
path.closePath();
this.path = path;
+ this.regionResource = regionResoruce;
}
+ public Resource regionResource;
double[] region;
Path2D path;
}
}
- static class RouteCondition extends Condition {
- public RouteCondition(Set<Resource> routePoints) {
- super();
+ public static class RouteCondition extends Condition {
+ public RouteCondition(Resource r, Resource routeResource, Set<Resource> routePoints) {
+ super(r);
this.routePoints = routePoints;
+ this.routeResource = routeResource;
}
+ public Resource routeResource;
Set<Resource> routePoints;
@Override
}
}
- static class DiagramCondition extends Condition {
- public DiagramCondition(Resource diagram) {
- super();
- this.diagram = diagram;
- }
-
- Resource diagram;
-
- @Override
- boolean match(ReadGraph graph, Resource r) throws DatabaseException {
- return graph.getSingleObject(r, L0.PartOf).equals(diagram);
- }
- }
-
- static class AggregateCondition extends Condition {
- static enum Type { DISJUNCTION, CONJUNCTION, NEGATION };
+ public static class AggregateCondition extends Condition {
+ public static enum Type { DISJUNCTION, CONJUNCTION, NEGATION };
- public AggregateCondition(Type type, List<Condition> conditions) {
- super();
+ public AggregateCondition(Resource r, Type type, List<Condition> conditions) {
+ super(r);
this.type = type;
this.conditions = conditions;
}
- Type type;
- List<Condition> conditions;
+ public Type type;
+ public List<Condition> conditions;
@Override
boolean match(ReadGraph graph, Resource r) throws DatabaseException {
return new ElementSelector(graph, resource);
}
}
+
+ public final static class SelectionExpressionRequest extends ResourceRead<String> {
+ public SelectionExpressionRequest(Resource condition) {
+ super(condition);
+ }
+
+ @Override
+ public String perform(ReadGraph graph) throws DatabaseException {
+ return ElementSelector.buildExpression(graph, resource);
+ }
+ }
+
+ public static final class SelectionConditionRequest extends ResourceRead<Condition> {
+ public SelectionConditionRequest(Resource resource) {
+ super(resource);
+ }
+
+ @Override
+ public Condition perform(ReadGraph graph) throws DatabaseException {
+ return buildCondition(graph, resource);
+ }
+ }
}