1 package org.simantics.graph.refactoring;
3 import java.io.BufferedInputStream;
4 import java.io.DataInputStream;
5 import java.io.InputStream;
6 import java.nio.file.Files;
7 import java.nio.file.Path;
8 import java.nio.file.Paths;
9 import java.nio.file.StandardOpenOption;
11 import org.simantics.databoard.binding.Binding;
12 import org.simantics.databoard.container.DataContainer;
13 import org.simantics.databoard.container.DataContainers;
14 import org.simantics.graph.representation.PrettyPrintTG;
15 import org.simantics.graph.representation.TransferableGraph1;
16 import org.slf4j.Logger;
17 import org.slf4j.LoggerFactory;
20 * @author Antti Villberg
23 public class FixExportedOntology {
25 private static final Logger LOGGER = LoggerFactory.getLogger(FixExportedOntology.class);
27 static TransferableGraph1 convertExportedSharedOntologyIntoBundleOntology(Path input, Path output) throws Exception {
28 LOGGER.info("Converting exported shared ontology\n\t{}\nto bundle-compatible ontology\n\t{}", input.toString(), output.toString());
29 try (InputStream is = new BufferedInputStream(Files.newInputStream(input), 128*1024)) {
30 Binding tgBinding = TransferableGraph1.BINDING;
31 DataContainer container = DataContainers.readFile(new DataInputStream(is), tgBinding);
32 TransferableGraph1 graph = (TransferableGraph1) container.content.getValue(tgBinding);
34 GraphRefactoringUtils.fixOntologyExport(graph);
35 container = TransferableGraphHasher.addHashToTG(container, graph);
37 DataContainers.writeFile(output.toFile(), container);
42 private static Path replaceExtension(Path p, String newExtension) {
43 String newName = p.getFileName().toString();
44 int lastDot = newName.lastIndexOf('.');
46 newName = newName.substring(0, lastDot);
47 return p.resolveSibling(newName + newExtension);
50 private static void createTg(Path input, Path output) throws Exception {
51 convertExportedSharedOntologyIntoBundleOntology(input, output);
54 public static void createTg(Path input) throws Exception {
55 createTg(input, replaceExtension(input, ".tg"));
58 public static void createTGAndPGraph(Path input) throws Exception {
59 createTGAndPGraph(input, true);
62 public static void createTGAndPGraph(Path input, boolean writePGraph) throws Exception {
63 TransferableGraph1 tg = convertExportedSharedOntologyIntoBundleOntology(input, replaceExtension(input, ".tg"));
65 String listing = PrettyPrintTG.print(tg, false);
66 Files.write(replaceExtension(input, ".pgraph"), listing.getBytes(),StandardOpenOption.CREATE);
70 public static void main(String[] args) throws Exception {
71 if (args.length == 0) {
72 System.out.println("Required arguments: <input .sharedLibrary file> [<output .tg file>]");
73 } else if (args.length == 1) {
74 Path input = Paths.get(args[0]);
75 createTGAndPGraph(input);
77 convertExportedSharedOntologyIntoBundleOntology(Paths.get(args[0]), Paths.get(args[1]));