Better support for ontological profiles

[simantics/platform.git] / bundles / org.simantics.diagram.profile / src / org / simantics / diagram / profile / view / ResourcePairImageRule.java

diff --git a/bundles/org.simantics.diagram.profile/src/org/simantics/diagram/profile/view/ResourcePairImageRule.java b/bundles/org.simantics.diagram.profile/src/org/simantics/diagram/profile/view/ResourcePairImageRule.java
index 6f8ebe0a618210ead71616f49587fd34a5b4d710..daf038a3972940a53b06637844ef0aa67b83fcdf 100644 (file)
--- a/bundles/org.simantics.diagram.profile/src/org/simantics/diagram/profile/view/ResourcePairImageRule.java
+++ b/bundles/org.simantics.diagram.profile/src/org/simantics/diagram/profile/view/ResourcePairImageRule.java
@@ -25,13 +25,13 @@ public class ResourcePairImageRule implements ImageRule {
 
     @Override
     public boolean isCompatible(Class<?> contentType) {
-        return contentType.equals(ResourcePair.class);
+        return contentType.equals(ProfileTuple.class);
     }
 
     @Override
     public Map<String, ImageDescriptor> getImage(ReadGraph graph, Object content) throws DatabaseException {
         
-        ImageDescriptorProvider provider = graph.getPossibleAdapter(((ResourcePair)content).getSecond(), ImageDescriptorProvider.class);
+        ImageDescriptorProvider provider = graph.getPossibleAdapter(((ProfileTuple)content).getEntry(), ImageDescriptorProvider.class);
         return provider != null ? Collections.singletonMap(ColumnKeys.SINGLE, provider.get()) : null;
         
     }