Combination of Simantics-platform related changes and fixes for district
[simantics/platform.git] / 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 }