bundles/org.simantics.diagram.svg/src/org/simantics/diagram/svg/export/DiagramToSVG.java

   1 package org.simantics.diagram.svg.export;\r
   2 \r
   3 \r
   4 import java.awt.Point;\r
   5 import java.util.concurrent.Semaphore;\r
   6 import java.util.concurrent.atomic.AtomicReference;\r
   7 \r
   8 import org.apache.batik.svggen.SVGGraphics2D;\r
   9 import org.simantics.Simantics;\r
  10 import org.simantics.db.ReadGraph;\r
  11 import org.simantics.db.Resource;\r
  12 import org.simantics.db.common.ResourceArray;\r
  13 import org.simantics.db.common.request.PossibleIndexRoot;\r
  14 import org.simantics.db.common.request.UniqueRead;\r
  15 import org.simantics.db.common.utils.NameUtils;\r
  16 import org.simantics.db.exception.DatabaseException;\r
  17 import org.simantics.db.exception.ValidationException;\r
  18 import org.simantics.db.management.ISessionContext;\r
  19 import org.simantics.db.request.Read;\r
  20 import org.simantics.diagram.elements.DiagramNodeUtil;\r
  21 import org.simantics.diagram.export.ImagePrinter;\r
  22 import org.simantics.diagram.stubs.DiagramResource;\r
  23 import org.simantics.g2d.canvas.impl.CanvasContext;\r
  24 import org.simantics.g2d.scenegraph.ICanvasSceneGraphProvider;\r
  25 import org.simantics.layer0.Layer0;\r
  26 import org.simantics.modeling.ModelingResources;\r
  27 import org.simantics.structural.stubs.StructuralResource2;\r
  28 import org.simantics.structural2.StructuralVariables;\r
  29 import org.simantics.utils.DataContainer;\r
  30 import org.simantics.utils.datastructures.Pair;\r
  31 import org.simantics.utils.page.MarginUtils.Margins;\r
  32 import org.simantics.utils.threads.ThreadUtils;\r
  33 import org.simantics.utils.threads.WorkerThread;\r
  34 \r
  35 \r
  36 public class DiagramToSVG {\r
  37         \r
  38         \r
  39         public static String diagramToSVG(Resource diagram) {\r
  40 \r
  41                 try {\r
  42                         ImagePrinter.ImageExportPlan exportPlan = new ImagePrinter.ImageExportPlan();\r
  43                         exportPlan.margin = 0.05;\r
  44                         exportPlan.dpi = 96.0;\r
  45                         exportPlan.diagram = diagram;\r
  46                         return render(diagram,exportPlan);\r
  47                 } catch (Exception e) {\r
  48                         e.printStackTrace();\r
  49                         return "<b>error</b> "+ e.getMessage();\r
  50                 }\r
  51  \r
  52                 \r
  53         }\r
  54         \r
  55         public static String diagramToSVG(Resource diagram, int width, int height) {\r
  56 \r
  57                 try {\r
  58                         ImagePrinter.ImageExportPlan exportPlan = new ImagePrinter.ImageExportPlan();\r
  59                         exportPlan.margin = 0.05;\r
  60                         exportPlan.size = new Point(width, height);\r
  61                         exportPlan.diagram = diagram;\r
  62                         return render(diagram,exportPlan);\r
  63                 } catch (Exception e) {\r
  64                         e.printStackTrace();\r
  65                         return "<b>error</b> "+ e.getMessage();\r
  66                 }\r
  67  \r
  68                 \r
  69         }\r
  70         \r
  71         public static String diagramToSVG(Resource diagram, double dpi) {\r
  72 \r
  73                 try {\r
  74                         ImagePrinter.ImageExportPlan exportPlan = new ImagePrinter.ImageExportPlan();\r
  75                         exportPlan.margin = 0.05;\r
  76                         exportPlan.dpi = dpi;\r
  77                         exportPlan.diagram = diagram;\r
  78                         return render(diagram,exportPlan);\r
  79                 } catch (Exception e) {\r
  80                         e.printStackTrace();\r
  81                         return "<b>error</b> "+ e.getMessage();\r
  82                 }\r
  83  \r
  84                 \r
  85         }\r
  86         \r
  87         public static String diagramToSVG(Resource diagram, ImagePrinter.ImageExportPlan exportPlan, Margins margins, SVGGraphics2D svgGenerator) {\r
  88 \r
  89                 try {\r
  90                         return render(diagram,exportPlan,margins,svgGenerator);\r
  91                 } catch (Exception e) {\r
  92                         e.printStackTrace();\r
  93                         return "<b>error</b> "+ e.getMessage();\r
  94                 }\r
  95                 \r
  96         }\r
  97         \r
  98         private static String render(final Resource input, final ImagePrinter.ImageExportPlan exportPlan) throws Exception {\r
  99                 return render(input, exportPlan, null, SVGBuilder.defaultSVGGenerator()); \r
 100         }\r
 101         \r
 102         private static String render(final Resource input, final ImagePrinter.ImageExportPlan exportPlan, Margins margins, SVGGraphics2D svgExporter) throws Exception {\r
 103                 Resource diagram = Simantics.getSession().syncRequest(new Read<Resource>() {\r
 104                         @Override\r
 105                         public Resource perform(ReadGraph graph) throws DatabaseException {\r
 106                                 DiagramResource DIA = DiagramResource.getInstance(graph);\r
 107                                 if (graph.isInstanceOf(input, DIA.Diagram))\r
 108                                         return input;\r
 109                                 StructuralResource2 SR = StructuralResource2.getInstance(graph);\r
 110                                 ModelingResources MOD = ModelingResources.getInstance(graph);\r
 111                                 Layer0 L0 = Layer0.getInstance(graph);\r
 112                                 if (graph.isInstanceOf(input, SR.Composite)) {\r
 113                                         Resource possibleDiagram = graph.getPossibleObject(input, MOD.CompositeToDiagram);\r
 114                                         if (possibleDiagram != null)\r
 115                                                 return possibleDiagram;\r
 116                                         for (Resource r : graph.getObjects(input, L0.ConsistsOf)) {\r
 117                                                 if (graph.isInstanceOf(r, SR.Composite)) {\r
 118                                                         possibleDiagram = graph.getPossibleObject(input, MOD.CompositeToDiagram);\r
 119                                                         if (possibleDiagram != null)\r
 120                                                                 return possibleDiagram;\r
 121                                                 }\r
 122                                         }\r
 123                                 }\r
 124                                 return null;\r
 125                         }\r
 126                 });\r
 127                 if (diagram == null)\r
 128                         throw new DatabaseException("Input " + input + " cannot be resolved as diagram");\r
 129                 \r
 130                 \r
 131         \r
 132                 \r
 133                 final WorkerThread thread = new WorkerThread("Diagram Image Painter");\r
 134                 thread.start();\r
 135                 \r
 136         final CanvasContext ctx = new CanvasContext(thread);\r
 137         final AtomicReference<ICanvasSceneGraphProvider> sgProvider = new AtomicReference<ICanvasSceneGraphProvider>();\r
 138                 final ISessionContext sessionContext = Simantics.getSessionContext();\r
 139                 final DataContainer<String> result = new DataContainer<String>(null);\r
 140                 final DataContainer<Exception> exception = new DataContainer<Exception>(null);\r
 141         try {\r
 142             final Semaphore done = new Semaphore(0);\r
 143             // IMPORTANT: Load diagram in a different thread than the canvas context thread!\r
 144             ThreadUtils.getBlockingWorkExecutor().execute(new Runnable() {\r
 145                 @Override\r
 146                 public void run() {\r
 147                     try {\r
 148                         Pair<Resource, String> modelAndRVI = sessionContext.getSession().syncRequest(new UniqueRead<Pair<Resource, String>>() {\r
 149                             @Override\r
 150                             public Pair<Resource, String> perform(ReadGraph graph) throws DatabaseException {\r
 151                                 return new Pair<Resource, String>( resolveModel(graph, exportPlan.diagram ), resolveRVI(graph, exportPlan.diagram) );\r
 152                             }\r
 153                         });\r
 154 \r
 155                         ICanvasSceneGraphProvider provider = DiagramNodeUtil.loadSceneGraphProvider(ctx, modelAndRVI.first, exportPlan.diagram, modelAndRVI.second);\r
 156                         sgProvider.set( provider );\r
 157                         \r
 158                         ThreadUtils.asyncExec(thread, new Runnable() {\r
 159                             @Override\r
 160                             public void run() {\r
 161                                 try {\r
 162                                     SVGBuilder chassis = margins != null ?\r
 163                                                 new SVGBuilder(exportPlan.dpi,exportPlan.size,margins) :\r
 164                                                 new SVGBuilder(exportPlan.dpi,exportPlan.size,exportPlan.margin);\r
 165                                     \r
 166                                     result.set(chassis.paint(ctx, svgExporter));\r
 167                                 } catch (Exception e) {\r
 168                                         exception.set(e);\r
 169                                 } finally {\r
 170                                     done.release();\r
 171                                 }\r
 172                             }\r
 173                         });\r
 174                     } catch (DatabaseException e) {\r
 175                         exception.set(e);\r
 176                         done.release();\r
 177                     } catch (Throwable e) {\r
 178                         exception.set(new DatabaseException(e));\r
 179                         done.release();\r
 180                     } finally {\r
 181                         done.release();\r
 182                     }\r
 183                 }\r
 184             });\r
 185 \r
 186             done.acquire(2);\r
 187             if (exception.get() != null)\r
 188                 throw exception.get();\r
 189             return result.get();\r
 190         } finally {\r
 191             if (sgProvider.get() != null)\r
 192                 sgProvider.get().dispose();\r
 193             ctx.dispose();\r
 194         }\r
 195         }\r
 196         \r
 197         private static Resource resolveModel(ReadGraph graph, Resource diagram) throws DatabaseException {\r
 198         ModelingResources mod = ModelingResources.getInstance(graph);\r
 199         Resource composite = graph.getSingleObject(diagram, mod.DiagramToComposite);\r
 200         Resource model = graph.syncRequest(new PossibleIndexRoot(composite));\r
 201         if (model == null)\r
 202             throw new ValidationException("no model found for composite " + NameUtils.getSafeName(graph, composite));\r
 203         return model;\r
 204     }\r
 205 \r
 206 \r
 207 \r
 208     private static String resolveRVI(ReadGraph graph, Resource diagram) throws DatabaseException {\r
 209         ModelingResources mod = ModelingResources.getInstance(graph);\r
 210         Resource composite = graph.getSingleObject(diagram, mod.DiagramToComposite);\r
 211         final ResourceArray compositePath = StructuralVariables.getCompositeArray(graph, composite);\r
 212         final ResourceArray variablePath = compositePath.removeFromBeginning(1);\r
 213         return StructuralVariables.getRVI(graph, variablePath);\r
 214     }\r
 215 \r
 216 }\r