bundles/org.simantics.diagram/src/org/simantics/diagram/export/DiagramContentTypeAction.java

   1 package org.simantics.diagram.export;\r
   2 \r
   3 import java.util.Collection;\r
   4 import java.util.Deque;\r
   5 import java.util.HashMap;\r
   6 import java.util.LinkedList;\r
   7 import java.util.Map;\r
   8 \r
   9 import org.simantics.NameLabelMode;\r
  10 import org.simantics.NameLabelUtil;\r
  11 import org.simantics.db.ReadGraph;\r
  12 import org.simantics.db.Resource;\r
  13 import org.simantics.db.common.request.UniqueRead;\r
  14 import org.simantics.db.exception.AssumptionException;\r
  15 import org.simantics.db.exception.DatabaseException;\r
  16 import org.simantics.db.exception.ManyObjectsForFunctionalRelationException;\r
  17 import org.simantics.db.exception.ServiceException;\r
  18 import org.simantics.db.exception.ValidationException;\r
  19 import org.simantics.db.layer0.request.PossibleModel;\r
  20 import org.simantics.db.request.Read;\r
  21 import org.simantics.export.core.ExportContext;\r
  22 import org.simantics.export.core.error.ExportException;\r
  23 import org.simantics.export.core.intf.ContentTypeAction;\r
  24 import org.simantics.layer0.Layer0;\r
  25 import org.simantics.simulation.ontology.SimulationResource;\r
  26 \r
  27 public class DiagramContentTypeAction implements ContentTypeAction {\r
  28         \r
  29         public static String getDiagramLabel(ReadGraph graph, Resource r) throws DatabaseException\r
  30         {\r
  31                 SimulationResource SIM = SimulationResource.getInstance(graph);\r
  32 \r
  33                 NameLabelMode mode = NameLabelUtil.getNameLabelMode(graph);\r
  34 \r
  35                 labelFailed:\r
  36                 try { \r
  37                         Resource model = graph.sync( new PossibleModel(r) );\r
  38                         if ( model == null ) break labelFailed;\r
  39                         Resource configuration = graph.getPossibleObject(model, SIM.HasConfiguration);\r
  40                         if ( configuration == null ) break labelFailed;\r
  41                         Deque<Resource> path = getPathTo(graph, r, model);\r
  42                         if ( path == null) break labelFailed;\r
  43                         path.remove(configuration);\r
  44 \r
  45                         StringBuilder sb = new StringBuilder();\r
  46                         for ( Resource node : path ) {\r
  47                                 if ( sb.length()>0 ) sb.append(" / ");\r
  48                                 String nodeLabel = NameLabelUtil.modalName(graph, node, mode);\r
  49                                 sb.append( nodeLabel );\r
  50                         }\r
  51                                 \r
  52                         return sb.toString();\r
  53                 } catch (AssumptionException e) {\r
  54                 } catch (ValidationException e) {\r
  55                 } catch (ServiceException e) {\r
  56                 }\r
  57                 \r
  58                 String uri = graph.getURI( r );\r
  59                 int c = uri.lastIndexOf('/');\r
  60                 if ( c>=0 ) {\r
  61                         String label = uri.substring(c+1);\r
  62                         if ( !label.isEmpty() ) return label;\r
  63                 }\r
  64                 \r
  65                 return uri;\r
  66         }\r
  67 \r
  68         @Override\r
  69         public Map<String, String> getLabels(ExportContext ctx, final Collection<String> uris) throws ExportException {\r
  70                 Read<Map<String, String>> req = new UniqueRead<Map<String, String>>() {\r
  71                         @Override\r
  72                         public Map<String, String> perform(ReadGraph graph) throws DatabaseException {\r
  73                                 Map<String, String> result = new HashMap<String, String>();\r
  74                                 \r
  75                                 for ( String uri : uris ) {\r
  76                                 Resource r = graph.getResource( uri );\r
  77                                         String label = getDiagramLabel(graph, r);\r
  78                                 result.put(uri, label);\r
  79                                 }\r
  80                                 \r
  81                                 return result;\r
  82                         }\r
  83                 };\r
  84                 \r
  85                 try {\r
  86                         return ctx.session.syncRequest( req );\r
  87                 } catch (DatabaseException e) {\r
  88                         throw new ExportException( e );\r
  89                 }\r
  90         }\r
  91         \r
  92         /**\r
  93          * Get all resource between start and end by following PartOf relation.\r
  94          * If there is no path, returns null.\r
  95          * Start is included, end is excluded.\r
  96          * \r
  97          * @param graph\r
  98          * @param start\r
  99          * @param end\r
 100          * @return path from end to start\r
 101          * @throws ServiceException \r
 102          * @throws ManyObjectsForFunctionalRelationException \r
 103          */\r
 104         static Deque<Resource> getPathTo(ReadGraph graph, Resource start, Resource end) throws DatabaseException {\r
 105                 LinkedList<Resource> result = new LinkedList<Resource>();\r
 106                 \r
 107                 Layer0 L0 = Layer0.getInstance(graph);\r
 108                 Resource pos = start;\r
 109                 while ( !pos.equals(end) ) {\r
 110                         result.add(0, pos);\r
 111                         pos = graph.getPossibleObject(pos, L0.PartOf);\r
 112                         if ( pos==null ) return null;\r
 113                 }\r
 114                 return result;\r
 115         }\r
 116 \r
 117 }\r