bundles/org.simantics.graph/src/org/simantics/graph/refactoring/FixExportedOntology.java

   1 package org.simantics.graph.refactoring;
   2
   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;
  10
  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
  17 /**
  18  * @author Antti Villberg
  19  * @since 1.24.0
  20  */
  21 public class FixExportedOntology {
  22
  23         static TransferableGraph1 convertExportedSharedOntologyIntoBundleOntology(Path input, Path output) throws Exception {
  24                 System.out.format("Converting exported shared ontology%n\t" + input.toString() + "%nto bundle-compatible ontology%n\t" + output.toString());
  25                 try (InputStream is = new BufferedInputStream(Files.newInputStream(input), 128*1024)) {
  26                         Binding tgBinding = TransferableGraph1.BINDING;
  27                         DataContainer container = DataContainers.readFile(new DataInputStream(is), tgBinding);
  28                         TransferableGraph1 graph = (TransferableGraph1) container.content.getValue(tgBinding);
  29
  30                         GraphRefactoringUtils.fixOntologyExport(graph);
  31                         container = TransferableGraphHasher.addHashToTG(container, graph);
  32
  33                         DataContainers.writeFile(output.toFile(), container);
  34                         return graph;
  35                 }
  36         }
  37
  38         public static void createTg(Path input) throws Exception {
  39             Path output = input.getParent().resolve("graph.tg");
  40             createTg(input, output);
  41         }
  42
  43         private static void createTg(Path input, Path output) throws Exception {
  44             convertExportedSharedOntologyIntoBundleOntology(input, output);
  45         }
  46
  47         public static void createTGAndPGraph(Path input) throws Exception {
  48         String newName = input.getFileName().toString();
  49         if (newName.contains("."))
  50             newName = newName.split("\\.")[0];
  51         Path parent = input.getParent();
  52         Path output1 = parent.resolve(newName + ".tg");
  53         TransferableGraph1 tg = convertExportedSharedOntologyIntoBundleOntology(input, output1);
  54         String listing = PrettyPrintTG.print(tg, false);
  55         Path output2 = parent.resolve(newName + ".pgraph");
  56         Files.write(output2, listing.getBytes(),StandardOpenOption.CREATE);
  57         }
  58
  59         public static void main(String[] args) throws Exception {
  60                 if (args.length == 0) {
  61                         System.out.println("Required arguments: <input .sharedLibrary file> [<output .tg file>]");
  62                 } else if (args.length == 1) {
  63                         Path input = Paths.get(args[0]);
  64                         createTGAndPGraph(input);
  65                 } else {
  66                         convertExportedSharedOntologyIntoBundleOntology(Paths.get(args[0]), Paths.get(args[1]));
  67                 }
  68         }
  69
  70 }