bp-scopviz/scopviz/src/main/java/de/tu_darmstadt/informatik/tk/scopviz/io/GraphMLExporter.java

package de.tu_darmstadt.informatik.tk.scopviz.io;

import java.io.FileOutputStream;
import java.io.IOException;

import de.tu_darmstadt.informatik.tk.scopviz.graphs.MyGraph;
import de.tu_darmstadt.informatik.tk.scopviz.main.Main;
import javafx.stage.FileChooser;
import javafx.stage.FileChooser.ExtensionFilter;
import javafx.stage.Stage;

/**
 * Exporter to write a given Graph object to a GraphML file on disk.
 * 
 * @author Jascha Bohne
 * @version 1.0
 * 
 */
public class GraphMLExporter {

	/**
	 * Exports the current state of the Graph to a GraphML file.
	 * 
	 * @param g
	 *            The Graphstream-Graph to be exported
	 * @param fileName
	 *            The Location on disk the File will be saved on
	 */
	public void writeGraph(final MyGraph g, final String fileName, boolean exportAsSingleGraph) {
		MyFileSinkGraphML writer = new MyFileSinkGraphML();
		String newFileName = fileName;
		if (g.isComposite() && !exportAsSingleGraph) {
			writer.exportGraphs(g.getAllSubGraphs(), fileName);
			return;
		}
		try {
			writer.writeAll(g, new FileOutputStream(newFileName));
		} catch (IOException e) {
			System.out.println("cannot Acces File or invalid path");
			e.printStackTrace();
		}
	}

	/**
	 * Exports the current state of the Graph to a GraphML file. Opens a
	 * FileSaveDialog
	 * 
	 * @param g
	 *            The Graphstream-Graph to be exported
	 * @param stage
	 *            The parent window of the save Window
	 */
	public void writeGraph(final MyGraph g, final Stage stage) {
		String fileName;
		FileChooser fileChooser = new FileChooser();
		fileChooser.setTitle("Saving graph");
		fileChooser.setInitialFileName("*.graphml");
		ExtensionFilter standard = new ExtensionFilter("GraphML Files", "*.graphml");
		fileChooser.getExtensionFilters().add(standard);
		fileChooser.getExtensionFilters().add(new ExtensionFilter("all Files", "*"));
		fileChooser.setSelectedExtensionFilter(standard);
		try {
			fileName = fileChooser.showSaveDialog(stage).getPath();
			Main.getInstance().getGraphManager().setCurrentPath(fileName);
			if (fileName != null) {
				writeGraph(g, fileName, false);
			}
		} catch (NullPointerException e) {

		}
	}

	/**
	 * Appends a string to the fileName before the fileExtension
	 * 
	 * @param fileName
	 *            the fileName
	 * @param append
	 *            the string that will be appended
	 */
	public String fileNameAppend(String fileName, String append) {
		String[] parts = fileName.split(".");
		if (parts.length < 2) {
			fileName = fileName.concat(append);
		} else {
			fileName = "";
			int i = 0;
			for (; i < parts.length - 1; i++) {
				fileName = fileName.concat(parts[0]);
			}
			fileName.concat(append);
			fileName.concat(parts[i]);
		}

		return fileName;
	}


	public void exportMapping(MyGraph g){
		Stage stage = Main.getInstance().getPrimaryStage();
		String fileName;
		FileChooser fileChooser = new FileChooser();
		fileChooser.setTitle("Saving graph");
		fileChooser.setInitialFileName("*.graphmlMap");
		ExtensionFilter standard = new ExtensionFilter("GraphML Mapping underlay Files", "*.graphmlMap");
		fileChooser.getExtensionFilters().add(standard);
		fileChooser.getExtensionFilters().add(new ExtensionFilter("all Files", "*.*"));
		fileChooser.setSelectedExtensionFilter(standard);
		try {
			fileName = fileChooser.showSaveDialog(stage).getPath();
			Main.getInstance().getGraphManager().setCurrentPath(fileName);
			if (fileName != null) {
				writeGraph(g, fileName, false);
			}
		} catch (NullPointerException e) {}
	}
}