public static void main(String[] args) {
Model model = Graph_Test.getExampleModel();
TrackScheme trackScheme = new TrackScheme(model, new SelectionModel(model));
trackScheme.render();
}
/** Saves XML+PNG format. */
protected void saveXmlPng(TrackSchemeFrame frame, String filename, Color bg) throws IOException {
final mxGraphComponent graphComponent = trackScheme.getGUI().graphComponent;
final mxGraph graph = trackScheme.getGraph();
// Creates the image for the PNG file
BufferedImage image =
mxCellRenderer.createBufferedImage(
graph, null, 1, bg, graphComponent.isAntiAlias(), null, graphComponent.getCanvas());
// Creates the URL-encoded XML data
mxCodec codec = new mxCodec();
String xml = URLEncoder.encode(mxXmlUtils.getXml(codec.encode(graph.getModel())), "UTF-8");
mxPngEncodeParam param = mxPngEncodeParam.getDefaultEncodeParam(image);
param.setCompressedText(new String[] {"mxGraphModel", xml});
// Saves as a PNG file
FileOutputStream outputStream = new FileOutputStream(new File(filename));
try {
mxPngImageEncoder encoder = new mxPngImageEncoder(outputStream, param);
if (image != null) {
encoder.encode(image);
} else {
JOptionPane.showMessageDialog(graphComponent, "No Image Data");
}
} finally {
outputStream.close();
}
}
private void exportGraphToPdf(mxGraph graph, String filename) {
Rectangle bounds =
new Rectangle(trackScheme.getGUI().graphComponent.getViewport().getViewSize());
// step 1
com.itextpdf.text.Rectangle pageSize =
new com.itextpdf.text.Rectangle(bounds.x, bounds.y, bounds.width, bounds.height);
com.itextpdf.text.Document document = new com.itextpdf.text.Document(pageSize);
// step 2
PdfWriter writer = null;
Graphics2D g2 = null;
try {
writer = PdfWriter.getInstance(document, new FileOutputStream(filename));
// step 3
document.open();
// step 4
PdfContentByte canvas = writer.getDirectContent();
g2 = canvas.createGraphics(pageSize.getWidth(), pageSize.getHeight());
trackScheme.getGUI().graphComponent.getViewport().paintComponents(g2);
} catch (FileNotFoundException e) {
e.printStackTrace();
} catch (DocumentException e) {
e.printStackTrace();
} finally {
g2.dispose();
document.close();
}
}
public void actionPerformed(ActionEvent e) {
final mxGraphComponent graphComponent = trackScheme.getGUI().graphComponent;
final mxGraph graph = trackScheme.getGraph();
FileFilter selectedFilter = null;
DefaultFileFilter xmlPngFilter = new DefaultFileFilter(".png", "PNG+XML file (.png)");
FileFilter vmlFileFilter = new DefaultFileFilter(".html", "VML file (.html)");
String filename = null;
boolean dialogShown = false;
String wd;
if (lastDir != null) {
wd = lastDir;
} else {
wd = System.getProperty("user.dir");
}
JFileChooser fc = new JFileChooser(wd);
// Adds the default file format
FileFilter defaultFilter = xmlPngFilter;
fc.addChoosableFileFilter(defaultFilter);
// Adds special vector graphics formats and HTML
fc.addChoosableFileFilter(new DefaultFileFilter(".pdf", "PDF file (.pdf)"));
fc.addChoosableFileFilter(new DefaultFileFilter(".svg", "SVG file (.svg)"));
fc.addChoosableFileFilter(new DefaultFileFilter(".html", "HTML file (.html)"));
fc.addChoosableFileFilter(vmlFileFilter);
fc.addChoosableFileFilter(new DefaultFileFilter(".txt", "Graph Drawing file (.txt)"));
fc.addChoosableFileFilter(new DefaultFileFilter(".mxe", "mxGraph Editor file (.mxe)"));
// Adds a filter for each supported image format
Object[] imageFormats = ImageIO.getReaderFormatNames();
// Finds all distinct extensions
HashSet<String> formats = new HashSet<String>();
for (int i = 0; i < imageFormats.length; i++) {
String ext = imageFormats[i].toString().toLowerCase();
formats.add(ext);
}
imageFormats = formats.toArray();
for (int i = 0; i < imageFormats.length; i++) {
String ext = imageFormats[i].toString();
fc.addChoosableFileFilter(
new DefaultFileFilter("." + ext, ext.toUpperCase() + " File (." + ext + ")"));
}
// Adds filter that accepts all supported image formats
fc.addChoosableFileFilter(new DefaultFileFilter.ImageFileFilter("All Images"));
fc.setFileFilter(defaultFilter);
int rc = fc.showDialog(null, "Save");
dialogShown = true;
if (rc != JFileChooser.APPROVE_OPTION) {
return;
} else {
lastDir = fc.getSelectedFile().getParent();
}
filename = fc.getSelectedFile().getAbsolutePath();
selectedFilter = fc.getFileFilter();
if (selectedFilter instanceof DefaultFileFilter) {
String ext = ((DefaultFileFilter) selectedFilter).getExtension();
if (!filename.toLowerCase().endsWith(ext)) {
filename += ext;
}
}
if (new File(filename).exists()
&& JOptionPane.showConfirmDialog(graphComponent, "Overwrite existing file?")
!= JOptionPane.YES_OPTION) {
return;
}
try {
String ext = filename.substring(filename.lastIndexOf('.') + 1);
if (ext.equalsIgnoreCase("svg")) {
mxSvgCanvas canvas =
(mxSvgCanvas)
mxCellRenderer.drawCells(
graph,
null,
1,
null,
new CanvasFactory() {
public mxICanvas createCanvas(int width, int height) {
TrackSchemeSvgCanvas canvas =
new TrackSchemeSvgCanvas(mxDomUtils.createSvgDocument(width, height));
canvas.setEmbedded(true);
return canvas;
}
});
mxUtils.writeFile(mxXmlUtils.getXml(canvas.getDocument()), filename);
} else if (selectedFilter == vmlFileFilter) {
mxUtils.writeFile(
mxXmlUtils.getXml(
mxCellRenderer.createVmlDocument(graph, null, 1, null, null).getDocumentElement()),
filename);
} else if (ext.equalsIgnoreCase("html")) {
mxUtils.writeFile(
mxXmlUtils.getXml(
mxCellRenderer.createHtmlDocument(graph, null, 1, null, null).getDocumentElement()),
filename);
} else if (ext.equalsIgnoreCase("mxe") || ext.equalsIgnoreCase("xml")) {
mxCodec codec = new mxCodec();
String xml = mxXmlUtils.getXml(codec.encode(graph.getModel()));
mxUtils.writeFile(xml, filename);
} else if (ext.equalsIgnoreCase("txt")) {
String content = mxGdCodec.encode(graph); // .getDocumentString();
mxUtils.writeFile(content, filename);
} else if (ext.equalsIgnoreCase("pdf")) {
exportGraphToPdf(graph, filename);
} else {
Color bg = null;
if ((!ext.equalsIgnoreCase("gif") && !ext.equalsIgnoreCase("png"))
|| JOptionPane.showConfirmDialog(graphComponent, "Transparent Background?")
!= JOptionPane.YES_OPTION) {
bg = graphComponent.getBackground();
}
if (selectedFilter == xmlPngFilter || (ext.equalsIgnoreCase("png") && !dialogShown)) {
saveXmlPng(trackScheme.getGUI(), filename, bg);
} else {
BufferedImage image =
mxCellRenderer.createBufferedImage(
graph,
null,
1,
bg,
graphComponent.isAntiAlias(),
null,
graphComponent.getCanvas());
if (image != null) {
ImageIO.write(image, ext, new File(filename));
} else {
JOptionPane.showMessageDialog(graphComponent, "No Image Data");
}
}
}
} catch (Throwable ex) {
ex.printStackTrace();
JOptionPane.showMessageDialog(
graphComponent, ex.toString(), "Error", JOptionPane.ERROR_MESSAGE);
}
}