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=90738f70d1c8243366659ff2901ce315247eafc9;hp=1c567b1e21d39294f00652a0e8467e33cf803044;hb=e0deb46135b4d11a6c8adf1d11189ca856a2ebb0;hpb=1941a8f086ccdc3017c84dd149418114a499aee4

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 1c567b1e2..90738f70d 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
@@ -40,17 +40,33 @@ public class FixExportedOntology {
 			
 		}
 	}
+	
+	public static void createTg(Path input) throws Exception {
+	    Path output = input.getParent().resolve("graph.tg");
+	    createTg(input, output);
+	}
+	
+	private static void createTg(Path input, Path output) throws Exception {
+	    convertExportedSharedOntologyIntoBundleOntology(input, output);
+	}
+	
+	public static void createTGAndPGraph(Path input) throws Exception {
+        Path output1 = input.getParent().resolve("graph.tg");
+        TransferableGraph1 tg = convertExportedSharedOntologyIntoBundleOntology(input, output1);
+        String listing = PrettyPrintTG.print(tg, false);
+        String newName = input.getFileName().toString();
+        if (newName.contains("."))
+            newName = newName.split("\\.")[0];
+        Path output2 = input.getParent().resolve(newName + ".pgraph");
+        Files.write(output2, 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]);
-			Path output1 = input.getParent().resolve("graph.tg");
-			TransferableGraph1 tg = convertExportedSharedOntologyIntoBundleOntology(input, output1);
-			String listing = PrettyPrintTG.print(tg, false);
-			Path output2 = Paths.get(args[0].substring(0, args[0].length() - ".sharedLibrary".length()) + ".pgraph");
-			Files.write(output2, listing.getBytes(),StandardOpenOption.CREATE);
+			createTGAndPGraph(input);
 		} else {
 			convertExportedSharedOntologyIntoBundleOntology(Paths.get(args[0]), Paths.get(args[1]));
 		}