bundles/org.simantics.modeling/src/org/simantics/modeling/svg/CreateSVGElement.java

   1 package org.simantics.modeling.svg;
   2
   3 import java.io.File;
   4 import java.io.IOException;
   5
   6 import org.simantics.databoard.Bindings;
   7 import org.simantics.db.Resource;
   8 import org.simantics.db.WriteGraph;
   9 import org.simantics.db.common.utils.OrderedSetUtils;
  10 import org.simantics.db.exception.DatabaseException;
  11 import org.simantics.diagram.stubs.DiagramResource;
  12 import org.simantics.diagram.stubs.G2DResource;
  13 import org.simantics.diagram.synchronization.graph.AddElement;
  14 import org.simantics.image2.ontology.ImageResource;
  15 import org.simantics.layer0.Layer0;
  16 import org.simantics.layer0.utils.direct.GraphUtils;
  17 import org.simantics.utils.FileUtils;
  18
  19 public class CreateSVGElement {
  20
  21     public static void createSVGElement(WriteGraph g, Resource diagram, String suffix, byte[] data, double mposX, double mposY) throws DatabaseException {
  22         createSVGElement(g, diagram, suffix, data, mposX, mposY);
  23     }
  24
  25     public static Resource createSVGElementR(WriteGraph g, Resource diagram, String suffix, byte[] data, double mposX, double mposY) throws DatabaseException {
  26
  27         Layer0 L0 = Layer0.getInstance(g);
  28         DiagramResource DIA = DiagramResource.getInstance(g);
  29         ImageResource IMAGE = ImageResource.getInstance(g);
  30         G2DResource G2D = G2DResource.getInstance(g);
  31
  32         Resource element;
  33         if("svg".equals(suffix)) {
  34             Resource transform = GraphUtils.create(g, L0.InstanceOf, G2D.Transform);
  35             g.claimValue(transform, new double[] {1.0, 0.0, 0.0, 1.0, mposX, mposY}, Bindings.DOUBLE_ARRAY);
  36             element = GraphUtils.create(g,
  37                     L0.InstanceOf, DIA.SVGElement,
  38                     G2D.HasSVGDocument, new String(data),
  39                     DIA.HasTransform, transform);
  40             AddElement.claimFreshElementName(g, diagram, element);
  41         } else if ("png".equals(suffix)) {
  42             Resource transform = GraphUtils.create(g, L0.InstanceOf, G2D.Transform);
  43             g.claimValue(transform, new double[] {1.0, 0.0, 0.0, 1.0, mposX, mposY}, Bindings.DOUBLE_ARRAY);
  44             Resource image = g.newResource();
  45             g.claim(image, L0.InstanceOf, IMAGE.PngImage);
  46             g.claimValue(image, data, Bindings.BYTE_ARRAY);
  47             element = GraphUtils.create(g,
  48                     L0.InstanceOf, DIA.RasterElement,
  49                     G2D.HasRasterImage, image,
  50                     DIA.HasTransform, transform);
  51             AddElement.claimFreshElementName(g, diagram, element);
  52         }
  53         else
  54             throw new DatabaseException("Unknown image format " + suffix);
  55         OrderedSetUtils.addFirst(g, diagram, element);
  56         g.claim(diagram, L0.ConsistsOf, element);
  57         return element;
  58     }
  59
  60     public static void importSVGElement(WriteGraph graph, Resource diagram, File file, double posX, double posY) throws DatabaseException, IOException {
  61         importSVGElementR(graph, diagram, file, posX, posY);
  62     }
  63
  64     public static Resource importSVGElementR(WriteGraph graph, Resource diagram, File file, double posX, double posY) throws DatabaseException, IOException {
  65         final byte[] data = FileUtils.readFile(file);
  66         return createSVGElementR(graph, diagram, suffix(file.getName()), data, posX, posY);
  67     }
  68
  69     private static String suffix(String fileName) {
  70         int len = fileName.length();
  71         if(len < 3) return null;
  72         else return fileName.substring(len-3,len).toLowerCase();
  73     }
  74
  75 }