X-Git-Url: https://gerrit.simantics.org/r/gitweb?p=simantics%2Fplatform.git;a=blobdiff_plain;f=bundles%2Forg.simantics.graph%2Fsrc%2Forg%2Fsimantics%2Fgraph%2Frefactoring%2FFixExportedOntology.java;h=33291cdf108e75cef32b35e64e69b19d8583c589;hp=09931e604bf6fdddf9ab751f20e0f910090210c6;hb=fac3394a0015e8f1b642bb59f97c814ea5d7ff1f;hpb=2a46c55805427cfb40ed0db7e33208bb5ffb3d7a

diff --git a/bundles/org.simantics.graph/src/org/simantics/graph/refactoring/FixExportedOntology.java b/bundles/org.simantics.graph/src/org/simantics/graph/refactoring/FixExportedOntology.java
index 09931e604..33291cdf1 100644
--- a/bundles/org.simantics.graph/src/org/simantics/graph/refactoring/FixExportedOntology.java
+++ b/bundles/org.simantics.graph/src/org/simantics/graph/refactoring/FixExportedOntology.java
@@ -1,51 +1,81 @@
-package org.simantics.graph.refactoring;
-
-import java.io.BufferedInputStream;
-import java.io.DataInput;
-import java.io.DataInputStream;
-import java.io.InputStream;
-import java.nio.file.Files;
-import java.nio.file.Path;
-import java.nio.file.Paths;
-
-import org.simantics.databoard.binding.Binding;
-import org.simantics.databoard.binding.mutable.Variant;
-import org.simantics.databoard.container.DataContainer;
-import org.simantics.databoard.container.DataContainers;
-import org.simantics.graph.representation.TransferableGraph1;
-
-/**
- * @author Antti Villberg
- * @since 1.24.0
- */
-public class FixExportedOntology {
-	
-	static void convertExportedSharedOntologyIntoBundleOntology(Path input, Path output) throws Exception {
-		System.out.format("Converting exported shared ontology%n\t" + input.toString() + "%nto bundle-compatible ontology%n\t" + output.toString());
-		try (InputStream is = new BufferedInputStream(Files.newInputStream(input), 128*1024)) {
-			DataInput dis = new DataInputStream(is);
-			org.simantics.databoard.container.DataContainer container = 
-					DataContainers.readFile(dis); 
-			Binding binding = TransferableGraph1.BINDING;
-			TransferableGraph1 graph = (TransferableGraph1)container.content.getValue(binding);
-			GraphRefactoringUtils.fixOntologyExport(graph);
-
-			DataContainers.writeFile(output.toFile(), new DataContainer(
-					container.format, container.version,
-					container.metadata, new Variant(TransferableGraph1.BINDING, graph)));
-		}
-	}
-
-	public static void main(String[] args) throws Exception {
-		if (args.length < 1) {
-			System.out.println("Required arguments: <input .sharedOntology file> [<output .tg file>]");
-		} else if (args.length < 2) {
-			Path input = Paths.get(args[0]);
-			Path output = input.getParent().resolve(input.getName(input.getNameCount()-1) + ".fixed");
-			convertExportedSharedOntologyIntoBundleOntology(input, output);
-		} else {
-			convertExportedSharedOntologyIntoBundleOntology(Paths.get(args[0]), Paths.get(args[1]));
-		}
-	}
-
-}
+package org.simantics.graph.refactoring;
+
+import java.io.BufferedInputStream;
+import java.io.DataInputStream;
+import java.io.InputStream;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.nio.file.StandardOpenOption;
+
+import org.simantics.databoard.binding.Binding;
+import org.simantics.databoard.container.DataContainer;
+import org.simantics.databoard.container.DataContainers;
+import org.simantics.graph.representation.PrettyPrintTG;
+import org.simantics.graph.representation.TransferableGraph1;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * @author Antti Villberg
+ * @since 1.24.0
+ */
+public class FixExportedOntology {
+
+	private static final Logger LOGGER = LoggerFactory.getLogger(FixExportedOntology.class);
+
+	static TransferableGraph1 convertExportedSharedOntologyIntoBundleOntology(Path input, Path output) throws Exception {
+		LOGGER.info("Converting exported shared ontology\n\t{}\nto bundle-compatible ontology\n\t{}", input.toString(), output.toString());
+		try (InputStream is = new BufferedInputStream(Files.newInputStream(input), 128*1024)) {
+			Binding tgBinding = TransferableGraph1.BINDING;
+			DataContainer container = DataContainers.readFile(new DataInputStream(is), tgBinding); 
+			TransferableGraph1 graph = (TransferableGraph1) container.content.getValue(tgBinding);
+
+			GraphRefactoringUtils.fixOntologyExport(graph);
+			container = TransferableGraphHasher.addHashToTG(container, graph);
+
+			DataContainers.writeFile(output.toFile(), container);
+			return graph;
+		}
+	}
+
+	private static Path replaceExtension(Path p, String newExtension) {
+		String newName = p.getFileName().toString();
+		int lastDot = newName.lastIndexOf('.');
+		if (lastDot > -1)
+			newName = newName.substring(0, lastDot);
+		return p.resolveSibling(newName + newExtension);
+	}
+
+	private static void createTg(Path input, Path output) throws Exception {
+		convertExportedSharedOntologyIntoBundleOntology(input, output);
+	}
+
+	public static void createTg(Path input) throws Exception {
+		createTg(input, replaceExtension(input, ".tg"));
+	}
+
+	public static void createTGAndPGraph(Path input) throws Exception {
+		createTGAndPGraph(input, true);
+	}
+
+	public static void createTGAndPGraph(Path input, boolean writePGraph) throws Exception {
+		TransferableGraph1 tg = convertExportedSharedOntologyIntoBundleOntology(input, replaceExtension(input, ".tg"));
+		if (writePGraph) {
+			String listing = PrettyPrintTG.print(tg, false);
+			Files.write(replaceExtension(input, ".pgraph"), listing.getBytes(),StandardOpenOption.CREATE);
+		}
+	}
+
+	public static void main(String[] args) throws Exception {
+		if (args.length == 0) {
+			System.out.println("Required arguments: <input .sharedLibrary file> [<output .tg file>]");
+		} else if (args.length == 1) {
+			Path input = Paths.get(args[0]);
+			createTGAndPGraph(input);
+		} else {
+			convertExportedSharedOntologyIntoBundleOntology(Paths.get(args[0]), Paths.get(args[1]));
+		}
+	}
+
+}